isl_qpolynomial_div: further normalize divs by reducing coefficients
[platform/upstream/isl.git] / isl_map.c
index 93ffa58..335fad0 100644 (file)
--- a/isl_map.c
+++ b/isl_map.c
 
 #include <string.h>
 #include <strings.h>
-#include "isl_ctx.h"
-#include "isl_blk.h"
+#include <isl/ctx.h>
+#include <isl/blk.h>
 #include "isl_dim_private.h"
 #include "isl_equalities.h"
-#include "isl_list.h"
-#include "isl_lp.h"
-#include "isl_seq.h"
-#include "isl_set.h"
-#include "isl_map.h"
+#include <isl/list.h>
+#include <isl/lp.h>
+#include <isl/seq.h>
+#include <isl/set.h>
+#include <isl/map.h>
 #include "isl_map_private.h"
 #include "isl_map_piplib.h"
 #include <isl_reordering.h>
 #include "isl_sample.h"
 #include "isl_tab.h"
-#include "isl_vec.h"
+#include <isl/vec.h>
 #include <isl_mat_private.h>
 
 /* Maps dst positions to src positions */
@@ -5571,12 +5571,54 @@ error:
        return NULL;
 }
 
+/* Construct the half-space x_pos >= 0.
+ */
+static __isl_give isl_basic_set *nonneg_halfspace(__isl_take isl_dim *dim,
+       int pos)
+{
+       int k;
+       isl_basic_set *nonneg;
+
+       nonneg = isl_basic_set_alloc_dim(dim, 0, 0, 1);
+       k = isl_basic_set_alloc_inequality(nonneg);
+       if (k < 0)
+               goto error;
+       isl_seq_clr(nonneg->ineq[k], 1 + isl_basic_set_total_dim(nonneg));
+       isl_int_set_si(nonneg->ineq[k][pos], 1);
+
+       return isl_basic_set_finalize(nonneg);
+error:
+       isl_basic_set_free(nonneg);
+       return NULL;
+}
+
+/* Construct the half-space x_pos <= -1.
+ */
+static __isl_give isl_basic_set *neg_halfspace(__isl_take isl_dim *dim, int pos)
+{
+       int k;
+       isl_basic_set *neg;
+
+       neg = isl_basic_set_alloc_dim(dim, 0, 0, 1);
+       k = isl_basic_set_alloc_inequality(neg);
+       if (k < 0)
+               goto error;
+       isl_seq_clr(neg->ineq[k], 1 + isl_basic_set_total_dim(neg));
+       isl_int_set_si(neg->ineq[k][0], -1);
+       isl_int_set_si(neg->ineq[k][pos], -1);
+
+       return isl_basic_set_finalize(neg);
+error:
+       isl_basic_set_free(neg);
+       return NULL;
+}
+
 __isl_give isl_set *isl_set_split_dims(__isl_take isl_set *set,
        enum isl_dim_type type, unsigned first, unsigned n)
 {
        int i;
-       isl_basic_set *nonneg = NULL;
-       isl_basic_set *neg = NULL;
+       isl_basic_set *nonneg;
+       isl_basic_set *neg;
 
        if (!set)
                return NULL;
@@ -5586,35 +5628,84 @@ __isl_give isl_set *isl_set_split_dims(__isl_take isl_set *set,
        isl_assert(set->ctx, first + n <= isl_set_dim(set, type), goto error);
 
        for (i = 0; i < n; ++i) {
-               int k;
-
-               neg = NULL;
-               nonneg = isl_basic_set_alloc_dim(isl_set_get_dim(set), 0, 0, 1);
-               k = isl_basic_set_alloc_inequality(nonneg);
-               if (k < 0)
-                       goto error;
-               isl_seq_clr(nonneg->ineq[k], 1 + isl_basic_set_total_dim(nonneg));
-               isl_int_set_si(nonneg->ineq[k][pos(set->dim, type) + first + i], 1);
-
-               neg = isl_basic_set_alloc_dim(isl_set_get_dim(set), 0, 0, 1);
-               k = isl_basic_set_alloc_inequality(neg);
-               if (k < 0)
-                       goto error;
-               isl_seq_clr(neg->ineq[k], 1 + isl_basic_set_total_dim(neg));
-               isl_int_set_si(neg->ineq[k][0], -1);
-               isl_int_set_si(neg->ineq[k][pos(set->dim, type) + first + i], -1);
+               nonneg = nonneg_halfspace(isl_set_get_dim(set),
+                                         pos(set->dim, type) + first + i);
+               neg = neg_halfspace(isl_set_get_dim(set),
+                                         pos(set->dim, type) + first + i);
 
                set = isl_set_intersect(set, isl_basic_set_union(nonneg, neg));
        }
 
        return set;
 error:
-       isl_basic_set_free(nonneg);
-       isl_basic_set_free(neg);
        isl_set_free(set);
        return NULL;
 }
 
+static int foreach_orthant(__isl_take isl_set *set, int *signs, int first,
+       int len, int (*fn)(__isl_take isl_set *orthant, int *signs, void *user),
+       void *user)
+{
+       isl_set *half;
+
+       if (!set)
+               return -1;
+       if (isl_set_fast_is_empty(set)) {
+               isl_set_free(set);
+               return 0;
+       }
+       if (first == len)
+               return fn(set, signs, user);
+
+       signs[first] = 1;
+       half = isl_set_from_basic_set(nonneg_halfspace(isl_set_get_dim(set),
+                                                       1 + first));
+       half = isl_set_intersect(half, isl_set_copy(set));
+       if (foreach_orthant(half, signs, first + 1, len, fn, user) < 0)
+               goto error;
+
+       signs[first] = -1;
+       half = isl_set_from_basic_set(neg_halfspace(isl_set_get_dim(set),
+                                                       1 + first));
+       half = isl_set_intersect(half, set);
+       return foreach_orthant(half, signs, first + 1, len, fn, user);
+error:
+       isl_set_free(set);
+       return -1;
+}
+
+/* Call "fn" on the intersections of "set" with each of the orthants
+ * (except for obviously empty intersections).  The orthant is identified
+ * by the signs array, with each entry having value 1 or -1 according
+ * to the sign of the corresponding variable.
+ */
+int isl_set_foreach_orthant(__isl_keep isl_set *set,
+       int (*fn)(__isl_take isl_set *orthant, int *signs, void *user),
+       void *user)
+{
+       unsigned nparam;
+       unsigned nvar;
+       int *signs;
+       int r;
+
+       if (!set)
+               return -1;
+       if (isl_set_fast_is_empty(set))
+               return 0;
+
+       nparam = isl_set_dim(set, isl_dim_param);
+       nvar = isl_set_dim(set, isl_dim_set);
+
+       signs = isl_alloc_array(set->ctx, int, nparam + nvar);
+
+       r = foreach_orthant(isl_set_copy(set), signs, 0, nparam + nvar,
+                           fn, user);
+
+       free(signs);
+
+       return r;
+}
+
 int isl_set_is_equal(struct isl_set *set1, struct isl_set *set2)
 {
        return isl_map_is_equal((struct isl_map *)set1, (struct isl_map *)set2);
@@ -7920,6 +8011,22 @@ error:
        return NULL;
 }
 
+__isl_give isl_mat *isl_basic_set_equalities_matrix(
+       __isl_keep isl_basic_set *bset, enum isl_dim_type c1,
+       enum isl_dim_type c2, enum isl_dim_type c3, enum isl_dim_type c4)
+{
+       return isl_basic_map_equalities_matrix((isl_basic_map *)bset,
+                                               c1, c2, c3, c4, isl_dim_in);
+}
+
+__isl_give isl_mat *isl_basic_set_inequalities_matrix(
+       __isl_keep isl_basic_set *bset, enum isl_dim_type c1,
+       enum isl_dim_type c2, enum isl_dim_type c3, enum isl_dim_type c4)
+{
+       return isl_basic_map_inequalities_matrix((isl_basic_map *)bset,
+                                                c1, c2, c3, c4, isl_dim_in);
+}
+
 __isl_give isl_basic_set *isl_basic_set_from_constraint_matrices(
        __isl_take isl_dim *dim,
        __isl_take isl_mat *eq, __isl_take isl_mat *ineq, enum isl_dim_type c1,