add isl_pw_aff_scale_down_val

[platform/upstream/isl.git] / isl_pw_templ.c

diff --git a/isl_pw_templ.c b/isl_pw_templ.c
index 394571d..7a4e421 100644 (file)
--- a/isl_pw_templ.c
+++ b/isl_pw_templ.c
@@ -538,9 +538,9 @@ static __isl_give PW *FN(PW,on_shared_domain_in)(__isl_take PW *pw1,
 
        n = pw1->n * pw2->n;
 #ifdef HAS_TYPE
-       res = FN(PW,alloc_size)(space, pw1->type, n);
+       res = FN(PW,alloc_size)(isl_space_copy(space), pw1->type, n);
 #else
-       res = FN(PW,alloc_size)(space, n);
+       res = FN(PW,alloc_size)(isl_space_copy(space), n);
 #endif
 
        for (i = 0; i < pw1->n; ++i) {
@@ -568,6 +568,7 @@ static __isl_give PW *FN(PW,on_shared_domain_in)(__isl_take PW *pw1,
                }
        }
 
+       isl_space_free(space);
        FN(PW,free)(pw1);
        FN(PW,free)(pw2);
        return res;
@@ -982,12 +983,14 @@ __isl_give PW *FN(PW,drop_dims)(__isl_take PW *pw,
        if (!pw->dim)
                goto error;
        for (i = 0; i < pw->n; ++i) {
-               pw->p[i].set = isl_set_drop(pw->p[i].set, set_type, first, n);
-               if (!pw->p[i].set)
-                       goto error;
                pw->p[i].FIELD = FN(EL,drop_dims)(pw->p[i].FIELD, type, first, n);
                if (!pw->p[i].FIELD)
                        goto error;
+               if (type == isl_dim_out)
+                       continue;
+               pw->p[i].set = isl_set_drop(pw->p[i].set, set_type, first, n);
+               if (!pw->p[i].set)
+                       goto error;
        }
 
        return pw;
@@ -1521,6 +1524,58 @@ error:
        return NULL;
 }
 
+/* Multiply the pieces of "pw" by "v" and return the result.
+ */
+__isl_give PW *FN(PW,scale_val)(__isl_take PW *pw, __isl_take isl_val *v)
+{
+       int i;
+
+       if (!pw || !v)
+               goto error;
+
+       if (isl_val_is_one(v)) {
+               isl_val_free(v);
+               return pw;
+       }
+       if (pw && DEFAULT_IS_ZERO && isl_val_is_zero(v)) {
+               PW *zero;
+               isl_space *space = FN(PW,get_space)(pw);
+#ifdef HAS_TYPE
+               zero = FN(PW,ZERO)(space, pw->type);
+#else
+               zero = FN(PW,ZERO)(space);
+#endif
+               FN(PW,free)(pw);
+               isl_val_free(v);
+               return zero;
+       }
+       if (pw->n == 0) {
+               isl_val_free(v);
+               return pw;
+       }
+       pw = FN(PW,cow)(pw);
+       if (!pw)
+               goto error;
+
+#ifdef HAS_TYPE
+       if (isl_val_is_neg(v))
+               pw->type = isl_fold_type_negate(pw->type);
+#endif
+       for (i = 0; i < pw->n; ++i) {
+               pw->p[i].FIELD = FN(EL,scale_val)(pw->p[i].FIELD,
+                                                   isl_val_copy(v));
+               if (!pw->p[i].FIELD)
+                       goto error;
+       }
+
+       isl_val_free(v);
+       return pw;
+error:
+       isl_val_free(v);
+       FN(PW,free)(pw);
+       return NULL;
+}
+
 __isl_give PW *FN(PW,scale)(__isl_take PW *pw, isl_int v)
 {
        return FN(PW,mul_isl_int)(pw, v);