add isl_pw_multi_aff_scale_multi_val

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

diff --git a/doc/user.pod b/doc/user.pod
index eb28452..b2e2ada 100644 (file)
--- a/doc/user.pod
+++ b/doc/user.pod
@@ -4389,6 +4389,10 @@ C<isl_multi_aff_sub> subtracts the second argument from the first.
        __isl_give isl_multi_aff *isl_multi_aff_scale_multi_val(
                __isl_take isl_multi_aff *ma,
                __isl_take isl_multi_val *mv);
+       __isl_give isl_pw_multi_aff *
+       isl_pw_multi_aff_scale_multi_val(
+               __isl_take isl_pw_multi_aff *pma,
+               __isl_take isl_multi_val *mv);
        __isl_give isl_multi_pw_aff *
        isl_multi_pw_aff_scale_multi_val(
                __isl_take isl_multi_pw_aff *mpa,

diff --git a/include/isl/aff.h b/include/isl/aff.h
index 2a08198..1f82c96 100644 (file)
--- a/include/isl/aff.h
+++ b/include/isl/aff.h
@@ -409,6 +409,8 @@ __isl_give isl_pw_multi_aff *isl_pw_multi_aff_sub(
 
 __isl_give isl_pw_multi_aff *isl_pw_multi_aff_scale_val(
        __isl_take isl_pw_multi_aff *pma, __isl_take isl_val *v);
+__isl_give isl_pw_multi_aff *isl_pw_multi_aff_scale_multi_val(
+       __isl_take isl_pw_multi_aff *pma, __isl_take isl_multi_val *mv);
 __isl_give isl_pw_multi_aff *isl_pw_multi_aff_scale_vec(
        __isl_take isl_pw_multi_aff *pma, __isl_take isl_vec *v);
 

diff --git a/isl_aff.c b/isl_aff.c
index 0ccab13..1582908 100644 (file)
--- a/isl_aff.c
+++ b/isl_aff.c
@@ -5283,3 +5283,42 @@ error:
        isl_union_pw_multi_aff_free(upma);
        return NULL;
 }
+
+/* Scale the elements of "pma" by the corresponding elements of "mv".
+ */
+__isl_give isl_pw_multi_aff *isl_pw_multi_aff_scale_multi_val(
+       __isl_take isl_pw_multi_aff *pma, __isl_take isl_multi_val *mv)
+{
+       int i;
+
+       pma = isl_pw_multi_aff_cow(pma);
+       if (!pma || !mv)
+               goto error;
+       if (!isl_space_tuple_match(pma->dim, isl_dim_out,
+                                       mv->space, isl_dim_set))
+               isl_die(isl_pw_multi_aff_get_ctx(pma), isl_error_invalid,
+                       "spaces don't match", goto error);
+       if (!isl_space_match(pma->dim, isl_dim_param,
+                                       mv->space, isl_dim_param)) {
+               pma = isl_pw_multi_aff_align_params(pma,
+                                           isl_multi_val_get_space(mv));
+               mv = isl_multi_val_align_params(mv,
+                                           isl_pw_multi_aff_get_space(pma));
+               if (!pma || !mv)
+                       goto error;
+       }
+
+       for (i = 0; i < pma->n; ++i) {
+               pma->p[i].maff = isl_multi_aff_scale_multi_val(pma->p[i].maff,
+                                                       isl_multi_val_copy(mv));
+               if (!pma->p[i].maff)
+                       goto error;
+       }
+
+       isl_multi_val_free(mv);
+       return pma;
+error:
+       isl_multi_val_free(mv);
+       isl_pw_multi_aff_free(pma);
+       return NULL;
+}