add isl_pw_aff_n_piece

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

diff --git a/doc/user.pod b/doc/user.pod
index d164c1c..c4649c8 100644 (file)
--- a/doc/user.pod
+++ b/doc/user.pod
@@ -2835,6 +2835,7 @@ The expression can be inspected using
        __isl_give isl_aff *isl_aff_get_div(
                __isl_keep isl_aff *aff, int pos);
 
+       int isl_pw_aff_n_piece(__isl_keep isl_pw_aff *pwaff);
        int isl_pw_aff_foreach_piece(__isl_keep isl_pw_aff *pwaff,
                int (*fn)(__isl_take isl_set *set,
                          __isl_take isl_aff *aff,

diff --git a/include/isl/aff.h b/include/isl/aff.h
index d1e968f..6ceaea8 100644 (file)
--- a/include/isl/aff.h
+++ b/include/isl/aff.h
@@ -193,6 +193,7 @@ __isl_give isl_pw_aff *isl_pw_aff_gist(__isl_take isl_pw_aff *pwaff,
 __isl_give isl_pw_aff *isl_pw_aff_gist_params(__isl_take isl_pw_aff *pwaff,
        __isl_take isl_set *context);
 
+int isl_pw_aff_n_piece(__isl_keep isl_pw_aff *pwaff);
 int isl_pw_aff_foreach_piece(__isl_keep isl_pw_aff *pwaff,
        int (*fn)(__isl_take isl_set *set, __isl_take isl_aff *aff,
                    void *user), void *user);

diff --git a/isl_pw_templ.c b/isl_pw_templ.c
index 448b082..840a4fb 100644 (file)
--- a/isl_pw_templ.c
+++ b/isl_pw_templ.c
@@ -1298,6 +1298,11 @@ error:
 }
 #endif
 
+int FN(PW,n_piece)(__isl_keep PW *pw)
+{
+       return pw ? pw->n : 0;
+}
+
 int FN(PW,foreach_piece)(__isl_keep PW *pw,
        int (*fn)(__isl_take isl_set *set, __isl_take EL *el, void *user),
        void *user)