add isl_basic_set_preimage_multi_aff
[platform/upstream/isl.git] / isl_aff.c
index 9ad26d6..9174069 100644 (file)
--- a/isl_aff.c
+++ b/isl_aff.c
@@ -1363,8 +1363,10 @@ __isl_give isl_aff *isl_aff_gist_params(__isl_take isl_aff *aff,
 
 /* Return a basic set containing those elements in the space
  * of aff where it is non-negative.
+ * If "rational" is set, then return a rational basic set.
  */
-__isl_give isl_basic_set *isl_aff_nonneg_basic_set(__isl_take isl_aff *aff)
+static __isl_give isl_basic_set *aff_nonneg_basic_set(
+       __isl_take isl_aff *aff, int rational)
 {
        isl_constraint *ineq;
        isl_basic_set *bset;
@@ -1372,10 +1374,20 @@ __isl_give isl_basic_set *isl_aff_nonneg_basic_set(__isl_take isl_aff *aff)
        ineq = isl_inequality_from_aff(aff);
 
        bset = isl_basic_set_from_constraint(ineq);
+       if (rational)
+               bset = isl_basic_set_set_rational(bset);
        bset = isl_basic_set_simplify(bset);
        return bset;
 }
 
+/* Return a basic set containing those elements in the space
+ * of aff where it is non-negative.
+ */
+__isl_give isl_basic_set *isl_aff_nonneg_basic_set(__isl_take isl_aff *aff)
+{
+       return aff_nonneg_basic_set(aff, 0);
+}
+
 /* Return a basic set containing those elements in the domain space
  * of aff where it is negative.
  */
@@ -1388,8 +1400,10 @@ __isl_give isl_basic_set *isl_aff_neg_basic_set(__isl_take isl_aff *aff)
 
 /* Return a basic set containing those elements in the space
  * of aff where it is zero.
+ * If "rational" is set, then return a rational basic set.
  */
-__isl_give isl_basic_set *isl_aff_zero_basic_set(__isl_take isl_aff *aff)
+static __isl_give isl_basic_set *aff_zero_basic_set(__isl_take isl_aff *aff,
+       int rational)
 {
        isl_constraint *ineq;
        isl_basic_set *bset;
@@ -1397,10 +1411,20 @@ __isl_give isl_basic_set *isl_aff_zero_basic_set(__isl_take isl_aff *aff)
        ineq = isl_equality_from_aff(aff);
 
        bset = isl_basic_set_from_constraint(ineq);
+       if (rational)
+               bset = isl_basic_set_set_rational(bset);
        bset = isl_basic_set_simplify(bset);
        return bset;
 }
 
+/* Return a basic set containing those elements in the space
+ * of aff where it is zero.
+ */
+__isl_give isl_basic_set *isl_aff_zero_basic_set(__isl_take isl_aff *aff)
+{
+       return aff_zero_basic_set(aff, 0);
+}
+
 /* Return a basic set containing those elements in the shared space
  * of aff1 and aff2 where aff1 is greater than or equal to aff2.
  */
@@ -1855,8 +1879,11 @@ __isl_give isl_set *isl_pw_aff_nonneg_set(__isl_take isl_pw_aff *pwaff)
        for (i = 0; i < pwaff->n; ++i) {
                isl_basic_set *bset;
                isl_set *set_i;
+               int rational;
 
-               bset = isl_aff_nonneg_basic_set(isl_aff_copy(pwaff->p[i].aff));
+               rational = isl_set_has_rational(pwaff->p[i].set);
+               bset = aff_nonneg_basic_set(isl_aff_copy(pwaff->p[i].aff),
+                                               rational);
                set_i = isl_set_from_basic_set(bset);
                set_i = isl_set_intersect(set_i, isl_set_copy(pwaff->p[i].set));
                set = isl_set_union_disjoint(set, set_i);
@@ -1885,8 +1912,11 @@ static __isl_give isl_set *pw_aff_zero_set(__isl_take isl_pw_aff *pwaff,
        for (i = 0; i < pwaff->n; ++i) {
                isl_basic_set *bset;
                isl_set *set_i, *zero;
+               int rational;
 
-               bset = isl_aff_zero_basic_set(isl_aff_copy(pwaff->p[i].aff));
+               rational = isl_set_has_rational(pwaff->p[i].set);
+               bset = aff_zero_basic_set(isl_aff_copy(pwaff->p[i].aff),
+                                               rational);
                zero = isl_set_from_basic_set(bset);
                set_i = isl_set_copy(pwaff->p[i].set);
                if (complement)
@@ -2509,6 +2539,50 @@ __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);
 }
 
+/* Mark the domains of "pwaff" as rational.
+ */
+__isl_give isl_pw_aff *isl_pw_aff_set_rational(__isl_take isl_pw_aff *pwaff)
+{
+       int i;
+
+       pwaff = isl_pw_aff_cow(pwaff);
+       if (!pwaff)
+               return NULL;
+       if (pwaff->n == 0)
+               return pwaff;
+
+       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 pwaff;
+}
+
+/* Mark the domains of the elements of "list" as rational.
+ */
+__isl_give isl_pw_aff_list *isl_pw_aff_list_set_rational(
+       __isl_take isl_pw_aff_list *list)
+{
+       int i;
+
+       if (!list)
+               return NULL;
+       if (list->n == 0)
+               return list;
+
+       for (i = 0; i < list->n; ++i) {
+               isl_pw_aff *pa;
+
+               pa = isl_pw_aff_list_get_pw_aff(list, i);
+               pa = isl_pw_aff_set_rational(pa);
+               list = isl_pw_aff_list_set_pw_aff(list, i, pa);
+       }
+
+       return list;
+}
+
 #undef BASE
 #define BASE aff
 
@@ -3343,6 +3417,77 @@ __isl_give isl_pw_multi_aff *isl_pw_multi_aff_substitute(
        return res;
 }
 
+/* Compute the preimage of the affine expression "src" under "ma"
+ * and put the result in "dst".  If "has_denom" is set (to one),
+ * then "src" and "dst" have an extra initial denominator.
+ * "n_div_ma" is the number of existentials in "ma"
+ * "n_div_bset" is the number of existentials in "src"
+ * The resulting "dst" (which is assumed to have been allocated by
+ * the caller) contains coefficients for both sets of existentials,
+ * first those in "ma" and then those in "src".
+ * f, c1, c2 and g are temporary objects that have been initialized
+ * by the caller.
+ *
+ * Let src represent the expression
+ *
+ *     (a(p) + b x + c(divs))/d
+ *
+ * and let ma represent the expressions
+ *
+ *     x_i = (r_i(p) + s_i(y) + t_i(divs'))/m_i
+ *
+ * We start out with the following expression for dst:
+ *
+ *     (a(p) + 0 y + 0 divs' + f \sum_i b_i x_i + c(divs))/d
+ *
+ * with the multiplication factor f initially equal to 1.
+ * For each x_i that we substitute, we multiply the numerator
+ * (and denominator) of dst by c_1 = m_i and add the numerator
+ * of the x_i expression multiplied by c_2 = f b_i,
+ * after removing the common factors of c_1 and c_2.
+ * The multiplication factor f also needs to be multiplied by c_1
+ * for the next x_j, j > i.
+ */
+void isl_seq_preimage(isl_int *dst, isl_int *src,
+       __isl_keep isl_multi_aff *ma, int n_div_ma, int n_div_bset,
+       isl_int f, isl_int c1, isl_int c2, isl_int g, int has_denom)
+{
+       int i;
+       int n_param, n_in, n_out;
+       int o_div_bset;
+
+       n_param = isl_multi_aff_dim(ma, isl_dim_param);
+       n_in = isl_multi_aff_dim(ma, isl_dim_in);
+       n_out = isl_multi_aff_dim(ma, isl_dim_out);
+
+       o_div_bset = has_denom + 1 + n_param + n_in + n_div_ma;
+
+       isl_seq_cpy(dst, src, has_denom + 1 + n_param);
+       isl_seq_clr(dst + has_denom + 1 + n_param, n_in + n_div_ma);
+       isl_seq_cpy(dst + o_div_bset,
+                   src + has_denom + 1 + n_param + n_out, n_div_bset);
+
+       isl_int_set_si(f, 1);
+
+       for (i = 0; i < n_out; ++i) {
+               if (isl_int_is_zero(src[has_denom + 1 + n_param + i]))
+                       continue;
+               isl_int_set(c1, ma->p[i]->v->el[0]);
+               isl_int_mul(c2, f, src[has_denom + 1 + n_param + i]);
+               isl_int_gcd(g, c1, c2);
+               isl_int_divexact(c1, c1, g);
+               isl_int_divexact(c2, c2, g);
+
+               isl_int_mul(f, f, c1);
+               isl_seq_combine(dst + has_denom, c1, dst + has_denom,
+                               c2, ma->p[i]->v->el + 1, ma->p[i]->v->size - 1);
+               isl_seq_scale(dst + o_div_bset,
+                               dst + o_div_bset, c1, n_div_bset);
+               if (has_denom)
+                       isl_int_mul(dst[0], dst[0], c1);
+       }
+}
+
 /* Extend the local space of "dst" to include the divs
  * in the local space of "src".
  */
@@ -3679,6 +3824,30 @@ error:
 }
 
 /* Given two aligned isl_pw_multi_affs A -> B and C -> D,
+ * construct an isl_pw_multi_aff (A * C) -> [B -> D].
+ */
+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)
+{
+       isl_space *space;
+
+       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);
+}
+
+/* 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,
  * construct an isl_pw_multi_aff (A * C) -> (B, D).
  */
 static __isl_give isl_pw_multi_aff *pw_multi_aff_flat_range_product(
@@ -3817,3 +3986,8 @@ error:
        isl_pw_aff_free(pa);
        return NULL;
 }
+
+#undef BASE
+#define BASE pw_aff
+
+#include <isl_multi_templ.c>