add isl_map_is_translation
[platform/upstream/isl.git] / isl_map.c
index a2d33d3..9b87bd8 100644 (file)
--- a/isl_map.c
+++ b/isl_map.c
@@ -55,6 +55,7 @@ static unsigned n(struct isl_dim *dim, enum isl_dim_type type)
        case isl_dim_in:        return dim->n_in;
        case isl_dim_out:       return dim->n_out;
        case isl_dim_all:       return dim->nparam + dim->n_in + dim->n_out;
+       default:                return 0;
        }
 }
 
@@ -64,6 +65,7 @@ static unsigned pos(struct isl_dim *dim, enum isl_dim_type type)
        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;
+       default:                return 0;
        }
 }
 
@@ -120,6 +122,7 @@ unsigned isl_basic_map_dim(const struct isl_basic_map *bmap,
        case isl_dim_out:       return isl_dim_size(bmap->dim, type);
        case isl_dim_div:       return bmap->n_div;
        case isl_dim_all:       return isl_basic_map_total_dim(bmap);
+       default:                return 0;
        }
 }
 
@@ -142,6 +145,7 @@ unsigned isl_basic_map_offset(struct isl_basic_map *bmap,
        case isl_dim_in:        return 1 + dim->nparam;
        case isl_dim_out:       return 1 + dim->nparam + dim->n_in;
        case isl_dim_div:       return 1 + dim->nparam + dim->n_in + dim->n_out;
+       default:                return 0;
        }
 }
 
@@ -269,6 +273,66 @@ struct isl_dim *isl_set_get_dim(struct isl_set *set)
        return isl_dim_copy(set->dim);
 }
 
+__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)
+{
+       if (!bmap)
+               return NULL;
+       bmap->dim = isl_dim_set_name(bmap->dim, type, pos, s);
+       if (!bmap->dim)
+               goto error;
+       return bmap;
+error:
+       isl_basic_map_free(bmap);
+       return NULL;
+}
+
+__isl_give isl_map *isl_map_set_dim_name(__isl_take isl_map *map,
+       enum isl_dim_type type, unsigned pos, const char *s)
+{
+       int i;
+
+       if (!map)
+               return NULL;
+
+       map->dim = isl_dim_set_name(map->dim, type, pos, s);
+       if (!map->dim)
+               goto error;
+
+       for (i = 0; i < map->n; ++i) {
+               map->p[i] = isl_basic_map_set_dim_name(map->p[i], type, pos, s);
+               if (!map->p[i])
+                       goto error;
+       }
+
+       return map;
+error:
+       isl_map_free(map);
+       return NULL;
+}
+
+__isl_give isl_basic_set *isl_basic_set_set_dim_name(
+       __isl_take isl_basic_set *bset,
+       enum isl_dim_type type, unsigned pos, const char *s)
+{
+       return (isl_basic_set *)isl_basic_map_set_dim_name(
+               (isl_basic_map *)bset, type, pos, s);
+}
+
+__isl_give isl_set *isl_set_set_dim_name(__isl_take isl_set *set,
+       enum isl_dim_type type, unsigned pos, const char *s)
+{
+       return (isl_set *)isl_map_set_dim_name((isl_map *)set, type, pos, s);
+}
+
+int isl_basic_map_is_rational(__isl_keep isl_basic_map *bmap)
+{
+       if (!bmap)
+               return -1;
+       return ISL_F_ISSET(bmap, ISL_BASIC_MAP_RATIONAL);
+}
+
 static struct isl_basic_map *basic_map_init(struct isl_ctx *ctx,
                struct isl_basic_map *bmap, unsigned extra,
                unsigned n_eq, unsigned n_ineq)
@@ -330,9 +394,6 @@ static struct isl_basic_map *basic_map_init(struct isl_ctx *ctx,
        bmap->sample = NULL;
 
        return bmap;
-error:
-       isl_basic_map_free(bmap);
-       return NULL;
 }
 
 struct isl_basic_set *isl_basic_set_alloc(struct isl_ctx *ctx,
@@ -465,7 +526,10 @@ struct isl_basic_map *isl_basic_map_copy(struct isl_basic_map *bmap)
                bmap->ref++;
                return bmap;
        }
-       return isl_basic_map_dup(bmap);
+       bmap = isl_basic_map_dup(bmap);
+       if (bmap)
+               ISL_F_SET(bmap, ISL_BASIC_SET_FINAL);
+       return bmap;
 }
 
 struct isl_map *isl_map_copy(struct isl_map *map)
