isl_printer_print_qpolynomial: properly print nested divs
[platform/upstream/isl.git] / isl_map.c
index 6b934f6..93ffa58 100644 (file)
--- a/isl_map.c
+++ b/isl_map.c
@@ -121,6 +121,7 @@ unsigned isl_basic_map_dim(const struct isl_basic_map *bmap,
        if (!bmap)
                return 0;
        switch (type) {
+       case isl_dim_cst:       return 1;
        case isl_dim_param:
        case isl_dim_in:
        case isl_dim_out:       return isl_dim_size(bmap->dim, type);
@@ -145,6 +146,7 @@ unsigned isl_basic_map_offset(struct isl_basic_map *bmap,
 {
        struct isl_dim *dim = bmap->dim;
        switch (type) {
+       case isl_dim_cst:       return 0;
        case isl_dim_param:     return 1;
        case isl_dim_in:        return 1 + dim->nparam;
        case isl_dim_out:       return 1 + dim->nparam + dim->n_in;
@@ -327,6 +329,12 @@ error:
        return NULL;
 }
 
+const char *isl_basic_map_get_tuple_name(__isl_keep isl_basic_map *bmap,
+       enum isl_dim_type type)
+{
+       return bmap ? isl_dim_get_tuple_name(bmap->dim, type) : NULL;
+}
+
 __isl_give isl_map *isl_map_set_tuple_name(__isl_take isl_map *map,
        enum isl_dim_type type, const char *s)
 {
@@ -352,6 +360,41 @@ error:
        return NULL;
 }
 
+const char *isl_map_get_tuple_name(__isl_keep isl_map *map,
+       enum isl_dim_type type)
+{
+       return map ? isl_dim_get_tuple_name(map->dim, type) : NULL;
+}
+
+__isl_give isl_set *isl_set_set_tuple_name(__isl_take isl_set *set,
+       const char *s)
+{
+       return (isl_set *)isl_map_set_tuple_name((isl_map *)set, isl_dim_set, s);
+}
+
+const char *isl_set_get_tuple_name(__isl_keep isl_set *set)
+{
+       return set ? isl_dim_get_tuple_name(set->dim, isl_dim_set) : NULL;
+}
+
+const char *isl_basic_map_get_dim_name(__isl_keep isl_basic_map *bmap,
+       enum isl_dim_type type, unsigned pos)
+{
+       return bmap ? isl_dim_get_name(bmap->dim, type, pos) : NULL;
+}
+
+const char *isl_map_get_dim_name(__isl_keep isl_map *map,
+       enum isl_dim_type type, unsigned pos)
+{
+       return map ? isl_dim_get_name(map->dim, type, pos) : NULL;
+}
+
+const char *isl_set_get_dim_name(__isl_keep isl_set *set,
+       enum isl_dim_type type, unsigned pos)
+{
+       return set ? isl_dim_get_name(set->dim, type, pos) : NULL;
+}
+
 __isl_give isl_basic_map *isl_basic_map_set_dim_name(
        __isl_take isl_basic_map *bmap,
        enum isl_dim_type type, unsigned pos, const char *s)
@@ -480,9 +523,12 @@ struct isl_basic_set *isl_basic_set_alloc_dim(struct isl_dim *dim,
        struct isl_basic_map *bmap;
        if (!dim)
                return NULL;
-       isl_assert(dim->ctx, dim->n_in == 0, return NULL);
+       isl_assert(dim->ctx, dim->n_in == 0, goto error);
        bmap = isl_basic_map_alloc_dim(dim, extra, n_eq, n_ineq);
        return (struct isl_basic_set *)bmap;
+error:
+       isl_dim_free(dim);
+       return NULL;
 }
 
 struct isl_basic_map *isl_basic_map_alloc_dim(struct isl_dim *dim,
@@ -1236,7 +1282,8 @@ struct isl_basic_set *isl_basic_set_swap_vars(
        isl_blk_free(bset->ctx, blk);
 
        ISL_F_CLR(bset, ISL_BASIC_SET_NORMALIZED);
-       return isl_basic_set_gauss(bset, NULL);
+       bset = isl_basic_set_gauss(bset, NULL);
+       return isl_basic_set_finalize(bset);
 error:
        isl_basic_set_free(bset);
        return NULL;
@@ -1363,23 +1410,22 @@ __isl_give isl_set *isl_set_eliminate_dims(__isl_take isl_set *set,
        return isl_set_eliminate(set, isl_dim_set, first, n);
 }
 
-/* Project out n dimensions starting at first using Fourier-Motzkin */
-struct isl_set *isl_set_remove_dims(struct isl_set *set,
-       unsigned first, unsigned n)
+__isl_give isl_basic_map *isl_basic_map_remove_divs(
+       __isl_take isl_basic_map *bmap)
 {
-       set = isl_set_eliminate_dims(set, first, n);
-       set = isl_set_drop_dims(set, first, n);
-       return set;
+       bmap = isl_basic_map_eliminate_vars(bmap, isl_dim_total(bmap->dim),
+                                               bmap->n_div);
+       if (!bmap)
+               return NULL;
+       bmap->n_div = 0;
+       return bmap;
 }
 
-struct isl_basic_set *isl_basic_set_remove_divs(struct isl_basic_set *bset)
+__isl_give isl_basic_set *isl_basic_set_remove_divs(
+       __isl_take isl_basic_set *bset)
 {
-       bset = isl_basic_set_eliminate_vars(bset, isl_dim_total(bset->dim),
-                                               bset->n_div);
-       if (!bset)
-               return NULL;
-       bset->n_div = 0;
-       return bset;
+       return (struct isl_basic_set *)isl_basic_map_remove_divs(
+                       (struct isl_basic_map *)bset);
 }
 
 struct isl_set *isl_set_remove_divs(struct isl_set *set)
@@ -1406,7 +1452,7 @@ error:
        return NULL;
 }
 
-struct isl_basic_map *isl_basic_map_remove(struct isl_basic_map *bmap,
+struct isl_basic_map *isl_basic_map_remove_dims(struct isl_basic_map *bmap,
        enum isl_dim_type type, unsigned first, unsigned n)
 {
        if (!bmap)
@@ -1428,14 +1474,99 @@ error:
        return NULL;
 }
 
-__isl_give isl_basic_set *isl_basic_set_remove(__isl_take isl_basic_set *bset,
+/* Return true if the definition of the given div (recursively) involves
+ * any of the given variables.
+ */
+static int div_involves_vars(__isl_keep isl_basic_map *bmap, int div,
+       unsigned first, unsigned n)
+{
+       int i;
+       unsigned div_offset = isl_basic_map_offset(bmap, isl_dim_div);
+
+       if (isl_int_is_zero(bmap->div[div][0]))
+               return 0;
+       if (isl_seq_first_non_zero(bmap->div[div] + 1 + first, n) >= 0)
+               return 1;
+
+       for (i = bmap->n_div - 1; i >= 0; --i) {
+               if (isl_int_is_zero(bmap->div[div][1 + div_offset + i]))
+                       continue;
+               if (div_involves_vars(bmap, i, first, n))
+                       return 1;
+       }
+
+       return 0;
+}
+
+/* Remove all divs (recursively) involving any of the given dimensions
+ * in their definitions.
+ */
+__isl_give isl_basic_map *isl_basic_map_remove_divs_involving_dims(
+       __isl_take isl_basic_map *bmap,
+       enum isl_dim_type type, unsigned first, unsigned n)
+{
+       int i;
+
+       if (!bmap)
+               return NULL;
+       isl_assert(bmap->ctx, first + n <= isl_basic_map_dim(bmap, type),
+                       goto error);
+       first += isl_basic_map_offset(bmap, type);
+
+       for (i = bmap->n_div - 1; i >= 0; --i) {
+               if (!div_involves_vars(bmap, i, first, n))
+                       continue;
+               bmap = isl_basic_map_remove_dims(bmap, isl_dim_div, i, 1);
+       }
+
+       return bmap;
+error:
+       isl_basic_map_free(bmap);
+       return NULL;
+}
+
+__isl_give isl_map *isl_map_remove_divs_involving_dims(__isl_take isl_map *map,
+       enum isl_dim_type type, unsigned first, unsigned n)
+{
+       int i;
+
+       if (!map)
+               return NULL;
+       if (map->n == 0)
+               return map;
+
+       map = isl_map_cow(map);
+       if (!map)
+               return NULL;
+
+       for (i = 0; i < map->n; ++i) {
+               map->p[i] = isl_basic_map_remove_divs_involving_dims(map->p[i],
+                                                               type, first, n);
+               if (!map->p[i])
+                       goto error;
+       }
+       return map;
+error:
+       isl_map_free(map);
+       return NULL;
+}
+
+__isl_give isl_set *isl_set_remove_divs_involving_dims(__isl_take isl_set *set,
+       enum isl_dim_type type, unsigned first, unsigned n)
+{
+       return (isl_set *)isl_map_remove_divs_involving_dims((isl_map *)set,
+                                                             type, first, n);
+}
+
+__isl_give isl_basic_set *isl_basic_set_remove_dims(
+       __isl_take isl_basic_set *bset,
        enum isl_dim_type type, unsigned first, unsigned n)
 {
        return (isl_basic_set *)
-               isl_basic_map_remove((isl_basic_map *)bset, type, first, n);
+           isl_basic_map_remove_dims((isl_basic_map *)bset, type, first, n);
 }
 
-struct isl_map *isl_map_remove(struct isl_map *map,
+struct isl_map *isl_map_remove_dims(struct isl_map *map,
        enum isl_dim_type type, unsigned first, unsigned n)
 {
        int i;
@@ -1461,27 +1592,17 @@ error:
        return NULL;
 }
 
-__isl_give isl_set *isl_set_remove(__isl_take isl_set *bset,
+__isl_give isl_set *isl_set_remove_dims(__isl_take isl_set *bset,
        enum isl_dim_type type, unsigned first, unsigned n)
 {
-       return (isl_set *)isl_map_remove((isl_map *)bset, type, first, n);
+       return (isl_set *)isl_map_remove_dims((isl_map *)bset, type, first, n);
 }
 
 /* Project out n inputs starting at first using Fourier-Motzkin */
 struct isl_map *isl_map_remove_inputs(struct isl_map *map,
        unsigned first, unsigned n)
 {
-       return isl_map_remove(map, isl_dim_in, first, n);
-}
-
-/* Project out n dimensions starting at first using Fourier-Motzkin */
-struct isl_basic_set *isl_basic_set_remove_dims(struct isl_basic_set *bset,
-       unsigned first, unsigned n)
-{
-       unsigned nparam = isl_basic_set_n_param(bset);
-       bset = isl_basic_set_eliminate_vars(bset, nparam + first, n);
-       bset = isl_basic_set_drop_dims(bset, first, n);
-       return bset;
+       return isl_map_remove_dims(map, isl_dim_in, first, n);
 }
 
 static void dump_term(struct isl_basic_map *bmap,
@@ -1649,8 +1770,8 @@ struct isl_set *isl_set_alloc_dim(struct isl_dim *dim, int n, unsigned flags)
 
        if (!dim)
                return NULL;
-       isl_assert(dim->ctx, dim->n_in == 0, return NULL);
-       isl_assert(dim->ctx, n >= 0, return NULL);
+       isl_assert(dim->ctx, dim->n_in == 0, goto error);
+       isl_assert(dim->ctx, n >= 0, goto error);
        set = isl_alloc(dim->ctx, struct isl_set,
                        sizeof(struct isl_set) +
                        (n - 1) * sizeof(struct isl_basic_set *));
@@ -2237,7 +2358,7 @@ __isl_give isl_set *isl_set_insert(__isl_take isl_set *set,
        return (isl_set *)isl_map_insert((isl_map *)set, type, pos, n);
 }
 
-__isl_give isl_map *isl_map_add(__isl_take isl_map *map,
+__isl_give isl_map *isl_map_add_dims(__isl_take isl_map *map,
                enum isl_dim_type type, unsigned n)
 {
        if (!map)
@@ -2245,13 +2366,13 @@ __isl_give isl_map *isl_map_add(__isl_take isl_map *map,
        return isl_map_insert(map, type, isl_map_dim(map, type), n);
 }
 
-__isl_give isl_set *isl_set_add(__isl_take isl_set *set,
+__isl_give isl_set *isl_set_add_dims(__isl_take isl_set *set,
                enum isl_dim_type type, unsigned n)
 {
        if (!set)
                return NULL;
        isl_assert(set->ctx, type != isl_dim_in, goto error);
-       return (isl_set *)isl_map_add((isl_map *)set, type, n);
+       return (isl_set *)isl_map_add_dims((isl_map *)set, type, n);
 error:
        isl_set_free(set);
        return NULL;
@@ -2474,7 +2595,7 @@ __isl_give isl_basic_map *isl_basic_map_project_out(
                return NULL;
 
        if (ISL_F_ISSET(bmap, ISL_BASIC_MAP_RATIONAL))
-               return isl_basic_map_remove(bmap, type, first, n);
+               return isl_basic_map_remove_dims(bmap, type, first, n);
 
        isl_assert(bmap->ctx, first + n <= isl_basic_map_dim(bmap, type),
                        goto error);
@@ -3201,6 +3322,86 @@ __isl_give isl_map *isl_map_lex_ge(__isl_take isl_dim *set_dim)
        return map_lex_gte(isl_dim_map(set_dim), 1);
 }
 
+__isl_give isl_map *isl_set_lex_le_set(__isl_take isl_set *set1,
+       __isl_take isl_set *set2)
+{
+       isl_map *map;
+       map = isl_map_lex_le(isl_set_get_dim(set1));
+       map = isl_map_intersect_domain(map, set1);
+       map = isl_map_intersect_range(map, set2);
+       return map;
+}
+
+__isl_give isl_map *isl_set_lex_lt_set(__isl_take isl_set *set1,
+       __isl_take isl_set *set2)
+{
+       isl_map *map;
+       map = isl_map_lex_lt(isl_set_get_dim(set1));
+       map = isl_map_intersect_domain(map, set1);
+       map = isl_map_intersect_range(map, set2);
+       return map;
+}
+
+__isl_give isl_map *isl_set_lex_ge_set(__isl_take isl_set *set1,
+       __isl_take isl_set *set2)
+{
+       isl_map *map;
+       map = isl_map_lex_ge(isl_set_get_dim(set1));
+       map = isl_map_intersect_domain(map, set1);
+       map = isl_map_intersect_range(map, set2);
+       return map;
+}
+
+__isl_give isl_map *isl_set_lex_gt_set(__isl_take isl_set *set1,
+       __isl_take isl_set *set2)
+{
+       isl_map *map;
+       map = isl_map_lex_gt(isl_set_get_dim(set1));
+       map = isl_map_intersect_domain(map, set1);
+       map = isl_map_intersect_range(map, set2);
+       return map;
+}
+
+__isl_give isl_map *isl_map_lex_le_map(__isl_take isl_map *map1,
+       __isl_take isl_map *map2)
+{
+       isl_map *map;
+       map = isl_map_lex_le(isl_dim_range(isl_map_get_dim(map1)));
+       map = isl_map_apply_domain(map, isl_map_reverse(map1));
+       map = isl_map_apply_range(map, isl_map_reverse(map2));
+       return map;
+}
+
+__isl_give isl_map *isl_map_lex_lt_map(__isl_take isl_map *map1,
+       __isl_take isl_map *map2)
+{
+       isl_map *map;
+       map = isl_map_lex_lt(isl_dim_range(isl_map_get_dim(map1)));
+       map = isl_map_apply_domain(map, isl_map_reverse(map1));
+       map = isl_map_apply_range(map, isl_map_reverse(map2));
+       return map;
+}
+
+__isl_give isl_map *isl_map_lex_ge_map(__isl_take isl_map *map1,
+       __isl_take isl_map *map2)
+{
+       isl_map *map;
+       map = isl_map_lex_ge(isl_dim_range(isl_map_get_dim(map1)));
+       map = isl_map_apply_domain(map, isl_map_reverse(map1));
+       map = isl_map_apply_range(map, isl_map_reverse(map2));
+       return map;
+}
+
+__isl_give isl_map *isl_map_lex_gt_map(__isl_take isl_map *map1,
+       __isl_take isl_map *map2)
+{
+       isl_map *map;
+       map = isl_map_lex_gt(isl_dim_range(isl_map_get_dim(map1)));
+       map = isl_map_apply_domain(map, isl_map_reverse(map1));
+       map = isl_map_apply_range(map, isl_map_reverse(map2));
+       return map;
+}
+
 struct isl_basic_map *isl_basic_map_from_basic_set(
                struct isl_basic_set *bset, struct isl_dim *dim)
 {
@@ -3292,7 +3493,9 @@ struct isl_basic_set *isl_basic_map_underlying_set(
        if (!bmap)
                goto error;
        if (bmap->dim->nparam == 0 && bmap->dim->n_in == 0 &&
-           bmap->n_div == 0 && !isl_dim_get_tuple_name(bmap->dim, isl_dim_out))
+           bmap->n_div == 0 &&
+           !isl_dim_is_named_or_nested(bmap->dim, isl_dim_in) &&
+           !isl_dim_is_named_or_nested(bmap->dim, isl_dim_out))
                return (struct isl_basic_set *)bmap;
        bmap = isl_basic_map_cow(bmap);
        if (!bmap)
@@ -3462,7 +3665,7 @@ struct isl_set *isl_set_to_underlying_set(struct isl_set *set)
        return (struct isl_set *)isl_map_underlying_set((struct isl_map *)set);
 }
 
-static __isl_give isl_basic_map *isl_basic_map_reset_dim(
+__isl_give isl_basic_map *isl_basic_map_reset_dim(
        __isl_take isl_basic_map *bmap, __isl_take isl_dim *dim)
 {
        bmap = isl_basic_map_cow(bmap);
@@ -3545,6 +3748,84 @@ struct isl_basic_set *isl_basic_map_range(struct isl_basic_map *bmap)
        return isl_basic_map_domain(isl_basic_map_reverse(bmap));
 }
 
+__isl_give isl_basic_map *isl_basic_map_domain_map(
+       __isl_take isl_basic_map *bmap)
+{
+       int i, k;
+       isl_dim *dim;
+       isl_basic_map *domain;
+       isl_basic_set *bset;
+       int nparam, n_in, n_out;
+       unsigned total;
+
+       nparam = isl_basic_map_dim(bmap, isl_dim_param);
+       n_in = isl_basic_map_dim(bmap, isl_dim_in);
+       n_out = isl_basic_map_dim(bmap, isl_dim_out);
+
+       dim = isl_dim_from_range(isl_dim_domain(isl_basic_map_get_dim(bmap)));
+       domain = isl_basic_map_universe(dim);
+
+       bmap = isl_basic_map_from_domain(isl_basic_map_wrap(bmap));
+       bmap = isl_basic_map_apply_range(bmap, domain);
+       bmap = isl_basic_map_extend_constraints(bmap, n_in, 0);
+
+       total = isl_basic_map_total_dim(bmap);
+
+       for (i = 0; i < n_in; ++i) {
+               k = isl_basic_map_alloc_equality(bmap);
+               if (k < 0)
+                       goto error;
+               isl_seq_clr(bmap->eq[k], 1 + total);
+               isl_int_set_si(bmap->eq[k][1 + nparam + i], -1);
+               isl_int_set_si(bmap->eq[k][1 + nparam + n_in + n_out + i], 1);
+       }
+
+       bmap = isl_basic_map_gauss(bmap, NULL);
+       return isl_basic_map_finalize(bmap);
+error:
+       isl_basic_map_free(bmap);
+       return NULL;
+}
+
+__isl_give isl_basic_map *isl_basic_map_range_map(
+       __isl_take isl_basic_map *bmap)
+{
+       int i, k;
+       isl_dim *dim;
+       isl_basic_map *range;
+       isl_basic_set *bset;
+       int nparam, n_in, n_out;
+       unsigned total;
+
+       nparam = isl_basic_map_dim(bmap, isl_dim_param);
+       n_in = isl_basic_map_dim(bmap, isl_dim_in);
+       n_out = isl_basic_map_dim(bmap, isl_dim_out);
+
+       dim = isl_dim_from_range(isl_dim_range(isl_basic_map_get_dim(bmap)));
+       range = isl_basic_map_universe(dim);
+
+       bmap = isl_basic_map_from_domain(isl_basic_map_wrap(bmap));
+       bmap = isl_basic_map_apply_range(bmap, range);
+       bmap = isl_basic_map_extend_constraints(bmap, n_out, 0);
+
+       total = isl_basic_map_total_dim(bmap);
+
+       for (i = 0; i < n_out; ++i) {
+               k = isl_basic_map_alloc_equality(bmap);
+               if (k < 0)
+                       goto error;
+               isl_seq_clr(bmap->eq[k], 1 + total);
+               isl_int_set_si(bmap->eq[k][1 + nparam + n_in + i], -1);
+               isl_int_set_si(bmap->eq[k][1 + nparam + n_in + n_out + i], 1);
+       }
+
+       bmap = isl_basic_map_gauss(bmap, NULL);
+       return isl_basic_map_finalize(bmap);
+error:
+       isl_basic_map_free(bmap);
+       return NULL;
+}
+
 struct isl_set *isl_map_range(struct isl_map *map)
 {
        int i;
@@ -3576,6 +3857,60 @@ error:
        return NULL;
 }
 
+__isl_give isl_map *isl_map_domain_map(__isl_take isl_map *map)
+{
+       int i;
+       isl_dim *domain_dim;
+
+       map = isl_map_cow(map);
+       if (!map)
+               return NULL;
+
+       domain_dim = isl_dim_from_range(isl_dim_domain(isl_map_get_dim(map)));
+       map->dim = isl_dim_from_domain(isl_dim_wrap(map->dim));
+       map->dim = isl_dim_join(map->dim, domain_dim);
+       if (!map->dim)
+               goto error;
+       for (i = 0; i < map->n; ++i) {
+               map->p[i] = isl_basic_map_domain_map(map->p[i]);
+               if (!map->p[i])
+                       goto error;
+       }
+       ISL_F_CLR(map, ISL_MAP_DISJOINT);
+       ISL_F_CLR(map, ISL_MAP_NORMALIZED);
+       return map;
+error:
+       isl_map_free(map);
+       return NULL;
+}
+
+__isl_give isl_map *isl_map_range_map(__isl_take isl_map *map)
+{
+       int i;
+       isl_dim *range_dim;
+
+       map = isl_map_cow(map);
+       if (!map)
+               return NULL;
+
+       range_dim = isl_dim_range(isl_map_get_dim(map));
+       map->dim = isl_dim_from_domain(isl_dim_wrap(map->dim));
+       map->dim = isl_dim_join(map->dim, range_dim);
+       if (!map->dim)
+               goto error;
+       for (i = 0; i < map->n; ++i) {
+               map->p[i] = isl_basic_map_range_map(map->p[i]);
+               if (!map->p[i])
+                       goto error;
+       }
+       ISL_F_CLR(map, ISL_MAP_DISJOINT);
+       ISL_F_CLR(map, ISL_MAP_NORMALIZED);
+       return map;
+error:
+       isl_map_free(map);
+       return NULL;
+}
+
 struct isl_map *isl_map_from_set(struct isl_set *set, struct isl_dim *dim)
 {
        int i;
@@ -3601,6 +3936,18 @@ error:
        return NULL;
 }
 
+__isl_give isl_basic_map *isl_basic_map_from_domain(
+       __isl_take isl_basic_set *bset)
+{
+       return isl_basic_map_reverse(isl_basic_map_from_range(bset));
+}
+
+__isl_give isl_basic_map *isl_basic_map_from_range(
+       __isl_take isl_basic_set *bset)
+{
+       return (isl_basic_map *)bset;
+}
+
 struct isl_map *isl_map_from_range(struct isl_set *set)
 {
        return (struct isl_map *)set;
@@ -5186,6 +5533,14 @@ struct isl_map *isl_map_identity_like_basic_map(struct isl_basic_map *model)
        return map_identity(isl_dim_copy(model->dim));
 }
 
+__isl_give isl_map *isl_set_identity(__isl_take isl_set *set)
+{
+       isl_dim *dim = isl_set_get_dim(set);
+       isl_map *id;
+       id = isl_map_identity(dim);
+       return isl_map_intersect_range(id, set);
+}
+
 /* Construct a basic set with all set dimensions having only non-negative
  * values.
  */
@@ -6643,6 +6998,11 @@ int isl_set_dim_is_unique(struct isl_set *set, unsigned dim)
        return 1;
 }
 
+int isl_set_n_basic_set(__isl_keep isl_set *set)
+{
+       return set ? set->n : 0;
+}
+
 int isl_map_foreach_basic_map(__isl_keep isl_map *map,
        int (*fn)(__isl_take isl_basic_map *bmap, void *user), void *user)
 {
@@ -7295,6 +7655,24 @@ error:
        return NULL;
 }
 
+__isl_give isl_set *isl_set_flatten(__isl_take isl_set *set)
+{
+       return (isl_set *)isl_map_flatten((isl_map *)set);
+}
+
+__isl_give isl_map *isl_set_flatten_map(__isl_take isl_set *set)
+{
+       isl_dim *dim, *flat_dim;
+       isl_map *map;
+
+       dim = isl_set_get_dim(set);
+       flat_dim = isl_dim_flatten(isl_dim_copy(dim));
+       map = map_identity(isl_dim_join(isl_dim_reverse(dim), flat_dim));
+       map = isl_map_intersect_domain(map, set);
+
+       return map;
+}
+
 /* Extend the given dim_map with mappings for the divs in bmap.
  */
 static __isl_give struct isl_dim_map *extend_dim_map(
@@ -7408,3 +7786,146 @@ __isl_give isl_set *isl_set_realign(__isl_take isl_set *set,
 {
        return (isl_set *)isl_map_realign((isl_map *)set, r);
 }
+
+__isl_give isl_mat *isl_basic_map_equalities_matrix(
+               __isl_keep isl_basic_map *bmap, enum isl_dim_type c1,
+               enum isl_dim_type c2, enum isl_dim_type c3,
+               enum isl_dim_type c4, enum isl_dim_type c5)
+{
+       enum isl_dim_type c[5] = { c1, c2, c3, c4, c5 };
+       struct isl_mat *mat;
+       int i, j, k;
+       int pos;
+
+       if (!bmap)
+               return NULL;
+       mat = isl_mat_alloc(bmap->ctx, bmap->n_eq,
+                               isl_basic_map_total_dim(bmap) + 1);
+       if (!mat)
+               return NULL;
+       for (i = 0; i < bmap->n_eq; ++i)
+               for (j = 0, pos = 0; j < 5; ++j) {
+                       int off = isl_basic_map_offset(bmap, c[j]);
+                       for (k = 0; k < isl_basic_map_dim(bmap, c[j]); ++k) {
+                               isl_int_set(mat->row[i][pos],
+                                           bmap->eq[i][off + k]);
+                               ++pos;
+                       }
+               }
+
+       return mat;
+}
+
+__isl_give isl_mat *isl_basic_map_inequalities_matrix(
+               __isl_keep isl_basic_map *bmap, enum isl_dim_type c1,
+               enum isl_dim_type c2, enum isl_dim_type c3,
+               enum isl_dim_type c4, enum isl_dim_type c5)
+{
+       enum isl_dim_type c[5] = { c1, c2, c3, c4, c5 };
+       struct isl_mat *mat;
+       int i, j, k;
+       int pos;
+
+       if (!bmap)
+               return NULL;
+       mat = isl_mat_alloc(bmap->ctx, bmap->n_ineq,
+                               isl_basic_map_total_dim(bmap) + 1);
+       if (!mat)
+               return NULL;
+       for (i = 0; i < bmap->n_ineq; ++i)
+               for (j = 0, pos = 0; j < 5; ++j) {
+                       int off = isl_basic_map_offset(bmap, c[j]);
+                       for (k = 0; k < isl_basic_map_dim(bmap, c[j]); ++k) {
+                               isl_int_set(mat->row[i][pos],
+                                           bmap->ineq[i][off + k]);
+                               ++pos;
+                       }
+               }
+
+       return mat;
+}
+
+__isl_give isl_basic_map *isl_basic_map_from_constraint_matrices(
+       __isl_take isl_dim *dim,
+       __isl_take isl_mat *eq, __isl_take isl_mat *ineq, enum isl_dim_type c1,
+       enum isl_dim_type c2, enum isl_dim_type c3,
+       enum isl_dim_type c4, enum isl_dim_type c5)
+{
+       enum isl_dim_type c[5] = { c1, c2, c3, c4, c5 };
+       isl_basic_map *bmap;
+       unsigned total;
+       unsigned extra;
+       int i, j, k, l;
+       int pos;
+
+       if (!dim || !eq || !ineq)
+               goto error;
+
+       if (eq->n_col != ineq->n_col)
+               isl_die(dim->ctx, isl_error_invalid,
+                       "equalities and inequalities matrices should have "
+                       "same number of columns", goto error);
+
+       total = 1 + isl_dim_total(dim);
+
+       if (eq->n_col < total)
+               isl_die(dim->ctx, isl_error_invalid,
+                       "number of columns too small", goto error);
+
+       extra = eq->n_col - total;
+
+       bmap = isl_basic_map_alloc_dim(isl_dim_copy(dim), extra,
+                                      eq->n_row, ineq->n_row);
+       if (!bmap)
+               goto error;
+       for (i = 0; i < extra; ++i)
+               if (isl_basic_map_alloc_div(bmap) < 0)
+                       goto error;
+       for (i = 0; i < eq->n_row; ++i) {
+               l = isl_basic_map_alloc_equality(bmap);
+               if (l < 0)
+                       goto error;
+               for (j = 0, pos = 0; j < 5; ++j) {
+                       int off = isl_basic_map_offset(bmap, c[j]);
+                       for (k = 0; k < isl_basic_map_dim(bmap, c[j]); ++k) {
+                               isl_int_set(bmap->eq[l][off + k], 
+                                           eq->row[i][pos]);
+                               ++pos;
+                       }
+               }
+       }
+       for (i = 0; i < ineq->n_row; ++i) {
+               l = isl_basic_map_alloc_inequality(bmap);
+               if (l < 0)
+                       goto error;
+               for (j = 0, pos = 0; j < 5; ++j) {
+                       int off = isl_basic_map_offset(bmap, c[j]);
+                       for (k = 0; k < isl_basic_map_dim(bmap, c[j]); ++k) {
+                               isl_int_set(bmap->ineq[l][off + k], 
+                                           ineq->row[i][pos]);
+                               ++pos;
+                       }
+               }
+       }
+
+       isl_dim_free(dim);
+       isl_mat_free(eq);
+       isl_mat_free(ineq);
+
+       return bmap;
+error:
+       isl_dim_free(dim);
+       isl_mat_free(eq);
+       isl_mat_free(ineq);
+       return NULL;
+}
+
+__isl_give isl_basic_set *isl_basic_set_from_constraint_matrices(
+       __isl_take isl_dim *dim,
+       __isl_take isl_mat *eq, __isl_take isl_mat *ineq, enum isl_dim_type c1,
+       enum isl_dim_type c2, enum isl_dim_type c3, enum isl_dim_type c4)
+{
+       return (isl_basic_set*)
+           isl_basic_map_from_constraint_matrices(dim, eq, ineq,
+                                                  c1, c2, c3, c4, isl_dim_in);
+}