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
19static
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
29static
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)(obj1: &pw, obj2: &ma);
43 ma = isl_multi_aff_align_divs(maff: ma);
44 n = FN(PW,n_piece)(pw);
45 if (n < 0 || !ma)
46 goto error;
47
48 space = isl_space_join(left: isl_multi_aff_get_space(multi: 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, pos: i);
56 domain = isl_set_preimage_multi_aff(set: domain,
57 ma: isl_multi_aff_copy(multi: ma));
58 pw = FN(PW,restore_domain_at)(pw, pos: i, domain);
59 el = FN(PW,take_base_at)(pw, pos: i);
60 el = FN(EL,pullback_multi_aff)(aff: el, ma: isl_multi_aff_copy(multi: ma));
61 pw = FN(PW,restore_base_at)(pw, pos: i, el);
62 }
63
64 pw = FN(PW,reset_space)(pw, space);
65 isl_multi_aff_free(multi: ma);
66 return pw;
67error:
68 isl_space_free(space);
69 isl_multi_aff_free(multi: 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 */
77static __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(left: 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 ma: isl_multi_aff_copy(multi: pma->p[0].maff));
98 res = FN(PW,intersect_domain)(pw: res, context: isl_set_copy(set: 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 ma: isl_multi_aff_copy(multi: pma->p[i].maff));
105 res_i = FN(PW,intersect_domain)(pw: res_i,
106 context: isl_set_copy(set: pma->p[i].set));
107 res = FN(PW,add_disjoint)(pw1: res, pw2: res_i);
108 }
109
110 isl_pw_multi_aff_free(pma);
111 FN(PW,free)(pw);
112 return res;
113error:
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)(obj1: &pw, obj2: &pma);
123 return FN(PW,pullback_pw_multi_aff_aligned)(pw, pma);
124}
125

source code of polly/lib/External/isl/isl_pw_pullback_templ.c