1 | /* |
2 | * Copyright 2011 Sven Verdoolaege |
3 | * Copyright 2012-2013 Ecole Normale Superieure |
4 | * |
5 | * Use of this software is governed by the MIT license |
6 | * |
7 | * Written by Sven Verdoolaege, |
8 | * Ecole Normale Superieure, 45 rue d’Ulm, 75230 Paris, France |
9 | */ |
10 | |
11 | #include <isl_multi_macro.h> |
12 | |
13 | /* Does the space of "domain" correspond to that of the domain of "multi"? |
14 | * The parameters do not need to be aligned. |
15 | */ |
16 | static isl_bool FN(MULTI(BASE),compatible_domain)( |
17 | __isl_keep MULTI(BASE) *multi, __isl_keep DOM *domain) |
18 | { |
19 | isl_bool ok; |
20 | isl_space *space, *domain_space; |
21 | |
22 | domain_space = FN(DOM,get_space)(map: domain); |
23 | space = FN(MULTI(BASE),get_space)(multi); |
24 | ok = isl_space_has_domain_tuples(space1: domain_space, space2: space); |
25 | isl_space_free(space); |
26 | isl_space_free(space: domain_space); |
27 | |
28 | return ok; |
29 | } |
30 | |
31 | /* Check that the space of "domain" corresponds to |
32 | * that of the domain of "multi", ignoring parameters. |
33 | */ |
34 | static isl_stat FN(MULTI(BASE),check_compatible_domain)( |
35 | __isl_keep MULTI(BASE) *multi, __isl_keep DOM *domain) |
36 | { |
37 | isl_bool ok; |
38 | |
39 | ok = FN(MULTI(BASE),compatible_domain)(multi, domain); |
40 | if (ok < 0) |
41 | return isl_stat_error; |
42 | if (!ok) |
43 | isl_die(FN(DOM,get_ctx)(domain), isl_error_invalid, |
44 | "incompatible spaces" , return isl_stat_error); |
45 | |
46 | return isl_stat_ok; |
47 | } |
48 | |
49 | /* Intersect the explicit domain of "multi" with "domain". |
50 | * |
51 | * The parameters of "multi" and "domain" are assumed to have been aligned. |
52 | * |
53 | * In the case of an isl_multi_union_pw_aff object, the explicit domain |
54 | * is allowed to have only constraints on the parameters, while |
55 | * "domain" contains actual domain elements. In this case, |
56 | * "domain" is intersected with those parameter constraints and |
57 | * then used as the explicit domain of "multi". |
58 | */ |
59 | static __isl_give MULTI(BASE) *FN(MULTI(BASE),domain_intersect_aligned)( |
60 | __isl_take MULTI(BASE) *multi, __isl_take DOM *domain) |
61 | { |
62 | isl_bool is_params; |
63 | DOM *multi_dom; |
64 | |
65 | if (FN(MULTI(BASE),check_compatible_domain)(multi, domain) < 0) |
66 | goto error; |
67 | if (FN(MULTI(BASE),check_has_explicit_domain)(multi) < 0) |
68 | goto error; |
69 | is_params = FN(DOM,is_params)(map: multi->u.dom); |
70 | if (is_params < 0) |
71 | goto error; |
72 | multi_dom = FN(MULTI(BASE),get_explicit_domain)(multi); |
73 | if (!is_params) { |
74 | domain = FN(DOM,intersect)(map1: multi_dom, map2: domain); |
75 | } else { |
76 | isl_set *params; |
77 | |
78 | params = FN(DOM,params)(map: multi_dom); |
79 | domain = FN(DOM,intersect_params)(map: domain, params); |
80 | } |
81 | multi = FN(MULTI(BASE),set_explicit_domain)(multi, dom: domain); |
82 | return multi; |
83 | error: |
84 | FN(MULTI(BASE),free)(multi); |
85 | FN(DOM,free)(map: domain); |
86 | return NULL; |
87 | } |
88 | |
89 | /* Intersect the explicit domain of "multi" with "domain". |
90 | * First align the parameters, if needed. |
91 | */ |
92 | static __isl_give MULTI(BASE) *FN(MULTI(BASE),domain_intersect)( |
93 | __isl_take MULTI(BASE) *multi, __isl_take DOM *domain) |
94 | { |
95 | return FN(FN(MULTI(BASE),align_params),DOMBASE)(multi, domain, |
96 | FN(MULTI(BASE),domain_intersect_aligned)); |
97 | } |
98 | |
99 | /* Intersect the domain of "multi" with "domain". |
100 | * |
101 | * If "multi" has an explicit domain, then only this domain |
102 | * needs to be intersected. |
103 | */ |
104 | __isl_give MULTI(BASE) *FN(MULTI(BASE),intersect_domain)( |
105 | __isl_take MULTI(BASE) *multi, __isl_take DOM *domain) |
106 | { |
107 | if (FN(MULTI(BASE),has_explicit_domain)(multi)) |
108 | return FN(MULTI(BASE),domain_intersect)(multi, domain); |
109 | return FN(FN(MULTI(BASE),apply),DOMBASE)(multi, set: domain, |
110 | fn: &FN(EL,intersect_domain)); |
111 | } |
112 | |
113 | /* Intersect the parameter domain of the explicit domain of "multi" |
114 | * with "domain". |
115 | */ |
116 | static __isl_give MULTI(BASE) *FN(MULTI(BASE),domain_intersect_params_aligned)( |
117 | __isl_take MULTI(BASE) *multi, __isl_take isl_set *domain) |
118 | { |
119 | DOM *multi_dom; |
120 | |
121 | multi_dom = FN(MULTI(BASE),get_explicit_domain)(multi); |
122 | multi_dom = FN(DOM,intersect_params)(map: multi_dom, params: domain); |
123 | multi = FN(MULTI(BASE),set_explicit_domain)(multi, dom: multi_dom); |
124 | |
125 | return multi; |
126 | } |
127 | |
128 | /* Intersect the parameter domain of the explicit domain of "multi" |
129 | * with "domain". |
130 | * First align the parameters, if needed. |
131 | */ |
132 | static __isl_give MULTI(BASE) *FN(MULTI(BASE),domain_intersect_params)( |
133 | __isl_take MULTI(BASE) *multi, __isl_take isl_set *domain) |
134 | { |
135 | return FN(FN(MULTI(BASE),align_params),set)(multi, domain, |
136 | FN(MULTI(BASE),domain_intersect_params_aligned)); |
137 | } |
138 | |
139 | /* Intersect the parameter domain of "multi" with "domain". |
140 | * |
141 | * If "multi" has an explicit domain, then only this domain |
142 | * needs to be intersected. |
143 | */ |
144 | __isl_give MULTI(BASE) *FN(MULTI(BASE),intersect_params)( |
145 | __isl_take MULTI(BASE) *multi, __isl_take isl_set *domain) |
146 | { |
147 | if (FN(MULTI(BASE),has_explicit_domain)(multi)) |
148 | return FN(MULTI(BASE),domain_intersect_params)(multi, domain); |
149 | return FN(MULTI(BASE),apply_set)(multi, set: domain, |
150 | fn: &FN(EL,intersect_params)); |
151 | } |
152 | |