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 */
FN(PW,pullback_multi_aff)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 */
FN(PW,pullback_pw_multi_aff_aligned)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
FN(PW,pullback_pw_multi_aff)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 }
125