1 /*
2 * Copyright 2017 Sven Verdoolaege
3 *
4 * Use of this software is governed by the MIT license
5 *
6 * Written by Sven Verdoolaege.
7 */
8
9 /* Initialize the explicit domain of "mpa".
10 *
11 * The explicit domain is initialized to a universe set
12 * in the domain space.
13 */
isl_multi_pw_aff_init_explicit_domain(__isl_take isl_multi_pw_aff * mpa)14 static __isl_give isl_multi_pw_aff *isl_multi_pw_aff_init_explicit_domain(
15 __isl_take isl_multi_pw_aff *mpa)
16 {
17 if (isl_multi_pw_aff_check_has_explicit_domain(mpa) < 0)
18 return isl_multi_pw_aff_free(mpa);
19 mpa->u.dom = isl_set_universe(isl_multi_pw_aff_get_domain_space(mpa));
20 if (!mpa->u.dom)
21 return isl_multi_pw_aff_free(mpa);
22 return mpa;
23 }
24
25 /* Intersect the domain of "dst" with the domain product
26 * of the explicit domains of "src1" and "src2".
27 * This function is only called if at least one of "src1" or "src2"
28 * has an explicit domain.
29 */
30 static __isl_give isl_multi_pw_aff *
isl_multi_pw_aff_intersect_explicit_domain_product(__isl_take isl_multi_pw_aff * dst,__isl_keep isl_multi_pw_aff * src1,__isl_keep isl_multi_pw_aff * src2)31 isl_multi_pw_aff_intersect_explicit_domain_product(
32 __isl_take isl_multi_pw_aff *dst, __isl_keep isl_multi_pw_aff *src1,
33 __isl_keep isl_multi_pw_aff *src2)
34 {
35 isl_space *space;
36 isl_set *dom;
37 isl_map *map;
38
39 if (!src1 || !src2)
40 return FN(isl_multi_pw_aff,free)(dst);
41 space = isl_multi_pw_aff_get_domain_space(dst);
42 dom = isl_set_universe(space);
43 map = isl_set_unwrap(dom);
44 if (isl_multi_pw_aff_has_explicit_domain(src1)) {
45 dom = isl_set_copy(src1->u.dom);
46 map = isl_map_intersect_domain(map, dom);
47 }
48 if (isl_multi_pw_aff_has_explicit_domain(src2)) {
49 dom = isl_set_copy(src2->u.dom);
50 map = isl_map_intersect_range(map, dom);
51 }
52 dom = isl_map_wrap(map);
53 dst = isl_multi_pw_aff_intersect_domain(dst, dom);
54 return dst;
55 }
56
57 /* Check whether the explicit domain of "mpa" has non-zero coefficients
58 * for any dimension in the given range or if any of these dimensions appear
59 * with non-zero coefficients in any of the integer divisions involved.
60 */
isl_multi_pw_aff_involves_explicit_domain_dims(__isl_keep isl_multi_pw_aff * mpa,enum isl_dim_type type,unsigned pos,unsigned n)61 isl_bool isl_multi_pw_aff_involves_explicit_domain_dims(
62 __isl_keep isl_multi_pw_aff *mpa,
63 enum isl_dim_type type, unsigned pos, unsigned n)
64 {
65 if (isl_multi_pw_aff_check_has_explicit_domain(mpa) < 0)
66 return isl_bool_error;
67 if (type == isl_dim_in)
68 type = isl_dim_set;
69 return isl_set_involves_dims(mpa->u.dom, type, pos, n);
70 }
71
72 /* Insert "n" dimensions of type "type" at position "pos"
73 * of the explicit domain of "mpa".
74 */
75 static __isl_give isl_multi_pw_aff *
isl_multi_pw_aff_insert_explicit_domain_dims(__isl_take isl_multi_pw_aff * mpa,enum isl_dim_type type,unsigned pos,unsigned n)76 isl_multi_pw_aff_insert_explicit_domain_dims(__isl_take isl_multi_pw_aff *mpa,
77 enum isl_dim_type type, unsigned pos, unsigned n)
78 {
79 if (isl_multi_pw_aff_check_has_explicit_domain(mpa) < 0)
80 return isl_multi_pw_aff_free(mpa);
81 mpa = isl_multi_pw_aff_cow(mpa);
82 if (!mpa)
83 return NULL;
84 if (type == isl_dim_in)
85 type = isl_dim_set;
86 mpa->u.dom = isl_set_insert_dims(mpa->u.dom, type, pos, n);
87 if (!mpa->u.dom)
88 return isl_multi_pw_aff_free(mpa);
89 return mpa;
90 }
91
92 /* Drop the "n" dimensions of type "type" starting at position "pos"
93 * of the explicit domain of "mpa".
94 */
95 static __isl_give isl_multi_pw_aff *
isl_multi_pw_aff_drop_explicit_domain_dims(__isl_take isl_multi_pw_aff * mpa,enum isl_dim_type type,unsigned pos,unsigned n)96 isl_multi_pw_aff_drop_explicit_domain_dims(__isl_take isl_multi_pw_aff *mpa,
97 enum isl_dim_type type, unsigned pos, unsigned n)
98 {
99 if (isl_multi_pw_aff_check_has_explicit_domain(mpa) < 0)
100 return isl_multi_pw_aff_free(mpa);
101 mpa = isl_multi_pw_aff_cow(mpa);
102 if (!mpa)
103 return NULL;
104 if (type == isl_dim_in)
105 type = isl_dim_set;
106 mpa->u.dom = isl_set_drop(mpa->u.dom, type, pos, n);
107 if (!mpa->u.dom)
108 return isl_multi_pw_aff_free(mpa);
109 return mpa;
110 }
111
112 /* Move the "n" dimensions of "src_type" starting at "src_pos" of
113 * of the explicit domain of "mpa" to dimensions of "dst_type" at "dst_pos".
114 */
isl_multi_pw_aff_move_explicit_domain_dims(__isl_take isl_multi_pw_aff * mpa,enum isl_dim_type dst_type,unsigned dst_pos,enum isl_dim_type src_type,unsigned src_pos,unsigned n)115 static __isl_give isl_multi_pw_aff *isl_multi_pw_aff_move_explicit_domain_dims(
116 __isl_take isl_multi_pw_aff *mpa,
117 enum isl_dim_type dst_type, unsigned dst_pos,
118 enum isl_dim_type src_type, unsigned src_pos, unsigned n)
119 {
120 if (isl_multi_pw_aff_check_has_explicit_domain(mpa) < 0)
121 return isl_multi_pw_aff_free(mpa);
122 mpa = isl_multi_pw_aff_cow(mpa);
123 if (!mpa)
124 return NULL;
125 if (dst_type == isl_dim_in)
126 dst_type = isl_dim_set;
127 if (src_type == isl_dim_in)
128 src_type = isl_dim_set;
129 mpa->u.dom = isl_set_move_dims(mpa->u.dom, dst_type, dst_pos,
130 src_type, src_pos, n);
131 if (!mpa->u.dom)
132 return isl_multi_pw_aff_free(mpa);
133 return mpa;
134 }
135