add isl_qpolynomial_fold_scale_val
[platform/upstream/isl.git] / isl_multi_templ.c
index b82dc4b..8e3c9d3 100644 (file)
@@ -8,6 +8,10 @@
  * Ecole Normale Superieure, 45 rue d’Ulm, 75230 Paris, France
  */
 
+#include <isl_space_private.h>
+#include <isl/set.h>
+#include <isl_reordering.h>
+
 #define xCAT(A,B) A ## B
 #define CAT(A,B) xCAT(A,B)
 #undef EL
@@ -223,16 +227,8 @@ __isl_give MULTI(BASE) *FN(FN(MULTI(BASE),set),BASE)(
                goto error;
 
        multi_space = FN(MULTI(BASE),get_space)(multi);
-       el_space = FN(EL,get_space)(el);
-
-       if (!isl_space_match(multi_space, isl_dim_param,
-                           el_space, isl_dim_param))
-               isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
-                       "parameters don't match", goto error);
-       if (!isl_space_tuple_match(multi_space, isl_dim_in,
-                           el_space, isl_dim_in))
-               isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
-                       "domains don't match", goto error);
+       if (FN(EL,check_match_domain_space)(el, multi_space) < 0)
+               goto error;
 
        if (pos < 0 || pos >= multi->n)
                isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
