add isl_qpolynomial_fold_foreach_qpolynomial

diff --git a/include/isl_polynomial.h b/include/isl_polynomial.h
index a39ecc4..c75a88a 100644 (file)
--- a/include/isl_polynomial.h
+++ b/include/isl_polynomial.h
@@ -215,6 +215,10 @@ __isl_give isl_qpolynomial_fold *isl_qpolynomial_fold_move_dims(
 __isl_give isl_qpolynomial *isl_qpolynomial_fold_eval(
        __isl_take isl_qpolynomial_fold *fold, __isl_take isl_point *pnt);
 
+int isl_qpolynomial_fold_foreach_qpolynomial(
+       __isl_keep isl_qpolynomial_fold *fold,
+       int (*fn)(__isl_take isl_qpolynomial *qp, void *user), void *user);
+
 void isl_qpolynomial_fold_print(__isl_keep isl_qpolynomial_fold *fold, FILE *out,
        unsigned output_format);
 

diff --git a/isl_fold.c b/isl_fold.c
index a0d241b..f04b4d4 100644 (file)
--- a/isl_fold.c
+++ b/isl_fold.c
@@ -803,6 +803,22 @@ error:
        return NULL;
 }
 
+int isl_qpolynomial_fold_foreach_qpolynomial(
+       __isl_keep isl_qpolynomial_fold *fold,
+       int (*fn)(__isl_take isl_qpolynomial *qp, void *user), void *user)
+{
+       int i;
+
+       if (!fold)
+               return -1;
+
+       for (i = 0; i < fold->n; ++i)
+               if (fn(isl_qpolynomial_copy(fold->qp[i]), user) < 0)
+                       return -1;
+
+       return 0;
+}
+
 __isl_give isl_qpolynomial_fold *isl_qpolynomial_fold_move_dims(
        __isl_take isl_qpolynomial_fold *fold,
        enum isl_dim_type dst_type, unsigned dst_pos,