@@ -2148,6 +2212,8 @@ __isl_give isl_basic_map *isl_basic_map_move_dims(
                if (!bmap->dim)
                        goto error;
 
+               bmap = isl_basic_map_finalize(bmap);
+
                return bmap;
        }
 
@@ -2200,6 +2266,14 @@ error:
        return NULL;
 }
 
+__isl_give isl_basic_set *isl_basic_set_move_dims(__isl_take isl_basic_set *bset,
+       enum isl_dim_type dst_type, unsigned dst_pos,
+       enum isl_dim_type src_type, unsigned src_pos, unsigned n)
+{
+       return (isl_basic_set *)isl_basic_map_move_dims(
+               (isl_basic_map *)bset, dst_type, dst_pos, src_type, src_pos, n);
+}
+
 __isl_give isl_set *isl_set_move_dims(__isl_take isl_set *set,
        enum isl_dim_type dst_type, unsigned dst_pos,
        enum isl_dim_type src_type, unsigned src_pos, unsigned n)
@@ -3088,27 +3162,43 @@ error:
  *
  *             f - m d >= n
  */
-int isl_basic_map_add_div_constraints(struct isl_basic_map *bmap, unsigned div)
+int isl_basic_map_add_div_constraints_var(__isl_keep isl_basic_map *bmap,
+       unsigned pos, isl_int *div)
 {
        int i, j;
        unsigned total = isl_basic_map_total_dim(bmap);
-       unsigned div_pos = 1 + total - bmap->n_div + div;
 
        i = isl_basic_map_alloc_inequality(bmap);
        if (i < 0)
                return -1;
-       isl_seq_cpy(bmap->ineq[i], bmap->div[div]+1, 1+total);
-       isl_int_neg(bmap->ineq[i][div_pos], bmap->div[div][0]);
+       isl_seq_cpy(bmap->ineq[i], div + 1, 1 + total);
+       isl_int_neg(bmap->ineq[i][1 + pos], div[0]);
 
        j = isl_basic_map_alloc_inequality(bmap);
        if (j < 0)
                return -1;
        isl_seq_neg(bmap->ineq[j], bmap->ineq[i], 1 + total);
-       isl_int_add(bmap->ineq[j][0], bmap->ineq[j][0], bmap->ineq[j][div_pos]);
+       isl_int_add(bmap->ineq[j][0], bmap->ineq[j][0], bmap->ineq[j][1 + pos]);
        isl_int_sub_ui(bmap->ineq[j][0], bmap->ineq[j][0], 1);
        return j;
 }
 
+int isl_basic_set_add_div_constraints_var(__isl_keep isl_basic_set *bset,
+       unsigned pos, isl_int *div)
+{
+       return isl_basic_map_add_div_constraints_var((isl_basic_map *)bset,
+                                                       pos, div);
+}
+
+int isl_basic_map_add_div_constraints(struct isl_basic_map *bmap, unsigned div)
+{
+       unsigned total = isl_basic_map_total_dim(bmap);
+       unsigned div_pos = total - bmap->n_div + div;
+
+       return isl_basic_map_add_div_constraints_var(bmap, div_pos,
+                                                       bmap->div[div]);
+}
+
 struct isl_basic_set *isl_basic_map_underlying_set(
                struct isl_basic_map *bmap)
 {
@@ -5920,6 +6010,13 @@ error:
        return NULL;
 }
 