@@ -345,7 +341,7 @@ __isl_give MULTI(BASE) *FN(MULTI(BASE),realign_domain)(
 
        multi = FN(MULTI(BASE),cow)(multi);
        if (!multi || !exp)
-               return NULL;
+               goto error;
 
        for (i = 0; i < multi->n; ++i) {
                multi->p[i] = FN(EL,realign_domain)(multi->p[i],
@@ -401,6 +397,7 @@ error:
        return NULL;
 }
 
+#ifndef NO_GIST
 static __isl_give MULTI(BASE) *FN(MULTI(BASE),align_params_multi_set_and)(
        __isl_take MULTI(BASE) *multi, __isl_take isl_set *set,
        __isl_give MULTI(BASE) *(*fn)(__isl_take MULTI(BASE) *multi,
@@ -464,6 +461,7 @@ __isl_give MULTI(BASE) *FN(MULTI(BASE),gist_params)(
        dom_context = isl_set_intersect_params(dom_context, context);
        return FN(MULTI(BASE),gist)(multi, dom_context);
 }
+#endif
 
 __isl_give MULTI(BASE) *FN(FN(MULTI(BASE),from),LIST(BASE))(
        __isl_take isl_space *space, __isl_take LIST(EL) *list)
@@ -497,6 +495,7 @@ error:
        return NULL;
 }
 
+#ifndef NO_IDENTITY
 /* Create a multi expression in the given space that maps each
  * input dimension to the corresponding output dimension.
  */
@@ -543,6 +542,7 @@ error:
        isl_space_free(space);
        return NULL;
 }
+#endif
 
 /* Construct a multi expression in the given space with value zero in
  * each of the output dimensions.
@@ -579,6 +579,7 @@ __isl_give MULTI(BASE) *FN(MULTI(BASE),zero)(__isl_take isl_space *space)
        return multi;
 }
 
+#ifndef NO_FROM_BASE
 __isl_give MULTI(BASE) *FN(FN(MULTI(BASE),from),BASE)(__isl_take EL *el)
 {
        MULTI(BASE) *multi;
@@ -588,6 +589,7 @@ __isl_give MULTI(BASE) *FN(FN(MULTI(BASE),from),BASE)(__isl_take EL *el)
 
        return multi;
 }
+#endif
 
 __isl_give MULTI(BASE) *FN(MULTI(BASE),drop_dims)(
        __isl_take MULTI(BASE) *multi,
@@ -701,3 +703,136 @@ __isl_give MULTI(BASE) *FN(MULTI(BASE),flat_range_product)(
        multi = FN(MULTI(BASE),flatten_range)(multi);
        return multi;
 }
+
+/* Given two multi expressions, "multi1"
+ *
+ *     [A] -> [B1 B2]
+ *
+ * where B2 starts at position "pos", and "multi2"
+ *
+ *     [A] -> [D]
+ *
+ * return the multi expression
+ *
+ *     [A] -> [B1 D B2]
+ */
+__isl_give MULTI(BASE) *FN(MULTI(BASE),range_splice)(
+       __isl_take MULTI(BASE) *multi1, unsigned pos,
+       __isl_take MULTI(BASE) *multi2)
+{
+       MULTI(BASE) *res;
+       unsigned dim;
+
+       if (!multi1 || !multi2)
+               goto error;
+
+       dim = FN(MULTI(BASE),dim)(multi1, isl_dim_out);
+       if (pos > dim)
+               isl_die(FN(MULTI(BASE),get_ctx)(multi1), isl_error_invalid,
+                       "index out of bounds", goto error);
+
+       res = FN(MULTI(BASE),copy)(multi1);
+       res = FN(MULTI(BASE),drop_dims)(res, isl_dim_out, pos, dim - pos);
+       multi1 = FN(MULTI(BASE),drop_dims)(multi1, isl_dim_out, 0, pos);
+
+       res = FN(MULTI(BASE),flat_range_product)(res, multi2);
+       res = FN(MULTI(BASE),flat_range_product)(res, multi1);
+
+       return res;
+error:
+       FN(MULTI(BASE),free)(multi1);
+       FN(MULTI(BASE),free)(multi2);
+       return NULL;
+}
+
+/* Given two multi expressions, "multi1"
+ *
+ *     [A1 A2] -> [B1 B2]
+ *
+ * where A2 starts at position "in_pos" and B2 starts at position "out_pos",
+ * and "multi2"
+ *
+ *     [C] -> [D]
+ *
+ * return the multi expression
+ *
+ *     [A1 C A2] -> [B1 D B2]
+ *
+ * We first insert input dimensions to obtain
+ *
+ *     [A1 C A2] -> [B1 B2]
+ *
+ * and
+ *
+ *     [A1 C A2] -> [D]
+ *
+ * and then apply range_splice.
+ */
+__isl_give MULTI(BASE) *FN(MULTI(BASE),splice)(
+       __isl_take MULTI(BASE) *multi1, unsigned in_pos, unsigned out_pos,
+       __isl_take MULTI(BASE) *multi2)
+{
+       unsigned n_in1;
+       unsigned n_in2;
+
+       if (!multi1 || !multi2)
+               goto error;
+
+       n_in1 = FN(MULTI(BASE),dim)(multi1, isl_dim_in);
+       if (in_pos > n_in1)
+               isl_die(FN(MULTI(BASE),get_ctx)(multi1), isl_error_invalid,
+                       "index out of bounds", goto error);
+
+       n_in2 = FN(MULTI(BASE),dim)(multi2, isl_dim_in);
+
+       multi1 = FN(MULTI(BASE),insert_dims)(multi1, isl_dim_in, in_pos, n_in2);
+       multi2 = FN(MULTI(BASE),insert_dims)(multi2, isl_dim_in, n_in2,
+                                               n_in1 - in_pos);
+       multi2 = FN(MULTI(BASE),insert_dims)(multi2, isl_dim_in, 0, in_pos);
+
+       return FN(MULTI(BASE),range_splice)(multi1, out_pos, multi2);
+error:
+       FN(MULTI(BASE),free)(multi1);
+       FN(MULTI(BASE),free)(multi2);
+       return NULL;
+}
+
+/* This function is currently only used from isl_aff.c
+ */
+static __isl_give MULTI(BASE) *FN(MULTI(BASE),bin_op)(
+       __isl_take MULTI(BASE) *multi1, __isl_take MULTI(BASE) *multi2,
+       __isl_give EL *(*fn)(__isl_take EL *, __isl_take EL *))
+       __attribute__ ((unused));
+
+/* Pairwise perform "fn" to the elements of "multi1" and "multi2" and
+ * return the result.
+ */
+static __isl_give MULTI(BASE) *FN(MULTI(BASE),bin_op)(
+       __isl_take MULTI(BASE) *multi1, __isl_take MULTI(BASE) *multi2,
+       __isl_give EL *(*fn)(__isl_take EL *, __isl_take EL *))
+{
+       int i;
+       isl_ctx *ctx;
+
+       multi1 = FN(MULTI(BASE),cow)(multi1);
+       if (!multi1 || !multi2)
+               goto error;
+
+       ctx = FN(MULTI(BASE),get_ctx)(multi1);
+       if (!isl_space_is_equal(multi1->space, multi2->space))
+               isl_die(ctx, isl_error_invalid,
+                       "spaces don't match", goto error);
+
+       for (i = 0; i < multi1->n; ++i) {
+               multi1->p[i] = fn(multi1->p[i], FN(EL,copy)(multi2->p[i]));
+               if (!multi1->p[i])
+                       goto error;
+       }
+
+       FN(MULTI(BASE),free)(multi2);
+       return multi1;
+error:
+       FN(MULTI(BASE),free)(multi1);
+       FN(MULTI(BASE),free)(multi2);
+       return NULL;
+}