add isl_pw_qpolynomial_fix_val

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

diff --git a/doc/user.pod b/doc/user.pod
index 7deae2d..13a4530 100644 (file)
--- a/doc/user.pod
+++ b/doc/user.pod
@@ -4790,6 +4790,10 @@ obviously equal, use
        __isl_give isl_qpolynomial *isl_qpolynomial_pow(
                __isl_take isl_qpolynomial *qp, unsigned exponent);
 
+       __isl_give isl_pw_qpolynomial *isl_pw_qpolynomial_fix_val(
+               __isl_take isl_pw_qpolynomial *pwqp,
+               enum isl_dim_type type, unsigned n,
+               __isl_take isl_val *v);
        __isl_give isl_pw_qpolynomial *
        isl_pw_qpolynomial_scale_val(
                __isl_take isl_pw_qpolynomial *pwqp,

diff --git a/include/isl/polynomial.h b/include/isl/polynomial.h
index 804ab7d..7aef73a 100644 (file)
--- a/include/isl/polynomial.h
+++ b/include/isl/polynomial.h
@@ -226,6 +226,9 @@ __isl_give isl_pw_qpolynomial *isl_pw_qpolynomial_move_dims(
 __isl_give isl_pw_qpolynomial *isl_pw_qpolynomial_fix_dim(
        __isl_take isl_pw_qpolynomial *pwqp,
        enum isl_dim_type type, unsigned n, isl_int v);
+__isl_give isl_pw_qpolynomial *isl_pw_qpolynomial_fix_val(
+       __isl_take isl_pw_qpolynomial *pwqp,
+       enum isl_dim_type type, unsigned n, __isl_take isl_val *v);
 
 __isl_give isl_qpolynomial *isl_pw_qpolynomial_eval(
        __isl_take isl_pw_qpolynomial *pwqp, __isl_take isl_point *pnt);

diff --git a/isl_fold.c b/isl_fold.c
index 4b7cd76..62ad065 100644 (file)
--- a/isl_fold.c
+++ b/isl_fold.c
@@ -17,6 +17,7 @@
 #include <isl/lp.h>
 #include <isl/seq.h>
 #include <isl_mat_private.h>
+#include <isl_val_private.h>
 #include <isl_config.h>
 
 enum isl_fold isl_fold_type_negate(enum isl_fold type)

diff --git a/isl_pw_templ.c b/isl_pw_templ.c
index 7a4e421..fb60651 100644 (file)
--- a/isl_pw_templ.c
+++ b/isl_pw_templ.c
@@ -1,4 +1,5 @@
 #include <isl/aff.h>
+#include <isl_val_private.h>
 
 #define xFN(TYPE,NAME) TYPE ## _ ## NAME
 #define FN(TYPE,NAME) xFN(TYPE,NAME)
@@ -1121,6 +1122,27 @@ error:
        return NULL;
 }
 
+/* Fix the value of the variable at position "pos" of type "type" of "pw"
+ * to be equal to "v".
+ */
+__isl_give PW *FN(PW,fix_val)(__isl_take PW *pw,
+       enum isl_dim_type type, unsigned pos, __isl_take isl_val *v)
+{
+       if (!v)
+               return FN(PW,free)(pw);
+       if (!isl_val_is_int(v))
+               isl_die(FN(PW,get_ctx)(pw), isl_error_invalid,
+                       "expecting integer value", goto error);
+
+       pw = FN(PW,fix_dim)(pw, type, pos, v->n);
+       isl_val_free(v);
+
+       return pw;
+error:
+       isl_val_free(v);
+       return FN(PW,free)(pw);
+}
+
 unsigned FN(PW,dim)(__isl_keep PW *pw, enum isl_dim_type type)
 {
        return pw ? isl_space_dim(pw->dim, type) : 0;