119 lines · c
1/*2 * Copyright 2011 Sven Verdoolaege3 * Copyright 2012-2013 Ecole Normale Superieure4 *5 * Use of this software is governed by the MIT license6 *7 * Written by Sven Verdoolaege,8 * Ecole Normale Superieure, 45 rue d'Ulm, 75230 Paris, France9 */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 dimension16 * in the given range or if any of these dimensions appear17 * with non-zero coefficients in any of the integer divisions involved.18 */19isl_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