add isl_pw_aff_mod_val

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

diff --git a/doc/user.pod b/doc/user.pod
index dbbfc8d..7deae2d 100644 (file)
--- a/doc/user.pod
+++ b/doc/user.pod
@@ -3790,6 +3790,9 @@ Operations include
                __isl_take isl_val *mod);
        __isl_give isl_pw_aff *isl_pw_aff_mod(
                __isl_take isl_pw_aff *pwaff, isl_int mod);
+       __isl_give isl_pw_aff *isl_pw_aff_mod_val(
+               __isl_take isl_pw_aff *pa,
+               __isl_take isl_val *mod);
        __isl_give isl_aff *isl_aff_scale(__isl_take isl_aff *aff,
                isl_int f);
        __isl_give isl_aff *isl_aff_scale_val(__isl_take isl_aff *aff,

diff --git a/include/isl/aff.h b/include/isl/aff.h
index cb2d3da..68776cf 100644 (file)
--- a/include/isl/aff.h
+++ b/include/isl/aff.h
@@ -208,6 +208,8 @@ __isl_give isl_pw_aff *isl_pw_aff_ceil(__isl_take isl_pw_aff *pwaff);
 __isl_give isl_pw_aff *isl_pw_aff_floor(__isl_take isl_pw_aff *pwaff);
 __isl_give isl_pw_aff *isl_pw_aff_mod(__isl_take isl_pw_aff *pwaff,
        isl_int mod);
+__isl_give isl_pw_aff *isl_pw_aff_mod_val(__isl_take isl_pw_aff *pa,
+       __isl_take isl_val *mod);
 __isl_give isl_pw_aff *isl_pw_aff_tdiv_q(__isl_take isl_pw_aff *pa1,
        __isl_take isl_pw_aff *pa2);
 __isl_give isl_pw_aff *isl_pw_aff_tdiv_r(__isl_take isl_pw_aff *pa1,

diff --git a/isl_aff.c b/isl_aff.c
index d9f0f71..0ccab13 100644 (file)
--- a/isl_aff.c
+++ b/isl_aff.c
@@ -1357,6 +1357,29 @@ __isl_give isl_pw_aff *isl_pw_aff_mod(__isl_take isl_pw_aff *pwaff, isl_int m)
        return res;
 }
 
+/* Compute
+ *
+ *     pa mod m = pa - m * floor(pa/m)
+ *
+ * with m an integer value.
+ */
+__isl_give isl_pw_aff *isl_pw_aff_mod_val(__isl_take isl_pw_aff *pa,
+       __isl_take isl_val *m)
+{
+       if (!pa || !m)
+               goto error;
+       if (!isl_val_is_int(m))
+               isl_die(isl_pw_aff_get_ctx(pa), isl_error_invalid,
+                       "expecting integer modulo", goto error);
+       pa = isl_pw_aff_mod(pa, m->n);
+       isl_val_free(m);
+       return pa;
+error:
+       isl_pw_aff_free(pa);
+       isl_val_free(m);
+       return NULL;
+}
+
 /* Given f, return ceil(f).
  * If f is an integer expression, then just return f.
  * Otherwise, let f be the expression