add isl_pw_aff_var_on_domain

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

diff --git a/doc/user.pod b/doc/user.pod
index 0d6e1a0..8acb32b 100644 (file)
--- a/doc/user.pod
+++ b/doc/user.pod
@@ -3155,6 +3155,9 @@ that is equal to a specified dimension on a given domain can be created using
        __isl_give isl_aff *isl_aff_var_on_domain(
                __isl_take isl_local_space *ls,
                enum isl_dim_type type, unsigned pos);
+       __isl_give isl_pw_aff *isl_pw_aff_var_on_domain(
+               __isl_take isl_local_space *ls,
+               enum isl_dim_type type, unsigned pos);
 
 Note that the space in which the resulting objects live is a map space
 with the given space as domain and a one-dimensional range.

diff --git a/include/isl/aff.h b/include/isl/aff.h
index 2ae0710..f799a08 100644 (file)
--- a/include/isl/aff.h
+++ b/include/isl/aff.h
@@ -122,6 +122,8 @@ __isl_give isl_pw_aff *isl_pw_aff_from_aff(__isl_take isl_aff *aff);
 __isl_give isl_pw_aff *isl_pw_aff_empty(__isl_take isl_space *dim);
 __isl_give isl_pw_aff *isl_pw_aff_alloc(__isl_take isl_set *set,
        __isl_take isl_aff *aff);
+__isl_give isl_pw_aff *isl_pw_aff_var_on_domain(__isl_take isl_local_space *ls,
+       enum isl_dim_type type, unsigned pos);
 
 __isl_give isl_pw_aff *isl_set_indicator_function(__isl_take isl_set *set);
 

diff --git a/isl_aff.c b/isl_aff.c
index 7017391..f9896c3 100644 (file)
--- a/isl_aff.c
+++ b/isl_aff.c
@@ -128,6 +128,15 @@ error:
        return NULL;
 }
 
+/* Return a piecewise affine expression that is equal to
+ * the specified dimension in "ls".
+ */
+__isl_give isl_pw_aff *isl_pw_aff_var_on_domain(__isl_take isl_local_space *ls,
+       enum isl_dim_type type, unsigned pos)
+{
+       return isl_pw_aff_from_aff(isl_aff_var_on_domain(ls, type, pos));
+}
+
 __isl_give isl_aff *isl_aff_copy(__isl_keep isl_aff *aff)
 {
        if (!aff)