add isl_qpolynomial_project_domain_on_params

Signed-off-by: Sven Verdoolaege <skimo@kotnet.org>

diff --git a/doc/user.pod b/doc/user.pod
index 5ffbf66..9fd3b4b 100644 (file)
--- a/doc/user.pod
+++ b/doc/user.pod
@@ -3124,6 +3124,9 @@ are returned in C<*n> and C<*d>, respectively.
                __isl_take isl_qpolynomial *qp,
                __isl_take isl_space *model);
 
+       __isl_give isl_qpolynomial *isl_qpolynomial_project_domain_on_params(
+               __isl_take isl_qpolynomial *qp);
+
        __isl_give isl_union_pw_qpolynomial *isl_union_pw_qpolynomial_coalesce(
                __isl_take isl_union_pw_qpolynomial *upwqp);
 

diff --git a/include/isl/polynomial.h b/include/isl/polynomial.h
index c3a177a..05f034b 100644 (file)
--- a/include/isl/polynomial.h
+++ b/include/isl/polynomial.h
@@ -78,6 +78,8 @@ __isl_give isl_qpolynomial *isl_qpolynomial_move_dims(
        __isl_take isl_qpolynomial *qp,
        enum isl_dim_type dst_type, unsigned dst_pos,
        enum isl_dim_type src_type, unsigned src_pos, unsigned n);
+__isl_give isl_qpolynomial *isl_qpolynomial_project_domain_on_params(
+       __isl_take isl_qpolynomial *qp);
 __isl_give isl_qpolynomial *isl_qpolynomial_drop_dims(
        __isl_take isl_qpolynomial *qp,
        enum isl_dim_type type, unsigned first, unsigned n);

diff --git a/isl_polynomial.c b/isl_polynomial.c
index 1597a63..2ed05a0 100644 (file)
--- a/isl_polynomial.c
+++ b/isl_polynomial.c
@@ -2407,6 +2407,31 @@ error:
        return NULL;
 }
 
+/* Project the domain of the quasi-polynomial onto its parameter space.
+ * The quasi-polynomial may not involve any of the domain dimensions.
+ */
+__isl_give isl_qpolynomial *isl_qpolynomial_project_domain_on_params(
+       __isl_take isl_qpolynomial *qp)
+{
+       isl_space *space;
+       unsigned n;
+       int involves;
+
+       n = isl_qpolynomial_dim(qp, isl_dim_set);
+       involves = isl_qpolynomial_involves_dims(qp, isl_dim_set, 0, n);
+       if (involves < 0)
+               return isl_qpolynomial_free(qp);
+       if (involves)
+               isl_die(isl_qpolynomial_get_ctx(qp), isl_error_invalid,
+                       "polynomial involves some of the domain dimensions",
+                       return isl_qpolynomial_free(qp));
+       qp = isl_qpolynomial_drop_dims(qp, isl_dim_set, 0, n);
+       space = isl_qpolynomial_get_space(qp);
+       space = isl_space_params(space);
+       qp = isl_qpolynomial_reset_space(qp, space);
+       return qp;
+}
+
 static __isl_give isl_qpolynomial *isl_qpolynomial_substitute_equalities_lifted(
        __isl_take isl_qpolynomial *qp, __isl_take isl_basic_set *eq)
 {