isl_multi_identity_templ.c

   1 /*
   2  * Copyright 2012      Ecole Normale Superieure
   3  *
   4  * Use of this software is governed by the MIT license
   5  *
   6  * Written by Sven Verdoolaege,
   7  * Ecole Normale Superieure, 45 rue d'Ulm, 75230 Paris, France
   8  */
   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 each
  16  * 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;
  61 error:
  62         isl_space_free(space);
  63         return NULL;
  64 }