97 lines · c
1/*2 * Copyright 2012 Ecole Normale Superieure3 *4 * Use of this software is governed by the MIT license5 *6 * Written by Sven Verdoolaege,7 * Ecole Normale Superieure, 45 rue d'Ulm, 75230 Paris, France8 */9 10#include <isl/space.h>11#include <isl/local_space.h>12 13#include <isl_multi_macro.h>14 15/* Create a multi expression in the given space that maps each16 * input dimension to the corresponding output dimension.17 */18__isl_give MULTI(BASE) *FN(MULTI(BASE),identity)(__isl_take isl_space *space)19{20 int i;21 isl_size n_in, n_out;22 isl_local_space *ls;23 MULTI(BASE) *multi;24 25 if (!space)26 return NULL;27 28 if (isl_space_is_set(space))29 isl_die(isl_space_get_ctx(space), isl_error_invalid,30 "expecting map space", goto error);31 32 n_in = isl_space_dim(space, isl_dim_in);33 n_out = isl_space_dim(space, isl_dim_out);34 if (n_in < 0 || n_out < 0)35 goto error;36 if (n_in != n_out)37 isl_die(isl_space_get_ctx(space), isl_error_invalid,38 "number of input and output dimensions needs to be "39 "the same", goto error);40 41 multi = FN(MULTI(BASE),alloc)(isl_space_copy(space));42 43 if (!n_out) {44 isl_space_free(space);45 return multi;46 }47 48 space = isl_space_domain(space);49 ls = isl_local_space_from_space(space);50 51 for (i = 0; i < n_out; ++i) {52 EL *el;53 el = FN(EL,var_on_domain)(isl_local_space_copy(ls),54 isl_dim_set, i);55 multi = FN(FN(MULTI(BASE),set),BASE)(multi, i, el);56 }57 58 isl_local_space_free(ls);59 60 return multi;61error:62 isl_space_free(space);63 return NULL;64}65 66/* Create a multi expression that maps elements in the given space67 * to themselves.68 */69__isl_give MULTI(BASE) *FN(MULTI(BASE),identity_on_domain_space)(70 __isl_take isl_space *space)71{72 return FN(MULTI(BASE),identity)(isl_space_map_from_set(space));73}74 75/* This function performs the same operation as76 * isl_multi_*_identity_on_domain_space,77 * but is considered as a function on an isl_space when exported.78 */79__isl_give MULTI(BASE) *FN(FN(isl_space_identity_multi,BASE),on_domain)(80 __isl_take isl_space *space)81{82 return FN(MULTI(BASE),identity_on_domain_space)(space);83}84 85/* Create a multi expression in the same space as "multi" that maps each86 * input dimension to the corresponding output dimension.87 */88__isl_give MULTI(BASE) *FN(FN(MULTI(BASE),identity_multi),BASE)(89 __isl_take MULTI(BASE) *multi)90{91 isl_space *space;92 93 space = FN(MULTI(BASE),get_space)(multi);94 FN(MULTI(BASE),free)(multi);95 return FN(MULTI(BASE),identity)(space);96}97