Home | History | Annotate | Line # | Download | only in dist 
      1 /*
      2  * Copyright 2011      Sven Verdoolaege
      3  * Copyright 2012-2013 Ecole Normale Superieure
      4  *
      5  * Use of this software is governed by the MIT license
      6  *
      7  * Written by Sven Verdoolaege,
      8  * Ecole Normale Superieure, 45 rue d'Ulm, 75230 Paris, France
      9  */
     10 
     11 #include <isl_space_private.h>
     12 
     13 #include <isl_multi_macro.h>
     14 
     15 /* Check whether "multi" has non-zero coefficients for any dimension
     16  * in the given range or if any of these dimensions appear
     17  * with non-zero coefficients in any of the integer divisions involved.
     18  */
     19 isl_bool FN(MULTI(BASE),involves_dims)(__isl_keep MULTI(BASE) *multi,
     20 	enum isl_dim_type type, unsigned first, unsigned n)
     21 {
     22 	int i;
     23 
     24 	if (!multi)
     25 		return isl_bool_error;
     26 	if (n == 0)
     27 		return isl_bool_false;
     28 
     29 	for (i = 0; i < multi->n; ++i) {
     30 		isl_bool involves;
     31 
     32 		involves = FN(EL,involves_dims)(multi->u.p[i], type, first, n);
     33 		if (involves < 0 || involves)
     34 			return involves;
     35 	}
     36 
     37 	if (FN(MULTI(BASE),has_explicit_domain)(multi))
     38 		return FN(MULTI(BASE),involves_explicit_domain_dims)(multi,
     39 								type, first, n);
     40 
     41 	return isl_bool_false;
     42 }
     43 
     44 __isl_give MULTI(BASE) *FN(MULTI(BASE),insert_dims)(
     45 	__isl_take MULTI(BASE) *multi,
     46 	enum isl_dim_type type, unsigned first, unsigned n)
     47 {
     48 	isl_space *space;
     49 	isl_size size;
     50 	int i;
     51 
     52 	size = FN(MULTI(BASE),size)(multi);
     53 	if (size < 0)
     54 		return FN(MULTI(BASE),free)(multi);
     55 	if (type == isl_dim_out)
     56 		isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
     57 			"cannot insert output/set dimensions",
     58 			return FN(MULTI(BASE),free)(multi));
     59 	if (n == 0 && !isl_space_is_named_or_nested(multi->space, type))
     60 		return multi;
     61 
     62 	space = FN(MULTI(BASE),take_space)(multi);
     63 	space = isl_space_insert_dims(space, type, first, n);
     64 	multi = FN(MULTI(BASE),restore_space)(multi, space);
     65 
     66 	if (FN(MULTI(BASE),has_explicit_domain)(multi))
     67 		multi = FN(MULTI(BASE),insert_explicit_domain_dims)(multi,
     68 								type, first, n);
     69 
     70 	for (i = 0; i < size; ++i) {
     71 		EL *el;
     72 
     73 		el = FN(MULTI(BASE),take_at)(multi, i);
     74 		el = FN(EL,insert_dims)(el, type, first, n);
     75 		multi = FN(MULTI(BASE),restore_at)(multi, i, el);
     76 	}
     77 
     78 	return multi;
     79 }
     80 
     81 __isl_give MULTI(BASE) *FN(MULTI(BASE),add_dims)(__isl_take MULTI(BASE) *multi,
     82 	enum isl_dim_type type, unsigned n)
     83 {
     84 	isl_size pos;
     85 
     86 	pos = FN(MULTI(BASE),dim)(multi, type);
     87 	if (pos < 0)
     88 		return FN(MULTI(BASE),free)(multi);
     89 
     90 	return FN(MULTI(BASE),insert_dims)(multi, type, pos, n);
     91 }
     92 
     93 /* Project the domain of "multi" onto its parameter space.
     94  * "multi" may not involve any of the domain dimensions.
     95  */
     96 __isl_give MULTI(BASE) *FN(MULTI(BASE),project_domain_on_params)(
     97 	__isl_take MULTI(BASE) *multi)
     98 {
     99 	isl_size n;
    100 	isl_bool involves;
    101 	isl_space *space;
    102 
    103 	n = FN(MULTI(BASE),dim)(multi, isl_dim_in);
    104 	if (n < 0)
    105 		return FN(MULTI(BASE),free)(multi);
    106 	involves = FN(MULTI(BASE),involves_dims)(multi, isl_dim_in, 0, n);
    107 	if (involves < 0)
    108 		return FN(MULTI(BASE),free)(multi);
    109 	if (involves)
    110 		isl_die(FN(MULTI(BASE),get_ctx)(multi), isl_error_invalid,
    111 		    "expression involves some of the domain dimensions",
    112 		    return FN(MULTI(BASE),free)(multi));
    113 	multi = FN(MULTI(BASE),drop_dims)(multi, isl_dim_in, 0, n);
    114 	space = FN(MULTI(BASE),get_domain_space)(multi);
    115 	space = isl_space_params(space);
    116 	multi = FN(MULTI(BASE),reset_domain_space)(multi, space);
    117 	return multi;
    118 }
    119