doc: fix typo
[platform/upstream/isl.git] / isl_map.c
index d70f3f3..561f1c6 100644 (file)
--- a/isl_map.c
+++ b/isl_map.c
@@ -33,6 +33,7 @@
 #include <isl_aff_private.h>
 #include <isl_options_private.h>
 #include <isl_morph.h>
+#include <isl_val_private.h>
 
 static unsigned n(__isl_keep isl_space *dim, enum isl_dim_type type)
 {
@@ -2727,6 +2728,9 @@ int isl_basic_map_contains(struct isl_basic_map *bmap, struct isl_vec *vec)
        unsigned total;
        isl_int s;
 
+       if (!bmap || !vec)
+               return -1;
+
        total = 1 + isl_basic_map_total_dim(bmap);
        if (total != vec->size)
                return -1;
@@ -3851,6 +3855,26 @@ error:
        return NULL;
 }
 
+/* Given a map A -> f(A) and an integer d, construct a map
+ * A -> floor(f(A)/d).
+ */
+__isl_give isl_map *isl_map_floordiv_val(__isl_take isl_map *map,
+       __isl_take isl_val *d)
+{
+       if (!map || !d)
+               goto error;
+       if (!isl_val_is_int(d))
+               isl_die(isl_val_get_ctx(d), isl_error_invalid,
+                       "expecting integer denominator", goto error);
+       map = isl_map_floordiv(map, d->n);
+       isl_val_free(d);
+       return map;
+error:
+       isl_map_free(map);
+       isl_val_free(d);
+       return NULL;
+}
+
 static struct isl_basic_map *var_equal(struct isl_basic_map *bmap, unsigned pos)
 {
        int i;
@@ -5328,6 +5352,39 @@ error:
        return NULL;
 }
 
