isl_pw_pullback_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_pw_macro.h>
  11
  12 #undef SUFFIX
  13 #define SUFFIX  multi_aff
  14 #undef ARG1
  15 #define ARG1    PW
  16 #undef ARG2
  17 #define ARG2    isl_multi_aff
  18
  19 static
  20 #include "isl_align_params_templ.c"
  21
  22 #undef SUFFIX
  23 #define SUFFIX  pw_multi_aff
  24 #undef ARG1
  25 #define ARG1    PW
  26 #undef ARG2
  27 #define ARG2    isl_pw_multi_aff
  28
  29 static
  30 #include "isl_align_params_templ.c"
  31
  32 /* Compute the pullback of "pw" by the function represented by "ma".
  33  * In other words, plug in "ma" in "pw".
  34  */
  35 __isl_give PW *FN(PW,pullback_multi_aff)(__isl_take PW *pw,
  36         __isl_take isl_multi_aff *ma)
  37 {
  38         int i;
  39         isl_size n;
  40         isl_space *space = NULL;
  41
  42         FN(PW,align_params_multi_aff)(&pw, &ma);
  43         ma = isl_multi_aff_align_divs(ma);
  44         n = FN(PW,n_piece)(pw);
  45         if (n < 0 || !ma)
  46                 goto error;
  47
  48         space = isl_space_join(isl_multi_aff_get_space(ma),
  49                                 FN(PW,get_space)(pw));
  50
  51         for (i = 0; i < n; ++i) {
  52                 isl_set *domain;
  53                 EL *el;
  54
  55                 domain = FN(PW,take_domain_at)(pw, i);
  56                 domain = isl_set_preimage_multi_aff(domain,
  57                                                     isl_multi_aff_copy(ma));
  58                 pw = FN(PW,restore_domain_at)(pw, i, domain);
  59                 el = FN(PW,take_base_at)(pw, i);
  60                 el = FN(EL,pullback_multi_aff)(el, isl_multi_aff_copy(ma));
  61                 pw = FN(PW,restore_base_at)(pw, i, el);
  62         }
  63
  64         pw = FN(PW,reset_space)(pw, space);
  65         isl_multi_aff_free(ma);
  66         return pw;
  67 error:
  68         isl_space_free(space);
  69         isl_multi_aff_free(ma);
  70         FN(PW,free)(pw);
  71         return NULL;
  72 }
  73
  74 /* Compute the pullback of "pw" by the function represented by "pma".
  75  * In other words, plug in "pma" in "pw".
  76  */
  77 static __isl_give PW *FN(PW,pullback_pw_multi_aff_aligned)(__isl_take PW *pw,
  78         __isl_take isl_pw_multi_aff *pma)
  79 {
  80         int i;
  81         PW *res;
  82
  83         if (!pma)
  84                 goto error;
  85
  86         if (pma->n == 0) {
  87                 isl_space *space;
  88                 space = isl_space_join(isl_pw_multi_aff_get_space(pma),
  89                                         FN(PW,get_space)(pw));
  90                 isl_pw_multi_aff_free(pma);
  91                 res = FN(PW,empty)(space);
  92                 FN(PW,free)(pw);
  93                 return res;
  94         }
  95
  96         res = FN(PW,pullback_multi_aff)(FN(PW,copy)(pw),
  97                                         isl_multi_aff_copy(pma->p[0].maff));
  98         res = FN(PW,intersect_domain)(res, isl_set_copy(pma->p[0].set));
  99
 100         for (i = 1; i < pma->n; ++i) {
 101                 PW *res_i;
 102
 103                 res_i = FN(PW,pullback_multi_aff)(FN(PW,copy)(pw),
 104                                         isl_multi_aff_copy(pma->p[i].maff));
 105                 res_i = FN(PW,intersect_domain)(res_i,
 106                                         isl_set_copy(pma->p[i].set));
 107                 res = FN(PW,add_disjoint)(res, res_i);
 108         }
 109
 110         isl_pw_multi_aff_free(pma);
 111         FN(PW,free)(pw);
 112         return res;
 113 error:
 114         isl_pw_multi_aff_free(pma);
 115         FN(PW,free)(pw);
 116         return NULL;
 117 }
 118
 119 __isl_give PW *FN(PW,pullback_pw_multi_aff)(__isl_take PW *pw,
 120         __isl_take isl_pw_multi_aff *pma)
 121 {
 122         FN(PW,align_params_pw_multi_aff)(&pw, &pma);
 123         return FN(PW,pullback_pw_multi_aff_aligned)(pw, pma);
 124 }