add isl_pw_aff_set_rational
[platform/upstream/isl.git] / isl_aff.c
index 5affddc..41bee5c 100644 (file)
--- a/isl_aff.c
+++ b/isl_aff.c
@@ -2509,45 +2509,32 @@ __isl_give isl_pw_aff *isl_pw_aff_list_max(__isl_take isl_pw_aff_list *list)
        return pw_aff_list_reduce(list, &isl_pw_aff_max);
 }
 
-#undef BASE
-#define BASE aff
-
-#include <isl_multi_templ.c>
-
-/* Construct an isl_multi_aff in the given space with value zero in
- * each of the output dimensions.
+/* Mark the domains of "pwaff" as rational.
  */
-__isl_give isl_multi_aff *isl_multi_aff_zero(__isl_take isl_space *space)
+__isl_give isl_pw_aff *isl_pw_aff_set_rational(__isl_take isl_pw_aff *pwaff)
 {
-       int n;
-       isl_multi_aff *ma;
+       int i;
 
-       if (!space)
+       pwaff = isl_pw_aff_cow(pwaff);
+       if (!pwaff)
                return NULL;
+       if (pwaff->n == 0)
+               return pwaff;
 
-       n = isl_space_dim(space , isl_dim_out);
-       ma = isl_multi_aff_alloc(isl_space_copy(space));
-
-       if (!n)
-               isl_space_free(space);
-       else {
-               int i;
-               isl_local_space *ls;
-               isl_aff *aff;
-
-               space = isl_space_domain(space);
-               ls = isl_local_space_from_space(space);
-               aff = isl_aff_zero_on_domain(ls);
-
-               for (i = 0; i < n; ++i)
-                       ma = isl_multi_aff_set_aff(ma, i, isl_aff_copy(aff));
-
-               isl_aff_free(aff);
+       for (i = 0; i < pwaff->n; ++i) {
+               pwaff->p[i].set = isl_set_set_rational(pwaff->p[i].set);
+               if (!pwaff->p[i].set)
+                       return isl_pw_aff_free(pwaff);
        }
 
-       return ma;
+       return pwaff;
 }
 
+#undef BASE
+#define BASE aff
+
+#include <isl_multi_templ.c>
+
 /* Create an isl_pw_multi_aff with the given isl_multi_aff on a universe
  * domain.
  */
@@ -2721,62 +2708,6 @@ int isl_multi_aff_plain_is_equal(__isl_keep isl_multi_aff *maff1,
        return 1;
 }
 
-__isl_give isl_multi_aff *isl_multi_aff_set_dim_name(
-       __isl_take isl_multi_aff *maff,
-       enum isl_dim_type type, unsigned pos, const char *s)
-{
-       int i;
-
-       maff = isl_multi_aff_cow(maff);
-       if (!maff)
-               return NULL;
-
-       maff->space = isl_space_set_dim_name(maff->space, type, pos, s);
-       if (!maff->space)
-               return isl_multi_aff_free(maff);
-
-       if (type == isl_dim_out)
-               return maff;
-       for (i = 0; i < maff->n; ++i) {
-               maff->p[i] = isl_aff_set_dim_name(maff->p[i], type, pos, s);
-               if (!maff->p[i])
-                       return isl_multi_aff_free(maff);
-       }
-
-       return maff;
-}
-
-__isl_give isl_multi_aff *isl_multi_aff_drop_dims(__isl_take isl_multi_aff *maff,
-       enum isl_dim_type type, unsigned first, unsigned n)
-{
-       int i;
-
-       maff = isl_multi_aff_cow(maff);
-       if (!maff)
-               return NULL;
-
-       maff->space = isl_space_drop_dims(maff->space, type, first, n);
-       if (!maff->space)
-               return isl_multi_aff_free(maff);
-
-       if (type == isl_dim_out) {
-               for (i = 0; i < n; ++i)
-                       isl_aff_free(maff->p[first + i]);
-               for (i = first; i + n < maff->n; ++i)
-                       maff->p[i] = maff->p[i + n];
-               maff->n -= n;
-               return maff;
-       }
-
-       for (i = 0; i < maff->n; ++i) {
-               maff->p[i] = isl_aff_drop_dims(maff->p[i], type, first, n);
-               if (!maff->p[i])
-                       return isl_multi_aff_free(maff);
-       }
-
-       return maff;
-}
-
 /* Return the set of domain elements where "ma1" is lexicographically
  * smaller than or equal to "ma2".
  */
@@ -3768,45 +3699,28 @@ error:
        return NULL;
 }
 
-/* Given two isl_multi_affs A -> B and C -> D,
- * construct an isl_multi_aff (A * C) -> (B, D).
+/* Given two aligned isl_pw_multi_affs A -> B and C -> D,
+ * construct an isl_pw_multi_aff (A * C) -> [B -> D].
  */
-__isl_give isl_multi_aff *isl_multi_aff_flat_range_product(
-       __isl_take isl_multi_aff *ma1, __isl_take isl_multi_aff *ma2)
+static __isl_give isl_pw_multi_aff *pw_multi_aff_range_product(
+       __isl_take isl_pw_multi_aff *pma1, __isl_take isl_pw_multi_aff *pma2)
 {
-       int i, n1, n2;
-       isl_aff *aff;
        isl_space *space;
-       isl_multi_aff *res;
-
-       if (!ma1 || !ma2)
-               goto error;
-
-       space = isl_space_range_product(isl_multi_aff_get_space(ma1),
-                                       isl_multi_aff_get_space(ma2));
-       space = isl_space_flatten_range(space);
-       res = isl_multi_aff_alloc(space);
-
-       n1 = isl_multi_aff_dim(ma1, isl_dim_out);
-       n2 = isl_multi_aff_dim(ma2, isl_dim_out);
 
-       for (i = 0; i < n1; ++i) {
-               aff = isl_multi_aff_get_aff(ma1, i);
-               res = isl_multi_aff_set_aff(res, i, aff);
-       }
-
-       for (i = 0; i < n2; ++i) {
-               aff = isl_multi_aff_get_aff(ma2, i);
-               res = isl_multi_aff_set_aff(res, n1 + i, aff);
-       }
+       space = isl_space_range_product(isl_pw_multi_aff_get_space(pma1),
+                                       isl_pw_multi_aff_get_space(pma2));
+       return isl_pw_multi_aff_on_shared_domain_in(pma1, pma2, space,
+                                           &isl_multi_aff_range_product);
+}
 
-       isl_multi_aff_free(ma1);
-       isl_multi_aff_free(ma2);
-       return res;
-error:
-       isl_multi_aff_free(ma1);
-       isl_multi_aff_free(ma2);
-       return NULL;
+/* Given two isl_pw_multi_affs A -> B and C -> D,
+ * construct an isl_pw_multi_aff (A * C) -> [B -> D].
+ */
+__isl_give isl_pw_multi_aff *isl_pw_multi_aff_range_product(
+       __isl_take isl_pw_multi_aff *pma1, __isl_take isl_pw_multi_aff *pma2)
+{
+       return isl_pw_multi_aff_align_params_pw_pw_and(pma1, pma2,
+                                           &pw_multi_aff_range_product);
 }
 
 /* Given two aligned isl_pw_multi_affs A -> B and C -> D,
@@ -3948,3 +3862,8 @@ error:
        isl_pw_aff_free(pa);
        return NULL;
 }
+
+#undef BASE
+#define BASE pw_aff
+
+#include <isl_multi_templ.c>