+/* Fix the value of the variable at position "pos" of type "type" of "bmap"
+ * to be equal to "v".
+ */
+__isl_give isl_basic_map *isl_basic_map_fix_val(__isl_take isl_basic_map *bmap,
+       enum isl_dim_type type, unsigned pos, __isl_take isl_val *v)
+{
+       if (!bmap || !v)
+               goto error;
+       if (!isl_val_is_int(v))
+               isl_die(isl_basic_map_get_ctx(bmap), isl_error_invalid,
+                       "expecting integer value", goto error);
+       if (pos >= isl_basic_map_dim(bmap, type))
+               isl_die(isl_basic_map_get_ctx(bmap), isl_error_invalid,
+                       "index out of bounds", goto error);
+       pos += isl_basic_map_offset(bmap, type);
+       bmap = isl_basic_map_fix_pos(bmap, pos, v->n);
+       isl_val_free(v);
+       return bmap;
+error:
+       isl_basic_map_free(bmap);
+       isl_val_free(v);
+       return NULL;
+}
+
+/* Fix the value of the variable at position "pos" of type "type" of "bset"
+ * to be equal to "v".
+ */
+__isl_give isl_basic_set *isl_basic_set_fix_val(__isl_take isl_basic_set *bset,
+       enum isl_dim_type type, unsigned pos, __isl_take isl_val *v)
+{
+       return isl_basic_map_fix_val(bset, type, pos, v);
+}
+
 struct isl_basic_set *isl_basic_set_fix_si(struct isl_basic_set *bset,
                enum isl_dim_type type, unsigned pos, int value)
 {
@@ -5467,6 +5524,48 @@ __isl_give isl_set *isl_set_fix(__isl_take isl_set *set,
        return (struct isl_set *)isl_map_fix((isl_map *)set, type, pos, value);
 }
 
+/* Fix the value of the variable at position "pos" of type "type" of "map"
+ * to be equal to "v".
+ */
+__isl_give isl_map *isl_map_fix_val(__isl_take isl_map *map,
+       enum isl_dim_type type, unsigned pos, __isl_take isl_val *v)
+{
+       int i;
+
+       map = isl_map_cow(map);
+       if (!map || !v)
+               goto error;
+
+       if (!isl_val_is_int(v))
+               isl_die(isl_map_get_ctx(map), isl_error_invalid,
+                       "expecting integer value", goto error);
+       if (pos >= isl_map_dim(map, type))
+               isl_die(isl_map_get_ctx(map), isl_error_invalid,
+                       "index out of bounds", goto error);
+       for (i = map->n - 1; i >= 0; --i) {
+               map->p[i] = isl_basic_map_fix_val(map->p[i], type, pos,
+                                                       isl_val_copy(v));
+               if (remove_if_empty(map, i) < 0)
+                       goto error;
+       }
+       ISL_F_CLR(map, ISL_MAP_NORMALIZED);
+       isl_val_free(v);
+       return map;
+error:
+       isl_map_free(map);
+       isl_val_free(v);
+       return NULL;
+}
+
+/* Fix the value of the variable at position "pos" of type "type" of "set"
+ * to be equal to "v".
+ */
+__isl_give isl_set *isl_set_fix_val(__isl_take isl_set *set,
+       enum isl_dim_type type, unsigned pos, __isl_take isl_val *v)
+{
+       return isl_map_fix_val(set, type, pos, v);
+}
+
 struct isl_map *isl_map_fix_input_si(struct isl_map *map,
                unsigned input, int value)
 {
@@ -5679,6 +5778,46 @@ __isl_give isl_set *isl_set_upper_bound(__isl_take isl_set *set,
        return isl_map_upper_bound(set, type, pos, value);
 }
 
+/* Force the values of the variable at position "pos" of type "type" of "set"
+ * to be no smaller than "value".
+ */
+__isl_give isl_set *isl_set_lower_bound_val(__isl_take isl_set *set,
+       enum isl_dim_type type, unsigned pos, __isl_take isl_val *value)
+{
+       if (!value)
+               goto error;
+       if (!isl_val_is_int(value))
+               isl_die(isl_set_get_ctx(set), isl_error_invalid,
+                       "expecting integer value", goto error);
+       set = isl_set_lower_bound(set, type, pos, value->n);
+       isl_val_free(value);
+       return set;
+error:
+       isl_val_free(value);
+       isl_set_free(set);
+       return NULL;
+}
+
+/* Force the values of the variable at position "pos" of type "type" of "set"
+ * to be no greater than "value".
+ */
+__isl_give isl_set *isl_set_upper_bound_val(__isl_take isl_set *set,
+       enum isl_dim_type type, unsigned pos, __isl_take isl_val *value)
+{
+       if (!value)
+               goto error;
+       if (!isl_val_is_int(value))
+               isl_die(isl_set_get_ctx(set), isl_error_invalid,
+                       "expecting integer value", goto error);
+       set = isl_set_upper_bound(set, type, pos, value->n);
+       isl_val_free(value);
+       return set;
+error:
+       isl_val_free(value);
+       isl_set_free(set);
+       return NULL;
+}
+
 struct isl_set *isl_set_lower_bound_dim(struct isl_set *set, unsigned dim,
                                        isl_int value)
 {
@@ -7931,6 +8070,8 @@ struct isl_basic_map *isl_basic_map_align_divs(
 
        src = isl_basic_map_order_divs(src);
        dst = isl_basic_map_cow(dst);
+       if (!dst)
+               return NULL;
        dst = isl_basic_map_extend_space(dst, isl_space_copy(dst->dim),
                        src->n_div, 0, 2 * src->n_div);
        if (!dst)
@@ -7978,8 +8119,11 @@ struct isl_map *isl_map_align_divs(struct isl_map *map)
 
        for (i = 1; i < map->n; ++i)
                map->p[0] = isl_basic_map_align_divs(map->p[0], map->p[i]);
-       for (i = 1; i < map->n; ++i)
+       for (i = 1; i < map->n; ++i) {
                map->p[i] = isl_basic_map_align_divs(map->p[i], map->p[0]);
+               if (!map->p[i])
+                       return isl_map_free(map);
+       }
 
        ISL_F_CLR(map, ISL_MAP_NORMALIZED);
        return map;
@@ -8327,6 +8471,35 @@ int isl_basic_map_plain_is_fixed(__isl_keep isl_basic_map *bmap,
                isl_basic_map_offset(bmap, type) - 1 + pos, val);
 }
 
+/* If "bmap" obviously lies on a hyperplane where the given dimension
+ * has a fixed value, then return that value.
+ * Otherwise return NaN.
+ */
+__isl_give isl_val *isl_basic_map_plain_get_val_if_fixed(
+       __isl_keep isl_basic_map *bmap,
+       enum isl_dim_type type, unsigned pos)
+{
+       isl_ctx *ctx;
+       isl_val *v;
+       int fixed;
+
+       if (!bmap)
+               return NULL;
+       ctx = isl_basic_map_get_ctx(bmap);
+       v = isl_val_alloc(ctx);
+       if (!v)
+               return NULL;
+       fixed = isl_basic_map_plain_is_fixed(bmap, type, pos, &v->n);
+       if (fixed < 0)
+               return isl_val_free(v);
+       if (fixed) {
+               isl_int_set_si(v->d, 1);
+               return v;
+       }
+       isl_val_free(v);
+       return isl_val_nan(ctx);
+}
+
 int isl_map_plain_is_fixed(__isl_keep isl_map *map,
        enum isl_dim_type type, unsigned pos, isl_int *val)
 {
@@ -8336,6 +8509,44 @@ int isl_map_plain_is_fixed(__isl_keep isl_map *map,
                map_offset(map, type) - 1 + pos, val);
 }
 
+/* If "map" obviously lies on a hyperplane where the given dimension
+ * has a fixed value, then return that value.
+ * Otherwise return NaN.
+ */
+__isl_give isl_val *isl_map_plain_get_val_if_fixed(__isl_keep isl_map *map,
+       enum isl_dim_type type, unsigned pos)
+{
+       isl_ctx *ctx;
+       isl_val *v;
+       int fixed;
+
+       if (!map)
+               return NULL;
+       ctx = isl_map_get_ctx(map);
+       v = isl_val_alloc(ctx);
+       if (!v)
+               return NULL;
+       fixed = isl_map_plain_is_fixed(map, type, pos, &v->n);
+       if (fixed < 0)
+               return isl_val_free(v);
+       if (fixed) {
+               isl_int_set_si(v->d, 1);
+               return v;
+       }
+       isl_val_free(v);
+       return isl_val_nan(ctx);
+}
+
+/* If "set" obviously lies on a hyperplane where the given dimension
+ * has a fixed value, then return that value.
+ * Otherwise return NaN.
+ */
+__isl_give isl_val *isl_set_plain_get_val_if_fixed(__isl_keep isl_set *set,
+       enum isl_dim_type type, unsigned pos)
+{
+       return isl_map_plain_get_val_if_fixed(set, type, pos);
+}
+
 int isl_set_plain_is_fixed(__isl_keep isl_set *set,
        enum isl_dim_type type, unsigned pos, isl_int *val)
 {