+__isl_give isl_basic_set *isl_basic_set_sort_constraints(
+       __isl_take isl_basic_set *bset)
+{
+       return (struct isl_basic_set *)isl_basic_map_sort_constraints(
+                                               (struct isl_basic_map *)bset);
+}
+
 struct isl_basic_map *isl_basic_map_normalize(struct isl_basic_map *bmap)
 {
        if (!bmap)
@@ -5990,6 +6087,13 @@ static int isl_basic_map_fast_is_equal(struct isl_basic_map *bmap1,
        return isl_basic_map_fast_cmp(bmap1, bmap2) == 0;
 }
 
+int isl_basic_set_fast_is_equal(__isl_keep isl_basic_set *bset1,
+       __isl_keep isl_basic_set *bset2)
+{
+       return isl_basic_map_fast_is_equal((isl_basic_map *)bset1,
+                                           (isl_basic_map *)bset2);
+}
+
 static int qsort_bmap_cmp(const void *p1, const void *p2)
 {
        const struct isl_basic_map *bmap1 = *(const struct isl_basic_map **)p1;
@@ -6414,6 +6518,8 @@ __isl_give isl_basic_set *isl_basic_set_lift(__isl_take isl_basic_set *bset)
        bset->dim = dim;
        bset->n_div = 0;
 
+       bset = isl_basic_set_finalize(bset);
+
        return bset;
 error:
        isl_basic_set_free(bset);
@@ -6601,3 +6707,122 @@ int isl_set_dim_is_bounded(__isl_keep isl_set *set,
 {
        return isl_map_dim_is_bounded((isl_map *)set, type, pos);
 }
+
+/* For each of the "n" variables starting at "first", determine
+ * the sign of the variable and put the results in the first "n"
+ * elements of the array "signs".
+ * Sign
+ *     1 means that the variable is non-negative
+ *     -1 means that the variable is non-positive
+ *     0 means the variable attains both positive and negative values.
+ */
+int isl_basic_set_vars_get_sign(__isl_keep isl_basic_set *bset,
+       unsigned first, unsigned n, int *signs)
+{
+       isl_vec *bound = NULL;
+       struct isl_tab *tab = NULL;
+       struct isl_tab_undo *snap;
+       int i;
+
+       if (!bset || !signs)
+               return -1;
+
+       bound = isl_vec_alloc(bset->ctx, 1 + isl_basic_set_total_dim(bset));
+       tab = isl_tab_from_basic_set(bset);
+       if (!bound || !tab)
+               goto error;
+
+       isl_seq_clr(bound->el, bound->size);
+       isl_int_set_si(bound->el[0], -1);
+
+       snap = isl_tab_snap(tab);
+       for (i = 0; i < n; ++i) {
+               int empty;
+
+               isl_int_set_si(bound->el[1 + first + i], -1);
+               if (isl_tab_add_ineq(tab, bound->el) < 0)
+                       goto error;
+               empty = tab->empty;
+               isl_int_set_si(bound->el[1 + first + i], 0);
+               if (isl_tab_rollback(tab, snap) < 0)
+                       goto error;
+
+               if (empty) {
+                       signs[i] = 1;
+                       continue;
+               }
+
+               isl_int_set_si(bound->el[1 + first + i], 1);
+               if (isl_tab_add_ineq(tab, bound->el) < 0)
+                       goto error;
+               empty = tab->empty;
+               isl_int_set_si(bound->el[1 + first + i], 0);
+               if (isl_tab_rollback(tab, snap) < 0)
+                       goto error;
+
+               signs[i] = empty ? -1 : 0;
+       }
+
+       isl_tab_free(tab);
+       isl_vec_free(bound);
+       return 0;
+error:
+       isl_tab_free(tab);
+       isl_vec_free(bound);
+       return -1;
+}
+
+int isl_basic_set_dims_get_sign(__isl_keep isl_basic_set *bset,
+       enum isl_dim_type type, unsigned first, unsigned n, int *signs)
+{
+       if (!bset || !signs)
+               return -1;
+       isl_assert(bset->ctx, first + n <= isl_basic_set_dim(bset, type),
+               return -1);
+
+       first += pos(bset->dim, type) - 1;
+       return isl_basic_set_vars_get_sign(bset, first, n, signs);
+}
+
+/* Check if the given map is single-valued.
+ * We simply compute
+ *
+ *     M \circ M^-1
+ *
+ * and check if the result is a subset of the identity mapping.
+ */
+int isl_map_is_single_valued(__isl_keep isl_map *map)
+{
+       isl_map *test;
+       isl_map *id;
+       int sv;
+
+       test = isl_map_reverse(isl_map_copy(map));
+       test = isl_map_apply_range(test, isl_map_copy(map));
+
+       id = isl_map_identity(isl_dim_range(isl_map_get_dim(map)));
+
+       sv = isl_map_is_subset(test, id);
+
+       isl_map_free(test);
+       isl_map_free(id);
+
+       return sv;
+}
+
+int isl_set_is_singleton(__isl_keep isl_set *set)
+{
+       return isl_map_is_single_valued((isl_map *)set);
+}
+
+int isl_map_is_translation(__isl_keep isl_map *map)
+{
+       int ok;
+       isl_set *delta;
+
+       delta = isl_map_deltas(isl_map_copy(map));
+       ok = isl_set_is_singleton(delta);
+       isl_set_free(delta);
+
+       return ok;
+}