xref: /llvm-project/polly/lib/External/isl/isl_map.c (revision a749e09e184b2b0b6dde71af01c82dd427b3e3e2)
1 /*
2  * Copyright 2008-2009 Katholieke Universiteit Leuven
3  * Copyright 2010      INRIA Saclay
4  * Copyright 2012-2014 Ecole Normale Superieure
5  * Copyright 2014      INRIA Rocquencourt
6  * Copyright 2016      INRIA Paris
7  * Copyright 2016      Sven Verdoolaege
8  * Copyright 2018-2019 Cerebras Systems
9  *
10  * Use of this software is governed by the MIT license
11  *
12  * Written by Sven Verdoolaege, K.U.Leuven, Departement
13  * Computerwetenschappen, Celestijnenlaan 200A, B-3001 Leuven, Belgium
14  * and INRIA Saclay - Ile-de-France, Parc Club Orsay Universite,
15  * ZAC des vignes, 4 rue Jacques Monod, 91893 Orsay, France
16  * and Ecole Normale Superieure, 45 rue d’Ulm, 75230 Paris, France
17  * and Inria Paris - Rocquencourt, Domaine de Voluceau - Rocquencourt,
18  * B.P. 105 - 78153 Le Chesnay, France
19  * and Centre de Recherche Inria de Paris, 2 rue Simone Iff - Voie DQ12,
20  * CS 42112, 75589 Paris Cedex 12, France
21  * and Cerebras Systems, 175 S San Antonio Rd, Los Altos, CA, USA
22  */
23 
24 #include <string.h>
25 #include <isl_ctx_private.h>
26 #include <isl_map_private.h>
27 #include <isl_blk.h>
28 #include <isl_id_private.h>
29 #include <isl/constraint.h>
30 #include "isl_space_private.h"
31 #include "isl_equalities.h"
32 #include <isl_lp_private.h>
33 #include <isl_seq.h>
34 #include <isl/set.h>
35 #include <isl/map.h>
36 #include <isl_reordering.h>
37 #include "isl_sample.h"
38 #include <isl_sort.h>
39 #include "isl_tab.h"
40 #include <isl/vec.h>
41 #include <isl_mat_private.h>
42 #include <isl_vec_private.h>
43 #include <isl_dim_map.h>
44 #include <isl_local_space_private.h>
45 #include <isl_aff_private.h>
46 #include <isl_options_private.h>
47 #include <isl_morph.h>
48 #include <isl_val_private.h>
49 #include <isl_printer_private.h>
50 
51 #include <bset_to_bmap.c>
52 #include <bset_from_bmap.c>
53 #include <set_to_map.c>
54 #include <set_from_map.c>
55 
56 /* Treat "bset" as a basic map.
57  * Internally, isl_basic_set is defined to isl_basic_map, so in practice,
58  * this function performs a redundant cast.
59  */
const_bset_to_bmap(__isl_keep const isl_basic_set * bset)60 static __isl_keep const isl_basic_map *const_bset_to_bmap(
61 	__isl_keep const isl_basic_set *bset)
62 {
63 	return (const isl_basic_map *) bset;
64 }
65 
66 #undef TYPE
67 #define TYPE	isl_basic_map
68 #include "has_single_reference_templ.c"
69 
pos(__isl_keep isl_space * space,enum isl_dim_type type)70 static unsigned pos(__isl_keep isl_space *space, enum isl_dim_type type)
71 {
72 	switch (type) {
73 	case isl_dim_param:	return 1;
74 	case isl_dim_in:	return 1 + space->nparam;
75 	case isl_dim_out:	return 1 + space->nparam + space->n_in;
76 	default:		return 0;
77 	}
78 }
79 
isl_basic_map_dim(__isl_keep isl_basic_map * bmap,enum isl_dim_type type)80 isl_size isl_basic_map_dim(__isl_keep isl_basic_map *bmap,
81 				enum isl_dim_type type)
82 {
83 	if (!bmap)
84 		return isl_size_error;
85 	switch (type) {
86 	case isl_dim_cst:	return 1;
87 	case isl_dim_param:
88 	case isl_dim_in:
89 	case isl_dim_out:	return isl_space_dim(bmap->dim, type);
90 	case isl_dim_div:	return bmap->n_div;
91 	case isl_dim_all:	return isl_basic_map_total_dim(bmap);
92 	default:		return 0;
93 	}
94 }
95 
96 /* Return the space of "map".
97  */
isl_map_peek_space(__isl_keep const isl_map * map)98 __isl_keep isl_space *isl_map_peek_space(__isl_keep const isl_map *map)
99 {
100 	return map ? map->dim : NULL;
101 }
102 
103 /* Return the space of "set".
104  */
isl_set_peek_space(__isl_keep isl_set * set)105 __isl_keep isl_space *isl_set_peek_space(__isl_keep isl_set *set)
106 {
107 	return isl_map_peek_space(set_to_map(set));
108 }
109 
isl_map_dim(__isl_keep isl_map * map,enum isl_dim_type type)110 isl_size isl_map_dim(__isl_keep isl_map *map, enum isl_dim_type type)
111 {
112 	return isl_space_dim(isl_map_peek_space(map), type);
113 }
114 
115 /* Return the dimensionality of the domain (tuple) of the map.
116  */
isl_map_domain_tuple_dim(__isl_keep isl_map * map)117 isl_size isl_map_domain_tuple_dim(__isl_keep isl_map *map)
118 {
119 	return isl_map_dim(map, isl_dim_in);
120 }
121 
122 /* Return the dimensionality of the range (tuple) of the map.
123  */
isl_map_range_tuple_dim(__isl_keep isl_map * map)124 isl_size isl_map_range_tuple_dim(__isl_keep isl_map *map)
125 {
126 	return isl_map_dim(map, isl_dim_out);
127 }
128 
isl_set_dim(__isl_keep isl_set * set,enum isl_dim_type type)129 isl_size isl_set_dim(__isl_keep isl_set *set, enum isl_dim_type type)
130 {
131 	return isl_map_dim(set_to_map(set), type);
132 }
133 
134 /* Return the dimensionality of the (tuple of the) set.
135  */
isl_set_tuple_dim(__isl_keep isl_set * set)136 isl_size isl_set_tuple_dim(__isl_keep isl_set *set)
137 {
138 	return isl_set_dim(set, isl_dim_set);
139 }
140 
141 /* Return the position of the variables of the given type
142  * within the sequence of variables of "bmap".
143  */
isl_basic_map_var_offset(__isl_keep isl_basic_map * bmap,enum isl_dim_type type)144 isl_size isl_basic_map_var_offset(__isl_keep isl_basic_map *bmap,
145 	enum isl_dim_type type)
146 {
147 	isl_space *space;
148 
149 	space = isl_basic_map_peek_space(bmap);
150 	if (!space)
151 		return isl_size_error;
152 
153 	switch (type) {
154 	case isl_dim_param:
155 	case isl_dim_in:
156 	case isl_dim_out:	return isl_space_offset(space, type);
157 	case isl_dim_div:	return isl_space_dim(space, isl_dim_all);
158 	case isl_dim_cst:
159 	default:
160 		isl_die(isl_basic_map_get_ctx(bmap), isl_error_invalid,
161 			"invalid dimension type", return isl_size_error);
162 	}
163 }
164 
165 /* Return the position of the variables of the given type
166  * within the sequence of variables of "bset".
167  */
isl_basic_set_var_offset(__isl_keep isl_basic_set * bset,enum isl_dim_type type)168 isl_size isl_basic_set_var_offset(__isl_keep isl_basic_set *bset,
169 	enum isl_dim_type type)
170 {
171 	return isl_basic_map_var_offset(bset_to_bmap(bset), type);
172 }
173 
174 /* Return the position of the coefficients of the variables of the given type
175  * within the sequence of coefficients of "bmap".
176  */
isl_basic_map_offset(__isl_keep isl_basic_map * bmap,enum isl_dim_type type)177 unsigned isl_basic_map_offset(__isl_keep isl_basic_map *bmap,
178 	enum isl_dim_type type)
179 {
180 	switch (type) {
181 	case isl_dim_cst:	return 0;
182 	case isl_dim_param:
183 	case isl_dim_in:
184 	case isl_dim_out:
185 	case isl_dim_div:	return 1 + isl_basic_map_var_offset(bmap, type);
186 	default:		return 0;
187 	}
188 }
189 
isl_basic_set_offset(__isl_keep isl_basic_set * bset,enum isl_dim_type type)190 unsigned isl_basic_set_offset(__isl_keep isl_basic_set *bset,
191 					enum isl_dim_type type)
192 {
193 	return isl_basic_map_offset(bset, type);
194 }
195 
map_offset(__isl_keep isl_map * map,enum isl_dim_type type)196 static unsigned map_offset(__isl_keep isl_map *map, enum isl_dim_type type)
197 {
198 	return pos(map->dim, type);
199 }
200 
isl_basic_set_dim(__isl_keep isl_basic_set * bset,enum isl_dim_type type)201 isl_size isl_basic_set_dim(__isl_keep isl_basic_set *bset,
202 				enum isl_dim_type type)
203 {
204 	return isl_basic_map_dim(bset, type);
205 }
206 
isl_basic_set_n_dim(__isl_keep isl_basic_set * bset)207 isl_size isl_basic_set_n_dim(__isl_keep isl_basic_set *bset)
208 {
209 	return isl_basic_set_dim(bset, isl_dim_set);
210 }
211 
isl_basic_set_n_param(__isl_keep isl_basic_set * bset)212 isl_size isl_basic_set_n_param(__isl_keep isl_basic_set *bset)
213 {
214 	return isl_basic_set_dim(bset, isl_dim_param);
215 }
216 
isl_basic_set_total_dim(__isl_keep const isl_basic_set * bset)217 isl_size isl_basic_set_total_dim(__isl_keep const isl_basic_set *bset)
218 {
219 	return isl_basic_map_total_dim(const_bset_to_bmap(bset));
220 }
221 
isl_set_n_dim(__isl_keep isl_set * set)222 isl_size isl_set_n_dim(__isl_keep isl_set *set)
223 {
224 	return isl_set_dim(set, isl_dim_set);
225 }
226 
isl_set_n_param(__isl_keep isl_set * set)227 isl_size isl_set_n_param(__isl_keep isl_set *set)
228 {
229 	return isl_set_dim(set, isl_dim_param);
230 }
231 
isl_basic_map_total_dim(__isl_keep const isl_basic_map * bmap)232 isl_size isl_basic_map_total_dim(__isl_keep const isl_basic_map *bmap)
233 {
234 	isl_size dim;
235 
236 	if (!bmap)
237 		return isl_size_error;
238 	dim = isl_space_dim(bmap->dim, isl_dim_all);
239 	if (dim < 0)
240 		return isl_size_error;
241 	return dim + bmap->n_div;
242 }
243 
244 /* Return the number of equality constraints in the description of "bmap".
245  * Return isl_size_error on error.
246  */
isl_basic_map_n_equality(__isl_keep isl_basic_map * bmap)247 isl_size isl_basic_map_n_equality(__isl_keep isl_basic_map *bmap)
248 {
249 	if (!bmap)
250 		return isl_size_error;
251 	return bmap->n_eq;
252 }
253 
254 /* Return the number of equality constraints in the description of "bset".
255  * Return isl_size_error on error.
256  */
isl_basic_set_n_equality(__isl_keep isl_basic_set * bset)257 isl_size isl_basic_set_n_equality(__isl_keep isl_basic_set *bset)
258 {
259 	return isl_basic_map_n_equality(bset_to_bmap(bset));
260 }
261 
262 /* Return the number of inequality constraints in the description of "bmap".
263  * Return isl_size_error on error.
264  */
isl_basic_map_n_inequality(__isl_keep isl_basic_map * bmap)265 isl_size isl_basic_map_n_inequality(__isl_keep isl_basic_map *bmap)
266 {
267 	if (!bmap)
268 		return isl_size_error;
269 	return bmap->n_ineq;
270 }
271 
272 /* Return the number of inequality constraints in the description of "bset".
273  * Return isl_size_error on error.
274  */
isl_basic_set_n_inequality(__isl_keep isl_basic_set * bset)275 isl_size isl_basic_set_n_inequality(__isl_keep isl_basic_set *bset)
276 {
277 	return isl_basic_map_n_inequality(bset_to_bmap(bset));
278 }
279 
280 /* Do "bmap1" and "bmap2" have the same parameters?
281  */
isl_basic_map_has_equal_params(__isl_keep isl_basic_map * bmap1,__isl_keep isl_basic_map * bmap2)282 static isl_bool isl_basic_map_has_equal_params(__isl_keep isl_basic_map *bmap1,
283 	__isl_keep isl_basic_map *bmap2)
284 {
285 	isl_space *space1, *space2;
286 
287 	space1 = isl_basic_map_peek_space(bmap1);
288 	space2 = isl_basic_map_peek_space(bmap2);
289 	return isl_space_has_equal_params(space1, space2);
290 }
291 
292 /* Do "map1" and "map2" have the same parameters?
293  */
isl_map_has_equal_params(__isl_keep isl_map * map1,__isl_keep isl_map * map2)294 isl_bool isl_map_has_equal_params(__isl_keep isl_map *map1,
295 	__isl_keep isl_map *map2)
296 {
297 	isl_space *space1, *space2;
298 
299 	space1 = isl_map_peek_space(map1);
300 	space2 = isl_map_peek_space(map2);
301 	return isl_space_has_equal_params(space1, space2);
302 }
303 
304 /* Do "map" and "set" have the same parameters?
305  */
isl_map_set_has_equal_params(__isl_keep isl_map * map,__isl_keep isl_set * set)306 static isl_bool isl_map_set_has_equal_params(__isl_keep isl_map *map,
307 	__isl_keep isl_set *set)
308 {
309 	return isl_map_has_equal_params(map, set_to_map(set));
310 }
311 
312 /* Is the tuple of type "type" of "bmap" the same as the single tuple of "bset"?
313  */
isl_basic_map_set_tuple_is_equal(__isl_keep isl_basic_map * bmap,enum isl_dim_type type,__isl_keep isl_basic_set * bset)314 static isl_bool isl_basic_map_set_tuple_is_equal(__isl_keep isl_basic_map *bmap,
315 	enum isl_dim_type type, __isl_keep isl_basic_set *bset)
316 {
317 	isl_space *bmap_space, *bset_space;
318 
319 	bmap_space = isl_basic_map_peek_space(bmap);
320 	bset_space = isl_basic_set_peek_space(bset);
321 	return isl_space_tuple_is_equal(bmap_space, type,
322 					bset_space, isl_dim_set);
323 }
324 
325 /* Is the tuple of type "type" of "map" the same as the single tuple of "set"?
326  */
isl_map_set_tuple_is_equal(__isl_keep isl_map * map,enum isl_dim_type type,__isl_keep isl_set * set)327 static isl_bool isl_map_set_tuple_is_equal(__isl_keep isl_map *map,
328 	enum isl_dim_type type, __isl_keep isl_set *set)
329 {
330 	return isl_map_tuple_is_equal(map, type, set_to_map(set), isl_dim_set);
331 }
332 
isl_map_compatible_domain(__isl_keep isl_map * map,__isl_keep isl_set * set)333 isl_bool isl_map_compatible_domain(__isl_keep isl_map *map,
334 	__isl_keep isl_set *set)
335 {
336 	isl_bool m;
337 	if (!map || !set)
338 		return isl_bool_error;
339 	m = isl_map_has_equal_params(map, set_to_map(set));
340 	if (m < 0 || !m)
341 		return m;
342 	return isl_map_set_tuple_is_equal(map, isl_dim_in, set);
343 }
344 
isl_basic_map_compatible_domain(__isl_keep isl_basic_map * bmap,__isl_keep isl_basic_set * bset)345 isl_bool isl_basic_map_compatible_domain(__isl_keep isl_basic_map *bmap,
346 	__isl_keep isl_basic_set *bset)
347 {
348 	isl_bool m;
349 	if (!bmap || !bset)
350 		return isl_bool_error;
351 	m = isl_basic_map_has_equal_params(bmap, bset_to_bmap(bset));
352 	if (m < 0 || !m)
353 		return m;
354 	return isl_basic_map_set_tuple_is_equal(bmap, isl_dim_in, bset);
355 }
356 
isl_map_compatible_range(__isl_keep isl_map * map,__isl_keep isl_set * set)357 isl_bool isl_map_compatible_range(__isl_keep isl_map *map,
358 	__isl_keep isl_set *set)
359 {
360 	isl_bool m;
361 	if (!map || !set)
362 		return isl_bool_error;
363 	m = isl_map_has_equal_params(map, set_to_map(set));
364 	if (m < 0 || !m)
365 		return m;
366 	return isl_map_set_tuple_is_equal(map, isl_dim_out, set);
367 }
368 
isl_basic_map_compatible_range(__isl_keep isl_basic_map * bmap,__isl_keep isl_basic_set * bset)369 isl_bool isl_basic_map_compatible_range(__isl_keep isl_basic_map *bmap,
370 	__isl_keep isl_basic_set *bset)
371 {
372 	isl_bool m;
373 	if (!bmap || !bset)
374 		return isl_bool_error;
375 	m = isl_basic_map_has_equal_params(bmap, bset_to_bmap(bset));
376 	if (m < 0 || !m)
377 		return m;
378 	return isl_basic_map_set_tuple_is_equal(bmap, isl_dim_out, bset);
379 }
380 
isl_basic_map_get_ctx(__isl_keep isl_basic_map * bmap)381 isl_ctx *isl_basic_map_get_ctx(__isl_keep isl_basic_map *bmap)
382 {
383 	return bmap ? bmap->ctx : NULL;
384 }
385 
isl_basic_set_get_ctx(__isl_keep isl_basic_set * bset)386 isl_ctx *isl_basic_set_get_ctx(__isl_keep isl_basic_set *bset)
387 {
388 	return bset ? bset->ctx : NULL;
389 }
390 
isl_map_get_ctx(__isl_keep isl_map * map)391 isl_ctx *isl_map_get_ctx(__isl_keep isl_map *map)
392 {
393 	return map ? map->ctx : NULL;
394 }
395 
isl_set_get_ctx(__isl_keep isl_set * set)396 isl_ctx *isl_set_get_ctx(__isl_keep isl_set *set)
397 {
398 	return set ? set->ctx : NULL;
399 }
400 
401 /* Return the space of "bmap".
402  */
isl_basic_map_peek_space(__isl_keep const isl_basic_map * bmap)403 __isl_keep isl_space *isl_basic_map_peek_space(
404 	__isl_keep const isl_basic_map *bmap)
405 {
406 	return bmap ? bmap->dim : NULL;
407 }
408 
409 /* Return the space of "bset".
410  */
isl_basic_set_peek_space(__isl_keep isl_basic_set * bset)411 __isl_keep isl_space *isl_basic_set_peek_space(__isl_keep isl_basic_set *bset)
412 {
413 	return isl_basic_map_peek_space(bset_to_bmap(bset));
414 }
415 
isl_basic_map_get_space(__isl_keep isl_basic_map * bmap)416 __isl_give isl_space *isl_basic_map_get_space(__isl_keep isl_basic_map *bmap)
417 {
418 	return isl_space_copy(isl_basic_map_peek_space(bmap));
419 }
420 
isl_basic_set_get_space(__isl_keep isl_basic_set * bset)421 __isl_give isl_space *isl_basic_set_get_space(__isl_keep isl_basic_set *bset)
422 {
423 	return isl_basic_map_get_space(bset_to_bmap(bset));
424 }
425 
426 /* Return the space of "bmap".
427  * This may be either a copy or the space itself
428  * if there is only one reference to "bmap".
429  * This allows the space to be modified inplace
430  * if both the basic map and its space have only a single reference.
431  * The caller is not allowed to modify "bmap" between this call and
432  * a subsequent call to isl_basic_map_restore_space.
433  * The only exception is that isl_basic_map_free can be called instead.
434  */
isl_basic_map_take_space(__isl_keep isl_basic_map * bmap)435 static __isl_give isl_space *isl_basic_map_take_space(
436 	__isl_keep isl_basic_map *bmap)
437 {
438 	isl_space *space;
439 
440 	if (!bmap)
441 		return NULL;
442 	if (bmap->ref != 1)
443 		return isl_basic_map_get_space(bmap);
444 	space = bmap->dim;
445 	bmap->dim = NULL;
446 	return space;
447 }
448 
449 /* Set the space of "bmap" to "space", where the space of "bmap" may be missing
450  * due to a preceding call to isl_basic_map_take_space.
451  * However, in this case, "bmap" only has a single reference and
452  * then the call to isl_basic_map_cow has no effect.
453  */
isl_basic_map_restore_space(__isl_take isl_basic_map * bmap,__isl_take isl_space * space)454 static __isl_give isl_basic_map *isl_basic_map_restore_space(
455 	__isl_take isl_basic_map *bmap, __isl_take isl_space *space)
456 {
457 	if (!bmap || !space)
458 		goto error;
459 
460 	if (bmap->dim == space) {
461 		isl_space_free(space);
462 		return bmap;
463 	}
464 
465 	bmap = isl_basic_map_cow(bmap);
466 	if (!bmap)
467 		goto error;
468 	isl_space_free(bmap->dim);
469 	bmap->dim = space;
470 
471 	return bmap;
472 error:
473 	isl_basic_map_free(bmap);
474 	isl_space_free(space);
475 	return NULL;
476 }
477 
478 /* Extract the divs in "bmap" as a matrix.
479  */
isl_basic_map_get_divs(__isl_keep isl_basic_map * bmap)480 __isl_give isl_mat *isl_basic_map_get_divs(__isl_keep isl_basic_map *bmap)
481 {
482 	int i;
483 	isl_ctx *ctx;
484 	isl_mat *div;
485 	isl_size v_div;
486 	unsigned cols;
487 
488 	v_div = isl_basic_map_var_offset(bmap, isl_dim_div);
489 	if (v_div < 0)
490 		return NULL;
491 
492 	ctx = isl_basic_map_get_ctx(bmap);
493 	cols = 1 + 1 + v_div + bmap->n_div;
494 	div = isl_mat_alloc(ctx, bmap->n_div, cols);
495 	if (!div)
496 		return NULL;
497 
498 	for (i = 0; i < bmap->n_div; ++i)
499 		isl_seq_cpy(div->row[i], bmap->div[i], cols);
500 
501 	return div;
502 }
503 
504 /* Extract the divs in "bset" as a matrix.
505  */
isl_basic_set_get_divs(__isl_keep isl_basic_set * bset)506 __isl_give isl_mat *isl_basic_set_get_divs(__isl_keep isl_basic_set *bset)
507 {
508 	return isl_basic_map_get_divs(bset);
509 }
510 
isl_basic_map_get_local_space(__isl_keep isl_basic_map * bmap)511 __isl_give isl_local_space *isl_basic_map_get_local_space(
512 	__isl_keep isl_basic_map *bmap)
513 {
514 	isl_mat *div;
515 
516 	if (!bmap)
517 		return NULL;
518 
519 	div = isl_basic_map_get_divs(bmap);
520 	return isl_local_space_alloc_div(isl_space_copy(bmap->dim), div);
521 }
522 
isl_basic_set_get_local_space(__isl_keep isl_basic_set * bset)523 __isl_give isl_local_space *isl_basic_set_get_local_space(
524 	__isl_keep isl_basic_set *bset)
525 {
526 	return isl_basic_map_get_local_space(bset);
527 }
528 
529 /* For each known div d = floor(f/m), add the constraints
530  *
531  *		f - m d >= 0
532  *		-(f-(m-1)) + m d >= 0
533  *
534  * Do not finalize the result.
535  */
add_known_div_constraints(__isl_take isl_basic_map * bmap)536 static __isl_give isl_basic_map *add_known_div_constraints(
537 	__isl_take isl_basic_map *bmap)
538 {
539 	int i;
540 	isl_size n_div;
541 
542 	n_div = isl_basic_map_dim(bmap, isl_dim_div);
543 	if (n_div < 0)
544 		return isl_basic_map_free(bmap);
545 	if (n_div == 0)
546 		return bmap;
547 	bmap = isl_basic_map_cow(bmap);
548 	bmap = isl_basic_map_extend_constraints(bmap, 0, 2 * n_div);
549 	if (!bmap)
550 		return NULL;
551 	for (i = 0; i < n_div; ++i) {
552 		if (isl_int_is_zero(bmap->div[i][0]))
553 			continue;
554 		bmap = isl_basic_map_add_div_constraints(bmap, i);
555 	}
556 
557 	return bmap;
558 }
559 
isl_basic_map_from_local_space(__isl_take isl_local_space * ls)560 __isl_give isl_basic_map *isl_basic_map_from_local_space(
561 	__isl_take isl_local_space *ls)
562 {
563 	int i;
564 	isl_size n_div;
565 	isl_basic_map *bmap;
566 
567 	n_div = isl_local_space_dim(ls, isl_dim_div);
568 	if (n_div < 0)
569 		ls = isl_local_space_free(ls);
570 	if (!ls)
571 		return NULL;
572 
573 	bmap = isl_basic_map_alloc_space(isl_local_space_get_space(ls),
574 					n_div, 0, 2 * n_div);
575 
576 	for (i = 0; i < n_div; ++i)
577 		if (isl_basic_map_alloc_div(bmap) < 0)
578 			goto error;
579 
580 	for (i = 0; i < n_div; ++i)
581 		isl_seq_cpy(bmap->div[i], ls->div->row[i], ls->div->n_col);
582 	bmap = add_known_div_constraints(bmap);
583 
584 	isl_local_space_free(ls);
585 	return bmap;
586 error:
587 	isl_local_space_free(ls);
588 	isl_basic_map_free(bmap);
589 	return NULL;
590 }
591 
isl_basic_set_from_local_space(__isl_take isl_local_space * ls)592 __isl_give isl_basic_set *isl_basic_set_from_local_space(
593 	__isl_take isl_local_space *ls)
594 {
595 	return isl_basic_map_from_local_space(ls);
596 }
597 
isl_map_get_space(__isl_keep isl_map * map)598 __isl_give isl_space *isl_map_get_space(__isl_keep isl_map *map)
599 {
600 	return isl_space_copy(isl_map_peek_space(map));
601 }
602 
isl_set_get_space(__isl_keep isl_set * set)603 __isl_give isl_space *isl_set_get_space(__isl_keep isl_set *set)
604 {
605 	if (!set)
606 		return NULL;
607 	return isl_space_copy(set->dim);
608 }
609 
610 /* Return the space of "map".
611  * This may be either a copy or the space itself
612  * if there is only one reference to "map".
613  * This allows the space to be modified inplace
614  * if both the map and its space have only a single reference.
615  * The caller is not allowed to modify "map" between this call and
616  * a subsequent call to isl_map_restore_space.
617  * The only exception is that isl_map_free can be called instead.
618  */
isl_map_take_space(__isl_keep isl_map * map)619 static __isl_give isl_space *isl_map_take_space(__isl_keep isl_map *map)
620 {
621 	isl_space *space;
622 
623 	if (!map)
624 		return NULL;
625 	if (map->ref != 1)
626 		return isl_map_get_space(map);
627 	space = map->dim;
628 	map->dim = NULL;
629 	return space;
630 }
631 
632 /* Set the space of "map" to "space", where the space of "map" may be missing
633  * due to a preceding call to isl_map_take_space.
634  * However, in this case, "map" only has a single reference and
635  * then the call to isl_map_cow has no effect.
636  */
isl_map_restore_space(__isl_take isl_map * map,__isl_take isl_space * space)637 static __isl_give isl_map *isl_map_restore_space(__isl_take isl_map *map,
638 	__isl_take isl_space *space)
639 {
640 	if (!map || !space)
641 		goto error;
642 
643 	if (map->dim == space) {
644 		isl_space_free(space);
645 		return map;
646 	}
647 
648 	map = isl_map_cow(map);
649 	if (!map)
650 		goto error;
651 	isl_space_free(map->dim);
652 	map->dim = space;
653 
654 	return map;
655 error:
656 	isl_map_free(map);
657 	isl_space_free(space);
658 	return NULL;
659 }
660 
isl_basic_map_set_tuple_name(__isl_take isl_basic_map * bmap,enum isl_dim_type type,const char * s)661 __isl_give isl_basic_map *isl_basic_map_set_tuple_name(
662 	__isl_take isl_basic_map *bmap, enum isl_dim_type type, const char *s)
663 {
664 	isl_space *space;
665 
666 	space = isl_basic_map_take_space(bmap);
667 	space = isl_space_set_tuple_name(space, type, s);
668 	bmap = isl_basic_map_restore_space(bmap, space);
669 	bmap = isl_basic_map_finalize(bmap);
670 	return bmap;
671 }
672 
isl_basic_set_set_tuple_name(__isl_take isl_basic_set * bset,const char * s)673 __isl_give isl_basic_set *isl_basic_set_set_tuple_name(
674 	__isl_take isl_basic_set *bset, const char *s)
675 {
676 	return isl_basic_map_set_tuple_name(bset, isl_dim_set, s);
677 }
678 
isl_basic_map_get_tuple_name(__isl_keep isl_basic_map * bmap,enum isl_dim_type type)679 const char *isl_basic_map_get_tuple_name(__isl_keep isl_basic_map *bmap,
680 	enum isl_dim_type type)
681 {
682 	return bmap ? isl_space_get_tuple_name(bmap->dim, type) : NULL;
683 }
684 
isl_map_set_tuple_name(__isl_take isl_map * map,enum isl_dim_type type,const char * s)685 __isl_give isl_map *isl_map_set_tuple_name(__isl_take isl_map *map,
686 	enum isl_dim_type type, const char *s)
687 {
688 	int i;
689 	isl_space *space;
690 
691 	map = isl_map_cow(map);
692 	if (!map)
693 		return NULL;
694 
695 	for (i = 0; i < map->n; ++i) {
696 		map->p[i] = isl_basic_map_set_tuple_name(map->p[i], type, s);
697 		if (!map->p[i])
698 			goto error;
699 	}
700 
701 	space = isl_map_take_space(map);
702 	space = isl_space_set_tuple_name(space, type, s);
703 	map = isl_map_restore_space(map, space);
704 
705 	return map;
706 error:
707 	isl_map_free(map);
708 	return NULL;
709 }
710 
711 /* Replace the identifier of the tuple of type "type" by "id".
712  */
isl_basic_map_set_tuple_id(__isl_take isl_basic_map * bmap,enum isl_dim_type type,__isl_take isl_id * id)713 __isl_give isl_basic_map *isl_basic_map_set_tuple_id(
714 	__isl_take isl_basic_map *bmap,
715 	enum isl_dim_type type, __isl_take isl_id *id)
716 {
717 	isl_space *space;
718 
719 	space = isl_basic_map_take_space(bmap);
720 	space = isl_space_set_tuple_id(space, type, id);
721 	bmap = isl_basic_map_restore_space(bmap, space);
722 	bmap = isl_basic_map_finalize(bmap);
723 	return bmap;
724 }
725 
726 /* Replace the identifier of the tuple by "id".
727  */
isl_basic_set_set_tuple_id(__isl_take isl_basic_set * bset,__isl_take isl_id * id)728 __isl_give isl_basic_set *isl_basic_set_set_tuple_id(
729 	__isl_take isl_basic_set *bset, __isl_take isl_id *id)
730 {
731 	return isl_basic_map_set_tuple_id(bset, isl_dim_set, id);
732 }
733 
734 /* Does the input or output tuple have a name?
735  */
isl_map_has_tuple_name(__isl_keep isl_map * map,enum isl_dim_type type)736 isl_bool isl_map_has_tuple_name(__isl_keep isl_map *map, enum isl_dim_type type)
737 {
738 	return map ? isl_space_has_tuple_name(map->dim, type) : isl_bool_error;
739 }
740 
isl_map_get_tuple_name(__isl_keep isl_map * map,enum isl_dim_type type)741 const char *isl_map_get_tuple_name(__isl_keep isl_map *map,
742 	enum isl_dim_type type)
743 {
744 	return map ? isl_space_get_tuple_name(map->dim, type) : NULL;
745 }
746 
isl_set_set_tuple_name(__isl_take isl_set * set,const char * s)747 __isl_give isl_set *isl_set_set_tuple_name(__isl_take isl_set *set,
748 	const char *s)
749 {
750 	return set_from_map(isl_map_set_tuple_name(set_to_map(set),
751 						isl_dim_set, s));
752 }
753 
isl_map_set_tuple_id(__isl_take isl_map * map,enum isl_dim_type type,__isl_take isl_id * id)754 __isl_give isl_map *isl_map_set_tuple_id(__isl_take isl_map *map,
755 	enum isl_dim_type type, __isl_take isl_id *id)
756 {
757 	isl_space *space;
758 
759 	space = isl_map_take_space(map);
760 	space = isl_space_set_tuple_id(space, type, id);
761 	map = isl_map_restore_space(map, space);
762 
763 	return isl_map_reset_space(map, isl_map_get_space(map));
764 }
765 
766 /* Replace the identifier of the domain tuple of "map" by "id".
767  */
isl_map_set_domain_tuple_id(__isl_take isl_map * map,__isl_take isl_id * id)768 __isl_give isl_map *isl_map_set_domain_tuple_id(__isl_take isl_map *map,
769 	__isl_take isl_id *id)
770 {
771 	return isl_map_set_tuple_id(map, isl_dim_in, id);
772 }
773 
774 /* Replace the identifier of the range tuple of "map" by "id".
775  */
isl_map_set_range_tuple_id(__isl_take isl_map * map,__isl_take isl_id * id)776 __isl_give isl_map *isl_map_set_range_tuple_id(__isl_take isl_map *map,
777 	__isl_take isl_id *id)
778 {
779 	return isl_map_set_tuple_id(map, isl_dim_out, id);
780 }
781 
isl_set_set_tuple_id(__isl_take isl_set * set,__isl_take isl_id * id)782 __isl_give isl_set *isl_set_set_tuple_id(__isl_take isl_set *set,
783 	__isl_take isl_id *id)
784 {
785 	return isl_map_set_tuple_id(set, isl_dim_set, id);
786 }
787 
isl_map_reset_tuple_id(__isl_take isl_map * map,enum isl_dim_type type)788 __isl_give isl_map *isl_map_reset_tuple_id(__isl_take isl_map *map,
789 	enum isl_dim_type type)
790 {
791 	isl_space *space;
792 
793 	space = isl_map_take_space(map);
794 	space = isl_space_reset_tuple_id(space, type);
795 	map = isl_map_restore_space(map, space);
796 
797 	return isl_map_reset_space(map, isl_map_get_space(map));
798 }
799 
isl_set_reset_tuple_id(__isl_take isl_set * set)800 __isl_give isl_set *isl_set_reset_tuple_id(__isl_take isl_set *set)
801 {
802 	return isl_map_reset_tuple_id(set, isl_dim_set);
803 }
804 
isl_map_has_tuple_id(__isl_keep isl_map * map,enum isl_dim_type type)805 isl_bool isl_map_has_tuple_id(__isl_keep isl_map *map, enum isl_dim_type type)
806 {
807 	return map ? isl_space_has_tuple_id(map->dim, type) : isl_bool_error;
808 }
809 
810 /* Does the domain tuple of "map" have an identifier?
811  */
isl_map_has_domain_tuple_id(__isl_keep isl_map * map)812 isl_bool isl_map_has_domain_tuple_id(__isl_keep isl_map *map)
813 {
814 	return isl_map_has_tuple_id(map, isl_dim_in);
815 }
816 
817 /* Does the range tuple of "map" have an identifier?
818  */
isl_map_has_range_tuple_id(__isl_keep isl_map * map)819 isl_bool isl_map_has_range_tuple_id(__isl_keep isl_map *map)
820 {
821 	return isl_map_has_tuple_id(map, isl_dim_out);
822 }
823 
isl_map_get_tuple_id(__isl_keep isl_map * map,enum isl_dim_type type)824 __isl_give isl_id *isl_map_get_tuple_id(__isl_keep isl_map *map,
825 	enum isl_dim_type type)
826 {
827 	return map ? isl_space_get_tuple_id(map->dim, type) : NULL;
828 }
829 
830 /* Return the identifier of the domain tuple of "map", assuming it has one.
831  */
isl_map_get_domain_tuple_id(__isl_keep isl_map * map)832 __isl_give isl_id *isl_map_get_domain_tuple_id(__isl_keep isl_map *map)
833 {
834 	return isl_map_get_tuple_id(map, isl_dim_in);
835 }
836 
837 /* Return the identifier of the range tuple of "map", assuming it has one.
838  */
isl_map_get_range_tuple_id(__isl_keep isl_map * map)839 __isl_give isl_id *isl_map_get_range_tuple_id(__isl_keep isl_map *map)
840 {
841 	return isl_map_get_tuple_id(map, isl_dim_out);
842 }
843 
isl_set_has_tuple_id(__isl_keep isl_set * set)844 isl_bool isl_set_has_tuple_id(__isl_keep isl_set *set)
845 {
846 	return isl_map_has_tuple_id(set, isl_dim_set);
847 }
848 
isl_set_get_tuple_id(__isl_keep isl_set * set)849 __isl_give isl_id *isl_set_get_tuple_id(__isl_keep isl_set *set)
850 {
851 	return isl_map_get_tuple_id(set, isl_dim_set);
852 }
853 
854 /* Does the set tuple have a name?
855  */
isl_set_has_tuple_name(__isl_keep isl_set * set)856 isl_bool isl_set_has_tuple_name(__isl_keep isl_set *set)
857 {
858 	if (!set)
859 		return isl_bool_error;
860 	return isl_space_has_tuple_name(set->dim, isl_dim_set);
861 }
862 
863 
isl_basic_set_get_tuple_name(__isl_keep isl_basic_set * bset)864 const char *isl_basic_set_get_tuple_name(__isl_keep isl_basic_set *bset)
865 {
866 	return bset ? isl_space_get_tuple_name(bset->dim, isl_dim_set) : NULL;
867 }
868 
isl_set_get_tuple_name(__isl_keep isl_set * set)869 const char *isl_set_get_tuple_name(__isl_keep isl_set *set)
870 {
871 	return set ? isl_space_get_tuple_name(set->dim, isl_dim_set) : NULL;
872 }
873 
isl_basic_map_get_dim_name(__isl_keep isl_basic_map * bmap,enum isl_dim_type type,unsigned pos)874 const char *isl_basic_map_get_dim_name(__isl_keep isl_basic_map *bmap,
875 	enum isl_dim_type type, unsigned pos)
876 {
877 	return bmap ? isl_space_get_dim_name(bmap->dim, type, pos) : NULL;
878 }
879 
isl_basic_set_get_dim_name(__isl_keep isl_basic_set * bset,enum isl_dim_type type,unsigned pos)880 const char *isl_basic_set_get_dim_name(__isl_keep isl_basic_set *bset,
881 	enum isl_dim_type type, unsigned pos)
882 {
883 	return bset ? isl_space_get_dim_name(bset->dim, type, pos) : NULL;
884 }
885 
886 /* Does the given dimension have a name?
887  */
isl_map_has_dim_name(__isl_keep isl_map * map,enum isl_dim_type type,unsigned pos)888 isl_bool isl_map_has_dim_name(__isl_keep isl_map *map,
889 	enum isl_dim_type type, unsigned pos)
890 {
891 	if (!map)
892 		return isl_bool_error;
893 	return isl_space_has_dim_name(map->dim, type, pos);
894 }
895 
isl_map_get_dim_name(__isl_keep isl_map * map,enum isl_dim_type type,unsigned pos)896 const char *isl_map_get_dim_name(__isl_keep isl_map *map,
897 	enum isl_dim_type type, unsigned pos)
898 {
899 	return map ? isl_space_get_dim_name(map->dim, type, pos) : NULL;
900 }
901 
isl_set_get_dim_name(__isl_keep isl_set * set,enum isl_dim_type type,unsigned pos)902 const char *isl_set_get_dim_name(__isl_keep isl_set *set,
903 	enum isl_dim_type type, unsigned pos)
904 {
905 	return set ? isl_space_get_dim_name(set->dim, type, pos) : NULL;
906 }
907 
908 /* Does the given dimension have a name?
909  */
isl_set_has_dim_name(__isl_keep isl_set * set,enum isl_dim_type type,unsigned pos)910 isl_bool isl_set_has_dim_name(__isl_keep isl_set *set,
911 	enum isl_dim_type type, unsigned pos)
912 {
913 	if (!set)
914 		return isl_bool_error;
915 	return isl_space_has_dim_name(set->dim, type, pos);
916 }
917 
isl_basic_map_set_dim_name(__isl_take isl_basic_map * bmap,enum isl_dim_type type,unsigned pos,const char * s)918 __isl_give isl_basic_map *isl_basic_map_set_dim_name(
919 	__isl_take isl_basic_map *bmap,
920 	enum isl_dim_type type, unsigned pos, const char *s)
921 {
922 	isl_space *space;
923 
924 	space = isl_basic_map_take_space(bmap);
925 	space = isl_space_set_dim_name(space, type, pos, s);
926 	bmap = isl_basic_map_restore_space(bmap, space);
927 	return isl_basic_map_finalize(bmap);
928 }
929 
isl_map_set_dim_name(__isl_take isl_map * map,enum isl_dim_type type,unsigned pos,const char * s)930 __isl_give isl_map *isl_map_set_dim_name(__isl_take isl_map *map,
931 	enum isl_dim_type type, unsigned pos, const char *s)
932 {
933 	int i;
934 	isl_space *space;
935 
936 	map = isl_map_cow(map);
937 	if (!map)
938 		return NULL;
939 
940 	for (i = 0; i < map->n; ++i) {
941 		map->p[i] = isl_basic_map_set_dim_name(map->p[i], type, pos, s);
942 		if (!map->p[i])
943 			goto error;
944 	}
945 
946 	space = isl_map_take_space(map);
947 	space = isl_space_set_dim_name(space, type, pos, s);
948 	map = isl_map_restore_space(map, space);
949 
950 	return map;
951 error:
952 	isl_map_free(map);
953 	return NULL;
954 }
955 
isl_basic_set_set_dim_name(__isl_take isl_basic_set * bset,enum isl_dim_type type,unsigned pos,const char * s)956 __isl_give isl_basic_set *isl_basic_set_set_dim_name(
957 	__isl_take isl_basic_set *bset,
958 	enum isl_dim_type type, unsigned pos, const char *s)
959 {
960 	return bset_from_bmap(isl_basic_map_set_dim_name(bset_to_bmap(bset),
961 							type, pos, s));
962 }
963 
isl_set_set_dim_name(__isl_take isl_set * set,enum isl_dim_type type,unsigned pos,const char * s)964 __isl_give isl_set *isl_set_set_dim_name(__isl_take isl_set *set,
965 	enum isl_dim_type type, unsigned pos, const char *s)
966 {
967 	return set_from_map(isl_map_set_dim_name(set_to_map(set),
968 							type, pos, s));
969 }
970 
isl_basic_map_has_dim_id(__isl_keep isl_basic_map * bmap,enum isl_dim_type type,unsigned pos)971 isl_bool isl_basic_map_has_dim_id(__isl_keep isl_basic_map *bmap,
972 	enum isl_dim_type type, unsigned pos)
973 {
974 	if (!bmap)
975 		return isl_bool_error;
976 	return isl_space_has_dim_id(bmap->dim, type, pos);
977 }
978 
isl_basic_set_get_dim_id(__isl_keep isl_basic_set * bset,enum isl_dim_type type,unsigned pos)979 __isl_give isl_id *isl_basic_set_get_dim_id(__isl_keep isl_basic_set *bset,
980 	enum isl_dim_type type, unsigned pos)
981 {
982 	return bset ? isl_space_get_dim_id(bset->dim, type, pos) : NULL;
983 }
984 
isl_map_has_dim_id(__isl_keep isl_map * map,enum isl_dim_type type,unsigned pos)985 isl_bool isl_map_has_dim_id(__isl_keep isl_map *map,
986 	enum isl_dim_type type, unsigned pos)
987 {
988 	return map ? isl_space_has_dim_id(map->dim, type, pos) : isl_bool_error;
989 }
990 
isl_map_get_dim_id(__isl_keep isl_map * map,enum isl_dim_type type,unsigned pos)991 __isl_give isl_id *isl_map_get_dim_id(__isl_keep isl_map *map,
992 	enum isl_dim_type type, unsigned pos)
993 {
994 	return map ? isl_space_get_dim_id(map->dim, type, pos) : NULL;
995 }
996 
isl_set_has_dim_id(__isl_keep isl_set * set,enum isl_dim_type type,unsigned pos)997 isl_bool isl_set_has_dim_id(__isl_keep isl_set *set,
998 	enum isl_dim_type type, unsigned pos)
999 {
1000 	return isl_map_has_dim_id(set, type, pos);
1001 }
1002 
isl_set_get_dim_id(__isl_keep isl_set * set,enum isl_dim_type type,unsigned pos)1003 __isl_give isl_id *isl_set_get_dim_id(__isl_keep isl_set *set,
1004 	enum isl_dim_type type, unsigned pos)
1005 {
1006 	return isl_map_get_dim_id(set, type, pos);
1007 }
1008 
isl_map_set_dim_id(__isl_take isl_map * map,enum isl_dim_type type,unsigned pos,__isl_take isl_id * id)1009 __isl_give isl_map *isl_map_set_dim_id(__isl_take isl_map *map,
1010 	enum isl_dim_type type, unsigned pos, __isl_take isl_id *id)
1011 {
1012 	isl_space *space;
1013 
1014 	space = isl_map_take_space(map);
1015 	space = isl_space_set_dim_id(space, type, pos, id);
1016 	map = isl_map_restore_space(map, space);
1017 
1018 	return isl_map_reset_space(map, isl_map_get_space(map));
1019 }
1020 
isl_set_set_dim_id(__isl_take isl_set * set,enum isl_dim_type type,unsigned pos,__isl_take isl_id * id)1021 __isl_give isl_set *isl_set_set_dim_id(__isl_take isl_set *set,
1022 	enum isl_dim_type type, unsigned pos, __isl_take isl_id *id)
1023 {
1024 	return isl_map_set_dim_id(set, type, pos, id);
1025 }
1026 
isl_map_find_dim_by_id(__isl_keep isl_map * map,enum isl_dim_type type,__isl_keep isl_id * id)1027 int isl_map_find_dim_by_id(__isl_keep isl_map *map, enum isl_dim_type type,
1028 	__isl_keep isl_id *id)
1029 {
1030 	if (!map)
1031 		return -1;
1032 	return isl_space_find_dim_by_id(map->dim, type, id);
1033 }
1034 
isl_set_find_dim_by_id(__isl_keep isl_set * set,enum isl_dim_type type,__isl_keep isl_id * id)1035 int isl_set_find_dim_by_id(__isl_keep isl_set *set, enum isl_dim_type type,
1036 	__isl_keep isl_id *id)
1037 {
1038 	return isl_map_find_dim_by_id(set, type, id);
1039 }
1040 
1041 /* Return the position of the dimension of the given type and name
1042  * in "bmap".
1043  * Return -1 if no such dimension can be found.
1044  */
isl_basic_map_find_dim_by_name(__isl_keep isl_basic_map * bmap,enum isl_dim_type type,const char * name)1045 int isl_basic_map_find_dim_by_name(__isl_keep isl_basic_map *bmap,
1046 	enum isl_dim_type type, const char *name)
1047 {
1048 	if (!bmap)
1049 		return -1;
1050 	return isl_space_find_dim_by_name(bmap->dim, type, name);
1051 }
1052 
isl_map_find_dim_by_name(__isl_keep isl_map * map,enum isl_dim_type type,const char * name)1053 int isl_map_find_dim_by_name(__isl_keep isl_map *map, enum isl_dim_type type,
1054 	const char *name)
1055 {
1056 	if (!map)
1057 		return -1;
1058 	return isl_space_find_dim_by_name(map->dim, type, name);
1059 }
1060 
isl_set_find_dim_by_name(__isl_keep isl_set * set,enum isl_dim_type type,const char * name)1061 int isl_set_find_dim_by_name(__isl_keep isl_set *set, enum isl_dim_type type,
1062 	const char *name)
1063 {
1064 	return isl_map_find_dim_by_name(set, type, name);
1065 }
1066 
1067 /* Check whether equality i of bset is a pure stride constraint
1068  * on a single dimension, i.e., of the form
1069  *
1070  *	v = k e
1071  *
1072  * with k a constant and e an existentially quantified variable.
1073  */
isl_basic_set_eq_is_stride(__isl_keep isl_basic_set * bset,int i)1074 isl_bool isl_basic_set_eq_is_stride(__isl_keep isl_basic_set *bset, int i)
1075 {
1076 	isl_size nparam;
1077 	isl_size d;
1078 	isl_size n_div;
1079 	int pos1;
1080 	int pos2;
1081 
1082 	nparam = isl_basic_set_dim(bset, isl_dim_param);
1083 	d = isl_basic_set_dim(bset, isl_dim_set);
1084 	n_div = isl_basic_set_dim(bset, isl_dim_div);
1085 	if (nparam < 0 || d < 0 || n_div < 0)
1086 		return isl_bool_error;
1087 
1088 	if (!isl_int_is_zero(bset->eq[i][0]))
1089 		return isl_bool_false;
1090 
1091 	if (isl_seq_first_non_zero(bset->eq[i] + 1, nparam) != -1)
1092 		return isl_bool_false;
1093 	pos1 = isl_seq_first_non_zero(bset->eq[i] + 1 + nparam, d);
1094 	if (pos1 == -1)
1095 		return isl_bool_false;
1096 	if (isl_seq_first_non_zero(bset->eq[i] + 1 + nparam + pos1 + 1,
1097 					d - pos1 - 1) != -1)
1098 		return isl_bool_false;
1099 
1100 	pos2 = isl_seq_first_non_zero(bset->eq[i] + 1 + nparam + d, n_div);
1101 	if (pos2 == -1)
1102 		return isl_bool_false;
1103 	if (isl_seq_first_non_zero(bset->eq[i] + 1 + nparam + d  + pos2 + 1,
1104 				   n_div - pos2 - 1) != -1)
1105 		return isl_bool_false;
1106 	if (!isl_int_is_one(bset->eq[i][1 + nparam + pos1]) &&
1107 	    !isl_int_is_negone(bset->eq[i][1 + nparam + pos1]))
1108 		return isl_bool_false;
1109 
1110 	return isl_bool_true;
1111 }
1112 
1113 /* Reset the user pointer on all identifiers of parameters and tuples
1114  * of the space of "map".
1115  */
isl_map_reset_user(__isl_take isl_map * map)1116 __isl_give isl_map *isl_map_reset_user(__isl_take isl_map *map)
1117 {
1118 	isl_space *space;
1119 
1120 	space = isl_map_get_space(map);
1121 	space = isl_space_reset_user(space);
1122 	map = isl_map_reset_space(map, space);
1123 
1124 	return map;
1125 }
1126 
1127 /* Reset the user pointer on all identifiers of parameters and tuples
1128  * of the space of "set".
1129  */
isl_set_reset_user(__isl_take isl_set * set)1130 __isl_give isl_set *isl_set_reset_user(__isl_take isl_set *set)
1131 {
1132 	return isl_map_reset_user(set);
1133 }
1134 
isl_basic_map_is_rational(__isl_keep isl_basic_map * bmap)1135 isl_bool isl_basic_map_is_rational(__isl_keep isl_basic_map *bmap)
1136 {
1137 	if (!bmap)
1138 		return isl_bool_error;
1139 	return ISL_F_ISSET(bmap, ISL_BASIC_MAP_RATIONAL);
1140 }
1141 
1142 /* Has "map" been marked as a rational map?
1143  * In particular, have all basic maps in "map" been marked this way?
1144  * An empty map is not considered to be rational.
1145  * Maps where only some of the basic maps are marked rational
1146  * are not allowed.
1147  */
isl_map_is_rational(__isl_keep isl_map * map)1148 isl_bool isl_map_is_rational(__isl_keep isl_map *map)
1149 {
1150 	int i;
1151 	isl_bool rational;
1152 
1153 	if (!map)
1154 		return isl_bool_error;
1155 	if (map->n == 0)
1156 		return isl_bool_false;
1157 	rational = isl_basic_map_is_rational(map->p[0]);
1158 	if (rational < 0)
1159 		return rational;
1160 	for (i = 1; i < map->n; ++i) {
1161 		isl_bool rational_i;
1162 
1163 		rational_i = isl_basic_map_is_rational(map->p[i]);
1164 		if (rational_i < 0)
1165 			return rational_i;
1166 		if (rational != rational_i)
1167 			isl_die(isl_map_get_ctx(map), isl_error_unsupported,
1168 				"mixed rational and integer basic maps "
1169 				"not supported", return isl_bool_error);
1170 	}
1171 
1172 	return rational;
1173 }
1174 
1175 /* Has "set" been marked as a rational set?
1176  * In particular, have all basic set in "set" been marked this way?
1177  * An empty set is not considered to be rational.
1178  * Sets where only some of the basic sets are marked rational
1179  * are not allowed.
1180  */
isl_set_is_rational(__isl_keep isl_set * set)1181 isl_bool isl_set_is_rational(__isl_keep isl_set *set)
1182 {
1183 	return isl_map_is_rational(set);
1184 }
1185 
isl_basic_set_is_rational(__isl_keep isl_basic_set * bset)1186 int isl_basic_set_is_rational(__isl_keep isl_basic_set *bset)
1187 {
1188 	return isl_basic_map_is_rational(bset);
1189 }
1190 
1191 /* Does "bmap" contain any rational points?
1192  *
1193  * If "bmap" has an equality for each dimension, equating the dimension
1194  * to an integer constant, then it has no rational points, even if it
1195  * is marked as rational.
1196  */
isl_basic_map_has_rational(__isl_keep isl_basic_map * bmap)1197 isl_bool isl_basic_map_has_rational(__isl_keep isl_basic_map *bmap)
1198 {
1199 	isl_bool has_rational = isl_bool_true;
1200 	isl_size total;
1201 
1202 	if (!bmap)
1203 		return isl_bool_error;
1204 	if (isl_basic_map_plain_is_empty(bmap))
1205 		return isl_bool_false;
1206 	if (!isl_basic_map_is_rational(bmap))
1207 		return isl_bool_false;
1208 	bmap = isl_basic_map_copy(bmap);
1209 	bmap = isl_basic_map_implicit_equalities(bmap);
1210 	total = isl_basic_map_dim(bmap, isl_dim_all);
1211 	if (total < 0)
1212 		return isl_bool_error;
1213 	if (bmap->n_eq == total) {
1214 		int i, j;
1215 		for (i = 0; i < bmap->n_eq; ++i) {
1216 			j = isl_seq_first_non_zero(bmap->eq[i] + 1, total);
1217 			if (j < 0)
1218 				break;
1219 			if (!isl_int_is_one(bmap->eq[i][1 + j]) &&
1220 			    !isl_int_is_negone(bmap->eq[i][1 + j]))
1221 				break;
1222 			j = isl_seq_first_non_zero(bmap->eq[i] + 1 + j + 1,
1223 						    total - j - 1);
1224 			if (j >= 0)
1225 				break;
1226 		}
1227 		if (i == bmap->n_eq)
1228 			has_rational = isl_bool_false;
1229 	}
1230 	isl_basic_map_free(bmap);
1231 
1232 	return has_rational;
1233 }
1234 
1235 /* Does "map" contain any rational points?
1236  */
isl_map_has_rational(__isl_keep isl_map * map)1237 isl_bool isl_map_has_rational(__isl_keep isl_map *map)
1238 {
1239 	int i;
1240 	isl_bool has_rational;
1241 
1242 	if (!map)
1243 		return isl_bool_error;
1244 	for (i = 0; i < map->n; ++i) {
1245 		has_rational = isl_basic_map_has_rational(map->p[i]);
1246 		if (has_rational < 0 || has_rational)
1247 			return has_rational;
1248 	}
1249 	return isl_bool_false;
1250 }
1251 
1252 /* Does "set" contain any rational points?
1253  */
isl_set_has_rational(__isl_keep isl_set * set)1254 isl_bool isl_set_has_rational(__isl_keep isl_set *set)
1255 {
1256 	return isl_map_has_rational(set);
1257 }
1258 
1259 /* Is this basic set a parameter domain?
1260  */
isl_basic_set_is_params(__isl_keep isl_basic_set * bset)1261 isl_bool isl_basic_set_is_params(__isl_keep isl_basic_set *bset)
1262 {
1263 	if (!bset)
1264 		return isl_bool_error;
1265 	return isl_space_is_params(bset->dim);
1266 }
1267 
1268 /* Is this set a parameter domain?
1269  */
isl_set_is_params(__isl_keep isl_set * set)1270 isl_bool isl_set_is_params(__isl_keep isl_set *set)
1271 {
1272 	if (!set)
1273 		return isl_bool_error;
1274 	return isl_space_is_params(set->dim);
1275 }
1276 
1277 /* Is this map actually a parameter domain?
1278  * Users should never call this function.  Outside of isl,
1279  * a map can never be a parameter domain.
1280  */
isl_map_is_params(__isl_keep isl_map * map)1281 isl_bool isl_map_is_params(__isl_keep isl_map *map)
1282 {
1283 	if (!map)
1284 		return isl_bool_error;
1285 	return isl_space_is_params(map->dim);
1286 }
1287 
basic_map_init(isl_ctx * ctx,__isl_take isl_basic_map * bmap,unsigned extra,unsigned n_eq,unsigned n_ineq)1288 static __isl_give isl_basic_map *basic_map_init(isl_ctx *ctx,
1289 	__isl_take isl_basic_map *bmap, unsigned extra,
1290 	unsigned n_eq, unsigned n_ineq)
1291 {
1292 	int i;
1293 	isl_space *space = isl_basic_map_peek_space(bmap);
1294 	isl_size n_var = isl_space_dim(space, isl_dim_all);
1295 	size_t row_size = 1 + n_var + extra;
1296 
1297 	bmap->ctx = ctx;
1298 	isl_ctx_ref(ctx);
1299 
1300 	if (n_var < 0)
1301 		return isl_basic_map_free(bmap);
1302 
1303 	bmap->block = isl_blk_alloc(ctx, (n_ineq + n_eq) * row_size);
1304 	if (isl_blk_is_error(bmap->block))
1305 		goto error;
1306 
1307 	bmap->ineq = isl_alloc_array(ctx, isl_int *, n_ineq + n_eq);
1308 	if ((n_ineq + n_eq) && !bmap->ineq)
1309 		goto error;
1310 
1311 	if (extra == 0) {
1312 		bmap->block2 = isl_blk_empty();
1313 		bmap->div = NULL;
1314 	} else {
1315 		bmap->block2 = isl_blk_alloc(ctx, extra * (1 + row_size));
1316 		if (isl_blk_is_error(bmap->block2))
1317 			goto error;
1318 
1319 		bmap->div = isl_alloc_array(ctx, isl_int *, extra);
1320 		if (!bmap->div)
1321 			goto error;
1322 	}
1323 
1324 	for (i = 0; i < n_ineq + n_eq; ++i)
1325 		bmap->ineq[i] = bmap->block.data + i * row_size;
1326 
1327 	for (i = 0; i < extra; ++i)
1328 		bmap->div[i] = bmap->block2.data + i * (1 + row_size);
1329 
1330 	bmap->ref = 1;
1331 	bmap->flags = 0;
1332 	bmap->c_size = n_eq + n_ineq;
1333 	bmap->eq = bmap->ineq + n_ineq;
1334 	bmap->extra = extra;
1335 	bmap->n_eq = 0;
1336 	bmap->n_ineq = 0;
1337 	bmap->n_div = 0;
1338 	bmap->sample = NULL;
1339 
1340 	return bmap;
1341 error:
1342 	isl_basic_map_free(bmap);
1343 	return NULL;
1344 }
1345 
isl_basic_set_alloc(isl_ctx * ctx,unsigned nparam,unsigned dim,unsigned extra,unsigned n_eq,unsigned n_ineq)1346 __isl_give isl_basic_set *isl_basic_set_alloc(isl_ctx *ctx,
1347 		unsigned nparam, unsigned dim, unsigned extra,
1348 		unsigned n_eq, unsigned n_ineq)
1349 {
1350 	struct isl_basic_map *bmap;
1351 	isl_space *space;
1352 
1353 	space = isl_space_set_alloc(ctx, nparam, dim);
1354 	if (!space)
1355 		return NULL;
1356 
1357 	bmap = isl_basic_map_alloc_space(space, extra, n_eq, n_ineq);
1358 	return bset_from_bmap(bmap);
1359 }
1360 
isl_basic_set_alloc_space(__isl_take isl_space * space,unsigned extra,unsigned n_eq,unsigned n_ineq)1361 __isl_give isl_basic_set *isl_basic_set_alloc_space(__isl_take isl_space *space,
1362 		unsigned extra, unsigned n_eq, unsigned n_ineq)
1363 {
1364 	struct isl_basic_map *bmap;
1365 	if (!space)
1366 		return NULL;
1367 	isl_assert(space->ctx, space->n_in == 0, goto error);
1368 	bmap = isl_basic_map_alloc_space(space, extra, n_eq, n_ineq);
1369 	return bset_from_bmap(bmap);
1370 error:
1371 	isl_space_free(space);
1372 	return NULL;
1373 }
1374 
isl_basic_map_alloc_space(__isl_take isl_space * space,unsigned extra,unsigned n_eq,unsigned n_ineq)1375 __isl_give isl_basic_map *isl_basic_map_alloc_space(__isl_take isl_space *space,
1376 		unsigned extra, unsigned n_eq, unsigned n_ineq)
1377 {
1378 	struct isl_basic_map *bmap;
1379 
1380 	if (!space)
1381 		return NULL;
1382 	bmap = isl_calloc_type(space->ctx, struct isl_basic_map);
1383 	if (!bmap)
1384 		goto error;
1385 	bmap->dim = space;
1386 
1387 	return basic_map_init(space->ctx, bmap, extra, n_eq, n_ineq);
1388 error:
1389 	isl_space_free(space);
1390 	return NULL;
1391 }
1392 
isl_basic_map_alloc(isl_ctx * ctx,unsigned nparam,unsigned in,unsigned out,unsigned extra,unsigned n_eq,unsigned n_ineq)1393 __isl_give isl_basic_map *isl_basic_map_alloc(isl_ctx *ctx,
1394 		unsigned nparam, unsigned in, unsigned out, unsigned extra,
1395 		unsigned n_eq, unsigned n_ineq)
1396 {
1397 	struct isl_basic_map *bmap;
1398 	isl_space *space;
1399 
1400 	space = isl_space_alloc(ctx, nparam, in, out);
1401 	if (!space)
1402 		return NULL;
1403 
1404 	bmap = isl_basic_map_alloc_space(space, extra, n_eq, n_ineq);
1405 	return bmap;
1406 }
1407 
dup_constraints(__isl_take isl_basic_map * dst,__isl_keep isl_basic_map * src)1408 static __isl_give isl_basic_map *dup_constraints(__isl_take isl_basic_map *dst,
1409 	__isl_keep isl_basic_map *src)
1410 {
1411 	int i;
1412 	isl_size total = isl_basic_map_dim(src, isl_dim_all);
1413 
1414 	if (!dst || total < 0)
1415 		return isl_basic_map_free(dst);
1416 
1417 	for (i = 0; i < src->n_eq; ++i) {
1418 		int j = isl_basic_map_alloc_equality(dst);
1419 		if (j < 0)
1420 			return isl_basic_map_free(dst);
1421 		isl_seq_cpy(dst->eq[j], src->eq[i], 1+total);
1422 	}
1423 
1424 	for (i = 0; i < src->n_ineq; ++i) {
1425 		int j = isl_basic_map_alloc_inequality(dst);
1426 		if (j < 0)
1427 			return isl_basic_map_free(dst);
1428 		isl_seq_cpy(dst->ineq[j], src->ineq[i], 1+total);
1429 	}
1430 
1431 	for (i = 0; i < src->n_div; ++i) {
1432 		int j = isl_basic_map_alloc_div(dst);
1433 		if (j < 0)
1434 			return isl_basic_map_free(dst);
1435 		isl_seq_cpy(dst->div[j], src->div[i], 1+1+total);
1436 	}
1437 	ISL_F_SET(dst, ISL_BASIC_SET_FINAL);
1438 	return dst;
1439 }
1440 
isl_basic_map_dup(__isl_keep isl_basic_map * bmap)1441 __isl_give isl_basic_map *isl_basic_map_dup(__isl_keep isl_basic_map *bmap)
1442 {
1443 	struct isl_basic_map *dup;
1444 
1445 	if (!bmap)
1446 		return NULL;
1447 	dup = isl_basic_map_alloc_space(isl_space_copy(bmap->dim),
1448 			bmap->n_div, bmap->n_eq, bmap->n_ineq);
1449 	dup = dup_constraints(dup, bmap);
1450 	if (!dup)
1451 		return NULL;
1452 	dup->flags = bmap->flags;
1453 	dup->sample = isl_vec_copy(bmap->sample);
1454 	return dup;
1455 }
1456 
isl_basic_set_dup(__isl_keep isl_basic_set * bset)1457 __isl_give isl_basic_set *isl_basic_set_dup(__isl_keep isl_basic_set *bset)
1458 {
1459 	struct isl_basic_map *dup;
1460 
1461 	dup = isl_basic_map_dup(bset_to_bmap(bset));
1462 	return bset_from_bmap(dup);
1463 }
1464 
isl_basic_set_copy(__isl_keep isl_basic_set * bset)1465 __isl_give isl_basic_set *isl_basic_set_copy(__isl_keep isl_basic_set *bset)
1466 {
1467 	return bset_from_bmap(isl_basic_map_copy(bset_to_bmap(bset)));
1468 }
1469 
isl_set_copy(__isl_keep isl_set * set)1470 __isl_give isl_set *isl_set_copy(__isl_keep isl_set *set)
1471 {
1472 	if (!set)
1473 		return NULL;
1474 
1475 	set->ref++;
1476 	return set;
1477 }
1478 
isl_basic_map_copy(__isl_keep isl_basic_map * bmap)1479 __isl_give isl_basic_map *isl_basic_map_copy(__isl_keep isl_basic_map *bmap)
1480 {
1481 	if (!bmap)
1482 		return NULL;
1483 
1484 	if (ISL_F_ISSET(bmap, ISL_BASIC_SET_FINAL)) {
1485 		bmap->ref++;
1486 		return bmap;
1487 	}
1488 	bmap = isl_basic_map_dup(bmap);
1489 	if (bmap)
1490 		ISL_F_SET(bmap, ISL_BASIC_SET_FINAL);
1491 	return bmap;
1492 }
1493 
isl_map_copy(__isl_keep isl_map * map)1494 __isl_give isl_map *isl_map_copy(__isl_keep isl_map *map)
1495 {
1496 	if (!map)
1497 		return NULL;
1498 
1499 	map->ref++;
1500 	return map;
1501 }
1502 
isl_basic_map_free(__isl_take isl_basic_map * bmap)1503 __isl_null isl_basic_map *isl_basic_map_free(__isl_take isl_basic_map *bmap)
1504 {
1505 	if (!bmap)
1506 		return NULL;
1507 
1508 	if (--bmap->ref > 0)
1509 		return NULL;
1510 
1511 	isl_ctx_deref(bmap->ctx);
1512 	free(bmap->div);
1513 	isl_blk_free(bmap->ctx, bmap->block2);
1514 	free(bmap->ineq);
1515 	isl_blk_free(bmap->ctx, bmap->block);
1516 	isl_vec_free(bmap->sample);
1517 	isl_space_free(bmap->dim);
1518 	free(bmap);
1519 
1520 	return NULL;
1521 }
1522 
isl_basic_set_free(__isl_take isl_basic_set * bset)1523 __isl_null isl_basic_set *isl_basic_set_free(__isl_take isl_basic_set *bset)
1524 {
1525 	return isl_basic_map_free(bset_to_bmap(bset));
1526 }
1527 
room_for_con(__isl_keep isl_basic_map * bmap,unsigned n)1528 static int room_for_con(__isl_keep isl_basic_map *bmap, unsigned n)
1529 {
1530 	return bmap->n_eq + bmap->n_ineq + n <= bmap->c_size;
1531 }
1532 
1533 /* Check that "bset" does not involve any parameters.
1534  */
isl_basic_set_check_no_params(__isl_keep isl_basic_set * bset)1535 isl_stat isl_basic_set_check_no_params(__isl_keep isl_basic_set *bset)
1536 {
1537 	isl_size nparam;
1538 
1539 	nparam = isl_basic_set_dim(bset, isl_dim_param);
1540 	if (nparam < 0)
1541 		return isl_stat_error;
1542 	if (nparam != 0)
1543 		isl_die(isl_basic_set_get_ctx(bset), isl_error_invalid,
1544 			"basic set should not have any parameters",
1545 			return isl_stat_error);
1546 	return isl_stat_ok;
1547 }
1548 
1549 /* Check that "bset" does not involve any local variables.
1550  */
isl_basic_set_check_no_locals(__isl_keep isl_basic_set * bset)1551 isl_stat isl_basic_set_check_no_locals(__isl_keep isl_basic_set *bset)
1552 {
1553 	isl_size n_div;
1554 
1555 	n_div = isl_basic_set_dim(bset, isl_dim_div);
1556 	if (n_div < 0)
1557 		return isl_stat_error;
1558 	if (n_div != 0)
1559 		isl_die(isl_basic_set_get_ctx(bset), isl_error_invalid,
1560 			"basic set should not have any local variables",
1561 			return isl_stat_error);
1562 	return isl_stat_ok;
1563 }
1564 
1565 #undef TYPE
1566 #define TYPE isl_map
1567 
1568 #include "isl_check_named_params_templ.c"
1569 
1570 #undef TYPE
1571 #define TYPE isl_basic_map
1572 
1573 static
1574 #include "isl_check_named_params_templ.c"
1575 
1576 /* Check that "bmap1" and "bmap2" have the same parameters,
1577  * reporting an error if they do not.
1578  */
isl_basic_map_check_equal_params(__isl_keep isl_basic_map * bmap1,__isl_keep isl_basic_map * bmap2)1579 static isl_stat isl_basic_map_check_equal_params(
1580 	__isl_keep isl_basic_map *bmap1, __isl_keep isl_basic_map *bmap2)
1581 {
1582 	isl_bool match;
1583 
1584 	match = isl_basic_map_has_equal_params(bmap1, bmap2);
1585 	if (match < 0)
1586 		return isl_stat_error;
1587 	if (!match)
1588 		isl_die(isl_basic_map_get_ctx(bmap1), isl_error_invalid,
1589 			"parameters don't match", return isl_stat_error);
1590 	return isl_stat_ok;
1591 }
1592 
1593 #undef TYPE
1594 #define TYPE	isl_map
1595 
1596 #include "isl_align_params_bin_templ.c"
1597 
1598 #undef SUFFIX
1599 #define SUFFIX	set
1600 #undef ARG1
1601 #define ARG1	isl_map
1602 #undef ARG2
1603 #define ARG2	isl_set
1604 
1605 #include "isl_align_params_templ.c"
1606 
isl_map_align_params_map_map_and_test(__isl_keep isl_map * map1,__isl_keep isl_map * map2,isl_bool (* fn)(__isl_keep isl_map * map1,__isl_keep isl_map * map2))1607 isl_bool isl_map_align_params_map_map_and_test(__isl_keep isl_map *map1,
1608 	__isl_keep isl_map *map2,
1609 	isl_bool (*fn)(__isl_keep isl_map *map1, __isl_keep isl_map *map2))
1610 {
1611 	isl_bool r;
1612 
1613 	if (!map1 || !map2)
1614 		return isl_bool_error;
1615 	if (isl_map_has_equal_params(map1, map2))
1616 		return fn(map1, map2);
1617 	if (isl_map_check_named_params(map1) < 0)
1618 		return isl_bool_error;
1619 	if (isl_map_check_named_params(map2) < 0)
1620 		return isl_bool_error;
1621 	map1 = isl_map_copy(map1);
1622 	map2 = isl_map_copy(map2);
1623 	map1 = isl_map_align_params(map1, isl_map_get_space(map2));
1624 	map2 = isl_map_align_params(map2, isl_map_get_space(map1));
1625 	r = fn(map1, map2);
1626 	isl_map_free(map1);
1627 	isl_map_free(map2);
1628 	return r;
1629 }
1630 
isl_basic_map_alloc_equality(__isl_keep isl_basic_map * bmap)1631 int isl_basic_map_alloc_equality(__isl_keep isl_basic_map *bmap)
1632 {
1633 	isl_size total;
1634 	struct isl_ctx *ctx;
1635 
1636 	total = isl_basic_map_dim(bmap, isl_dim_all);
1637 	if (total < 0)
1638 		return -1;
1639 	ctx = bmap->ctx;
1640 	isl_assert(ctx, room_for_con(bmap, 1), return -1);
1641 	isl_assert(ctx, (bmap->eq - bmap->ineq) + bmap->n_eq <= bmap->c_size,
1642 			return -1);
1643 	ISL_F_CLR(bmap, ISL_BASIC_MAP_NO_REDUNDANT);
1644 	ISL_F_CLR(bmap, ISL_BASIC_MAP_NO_IMPLICIT);
1645 	ISL_F_CLR(bmap, ISL_BASIC_MAP_ALL_EQUALITIES);
1646 	ISL_F_CLR(bmap, ISL_BASIC_MAP_NORMALIZED_DIVS);
1647 	if ((bmap->eq - bmap->ineq) + bmap->n_eq == bmap->c_size) {
1648 		isl_int *t;
1649 		int j = isl_basic_map_alloc_inequality(bmap);
1650 		if (j < 0)
1651 			return -1;
1652 		t = bmap->ineq[j];
1653 		bmap->ineq[j] = bmap->ineq[bmap->n_ineq - 1];
1654 		bmap->ineq[bmap->n_ineq - 1] = bmap->eq[-1];
1655 		bmap->eq[-1] = t;
1656 		bmap->n_eq++;
1657 		bmap->n_ineq--;
1658 		bmap->eq--;
1659 		return 0;
1660 	}
1661 	isl_seq_clr(bmap->eq[bmap->n_eq] + 1 + total,
1662 		      bmap->extra - bmap->n_div);
1663 	return bmap->n_eq++;
1664 }
1665 
isl_basic_set_alloc_equality(__isl_keep isl_basic_set * bset)1666 int isl_basic_set_alloc_equality(__isl_keep isl_basic_set *bset)
1667 {
1668 	return isl_basic_map_alloc_equality(bset_to_bmap(bset));
1669 }
1670 
isl_basic_map_free_equality(__isl_take isl_basic_map * bmap,unsigned n)1671 __isl_give isl_basic_map *isl_basic_map_free_equality(
1672 	__isl_take isl_basic_map *bmap, unsigned n)
1673 {
1674 	if (!bmap)
1675 		return NULL;
1676 	if (n > bmap->n_eq)
1677 		isl_die(isl_basic_map_get_ctx(bmap), isl_error_invalid,
1678 			"invalid number of equalities",
1679 			isl_basic_map_free(bmap));
1680 	bmap->n_eq -= n;
1681 	return bmap;
1682 }
1683 
isl_basic_set_free_equality(__isl_take isl_basic_set * bset,unsigned n)1684 __isl_give isl_basic_set *isl_basic_set_free_equality(
1685 	__isl_take isl_basic_set *bset, unsigned n)
1686 {
1687 	return bset_from_bmap(isl_basic_map_free_equality(bset_to_bmap(bset),
1688 							    n));
1689 }
1690 
1691 /* Drop the equality constraint at position "pos",
1692  * preserving the order of the other equality constraints.
1693  */
isl_basic_map_drop_equality(__isl_keep isl_basic_map * bmap,unsigned pos)1694 int isl_basic_map_drop_equality(__isl_keep isl_basic_map *bmap, unsigned pos)
1695 {
1696 	isl_int *t;
1697 	int r;
1698 
1699 	if (!bmap)
1700 		return -1;
1701 	isl_assert(bmap->ctx, pos < bmap->n_eq, return -1);
1702 
1703 	t = bmap->eq[pos];
1704 	bmap->n_eq--;
1705 	for (r = pos; r < bmap->n_eq; ++r)
1706 		bmap->eq[r] = bmap->eq[r + 1];
1707 	bmap->eq[bmap->n_eq] = t;
1708 
1709 	return 0;
1710 }
1711 
1712 /* Turn inequality "pos" of "bmap" into an equality.
1713  *
1714  * In particular, we move the inequality in front of the equalities
1715  * and move the last inequality in the position of the moved inequality.
1716  * Note that isl_tab_make_equalities_explicit depends on this particular
1717  * change in the ordering of the constraints.
1718  */
isl_basic_map_inequality_to_equality(__isl_keep isl_basic_map * bmap,unsigned pos)1719 void isl_basic_map_inequality_to_equality(
1720 		__isl_keep isl_basic_map *bmap, unsigned pos)
1721 {
1722 	isl_int *t;
1723 
1724 	t = bmap->ineq[pos];
1725 	bmap->ineq[pos] = bmap->ineq[bmap->n_ineq - 1];
1726 	bmap->ineq[bmap->n_ineq - 1] = bmap->eq[-1];
1727 	bmap->eq[-1] = t;
1728 	bmap->n_eq++;
1729 	bmap->n_ineq--;
1730 	bmap->eq--;
1731 	ISL_F_CLR(bmap, ISL_BASIC_MAP_NO_REDUNDANT);
1732 	ISL_F_CLR(bmap, ISL_BASIC_MAP_SORTED);
1733 	ISL_F_CLR(bmap, ISL_BASIC_MAP_NORMALIZED_DIVS);
1734 	ISL_F_CLR(bmap, ISL_BASIC_MAP_ALL_EQUALITIES);
1735 }
1736 
room_for_ineq(__isl_keep isl_basic_map * bmap,unsigned n)1737 static int room_for_ineq(__isl_keep isl_basic_map *bmap, unsigned n)
1738 {
1739 	return bmap->n_ineq + n <= bmap->eq - bmap->ineq;
1740 }
1741 
isl_basic_map_alloc_inequality(__isl_keep isl_basic_map * bmap)1742 int isl_basic_map_alloc_inequality(__isl_keep isl_basic_map *bmap)
1743 {
1744 	isl_size total;
1745 	struct isl_ctx *ctx;
1746 
1747 	total = isl_basic_map_dim(bmap, isl_dim_all);
1748 	if (total < 0)
1749 		return -1;
1750 	ctx = bmap->ctx;
1751 	isl_assert(ctx, room_for_ineq(bmap, 1), return -1);
1752 	ISL_F_CLR(bmap, ISL_BASIC_MAP_NO_IMPLICIT);
1753 	ISL_F_CLR(bmap, ISL_BASIC_MAP_NO_REDUNDANT);
1754 	ISL_F_CLR(bmap, ISL_BASIC_MAP_SORTED);
1755 	ISL_F_CLR(bmap, ISL_BASIC_MAP_ALL_EQUALITIES);
1756 	isl_seq_clr(bmap->ineq[bmap->n_ineq] + 1 + total,
1757 		      bmap->extra - bmap->n_div);
1758 	return bmap->n_ineq++;
1759 }
1760 
isl_basic_set_alloc_inequality(__isl_keep isl_basic_set * bset)1761 int isl_basic_set_alloc_inequality(__isl_keep isl_basic_set *bset)
1762 {
1763 	return isl_basic_map_alloc_inequality(bset_to_bmap(bset));
1764 }
1765 
isl_basic_map_free_inequality(__isl_take isl_basic_map * bmap,unsigned n)1766 __isl_give isl_basic_map *isl_basic_map_free_inequality(
1767 	__isl_take isl_basic_map *bmap, unsigned n)
1768 {
1769 	if (!bmap)
1770 		return NULL;
1771 	if (n > bmap->n_ineq)
1772 		isl_die(isl_basic_map_get_ctx(bmap), isl_error_invalid,
1773 			"invalid number of inequalities",
1774 			return isl_basic_map_free(bmap));
1775 	bmap->n_ineq -= n;
1776 	return bmap;
1777 }
1778 
isl_basic_set_free_inequality(__isl_take isl_basic_set * bset,unsigned n)1779 __isl_give isl_basic_set *isl_basic_set_free_inequality(
1780 	__isl_take isl_basic_set *bset, unsigned n)
1781 {
1782 	return bset_from_bmap(isl_basic_map_free_inequality(bset_to_bmap(bset),
1783 							    n));
1784 }
1785 
isl_basic_map_drop_inequality(__isl_keep isl_basic_map * bmap,unsigned pos)1786 int isl_basic_map_drop_inequality(__isl_keep isl_basic_map *bmap, unsigned pos)
1787 {
1788 	isl_int *t;
1789 	if (!bmap)
1790 		return -1;
1791 	isl_assert(bmap->ctx, pos < bmap->n_ineq, return -1);
1792 
1793 	if (pos != bmap->n_ineq - 1) {
1794 		t = bmap->ineq[pos];
1795 		bmap->ineq[pos] = bmap->ineq[bmap->n_ineq - 1];
1796 		bmap->ineq[bmap->n_ineq - 1] = t;
1797 		ISL_F_CLR(bmap, ISL_BASIC_MAP_SORTED);
1798 	}
1799 	bmap->n_ineq--;
1800 	return 0;
1801 }
1802 
isl_basic_set_drop_inequality(__isl_keep isl_basic_set * bset,unsigned pos)1803 int isl_basic_set_drop_inequality(__isl_keep isl_basic_set *bset, unsigned pos)
1804 {
1805 	return isl_basic_map_drop_inequality(bset_to_bmap(bset), pos);
1806 }
1807 
isl_basic_map_add_eq(__isl_take isl_basic_map * bmap,isl_int * eq)1808 __isl_give isl_basic_map *isl_basic_map_add_eq(__isl_take isl_basic_map *bmap,
1809 	isl_int *eq)
1810 {
1811 	isl_bool empty;
1812 	isl_size total;
1813 	int k;
1814 
1815 	empty = isl_basic_map_plain_is_empty(bmap);
1816 	if (empty < 0)
1817 		return isl_basic_map_free(bmap);
1818 	if (empty)
1819 		return bmap;
1820 
1821 	bmap = isl_basic_map_cow(bmap);
1822 	bmap = isl_basic_map_extend_constraints(bmap, 1, 0);
1823 	total = isl_basic_map_dim(bmap, isl_dim_all);
1824 	if (total < 0)
1825 		return isl_basic_map_free(bmap);
1826 	k = isl_basic_map_alloc_equality(bmap);
1827 	if (k < 0)
1828 		goto error;
1829 	isl_seq_cpy(bmap->eq[k], eq, 1 + total);
1830 	return bmap;
1831 error:
1832 	isl_basic_map_free(bmap);
1833 	return NULL;
1834 }
1835 
isl_basic_set_add_eq(__isl_take isl_basic_set * bset,isl_int * eq)1836 __isl_give isl_basic_set *isl_basic_set_add_eq(__isl_take isl_basic_set *bset,
1837 	isl_int *eq)
1838 {
1839 	return bset_from_bmap(isl_basic_map_add_eq(bset_to_bmap(bset), eq));
1840 }
1841 
isl_basic_map_add_ineq(__isl_take isl_basic_map * bmap,isl_int * ineq)1842 __isl_give isl_basic_map *isl_basic_map_add_ineq(__isl_take isl_basic_map *bmap,
1843 	isl_int *ineq)
1844 {
1845 	isl_size total;
1846 	int k;
1847 
1848 	bmap = isl_basic_map_cow(bmap);
1849 	bmap = isl_basic_map_extend_constraints(bmap, 0, 1);
1850 	total = isl_basic_map_dim(bmap, isl_dim_all);
1851 	if (total < 0)
1852 		return isl_basic_map_free(bmap);
1853 	k = isl_basic_map_alloc_inequality(bmap);
1854 	if (k < 0)
1855 		goto error;
1856 	isl_seq_cpy(bmap->ineq[k], ineq, 1 + total);
1857 	return bmap;
1858 error:
1859 	isl_basic_map_free(bmap);
1860 	return NULL;
1861 }
1862 
isl_basic_set_add_ineq(__isl_take isl_basic_set * bset,isl_int * ineq)1863 __isl_give isl_basic_set *isl_basic_set_add_ineq(__isl_take isl_basic_set *bset,
1864 	isl_int *ineq)
1865 {
1866 	return bset_from_bmap(isl_basic_map_add_ineq(bset_to_bmap(bset), ineq));
1867 }
1868 
isl_basic_map_alloc_div(__isl_keep isl_basic_map * bmap)1869 int isl_basic_map_alloc_div(__isl_keep isl_basic_map *bmap)
1870 {
1871 	isl_size total;
1872 
1873 	total = isl_basic_map_dim(bmap, isl_dim_all);
1874 	if (total < 0)
1875 		return -1;
1876 	isl_assert(bmap->ctx, bmap->n_div < bmap->extra, return -1);
1877 	isl_seq_clr(bmap->div[bmap->n_div] + 1 + 1 + total,
1878 		      bmap->extra - bmap->n_div);
1879 	ISL_F_CLR(bmap, ISL_BASIC_MAP_NORMALIZED_DIVS);
1880 	return bmap->n_div++;
1881 }
1882 
isl_basic_set_alloc_div(__isl_keep isl_basic_set * bset)1883 int isl_basic_set_alloc_div(__isl_keep isl_basic_set *bset)
1884 {
1885 	return isl_basic_map_alloc_div(bset_to_bmap(bset));
1886 }
1887 
1888 #undef TYPE
1889 #define TYPE	isl_basic_map
1890 #include "check_type_range_templ.c"
1891 
1892 /* Check that there are "n" dimensions of type "type" starting at "first"
1893  * in "bset".
1894  */
isl_basic_set_check_range(__isl_keep isl_basic_set * bset,enum isl_dim_type type,unsigned first,unsigned n)1895 isl_stat isl_basic_set_check_range(__isl_keep isl_basic_set *bset,
1896 	enum isl_dim_type type, unsigned first, unsigned n)
1897 {
1898 	return isl_basic_map_check_range(bset_to_bmap(bset),
1899 					type, first, n);
1900 }
1901 
1902 /* Insert an extra integer division, prescribed by "div", to "bmap"
1903  * at (integer division) position "pos".
1904  *
1905  * The integer division is first added at the end and then moved
1906  * into the right position.
1907  */
isl_basic_map_insert_div(__isl_take isl_basic_map * bmap,int pos,__isl_keep isl_vec * div)1908 __isl_give isl_basic_map *isl_basic_map_insert_div(
1909 	__isl_take isl_basic_map *bmap, int pos, __isl_keep isl_vec *div)
1910 {
1911 	int i, k;
1912 	isl_size total;
1913 
1914 	bmap = isl_basic_map_cow(bmap);
1915 	total = isl_basic_map_dim(bmap, isl_dim_all);
1916 	if (total < 0 || !div)
1917 		return isl_basic_map_free(bmap);
1918 
1919 	if (div->size != 1 + 1 + total)
1920 		isl_die(isl_basic_map_get_ctx(bmap), isl_error_invalid,
1921 			"unexpected size", return isl_basic_map_free(bmap));
1922 	if (isl_basic_map_check_range(bmap, isl_dim_div, pos, 0) < 0)
1923 		return isl_basic_map_free(bmap);
1924 
1925 	bmap = isl_basic_map_extend(bmap, 1, 0, 2);
1926 	k = isl_basic_map_alloc_div(bmap);
1927 	if (k < 0)
1928 		return isl_basic_map_free(bmap);
1929 	isl_seq_cpy(bmap->div[k], div->el, div->size);
1930 	isl_int_set_si(bmap->div[k][div->size], 0);
1931 
1932 	for (i = k; i > pos; --i)
1933 		bmap = isl_basic_map_swap_div(bmap, i, i - 1);
1934 
1935 	return bmap;
1936 }
1937 
isl_basic_map_free_div(__isl_keep isl_basic_map * bmap,unsigned n)1938 isl_stat isl_basic_map_free_div(__isl_keep isl_basic_map *bmap, unsigned n)
1939 {
1940 	if (!bmap)
1941 		return isl_stat_error;
1942 	isl_assert(bmap->ctx, n <= bmap->n_div, return isl_stat_error);
1943 	bmap->n_div -= n;
1944 	return isl_stat_ok;
1945 }
1946 
add_constraints(__isl_take isl_basic_map * bmap1,__isl_take isl_basic_map * bmap2,unsigned i_pos,unsigned o_pos)1947 static __isl_give isl_basic_map *add_constraints(
1948 	__isl_take isl_basic_map *bmap1, __isl_take isl_basic_map *bmap2,
1949 	unsigned i_pos, unsigned o_pos)
1950 {
1951 	isl_size total, n_param, n_in, n_out, n_div;
1952 	unsigned o_in, o_out;
1953 	isl_ctx *ctx;
1954 	isl_space *space;
1955 	struct isl_dim_map *dim_map;
1956 
1957 	space = isl_basic_map_peek_space(bmap2);
1958 	if (!bmap1 || !space)
1959 		goto error;
1960 
1961 	total = isl_basic_map_dim(bmap1, isl_dim_all);
1962 	n_param = isl_basic_map_dim(bmap2, isl_dim_param);
1963 	n_in = isl_basic_map_dim(bmap2, isl_dim_in);
1964 	o_in = isl_basic_map_offset(bmap1, isl_dim_in) - 1 + i_pos;
1965 	n_out = isl_basic_map_dim(bmap2, isl_dim_out);
1966 	o_out = isl_basic_map_offset(bmap1, isl_dim_out) - 1 + o_pos;
1967 	n_div = isl_basic_map_dim(bmap2, isl_dim_div);
1968 	if (total < 0 || n_param < 0 || n_in < 0 || n_out < 0 || n_div < 0)
1969 		goto error;
1970 	ctx = isl_basic_map_get_ctx(bmap1);
1971 	dim_map = isl_dim_map_alloc(ctx, total + n_div);
1972 	isl_dim_map_dim_range(dim_map, space, isl_dim_param, 0, n_param, 0);
1973 	isl_dim_map_dim_range(dim_map, space, isl_dim_in, 0, n_in, o_in);
1974 	isl_dim_map_dim_range(dim_map, space, isl_dim_out, 0, n_out, o_out);
1975 	isl_dim_map_div(dim_map, bmap2, total);
1976 
1977 	return isl_basic_map_add_constraints_dim_map(bmap1, bmap2, dim_map);
1978 error:
1979 	isl_basic_map_free(bmap1);
1980 	isl_basic_map_free(bmap2);
1981 	return NULL;
1982 }
1983 
isl_basic_map_extend(__isl_take isl_basic_map * base,unsigned extra,unsigned n_eq,unsigned n_ineq)1984 __isl_give isl_basic_map *isl_basic_map_extend(__isl_take isl_basic_map *base,
1985 	unsigned extra, unsigned n_eq, unsigned n_ineq)
1986 {
1987 	isl_space *space;
1988 	struct isl_basic_map *ext;
1989 	unsigned flags;
1990 	int dims_ok;
1991 
1992 	if (!base)
1993 		goto error;
1994 
1995 	dims_ok = base->extra >= base->n_div + extra;
1996 
1997 	if (dims_ok && room_for_con(base, n_eq + n_ineq) &&
1998 		       room_for_ineq(base, n_ineq))
1999 		return base;
2000 
2001 	extra += base->extra;
2002 	n_eq += base->n_eq;
2003 	n_ineq += base->n_ineq;
2004 
2005 	space = isl_basic_map_get_space(base);
2006 	ext = isl_basic_map_alloc_space(space, extra, n_eq, n_ineq);
2007 	if (!ext)
2008 		goto error;
2009 
2010 	if (dims_ok)
2011 		ext->sample = isl_vec_copy(base->sample);
2012 	flags = base->flags;
2013 	ext = add_constraints(ext, base, 0, 0);
2014 	if (ext) {
2015 		ext->flags = flags;
2016 		ISL_F_CLR(ext, ISL_BASIC_SET_FINAL);
2017 	}
2018 
2019 	return ext;
2020 
2021 error:
2022 	isl_basic_map_free(base);
2023 	return NULL;
2024 }
2025 
isl_basic_set_extend(__isl_take isl_basic_set * base,unsigned extra,unsigned n_eq,unsigned n_ineq)2026 __isl_give isl_basic_set *isl_basic_set_extend(__isl_take isl_basic_set *base,
2027 	unsigned extra, unsigned n_eq, unsigned n_ineq)
2028 {
2029 	return bset_from_bmap(isl_basic_map_extend(bset_to_bmap(base),
2030 						    extra, n_eq, n_ineq));
2031 }
2032 
isl_basic_map_extend_constraints(__isl_take isl_basic_map * base,unsigned n_eq,unsigned n_ineq)2033 __isl_give isl_basic_map *isl_basic_map_extend_constraints(
2034 	__isl_take isl_basic_map *base, unsigned n_eq, unsigned n_ineq)
2035 {
2036 	return isl_basic_map_extend(base, 0, n_eq, n_ineq);
2037 }
2038 
isl_basic_set_extend_constraints(__isl_take isl_basic_set * base,unsigned n_eq,unsigned n_ineq)2039 __isl_give isl_basic_set *isl_basic_set_extend_constraints(
2040 	__isl_take isl_basic_set *base, unsigned n_eq, unsigned n_ineq)
2041 {
2042 	isl_basic_map *bmap = bset_to_bmap(base);
2043 	bmap = isl_basic_map_extend_constraints(bmap, n_eq, n_ineq);
2044 	return bset_from_bmap(bmap);
2045 }
2046 
isl_basic_set_cow(__isl_take isl_basic_set * bset)2047 __isl_give isl_basic_set *isl_basic_set_cow(__isl_take isl_basic_set *bset)
2048 {
2049 	return bset_from_bmap(isl_basic_map_cow(bset_to_bmap(bset)));
2050 }
2051 
isl_basic_map_cow(__isl_take isl_basic_map * bmap)2052 __isl_give isl_basic_map *isl_basic_map_cow(__isl_take isl_basic_map *bmap)
2053 {
2054 	if (!bmap)
2055 		return NULL;
2056 
2057 	if (bmap->ref > 1) {
2058 		bmap->ref--;
2059 		bmap = isl_basic_map_dup(bmap);
2060 	}
2061 	if (bmap) {
2062 		ISL_F_CLR(bmap, ISL_BASIC_SET_FINAL);
2063 		ISL_F_CLR(bmap, ISL_BASIC_MAP_REDUCED_COEFFICIENTS);
2064 	}
2065 	return bmap;
2066 }
2067 
2068 /* Clear all cached information in "map", either because it is about
2069  * to be modified or because it is being freed.
2070  * Always return the same pointer that is passed in.
2071  * This is needed for the use in isl_map_free.
2072  */
clear_caches(__isl_take isl_map * map)2073 static __isl_give isl_map *clear_caches(__isl_take isl_map *map)
2074 {
2075 	isl_basic_map_free(map->cached_simple_hull[0]);
2076 	isl_basic_map_free(map->cached_simple_hull[1]);
2077 	map->cached_simple_hull[0] = NULL;
2078 	map->cached_simple_hull[1] = NULL;
2079 	return map;
2080 }
2081 
isl_set_cow(__isl_take isl_set * set)2082 __isl_give isl_set *isl_set_cow(__isl_take isl_set *set)
2083 {
2084 	return isl_map_cow(set);
2085 }
2086 
2087 /* Return an isl_map that is equal to "map" and that has only
2088  * a single reference.
2089  *
2090  * If the original input already has only one reference, then
2091  * simply return it, but clear all cached information, since
2092  * it may be rendered invalid by the operations that will be
2093  * performed on the result.
2094  *
2095  * Otherwise, create a duplicate (without any cached information).
2096  */
isl_map_cow(__isl_take isl_map * map)2097 __isl_give isl_map *isl_map_cow(__isl_take isl_map *map)
2098 {
2099 	if (!map)
2100 		return NULL;
2101 
2102 	if (map->ref == 1)
2103 		return clear_caches(map);
2104 	map->ref--;
2105 	return isl_map_dup(map);
2106 }
2107 
swap_vars(struct isl_blk blk,isl_int * a,unsigned a_len,unsigned b_len)2108 static void swap_vars(struct isl_blk blk, isl_int *a,
2109 			unsigned a_len, unsigned b_len)
2110 {
2111 	isl_seq_cpy(blk.data, a+a_len, b_len);
2112 	isl_seq_cpy(blk.data+b_len, a, a_len);
2113 	isl_seq_cpy(a, blk.data, b_len+a_len);
2114 }
2115 
isl_basic_map_swap_vars(__isl_take isl_basic_map * bmap,unsigned pos,unsigned n1,unsigned n2)2116 static __isl_give isl_basic_map *isl_basic_map_swap_vars(
2117 	__isl_take isl_basic_map *bmap, unsigned pos, unsigned n1, unsigned n2)
2118 {
2119 	int i;
2120 	struct isl_blk blk;
2121 
2122 	if (isl_basic_map_check_range(bmap, isl_dim_all, pos - 1, n1 + n2) < 0)
2123 		goto error;
2124 
2125 	if (n1 == 0 || n2 == 0)
2126 		return bmap;
2127 
2128 	bmap = isl_basic_map_cow(bmap);
2129 	if (!bmap)
2130 		return NULL;
2131 
2132 	blk = isl_blk_alloc(bmap->ctx, n1 + n2);
2133 	if (isl_blk_is_error(blk))
2134 		goto error;
2135 
2136 	for (i = 0; i < bmap->n_eq; ++i)
2137 		swap_vars(blk,
2138 			  bmap->eq[i] + pos, n1, n2);
2139 
2140 	for (i = 0; i < bmap->n_ineq; ++i)
2141 		swap_vars(blk,
2142 			  bmap->ineq[i] + pos, n1, n2);
2143 
2144 	for (i = 0; i < bmap->n_div; ++i)
2145 		swap_vars(blk,
2146 			  bmap->div[i]+1 + pos, n1, n2);
2147 
2148 	isl_blk_free(bmap->ctx, blk);
2149 
2150 	ISL_F_CLR(bmap, ISL_BASIC_SET_SORTED);
2151 	bmap = isl_basic_map_gauss(bmap, NULL);
2152 	return isl_basic_map_finalize(bmap);
2153 error:
2154 	isl_basic_map_free(bmap);
2155 	return NULL;
2156 }
2157 
2158 /* The given basic map has turned out to be empty.
2159  * Explicitly mark it as such and change the representation
2160  * to a canonical representation of the empty basic map.
2161  * Since the basic map has conflicting constraints,
2162  * it must have at least one constraint, except perhaps
2163  * if it was already explicitly marked as being empty.
2164  * Do nothing in the latter case, i.e., if it has been marked empty and
2165  * has no constraints.
2166  */
isl_basic_map_set_to_empty(__isl_take isl_basic_map * bmap)2167 __isl_give isl_basic_map *isl_basic_map_set_to_empty(
2168 	__isl_take isl_basic_map *bmap)
2169 {
2170 	int i = 0;
2171 	isl_bool empty;
2172 	isl_size n;
2173 	isl_size total;
2174 
2175 	n = isl_basic_map_n_constraint(bmap);
2176 	empty = isl_basic_map_plain_is_empty(bmap);
2177 	if (n < 0 || empty < 0)
2178 		return isl_basic_map_free(bmap);
2179 	if (n == 0 && empty)
2180 		return bmap;
2181 	total = isl_basic_map_dim(bmap, isl_dim_all);
2182 	if (total < 0)
2183 		return isl_basic_map_free(bmap);
2184 	if (isl_basic_map_free_div(bmap, bmap->n_div) < 0)
2185 		return isl_basic_map_free(bmap);
2186 	bmap = isl_basic_map_free_inequality(bmap, bmap->n_ineq);
2187 	if (!bmap)
2188 		return NULL;
2189 	if (bmap->n_eq > 0) {
2190 		bmap = isl_basic_map_free_equality(bmap, bmap->n_eq - 1);
2191 		if (!bmap)
2192 			return NULL;
2193 	} else {
2194 		i = isl_basic_map_alloc_equality(bmap);
2195 		if (i < 0)
2196 			goto error;
2197 	}
2198 	isl_int_set_si(bmap->eq[i][0], 1);
2199 	isl_seq_clr(bmap->eq[i]+1, total);
2200 	ISL_F_SET(bmap, ISL_BASIC_MAP_EMPTY);
2201 	isl_vec_free(bmap->sample);
2202 	bmap->sample = NULL;
2203 	return isl_basic_map_finalize(bmap);
2204 error:
2205 	isl_basic_map_free(bmap);
2206 	return NULL;
2207 }
2208 
isl_basic_set_set_to_empty(__isl_take isl_basic_set * bset)2209 __isl_give isl_basic_set *isl_basic_set_set_to_empty(
2210 	__isl_take isl_basic_set *bset)
2211 {
2212 	return bset_from_bmap(isl_basic_map_set_to_empty(bset_to_bmap(bset)));
2213 }
2214 
isl_basic_map_set_rational(__isl_take isl_basic_map * bmap)2215 __isl_give isl_basic_map *isl_basic_map_set_rational(
2216 	__isl_take isl_basic_map *bmap)
2217 {
2218 	if (!bmap)
2219 		return NULL;
2220 
2221 	if (ISL_F_ISSET(bmap, ISL_BASIC_MAP_RATIONAL))
2222 		return bmap;
2223 
2224 	bmap = isl_basic_map_cow(bmap);
2225 	if (!bmap)
2226 		return NULL;
2227 
2228 	ISL_F_SET(bmap, ISL_BASIC_MAP_RATIONAL);
2229 
2230 	return isl_basic_map_finalize(bmap);
2231 }
2232 
isl_basic_set_set_rational(__isl_take isl_basic_set * bset)2233 __isl_give isl_basic_set *isl_basic_set_set_rational(
2234 	__isl_take isl_basic_set *bset)
2235 {
2236 	return isl_basic_map_set_rational(bset);
2237 }
2238 
isl_basic_set_set_integral(__isl_take isl_basic_set * bset)2239 __isl_give isl_basic_set *isl_basic_set_set_integral(
2240 	__isl_take isl_basic_set *bset)
2241 {
2242 	if (!bset)
2243 		return NULL;
2244 
2245 	if (!ISL_F_ISSET(bset, ISL_BASIC_MAP_RATIONAL))
2246 		return bset;
2247 
2248 	bset = isl_basic_set_cow(bset);
2249 	if (!bset)
2250 		return NULL;
2251 
2252 	ISL_F_CLR(bset, ISL_BASIC_MAP_RATIONAL);
2253 
2254 	return isl_basic_set_finalize(bset);
2255 }
2256 
isl_map_set_rational(__isl_take isl_map * map)2257 __isl_give isl_map *isl_map_set_rational(__isl_take isl_map *map)
2258 {
2259 	int i;
2260 
2261 	map = isl_map_cow(map);
2262 	if (!map)
2263 		return NULL;
2264 	for (i = 0; i < map->n; ++i) {
2265 		map->p[i] = isl_basic_map_set_rational(map->p[i]);
2266 		if (!map->p[i])
2267 			goto error;
2268 	}
2269 	return map;
2270 error:
2271 	isl_map_free(map);
2272 	return NULL;
2273 }
2274 
isl_set_set_rational(__isl_take isl_set * set)2275 __isl_give isl_set *isl_set_set_rational(__isl_take isl_set *set)
2276 {
2277 	return isl_map_set_rational(set);
2278 }
2279 
2280 /* Swap divs "a" and "b" in "bmap" (without modifying any of the constraints
2281  * of "bmap").
2282  */
swap_div(__isl_keep isl_basic_map * bmap,int a,int b)2283 static void swap_div(__isl_keep isl_basic_map *bmap, int a, int b)
2284 {
2285 	isl_int *t = bmap->div[a];
2286 	bmap->div[a] = bmap->div[b];
2287 	bmap->div[b] = t;
2288 }
2289 
2290 /* Swap divs "a" and "b" in "bmap" and adjust the constraints and
2291  * div definitions accordingly.
2292  */
isl_basic_map_swap_div(__isl_take isl_basic_map * bmap,int a,int b)2293 __isl_give isl_basic_map *isl_basic_map_swap_div(__isl_take isl_basic_map *bmap,
2294 	int a, int b)
2295 {
2296 	int i;
2297 	isl_size off;
2298 
2299 	off = isl_basic_map_var_offset(bmap, isl_dim_div);
2300 	if (off < 0)
2301 		return isl_basic_map_free(bmap);
2302 
2303 	swap_div(bmap, a, b);
2304 
2305 	for (i = 0; i < bmap->n_eq; ++i)
2306 		isl_int_swap(bmap->eq[i][1+off+a], bmap->eq[i][1+off+b]);
2307 
2308 	for (i = 0; i < bmap->n_ineq; ++i)
2309 		isl_int_swap(bmap->ineq[i][1+off+a], bmap->ineq[i][1+off+b]);
2310 
2311 	for (i = 0; i < bmap->n_div; ++i)
2312 		isl_int_swap(bmap->div[i][1+1+off+a], bmap->div[i][1+1+off+b]);
2313 	ISL_F_CLR(bmap, ISL_BASIC_MAP_SORTED);
2314 
2315 	return bmap;
2316 }
2317 
constraint_drop_vars(isl_int * c,unsigned n,unsigned rem)2318 static void constraint_drop_vars(isl_int *c, unsigned n, unsigned rem)
2319 {
2320 	isl_seq_cpy(c, c + n, rem);
2321 	isl_seq_clr(c + rem, n);
2322 }
2323 
2324 /* Drop n dimensions starting at first.
2325  *
2326  * In principle, this frees up some extra variables as the number
2327  * of columns remains constant, but we would have to extend
2328  * the div array too as the number of rows in this array is assumed
2329  * to be equal to extra.
2330  */
isl_basic_set_drop_dims(__isl_take isl_basic_set * bset,unsigned first,unsigned n)2331 __isl_give isl_basic_set *isl_basic_set_drop_dims(
2332 	__isl_take isl_basic_set *bset, unsigned first, unsigned n)
2333 {
2334 	return isl_basic_map_drop(bset_to_bmap(bset), isl_dim_set, first, n);
2335 }
2336 
2337 /* Move "n" divs starting at "first" to the end of the list of divs.
2338  */
move_divs_last(__isl_take isl_basic_map * bmap,unsigned first,unsigned n)2339 static __isl_give isl_basic_map *move_divs_last(__isl_take isl_basic_map *bmap,
2340 	unsigned first, unsigned n)
2341 {
2342 	isl_int **div;
2343 	int i;
2344 
2345 	if (first + n == bmap->n_div)
2346 		return bmap;
2347 
2348 	div = isl_alloc_array(bmap->ctx, isl_int *, n);
2349 	if (!div)
2350 		goto error;
2351 	for (i = 0; i < n; ++i)
2352 		div[i] = bmap->div[first + i];
2353 	for (i = 0; i < bmap->n_div - first - n; ++i)
2354 		bmap->div[first + i] = bmap->div[first + n + i];
2355 	for (i = 0; i < n; ++i)
2356 		bmap->div[bmap->n_div - n + i] = div[i];
2357 	free(div);
2358 	return bmap;
2359 error:
2360 	isl_basic_map_free(bmap);
2361 	return NULL;
2362 }
2363 
2364 #undef TYPE
2365 #define TYPE	isl_map
2366 static
2367 #include "check_type_range_templ.c"
2368 
2369 /* Check that there are "n" dimensions of type "type" starting at "first"
2370  * in "set".
2371  */
isl_set_check_range(__isl_keep isl_set * set,enum isl_dim_type type,unsigned first,unsigned n)2372 isl_stat isl_set_check_range(__isl_keep isl_set *set,
2373 	enum isl_dim_type type, unsigned first, unsigned n)
2374 {
2375 	return isl_map_check_range(set_to_map(set), type, first, n);
2376 }
2377 
2378 /* Drop "n" dimensions of type "type" starting at "first".
2379  * Perform the core computation, without cowing or
2380  * simplifying and finalizing the result.
2381  *
2382  * In principle, this frees up some extra variables as the number
2383  * of columns remains constant, but we would have to extend
2384  * the div array too as the number of rows in this array is assumed
2385  * to be equal to extra.
2386  */
isl_basic_map_drop_core(__isl_take isl_basic_map * bmap,enum isl_dim_type type,unsigned first,unsigned n)2387 __isl_give isl_basic_map *isl_basic_map_drop_core(
2388 	__isl_take isl_basic_map *bmap, enum isl_dim_type type,
2389 	unsigned first, unsigned n)
2390 {
2391 	int i;
2392 	unsigned offset;
2393 	unsigned left;
2394 	isl_size total;
2395 
2396 	if (isl_basic_map_check_range(bmap, type, first, n) < 0)
2397 		return isl_basic_map_free(bmap);
2398 
2399 	total = isl_basic_map_dim(bmap, isl_dim_all);
2400 	if (total < 0)
2401 		return isl_basic_map_free(bmap);
2402 
2403 	offset = isl_basic_map_offset(bmap, type) + first;
2404 	left = total - (offset - 1) - n;
2405 	for (i = 0; i < bmap->n_eq; ++i)
2406 		constraint_drop_vars(bmap->eq[i]+offset, n, left);
2407 
2408 	for (i = 0; i < bmap->n_ineq; ++i)
2409 		constraint_drop_vars(bmap->ineq[i]+offset, n, left);
2410 
2411 	for (i = 0; i < bmap->n_div; ++i)
2412 		constraint_drop_vars(bmap->div[i]+1+offset, n, left);
2413 
2414 	if (type == isl_dim_div) {
2415 		bmap = move_divs_last(bmap, first, n);
2416 		if (!bmap)
2417 			return NULL;
2418 		if (isl_basic_map_free_div(bmap, n) < 0)
2419 			return isl_basic_map_free(bmap);
2420 	} else
2421 		bmap->dim = isl_space_drop_dims(bmap->dim, type, first, n);
2422 	if (!bmap->dim)
2423 		return isl_basic_map_free(bmap);
2424 
2425 	ISL_F_CLR(bmap, ISL_BASIC_MAP_NO_REDUNDANT);
2426 	ISL_F_CLR(bmap, ISL_BASIC_MAP_SORTED);
2427 	return bmap;
2428 }
2429 
2430 /* Drop "n" dimensions of type "type" starting at "first".
2431  *
2432  * In principle, this frees up some extra variables as the number
2433  * of columns remains constant, but we would have to extend
2434  * the div array too as the number of rows in this array is assumed
2435  * to be equal to extra.
2436  */
isl_basic_map_drop(__isl_take isl_basic_map * bmap,enum isl_dim_type type,unsigned first,unsigned n)2437 __isl_give isl_basic_map *isl_basic_map_drop(__isl_take isl_basic_map *bmap,
2438 	enum isl_dim_type type, unsigned first, unsigned n)
2439 {
2440 	if (!bmap)
2441 		return NULL;
2442 	if (n == 0 && !isl_space_is_named_or_nested(bmap->dim, type))
2443 		return bmap;
2444 
2445 	bmap = isl_basic_map_cow(bmap);
2446 	if (!bmap)
2447 		return NULL;
2448 
2449 	bmap = isl_basic_map_drop_core(bmap, type, first, n);
2450 
2451 	bmap = isl_basic_map_simplify(bmap);
2452 	return isl_basic_map_finalize(bmap);
2453 }
2454 
isl_basic_set_drop(__isl_take isl_basic_set * bset,enum isl_dim_type type,unsigned first,unsigned n)2455 __isl_give isl_basic_set *isl_basic_set_drop(__isl_take isl_basic_set *bset,
2456 	enum isl_dim_type type, unsigned first, unsigned n)
2457 {
2458 	return bset_from_bmap(isl_basic_map_drop(bset_to_bmap(bset),
2459 							type, first, n));
2460 }
2461 
2462 /* No longer consider "map" to be normalized.
2463  */
isl_map_unmark_normalized(__isl_take isl_map * map)2464 static __isl_give isl_map *isl_map_unmark_normalized(__isl_take isl_map *map)
2465 {
2466 	if (!map)
2467 		return NULL;
2468 	ISL_F_CLR(map, ISL_MAP_NORMALIZED);
2469 	return map;
2470 }
2471 
isl_map_drop(__isl_take isl_map * map,enum isl_dim_type type,unsigned first,unsigned n)2472 __isl_give isl_map *isl_map_drop(__isl_take isl_map *map,
2473 	enum isl_dim_type type, unsigned first, unsigned n)
2474 {
2475 	int i;
2476 	isl_space *space;
2477 
2478 	if (isl_map_check_range(map, type, first, n) < 0)
2479 		return isl_map_free(map);
2480 
2481 	if (n == 0 && !isl_space_is_named_or_nested(map->dim, type))
2482 		return map;
2483 	map = isl_map_cow(map);
2484 	if (!map)
2485 		goto error;
2486 
2487 	for (i = 0; i < map->n; ++i) {
2488 		map->p[i] = isl_basic_map_drop(map->p[i], type, first, n);
2489 		if (!map->p[i])
2490 			goto error;
2491 	}
2492 	map = isl_map_unmark_normalized(map);
2493 
2494 	space = isl_map_take_space(map);
2495 	space = isl_space_drop_dims(space, type, first, n);
2496 	map = isl_map_restore_space(map, space);
2497 
2498 	return map;
2499 error:
2500 	isl_map_free(map);
2501 	return NULL;
2502 }
2503 
isl_set_drop(__isl_take isl_set * set,enum isl_dim_type type,unsigned first,unsigned n)2504 __isl_give isl_set *isl_set_drop(__isl_take isl_set *set,
2505 	enum isl_dim_type type, unsigned first, unsigned n)
2506 {
2507 	return set_from_map(isl_map_drop(set_to_map(set), type, first, n));
2508 }
2509 
2510 /* Drop the integer division at position "div", which is assumed
2511  * not to appear in any of the constraints or
2512  * in any of the other integer divisions.
2513  *
2514  * Since the integer division is redundant, there is no need to cow.
2515  */
isl_basic_map_drop_div(__isl_take isl_basic_map * bmap,unsigned div)2516 __isl_give isl_basic_map *isl_basic_map_drop_div(
2517 	__isl_take isl_basic_map *bmap, unsigned div)
2518 {
2519 	return isl_basic_map_drop_core(bmap, isl_dim_div, div, 1);
2520 }
2521 
2522 /* Eliminate the specified n dimensions starting at first from the
2523  * constraints, without removing the dimensions from the space.
2524  * If the set is rational, the dimensions are eliminated using Fourier-Motzkin.
2525  */
isl_map_eliminate(__isl_take isl_map * map,enum isl_dim_type type,unsigned first,unsigned n)2526 __isl_give isl_map *isl_map_eliminate(__isl_take isl_map *map,
2527 	enum isl_dim_type type, unsigned first, unsigned n)
2528 {
2529 	int i;
2530 
2531 	if (n == 0)
2532 		return map;
2533 
2534 	if (isl_map_check_range(map, type, first, n) < 0)
2535 		return isl_map_free(map);
2536 
2537 	map = isl_map_cow(map);
2538 	if (!map)
2539 		return NULL;
2540 
2541 	for (i = 0; i < map->n; ++i) {
2542 		map->p[i] = isl_basic_map_eliminate(map->p[i], type, first, n);
2543 		if (!map->p[i])
2544 			goto error;
2545 	}
2546 	return map;
2547 error:
2548 	isl_map_free(map);
2549 	return NULL;
2550 }
2551 
2552 /* Eliminate the specified n dimensions starting at first from the
2553  * constraints, without removing the dimensions from the space.
2554  * If the set is rational, the dimensions are eliminated using Fourier-Motzkin.
2555  */
isl_set_eliminate(__isl_take isl_set * set,enum isl_dim_type type,unsigned first,unsigned n)2556 __isl_give isl_set *isl_set_eliminate(__isl_take isl_set *set,
2557 	enum isl_dim_type type, unsigned first, unsigned n)
2558 {
2559 	return set_from_map(isl_map_eliminate(set_to_map(set), type, first, n));
2560 }
2561 
2562 /* Eliminate the specified n dimensions starting at first from the
2563  * constraints, without removing the dimensions from the space.
2564  * If the set is rational, the dimensions are eliminated using Fourier-Motzkin.
2565  */
isl_set_eliminate_dims(__isl_take isl_set * set,unsigned first,unsigned n)2566 __isl_give isl_set *isl_set_eliminate_dims(__isl_take isl_set *set,
2567 	unsigned first, unsigned n)
2568 {
2569 	return isl_set_eliminate(set, isl_dim_set, first, n);
2570 }
2571 
isl_basic_map_remove_divs(__isl_take isl_basic_map * bmap)2572 __isl_give isl_basic_map *isl_basic_map_remove_divs(
2573 	__isl_take isl_basic_map *bmap)
2574 {
2575 	isl_size v_div;
2576 
2577 	v_div = isl_basic_map_var_offset(bmap, isl_dim_div);
2578 	if (v_div < 0)
2579 		return isl_basic_map_free(bmap);
2580 	bmap = isl_basic_map_eliminate_vars(bmap, v_div, bmap->n_div);
2581 	if (!bmap)
2582 		return NULL;
2583 	bmap->n_div = 0;
2584 	return isl_basic_map_finalize(bmap);
2585 }
2586 
isl_basic_set_remove_divs(__isl_take isl_basic_set * bset)2587 __isl_give isl_basic_set *isl_basic_set_remove_divs(
2588 	__isl_take isl_basic_set *bset)
2589 {
2590 	return bset_from_bmap(isl_basic_map_remove_divs(bset_to_bmap(bset)));
2591 }
2592 
isl_map_remove_divs(__isl_take isl_map * map)2593 __isl_give isl_map *isl_map_remove_divs(__isl_take isl_map *map)
2594 {
2595 	int i;
2596 
2597 	if (!map)
2598 		return NULL;
2599 	if (map->n == 0)
2600 		return map;
2601 
2602 	map = isl_map_cow(map);
2603 	if (!map)
2604 		return NULL;
2605 
2606 	for (i = 0; i < map->n; ++i) {
2607 		map->p[i] = isl_basic_map_remove_divs(map->p[i]);
2608 		if (!map->p[i])
2609 			goto error;
2610 	}
2611 	return map;
2612 error:
2613 	isl_map_free(map);
2614 	return NULL;
2615 }
2616 
isl_set_remove_divs(__isl_take isl_set * set)2617 __isl_give isl_set *isl_set_remove_divs(__isl_take isl_set *set)
2618 {
2619 	return isl_map_remove_divs(set);
2620 }
2621 
isl_basic_map_remove_dims(__isl_take isl_basic_map * bmap,enum isl_dim_type type,unsigned first,unsigned n)2622 __isl_give isl_basic_map *isl_basic_map_remove_dims(
2623 	__isl_take isl_basic_map *bmap, enum isl_dim_type type,
2624 	unsigned first, unsigned n)
2625 {
2626 	if (isl_basic_map_check_range(bmap, type, first, n) < 0)
2627 		return isl_basic_map_free(bmap);
2628 	if (n == 0 && !isl_space_is_named_or_nested(bmap->dim, type))
2629 		return bmap;
2630 	bmap = isl_basic_map_eliminate_vars(bmap,
2631 			isl_basic_map_offset(bmap, type) - 1 + first, n);
2632 	if (!bmap)
2633 		return bmap;
2634 	if (ISL_F_ISSET(bmap, ISL_BASIC_MAP_EMPTY) && type == isl_dim_div)
2635 		return bmap;
2636 	bmap = isl_basic_map_drop(bmap, type, first, n);
2637 	return bmap;
2638 }
2639 
2640 /* Return true if the definition of the given div (recursively) involves
2641  * any of the given variables.
2642  */
div_involves_vars(__isl_keep isl_basic_map * bmap,int div,unsigned first,unsigned n)2643 static isl_bool div_involves_vars(__isl_keep isl_basic_map *bmap, int div,
2644 	unsigned first, unsigned n)
2645 {
2646 	int i;
2647 	unsigned div_offset = isl_basic_map_offset(bmap, isl_dim_div);
2648 
2649 	if (isl_int_is_zero(bmap->div[div][0]))
2650 		return isl_bool_false;
2651 	if (isl_seq_first_non_zero(bmap->div[div] + 1 + first, n) >= 0)
2652 		return isl_bool_true;
2653 
2654 	for (i = bmap->n_div - 1; i >= 0; --i) {
2655 		isl_bool involves;
2656 
2657 		if (isl_int_is_zero(bmap->div[div][1 + div_offset + i]))
2658 			continue;
2659 		involves = div_involves_vars(bmap, i, first, n);
2660 		if (involves < 0 || involves)
2661 			return involves;
2662 	}
2663 
2664 	return isl_bool_false;
2665 }
2666 
2667 /* Try and add a lower and/or upper bound on "div" to "bmap"
2668  * based on inequality "i".
2669  * "total" is the total number of variables (excluding the divs).
2670  * "v" is a temporary object that can be used during the calculations.
2671  * If "lb" is set, then a lower bound should be constructed.
2672  * If "ub" is set, then an upper bound should be constructed.
2673  *
2674  * The calling function has already checked that the inequality does not
2675  * reference "div", but we still need to check that the inequality is
2676  * of the right form.  We'll consider the case where we want to construct
2677  * a lower bound.  The construction of upper bounds is similar.
2678  *
2679  * Let "div" be of the form
2680  *
2681  *	q = floor((a + f(x))/d)
2682  *
2683  * We essentially check if constraint "i" is of the form
2684  *
2685  *	b + f(x) >= 0
2686  *
2687  * so that we can use it to derive a lower bound on "div".
2688  * However, we allow a slightly more general form
2689  *
2690  *	b + g(x) >= 0
2691  *
2692  * with the condition that the coefficients of g(x) - f(x) are all
2693  * divisible by d.
2694  * Rewriting this constraint as
2695  *
2696  *	0 >= -b - g(x)
2697  *
2698  * adding a + f(x) to both sides and dividing by d, we obtain
2699  *
2700  *	(a + f(x))/d >= (a-b)/d + (f(x)-g(x))/d
2701  *
2702  * Taking the floor on both sides, we obtain
2703  *
2704  *	q >= floor((a-b)/d) + (f(x)-g(x))/d
2705  *
2706  * or
2707  *
2708  *	(g(x)-f(x))/d + ceil((b-a)/d) + q >= 0
2709  *
2710  * In the case of an upper bound, we construct the constraint
2711  *
2712  *	(g(x)+f(x))/d + floor((b+a)/d) - q >= 0
2713  *
2714  */
insert_bounds_on_div_from_ineq(__isl_take isl_basic_map * bmap,int div,int i,unsigned total,isl_int v,int lb,int ub)2715 static __isl_give isl_basic_map *insert_bounds_on_div_from_ineq(
2716 	__isl_take isl_basic_map *bmap, int div, int i,
2717 	unsigned total, isl_int v, int lb, int ub)
2718 {
2719 	int j;
2720 
2721 	for (j = 0; (lb || ub) && j < total + bmap->n_div; ++j) {
2722 		if (lb) {
2723 			isl_int_sub(v, bmap->ineq[i][1 + j],
2724 					bmap->div[div][1 + 1 + j]);
2725 			lb = isl_int_is_divisible_by(v, bmap->div[div][0]);
2726 		}
2727 		if (ub) {
2728 			isl_int_add(v, bmap->ineq[i][1 + j],
2729 					bmap->div[div][1 + 1 + j]);
2730 			ub = isl_int_is_divisible_by(v, bmap->div[div][0]);
2731 		}
2732 	}
2733 	if (!lb && !ub)
2734 		return bmap;
2735 
2736 	bmap = isl_basic_map_cow(bmap);
2737 	bmap = isl_basic_map_extend_constraints(bmap, 0, lb + ub);
2738 	if (lb) {
2739 		int k = isl_basic_map_alloc_inequality(bmap);
2740 		if (k < 0)
2741 			goto error;
2742 		for (j = 0; j < 1 + total + bmap->n_div; ++j) {
2743 			isl_int_sub(bmap->ineq[k][j], bmap->ineq[i][j],
2744 					bmap->div[div][1 + j]);
2745 			isl_int_cdiv_q(bmap->ineq[k][j],
2746 					bmap->ineq[k][j], bmap->div[div][0]);
2747 		}
2748 		isl_int_set_si(bmap->ineq[k][1 + total + div], 1);
2749 	}
2750 	if (ub) {
2751 		int k = isl_basic_map_alloc_inequality(bmap);
2752 		if (k < 0)
2753 			goto error;
2754 		for (j = 0; j < 1 + total + bmap->n_div; ++j) {
2755 			isl_int_add(bmap->ineq[k][j], bmap->ineq[i][j],
2756 					bmap->div[div][1 + j]);
2757 			isl_int_fdiv_q(bmap->ineq[k][j],
2758 					bmap->ineq[k][j], bmap->div[div][0]);
2759 		}
2760 		isl_int_set_si(bmap->ineq[k][1 + total + div], -1);
2761 	}
2762 
2763 	return bmap;
2764 error:
2765 	isl_basic_map_free(bmap);
2766 	return NULL;
2767 }
2768 
2769 /* This function is called right before "div" is eliminated from "bmap"
2770  * using Fourier-Motzkin.
2771  * Look through the constraints of "bmap" for constraints on the argument
2772  * of the integer division and use them to construct constraints on the
2773  * integer division itself.  These constraints can then be combined
2774  * during the Fourier-Motzkin elimination.
2775  * Note that it is only useful to introduce lower bounds on "div"
2776  * if "bmap" already contains upper bounds on "div" as the newly
2777  * introduce lower bounds can then be combined with the pre-existing
2778  * upper bounds.  Similarly for upper bounds.
2779  * We therefore first check if "bmap" contains any lower and/or upper bounds
2780  * on "div".
2781  *
2782  * It is interesting to note that the introduction of these constraints
2783  * can indeed lead to more accurate results, even when compared to
2784  * deriving constraints on the argument of "div" from constraints on "div".
2785  * Consider, for example, the set
2786  *
2787  *	{ [i,j,k] : 3 + i + 2j >= 0 and 2 * [(i+2j)/4] <= k }
2788  *
2789  * The second constraint can be rewritten as
2790  *
2791  *	2 * [(-i-2j+3)/4] + k >= 0
2792  *
2793  * from which we can derive
2794  *
2795  *	-i - 2j + 3 >= -2k
2796  *
2797  * or
2798  *
2799  *	i + 2j <= 3 + 2k
2800  *
2801  * Combined with the first constraint, we obtain
2802  *
2803  *	-3 <= 3 + 2k	or	k >= -3
2804  *
2805  * If, on the other hand we derive a constraint on [(i+2j)/4] from
2806  * the first constraint, we obtain
2807  *
2808  *	[(i + 2j)/4] >= [-3/4] = -1
2809  *
2810  * Combining this constraint with the second constraint, we obtain
2811  *
2812  *	k >= -2
2813  */
insert_bounds_on_div(__isl_take isl_basic_map * bmap,int div)2814 static __isl_give isl_basic_map *insert_bounds_on_div(
2815 	__isl_take isl_basic_map *bmap, int div)
2816 {
2817 	int i;
2818 	int check_lb, check_ub;
2819 	isl_int v;
2820 	isl_size v_div;
2821 
2822 	if (!bmap)
2823 		return NULL;
2824 
2825 	if (isl_int_is_zero(bmap->div[div][0]))
2826 		return bmap;
2827 
2828 	v_div = isl_basic_map_var_offset(bmap, isl_dim_div);
2829 	if (v_div < 0)
2830 		return isl_basic_map_free(bmap);
2831 
2832 	check_lb = 0;
2833 	check_ub = 0;
2834 	for (i = 0; (!check_lb || !check_ub) && i < bmap->n_ineq; ++i) {
2835 		int s = isl_int_sgn(bmap->ineq[i][1 + v_div + div]);
2836 		if (s > 0)
2837 			check_ub = 1;
2838 		if (s < 0)
2839 			check_lb = 1;
2840 	}
2841 
2842 	if (!check_lb && !check_ub)
2843 		return bmap;
2844 
2845 	isl_int_init(v);
2846 
2847 	for (i = 0; bmap && i < bmap->n_ineq; ++i) {
2848 		if (!isl_int_is_zero(bmap->ineq[i][1 + v_div + div]))
2849 			continue;
2850 
2851 		bmap = insert_bounds_on_div_from_ineq(bmap, div, i, v_div, v,
2852 							check_lb, check_ub);
2853 	}
2854 
2855 	isl_int_clear(v);
2856 
2857 	return bmap;
2858 }
2859 
2860 /* Remove all divs (recursively) involving any of the given dimensions
2861  * in their definitions.
2862  */
isl_basic_map_remove_divs_involving_dims(__isl_take isl_basic_map * bmap,enum isl_dim_type type,unsigned first,unsigned n)2863 __isl_give isl_basic_map *isl_basic_map_remove_divs_involving_dims(
2864 	__isl_take isl_basic_map *bmap,
2865 	enum isl_dim_type type, unsigned first, unsigned n)
2866 {
2867 	int i;
2868 
2869 	if (isl_basic_map_check_range(bmap, type, first, n) < 0)
2870 		return isl_basic_map_free(bmap);
2871 	first += isl_basic_map_offset(bmap, type);
2872 
2873 	for (i = bmap->n_div - 1; i >= 0; --i) {
2874 		isl_bool involves;
2875 
2876 		involves = div_involves_vars(bmap, i, first, n);
2877 		if (involves < 0)
2878 			return isl_basic_map_free(bmap);
2879 		if (!involves)
2880 			continue;
2881 		bmap = insert_bounds_on_div(bmap, i);
2882 		bmap = isl_basic_map_remove_dims(bmap, isl_dim_div, i, 1);
2883 		if (!bmap)
2884 			return NULL;
2885 		i = bmap->n_div;
2886 	}
2887 
2888 	return bmap;
2889 }
2890 
isl_basic_set_remove_divs_involving_dims(__isl_take isl_basic_set * bset,enum isl_dim_type type,unsigned first,unsigned n)2891 __isl_give isl_basic_set *isl_basic_set_remove_divs_involving_dims(
2892 	__isl_take isl_basic_set *bset,
2893 	enum isl_dim_type type, unsigned first, unsigned n)
2894 {
2895 	return isl_basic_map_remove_divs_involving_dims(bset, type, first, n);
2896 }
2897 
isl_map_remove_divs_involving_dims(__isl_take isl_map * map,enum isl_dim_type type,unsigned first,unsigned n)2898 __isl_give isl_map *isl_map_remove_divs_involving_dims(__isl_take isl_map *map,
2899 	enum isl_dim_type type, unsigned first, unsigned n)
2900 {
2901 	int i;
2902 
2903 	if (!map)
2904 		return NULL;
2905 	if (map->n == 0)
2906 		return map;
2907 
2908 	map = isl_map_cow(map);
2909 	if (!map)
2910 		return NULL;
2911 
2912 	for (i = 0; i < map->n; ++i) {
2913 		map->p[i] = isl_basic_map_remove_divs_involving_dims(map->p[i],
2914 								type, first, n);
2915 		if (!map->p[i])
2916 			goto error;
2917 	}
2918 	return map;
2919 error:
2920 	isl_map_free(map);
2921 	return NULL;
2922 }
2923 
isl_set_remove_divs_involving_dims(__isl_take isl_set * set,enum isl_dim_type type,unsigned first,unsigned n)2924 __isl_give isl_set *isl_set_remove_divs_involving_dims(__isl_take isl_set *set,
2925 	enum isl_dim_type type, unsigned first, unsigned n)
2926 {
2927 	return set_from_map(isl_map_remove_divs_involving_dims(set_to_map(set),
2928 							      type, first, n));
2929 }
2930 
2931 /* Does the description of "bmap" depend on the specified dimensions?
2932  * We also check whether the dimensions appear in any of the div definitions.
2933  * In principle there is no need for this check.  If the dimensions appear
2934  * in a div definition, they also appear in the defining constraints of that
2935  * div.
2936  */
isl_basic_map_involves_dims(__isl_keep isl_basic_map * bmap,enum isl_dim_type type,unsigned first,unsigned n)2937 isl_bool isl_basic_map_involves_dims(__isl_keep isl_basic_map *bmap,
2938 	enum isl_dim_type type, unsigned first, unsigned n)
2939 {
2940 	int i;
2941 
2942 	if (isl_basic_map_check_range(bmap, type, first, n) < 0)
2943 		return isl_bool_error;
2944 
2945 	first += isl_basic_map_offset(bmap, type);
2946 	for (i = 0; i < bmap->n_eq; ++i)
2947 		if (isl_seq_first_non_zero(bmap->eq[i] + first, n) >= 0)
2948 			return isl_bool_true;
2949 	for (i = 0; i < bmap->n_ineq; ++i)
2950 		if (isl_seq_first_non_zero(bmap->ineq[i] + first, n) >= 0)
2951 			return isl_bool_true;
2952 	for (i = 0; i < bmap->n_div; ++i) {
2953 		if (isl_int_is_zero(bmap->div[i][0]))
2954 			continue;
2955 		if (isl_seq_first_non_zero(bmap->div[i] + 1 + first, n) >= 0)
2956 			return isl_bool_true;
2957 	}
2958 
2959 	return isl_bool_false;
2960 }
2961 
isl_map_involves_dims(__isl_keep isl_map * map,enum isl_dim_type type,unsigned first,unsigned n)2962 isl_bool isl_map_involves_dims(__isl_keep isl_map *map,
2963 	enum isl_dim_type type, unsigned first, unsigned n)
2964 {
2965 	int i;
2966 
2967 	if (isl_map_check_range(map, type, first, n) < 0)
2968 		return isl_bool_error;
2969 
2970 	for (i = 0; i < map->n; ++i) {
2971 		isl_bool involves = isl_basic_map_involves_dims(map->p[i],
2972 							    type, first, n);
2973 		if (involves < 0 || involves)
2974 			return involves;
2975 	}
2976 
2977 	return isl_bool_false;
2978 }
2979 
isl_basic_set_involves_dims(__isl_keep isl_basic_set * bset,enum isl_dim_type type,unsigned first,unsigned n)2980 isl_bool isl_basic_set_involves_dims(__isl_keep isl_basic_set *bset,
2981 	enum isl_dim_type type, unsigned first, unsigned n)
2982 {
2983 	return isl_basic_map_involves_dims(bset, type, first, n);
2984 }
2985 
isl_set_involves_dims(__isl_keep isl_set * set,enum isl_dim_type type,unsigned first,unsigned n)2986 isl_bool isl_set_involves_dims(__isl_keep isl_set *set,
2987 	enum isl_dim_type type, unsigned first, unsigned n)
2988 {
2989 	return isl_map_involves_dims(set, type, first, n);
2990 }
2991 
2992 /* Does "bset" involve any local variables, i.e., integer divisions?
2993  */
isl_basic_set_involves_locals(__isl_keep isl_basic_set * bset)2994 static isl_bool isl_basic_set_involves_locals(__isl_keep isl_basic_set *bset)
2995 {
2996 	isl_size n;
2997 
2998 	n = isl_basic_set_dim(bset, isl_dim_div);
2999 	if (n < 0)
3000 		return isl_bool_error;
3001 	return isl_bool_ok(n > 0);
3002 }
3003 
3004 /* isl_set_every_basic_set callback that checks whether "bset"
3005  * is free of local variables.
3006  */
basic_set_no_locals(__isl_keep isl_basic_set * bset,void * user)3007 static isl_bool basic_set_no_locals(__isl_keep isl_basic_set *bset, void *user)
3008 {
3009 	return isl_bool_not(isl_basic_set_involves_locals(bset));
3010 }
3011 
3012 /* Does "set" involve any local variables, i.e., integer divisions?
3013  */
isl_set_involves_locals(__isl_keep isl_set * set)3014 isl_bool isl_set_involves_locals(__isl_keep isl_set *set)
3015 {
3016 	isl_bool no_locals;
3017 
3018 	no_locals = isl_set_every_basic_set(set, &basic_set_no_locals, NULL);
3019 	return isl_bool_not(no_locals);
3020 }
3021 
3022 /* Drop all constraints in bmap that involve any of the dimensions
3023  * first to first+n-1.
3024  * This function only performs the actual removal of constraints.
3025  *
3026  * This function should not call finalize since it is used by
3027  * remove_redundant_divs, which in turn is called by isl_basic_map_finalize.
3028  */
isl_basic_map_drop_constraints_involving(__isl_take isl_basic_map * bmap,unsigned first,unsigned n)3029 __isl_give isl_basic_map *isl_basic_map_drop_constraints_involving(
3030 	__isl_take isl_basic_map *bmap, unsigned first, unsigned n)
3031 {
3032 	int i;
3033 
3034 	if (n == 0)
3035 		return bmap;
3036 
3037 	bmap = isl_basic_map_cow(bmap);
3038 
3039 	if (!bmap)
3040 		return NULL;
3041 
3042 	for (i = bmap->n_eq - 1; i >= 0; --i) {
3043 		if (isl_seq_first_non_zero(bmap->eq[i] + 1 + first, n) == -1)
3044 			continue;
3045 		if (isl_basic_map_drop_equality(bmap, i) < 0)
3046 			return isl_basic_map_free(bmap);
3047 	}
3048 
3049 	for (i = bmap->n_ineq - 1; i >= 0; --i) {
3050 		if (isl_seq_first_non_zero(bmap->ineq[i] + 1 + first, n) == -1)
3051 			continue;
3052 		if (isl_basic_map_drop_inequality(bmap, i) < 0)
3053 			return isl_basic_map_free(bmap);
3054 	}
3055 
3056 	return bmap;
3057 }
3058 
3059 /* Drop all constraints in bset that involve any of the dimensions
3060  * first to first+n-1.
3061  * This function only performs the actual removal of constraints.
3062  */
isl_basic_set_drop_constraints_involving(__isl_take isl_basic_set * bset,unsigned first,unsigned n)3063 __isl_give isl_basic_set *isl_basic_set_drop_constraints_involving(
3064 	__isl_take isl_basic_set *bset, unsigned first, unsigned n)
3065 {
3066 	return isl_basic_map_drop_constraints_involving(bset, first, n);
3067 }
3068 
3069 /* Drop all constraints in bmap that do not involve any of the dimensions
3070  * first to first + n - 1 of the given type.
3071  */
isl_basic_map_drop_constraints_not_involving_dims(__isl_take isl_basic_map * bmap,enum isl_dim_type type,unsigned first,unsigned n)3072 __isl_give isl_basic_map *isl_basic_map_drop_constraints_not_involving_dims(
3073 	__isl_take isl_basic_map *bmap,
3074 	enum isl_dim_type type, unsigned first, unsigned n)
3075 {
3076 	int i;
3077 
3078 	if (n == 0) {
3079 		isl_space *space = isl_basic_map_get_space(bmap);
3080 		isl_basic_map_free(bmap);
3081 		return isl_basic_map_universe(space);
3082 	}
3083 	bmap = isl_basic_map_cow(bmap);
3084 	if (!bmap)
3085 		return NULL;
3086 
3087 	if (isl_basic_map_check_range(bmap, type, first, n) < 0)
3088 		return isl_basic_map_free(bmap);
3089 
3090 	first += isl_basic_map_offset(bmap, type) - 1;
3091 
3092 	for (i = bmap->n_eq - 1; i >= 0; --i) {
3093 		if (isl_seq_first_non_zero(bmap->eq[i] + 1 + first, n) != -1)
3094 			continue;
3095 		if (isl_basic_map_drop_equality(bmap, i) < 0)
3096 			return isl_basic_map_free(bmap);
3097 	}
3098 
3099 	for (i = bmap->n_ineq - 1; i >= 0; --i) {
3100 		if (isl_seq_first_non_zero(bmap->ineq[i] + 1 + first, n) != -1)
3101 			continue;
3102 		if (isl_basic_map_drop_inequality(bmap, i) < 0)
3103 			return isl_basic_map_free(bmap);
3104 	}
3105 
3106 	bmap = isl_basic_map_add_known_div_constraints(bmap);
3107 	return bmap;
3108 }
3109 
3110 /* Drop all constraints in bset that do not involve any of the dimensions
3111  * first to first + n - 1 of the given type.
3112  */
isl_basic_set_drop_constraints_not_involving_dims(__isl_take isl_basic_set * bset,enum isl_dim_type type,unsigned first,unsigned n)3113 __isl_give isl_basic_set *isl_basic_set_drop_constraints_not_involving_dims(
3114 	__isl_take isl_basic_set *bset,
3115 	enum isl_dim_type type, unsigned first, unsigned n)
3116 {
3117 	return isl_basic_map_drop_constraints_not_involving_dims(bset,
3118 							    type, first, n);
3119 }
3120 
3121 /* Drop all constraints in bmap that involve any of the dimensions
3122  * first to first + n - 1 of the given type.
3123  */
isl_basic_map_drop_constraints_involving_dims(__isl_take isl_basic_map * bmap,enum isl_dim_type type,unsigned first,unsigned n)3124 __isl_give isl_basic_map *isl_basic_map_drop_constraints_involving_dims(
3125 	__isl_take isl_basic_map *bmap,
3126 	enum isl_dim_type type, unsigned first, unsigned n)
3127 {
3128 	if (!bmap)
3129 		return NULL;
3130 	if (n == 0)
3131 		return bmap;
3132 
3133 	if (isl_basic_map_check_range(bmap, type, first, n) < 0)
3134 		return isl_basic_map_free(bmap);
3135 
3136 	bmap = isl_basic_map_remove_divs_involving_dims(bmap, type, first, n);
3137 	first += isl_basic_map_offset(bmap, type) - 1;
3138 	bmap = isl_basic_map_drop_constraints_involving(bmap, first, n);
3139 	bmap = isl_basic_map_add_known_div_constraints(bmap);
3140 	return bmap;
3141 }
3142 
3143 /* Drop all constraints in bset that involve any of the dimensions
3144  * first to first + n - 1 of the given type.
3145  */
isl_basic_set_drop_constraints_involving_dims(__isl_take isl_basic_set * bset,enum isl_dim_type type,unsigned first,unsigned n)3146 __isl_give isl_basic_set *isl_basic_set_drop_constraints_involving_dims(
3147 	__isl_take isl_basic_set *bset,
3148 	enum isl_dim_type type, unsigned first, unsigned n)
3149 {
3150 	return isl_basic_map_drop_constraints_involving_dims(bset,
3151 							    type, first, n);
3152 }
3153 
3154 /* Drop constraints from "map" by applying "drop" to each basic map.
3155  */
drop_constraints(__isl_take isl_map * map,enum isl_dim_type type,unsigned first,unsigned n,__isl_give isl_basic_map * (* drop)(__isl_take isl_basic_map * bmap,enum isl_dim_type type,unsigned first,unsigned n))3156 static __isl_give isl_map *drop_constraints(__isl_take isl_map *map,
3157 	enum isl_dim_type type, unsigned first, unsigned n,
3158 	__isl_give isl_basic_map *(*drop)(__isl_take isl_basic_map *bmap,
3159 		enum isl_dim_type type, unsigned first, unsigned n))
3160 {
3161 	int i;
3162 
3163 	if (isl_map_check_range(map, type, first, n) < 0)
3164 		return isl_map_free(map);
3165 
3166 	map = isl_map_cow(map);
3167 	if (!map)
3168 		return NULL;
3169 
3170 	for (i = 0; i < map->n; ++i) {
3171 		map->p[i] = drop(map->p[i], type, first, n);
3172 		if (!map->p[i])
3173 			return isl_map_free(map);
3174 	}
3175 
3176 	if (map->n > 1)
3177 		ISL_F_CLR(map, ISL_MAP_DISJOINT);
3178 
3179 	return map;
3180 }
3181 
3182 /* Drop all constraints in map that involve any of the dimensions
3183  * first to first + n - 1 of the given type.
3184  */
isl_map_drop_constraints_involving_dims(__isl_take isl_map * map,enum isl_dim_type type,unsigned first,unsigned n)3185 __isl_give isl_map *isl_map_drop_constraints_involving_dims(
3186 	__isl_take isl_map *map,
3187 	enum isl_dim_type type, unsigned first, unsigned n)
3188 {
3189 	if (n == 0)
3190 		return map;
3191 	return drop_constraints(map, type, first, n,
3192 				&isl_basic_map_drop_constraints_involving_dims);
3193 }
3194 
3195 /* Drop all constraints in "map" that do not involve any of the dimensions
3196  * first to first + n - 1 of the given type.
3197  */
isl_map_drop_constraints_not_involving_dims(__isl_take isl_map * map,enum isl_dim_type type,unsigned first,unsigned n)3198 __isl_give isl_map *isl_map_drop_constraints_not_involving_dims(
3199 	__isl_take isl_map *map,
3200 	enum isl_dim_type type, unsigned first, unsigned n)
3201 {
3202 	if (n == 0) {
3203 		isl_space *space = isl_map_get_space(map);
3204 		isl_map_free(map);
3205 		return isl_map_universe(space);
3206 	}
3207 	return drop_constraints(map, type, first, n,
3208 			    &isl_basic_map_drop_constraints_not_involving_dims);
3209 }
3210 
3211 /* Drop all constraints in set that involve any of the dimensions
3212  * first to first + n - 1 of the given type.
3213  */
isl_set_drop_constraints_involving_dims(__isl_take isl_set * set,enum isl_dim_type type,unsigned first,unsigned n)3214 __isl_give isl_set *isl_set_drop_constraints_involving_dims(
3215 	__isl_take isl_set *set,
3216 	enum isl_dim_type type, unsigned first, unsigned n)
3217 {
3218 	return isl_map_drop_constraints_involving_dims(set, type, first, n);
3219 }
3220 
3221 /* Drop all constraints in "set" that do not involve any of the dimensions
3222  * first to first + n - 1 of the given type.
3223  */
isl_set_drop_constraints_not_involving_dims(__isl_take isl_set * set,enum isl_dim_type type,unsigned first,unsigned n)3224 __isl_give isl_set *isl_set_drop_constraints_not_involving_dims(
3225 	__isl_take isl_set *set,
3226 	enum isl_dim_type type, unsigned first, unsigned n)
3227 {
3228 	return isl_map_drop_constraints_not_involving_dims(set, type, first, n);
3229 }
3230 
3231 /* Does local variable "div" of "bmap" have a complete explicit representation?
3232  * Having a complete explicit representation requires not only
3233  * an explicit representation, but also that all local variables
3234  * that appear in this explicit representation in turn have
3235  * a complete explicit representation.
3236  */
isl_basic_map_div_is_known(__isl_keep isl_basic_map * bmap,int div)3237 isl_bool isl_basic_map_div_is_known(__isl_keep isl_basic_map *bmap, int div)
3238 {
3239 	int i;
3240 	unsigned div_offset = isl_basic_map_offset(bmap, isl_dim_div);
3241 	isl_bool marked;
3242 
3243 	marked = isl_basic_map_div_is_marked_unknown(bmap, div);
3244 	if (marked < 0 || marked)
3245 		return isl_bool_not(marked);
3246 
3247 	for (i = bmap->n_div - 1; i >= 0; --i) {
3248 		isl_bool known;
3249 
3250 		if (isl_int_is_zero(bmap->div[div][1 + div_offset + i]))
3251 			continue;
3252 		known = isl_basic_map_div_is_known(bmap, i);
3253 		if (known < 0 || !known)
3254 			return known;
3255 	}
3256 
3257 	return isl_bool_true;
3258 }
3259 
3260 /* Remove all divs that are unknown or defined in terms of unknown divs.
3261  */
isl_basic_map_remove_unknown_divs(__isl_take isl_basic_map * bmap)3262 __isl_give isl_basic_map *isl_basic_map_remove_unknown_divs(
3263 	__isl_take isl_basic_map *bmap)
3264 {
3265 	int i;
3266 
3267 	if (!bmap)
3268 		return NULL;
3269 
3270 	for (i = bmap->n_div - 1; i >= 0; --i) {
3271 		if (isl_basic_map_div_is_known(bmap, i))
3272 			continue;
3273 		bmap = isl_basic_map_remove_dims(bmap, isl_dim_div, i, 1);
3274 		if (!bmap)
3275 			return NULL;
3276 		i = bmap->n_div;
3277 	}
3278 
3279 	return bmap;
3280 }
3281 
3282 /* Remove all divs that are unknown or defined in terms of unknown divs.
3283  */
isl_basic_set_remove_unknown_divs(__isl_take isl_basic_set * bset)3284 __isl_give isl_basic_set *isl_basic_set_remove_unknown_divs(
3285 	__isl_take isl_basic_set *bset)
3286 {
3287 	return isl_basic_map_remove_unknown_divs(bset);
3288 }
3289 
isl_map_remove_unknown_divs(__isl_take isl_map * map)3290 __isl_give isl_map *isl_map_remove_unknown_divs(__isl_take isl_map *map)
3291 {
3292 	int i;
3293 
3294 	if (!map)
3295 		return NULL;
3296 	if (map->n == 0)
3297 		return map;
3298 
3299 	map = isl_map_cow(map);
3300 	if (!map)
3301 		return NULL;
3302 
3303 	for (i = 0; i < map->n; ++i) {
3304 		map->p[i] = isl_basic_map_remove_unknown_divs(map->p[i]);
3305 		if (!map->p[i])
3306 			goto error;
3307 	}
3308 	return map;
3309 error:
3310 	isl_map_free(map);
3311 	return NULL;
3312 }
3313 
isl_set_remove_unknown_divs(__isl_take isl_set * set)3314 __isl_give isl_set *isl_set_remove_unknown_divs(__isl_take isl_set *set)
3315 {
3316 	return set_from_map(isl_map_remove_unknown_divs(set_to_map(set)));
3317 }
3318 
isl_basic_set_remove_dims(__isl_take isl_basic_set * bset,enum isl_dim_type type,unsigned first,unsigned n)3319 __isl_give isl_basic_set *isl_basic_set_remove_dims(
3320 	__isl_take isl_basic_set *bset,
3321 	enum isl_dim_type type, unsigned first, unsigned n)
3322 {
3323 	isl_basic_map *bmap = bset_to_bmap(bset);
3324 	bmap = isl_basic_map_remove_dims(bmap, type, first, n);
3325 	return bset_from_bmap(bmap);
3326 }
3327 
isl_map_remove_dims(__isl_take isl_map * map,enum isl_dim_type type,unsigned first,unsigned n)3328 __isl_give isl_map *isl_map_remove_dims(__isl_take isl_map *map,
3329 	enum isl_dim_type type, unsigned first, unsigned n)
3330 {
3331 	int i;
3332 
3333 	if (n == 0)
3334 		return map;
3335 
3336 	map = isl_map_cow(map);
3337 	if (isl_map_check_range(map, type, first, n) < 0)
3338 		return isl_map_free(map);
3339 
3340 	for (i = 0; i < map->n; ++i) {
3341 		map->p[i] = isl_basic_map_eliminate_vars(map->p[i],
3342 			isl_basic_map_offset(map->p[i], type) - 1 + first, n);
3343 		if (!map->p[i])
3344 			goto error;
3345 	}
3346 	map = isl_map_drop(map, type, first, n);
3347 	return map;
3348 error:
3349 	isl_map_free(map);
3350 	return NULL;
3351 }
3352 
isl_set_remove_dims(__isl_take isl_set * bset,enum isl_dim_type type,unsigned first,unsigned n)3353 __isl_give isl_set *isl_set_remove_dims(__isl_take isl_set *bset,
3354 	enum isl_dim_type type, unsigned first, unsigned n)
3355 {
3356 	return set_from_map(isl_map_remove_dims(set_to_map(bset),
3357 						type, first, n));
3358 }
3359 
3360 /* Project out n inputs starting at first using Fourier-Motzkin */
isl_map_remove_inputs(__isl_take isl_map * map,unsigned first,unsigned n)3361 __isl_give isl_map *isl_map_remove_inputs(__isl_take isl_map *map,
3362 	unsigned first, unsigned n)
3363 {
3364 	return isl_map_remove_dims(map, isl_dim_in, first, n);
3365 }
3366 
isl_basic_set_print_internal(__isl_keep isl_basic_set * bset,FILE * out,int indent)3367 void isl_basic_set_print_internal(__isl_keep isl_basic_set *bset,
3368 	FILE *out, int indent)
3369 {
3370 	isl_printer *p;
3371 
3372 	if (!bset) {
3373 		fprintf(out, "null basic set\n");
3374 		return;
3375 	}
3376 
3377 	fprintf(out, "%*s", indent, "");
3378 	fprintf(out, "ref: %d, nparam: %d, dim: %d, extra: %d, flags: %x\n",
3379 			bset->ref, bset->dim->nparam, bset->dim->n_out,
3380 			bset->extra, bset->flags);
3381 
3382 	p = isl_printer_to_file(isl_basic_set_get_ctx(bset), out);
3383 	p = isl_printer_set_dump(p, 1);
3384 	p = isl_printer_set_indent(p, indent);
3385 	p = isl_printer_start_line(p);
3386 	p = isl_printer_print_basic_set(p, bset);
3387 	p = isl_printer_end_line(p);
3388 	isl_printer_free(p);
3389 }
3390 
isl_basic_map_print_internal(__isl_keep isl_basic_map * bmap,FILE * out,int indent)3391 void isl_basic_map_print_internal(__isl_keep isl_basic_map *bmap,
3392 	FILE *out, int indent)
3393 {
3394 	isl_printer *p;
3395 
3396 	if (!bmap) {
3397 		fprintf(out, "null basic map\n");
3398 		return;
3399 	}
3400 
3401 	fprintf(out, "%*s", indent, "");
3402 	fprintf(out, "ref: %d, nparam: %d, in: %d, out: %d, extra: %d, "
3403 			"flags: %x, n_name: %d\n",
3404 		bmap->ref,
3405 		bmap->dim->nparam, bmap->dim->n_in, bmap->dim->n_out,
3406 		bmap->extra, bmap->flags, bmap->dim->n_id);
3407 
3408 	p = isl_printer_to_file(isl_basic_map_get_ctx(bmap), out);
3409 	p = isl_printer_set_dump(p, 1);
3410 	p = isl_printer_set_indent(p, indent);
3411 	p = isl_printer_start_line(p);
3412 	p = isl_printer_print_basic_map(p, bmap);
3413 	p = isl_printer_end_line(p);
3414 	isl_printer_free(p);
3415 }
3416 
isl_inequality_negate(__isl_take isl_basic_map * bmap,unsigned pos)3417 __isl_give isl_basic_map *isl_inequality_negate(__isl_take isl_basic_map *bmap,
3418 	unsigned pos)
3419 {
3420 	isl_size total;
3421 
3422 	total = isl_basic_map_dim(bmap, isl_dim_all);
3423 	if (total < 0)
3424 		return isl_basic_map_free(bmap);
3425 	if (pos >= bmap->n_ineq)
3426 		isl_die(isl_basic_map_get_ctx(bmap), isl_error_invalid,
3427 			"invalid position", return isl_basic_map_free(bmap));
3428 	isl_seq_neg(bmap->ineq[pos], bmap->ineq[pos], 1 + total);
3429 	isl_int_sub_ui(bmap->ineq[pos][0], bmap->ineq[pos][0], 1);
3430 	ISL_F_CLR(bmap, ISL_BASIC_MAP_NO_REDUNDANT);
3431 	ISL_F_CLR(bmap, ISL_BASIC_MAP_SORTED);
3432 	return bmap;
3433 }
3434 
isl_set_alloc_space(__isl_take isl_space * space,int n,unsigned flags)3435 __isl_give isl_set *isl_set_alloc_space(__isl_take isl_space *space, int n,
3436 	unsigned flags)
3437 {
3438 	if (isl_space_check_is_set(space) < 0)
3439 		goto error;
3440 	return isl_map_alloc_space(space, n, flags);
3441 error:
3442 	isl_space_free(space);
3443 	return NULL;
3444 }
3445 
3446 /* Make sure "map" has room for at least "n" more basic maps.
3447  */
isl_map_grow(__isl_take isl_map * map,int n)3448 __isl_give isl_map *isl_map_grow(__isl_take isl_map *map, int n)
3449 {
3450 	int i;
3451 	struct isl_map *grown = NULL;
3452 
3453 	if (!map)
3454 		return NULL;
3455 	isl_assert(map->ctx, n >= 0, goto error);
3456 	if (map->n + n <= map->size)
3457 		return map;
3458 	grown = isl_map_alloc_space(isl_map_get_space(map), map->n + n, map->flags);
3459 	if (!grown)
3460 		goto error;
3461 	for (i = 0; i < map->n; ++i) {
3462 		grown->p[i] = isl_basic_map_copy(map->p[i]);
3463 		if (!grown->p[i])
3464 			goto error;
3465 		grown->n++;
3466 	}
3467 	isl_map_free(map);
3468 	return grown;
3469 error:
3470 	isl_map_free(grown);
3471 	isl_map_free(map);
3472 	return NULL;
3473 }
3474 
3475 /* Make sure "set" has room for at least "n" more basic sets.
3476  */
isl_set_grow(__isl_take isl_set * set,int n)3477 __isl_give isl_set *isl_set_grow(__isl_take isl_set *set, int n)
3478 {
3479 	return set_from_map(isl_map_grow(set_to_map(set), n));
3480 }
3481 
isl_set_from_basic_set(__isl_take isl_basic_set * bset)3482 __isl_give isl_set *isl_set_from_basic_set(__isl_take isl_basic_set *bset)
3483 {
3484 	return isl_map_from_basic_map(bset);
3485 }
3486 
3487 /* This function performs the same operation as isl_set_from_basic_set,
3488  * but is considered as a function on an isl_basic_set when exported.
3489  */
isl_basic_set_to_set(__isl_take isl_basic_set * bset)3490 __isl_give isl_set *isl_basic_set_to_set(__isl_take isl_basic_set *bset)
3491 {
3492 	return isl_set_from_basic_set(bset);
3493 }
3494 
isl_map_from_basic_map(__isl_take isl_basic_map * bmap)3495 __isl_give isl_map *isl_map_from_basic_map(__isl_take isl_basic_map *bmap)
3496 {
3497 	struct isl_map *map;
3498 
3499 	if (!bmap)
3500 		return NULL;
3501 
3502 	map = isl_map_alloc_space(isl_space_copy(bmap->dim), 1, ISL_MAP_DISJOINT);
3503 	return isl_map_add_basic_map(map, bmap);
3504 }
3505 
isl_set_add_basic_set(__isl_take isl_set * set,__isl_take isl_basic_set * bset)3506 __isl_give isl_set *isl_set_add_basic_set(__isl_take isl_set *set,
3507 						__isl_take isl_basic_set *bset)
3508 {
3509 	return set_from_map(isl_map_add_basic_map(set_to_map(set),
3510 						bset_to_bmap(bset)));
3511 }
3512 
isl_set_free(__isl_take isl_set * set)3513 __isl_null isl_set *isl_set_free(__isl_take isl_set *set)
3514 {
3515 	return isl_map_free(set);
3516 }
3517 
isl_set_print_internal(__isl_keep isl_set * set,FILE * out,int indent)3518 void isl_set_print_internal(__isl_keep isl_set *set, FILE *out, int indent)
3519 {
3520 	int i;
3521 
3522 	if (!set) {
3523 		fprintf(out, "null set\n");
3524 		return;
3525 	}
3526 
3527 	fprintf(out, "%*s", indent, "");
3528 	fprintf(out, "ref: %d, n: %d, nparam: %d, dim: %d, flags: %x\n",
3529 			set->ref, set->n, set->dim->nparam, set->dim->n_out,
3530 			set->flags);
3531 	for (i = 0; i < set->n; ++i) {
3532 		fprintf(out, "%*s", indent, "");
3533 		fprintf(out, "basic set %d:\n", i);
3534 		isl_basic_set_print_internal(set->p[i], out, indent+4);
3535 	}
3536 }
3537 
isl_map_print_internal(__isl_keep isl_map * map,FILE * out,int indent)3538 void isl_map_print_internal(__isl_keep isl_map *map, FILE *out, int indent)
3539 {
3540 	int i;
3541 
3542 	if (!map) {
3543 		fprintf(out, "null map\n");
3544 		return;
3545 	}
3546 
3547 	fprintf(out, "%*s", indent, "");
3548 	fprintf(out, "ref: %d, n: %d, nparam: %d, in: %d, out: %d, "
3549 		     "flags: %x, n_name: %d\n",
3550 			map->ref, map->n, map->dim->nparam, map->dim->n_in,
3551 			map->dim->n_out, map->flags, map->dim->n_id);
3552 	for (i = 0; i < map->n; ++i) {
3553 		fprintf(out, "%*s", indent, "");
3554 		fprintf(out, "basic map %d:\n", i);
3555 		isl_basic_map_print_internal(map->p[i], out, indent+4);
3556 	}
3557 }
3558 
3559 /* Check that the space of "bset" is the same as that of the domain of "bmap".
3560  */
isl_basic_map_check_compatible_domain(__isl_keep isl_basic_map * bmap,__isl_keep isl_basic_set * bset)3561 static isl_stat isl_basic_map_check_compatible_domain(
3562 	__isl_keep isl_basic_map *bmap, __isl_keep isl_basic_set *bset)
3563 {
3564 	isl_bool ok;
3565 
3566 	ok = isl_basic_map_compatible_domain(bmap, bset);
3567 	if (ok < 0)
3568 		return isl_stat_error;
3569 	if (!ok)
3570 		isl_die(isl_basic_set_get_ctx(bset), isl_error_invalid,
3571 			"incompatible spaces", return isl_stat_error);
3572 
3573 	return isl_stat_ok;
3574 }
3575 
isl_basic_map_intersect_domain(__isl_take isl_basic_map * bmap,__isl_take isl_basic_set * bset)3576 __isl_give isl_basic_map *isl_basic_map_intersect_domain(
3577 	__isl_take isl_basic_map *bmap, __isl_take isl_basic_set *bset)
3578 {
3579 	struct isl_basic_map *bmap_domain;
3580 	isl_size dim;
3581 
3582 	if (isl_basic_map_check_equal_params(bmap, bset_to_bmap(bset)) < 0)
3583 		goto error;
3584 
3585 	dim = isl_basic_set_dim(bset, isl_dim_set);
3586 	if (dim < 0)
3587 		goto error;
3588 	if (dim != 0 &&
3589 	    isl_basic_map_check_compatible_domain(bmap, bset) < 0)
3590 		goto error;
3591 
3592 	bmap = isl_basic_map_cow(bmap);
3593 	if (!bmap)
3594 		goto error;
3595 	bmap = isl_basic_map_extend(bmap,
3596 			bset->n_div, bset->n_eq, bset->n_ineq);
3597 	bmap_domain = isl_basic_map_from_domain(bset);
3598 	bmap = add_constraints(bmap, bmap_domain, 0, 0);
3599 
3600 	bmap = isl_basic_map_simplify(bmap);
3601 	return isl_basic_map_finalize(bmap);
3602 error:
3603 	isl_basic_map_free(bmap);
3604 	isl_basic_set_free(bset);
3605 	return NULL;
3606 }
3607 
3608 /* Check that the space of "bset" is the same as that of the range of "bmap".
3609  */
isl_basic_map_check_compatible_range(__isl_keep isl_basic_map * bmap,__isl_keep isl_basic_set * bset)3610 static isl_stat isl_basic_map_check_compatible_range(
3611 	__isl_keep isl_basic_map *bmap, __isl_keep isl_basic_set *bset)
3612 {
3613 	isl_bool ok;
3614 
3615 	ok = isl_basic_map_compatible_range(bmap, bset);
3616 	if (ok < 0)
3617 		return isl_stat_error;
3618 	if (!ok)
3619 		isl_die(isl_basic_set_get_ctx(bset), isl_error_invalid,
3620 			"incompatible spaces", return isl_stat_error);
3621 
3622 	return isl_stat_ok;
3623 }
3624 
isl_basic_map_intersect_range(__isl_take isl_basic_map * bmap,__isl_take isl_basic_set * bset)3625 __isl_give isl_basic_map *isl_basic_map_intersect_range(
3626 	__isl_take isl_basic_map *bmap, __isl_take isl_basic_set *bset)
3627 {
3628 	struct isl_basic_map *bmap_range;
3629 	isl_size dim;
3630 
3631 	if (isl_basic_map_check_equal_params(bmap, bset_to_bmap(bset)) < 0)
3632 		goto error;
3633 
3634 	dim = isl_basic_set_dim(bset, isl_dim_set);
3635 	if (dim < 0)
3636 		goto error;
3637 	if (dim != 0 && isl_basic_map_check_compatible_range(bmap, bset) < 0)
3638 		goto error;
3639 
3640 	if (isl_basic_set_plain_is_universe(bset)) {
3641 		isl_basic_set_free(bset);
3642 		return bmap;
3643 	}
3644 
3645 	bmap = isl_basic_map_cow(bmap);
3646 	if (!bmap)
3647 		goto error;
3648 	bmap = isl_basic_map_extend(bmap,
3649 			bset->n_div, bset->n_eq, bset->n_ineq);
3650 	bmap_range = bset_to_bmap(bset);
3651 	bmap = add_constraints(bmap, bmap_range, 0, 0);
3652 
3653 	bmap = isl_basic_map_simplify(bmap);
3654 	return isl_basic_map_finalize(bmap);
3655 error:
3656 	isl_basic_map_free(bmap);
3657 	isl_basic_set_free(bset);
3658 	return NULL;
3659 }
3660 
isl_basic_map_contains(__isl_keep isl_basic_map * bmap,__isl_keep isl_vec * vec)3661 isl_bool isl_basic_map_contains(__isl_keep isl_basic_map *bmap,
3662 	__isl_keep isl_vec *vec)
3663 {
3664 	int i;
3665 	isl_size total;
3666 	isl_int s;
3667 
3668 	total = isl_basic_map_dim(bmap, isl_dim_all);
3669 	if (total < 0 || !vec)
3670 		return isl_bool_error;
3671 
3672 	if (1 + total != vec->size)
3673 		return isl_bool_false;
3674 
3675 	isl_int_init(s);
3676 
3677 	for (i = 0; i < bmap->n_eq; ++i) {
3678 		isl_seq_inner_product(vec->el, bmap->eq[i], 1 + total, &s);
3679 		if (!isl_int_is_zero(s)) {
3680 			isl_int_clear(s);
3681 			return isl_bool_false;
3682 		}
3683 	}
3684 
3685 	for (i = 0; i < bmap->n_ineq; ++i) {
3686 		isl_seq_inner_product(vec->el, bmap->ineq[i], 1 + total, &s);
3687 		if (isl_int_is_neg(s)) {
3688 			isl_int_clear(s);
3689 			return isl_bool_false;
3690 		}
3691 	}
3692 
3693 	isl_int_clear(s);
3694 
3695 	return isl_bool_true;
3696 }
3697 
isl_basic_set_contains(__isl_keep isl_basic_set * bset,__isl_keep isl_vec * vec)3698 isl_bool isl_basic_set_contains(__isl_keep isl_basic_set *bset,
3699 	__isl_keep isl_vec *vec)
3700 {
3701 	return isl_basic_map_contains(bset_to_bmap(bset), vec);
3702 }
3703 
isl_basic_map_intersect(__isl_take isl_basic_map * bmap1,__isl_take isl_basic_map * bmap2)3704 __isl_give isl_basic_map *isl_basic_map_intersect(
3705 	__isl_take isl_basic_map *bmap1, __isl_take isl_basic_map *bmap2)
3706 {
3707 	struct isl_vec *sample = NULL;
3708 	isl_space *space1, *space2;
3709 	isl_size dim1, dim2, nparam1, nparam2;
3710 
3711 	if (isl_basic_map_check_equal_params(bmap1, bmap2) < 0)
3712 		goto error;
3713 	space1 = isl_basic_map_peek_space(bmap1);
3714 	space2 = isl_basic_map_peek_space(bmap2);
3715 	dim1 = isl_space_dim(space1, isl_dim_all);
3716 	dim2 = isl_space_dim(space2, isl_dim_all);
3717 	nparam1 = isl_space_dim(space1, isl_dim_param);
3718 	nparam2 = isl_space_dim(space2, isl_dim_param);
3719 	if (dim1 < 0 || dim2 < 0 || nparam1 < 0 || nparam2 < 0)
3720 		goto error;
3721 	if (dim1 == nparam1 && dim2 != nparam2)
3722 		return isl_basic_map_intersect(bmap2, bmap1);
3723 
3724 	if (dim2 != nparam2 &&
3725 	    isl_basic_map_check_equal_space(bmap1, bmap2) < 0)
3726 		goto error;
3727 
3728 	if (isl_basic_map_plain_is_empty(bmap1)) {
3729 		isl_basic_map_free(bmap2);
3730 		return bmap1;
3731 	}
3732 	if (isl_basic_map_plain_is_empty(bmap2)) {
3733 		isl_basic_map_free(bmap1);
3734 		return bmap2;
3735 	}
3736 
3737 	if (bmap1->sample &&
3738 	    isl_basic_map_contains(bmap1, bmap1->sample) > 0 &&
3739 	    isl_basic_map_contains(bmap2, bmap1->sample) > 0)
3740 		sample = isl_vec_copy(bmap1->sample);
3741 	else if (bmap2->sample &&
3742 	    isl_basic_map_contains(bmap1, bmap2->sample) > 0 &&
3743 	    isl_basic_map_contains(bmap2, bmap2->sample) > 0)
3744 		sample = isl_vec_copy(bmap2->sample);
3745 
3746 	bmap1 = isl_basic_map_cow(bmap1);
3747 	if (!bmap1)
3748 		goto error;
3749 	bmap1 = isl_basic_map_extend(bmap1,
3750 			bmap2->n_div, bmap2->n_eq, bmap2->n_ineq);
3751 	bmap1 = add_constraints(bmap1, bmap2, 0, 0);
3752 
3753 	if (!bmap1)
3754 		isl_vec_free(sample);
3755 	else if (sample) {
3756 		isl_vec_free(bmap1->sample);
3757 		bmap1->sample = sample;
3758 	}
3759 
3760 	bmap1 = isl_basic_map_simplify(bmap1);
3761 	return isl_basic_map_finalize(bmap1);
3762 error:
3763 	if (sample)
3764 		isl_vec_free(sample);
3765 	isl_basic_map_free(bmap1);
3766 	isl_basic_map_free(bmap2);
3767 	return NULL;
3768 }
3769 
isl_basic_set_intersect(__isl_take isl_basic_set * bset1,__isl_take isl_basic_set * bset2)3770 __isl_give isl_basic_set *isl_basic_set_intersect(
3771 	__isl_take isl_basic_set *bset1, __isl_take isl_basic_set *bset2)
3772 {
3773 	return bset_from_bmap(isl_basic_map_intersect(bset_to_bmap(bset1),
3774 							bset_to_bmap(bset2)));
3775 }
3776 
isl_basic_set_intersect_params(__isl_take isl_basic_set * bset1,__isl_take isl_basic_set * bset2)3777 __isl_give isl_basic_set *isl_basic_set_intersect_params(
3778 	__isl_take isl_basic_set *bset1, __isl_take isl_basic_set *bset2)
3779 {
3780 	return isl_basic_set_intersect(bset1, bset2);
3781 }
3782 
3783 /* Does "map" consist of a single disjunct, without any local variables?
3784  */
is_convex_no_locals(__isl_keep isl_map * map)3785 static isl_bool is_convex_no_locals(__isl_keep isl_map *map)
3786 {
3787 	isl_size n_div;
3788 
3789 	if (!map)
3790 		return isl_bool_error;
3791 	if (map->n != 1)
3792 		return isl_bool_false;
3793 	n_div = isl_basic_map_dim(map->p[0], isl_dim_div);
3794 	if (n_div < 0)
3795 		return isl_bool_error;
3796 	if (n_div != 0)
3797 		return isl_bool_false;
3798 	return isl_bool_true;
3799 }
3800 
3801 /* Check that "map" consists of a single disjunct, without any local variables.
3802  */
check_convex_no_locals(__isl_keep isl_map * map)3803 static isl_stat check_convex_no_locals(__isl_keep isl_map *map)
3804 {
3805 	isl_bool ok;
3806 
3807 	ok = is_convex_no_locals(map);
3808 	if (ok < 0)
3809 		return isl_stat_error;
3810 	if (ok)
3811 		return isl_stat_ok;
3812 
3813 	isl_die(isl_map_get_ctx(map), isl_error_internal,
3814 		"unexpectedly not convex or involving local variables",
3815 		return isl_stat_error);
3816 }
3817 
3818 /* Special case of isl_map_intersect, where both map1 and map2
3819  * are convex, without any divs and such that either map1 or map2
3820  * contains a single constraint.  This constraint is then simply
3821  * added to the other map.
3822  */
map_intersect_add_constraint(__isl_take isl_map * map1,__isl_take isl_map * map2)3823 static __isl_give isl_map *map_intersect_add_constraint(
3824 	__isl_take isl_map *map1, __isl_take isl_map *map2)
3825 {
3826 	if (check_convex_no_locals(map1) < 0 ||
3827 	    check_convex_no_locals(map2) < 0)
3828 		goto error;
3829 
3830 	if (map2->p[0]->n_eq + map2->p[0]->n_ineq != 1)
3831 		return isl_map_intersect(map2, map1);
3832 
3833 	map1 = isl_map_cow(map1);
3834 	if (!map1)
3835 		goto error;
3836 	if (isl_map_plain_is_empty(map1)) {
3837 		isl_map_free(map2);
3838 		return map1;
3839 	}
3840 	if (map2->p[0]->n_eq == 1)
3841 		map1->p[0] = isl_basic_map_add_eq(map1->p[0], map2->p[0]->eq[0]);
3842 	else
3843 		map1->p[0] = isl_basic_map_add_ineq(map1->p[0],
3844 							map2->p[0]->ineq[0]);
3845 
3846 	map1->p[0] = isl_basic_map_simplify(map1->p[0]);
3847 	map1->p[0] = isl_basic_map_finalize(map1->p[0]);
3848 	if (!map1->p[0])
3849 		goto error;
3850 
3851 	if (isl_basic_map_plain_is_empty(map1->p[0])) {
3852 		isl_basic_map_free(map1->p[0]);
3853 		map1->n = 0;
3854 	}
3855 
3856 	isl_map_free(map2);
3857 
3858 	map1 = isl_map_unmark_normalized(map1);
3859 	return map1;
3860 error:
3861 	isl_map_free(map1);
3862 	isl_map_free(map2);
3863 	return NULL;
3864 }
3865 
3866 /* map2 may be either a parameter domain or a map living in the same
3867  * space as map1.
3868  */
map_intersect_internal(__isl_take isl_map * map1,__isl_take isl_map * map2)3869 static __isl_give isl_map *map_intersect_internal(__isl_take isl_map *map1,
3870 	__isl_take isl_map *map2)
3871 {
3872 	unsigned flags = 0;
3873 	isl_bool equal;
3874 	isl_map *result;
3875 	int i, j;
3876 	isl_size dim2, nparam2;
3877 
3878 	if (!map1 || !map2)
3879 		goto error;
3880 
3881 	if ((isl_map_plain_is_empty(map1) ||
3882 	     isl_map_plain_is_universe(map2)) &&
3883 	    isl_space_is_equal(map1->dim, map2->dim)) {
3884 		isl_map_free(map2);
3885 		return map1;
3886 	}
3887 	if ((isl_map_plain_is_empty(map2) ||
3888 	     isl_map_plain_is_universe(map1)) &&
3889 	    isl_space_is_equal(map1->dim, map2->dim)) {
3890 		isl_map_free(map1);
3891 		return map2;
3892 	}
3893 
3894 	if (is_convex_no_locals(map1) == isl_bool_true &&
3895 	    is_convex_no_locals(map2) == isl_bool_true &&
3896 	    isl_space_is_equal(map1->dim, map2->dim) &&
3897 	    (map1->p[0]->n_eq + map1->p[0]->n_ineq == 1 ||
3898 	     map2->p[0]->n_eq + map2->p[0]->n_ineq == 1))
3899 		return map_intersect_add_constraint(map1, map2);
3900 
3901 	equal = isl_map_plain_is_equal(map1, map2);
3902 	if (equal < 0)
3903 		goto error;
3904 	if (equal) {
3905 		isl_map_free(map2);
3906 		return map1;
3907 	}
3908 
3909 	dim2 = isl_map_dim(map2, isl_dim_all);
3910 	nparam2 = isl_map_dim(map2, isl_dim_param);
3911 	if (dim2 < 0 || nparam2 < 0)
3912 		goto error;
3913 	if (dim2 != nparam2)
3914 		isl_assert(map1->ctx,
3915 			    isl_space_is_equal(map1->dim, map2->dim), goto error);
3916 
3917 	if (ISL_F_ISSET(map1, ISL_MAP_DISJOINT) &&
3918 	    ISL_F_ISSET(map2, ISL_MAP_DISJOINT))
3919 		ISL_FL_SET(flags, ISL_MAP_DISJOINT);
3920 
3921 	result = isl_map_alloc_space(isl_space_copy(map1->dim),
3922 				map1->n * map2->n, flags);
3923 	if (!result)
3924 		goto error;
3925 	for (i = 0; i < map1->n; ++i)
3926 		for (j = 0; j < map2->n; ++j) {
3927 			struct isl_basic_map *part;
3928 			part = isl_basic_map_intersect(
3929 				    isl_basic_map_copy(map1->p[i]),
3930 				    isl_basic_map_copy(map2->p[j]));
3931 			if (isl_basic_map_is_empty(part) < 0)
3932 				part = isl_basic_map_free(part);
3933 			result = isl_map_add_basic_map(result, part);
3934 			if (!result)
3935 				goto error;
3936 		}
3937 	isl_map_free(map1);
3938 	isl_map_free(map2);
3939 	return result;
3940 error:
3941 	isl_map_free(map1);
3942 	isl_map_free(map2);
3943 	return NULL;
3944 }
3945 
map_intersect(__isl_take isl_map * map1,__isl_take isl_map * map2)3946 static __isl_give isl_map *map_intersect(__isl_take isl_map *map1,
3947 	__isl_take isl_map *map2)
3948 {
3949 	if (isl_map_check_equal_space(map1, map2) < 0)
3950 		goto error;
3951 	return map_intersect_internal(map1, map2);
3952 error:
3953 	isl_map_free(map1);
3954 	isl_map_free(map2);
3955 	return NULL;
3956 }
3957 
isl_map_intersect(__isl_take isl_map * map1,__isl_take isl_map * map2)3958 __isl_give isl_map *isl_map_intersect(__isl_take isl_map *map1,
3959 	__isl_take isl_map *map2)
3960 {
3961 	isl_map_align_params_bin(&map1, &map2);
3962 	return map_intersect(map1, map2);
3963 }
3964 
isl_set_intersect(__isl_take isl_set * set1,__isl_take isl_set * set2)3965 __isl_give isl_set *isl_set_intersect(__isl_take isl_set *set1,
3966 	__isl_take isl_set *set2)
3967 {
3968 	return set_from_map(isl_map_intersect(set_to_map(set1),
3969 					      set_to_map(set2)));
3970 }
3971 
3972 /* map_intersect_internal accepts intersections
3973  * with parameter domains, so we can just call that function.
3974  */
isl_map_intersect_params(__isl_take isl_map * map,__isl_take isl_set * params)3975 __isl_give isl_map *isl_map_intersect_params(__isl_take isl_map *map,
3976 	__isl_take isl_set *params)
3977 {
3978 	isl_map_align_params_set(&map, &params);
3979 	return map_intersect_internal(map, params);
3980 }
3981 
isl_set_intersect_params(__isl_take isl_set * set,__isl_take isl_set * params)3982 __isl_give isl_set *isl_set_intersect_params(__isl_take isl_set *set,
3983 		__isl_take isl_set *params)
3984 {
3985 	return isl_map_intersect_params(set, params);
3986 }
3987 
isl_basic_map_reverse(__isl_take isl_basic_map * bmap)3988 __isl_give isl_basic_map *isl_basic_map_reverse(__isl_take isl_basic_map *bmap)
3989 {
3990 	isl_space *space;
3991 	unsigned pos;
3992 	isl_size n1, n2;
3993 
3994 	if (!bmap)
3995 		return NULL;
3996 	bmap = isl_basic_map_cow(bmap);
3997 	if (!bmap)
3998 		return NULL;
3999 	space = isl_space_reverse(isl_space_copy(bmap->dim));
4000 	pos = isl_basic_map_offset(bmap, isl_dim_in);
4001 	n1 = isl_basic_map_dim(bmap, isl_dim_in);
4002 	n2 = isl_basic_map_dim(bmap, isl_dim_out);
4003 	if (n1 < 0 || n2 < 0)
4004 		bmap = isl_basic_map_free(bmap);
4005 	bmap = isl_basic_map_swap_vars(bmap, pos, n1, n2);
4006 	return isl_basic_map_reset_space(bmap, space);
4007 }
4008 
4009 /* Given a basic map A -> (B -> C), return the corresponding basic map
4010  * A -> (C -> B).
4011  */
isl_basic_map_range_reverse(__isl_take isl_basic_map * bmap)4012 static __isl_give isl_basic_map *isl_basic_map_range_reverse(
4013 	__isl_take isl_basic_map *bmap)
4014 {
4015 	isl_space *space;
4016 	isl_size offset, n1, n2;
4017 
4018 	space = isl_basic_map_peek_space(bmap);
4019 	if (isl_space_check_range_is_wrapping(space) < 0)
4020 		return isl_basic_map_free(bmap);
4021 	offset = isl_basic_map_var_offset(bmap, isl_dim_out);
4022 	n1 = isl_space_wrapped_dim(space, isl_dim_out, isl_dim_in);
4023 	n2 = isl_space_wrapped_dim(space, isl_dim_out, isl_dim_out);
4024 	if (offset < 0 || n1 < 0 || n2 < 0)
4025 		return isl_basic_map_free(bmap);
4026 
4027 	bmap = isl_basic_map_swap_vars(bmap, 1 + offset, n1, n2);
4028 
4029 	space = isl_basic_map_take_space(bmap);
4030 	space = isl_space_range_reverse(space);
4031 	bmap = isl_basic_map_restore_space(bmap, space);
4032 
4033 	return bmap;
4034 }
4035 
basic_map_space_reset(__isl_take isl_basic_map * bmap,enum isl_dim_type type)4036 static __isl_give isl_basic_map *basic_map_space_reset(
4037 	__isl_take isl_basic_map *bmap, enum isl_dim_type type)
4038 {
4039 	isl_space *space;
4040 
4041 	if (!bmap)
4042 		return NULL;
4043 	if (!isl_space_is_named_or_nested(bmap->dim, type))
4044 		return bmap;
4045 
4046 	space = isl_basic_map_get_space(bmap);
4047 	space = isl_space_reset(space, type);
4048 	bmap = isl_basic_map_reset_space(bmap, space);
4049 	return bmap;
4050 }
4051 
isl_basic_map_insert_dims(__isl_take isl_basic_map * bmap,enum isl_dim_type type,unsigned pos,unsigned n)4052 __isl_give isl_basic_map *isl_basic_map_insert_dims(
4053 	__isl_take isl_basic_map *bmap, enum isl_dim_type type,
4054 	unsigned pos, unsigned n)
4055 {
4056 	isl_bool rational, is_empty;
4057 	isl_space *res_space;
4058 	struct isl_basic_map *res;
4059 	struct isl_dim_map *dim_map;
4060 	isl_size total;
4061 	unsigned off;
4062 	enum isl_dim_type t;
4063 
4064 	if (n == 0)
4065 		return basic_map_space_reset(bmap, type);
4066 
4067 	is_empty = isl_basic_map_plain_is_empty(bmap);
4068 	total = isl_basic_map_dim(bmap, isl_dim_all);
4069 	if (is_empty < 0 || total < 0)
4070 		return isl_basic_map_free(bmap);
4071 	res_space = isl_space_insert_dims(isl_basic_map_get_space(bmap),
4072 					type, pos, n);
4073 	if (!res_space)
4074 		return isl_basic_map_free(bmap);
4075 	if (is_empty) {
4076 		isl_basic_map_free(bmap);
4077 		return isl_basic_map_empty(res_space);
4078 	}
4079 
4080 	dim_map = isl_dim_map_alloc(bmap->ctx, total + n);
4081 	off = 0;
4082 	for (t = isl_dim_param; t <= isl_dim_out; ++t) {
4083 		isl_size dim;
4084 
4085 		if (t != type) {
4086 			isl_dim_map_dim(dim_map, bmap->dim, t, off);
4087 		} else {
4088 			isl_size size = isl_basic_map_dim(bmap, t);
4089 			if (size < 0)
4090 				dim_map = isl_dim_map_free(dim_map);
4091 			isl_dim_map_dim_range(dim_map, bmap->dim, t,
4092 						0, pos, off);
4093 			isl_dim_map_dim_range(dim_map, bmap->dim, t,
4094 						pos, size - pos, off + pos + n);
4095 		}
4096 		dim = isl_space_dim(res_space, t);
4097 		if (dim < 0)
4098 			dim_map = isl_dim_map_free(dim_map);
4099 		off += dim;
4100 	}
4101 	isl_dim_map_div(dim_map, bmap, off);
4102 
4103 	res = isl_basic_map_alloc_space(res_space,
4104 			bmap->n_div, bmap->n_eq, bmap->n_ineq);
4105 	rational = isl_basic_map_is_rational(bmap);
4106 	if (rational < 0)
4107 		res = isl_basic_map_free(res);
4108 	if (rational)
4109 		res = isl_basic_map_set_rational(res);
4110 	res = isl_basic_map_add_constraints_dim_map(res, bmap, dim_map);
4111 	return isl_basic_map_finalize(res);
4112 }
4113 
isl_basic_set_insert_dims(__isl_take isl_basic_set * bset,enum isl_dim_type type,unsigned pos,unsigned n)4114 __isl_give isl_basic_set *isl_basic_set_insert_dims(
4115 	__isl_take isl_basic_set *bset,
4116 	enum isl_dim_type type, unsigned pos, unsigned n)
4117 {
4118 	return isl_basic_map_insert_dims(bset, type, pos, n);
4119 }
4120 
isl_basic_map_add_dims(__isl_take isl_basic_map * bmap,enum isl_dim_type type,unsigned n)4121 __isl_give isl_basic_map *isl_basic_map_add_dims(__isl_take isl_basic_map *bmap,
4122 		enum isl_dim_type type, unsigned n)
4123 {
4124 	isl_size dim;
4125 
4126 	dim = isl_basic_map_dim(bmap, type);
4127 	if (dim < 0)
4128 		return isl_basic_map_free(bmap);
4129 	return isl_basic_map_insert_dims(bmap, type, dim, n);
4130 }
4131 
isl_basic_set_add_dims(__isl_take isl_basic_set * bset,enum isl_dim_type type,unsigned n)4132 __isl_give isl_basic_set *isl_basic_set_add_dims(__isl_take isl_basic_set *bset,
4133 		enum isl_dim_type type, unsigned n)
4134 {
4135 	if (!bset)
4136 		return NULL;
4137 	isl_assert(bset->ctx, type != isl_dim_in, goto error);
4138 	return isl_basic_map_add_dims(bset, type, n);
4139 error:
4140 	isl_basic_set_free(bset);
4141 	return NULL;
4142 }
4143 
map_space_reset(__isl_take isl_map * map,enum isl_dim_type type)4144 static __isl_give isl_map *map_space_reset(__isl_take isl_map *map,
4145 	enum isl_dim_type type)
4146 {
4147 	isl_space *space;
4148 
4149 	if (!map || !isl_space_is_named_or_nested(map->dim, type))
4150 		return map;
4151 
4152 	space = isl_map_get_space(map);
4153 	space = isl_space_reset(space, type);
4154 	map = isl_map_reset_space(map, space);
4155 	return map;
4156 }
4157 
isl_map_insert_dims(__isl_take isl_map * map,enum isl_dim_type type,unsigned pos,unsigned n)4158 __isl_give isl_map *isl_map_insert_dims(__isl_take isl_map *map,
4159 		enum isl_dim_type type, unsigned pos, unsigned n)
4160 {
4161 	int i;
4162 	isl_space *space;
4163 
4164 	if (n == 0)
4165 		return map_space_reset(map, type);
4166 
4167 	map = isl_map_cow(map);
4168 	if (!map)
4169 		return NULL;
4170 
4171 	for (i = 0; i < map->n; ++i) {
4172 		map->p[i] = isl_basic_map_insert_dims(map->p[i], type, pos, n);
4173 		if (!map->p[i])
4174 			goto error;
4175 	}
4176 
4177 	space = isl_map_take_space(map);
4178 	space = isl_space_insert_dims(space, type, pos, n);
4179 	map = isl_map_restore_space(map, space);
4180 
4181 	return map;
4182 error:
4183 	isl_map_free(map);
4184 	return NULL;
4185 }
4186 
isl_set_insert_dims(__isl_take isl_set * set,enum isl_dim_type type,unsigned pos,unsigned n)4187 __isl_give isl_set *isl_set_insert_dims(__isl_take isl_set *set,
4188 		enum isl_dim_type type, unsigned pos, unsigned n)
4189 {
4190 	return isl_map_insert_dims(set, type, pos, n);
4191 }
4192 
isl_map_add_dims(__isl_take isl_map * map,enum isl_dim_type type,unsigned n)4193 __isl_give isl_map *isl_map_add_dims(__isl_take isl_map *map,
4194 		enum isl_dim_type type, unsigned n)
4195 {
4196 	isl_size dim;
4197 
4198 	dim = isl_map_dim(map, type);
4199 	if (dim < 0)
4200 		return isl_map_free(map);
4201 	return isl_map_insert_dims(map, type, dim, n);
4202 }
4203 
isl_set_add_dims(__isl_take isl_set * set,enum isl_dim_type type,unsigned n)4204 __isl_give isl_set *isl_set_add_dims(__isl_take isl_set *set,
4205 		enum isl_dim_type type, unsigned n)
4206 {
4207 	if (!set)
4208 		return NULL;
4209 	isl_assert(set->ctx, type != isl_dim_in, goto error);
4210 	return set_from_map(isl_map_add_dims(set_to_map(set), type, n));
4211 error:
4212 	isl_set_free(set);
4213 	return NULL;
4214 }
4215 
isl_basic_map_move_dims(__isl_take isl_basic_map * bmap,enum isl_dim_type dst_type,unsigned dst_pos,enum isl_dim_type src_type,unsigned src_pos,unsigned n)4216 __isl_give isl_basic_map *isl_basic_map_move_dims(
4217 	__isl_take isl_basic_map *bmap,
4218 	enum isl_dim_type dst_type, unsigned dst_pos,
4219 	enum isl_dim_type src_type, unsigned src_pos, unsigned n)
4220 {
4221 	isl_space *space;
4222 	struct isl_dim_map *dim_map;
4223 	struct isl_basic_map *res;
4224 	enum isl_dim_type t;
4225 	isl_size total;
4226 	unsigned off;
4227 
4228 	if (!bmap)
4229 		return NULL;
4230 	if (n == 0) {
4231 		bmap = isl_basic_map_reset(bmap, src_type);
4232 		bmap = isl_basic_map_reset(bmap, dst_type);
4233 		return bmap;
4234 	}
4235 
4236 	if (isl_basic_map_check_range(bmap, src_type, src_pos, n) < 0)
4237 		return isl_basic_map_free(bmap);
4238 
4239 	if (dst_type == src_type && dst_pos == src_pos)
4240 		return bmap;
4241 
4242 	isl_assert(bmap->ctx, dst_type != src_type, goto error);
4243 
4244 	if (pos(bmap->dim, dst_type) + dst_pos ==
4245 	    pos(bmap->dim, src_type) + src_pos +
4246 					    ((src_type < dst_type) ? n : 0)) {
4247 		space = isl_basic_map_take_space(bmap);
4248 		space = isl_space_move_dims(space, dst_type, dst_pos,
4249 						src_type, src_pos, n);
4250 		bmap = isl_basic_map_restore_space(bmap, space);
4251 		bmap = isl_basic_map_finalize(bmap);
4252 
4253 		return bmap;
4254 	}
4255 
4256 	total = isl_basic_map_dim(bmap, isl_dim_all);
4257 	if (total < 0)
4258 		return isl_basic_map_free(bmap);
4259 	dim_map = isl_dim_map_alloc(bmap->ctx, total);
4260 
4261 	off = 0;
4262 	space = isl_basic_map_peek_space(bmap);
4263 	for (t = isl_dim_param; t <= isl_dim_out; ++t) {
4264 		isl_size size = isl_space_dim(space, t);
4265 		if (size < 0)
4266 			dim_map = isl_dim_map_free(dim_map);
4267 		if (t == dst_type) {
4268 			isl_dim_map_dim_range(dim_map, space, t,
4269 					    0, dst_pos, off);
4270 			off += dst_pos;
4271 			isl_dim_map_dim_range(dim_map, space, src_type,
4272 					    src_pos, n, off);
4273 			off += n;
4274 			isl_dim_map_dim_range(dim_map, space, t,
4275 					    dst_pos, size - dst_pos, off);
4276 			off += size - dst_pos;
4277 		} else if (t == src_type) {
4278 			isl_dim_map_dim_range(dim_map, space, t,
4279 					    0, src_pos, off);
4280 			off += src_pos;
4281 			isl_dim_map_dim_range(dim_map, space, t,
4282 					src_pos + n, size - src_pos - n, off);
4283 			off += size - src_pos - n;
4284 		} else {
4285 			isl_dim_map_dim(dim_map, space, t, off);
4286 			off += size;
4287 		}
4288 	}
4289 	isl_dim_map_div(dim_map, bmap, off);
4290 
4291 	res = isl_basic_map_alloc_space(isl_basic_map_get_space(bmap),
4292 			bmap->n_div, bmap->n_eq, bmap->n_ineq);
4293 	bmap = isl_basic_map_add_constraints_dim_map(res, bmap, dim_map);
4294 	space = isl_basic_map_take_space(bmap);
4295 	space = isl_space_move_dims(space, dst_type, dst_pos,
4296 					src_type, src_pos, n);
4297 	bmap = isl_basic_map_restore_space(bmap, space);
4298 	if (!bmap)
4299 		goto error;
4300 
4301 	ISL_F_CLR(bmap, ISL_BASIC_MAP_SORTED);
4302 	bmap = isl_basic_map_gauss(bmap, NULL);
4303 	bmap = isl_basic_map_finalize(bmap);
4304 
4305 	return bmap;
4306 error:
4307 	isl_basic_map_free(bmap);
4308 	return NULL;
4309 }
4310 
isl_basic_set_move_dims(__isl_take isl_basic_set * bset,enum isl_dim_type dst_type,unsigned dst_pos,enum isl_dim_type src_type,unsigned src_pos,unsigned n)4311 __isl_give isl_basic_set *isl_basic_set_move_dims(__isl_take isl_basic_set *bset,
4312 	enum isl_dim_type dst_type, unsigned dst_pos,
4313 	enum isl_dim_type src_type, unsigned src_pos, unsigned n)
4314 {
4315 	isl_basic_map *bmap = bset_to_bmap(bset);
4316 	bmap = isl_basic_map_move_dims(bmap, dst_type, dst_pos,
4317 					src_type, src_pos, n);
4318 	return bset_from_bmap(bmap);
4319 }
4320 
isl_set_move_dims(__isl_take isl_set * set,enum isl_dim_type dst_type,unsigned dst_pos,enum isl_dim_type src_type,unsigned src_pos,unsigned n)4321 __isl_give isl_set *isl_set_move_dims(__isl_take isl_set *set,
4322 	enum isl_dim_type dst_type, unsigned dst_pos,
4323 	enum isl_dim_type src_type, unsigned src_pos, unsigned n)
4324 {
4325 	if (!set)
4326 		return NULL;
4327 	isl_assert(set->ctx, dst_type != isl_dim_in, goto error);
4328 	return set_from_map(isl_map_move_dims(set_to_map(set),
4329 				    dst_type, dst_pos, src_type, src_pos, n));
4330 error:
4331 	isl_set_free(set);
4332 	return NULL;
4333 }
4334 
isl_map_move_dims(__isl_take isl_map * map,enum isl_dim_type dst_type,unsigned dst_pos,enum isl_dim_type src_type,unsigned src_pos,unsigned n)4335 __isl_give isl_map *isl_map_move_dims(__isl_take isl_map *map,
4336 	enum isl_dim_type dst_type, unsigned dst_pos,
4337 	enum isl_dim_type src_type, unsigned src_pos, unsigned n)
4338 {
4339 	int i;
4340 	isl_space *space;
4341 
4342 	if (n == 0) {
4343 		map = isl_map_reset(map, src_type);
4344 		map = isl_map_reset(map, dst_type);
4345 		return map;
4346 	}
4347 
4348 	if (isl_map_check_range(map, src_type, src_pos, n))
4349 		return isl_map_free(map);
4350 
4351 	if (dst_type == src_type && dst_pos == src_pos)
4352 		return map;
4353 
4354 	isl_assert(map->ctx, dst_type != src_type, goto error);
4355 
4356 	map = isl_map_cow(map);
4357 	if (!map)
4358 		return NULL;
4359 
4360 	for (i = 0; i < map->n; ++i) {
4361 		map->p[i] = isl_basic_map_move_dims(map->p[i],
4362 						dst_type, dst_pos,
4363 						src_type, src_pos, n);
4364 		if (!map->p[i])
4365 			goto error;
4366 	}
4367 
4368 	space = isl_map_take_space(map);
4369 	space = isl_space_move_dims(space, dst_type, dst_pos,
4370 					    src_type, src_pos, n);
4371 	map = isl_map_restore_space(map, space);
4372 
4373 	return map;
4374 error:
4375 	isl_map_free(map);
4376 	return NULL;
4377 }
4378 
4379 /* Move the specified dimensions to the last columns right before
4380  * the divs.  Don't change the dimension specification of bmap.
4381  * That's the responsibility of the caller.
4382  */
move_last(__isl_take isl_basic_map * bmap,enum isl_dim_type type,unsigned first,unsigned n)4383 static __isl_give isl_basic_map *move_last(__isl_take isl_basic_map *bmap,
4384 	enum isl_dim_type type, unsigned first, unsigned n)
4385 {
4386 	isl_space *space;
4387 	struct isl_dim_map *dim_map;
4388 	struct isl_basic_map *res;
4389 	enum isl_dim_type t;
4390 	isl_size total;
4391 	unsigned off;
4392 
4393 	if (!bmap)
4394 		return NULL;
4395 	if (isl_basic_map_offset(bmap, type) + first + n ==
4396 				isl_basic_map_offset(bmap, isl_dim_div))
4397 		return bmap;
4398 
4399 	total = isl_basic_map_dim(bmap, isl_dim_all);
4400 	if (total < 0)
4401 		return isl_basic_map_free(bmap);
4402 	dim_map = isl_dim_map_alloc(bmap->ctx, total);
4403 
4404 	off = 0;
4405 	space = isl_basic_map_peek_space(bmap);
4406 	for (t = isl_dim_param; t <= isl_dim_out; ++t) {
4407 		isl_size size = isl_space_dim(space, t);
4408 		if (size < 0)
4409 			dim_map = isl_dim_map_free(dim_map);
4410 		if (t == type) {
4411 			isl_dim_map_dim_range(dim_map, space, t,
4412 					    0, first, off);
4413 			off += first;
4414 			isl_dim_map_dim_range(dim_map, space, t,
4415 					    first, n, total - bmap->n_div - n);
4416 			isl_dim_map_dim_range(dim_map, space, t,
4417 					    first + n, size - (first + n), off);
4418 			off += size - (first + n);
4419 		} else {
4420 			isl_dim_map_dim(dim_map, space, t, off);
4421 			off += size;
4422 		}
4423 	}
4424 	isl_dim_map_div(dim_map, bmap, off + n);
4425 
4426 	res = isl_basic_map_alloc_space(isl_basic_map_get_space(bmap),
4427 			bmap->n_div, bmap->n_eq, bmap->n_ineq);
4428 	res = isl_basic_map_add_constraints_dim_map(res, bmap, dim_map);
4429 	return res;
4430 }
4431 
4432 /* Insert "n" rows in the divs of "bmap".
4433  *
4434  * The number of columns is not changed, which means that the last
4435  * dimensions of "bmap" are being reintepreted as the new divs.
4436  * The space of "bmap" is not adjusted, however, which means
4437  * that "bmap" is left in an inconsistent state.  Removing "n" dimensions
4438  * from the space of "bmap" is the responsibility of the caller.
4439  */
insert_div_rows(__isl_take isl_basic_map * bmap,int n)4440 static __isl_give isl_basic_map *insert_div_rows(__isl_take isl_basic_map *bmap,
4441 	int n)
4442 {
4443 	int i;
4444 	size_t row_size;
4445 	isl_int **new_div;
4446 	isl_int *old;
4447 
4448 	bmap = isl_basic_map_cow(bmap);
4449 	if (!bmap)
4450 		return NULL;
4451 
4452 	row_size = isl_basic_map_offset(bmap, isl_dim_div) + bmap->extra;
4453 	old = bmap->block2.data;
4454 	bmap->block2 = isl_blk_extend(bmap->ctx, bmap->block2,
4455 					(bmap->extra + n) * (1 + row_size));
4456 	if (!bmap->block2.data)
4457 		return isl_basic_map_free(bmap);
4458 	new_div = isl_alloc_array(bmap->ctx, isl_int *, bmap->extra + n);
4459 	if (!new_div)
4460 		return isl_basic_map_free(bmap);
4461 	for (i = 0; i < n; ++i) {
4462 		new_div[i] = bmap->block2.data +
4463 				(bmap->extra + i) * (1 + row_size);
4464 		isl_seq_clr(new_div[i], 1 + row_size);
4465 	}
4466 	for (i = 0; i < bmap->extra; ++i)
4467 		new_div[n + i] = bmap->block2.data + (bmap->div[i] - old);
4468 	free(bmap->div);
4469 	bmap->div = new_div;
4470 	bmap->n_div += n;
4471 	bmap->extra += n;
4472 
4473 	return bmap;
4474 }
4475 
4476 /* Drop constraints from "bmap" that only involve the variables
4477  * of "type" in the range [first, first + n] that are not related
4478  * to any of the variables outside that interval.
4479  * These constraints cannot influence the values for the variables
4480  * outside the interval, except in case they cause "bmap" to be empty.
4481  * Only drop the constraints if "bmap" is known to be non-empty.
4482  */
drop_irrelevant_constraints(__isl_take isl_basic_map * bmap,enum isl_dim_type type,unsigned first,unsigned n)4483 static __isl_give isl_basic_map *drop_irrelevant_constraints(
4484 	__isl_take isl_basic_map *bmap, enum isl_dim_type type,
4485 	unsigned first, unsigned n)
4486 {
4487 	int i;
4488 	int *groups;
4489 	isl_size dim, n_div;
4490 	isl_bool non_empty;
4491 
4492 	non_empty = isl_basic_map_plain_is_non_empty(bmap);
4493 	if (non_empty < 0)
4494 		return isl_basic_map_free(bmap);
4495 	if (!non_empty)
4496 		return bmap;
4497 
4498 	dim = isl_basic_map_dim(bmap, isl_dim_all);
4499 	n_div = isl_basic_map_dim(bmap, isl_dim_div);
4500 	if (dim < 0 || n_div < 0)
4501 		return isl_basic_map_free(bmap);
4502 	groups = isl_calloc_array(isl_basic_map_get_ctx(bmap), int, dim);
4503 	if (!groups)
4504 		return isl_basic_map_free(bmap);
4505 	first += isl_basic_map_offset(bmap, type) - 1;
4506 	for (i = 0; i < first; ++i)
4507 		groups[i] = -1;
4508 	for (i = first + n; i < dim - n_div; ++i)
4509 		groups[i] = -1;
4510 
4511 	bmap = isl_basic_map_drop_unrelated_constraints(bmap, groups);
4512 
4513 	return bmap;
4514 }
4515 
4516 /* Turn the n dimensions of type type, starting at first
4517  * into existentially quantified variables.
4518  *
4519  * If a subset of the projected out variables are unrelated
4520  * to any of the variables that remain, then the constraints
4521  * involving this subset are simply dropped first.
4522  */
isl_basic_map_project_out(__isl_take isl_basic_map * bmap,enum isl_dim_type type,unsigned first,unsigned n)4523 __isl_give isl_basic_map *isl_basic_map_project_out(
4524 		__isl_take isl_basic_map *bmap,
4525 		enum isl_dim_type type, unsigned first, unsigned n)
4526 {
4527 	isl_bool empty;
4528 	isl_space *space;
4529 
4530 	if (n == 0)
4531 		return basic_map_space_reset(bmap, type);
4532 	if (type == isl_dim_div)
4533 		isl_die(isl_basic_map_get_ctx(bmap), isl_error_invalid,
4534 			"cannot project out existentially quantified variables",
4535 			return isl_basic_map_free(bmap));
4536 
4537 	empty = isl_basic_map_plain_is_empty(bmap);
4538 	if (empty < 0)
4539 		return isl_basic_map_free(bmap);
4540 	if (empty)
4541 		bmap = isl_basic_map_set_to_empty(bmap);
4542 
4543 	bmap = drop_irrelevant_constraints(bmap, type, first, n);
4544 	if (!bmap)
4545 		return NULL;
4546 
4547 	if (ISL_F_ISSET(bmap, ISL_BASIC_MAP_RATIONAL))
4548 		return isl_basic_map_remove_dims(bmap, type, first, n);
4549 
4550 	if (isl_basic_map_check_range(bmap, type, first, n) < 0)
4551 		return isl_basic_map_free(bmap);
4552 
4553 	bmap = move_last(bmap, type, first, n);
4554 	bmap = isl_basic_map_cow(bmap);
4555 	bmap = insert_div_rows(bmap, n);
4556 
4557 	space = isl_basic_map_take_space(bmap);
4558 	space = isl_space_drop_dims(space, type, first, n);
4559 	bmap = isl_basic_map_restore_space(bmap, space);
4560 	bmap = isl_basic_map_simplify(bmap);
4561 	bmap = isl_basic_map_drop_redundant_divs(bmap);
4562 	return isl_basic_map_finalize(bmap);
4563 }
4564 
4565 /* Turn the n dimensions of type type, starting at first
4566  * into existentially quantified variables.
4567  */
isl_basic_set_project_out(__isl_take isl_basic_set * bset,enum isl_dim_type type,unsigned first,unsigned n)4568 __isl_give isl_basic_set *isl_basic_set_project_out(
4569 	__isl_take isl_basic_set *bset, enum isl_dim_type type,
4570 	unsigned first, unsigned n)
4571 {
4572 	return bset_from_bmap(isl_basic_map_project_out(bset_to_bmap(bset),
4573 							type, first, n));
4574 }
4575 
4576 /* Turn the n dimensions of type type, starting at first
4577  * into existentially quantified variables.
4578  */
isl_map_project_out(__isl_take isl_map * map,enum isl_dim_type type,unsigned first,unsigned n)4579 __isl_give isl_map *isl_map_project_out(__isl_take isl_map *map,
4580 		enum isl_dim_type type, unsigned first, unsigned n)
4581 {
4582 	int i;
4583 	isl_space *space;
4584 
4585 	if (n == 0)
4586 		return map_space_reset(map, type);
4587 
4588 	if (isl_map_check_range(map, type, first, n) < 0)
4589 		return isl_map_free(map);
4590 
4591 	map = isl_map_cow(map);
4592 	if (!map)
4593 		return NULL;
4594 
4595 	for (i = 0; i < map->n; ++i) {
4596 		map->p[i] = isl_basic_map_project_out(map->p[i], type, first, n);
4597 		if (!map->p[i])
4598 			goto error;
4599 	}
4600 
4601 	if (map->n > 1)
4602 		ISL_F_CLR(map, ISL_MAP_DISJOINT);
4603 	map = isl_map_unmark_normalized(map);
4604 
4605 	space = isl_map_take_space(map);
4606 	space = isl_space_drop_dims(space, type, first, n);
4607 	map = isl_map_restore_space(map, space);
4608 
4609 	return map;
4610 error:
4611 	isl_map_free(map);
4612 	return NULL;
4613 }
4614 
4615 #undef TYPE
4616 #define TYPE	isl_map
4617 #include "isl_project_out_all_params_templ.c"
4618 #include "isl_project_out_param_templ.c"
4619 
4620 /* Turn all the dimensions of type "type", except the "n" starting at "first"
4621  * into existentially quantified variables.
4622  */
isl_map_project_onto(__isl_take isl_map * map,enum isl_dim_type type,unsigned first,unsigned n)4623 __isl_give isl_map *isl_map_project_onto(__isl_take isl_map *map,
4624 	enum isl_dim_type type, unsigned first, unsigned n)
4625 {
4626 	isl_size dim;
4627 
4628 	dim = isl_map_dim(map, type);
4629 	if (isl_map_check_range(map, type, first, n) < 0 || dim < 0)
4630 		return isl_map_free(map);
4631 	map = isl_map_project_out(map, type, first + n, dim - (first + n));
4632 	map = isl_map_project_out(map, type, 0, first);
4633 	return map;
4634 }
4635 
4636 /* Turn the n dimensions of type type, starting at first
4637  * into existentially quantified variables.
4638  */
isl_set_project_out(__isl_take isl_set * set,enum isl_dim_type type,unsigned first,unsigned n)4639 __isl_give isl_set *isl_set_project_out(__isl_take isl_set *set,
4640 		enum isl_dim_type type, unsigned first, unsigned n)
4641 {
4642 	return set_from_map(isl_map_project_out(set_to_map(set),
4643 						type, first, n));
4644 }
4645 
4646 /* If "set" involves a parameter with identifier "id",
4647  * then turn it into an existentially quantified variable.
4648  */
isl_set_project_out_param_id(__isl_take isl_set * set,__isl_take isl_id * id)4649 __isl_give isl_set *isl_set_project_out_param_id(__isl_take isl_set *set,
4650 	__isl_take isl_id *id)
4651 {
4652 	return set_from_map(isl_map_project_out_param_id(set_to_map(set), id));
4653 }
4654 
4655 /* If "set" involves any of the parameters with identifiers in "list",
4656  * then turn them into existentially quantified variables.
4657  */
isl_set_project_out_param_id_list(__isl_take isl_set * set,__isl_take isl_id_list * list)4658 __isl_give isl_set *isl_set_project_out_param_id_list(__isl_take isl_set *set,
4659 	__isl_take isl_id_list *list)
4660 {
4661 	isl_map *map;
4662 
4663 	map = set_to_map(set);
4664 	map = isl_map_project_out_param_id_list(map, list);
4665 	return set_from_map(map);
4666 }
4667 
4668 /* Project out all parameters from "set" by existentially quantifying
4669  * over them.
4670  */
isl_set_project_out_all_params(__isl_take isl_set * set)4671 __isl_give isl_set *isl_set_project_out_all_params(__isl_take isl_set *set)
4672 {
4673 	return set_from_map(isl_map_project_out_all_params(set_to_map(set)));
4674 }
4675 
4676 /* Return a map that projects the elements in "set" onto their
4677  * "n" set dimensions starting at "first".
4678  * "type" should be equal to isl_dim_set.
4679  */
isl_set_project_onto_map(__isl_take isl_set * set,enum isl_dim_type type,unsigned first,unsigned n)4680 __isl_give isl_map *isl_set_project_onto_map(__isl_take isl_set *set,
4681 	enum isl_dim_type type, unsigned first, unsigned n)
4682 {
4683 	int i;
4684 	isl_map *map;
4685 
4686 	if (type != isl_dim_set)
4687 		isl_die(isl_set_get_ctx(set), isl_error_invalid,
4688 			"only set dimensions can be projected out", goto error);
4689 	if (isl_set_check_range(set, type, first, n) < 0)
4690 		return isl_set_free(set);
4691 
4692 	map = isl_map_from_domain(set);
4693 	map = isl_map_add_dims(map, isl_dim_out, n);
4694 	for (i = 0; i < n; ++i)
4695 		map = isl_map_equate(map, isl_dim_in, first + i,
4696 					isl_dim_out, i);
4697 	return map;
4698 error:
4699 	isl_set_free(set);
4700 	return NULL;
4701 }
4702 
add_divs(__isl_take isl_basic_map * bmap,unsigned n)4703 static __isl_give isl_basic_map *add_divs(__isl_take isl_basic_map *bmap,
4704 	unsigned n)
4705 {
4706 	int i, j;
4707 	isl_size total;
4708 
4709 	total = isl_basic_map_dim(bmap, isl_dim_all);
4710 	if (total < 0)
4711 		return isl_basic_map_free(bmap);
4712 	for (i = 0; i < n; ++i) {
4713 		j = isl_basic_map_alloc_div(bmap);
4714 		if (j < 0)
4715 			goto error;
4716 		isl_seq_clr(bmap->div[j], 1 + 1 + total);
4717 	}
4718 	return bmap;
4719 error:
4720 	isl_basic_map_free(bmap);
4721 	return NULL;
4722 }
4723 
4724 /* Does "bmap2" apply to the range of "bmap1" (ignoring parameters)?
4725  */
isl_basic_map_applies_range(__isl_keep isl_basic_map * bmap1,__isl_keep isl_basic_map * bmap2)4726 isl_bool isl_basic_map_applies_range(__isl_keep isl_basic_map *bmap1,
4727 	__isl_keep isl_basic_map *bmap2)
4728 {
4729 	isl_space *space1, *space2;
4730 
4731 	space1 = isl_basic_map_peek_space(bmap1);
4732 	space2 = isl_basic_map_peek_space(bmap2);
4733 	return isl_space_tuple_is_equal(space1, isl_dim_out,
4734 					space2, isl_dim_in);
4735 }
4736 
4737 /* Check that "bmap2" applies to the range of "bmap1" (ignoring parameters).
4738  */
isl_basic_map_check_applies_range(__isl_keep isl_basic_map * bmap1,__isl_keep isl_basic_map * bmap2)4739 static isl_stat isl_basic_map_check_applies_range(
4740 	__isl_keep isl_basic_map *bmap1, __isl_keep isl_basic_map *bmap2)
4741 {
4742 	isl_bool equal;
4743 
4744 	equal = isl_basic_map_applies_range(bmap1, bmap2);
4745 	if (equal < 0)
4746 		return isl_stat_error;
4747 	if (!equal)
4748 		isl_die(isl_basic_map_get_ctx(bmap1), isl_error_invalid,
4749 			"spaces don't match", return isl_stat_error);
4750 	return isl_stat_ok;
4751 }
4752 
isl_basic_map_apply_range(__isl_take isl_basic_map * bmap1,__isl_take isl_basic_map * bmap2)4753 __isl_give isl_basic_map *isl_basic_map_apply_range(
4754 	__isl_take isl_basic_map *bmap1, __isl_take isl_basic_map *bmap2)
4755 {
4756 	isl_space *space_result = NULL;
4757 	struct isl_basic_map *bmap;
4758 	isl_size n_in, n_out, n, nparam;
4759 	unsigned total, pos;
4760 	struct isl_dim_map *dim_map1, *dim_map2;
4761 
4762 	if (isl_basic_map_check_equal_params(bmap1, bmap2) < 0)
4763 		goto error;
4764 	if (isl_basic_map_check_applies_range(bmap1, bmap2) < 0)
4765 		goto error;
4766 
4767 	n_in = isl_basic_map_dim(bmap1, isl_dim_in);
4768 	n_out = isl_basic_map_dim(bmap2, isl_dim_out);
4769 	n = isl_basic_map_dim(bmap1, isl_dim_out);
4770 	nparam = isl_basic_map_dim(bmap1, isl_dim_param);
4771 	if (n_in < 0 || n_out < 0 || n < 0 || nparam < 0)
4772 		goto error;
4773 
4774 	space_result = isl_space_join(isl_basic_map_get_space(bmap1),
4775 				  isl_basic_map_get_space(bmap2));
4776 
4777 	total = nparam + n_in + n_out + bmap1->n_div + bmap2->n_div + n;
4778 	dim_map1 = isl_dim_map_alloc(bmap1->ctx, total);
4779 	dim_map2 = isl_dim_map_alloc(bmap1->ctx, total);
4780 	isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_param, pos = 0);
4781 	isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_param, pos = 0);
4782 	isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_in, pos += nparam);
4783 	isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_out, pos += n_in);
4784 	isl_dim_map_div(dim_map1, bmap1, pos += n_out);
4785 	isl_dim_map_div(dim_map2, bmap2, pos += bmap1->n_div);
4786 	isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_out, pos += bmap2->n_div);
4787 	isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_in, pos);
4788 
4789 	bmap = isl_basic_map_alloc_space(space_result,
4790 			bmap1->n_div + bmap2->n_div + n,
4791 			bmap1->n_eq + bmap2->n_eq,
4792 			bmap1->n_ineq + bmap2->n_ineq);
4793 	bmap = isl_basic_map_add_constraints_dim_map(bmap, bmap1, dim_map1);
4794 	bmap = isl_basic_map_add_constraints_dim_map(bmap, bmap2, dim_map2);
4795 	bmap = add_divs(bmap, n);
4796 	bmap = isl_basic_map_simplify(bmap);
4797 	bmap = isl_basic_map_drop_redundant_divs(bmap);
4798 	return isl_basic_map_finalize(bmap);
4799 error:
4800 	isl_basic_map_free(bmap1);
4801 	isl_basic_map_free(bmap2);
4802 	return NULL;
4803 }
4804 
isl_basic_set_apply(__isl_take isl_basic_set * bset,__isl_take isl_basic_map * bmap)4805 __isl_give isl_basic_set *isl_basic_set_apply(__isl_take isl_basic_set *bset,
4806 	__isl_take isl_basic_map *bmap)
4807 {
4808 	if (isl_basic_map_check_compatible_domain(bmap, bset) < 0)
4809 		goto error;
4810 
4811 	return bset_from_bmap(isl_basic_map_apply_range(bset_to_bmap(bset),
4812 							bmap));
4813 error:
4814 	isl_basic_set_free(bset);
4815 	isl_basic_map_free(bmap);
4816 	return NULL;
4817 }
4818 
isl_basic_map_apply_domain(__isl_take isl_basic_map * bmap1,__isl_take isl_basic_map * bmap2)4819 __isl_give isl_basic_map *isl_basic_map_apply_domain(
4820 	__isl_take isl_basic_map *bmap1, __isl_take isl_basic_map *bmap2)
4821 {
4822 	if (isl_basic_map_check_equal_params(bmap1, bmap2) < 0)
4823 		goto error;
4824 	if (!isl_space_tuple_is_equal(bmap1->dim, isl_dim_in,
4825 					bmap2->dim, isl_dim_in))
4826 		isl_die(isl_basic_map_get_ctx(bmap1), isl_error_invalid,
4827 			"spaces don't match", goto error);
4828 
4829 	bmap1 = isl_basic_map_reverse(bmap1);
4830 	bmap1 = isl_basic_map_apply_range(bmap1, bmap2);
4831 	return isl_basic_map_reverse(bmap1);
4832 error:
4833 	isl_basic_map_free(bmap1);
4834 	isl_basic_map_free(bmap2);
4835 	return NULL;
4836 }
4837 
4838 /* Given two basic maps A -> f(A) and B -> g(B), construct a basic map
4839  * A \cap B -> f(A) + f(B)
4840  */
isl_basic_map_sum(__isl_take isl_basic_map * bmap1,__isl_take isl_basic_map * bmap2)4841 __isl_give isl_basic_map *isl_basic_map_sum(__isl_take isl_basic_map *bmap1,
4842 	__isl_take isl_basic_map *bmap2)
4843 {
4844 	isl_size n_in, n_out, nparam;
4845 	unsigned total, pos;
4846 	struct isl_basic_map *bmap = NULL;
4847 	struct isl_dim_map *dim_map1, *dim_map2;
4848 	int i;
4849 
4850 	if (isl_basic_map_check_equal_space(bmap1, bmap2) < 0)
4851 		goto error;
4852 
4853 	nparam = isl_basic_map_dim(bmap1, isl_dim_param);
4854 	n_in = isl_basic_map_dim(bmap1, isl_dim_in);
4855 	n_out = isl_basic_map_dim(bmap1, isl_dim_out);
4856 	if (nparam < 0 || n_in < 0 || n_out < 0)
4857 		goto error;
4858 
4859 	total = nparam + n_in + n_out + bmap1->n_div + bmap2->n_div + 2 * n_out;
4860 	dim_map1 = isl_dim_map_alloc(bmap1->ctx, total);
4861 	dim_map2 = isl_dim_map_alloc(bmap2->ctx, total);
4862 	isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_param, pos = 0);
4863 	isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_param, pos);
4864 	isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_in, pos += nparam);
4865 	isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_in, pos);
4866 	isl_dim_map_div(dim_map1, bmap1, pos += n_in + n_out);
4867 	isl_dim_map_div(dim_map2, bmap2, pos += bmap1->n_div);
4868 	isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_out, pos += bmap2->n_div);
4869 	isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_out, pos += n_out);
4870 
4871 	bmap = isl_basic_map_alloc_space(isl_space_copy(bmap1->dim),
4872 			bmap1->n_div + bmap2->n_div + 2 * n_out,
4873 			bmap1->n_eq + bmap2->n_eq + n_out,
4874 			bmap1->n_ineq + bmap2->n_ineq);
4875 	for (i = 0; i < n_out; ++i) {
4876 		int j = isl_basic_map_alloc_equality(bmap);
4877 		if (j < 0)
4878 			goto error;
4879 		isl_seq_clr(bmap->eq[j], 1+total);
4880 		isl_int_set_si(bmap->eq[j][1+nparam+n_in+i], -1);
4881 		isl_int_set_si(bmap->eq[j][1+pos+i], 1);
4882 		isl_int_set_si(bmap->eq[j][1+pos-n_out+i], 1);
4883 	}
4884 	bmap = isl_basic_map_add_constraints_dim_map(bmap, bmap1, dim_map1);
4885 	bmap = isl_basic_map_add_constraints_dim_map(bmap, bmap2, dim_map2);
4886 	bmap = add_divs(bmap, 2 * n_out);
4887 
4888 	bmap = isl_basic_map_simplify(bmap);
4889 	return isl_basic_map_finalize(bmap);
4890 error:
4891 	isl_basic_map_free(bmap);
4892 	isl_basic_map_free(bmap1);
4893 	isl_basic_map_free(bmap2);
4894 	return NULL;
4895 }
4896 
4897 /* Given two maps A -> f(A) and B -> g(B), construct a map
4898  * A \cap B -> f(A) + f(B)
4899  */
isl_map_sum(__isl_take isl_map * map1,__isl_take isl_map * map2)4900 __isl_give isl_map *isl_map_sum(__isl_take isl_map *map1,
4901 	__isl_take isl_map *map2)
4902 {
4903 	struct isl_map *result;
4904 	int i, j;
4905 
4906 	if (isl_map_check_equal_space(map1, map2) < 0)
4907 		goto error;
4908 
4909 	result = isl_map_alloc_space(isl_space_copy(map1->dim),
4910 				map1->n * map2->n, 0);
4911 	if (!result)
4912 		goto error;
4913 	for (i = 0; i < map1->n; ++i)
4914 		for (j = 0; j < map2->n; ++j) {
4915 			struct isl_basic_map *part;
4916 			part = isl_basic_map_sum(
4917 				    isl_basic_map_copy(map1->p[i]),
4918 				    isl_basic_map_copy(map2->p[j]));
4919 			if (isl_basic_map_is_empty(part))
4920 				isl_basic_map_free(part);
4921 			else
4922 				result = isl_map_add_basic_map(result, part);
4923 			if (!result)
4924 				goto error;
4925 		}
4926 	isl_map_free(map1);
4927 	isl_map_free(map2);
4928 	return result;
4929 error:
4930 	isl_map_free(map1);
4931 	isl_map_free(map2);
4932 	return NULL;
4933 }
4934 
isl_set_sum(__isl_take isl_set * set1,__isl_take isl_set * set2)4935 __isl_give isl_set *isl_set_sum(__isl_take isl_set *set1,
4936 	__isl_take isl_set *set2)
4937 {
4938 	return set_from_map(isl_map_sum(set_to_map(set1), set_to_map(set2)));
4939 }
4940 
4941 /* Given a basic map A -> f(A), construct A -> -f(A).
4942  */
isl_basic_map_neg(__isl_take isl_basic_map * bmap)4943 __isl_give isl_basic_map *isl_basic_map_neg(__isl_take isl_basic_map *bmap)
4944 {
4945 	int i, j;
4946 	unsigned off;
4947 	isl_size n;
4948 
4949 	bmap = isl_basic_map_cow(bmap);
4950 	n = isl_basic_map_dim(bmap, isl_dim_out);
4951 	if (n < 0)
4952 		return isl_basic_map_free(bmap);
4953 
4954 	off = isl_basic_map_offset(bmap, isl_dim_out);
4955 	for (i = 0; i < bmap->n_eq; ++i)
4956 		for (j = 0; j < n; ++j)
4957 			isl_int_neg(bmap->eq[i][off+j], bmap->eq[i][off+j]);
4958 	for (i = 0; i < bmap->n_ineq; ++i)
4959 		for (j = 0; j < n; ++j)
4960 			isl_int_neg(bmap->ineq[i][off+j], bmap->ineq[i][off+j]);
4961 	for (i = 0; i < bmap->n_div; ++i)
4962 		for (j = 0; j < n; ++j)
4963 			isl_int_neg(bmap->div[i][1+off+j], bmap->div[i][1+off+j]);
4964 	bmap = isl_basic_map_gauss(bmap, NULL);
4965 	return isl_basic_map_finalize(bmap);
4966 }
4967 
isl_basic_set_neg(__isl_take isl_basic_set * bset)4968 __isl_give isl_basic_set *isl_basic_set_neg(__isl_take isl_basic_set *bset)
4969 {
4970 	return isl_basic_map_neg(bset);
4971 }
4972 
4973 /* Given a map A -> f(A), construct A -> -f(A).
4974  */
isl_map_neg(__isl_take isl_map * map)4975 __isl_give isl_map *isl_map_neg(__isl_take isl_map *map)
4976 {
4977 	int i;
4978 
4979 	map = isl_map_cow(map);
4980 	if (!map)
4981 		return NULL;
4982 
4983 	for (i = 0; i < map->n; ++i) {
4984 		map->p[i] = isl_basic_map_neg(map->p[i]);
4985 		if (!map->p[i])
4986 			goto error;
4987 	}
4988 
4989 	return map;
4990 error:
4991 	isl_map_free(map);
4992 	return NULL;
4993 }
4994 
isl_set_neg(__isl_take isl_set * set)4995 __isl_give isl_set *isl_set_neg(__isl_take isl_set *set)
4996 {
4997 	return set_from_map(isl_map_neg(set_to_map(set)));
4998 }
4999 
5000 /* Given a basic map A -> f(A) and an integer d, construct a basic map
5001  * A -> floor(f(A)/d).
5002  */
isl_basic_map_floordiv(__isl_take isl_basic_map * bmap,isl_int d)5003 __isl_give isl_basic_map *isl_basic_map_floordiv(__isl_take isl_basic_map *bmap,
5004 		isl_int d)
5005 {
5006 	isl_size n_in, n_out, nparam;
5007 	unsigned total, pos;
5008 	struct isl_basic_map *result = NULL;
5009 	struct isl_dim_map *dim_map;
5010 	int i;
5011 
5012 	nparam = isl_basic_map_dim(bmap, isl_dim_param);
5013 	n_in = isl_basic_map_dim(bmap, isl_dim_in);
5014 	n_out = isl_basic_map_dim(bmap, isl_dim_out);
5015 	if (nparam < 0 || n_in < 0 || n_out < 0)
5016 		return isl_basic_map_free(bmap);
5017 
5018 	total = nparam + n_in + n_out + bmap->n_div + n_out;
5019 	dim_map = isl_dim_map_alloc(bmap->ctx, total);
5020 	isl_dim_map_dim(dim_map, bmap->dim, isl_dim_param, pos = 0);
5021 	isl_dim_map_dim(dim_map, bmap->dim, isl_dim_in, pos += nparam);
5022 	isl_dim_map_div(dim_map, bmap, pos += n_in + n_out);
5023 	isl_dim_map_dim(dim_map, bmap->dim, isl_dim_out, pos += bmap->n_div);
5024 
5025 	result = isl_basic_map_alloc_space(isl_space_copy(bmap->dim),
5026 			bmap->n_div + n_out,
5027 			bmap->n_eq, bmap->n_ineq + 2 * n_out);
5028 	result = isl_basic_map_add_constraints_dim_map(result, bmap, dim_map);
5029 	result = add_divs(result, n_out);
5030 	for (i = 0; i < n_out; ++i) {
5031 		int j;
5032 		j = isl_basic_map_alloc_inequality(result);
5033 		if (j < 0)
5034 			goto error;
5035 		isl_seq_clr(result->ineq[j], 1+total);
5036 		isl_int_neg(result->ineq[j][1+nparam+n_in+i], d);
5037 		isl_int_set_si(result->ineq[j][1+pos+i], 1);
5038 		j = isl_basic_map_alloc_inequality(result);
5039 		if (j < 0)
5040 			goto error;
5041 		isl_seq_clr(result->ineq[j], 1+total);
5042 		isl_int_set(result->ineq[j][1+nparam+n_in+i], d);
5043 		isl_int_set_si(result->ineq[j][1+pos+i], -1);
5044 		isl_int_sub_ui(result->ineq[j][0], d, 1);
5045 	}
5046 
5047 	result = isl_basic_map_simplify(result);
5048 	return isl_basic_map_finalize(result);
5049 error:
5050 	isl_basic_map_free(result);
5051 	return NULL;
5052 }
5053 
5054 /* Given a map A -> f(A) and an integer d, construct a map
5055  * A -> floor(f(A)/d).
5056  */
isl_map_floordiv(__isl_take isl_map * map,isl_int d)5057 __isl_give isl_map *isl_map_floordiv(__isl_take isl_map *map, isl_int d)
5058 {
5059 	int i;
5060 
5061 	map = isl_map_cow(map);
5062 	if (!map)
5063 		return NULL;
5064 
5065 	ISL_F_CLR(map, ISL_MAP_DISJOINT);
5066 	for (i = 0; i < map->n; ++i) {
5067 		map->p[i] = isl_basic_map_floordiv(map->p[i], d);
5068 		if (!map->p[i])
5069 			goto error;
5070 	}
5071 	map = isl_map_unmark_normalized(map);
5072 
5073 	return map;
5074 error:
5075 	isl_map_free(map);
5076 	return NULL;
5077 }
5078 
5079 /* Given a map A -> f(A) and an integer d, construct a map
5080  * A -> floor(f(A)/d).
5081  */
isl_map_floordiv_val(__isl_take isl_map * map,__isl_take isl_val * d)5082 __isl_give isl_map *isl_map_floordiv_val(__isl_take isl_map *map,
5083 	__isl_take isl_val *d)
5084 {
5085 	if (!map || !d)
5086 		goto error;
5087 	if (!isl_val_is_int(d))
5088 		isl_die(isl_val_get_ctx(d), isl_error_invalid,
5089 			"expecting integer denominator", goto error);
5090 	map = isl_map_floordiv(map, d->n);
5091 	isl_val_free(d);
5092 	return map;
5093 error:
5094 	isl_map_free(map);
5095 	isl_val_free(d);
5096 	return NULL;
5097 }
5098 
var_equal(__isl_take isl_basic_map * bmap,unsigned pos)5099 static __isl_give isl_basic_map *var_equal(__isl_take isl_basic_map *bmap,
5100 	unsigned pos)
5101 {
5102 	int i;
5103 	isl_size nparam;
5104 	isl_size n_in;
5105 	isl_size total;
5106 
5107 	total = isl_basic_map_dim(bmap, isl_dim_all);
5108 	nparam = isl_basic_map_dim(bmap, isl_dim_param);
5109 	n_in = isl_basic_map_dim(bmap, isl_dim_in);
5110 	if (total < 0 || nparam < 0 || n_in < 0)
5111 		return isl_basic_map_free(bmap);
5112 	i = isl_basic_map_alloc_equality(bmap);
5113 	if (i < 0)
5114 		goto error;
5115 	isl_seq_clr(bmap->eq[i], 1 + total);
5116 	isl_int_set_si(bmap->eq[i][1+nparam+pos], -1);
5117 	isl_int_set_si(bmap->eq[i][1+nparam+n_in+pos], 1);
5118 	return isl_basic_map_finalize(bmap);
5119 error:
5120 	isl_basic_map_free(bmap);
5121 	return NULL;
5122 }
5123 
5124 /* Add a constraint to "bmap" expressing i_pos < o_pos
5125  */
var_less(__isl_take isl_basic_map * bmap,unsigned pos)5126 static __isl_give isl_basic_map *var_less(__isl_take isl_basic_map *bmap,
5127 	unsigned pos)
5128 {
5129 	int i;
5130 	isl_size nparam;
5131 	isl_size n_in;
5132 	isl_size total;
5133 
5134 	total = isl_basic_map_dim(bmap, isl_dim_all);
5135 	nparam = isl_basic_map_dim(bmap, isl_dim_param);
5136 	n_in = isl_basic_map_dim(bmap, isl_dim_in);
5137 	if (total < 0 || nparam < 0 || n_in < 0)
5138 		return isl_basic_map_free(bmap);
5139 	i = isl_basic_map_alloc_inequality(bmap);
5140 	if (i < 0)
5141 		goto error;
5142 	isl_seq_clr(bmap->ineq[i], 1 + total);
5143 	isl_int_set_si(bmap->ineq[i][0], -1);
5144 	isl_int_set_si(bmap->ineq[i][1+nparam+pos], -1);
5145 	isl_int_set_si(bmap->ineq[i][1+nparam+n_in+pos], 1);
5146 	return isl_basic_map_finalize(bmap);
5147 error:
5148 	isl_basic_map_free(bmap);
5149 	return NULL;
5150 }
5151 
5152 /* Add a constraint to "bmap" expressing i_pos <= o_pos
5153  */
var_less_or_equal(__isl_take isl_basic_map * bmap,unsigned pos)5154 static __isl_give isl_basic_map *var_less_or_equal(
5155 	__isl_take isl_basic_map *bmap, unsigned pos)
5156 {
5157 	int i;
5158 	isl_size nparam;
5159 	isl_size n_in;
5160 	isl_size total;
5161 
5162 	total = isl_basic_map_dim(bmap, isl_dim_all);
5163 	nparam = isl_basic_map_dim(bmap, isl_dim_param);
5164 	n_in = isl_basic_map_dim(bmap, isl_dim_in);
5165 	if (total < 0 || nparam < 0 || n_in < 0)
5166 		return isl_basic_map_free(bmap);
5167 	i = isl_basic_map_alloc_inequality(bmap);
5168 	if (i < 0)
5169 		goto error;
5170 	isl_seq_clr(bmap->ineq[i], 1 + total);
5171 	isl_int_set_si(bmap->ineq[i][1+nparam+pos], -1);
5172 	isl_int_set_si(bmap->ineq[i][1+nparam+n_in+pos], 1);
5173 	return isl_basic_map_finalize(bmap);
5174 error:
5175 	isl_basic_map_free(bmap);
5176 	return NULL;
5177 }
5178 
5179 /* Add a constraint to "bmap" expressing i_pos > o_pos
5180  */
var_more(__isl_take isl_basic_map * bmap,unsigned pos)5181 static __isl_give isl_basic_map *var_more(__isl_take isl_basic_map *bmap,
5182 	unsigned pos)
5183 {
5184 	int i;
5185 	isl_size nparam;
5186 	isl_size n_in;
5187 	isl_size total;
5188 
5189 	total = isl_basic_map_dim(bmap, isl_dim_all);
5190 	nparam = isl_basic_map_dim(bmap, isl_dim_param);
5191 	n_in = isl_basic_map_dim(bmap, isl_dim_in);
5192 	if (total < 0 || nparam < 0 || n_in < 0)
5193 		return isl_basic_map_free(bmap);
5194 	i = isl_basic_map_alloc_inequality(bmap);
5195 	if (i < 0)
5196 		goto error;
5197 	isl_seq_clr(bmap->ineq[i], 1 + total);
5198 	isl_int_set_si(bmap->ineq[i][0], -1);
5199 	isl_int_set_si(bmap->ineq[i][1+nparam+pos], 1);
5200 	isl_int_set_si(bmap->ineq[i][1+nparam+n_in+pos], -1);
5201 	return isl_basic_map_finalize(bmap);
5202 error:
5203 	isl_basic_map_free(bmap);
5204 	return NULL;
5205 }
5206 
5207 /* Add a constraint to "bmap" expressing i_pos >= o_pos
5208  */
var_more_or_equal(__isl_take isl_basic_map * bmap,unsigned pos)5209 static __isl_give isl_basic_map *var_more_or_equal(
5210 	__isl_take isl_basic_map *bmap, unsigned pos)
5211 {
5212 	int i;
5213 	isl_size nparam;
5214 	isl_size n_in;
5215 	isl_size total;
5216 
5217 	total = isl_basic_map_dim(bmap, isl_dim_all);
5218 	nparam = isl_basic_map_dim(bmap, isl_dim_param);
5219 	n_in = isl_basic_map_dim(bmap, isl_dim_in);
5220 	if (total < 0 || nparam < 0 || n_in < 0)
5221 		return isl_basic_map_free(bmap);
5222 	i = isl_basic_map_alloc_inequality(bmap);
5223 	if (i < 0)
5224 		goto error;
5225 	isl_seq_clr(bmap->ineq[i], 1 + total);
5226 	isl_int_set_si(bmap->ineq[i][1+nparam+pos], 1);
5227 	isl_int_set_si(bmap->ineq[i][1+nparam+n_in+pos], -1);
5228 	return isl_basic_map_finalize(bmap);
5229 error:
5230 	isl_basic_map_free(bmap);
5231 	return NULL;
5232 }
5233 
isl_basic_map_equal(__isl_take isl_space * space,unsigned n_equal)5234 __isl_give isl_basic_map *isl_basic_map_equal(
5235 	__isl_take isl_space *space, unsigned n_equal)
5236 {
5237 	int i;
5238 	struct isl_basic_map *bmap;
5239 	bmap = isl_basic_map_alloc_space(space, 0, n_equal, 0);
5240 	if (!bmap)
5241 		return NULL;
5242 	for (i = 0; i < n_equal && bmap; ++i)
5243 		bmap = var_equal(bmap, i);
5244 	return isl_basic_map_finalize(bmap);
5245 }
5246 
5247 /* Return a relation on of dimension "space" expressing i_[0..pos] << o_[0..pos]
5248  */
isl_basic_map_less_at(__isl_take isl_space * space,unsigned pos)5249 __isl_give isl_basic_map *isl_basic_map_less_at(__isl_take isl_space *space,
5250 	unsigned pos)
5251 {
5252 	int i;
5253 	struct isl_basic_map *bmap;
5254 	bmap = isl_basic_map_alloc_space(space, 0, pos, 1);
5255 	if (!bmap)
5256 		return NULL;
5257 	for (i = 0; i < pos && bmap; ++i)
5258 		bmap = var_equal(bmap, i);
5259 	if (bmap)
5260 		bmap = var_less(bmap, pos);
5261 	return isl_basic_map_finalize(bmap);
5262 }
5263 
5264 /* Return a relation on "space" expressing i_[0..pos] <<= o_[0..pos]
5265  */
isl_basic_map_less_or_equal_at(__isl_take isl_space * space,unsigned pos)5266 __isl_give isl_basic_map *isl_basic_map_less_or_equal_at(
5267 	__isl_take isl_space *space, unsigned pos)
5268 {
5269 	int i;
5270 	isl_basic_map *bmap;
5271 
5272 	bmap = isl_basic_map_alloc_space(space, 0, pos, 1);
5273 	for (i = 0; i < pos; ++i)
5274 		bmap = var_equal(bmap, i);
5275 	bmap = var_less_or_equal(bmap, pos);
5276 	return isl_basic_map_finalize(bmap);
5277 }
5278 
5279 /* Return a relation on "space" expressing i_pos > o_pos
5280  */
isl_basic_map_more_at(__isl_take isl_space * space,unsigned pos)5281 __isl_give isl_basic_map *isl_basic_map_more_at(__isl_take isl_space *space,
5282 	unsigned pos)
5283 {
5284 	int i;
5285 	struct isl_basic_map *bmap;
5286 	bmap = isl_basic_map_alloc_space(space, 0, pos, 1);
5287 	if (!bmap)
5288 		return NULL;
5289 	for (i = 0; i < pos && bmap; ++i)
5290 		bmap = var_equal(bmap, i);
5291 	if (bmap)
5292 		bmap = var_more(bmap, pos);
5293 	return isl_basic_map_finalize(bmap);
5294 }
5295 
5296 /* Return a relation on "space" expressing i_[0..pos] >>= o_[0..pos]
5297  */
isl_basic_map_more_or_equal_at(__isl_take isl_space * space,unsigned pos)5298 __isl_give isl_basic_map *isl_basic_map_more_or_equal_at(
5299 	__isl_take isl_space *space, unsigned pos)
5300 {
5301 	int i;
5302 	isl_basic_map *bmap;
5303 
5304 	bmap = isl_basic_map_alloc_space(space, 0, pos, 1);
5305 	for (i = 0; i < pos; ++i)
5306 		bmap = var_equal(bmap, i);
5307 	bmap = var_more_or_equal(bmap, pos);
5308 	return isl_basic_map_finalize(bmap);
5309 }
5310 
map_lex_lte_first(__isl_take isl_space * space,unsigned n,int equal)5311 static __isl_give isl_map *map_lex_lte_first(__isl_take isl_space *space,
5312 	unsigned n, int equal)
5313 {
5314 	struct isl_map *map;
5315 	int i;
5316 
5317 	if (n == 0 && equal)
5318 		return isl_map_universe(space);
5319 
5320 	map = isl_map_alloc_space(isl_space_copy(space), n, ISL_MAP_DISJOINT);
5321 
5322 	for (i = 0; i + 1 < n; ++i)
5323 		map = isl_map_add_basic_map(map,
5324 				  isl_basic_map_less_at(isl_space_copy(space), i));
5325 	if (n > 0) {
5326 		if (equal)
5327 			map = isl_map_add_basic_map(map,
5328 			      isl_basic_map_less_or_equal_at(space, n - 1));
5329 		else
5330 			map = isl_map_add_basic_map(map,
5331 			      isl_basic_map_less_at(space, n - 1));
5332 	} else
5333 		isl_space_free(space);
5334 
5335 	return map;
5336 }
5337 
map_lex_lte(__isl_take isl_space * space,int equal)5338 static __isl_give isl_map *map_lex_lte(__isl_take isl_space *space, int equal)
5339 {
5340 	if (!space)
5341 		return NULL;
5342 	return map_lex_lte_first(space, space->n_out, equal);
5343 }
5344 
isl_map_lex_lt_first(__isl_take isl_space * space,unsigned n)5345 __isl_give isl_map *isl_map_lex_lt_first(__isl_take isl_space *space,
5346 	unsigned n)
5347 {
5348 	return map_lex_lte_first(space, n, 0);
5349 }
5350 
isl_map_lex_le_first(__isl_take isl_space * space,unsigned n)5351 __isl_give isl_map *isl_map_lex_le_first(__isl_take isl_space *space,
5352 	unsigned n)
5353 {
5354 	return map_lex_lte_first(space, n, 1);
5355 }
5356 
isl_map_lex_lt(__isl_take isl_space * set_space)5357 __isl_give isl_map *isl_map_lex_lt(__isl_take isl_space *set_space)
5358 {
5359 	return map_lex_lte(isl_space_map_from_set(set_space), 0);
5360 }
5361 
isl_map_lex_le(__isl_take isl_space * set_space)5362 __isl_give isl_map *isl_map_lex_le(__isl_take isl_space *set_space)
5363 {
5364 	return map_lex_lte(isl_space_map_from_set(set_space), 1);
5365 }
5366 
map_lex_gte_first(__isl_take isl_space * space,unsigned n,int equal)5367 static __isl_give isl_map *map_lex_gte_first(__isl_take isl_space *space,
5368 	unsigned n, int equal)
5369 {
5370 	struct isl_map *map;
5371 	int i;
5372 
5373 	if (n == 0 && equal)
5374 		return isl_map_universe(space);
5375 
5376 	map = isl_map_alloc_space(isl_space_copy(space), n, ISL_MAP_DISJOINT);
5377 
5378 	for (i = 0; i + 1 < n; ++i)
5379 		map = isl_map_add_basic_map(map,
5380 				  isl_basic_map_more_at(isl_space_copy(space), i));
5381 	if (n > 0) {
5382 		if (equal)
5383 			map = isl_map_add_basic_map(map,
5384 			      isl_basic_map_more_or_equal_at(space, n - 1));
5385 		else
5386 			map = isl_map_add_basic_map(map,
5387 			      isl_basic_map_more_at(space, n - 1));
5388 	} else
5389 		isl_space_free(space);
5390 
5391 	return map;
5392 }
5393 
map_lex_gte(__isl_take isl_space * space,int equal)5394 static __isl_give isl_map *map_lex_gte(__isl_take isl_space *space, int equal)
5395 {
5396 	if (!space)
5397 		return NULL;
5398 	return map_lex_gte_first(space, space->n_out, equal);
5399 }
5400 
isl_map_lex_gt_first(__isl_take isl_space * space,unsigned n)5401 __isl_give isl_map *isl_map_lex_gt_first(__isl_take isl_space *space,
5402 	unsigned n)
5403 {
5404 	return map_lex_gte_first(space, n, 0);
5405 }
5406 
isl_map_lex_ge_first(__isl_take isl_space * space,unsigned n)5407 __isl_give isl_map *isl_map_lex_ge_first(__isl_take isl_space *space,
5408 	unsigned n)
5409 {
5410 	return map_lex_gte_first(space, n, 1);
5411 }
5412 
isl_map_lex_gt(__isl_take isl_space * set_space)5413 __isl_give isl_map *isl_map_lex_gt(__isl_take isl_space *set_space)
5414 {
5415 	return map_lex_gte(isl_space_map_from_set(set_space), 0);
5416 }
5417 
isl_map_lex_ge(__isl_take isl_space * set_space)5418 __isl_give isl_map *isl_map_lex_ge(__isl_take isl_space *set_space)
5419 {
5420 	return map_lex_gte(isl_space_map_from_set(set_space), 1);
5421 }
5422 
isl_set_lex_le_set(__isl_take isl_set * set1,__isl_take isl_set * set2)5423 __isl_give isl_map *isl_set_lex_le_set(__isl_take isl_set *set1,
5424 	__isl_take isl_set *set2)
5425 {
5426 	isl_map *map;
5427 	map = isl_map_lex_le(isl_set_get_space(set1));
5428 	map = isl_map_intersect_domain(map, set1);
5429 	map = isl_map_intersect_range(map, set2);
5430 	return map;
5431 }
5432 
isl_set_lex_lt_set(__isl_take isl_set * set1,__isl_take isl_set * set2)5433 __isl_give isl_map *isl_set_lex_lt_set(__isl_take isl_set *set1,
5434 	__isl_take isl_set *set2)
5435 {
5436 	isl_map *map;
5437 	map = isl_map_lex_lt(isl_set_get_space(set1));
5438 	map = isl_map_intersect_domain(map, set1);
5439 	map = isl_map_intersect_range(map, set2);
5440 	return map;
5441 }
5442 
isl_set_lex_ge_set(__isl_take isl_set * set1,__isl_take isl_set * set2)5443 __isl_give isl_map *isl_set_lex_ge_set(__isl_take isl_set *set1,
5444 	__isl_take isl_set *set2)
5445 {
5446 	isl_map *map;
5447 	map = isl_map_lex_ge(isl_set_get_space(set1));
5448 	map = isl_map_intersect_domain(map, set1);
5449 	map = isl_map_intersect_range(map, set2);
5450 	return map;
5451 }
5452 
isl_set_lex_gt_set(__isl_take isl_set * set1,__isl_take isl_set * set2)5453 __isl_give isl_map *isl_set_lex_gt_set(__isl_take isl_set *set1,
5454 	__isl_take isl_set *set2)
5455 {
5456 	isl_map *map;
5457 	map = isl_map_lex_gt(isl_set_get_space(set1));
5458 	map = isl_map_intersect_domain(map, set1);
5459 	map = isl_map_intersect_range(map, set2);
5460 	return map;
5461 }
5462 
isl_map_lex_le_map(__isl_take isl_map * map1,__isl_take isl_map * map2)5463 __isl_give isl_map *isl_map_lex_le_map(__isl_take isl_map *map1,
5464 	__isl_take isl_map *map2)
5465 {
5466 	isl_map *map;
5467 	map = isl_map_lex_le(isl_space_range(isl_map_get_space(map1)));
5468 	map = isl_map_apply_domain(map, isl_map_reverse(map1));
5469 	map = isl_map_apply_range(map, isl_map_reverse(map2));
5470 	return map;
5471 }
5472 
isl_map_lex_lt_map(__isl_take isl_map * map1,__isl_take isl_map * map2)5473 __isl_give isl_map *isl_map_lex_lt_map(__isl_take isl_map *map1,
5474 	__isl_take isl_map *map2)
5475 {
5476 	isl_map *map;
5477 	map = isl_map_lex_lt(isl_space_range(isl_map_get_space(map1)));
5478 	map = isl_map_apply_domain(map, isl_map_reverse(map1));
5479 	map = isl_map_apply_range(map, isl_map_reverse(map2));
5480 	return map;
5481 }
5482 
isl_map_lex_ge_map(__isl_take isl_map * map1,__isl_take isl_map * map2)5483 __isl_give isl_map *isl_map_lex_ge_map(__isl_take isl_map *map1,
5484 	__isl_take isl_map *map2)
5485 {
5486 	isl_map *map;
5487 	map = isl_map_lex_ge(isl_space_range(isl_map_get_space(map1)));
5488 	map = isl_map_apply_domain(map, isl_map_reverse(map1));
5489 	map = isl_map_apply_range(map, isl_map_reverse(map2));
5490 	return map;
5491 }
5492 
isl_map_lex_gt_map(__isl_take isl_map * map1,__isl_take isl_map * map2)5493 __isl_give isl_map *isl_map_lex_gt_map(__isl_take isl_map *map1,
5494 	__isl_take isl_map *map2)
5495 {
5496 	isl_map *map;
5497 	map = isl_map_lex_gt(isl_space_range(isl_map_get_space(map1)));
5498 	map = isl_map_apply_domain(map, isl_map_reverse(map1));
5499 	map = isl_map_apply_range(map, isl_map_reverse(map2));
5500 	return map;
5501 }
5502 
5503 /* For the div d = floor(f/m) at position "div", add the constraint
5504  *
5505  *		f - m d >= 0
5506  */
add_upper_div_constraint(__isl_take isl_basic_map * bmap,unsigned div)5507 static __isl_give isl_basic_map *add_upper_div_constraint(
5508 	__isl_take isl_basic_map *bmap, unsigned div)
5509 {
5510 	int i;
5511 	isl_size v_div = isl_basic_map_var_offset(bmap, isl_dim_div);
5512 	isl_size n_div;
5513 	unsigned pos;
5514 
5515 	n_div = isl_basic_map_dim(bmap, isl_dim_div);
5516 	if (v_div < 0 || n_div < 0)
5517 		return isl_basic_map_free(bmap);
5518 	pos = v_div + div;
5519 	i = isl_basic_map_alloc_inequality(bmap);
5520 	if (i < 0)
5521 		return isl_basic_map_free(bmap);
5522 	isl_seq_cpy(bmap->ineq[i], bmap->div[div] + 1, 1 + v_div + n_div);
5523 	isl_int_neg(bmap->ineq[i][1 + pos], bmap->div[div][0]);
5524 
5525 	return bmap;
5526 }
5527 
5528 /* For the div d = floor(f/m) at position "div", add the constraint
5529  *
5530  *		-(f-(m-1)) + m d >= 0
5531  */
add_lower_div_constraint(__isl_take isl_basic_map * bmap,unsigned div)5532 static __isl_give isl_basic_map *add_lower_div_constraint(
5533 	__isl_take isl_basic_map *bmap, unsigned div)
5534 {
5535 	int i;
5536 	isl_size v_div = isl_basic_map_var_offset(bmap, isl_dim_div);
5537 	isl_size n_div;
5538 	unsigned pos;
5539 
5540 	n_div = isl_basic_map_dim(bmap, isl_dim_div);
5541 	if (v_div < 0 || n_div < 0)
5542 		return isl_basic_map_free(bmap);
5543 	pos = v_div + div;
5544 	i = isl_basic_map_alloc_inequality(bmap);
5545 	if (i < 0)
5546 		return isl_basic_map_free(bmap);
5547 	isl_seq_neg(bmap->ineq[i], bmap->div[div] + 1, 1 + v_div + n_div);
5548 	isl_int_set(bmap->ineq[i][1 + pos], bmap->div[div][0]);
5549 	isl_int_add(bmap->ineq[i][0], bmap->ineq[i][0], bmap->ineq[i][1 + pos]);
5550 	isl_int_sub_ui(bmap->ineq[i][0], bmap->ineq[i][0], 1);
5551 
5552 	return bmap;
5553 }
5554 
5555 /* For the div d = floor(f/m) at position "pos", add the constraints
5556  *
5557  *		f - m d >= 0
5558  *		-(f-(m-1)) + m d >= 0
5559  *
5560  * Note that the second constraint is the negation of
5561  *
5562  *		f - m d >= m
5563  */
isl_basic_map_add_div_constraints(__isl_take isl_basic_map * bmap,unsigned pos)5564 __isl_give isl_basic_map *isl_basic_map_add_div_constraints(
5565 	__isl_take isl_basic_map *bmap, unsigned pos)
5566 {
5567 	bmap = add_upper_div_constraint(bmap, pos);
5568 	bmap = add_lower_div_constraint(bmap, pos);
5569 	return bmap;
5570 }
5571 
5572 /* For each known div d = floor(f/m), add the constraints
5573  *
5574  *		f - m d >= 0
5575  *		-(f-(m-1)) + m d >= 0
5576  *
5577  * Remove duplicate constraints in case of some these div constraints
5578  * already appear in "bmap".
5579  */
isl_basic_map_add_known_div_constraints(__isl_take isl_basic_map * bmap)5580 __isl_give isl_basic_map *isl_basic_map_add_known_div_constraints(
5581 	__isl_take isl_basic_map *bmap)
5582 {
5583 	isl_size n_div;
5584 
5585 	n_div = isl_basic_map_dim(bmap, isl_dim_div);
5586 	if (n_div < 0)
5587 		return isl_basic_map_free(bmap);
5588 	if (n_div == 0)
5589 		return bmap;
5590 
5591 	bmap = add_known_div_constraints(bmap);
5592 	bmap = isl_basic_map_remove_duplicate_constraints(bmap, NULL, 0);
5593 	bmap = isl_basic_map_finalize(bmap);
5594 	return bmap;
5595 }
5596 
5597 /* Add the div constraint of sign "sign" for div "div" of "bmap".
5598  *
5599  * In particular, if this div is of the form d = floor(f/m),
5600  * then add the constraint
5601  *
5602  *		f - m d >= 0
5603  *
5604  * if sign < 0 or the constraint
5605  *
5606  *		-(f-(m-1)) + m d >= 0
5607  *
5608  * if sign > 0.
5609  */
isl_basic_map_add_div_constraint(__isl_take isl_basic_map * bmap,unsigned div,int sign)5610 __isl_give isl_basic_map *isl_basic_map_add_div_constraint(
5611 	__isl_take isl_basic_map *bmap, unsigned div, int sign)
5612 {
5613 	if (sign < 0)
5614 		return add_upper_div_constraint(bmap, div);
5615 	else
5616 		return add_lower_div_constraint(bmap, div);
5617 }
5618 
isl_basic_map_underlying_set(__isl_take isl_basic_map * bmap)5619 __isl_give isl_basic_set *isl_basic_map_underlying_set(
5620 	__isl_take isl_basic_map *bmap)
5621 {
5622 	isl_space *space;
5623 
5624 	if (!bmap)
5625 		goto error;
5626 	if (bmap->dim->nparam == 0 && bmap->dim->n_in == 0 &&
5627 	    bmap->n_div == 0 &&
5628 	    !isl_space_is_named_or_nested(bmap->dim, isl_dim_in) &&
5629 	    !isl_space_is_named_or_nested(bmap->dim, isl_dim_out))
5630 		return bset_from_bmap(bmap);
5631 	bmap = isl_basic_map_cow(bmap);
5632 	if (!bmap)
5633 		return NULL;
5634 	space = isl_basic_map_take_space(bmap);
5635 	space = isl_space_underlying(space, bmap->n_div);
5636 	bmap = isl_basic_map_restore_space(bmap, space);
5637 	if (!bmap)
5638 		return NULL;
5639 	bmap->extra -= bmap->n_div;
5640 	bmap->n_div = 0;
5641 	bmap = isl_basic_map_finalize(bmap);
5642 	return bset_from_bmap(bmap);
5643 error:
5644 	isl_basic_map_free(bmap);
5645 	return NULL;
5646 }
5647 
isl_basic_set_underlying_set(__isl_take isl_basic_set * bset)5648 __isl_give isl_basic_set *isl_basic_set_underlying_set(
5649 		__isl_take isl_basic_set *bset)
5650 {
5651 	return isl_basic_map_underlying_set(bset_to_bmap(bset));
5652 }
5653 
5654 /* Replace each element in "list" by the result of applying
5655  * isl_basic_map_underlying_set to the element.
5656  */
isl_basic_map_list_underlying_set(__isl_take isl_basic_map_list * list)5657 __isl_give isl_basic_set_list *isl_basic_map_list_underlying_set(
5658 	__isl_take isl_basic_map_list *list)
5659 {
5660 	int i;
5661 	isl_size n;
5662 
5663 	n = isl_basic_map_list_n_basic_map(list);
5664 	if (n < 0)
5665 		goto error;
5666 
5667 	for (i = 0; i < n; ++i) {
5668 		isl_basic_map *bmap;
5669 		isl_basic_set *bset;
5670 
5671 		bmap = isl_basic_map_list_get_basic_map(list, i);
5672 		bset = isl_basic_set_underlying_set(bmap);
5673 		list = isl_basic_set_list_set_basic_set(list, i, bset);
5674 	}
5675 
5676 	return list;
5677 error:
5678 	isl_basic_map_list_free(list);
5679 	return NULL;
5680 }
5681 
isl_basic_map_overlying_set(__isl_take isl_basic_set * bset,__isl_take isl_basic_map * like)5682 __isl_give isl_basic_map *isl_basic_map_overlying_set(
5683 	__isl_take isl_basic_set *bset, __isl_take isl_basic_map *like)
5684 {
5685 	struct isl_basic_map *bmap;
5686 	struct isl_ctx *ctx;
5687 	isl_size dim, bmap_total;
5688 	unsigned total;
5689 	int i;
5690 
5691 	if (!bset || !like)
5692 		goto error;
5693 	ctx = bset->ctx;
5694 	if (isl_basic_set_check_no_params(bset) < 0 ||
5695 	    isl_basic_set_check_no_locals(bset) < 0)
5696 		goto error;
5697 	dim = isl_basic_set_dim(bset, isl_dim_set);
5698 	bmap_total = isl_basic_map_dim(like, isl_dim_all);
5699 	if (dim < 0 || bmap_total < 0)
5700 		goto error;
5701 	isl_assert(ctx, dim == bmap_total, goto error);
5702 	if (like->n_div == 0) {
5703 		isl_space *space = isl_basic_map_get_space(like);
5704 		isl_basic_map_free(like);
5705 		return isl_basic_map_reset_space(bset, space);
5706 	}
5707 	bset = isl_basic_set_cow(bset);
5708 	if (!bset)
5709 		goto error;
5710 	total = dim + bset->extra;
5711 	bmap = bset_to_bmap(bset);
5712 	isl_space_free(isl_basic_map_take_space(bmap));
5713 	bmap = isl_basic_map_restore_space(bmap, isl_basic_map_get_space(like));
5714 	if (!bmap)
5715 		goto error;
5716 	bmap->n_div = like->n_div;
5717 	bmap->extra += like->n_div;
5718 	if (bmap->extra) {
5719 		unsigned ltotal;
5720 		isl_int **div;
5721 		ltotal = total - bmap->extra + like->extra;
5722 		if (ltotal > total)
5723 			ltotal = total;
5724 		bmap->block2 = isl_blk_extend(ctx, bmap->block2,
5725 					bmap->extra * (1 + 1 + total));
5726 		if (isl_blk_is_error(bmap->block2))
5727 			goto error;
5728 		div = isl_realloc_array(ctx, bmap->div, isl_int *, bmap->extra);
5729 		if (!div)
5730 			goto error;
5731 		bmap->div = div;
5732 		for (i = 0; i < bmap->extra; ++i)
5733 			bmap->div[i] = bmap->block2.data + i * (1 + 1 + total);
5734 		for (i = 0; i < like->n_div; ++i) {
5735 			isl_seq_cpy(bmap->div[i], like->div[i], 1 + 1 + ltotal);
5736 			isl_seq_clr(bmap->div[i]+1+1+ltotal, total - ltotal);
5737 		}
5738 		bmap = isl_basic_map_add_known_div_constraints(bmap);
5739 	}
5740 	isl_basic_map_free(like);
5741 	bmap = isl_basic_map_simplify(bmap);
5742 	bmap = isl_basic_map_finalize(bmap);
5743 	return bmap;
5744 error:
5745 	isl_basic_map_free(like);
5746 	isl_basic_set_free(bset);
5747 	return NULL;
5748 }
5749 
isl_basic_set_from_underlying_set(__isl_take isl_basic_set * bset,__isl_take isl_basic_set * like)5750 __isl_give isl_basic_set *isl_basic_set_from_underlying_set(
5751 	__isl_take isl_basic_set *bset, __isl_take isl_basic_set *like)
5752 {
5753 	return bset_from_bmap(isl_basic_map_overlying_set(bset,
5754 							bset_to_bmap(like)));
5755 }
5756 
isl_map_underlying_set(__isl_take isl_map * map)5757 __isl_give isl_set *isl_map_underlying_set(__isl_take isl_map *map)
5758 {
5759 	int i;
5760 
5761 	map = isl_map_cow(map);
5762 	if (!map)
5763 		return NULL;
5764 	map->dim = isl_space_cow(map->dim);
5765 	if (!map->dim)
5766 		goto error;
5767 
5768 	for (i = 1; i < map->n; ++i)
5769 		isl_assert(map->ctx, map->p[0]->n_div == map->p[i]->n_div,
5770 				goto error);
5771 	for (i = 0; i < map->n; ++i) {
5772 		map->p[i] = bset_to_bmap(
5773 				isl_basic_map_underlying_set(map->p[i]));
5774 		if (!map->p[i])
5775 			goto error;
5776 	}
5777 	if (map->n == 0)
5778 		map->dim = isl_space_underlying(map->dim, 0);
5779 	else {
5780 		isl_space_free(map->dim);
5781 		map->dim = isl_space_copy(map->p[0]->dim);
5782 	}
5783 	if (!map->dim)
5784 		goto error;
5785 	return set_from_map(map);
5786 error:
5787 	isl_map_free(map);
5788 	return NULL;
5789 }
5790 
5791 /* Replace the space of "bmap" by "space".
5792  *
5793  * If the space of "bmap" is identical to "space" (including the identifiers
5794  * of the input and output dimensions), then simply return the original input.
5795  */
isl_basic_map_reset_space(__isl_take isl_basic_map * bmap,__isl_take isl_space * space)5796 __isl_give isl_basic_map *isl_basic_map_reset_space(
5797 	__isl_take isl_basic_map *bmap, __isl_take isl_space *space)
5798 {
5799 	isl_bool equal;
5800 	isl_space *bmap_space;
5801 
5802 	bmap_space = isl_basic_map_peek_space(bmap);
5803 	equal = isl_space_is_equal(bmap_space, space);
5804 	if (equal >= 0 && equal)
5805 		equal = isl_space_has_equal_ids(bmap_space, space);
5806 	if (equal < 0)
5807 		goto error;
5808 	if (equal) {
5809 		isl_space_free(space);
5810 		return bmap;
5811 	}
5812 	isl_space_free(isl_basic_map_take_space(bmap));
5813 	bmap = isl_basic_map_restore_space(bmap, space);
5814 
5815 	bmap = isl_basic_map_finalize(bmap);
5816 
5817 	return bmap;
5818 error:
5819 	isl_basic_map_free(bmap);
5820 	isl_space_free(space);
5821 	return NULL;
5822 }
5823 
isl_basic_set_reset_space(__isl_take isl_basic_set * bset,__isl_take isl_space * space)5824 __isl_give isl_basic_set *isl_basic_set_reset_space(
5825 	__isl_take isl_basic_set *bset, __isl_take isl_space *space)
5826 {
5827 	return bset_from_bmap(isl_basic_map_reset_space(bset_to_bmap(bset),
5828 							space));
5829 }
5830 
5831 /* Check that the total dimensions of "map" and "space" are the same.
5832  */
check_map_space_equal_total_dim(__isl_keep isl_map * map,__isl_keep isl_space * space)5833 static isl_stat check_map_space_equal_total_dim(__isl_keep isl_map *map,
5834 	__isl_keep isl_space *space)
5835 {
5836 	isl_size dim1, dim2;
5837 
5838 	dim1 = isl_map_dim(map, isl_dim_all);
5839 	dim2 = isl_space_dim(space, isl_dim_all);
5840 	if (dim1 < 0 || dim2 < 0)
5841 		return isl_stat_error;
5842 	if (dim1 == dim2)
5843 		return isl_stat_ok;
5844 	isl_die(isl_map_get_ctx(map), isl_error_invalid,
5845 		"total dimensions do not match", return isl_stat_error);
5846 }
5847 
isl_map_reset_space(__isl_take isl_map * map,__isl_take isl_space * space)5848 __isl_give isl_map *isl_map_reset_space(__isl_take isl_map *map,
5849 	__isl_take isl_space *space)
5850 {
5851 	int i;
5852 
5853 	map = isl_map_cow(map);
5854 	if (!map || !space)
5855 		goto error;
5856 
5857 	for (i = 0; i < map->n; ++i) {
5858 		map->p[i] = isl_basic_map_reset_space(map->p[i],
5859 						    isl_space_copy(space));
5860 		if (!map->p[i])
5861 			goto error;
5862 	}
5863 	isl_space_free(isl_map_take_space(map));
5864 	map = isl_map_restore_space(map, space);
5865 
5866 	return map;
5867 error:
5868 	isl_map_free(map);
5869 	isl_space_free(space);
5870 	return NULL;
5871 }
5872 
5873 /* Replace the space of "map" by "space", without modifying
5874  * the dimension of "map".
5875  *
5876  * If the space of "map" is identical to "space" (including the identifiers
5877  * of the input and output dimensions), then simply return the original input.
5878  */
isl_map_reset_equal_dim_space(__isl_take isl_map * map,__isl_take isl_space * space)5879 __isl_give isl_map *isl_map_reset_equal_dim_space(__isl_take isl_map *map,
5880 	__isl_take isl_space *space)
5881 {
5882 	isl_bool equal;
5883 	isl_space *map_space;
5884 
5885 	map_space = isl_map_peek_space(map);
5886 	equal = isl_space_is_equal(map_space, space);
5887 	if (equal >= 0 && equal)
5888 		equal = isl_space_has_equal_ids(map_space, space);
5889 	if (equal < 0)
5890 		goto error;
5891 	if (equal) {
5892 		isl_space_free(space);
5893 		return map;
5894 	}
5895 	if (check_map_space_equal_total_dim(map, space) < 0)
5896 		goto error;
5897 	return isl_map_reset_space(map, space);
5898 error:
5899 	isl_map_free(map);
5900 	isl_space_free(space);
5901 	return NULL;
5902 }
5903 
isl_set_reset_space(__isl_take isl_set * set,__isl_take isl_space * space)5904 __isl_give isl_set *isl_set_reset_space(__isl_take isl_set *set,
5905 	__isl_take isl_space *space)
5906 {
5907 	return set_from_map(isl_map_reset_space(set_to_map(set), space));
5908 }
5909 
5910 /* Compute the parameter domain of the given basic set.
5911  */
isl_basic_set_params(__isl_take isl_basic_set * bset)5912 __isl_give isl_basic_set *isl_basic_set_params(__isl_take isl_basic_set *bset)
5913 {
5914 	isl_bool is_params;
5915 	isl_space *space;
5916 	isl_size n;
5917 
5918 	is_params = isl_basic_set_is_params(bset);
5919 	if (is_params < 0)
5920 		return isl_basic_set_free(bset);
5921 	if (is_params)
5922 		return bset;
5923 
5924 	n = isl_basic_set_dim(bset, isl_dim_set);
5925 	if (n < 0)
5926 		return isl_basic_set_free(bset);
5927 	bset = isl_basic_set_project_out(bset, isl_dim_set, 0, n);
5928 	space = isl_basic_set_get_space(bset);
5929 	space = isl_space_params(space);
5930 	bset = isl_basic_set_reset_space(bset, space);
5931 	return bset;
5932 }
5933 
5934 /* Construct a zero-dimensional basic set with the given parameter domain.
5935  */
isl_basic_set_from_params(__isl_take isl_basic_set * bset)5936 __isl_give isl_basic_set *isl_basic_set_from_params(
5937 	__isl_take isl_basic_set *bset)
5938 {
5939 	isl_space *space;
5940 	space = isl_basic_set_get_space(bset);
5941 	space = isl_space_set_from_params(space);
5942 	bset = isl_basic_set_reset_space(bset, space);
5943 	return bset;
5944 }
5945 
5946 /* Compute the parameter domain of the given set.
5947  */
isl_set_params(__isl_take isl_set * set)5948 __isl_give isl_set *isl_set_params(__isl_take isl_set *set)
5949 {
5950 	return isl_map_params(set_to_map(set));
5951 }
5952 
5953 /* Construct a zero-dimensional set with the given parameter domain.
5954  */
isl_set_from_params(__isl_take isl_set * set)5955 __isl_give isl_set *isl_set_from_params(__isl_take isl_set *set)
5956 {
5957 	isl_space *space;
5958 	space = isl_set_get_space(set);
5959 	space = isl_space_set_from_params(space);
5960 	set = isl_set_reset_space(set, space);
5961 	return set;
5962 }
5963 
5964 /* Compute the parameter domain of the given map.
5965  */
isl_map_params(__isl_take isl_map * map)5966 __isl_give isl_set *isl_map_params(__isl_take isl_map *map)
5967 {
5968 	isl_space *space;
5969 	isl_size n_in, n_out;
5970 
5971 	n_in = isl_map_dim(map, isl_dim_in);
5972 	n_out = isl_map_dim(map, isl_dim_out);
5973 	if (n_in < 0 || n_out < 0)
5974 		return isl_map_free(map);
5975 	map = isl_map_project_out(map, isl_dim_in, 0, n_in);
5976 	map = isl_map_project_out(map, isl_dim_out, 0, n_out);
5977 	space = isl_map_get_space(map);
5978 	space = isl_space_params(space);
5979 	map = isl_map_reset_space(map, space);
5980 	return map;
5981 }
5982 
isl_basic_map_domain(__isl_take isl_basic_map * bmap)5983 __isl_give isl_basic_set *isl_basic_map_domain(__isl_take isl_basic_map *bmap)
5984 {
5985 	isl_space *space;
5986 	isl_size n_out;
5987 
5988 	n_out = isl_basic_map_dim(bmap, isl_dim_out);
5989 	if (n_out < 0)
5990 		return isl_basic_map_free(bmap);
5991 	space = isl_space_domain(isl_basic_map_get_space(bmap));
5992 
5993 	bmap = isl_basic_map_project_out(bmap, isl_dim_out, 0, n_out);
5994 
5995 	return isl_basic_map_reset_space(bmap, space);
5996 }
5997 
isl_basic_map_may_be_set(__isl_keep isl_basic_map * bmap)5998 isl_bool isl_basic_map_may_be_set(__isl_keep isl_basic_map *bmap)
5999 {
6000 	if (!bmap)
6001 		return isl_bool_error;
6002 	return isl_space_may_be_set(bmap->dim);
6003 }
6004 
6005 /* Is this basic map actually a set?
6006  * Users should never call this function.  Outside of isl,
6007  * the type should indicate whether something is a set or a map.
6008  */
isl_basic_map_is_set(__isl_keep isl_basic_map * bmap)6009 isl_bool isl_basic_map_is_set(__isl_keep isl_basic_map *bmap)
6010 {
6011 	if (!bmap)
6012 		return isl_bool_error;
6013 	return isl_space_is_set(bmap->dim);
6014 }
6015 
isl_basic_map_range(__isl_take isl_basic_map * bmap)6016 __isl_give isl_basic_set *isl_basic_map_range(__isl_take isl_basic_map *bmap)
6017 {
6018 	isl_bool is_set;
6019 
6020 	is_set = isl_basic_map_is_set(bmap);
6021 	if (is_set < 0)
6022 		goto error;
6023 	if (is_set)
6024 		return bmap;
6025 	return isl_basic_map_domain(isl_basic_map_reverse(bmap));
6026 error:
6027 	isl_basic_map_free(bmap);
6028 	return NULL;
6029 }
6030 
isl_basic_map_domain_map(__isl_take isl_basic_map * bmap)6031 __isl_give isl_basic_map *isl_basic_map_domain_map(
6032 	__isl_take isl_basic_map *bmap)
6033 {
6034 	int i;
6035 	isl_space *space;
6036 	isl_basic_map *domain;
6037 	isl_size nparam, n_in, n_out;
6038 
6039 	nparam = isl_basic_map_dim(bmap, isl_dim_param);
6040 	n_in = isl_basic_map_dim(bmap, isl_dim_in);
6041 	n_out = isl_basic_map_dim(bmap, isl_dim_out);
6042 	if (nparam < 0 || n_in < 0 || n_out < 0)
6043 		return isl_basic_map_free(bmap);
6044 
6045 	space = isl_basic_map_get_space(bmap);
6046 	space = isl_space_from_range(isl_space_domain(space));
6047 	domain = isl_basic_map_universe(space);
6048 
6049 	bmap = isl_basic_map_from_domain(isl_basic_map_wrap(bmap));
6050 	bmap = isl_basic_map_apply_range(bmap, domain);
6051 	bmap = isl_basic_map_extend_constraints(bmap, n_in, 0);
6052 
6053 	for (i = 0; i < n_in; ++i)
6054 		bmap = isl_basic_map_equate(bmap, isl_dim_in, i,
6055 						    isl_dim_out, i);
6056 
6057 	bmap = isl_basic_map_gauss(bmap, NULL);
6058 	return isl_basic_map_finalize(bmap);
6059 }
6060 
isl_basic_map_range_map(__isl_take isl_basic_map * bmap)6061 __isl_give isl_basic_map *isl_basic_map_range_map(
6062 	__isl_take isl_basic_map *bmap)
6063 {
6064 	int i;
6065 	isl_space *space;
6066 	isl_basic_map *range;
6067 	isl_size nparam, n_in, n_out;
6068 
6069 	nparam = isl_basic_map_dim(bmap, isl_dim_param);
6070 	n_in = isl_basic_map_dim(bmap, isl_dim_in);
6071 	n_out = isl_basic_map_dim(bmap, isl_dim_out);
6072 	if (nparam < 0 || n_in < 0 || n_out < 0)
6073 		return isl_basic_map_free(bmap);
6074 
6075 	space = isl_basic_map_get_space(bmap);
6076 	space = isl_space_from_range(isl_space_range(space));
6077 	range = isl_basic_map_universe(space);
6078 
6079 	bmap = isl_basic_map_from_domain(isl_basic_map_wrap(bmap));
6080 	bmap = isl_basic_map_apply_range(bmap, range);
6081 	bmap = isl_basic_map_extend_constraints(bmap, n_out, 0);
6082 
6083 	for (i = 0; i < n_out; ++i)
6084 		bmap = isl_basic_map_equate(bmap, isl_dim_in, n_in + i,
6085 						    isl_dim_out, i);
6086 
6087 	bmap = isl_basic_map_gauss(bmap, NULL);
6088 	return isl_basic_map_finalize(bmap);
6089 }
6090 
isl_map_may_be_set(__isl_keep isl_map * map)6091 int isl_map_may_be_set(__isl_keep isl_map *map)
6092 {
6093 	if (!map)
6094 		return -1;
6095 	return isl_space_may_be_set(map->dim);
6096 }
6097 
6098 /* Is this map actually a set?
6099  * Users should never call this function.  Outside of isl,
6100  * the type should indicate whether something is a set or a map.
6101  */
isl_map_is_set(__isl_keep isl_map * map)6102 isl_bool isl_map_is_set(__isl_keep isl_map *map)
6103 {
6104 	if (!map)
6105 		return isl_bool_error;
6106 	return isl_space_is_set(map->dim);
6107 }
6108 
isl_map_range(__isl_take isl_map * map)6109 __isl_give isl_set *isl_map_range(__isl_take isl_map *map)
6110 {
6111 	isl_space *space;
6112 	isl_size n_in;
6113 
6114 	n_in = isl_map_dim(map, isl_dim_in);
6115 	if (n_in < 0)
6116 		return set_from_map(isl_map_free(map));
6117 	space = isl_space_range(isl_map_get_space(map));
6118 
6119 	map = isl_map_project_out(map, isl_dim_in, 0, n_in);
6120 
6121 	return set_from_map(isl_map_reset_space(map, space));
6122 }
6123 
6124 /* Transform "map" by applying "fn_space" to its space and "fn_bmap"
6125  * to each of its basic maps.
6126  */
isl_map_transform(__isl_take isl_map * map,__isl_give isl_space * (* fn_space)(__isl_take isl_space * space),__isl_give isl_basic_map * (* fn_bmap)(__isl_take isl_basic_map * bmap))6127 static __isl_give isl_map *isl_map_transform(__isl_take isl_map *map,
6128 	__isl_give isl_space *(*fn_space)(__isl_take isl_space *space),
6129 	__isl_give isl_basic_map *(*fn_bmap)(__isl_take isl_basic_map *bmap))
6130 {
6131 	int i;
6132 	isl_space *space;
6133 
6134 	map = isl_map_cow(map);
6135 	if (!map)
6136 		return NULL;
6137 
6138 	for (i = 0; i < map->n; ++i) {
6139 		map->p[i] = fn_bmap(map->p[i]);
6140 		if (!map->p[i])
6141 			return isl_map_free(map);
6142 	}
6143 	map = isl_map_unmark_normalized(map);
6144 
6145 	space = isl_map_take_space(map);
6146 	space = fn_space(space);
6147 	map = isl_map_restore_space(map, space);
6148 
6149 	return map;
6150 }
6151 
isl_map_domain_map(__isl_take isl_map * map)6152 __isl_give isl_map *isl_map_domain_map(__isl_take isl_map *map)
6153 {
6154 	return isl_map_transform(map, &isl_space_domain_map,
6155 					&isl_basic_map_domain_map);
6156 }
6157 
isl_map_range_map(__isl_take isl_map * map)6158 __isl_give isl_map *isl_map_range_map(__isl_take isl_map *map)
6159 {
6160 	return isl_map_transform(map, &isl_space_range_map,
6161 					&isl_basic_map_range_map);
6162 }
6163 
6164 /* Given a wrapped map of the form A[B -> C],
6165  * return the map A[B -> C] -> B.
6166  */
isl_set_wrapped_domain_map(__isl_take isl_set * set)6167 __isl_give isl_map *isl_set_wrapped_domain_map(__isl_take isl_set *set)
6168 {
6169 	isl_id *id;
6170 	isl_map *map;
6171 
6172 	if (!set)
6173 		return NULL;
6174 	if (!isl_set_has_tuple_id(set))
6175 		return isl_map_domain_map(isl_set_unwrap(set));
6176 
6177 	id = isl_set_get_tuple_id(set);
6178 	map = isl_map_domain_map(isl_set_unwrap(set));
6179 	map = isl_map_set_tuple_id(map, isl_dim_in, id);
6180 
6181 	return map;
6182 }
6183 
isl_basic_map_from_domain(__isl_take isl_basic_set * bset)6184 __isl_give isl_basic_map *isl_basic_map_from_domain(
6185 	__isl_take isl_basic_set *bset)
6186 {
6187 	return isl_basic_map_reverse(isl_basic_map_from_range(bset));
6188 }
6189 
isl_basic_map_from_range(__isl_take isl_basic_set * bset)6190 __isl_give isl_basic_map *isl_basic_map_from_range(
6191 	__isl_take isl_basic_set *bset)
6192 {
6193 	isl_space *space;
6194 	space = isl_basic_set_get_space(bset);
6195 	space = isl_space_from_range(space);
6196 	bset = isl_basic_set_reset_space(bset, space);
6197 	return bset_to_bmap(bset);
6198 }
6199 
6200 /* Create a relation with the given set as range.
6201  * The domain of the created relation is a zero-dimensional
6202  * flat anonymous space.
6203  */
isl_map_from_range(__isl_take isl_set * set)6204 __isl_give isl_map *isl_map_from_range(__isl_take isl_set *set)
6205 {
6206 	isl_space *space;
6207 	space = isl_set_get_space(set);
6208 	space = isl_space_from_range(space);
6209 	set = isl_set_reset_space(set, space);
6210 	return set_to_map(set);
6211 }
6212 
6213 /* Create a relation with the given set as domain.
6214  * The range of the created relation is a zero-dimensional
6215  * flat anonymous space.
6216  */
isl_map_from_domain(__isl_take isl_set * set)6217 __isl_give isl_map *isl_map_from_domain(__isl_take isl_set *set)
6218 {
6219 	return isl_map_reverse(isl_map_from_range(set));
6220 }
6221 
isl_basic_map_from_domain_and_range(__isl_take isl_basic_set * domain,__isl_take isl_basic_set * range)6222 __isl_give isl_basic_map *isl_basic_map_from_domain_and_range(
6223 	__isl_take isl_basic_set *domain, __isl_take isl_basic_set *range)
6224 {
6225 	return isl_basic_map_apply_range(isl_basic_map_reverse(domain), range);
6226 }
6227 
isl_map_from_domain_and_range(__isl_take isl_set * domain,__isl_take isl_set * range)6228 __isl_give isl_map *isl_map_from_domain_and_range(__isl_take isl_set *domain,
6229 	__isl_take isl_set *range)
6230 {
6231 	return isl_map_apply_range(isl_map_reverse(domain), range);
6232 }
6233 
6234 /* Return a newly allocated isl_map with given space and flags and
6235  * room for "n" basic maps.
6236  * Make sure that all cached information is cleared.
6237  */
isl_map_alloc_space(__isl_take isl_space * space,int n,unsigned flags)6238 __isl_give isl_map *isl_map_alloc_space(__isl_take isl_space *space, int n,
6239 	unsigned flags)
6240 {
6241 	struct isl_map *map;
6242 
6243 	if (!space)
6244 		return NULL;
6245 	if (n < 0)
6246 		isl_die(space->ctx, isl_error_internal,
6247 			"negative number of basic maps", goto error);
6248 	map = isl_calloc(space->ctx, struct isl_map,
6249 			sizeof(struct isl_map) +
6250 			(n - 1) * sizeof(struct isl_basic_map *));
6251 	if (!map)
6252 		goto error;
6253 
6254 	map->ctx = space->ctx;
6255 	isl_ctx_ref(map->ctx);
6256 	map->ref = 1;
6257 	map->size = n;
6258 	map->n = 0;
6259 	map->dim = space;
6260 	map->flags = flags;
6261 	return map;
6262 error:
6263 	isl_space_free(space);
6264 	return NULL;
6265 }
6266 
isl_basic_map_empty(__isl_take isl_space * space)6267 __isl_give isl_basic_map *isl_basic_map_empty(__isl_take isl_space *space)
6268 {
6269 	struct isl_basic_map *bmap;
6270 	bmap = isl_basic_map_alloc_space(space, 0, 1, 0);
6271 	bmap = isl_basic_map_set_to_empty(bmap);
6272 	return bmap;
6273 }
6274 
isl_basic_set_empty(__isl_take isl_space * space)6275 __isl_give isl_basic_set *isl_basic_set_empty(__isl_take isl_space *space)
6276 {
6277 	struct isl_basic_set *bset;
6278 	bset = isl_basic_set_alloc_space(space, 0, 1, 0);
6279 	bset = isl_basic_set_set_to_empty(bset);
6280 	return bset;
6281 }
6282 
isl_basic_map_universe(__isl_take isl_space * space)6283 __isl_give isl_basic_map *isl_basic_map_universe(__isl_take isl_space *space)
6284 {
6285 	struct isl_basic_map *bmap;
6286 	bmap = isl_basic_map_alloc_space(space, 0, 0, 0);
6287 	bmap = isl_basic_map_finalize(bmap);
6288 	return bmap;
6289 }
6290 
isl_basic_set_universe(__isl_take isl_space * space)6291 __isl_give isl_basic_set *isl_basic_set_universe(__isl_take isl_space *space)
6292 {
6293 	struct isl_basic_set *bset;
6294 	bset = isl_basic_set_alloc_space(space, 0, 0, 0);
6295 	bset = isl_basic_set_finalize(bset);
6296 	return bset;
6297 }
6298 
isl_basic_map_nat_universe(__isl_take isl_space * space)6299 __isl_give isl_basic_map *isl_basic_map_nat_universe(
6300 	__isl_take isl_space *space)
6301 {
6302 	int i;
6303 	isl_size total = isl_space_dim(space, isl_dim_all);
6304 	isl_basic_map *bmap;
6305 
6306 	if (total < 0)
6307 		space = isl_space_free(space);
6308 	bmap = isl_basic_map_alloc_space(space, 0, 0, total);
6309 	for (i = 0; i < total; ++i) {
6310 		int k = isl_basic_map_alloc_inequality(bmap);
6311 		if (k < 0)
6312 			goto error;
6313 		isl_seq_clr(bmap->ineq[k], 1 + total);
6314 		isl_int_set_si(bmap->ineq[k][1 + i], 1);
6315 	}
6316 	return bmap;
6317 error:
6318 	isl_basic_map_free(bmap);
6319 	return NULL;
6320 }
6321 
isl_basic_set_nat_universe(__isl_take isl_space * space)6322 __isl_give isl_basic_set *isl_basic_set_nat_universe(
6323 	__isl_take isl_space *space)
6324 {
6325 	return isl_basic_map_nat_universe(space);
6326 }
6327 
isl_map_nat_universe(__isl_take isl_space * space)6328 __isl_give isl_map *isl_map_nat_universe(__isl_take isl_space *space)
6329 {
6330 	return isl_map_from_basic_map(isl_basic_map_nat_universe(space));
6331 }
6332 
isl_set_nat_universe(__isl_take isl_space * space)6333 __isl_give isl_set *isl_set_nat_universe(__isl_take isl_space *space)
6334 {
6335 	return isl_map_nat_universe(space);
6336 }
6337 
isl_map_empty(__isl_take isl_space * space)6338 __isl_give isl_map *isl_map_empty(__isl_take isl_space *space)
6339 {
6340 	return isl_map_alloc_space(space, 0, ISL_MAP_DISJOINT);
6341 }
6342 
isl_set_empty(__isl_take isl_space * space)6343 __isl_give isl_set *isl_set_empty(__isl_take isl_space *space)
6344 {
6345 	return isl_set_alloc_space(space, 0, ISL_MAP_DISJOINT);
6346 }
6347 
isl_map_universe(__isl_take isl_space * space)6348 __isl_give isl_map *isl_map_universe(__isl_take isl_space *space)
6349 {
6350 	struct isl_map *map;
6351 	if (!space)
6352 		return NULL;
6353 	map = isl_map_alloc_space(isl_space_copy(space), 1, ISL_MAP_DISJOINT);
6354 	map = isl_map_add_basic_map(map, isl_basic_map_universe(space));
6355 	return map;
6356 }
6357 
6358 /* This function performs the same operation as isl_map_universe,
6359  * but is considered as a function on an isl_space when exported.
6360  */
isl_space_universe_map(__isl_take isl_space * space)6361 __isl_give isl_map *isl_space_universe_map(__isl_take isl_space *space)
6362 {
6363 	return isl_map_universe(space);
6364 }
6365 
isl_set_universe(__isl_take isl_space * space)6366 __isl_give isl_set *isl_set_universe(__isl_take isl_space *space)
6367 {
6368 	struct isl_set *set;
6369 	if (!space)
6370 		return NULL;
6371 	set = isl_set_alloc_space(isl_space_copy(space), 1, ISL_MAP_DISJOINT);
6372 	set = isl_set_add_basic_set(set, isl_basic_set_universe(space));
6373 	return set;
6374 }
6375 
6376 /* This function performs the same operation as isl_set_universe,
6377  * but is considered as a function on an isl_space when exported.
6378  */
isl_space_universe_set(__isl_take isl_space * space)6379 __isl_give isl_set *isl_space_universe_set(__isl_take isl_space *space)
6380 {
6381 	return isl_set_universe(space);
6382 }
6383 
isl_map_dup(__isl_keep isl_map * map)6384 __isl_give isl_map *isl_map_dup(__isl_keep isl_map *map)
6385 {
6386 	int i;
6387 	struct isl_map *dup;
6388 
6389 	if (!map)
6390 		return NULL;
6391 	dup = isl_map_alloc_space(isl_space_copy(map->dim), map->n, map->flags);
6392 	for (i = 0; i < map->n; ++i)
6393 		dup = isl_map_add_basic_map(dup, isl_basic_map_copy(map->p[i]));
6394 	return dup;
6395 }
6396 
isl_map_add_basic_map(__isl_take isl_map * map,__isl_take isl_basic_map * bmap)6397 __isl_give isl_map *isl_map_add_basic_map(__isl_take isl_map *map,
6398 						__isl_take isl_basic_map *bmap)
6399 {
6400 	if (!bmap || !map)
6401 		goto error;
6402 	if (isl_basic_map_plain_is_empty(bmap)) {
6403 		isl_basic_map_free(bmap);
6404 		return map;
6405 	}
6406 	if (isl_map_basic_map_check_equal_space(map, bmap) < 0)
6407 		goto error;
6408 	isl_assert(map->ctx, map->n < map->size, goto error);
6409 	map->p[map->n] = bmap;
6410 	map->n++;
6411 	map = isl_map_unmark_normalized(map);
6412 	return map;
6413 error:
6414 	if (map)
6415 		isl_map_free(map);
6416 	if (bmap)
6417 		isl_basic_map_free(bmap);
6418 	return NULL;
6419 }
6420 
isl_map_free(__isl_take isl_map * map)6421 __isl_null isl_map *isl_map_free(__isl_take isl_map *map)
6422 {
6423 	int i;
6424 
6425 	if (!map)
6426 		return NULL;
6427 
6428 	if (--map->ref > 0)
6429 		return NULL;
6430 
6431 	clear_caches(map);
6432 	isl_ctx_deref(map->ctx);
6433 	for (i = 0; i < map->n; ++i)
6434 		isl_basic_map_free(map->p[i]);
6435 	isl_space_free(map->dim);
6436 	free(map);
6437 
6438 	return NULL;
6439 }
6440 
isl_basic_map_fix_pos_si(__isl_take isl_basic_map * bmap,unsigned pos,int value)6441 static __isl_give isl_basic_map *isl_basic_map_fix_pos_si(
6442 	__isl_take isl_basic_map *bmap, unsigned pos, int value)
6443 {
6444 	int j;
6445 	isl_size total;
6446 
6447 	total = isl_basic_map_dim(bmap, isl_dim_all);
6448 	if (total < 0)
6449 		return isl_basic_map_free(bmap);
6450 
6451 	bmap = isl_basic_map_cow(bmap);
6452 	bmap = isl_basic_map_extend_constraints(bmap, 1, 0);
6453 	j = isl_basic_map_alloc_equality(bmap);
6454 	if (j < 0)
6455 		goto error;
6456 	isl_seq_clr(bmap->eq[j] + 1, total);
6457 	isl_int_set_si(bmap->eq[j][pos], -1);
6458 	isl_int_set_si(bmap->eq[j][0], value);
6459 	bmap = isl_basic_map_simplify(bmap);
6460 	return isl_basic_map_finalize(bmap);
6461 error:
6462 	isl_basic_map_free(bmap);
6463 	return NULL;
6464 }
6465 
isl_basic_map_fix_pos(__isl_take isl_basic_map * bmap,unsigned pos,isl_int value)6466 static __isl_give isl_basic_map *isl_basic_map_fix_pos(
6467 	__isl_take isl_basic_map *bmap, unsigned pos, isl_int value)
6468 {
6469 	int j;
6470 	isl_size total;
6471 
6472 	total = isl_basic_map_dim(bmap, isl_dim_all);
6473 	if (total < 0)
6474 		return isl_basic_map_free(bmap);
6475 
6476 	bmap = isl_basic_map_cow(bmap);
6477 	bmap = isl_basic_map_extend_constraints(bmap, 1, 0);
6478 	j = isl_basic_map_alloc_equality(bmap);
6479 	if (j < 0)
6480 		goto error;
6481 	isl_seq_clr(bmap->eq[j] + 1, total);
6482 	isl_int_set_si(bmap->eq[j][pos], -1);
6483 	isl_int_set(bmap->eq[j][0], value);
6484 	bmap = isl_basic_map_simplify(bmap);
6485 	return isl_basic_map_finalize(bmap);
6486 error:
6487 	isl_basic_map_free(bmap);
6488 	return NULL;
6489 }
6490 
isl_basic_map_fix_si(__isl_take isl_basic_map * bmap,enum isl_dim_type type,unsigned pos,int value)6491 __isl_give isl_basic_map *isl_basic_map_fix_si(__isl_take isl_basic_map *bmap,
6492 		enum isl_dim_type type, unsigned pos, int value)
6493 {
6494 	if (isl_basic_map_check_range(bmap, type, pos, 1) < 0)
6495 		return isl_basic_map_free(bmap);
6496 	return isl_basic_map_fix_pos_si(bmap,
6497 		isl_basic_map_offset(bmap, type) + pos, value);
6498 }
6499 
isl_basic_map_fix(__isl_take isl_basic_map * bmap,enum isl_dim_type type,unsigned pos,isl_int value)6500 __isl_give isl_basic_map *isl_basic_map_fix(__isl_take isl_basic_map *bmap,
6501 		enum isl_dim_type type, unsigned pos, isl_int value)
6502 {
6503 	if (isl_basic_map_check_range(bmap, type, pos, 1) < 0)
6504 		return isl_basic_map_free(bmap);
6505 	return isl_basic_map_fix_pos(bmap,
6506 		isl_basic_map_offset(bmap, type) + pos, value);
6507 }
6508 
6509 /* Fix the value of the variable at position "pos" of type "type" of "bmap"
6510  * to be equal to "v".
6511  */
isl_basic_map_fix_val(__isl_take isl_basic_map * bmap,enum isl_dim_type type,unsigned pos,__isl_take isl_val * v)6512 __isl_give isl_basic_map *isl_basic_map_fix_val(__isl_take isl_basic_map *bmap,
6513 	enum isl_dim_type type, unsigned pos, __isl_take isl_val *v)
6514 {
6515 	if (!bmap || !v)
6516 		goto error;
6517 	if (!isl_val_is_int(v))
6518 		isl_die(isl_basic_map_get_ctx(bmap), isl_error_invalid,
6519 			"expecting integer value", goto error);
6520 	if (isl_basic_map_check_range(bmap, type, pos, 1) < 0)
6521 		goto error;
6522 	pos += isl_basic_map_offset(bmap, type);
6523 	bmap = isl_basic_map_fix_pos(bmap, pos, v->n);
6524 	isl_val_free(v);
6525 	return bmap;
6526 error:
6527 	isl_basic_map_free(bmap);
6528 	isl_val_free(v);
6529 	return NULL;
6530 }
6531 
6532 /* Fix the value of the variable at position "pos" of type "type" of "bset"
6533  * to be equal to "v".
6534  */
isl_basic_set_fix_val(__isl_take isl_basic_set * bset,enum isl_dim_type type,unsigned pos,__isl_take isl_val * v)6535 __isl_give isl_basic_set *isl_basic_set_fix_val(__isl_take isl_basic_set *bset,
6536 	enum isl_dim_type type, unsigned pos, __isl_take isl_val *v)
6537 {
6538 	return isl_basic_map_fix_val(bset, type, pos, v);
6539 }
6540 
isl_basic_set_fix_si(__isl_take isl_basic_set * bset,enum isl_dim_type type,unsigned pos,int value)6541 __isl_give isl_basic_set *isl_basic_set_fix_si(__isl_take isl_basic_set *bset,
6542 		enum isl_dim_type type, unsigned pos, int value)
6543 {
6544 	return bset_from_bmap(isl_basic_map_fix_si(bset_to_bmap(bset),
6545 						    type, pos, value));
6546 }
6547 
isl_basic_set_fix(__isl_take isl_basic_set * bset,enum isl_dim_type type,unsigned pos,isl_int value)6548 __isl_give isl_basic_set *isl_basic_set_fix(__isl_take isl_basic_set *bset,
6549 		enum isl_dim_type type, unsigned pos, isl_int value)
6550 {
6551 	return bset_from_bmap(isl_basic_map_fix(bset_to_bmap(bset),
6552 						    type, pos, value));
6553 }
6554 
6555 /* Remove the basic map at position "i" from "map" if this basic map
6556  * is (obviously) empty.
6557  */
remove_if_empty(__isl_take isl_map * map,int i)6558 static __isl_give isl_map *remove_if_empty(__isl_take isl_map *map, int i)
6559 {
6560 	isl_bool empty;
6561 
6562 	if (!map)
6563 		return NULL;
6564 
6565 	empty = isl_basic_map_plain_is_empty(map->p[i]);
6566 	if (empty < 0)
6567 		return isl_map_free(map);
6568 	if (!empty)
6569 		return map;
6570 
6571 	isl_basic_map_free(map->p[i]);
6572 	map->n--;
6573 	if (i != map->n) {
6574 		map->p[i] = map->p[map->n];
6575 		map = isl_map_unmark_normalized(map);
6576 
6577 	}
6578 
6579 	return map;
6580 }
6581 
6582 /* Perform "fn" on each basic map of "map", where we may not be holding
6583  * the only reference to "map".
6584  * In particular, "fn" should be a semantics preserving operation
6585  * that we want to apply to all copies of "map".  We therefore need
6586  * to be careful not to modify "map" in a way that breaks "map"
6587  * in case anything goes wrong.
6588  */
isl_map_inline_foreach_basic_map(__isl_take isl_map * map,__isl_give isl_basic_map * (* fn)(__isl_take isl_basic_map * bmap))6589 __isl_give isl_map *isl_map_inline_foreach_basic_map(__isl_take isl_map *map,
6590 	__isl_give isl_basic_map *(*fn)(__isl_take isl_basic_map *bmap))
6591 {
6592 	struct isl_basic_map *bmap;
6593 	int i;
6594 
6595 	if (!map)
6596 		return NULL;
6597 
6598 	for (i = map->n - 1; i >= 0; --i) {
6599 		bmap = isl_basic_map_copy(map->p[i]);
6600 		bmap = fn(bmap);
6601 		if (!bmap)
6602 			goto error;
6603 		isl_basic_map_free(map->p[i]);
6604 		map->p[i] = bmap;
6605 		map = remove_if_empty(map, i);
6606 		if (!map)
6607 			return NULL;
6608 	}
6609 
6610 	return map;
6611 error:
6612 	isl_map_free(map);
6613 	return NULL;
6614 }
6615 
isl_map_fix_si(__isl_take isl_map * map,enum isl_dim_type type,unsigned pos,int value)6616 __isl_give isl_map *isl_map_fix_si(__isl_take isl_map *map,
6617 		enum isl_dim_type type, unsigned pos, int value)
6618 {
6619 	int i;
6620 
6621 	map = isl_map_cow(map);
6622 	if (isl_map_check_range(map, type, pos, 1) < 0)
6623 		return isl_map_free(map);
6624 	for (i = map->n - 1; i >= 0; --i) {
6625 		map->p[i] = isl_basic_map_fix_si(map->p[i], type, pos, value);
6626 		map = remove_if_empty(map, i);
6627 		if (!map)
6628 			return NULL;
6629 	}
6630 	map = isl_map_unmark_normalized(map);
6631 	return map;
6632 }
6633 
isl_set_fix_si(__isl_take isl_set * set,enum isl_dim_type type,unsigned pos,int value)6634 __isl_give isl_set *isl_set_fix_si(__isl_take isl_set *set,
6635 		enum isl_dim_type type, unsigned pos, int value)
6636 {
6637 	return set_from_map(isl_map_fix_si(set_to_map(set), type, pos, value));
6638 }
6639 
isl_map_fix(__isl_take isl_map * map,enum isl_dim_type type,unsigned pos,isl_int value)6640 __isl_give isl_map *isl_map_fix(__isl_take isl_map *map,
6641 		enum isl_dim_type type, unsigned pos, isl_int value)
6642 {
6643 	int i;
6644 
6645 	map = isl_map_cow(map);
6646 	if (isl_map_check_range(map, type, pos, 1) < 0)
6647 		return isl_map_free(map);
6648 	for (i = 0; i < map->n; ++i) {
6649 		map->p[i] = isl_basic_map_fix(map->p[i], type, pos, value);
6650 		if (!map->p[i])
6651 			goto error;
6652 	}
6653 	map = isl_map_unmark_normalized(map);
6654 	return map;
6655 error:
6656 	isl_map_free(map);
6657 	return NULL;
6658 }
6659 
isl_set_fix(__isl_take isl_set * set,enum isl_dim_type type,unsigned pos,isl_int value)6660 __isl_give isl_set *isl_set_fix(__isl_take isl_set *set,
6661 		enum isl_dim_type type, unsigned pos, isl_int value)
6662 {
6663 	return set_from_map(isl_map_fix(set_to_map(set), type, pos, value));
6664 }
6665 
6666 /* Fix the value of the variable at position "pos" of type "type" of "map"
6667  * to be equal to "v".
6668  */
isl_map_fix_val(__isl_take isl_map * map,enum isl_dim_type type,unsigned pos,__isl_take isl_val * v)6669 __isl_give isl_map *isl_map_fix_val(__isl_take isl_map *map,
6670 	enum isl_dim_type type, unsigned pos, __isl_take isl_val *v)
6671 {
6672 	int i;
6673 
6674 	map = isl_map_cow(map);
6675 	if (!map || !v)
6676 		goto error;
6677 
6678 	if (!isl_val_is_int(v))
6679 		isl_die(isl_map_get_ctx(map), isl_error_invalid,
6680 			"expecting integer value", goto error);
6681 	if (isl_map_check_range(map, type, pos, 1) < 0)
6682 		goto error;
6683 	for (i = map->n - 1; i >= 0; --i) {
6684 		map->p[i] = isl_basic_map_fix_val(map->p[i], type, pos,
6685 							isl_val_copy(v));
6686 		map = remove_if_empty(map, i);
6687 		if (!map)
6688 			goto error;
6689 	}
6690 	map = isl_map_unmark_normalized(map);
6691 	isl_val_free(v);
6692 	return map;
6693 error:
6694 	isl_map_free(map);
6695 	isl_val_free(v);
6696 	return NULL;
6697 }
6698 
6699 /* Fix the value of the variable at position "pos" of type "type" of "set"
6700  * to be equal to "v".
6701  */
isl_set_fix_val(__isl_take isl_set * set,enum isl_dim_type type,unsigned pos,__isl_take isl_val * v)6702 __isl_give isl_set *isl_set_fix_val(__isl_take isl_set *set,
6703 	enum isl_dim_type type, unsigned pos, __isl_take isl_val *v)
6704 {
6705 	return isl_map_fix_val(set, type, pos, v);
6706 }
6707 
isl_map_fix_input_si(__isl_take isl_map * map,unsigned input,int value)6708 __isl_give isl_map *isl_map_fix_input_si(__isl_take isl_map *map,
6709 		unsigned input, int value)
6710 {
6711 	return isl_map_fix_si(map, isl_dim_in, input, value);
6712 }
6713 
isl_set_fix_dim_si(__isl_take isl_set * set,unsigned dim,int value)6714 __isl_give isl_set *isl_set_fix_dim_si(__isl_take isl_set *set, unsigned dim,
6715 	int value)
6716 {
6717 	return set_from_map(isl_map_fix_si(set_to_map(set),
6718 						isl_dim_set, dim, value));
6719 }
6720 
basic_map_bound_si(__isl_take isl_basic_map * bmap,enum isl_dim_type type,unsigned pos,int value,int upper)6721 static __isl_give isl_basic_map *basic_map_bound_si(
6722 	__isl_take isl_basic_map *bmap,
6723 	enum isl_dim_type type, unsigned pos, int value, int upper)
6724 {
6725 	int j;
6726 	isl_size total;
6727 
6728 	if (isl_basic_map_check_range(bmap, type, pos, 1) < 0)
6729 		return isl_basic_map_free(bmap);
6730 	total = isl_basic_map_dim(bmap, isl_dim_all);
6731 	if (total < 0)
6732 		return isl_basic_map_free(bmap);
6733 	pos += isl_basic_map_offset(bmap, type);
6734 	bmap = isl_basic_map_cow(bmap);
6735 	bmap = isl_basic_map_extend_constraints(bmap, 0, 1);
6736 	j = isl_basic_map_alloc_inequality(bmap);
6737 	if (j < 0)
6738 		goto error;
6739 	isl_seq_clr(bmap->ineq[j], 1 + total);
6740 	if (upper) {
6741 		isl_int_set_si(bmap->ineq[j][pos], -1);
6742 		isl_int_set_si(bmap->ineq[j][0], value);
6743 	} else {
6744 		isl_int_set_si(bmap->ineq[j][pos], 1);
6745 		isl_int_set_si(bmap->ineq[j][0], -value);
6746 	}
6747 	bmap = isl_basic_map_simplify(bmap);
6748 	return isl_basic_map_finalize(bmap);
6749 error:
6750 	isl_basic_map_free(bmap);
6751 	return NULL;
6752 }
6753 
isl_basic_map_lower_bound_si(__isl_take isl_basic_map * bmap,enum isl_dim_type type,unsigned pos,int value)6754 __isl_give isl_basic_map *isl_basic_map_lower_bound_si(
6755 	__isl_take isl_basic_map *bmap,
6756 	enum isl_dim_type type, unsigned pos, int value)
6757 {
6758 	return basic_map_bound_si(bmap, type, pos, value, 0);
6759 }
6760 
6761 /* Constrain the values of the given dimension to be no greater than "value".
6762  */
isl_basic_map_upper_bound_si(__isl_take isl_basic_map * bmap,enum isl_dim_type type,unsigned pos,int value)6763 __isl_give isl_basic_map *isl_basic_map_upper_bound_si(
6764 	__isl_take isl_basic_map *bmap,
6765 	enum isl_dim_type type, unsigned pos, int value)
6766 {
6767 	return basic_map_bound_si(bmap, type, pos, value, 1);
6768 }
6769 
map_bound_si(__isl_take isl_map * map,enum isl_dim_type type,unsigned pos,int value,int upper)6770 static __isl_give isl_map *map_bound_si(__isl_take isl_map *map,
6771 	enum isl_dim_type type, unsigned pos, int value, int upper)
6772 {
6773 	int i;
6774 
6775 	map = isl_map_cow(map);
6776 	if (isl_map_check_range(map, type, pos, 1) < 0)
6777 		return isl_map_free(map);
6778 	for (i = 0; i < map->n; ++i) {
6779 		map->p[i] = basic_map_bound_si(map->p[i],
6780 						 type, pos, value, upper);
6781 		if (!map->p[i])
6782 			goto error;
6783 	}
6784 	map = isl_map_unmark_normalized(map);
6785 	return map;
6786 error:
6787 	isl_map_free(map);
6788 	return NULL;
6789 }
6790 
isl_map_lower_bound_si(__isl_take isl_map * map,enum isl_dim_type type,unsigned pos,int value)6791 __isl_give isl_map *isl_map_lower_bound_si(__isl_take isl_map *map,
6792 	enum isl_dim_type type, unsigned pos, int value)
6793 {
6794 	return map_bound_si(map, type, pos, value, 0);
6795 }
6796 
isl_map_upper_bound_si(__isl_take isl_map * map,enum isl_dim_type type,unsigned pos,int value)6797 __isl_give isl_map *isl_map_upper_bound_si(__isl_take isl_map *map,
6798 	enum isl_dim_type type, unsigned pos, int value)
6799 {
6800 	return map_bound_si(map, type, pos, value, 1);
6801 }
6802 
isl_set_lower_bound_si(__isl_take isl_set * set,enum isl_dim_type type,unsigned pos,int value)6803 __isl_give isl_set *isl_set_lower_bound_si(__isl_take isl_set *set,
6804 		enum isl_dim_type type, unsigned pos, int value)
6805 {
6806 	return set_from_map(isl_map_lower_bound_si(set_to_map(set),
6807 							type, pos, value));
6808 }
6809 
isl_set_upper_bound_si(__isl_take isl_set * set,enum isl_dim_type type,unsigned pos,int value)6810 __isl_give isl_set *isl_set_upper_bound_si(__isl_take isl_set *set,
6811 	enum isl_dim_type type, unsigned pos, int value)
6812 {
6813 	return isl_map_upper_bound_si(set, type, pos, value);
6814 }
6815 
6816 /* Bound the given variable of "bmap" from below (or above is "upper"
6817  * is set) to "value".
6818  */
basic_map_bound(__isl_take isl_basic_map * bmap,enum isl_dim_type type,unsigned pos,isl_int value,int upper)6819 static __isl_give isl_basic_map *basic_map_bound(
6820 	__isl_take isl_basic_map *bmap,
6821 	enum isl_dim_type type, unsigned pos, isl_int value, int upper)
6822 {
6823 	int j;
6824 	isl_size total;
6825 
6826 	if (isl_basic_map_check_range(bmap, type, pos, 1) < 0)
6827 		return isl_basic_map_free(bmap);
6828 	total = isl_basic_map_dim(bmap, isl_dim_all);
6829 	if (total < 0)
6830 		return isl_basic_map_free(bmap);
6831 	pos += isl_basic_map_offset(bmap, type);
6832 	bmap = isl_basic_map_cow(bmap);
6833 	bmap = isl_basic_map_extend_constraints(bmap, 0, 1);
6834 	j = isl_basic_map_alloc_inequality(bmap);
6835 	if (j < 0)
6836 		goto error;
6837 	isl_seq_clr(bmap->ineq[j], 1 + total);
6838 	if (upper) {
6839 		isl_int_set_si(bmap->ineq[j][pos], -1);
6840 		isl_int_set(bmap->ineq[j][0], value);
6841 	} else {
6842 		isl_int_set_si(bmap->ineq[j][pos], 1);
6843 		isl_int_neg(bmap->ineq[j][0], value);
6844 	}
6845 	bmap = isl_basic_map_simplify(bmap);
6846 	return isl_basic_map_finalize(bmap);
6847 error:
6848 	isl_basic_map_free(bmap);
6849 	return NULL;
6850 }
6851 
6852 /* Bound the given variable of "map" from below (or above is "upper"
6853  * is set) to "value".
6854  */
map_bound(__isl_take isl_map * map,enum isl_dim_type type,unsigned pos,isl_int value,int upper)6855 static __isl_give isl_map *map_bound(__isl_take isl_map *map,
6856 	enum isl_dim_type type, unsigned pos, isl_int value, int upper)
6857 {
6858 	int i;
6859 
6860 	map = isl_map_cow(map);
6861 	if (isl_map_check_range(map, type, pos, 1) < 0)
6862 		return isl_map_free(map);
6863 	for (i = map->n - 1; i >= 0; --i) {
6864 		map->p[i] = basic_map_bound(map->p[i], type, pos, value, upper);
6865 		map = remove_if_empty(map, i);
6866 		if (!map)
6867 			return NULL;
6868 	}
6869 	map = isl_map_unmark_normalized(map);
6870 	return map;
6871 }
6872 
isl_map_lower_bound(__isl_take isl_map * map,enum isl_dim_type type,unsigned pos,isl_int value)6873 __isl_give isl_map *isl_map_lower_bound(__isl_take isl_map *map,
6874 	enum isl_dim_type type, unsigned pos, isl_int value)
6875 {
6876 	return map_bound(map, type, pos, value, 0);
6877 }
6878 
isl_map_upper_bound(__isl_take isl_map * map,enum isl_dim_type type,unsigned pos,isl_int value)6879 __isl_give isl_map *isl_map_upper_bound(__isl_take isl_map *map,
6880 	enum isl_dim_type type, unsigned pos, isl_int value)
6881 {
6882 	return map_bound(map, type, pos, value, 1);
6883 }
6884 
isl_set_lower_bound(__isl_take isl_set * set,enum isl_dim_type type,unsigned pos,isl_int value)6885 __isl_give isl_set *isl_set_lower_bound(__isl_take isl_set *set,
6886 	enum isl_dim_type type, unsigned pos, isl_int value)
6887 {
6888 	return isl_map_lower_bound(set, type, pos, value);
6889 }
6890 
isl_set_upper_bound(__isl_take isl_set * set,enum isl_dim_type type,unsigned pos,isl_int value)6891 __isl_give isl_set *isl_set_upper_bound(__isl_take isl_set *set,
6892 	enum isl_dim_type type, unsigned pos, isl_int value)
6893 {
6894 	return isl_map_upper_bound(set, type, pos, value);
6895 }
6896 
6897 /* Force the values of the variable at position "pos" of type "type" of "map"
6898  * to be no smaller than "value".
6899  */
isl_map_lower_bound_val(__isl_take isl_map * map,enum isl_dim_type type,unsigned pos,__isl_take isl_val * value)6900 __isl_give isl_map *isl_map_lower_bound_val(__isl_take isl_map *map,
6901 	enum isl_dim_type type, unsigned pos, __isl_take isl_val *value)
6902 {
6903 	if (!value)
6904 		goto error;
6905 	if (!isl_val_is_int(value))
6906 		isl_die(isl_map_get_ctx(map), isl_error_invalid,
6907 			"expecting integer value", goto error);
6908 	map = isl_map_lower_bound(map, type, pos, value->n);
6909 	isl_val_free(value);
6910 	return map;
6911 error:
6912 	isl_val_free(value);
6913 	isl_map_free(map);
6914 	return NULL;
6915 }
6916 
6917 /* Force the values of the variable at position "pos" of type "type" of "set"
6918  * to be no smaller than "value".
6919  */
isl_set_lower_bound_val(__isl_take isl_set * set,enum isl_dim_type type,unsigned pos,__isl_take isl_val * value)6920 __isl_give isl_set *isl_set_lower_bound_val(__isl_take isl_set *set,
6921 	enum isl_dim_type type, unsigned pos, __isl_take isl_val *value)
6922 {
6923 	isl_map *map;
6924 
6925 	map = set_to_map(set);
6926 	return set_from_map(isl_map_lower_bound_val(map, type, pos, value));
6927 }
6928 
6929 /* Force the values of the variable at position "pos" of type "type" of "map"
6930  * to be no greater than "value".
6931  */
isl_map_upper_bound_val(__isl_take isl_map * map,enum isl_dim_type type,unsigned pos,__isl_take isl_val * value)6932 __isl_give isl_map *isl_map_upper_bound_val(__isl_take isl_map *map,
6933 	enum isl_dim_type type, unsigned pos, __isl_take isl_val *value)
6934 {
6935 	if (!value)
6936 		goto error;
6937 	if (!isl_val_is_int(value))
6938 		isl_die(isl_map_get_ctx(map), isl_error_invalid,
6939 			"expecting integer value", goto error);
6940 	map = isl_map_upper_bound(map, type, pos, value->n);
6941 	isl_val_free(value);
6942 	return map;
6943 error:
6944 	isl_val_free(value);
6945 	isl_map_free(map);
6946 	return NULL;
6947 }
6948 
6949 /* Force the values of the variable at position "pos" of type "type" of "set"
6950  * to be no greater than "value".
6951  */
isl_set_upper_bound_val(__isl_take isl_set * set,enum isl_dim_type type,unsigned pos,__isl_take isl_val * value)6952 __isl_give isl_set *isl_set_upper_bound_val(__isl_take isl_set *set,
6953 	enum isl_dim_type type, unsigned pos, __isl_take isl_val *value)
6954 {
6955 	isl_map *map;
6956 
6957 	map = set_to_map(set);
6958 	return set_from_map(isl_map_upper_bound_val(map, type, pos, value));
6959 }
6960 
6961 /* If "mv" has an explicit domain, then intersect the domain of "map"
6962  * with this explicit domain.
6963  *
6964  * An isl_multi_val object never has an explicit domain,
6965  * so simply return "map".
6966  */
isl_map_intersect_multi_val_explicit_domain(__isl_take isl_map * map,__isl_keep isl_multi_val * mv)6967 static __isl_give isl_map *isl_map_intersect_multi_val_explicit_domain(
6968 	__isl_take isl_map *map, __isl_keep isl_multi_val *mv)
6969 {
6970 	return map;
6971 }
6972 
6973 #undef BASE
6974 #define BASE	val
6975 #include "isl_map_bound_templ.c"
6976 
6977 /* Apply "map_bound" to "set" with the corresponding value in "bound"
6978  * for each set dimension, by treating the set as a map.
6979  */
set_bound_multi_val(__isl_take isl_set * set,__isl_take isl_multi_val * bound,__isl_give isl_map * map_bound (__isl_take isl_map * map,unsigned pos,__isl_take isl_val * value))6980 static __isl_give isl_set *set_bound_multi_val(__isl_take isl_set *set,
6981 	__isl_take isl_multi_val *bound,
6982 	__isl_give isl_map *map_bound(__isl_take isl_map *map,
6983 		unsigned pos, __isl_take isl_val *value))
6984 {
6985 	isl_map *map;
6986 
6987 	map = set_to_map(set);
6988 	return set_from_map(map_bound_multi_val(map, bound, map_bound));
6989 }
6990 
6991 #undef BASE
6992 #define BASE	pw_aff
6993 #include "isl_map_bound_templ.c"
6994 
6995 /* Apply "map_bound" to "set" with the corresponding value in "bound"
6996  * for each set dimension, by converting the set and the bound
6997  * to objects living in a map space.
6998  */
set_bound_multi_pw_aff(__isl_take isl_set * set,__isl_take isl_multi_pw_aff * bound,__isl_give isl_map * set_bound (__isl_take isl_map * map,unsigned pos,__isl_take TYPE * value))6999 static __isl_give isl_set *set_bound_multi_pw_aff(__isl_take isl_set *set,
7000 	__isl_take isl_multi_pw_aff *bound,
7001 	__isl_give isl_map *set_bound(__isl_take isl_map *map,
7002 		unsigned pos, __isl_take TYPE *value))
7003 {
7004 	isl_map *map;
7005 
7006 	map = isl_map_from_range(set);
7007 	bound = isl_multi_pw_aff_from_range(bound);
7008 	map = map_bound_multi_pw_aff(map, bound, set_bound);
7009 	return isl_map_range(map);
7010 }
7011 
7012 /* Wrapper around isl_map_lower_bound_val for use in map_bound_multi_val,
7013  * setting a bound on the given output dimension.
7014  */
map_lower_bound_val(__isl_take isl_map * map,unsigned pos,__isl_take isl_val * v)7015 static __isl_give isl_map *map_lower_bound_val(__isl_take isl_map *map,
7016 	unsigned pos, __isl_take isl_val *v)
7017 {
7018 	return isl_map_lower_bound_val(map, isl_dim_out, pos, v);
7019 }
7020 
7021 /* Force the values of the set dimensions of "set"
7022  * to be no smaller than the corresponding values in "lower".
7023  */
isl_set_lower_bound_multi_val(__isl_take isl_set * set,__isl_take isl_multi_val * lower)7024 __isl_give isl_set *isl_set_lower_bound_multi_val(__isl_take isl_set *set,
7025 	__isl_take isl_multi_val *lower)
7026 {
7027 	return set_bound_multi_val(set, lower, &map_lower_bound_val);
7028 }
7029 
7030 /* Wrapper around isl_map_upper_bound_val for use in map_bound_multi_val,
7031  * setting a bound on the given output dimension.
7032  */
map_upper_bound_val(__isl_take isl_map * map,unsigned pos,__isl_take isl_val * v)7033 static __isl_give isl_map *map_upper_bound_val(__isl_take isl_map *map,
7034 	unsigned pos, __isl_take isl_val *v)
7035 {
7036 	return isl_map_upper_bound_val(map, isl_dim_out, pos, v);
7037 }
7038 
7039 /* Force the values of the set dimensions of "set"
7040  * to be no greater than the corresponding values in "upper".
7041  */
isl_set_upper_bound_multi_val(__isl_take isl_set * set,__isl_take isl_multi_val * upper)7042 __isl_give isl_set *isl_set_upper_bound_multi_val(__isl_take isl_set *set,
7043 	__isl_take isl_multi_val *upper)
7044 {
7045 	return set_bound_multi_val(set, upper, &map_upper_bound_val);
7046 }
7047 
7048 /* Force the symbolic constant expression "bound"
7049  * to satisfy the relation "order" with respect to
7050  * the output variable at position "pos" of "map".
7051  *
7052  * Create an affine expression representing the output variable
7053  * in terms of the range and
7054  * compare it using "order" to "bound" (defined on the domain).
7055  * The result is a relation between elements in domain and range that
7056  * can be intersected with "map".
7057  */
map_bound_pw_aff(__isl_take isl_map * map,unsigned pos,__isl_take isl_pw_aff * bound,__isl_give isl_map * (* order)(__isl_take isl_pw_aff * pa1,__isl_take isl_pw_aff * pa2))7058 static __isl_give isl_map *map_bound_pw_aff(__isl_take isl_map *map,
7059 	unsigned pos, __isl_take isl_pw_aff *bound,
7060 	__isl_give isl_map *(*order)(__isl_take isl_pw_aff *pa1,
7061 		__isl_take isl_pw_aff *pa2))
7062 {
7063 	isl_space *space;
7064 	isl_local_space *ls;
7065 	isl_pw_aff *var;
7066 
7067 	space = isl_space_range(isl_map_get_space(map));
7068 	ls = isl_local_space_from_space(space);
7069 	var = isl_pw_aff_var_on_domain(ls, isl_dim_set, pos);
7070 	map = isl_map_intersect(map, order(bound, var));
7071 	return map;
7072 }
7073 
7074 /* Force the values of the output variable at position "pos" of "map"
7075  * to be no smaller than the symbolic constant expression "lower".
7076  */
map_lower_bound_pw_aff(__isl_take isl_map * map,unsigned pos,__isl_take isl_pw_aff * lower)7077 static __isl_give isl_map *map_lower_bound_pw_aff(__isl_take isl_map *map,
7078 	unsigned pos, __isl_take isl_pw_aff *lower)
7079 {
7080 	return map_bound_pw_aff(map, pos, lower, &isl_pw_aff_le_map);
7081 }
7082 
7083 /* Force the values of the output variable at position "pos" of "map"
7084  * to be no greater than the symbolic constant expression "upper".
7085  */
map_upper_bound_pw_aff(__isl_take isl_map * map,unsigned pos,__isl_take isl_pw_aff * upper)7086 static __isl_give isl_map *map_upper_bound_pw_aff(__isl_take isl_map *map,
7087 	unsigned pos, __isl_take isl_pw_aff *upper)
7088 {
7089 	return map_bound_pw_aff(map, pos, upper, &isl_pw_aff_ge_map);
7090 }
7091 
7092 /* Force the values of the set dimensions of "set"
7093  * to be no smaller than the corresponding constant symbolic expressions
7094  * in "lower".
7095  */
isl_set_lower_bound_multi_pw_aff(__isl_take isl_set * set,__isl_take isl_multi_pw_aff * lower)7096 __isl_give isl_set *isl_set_lower_bound_multi_pw_aff(__isl_take isl_set *set,
7097 	__isl_take isl_multi_pw_aff *lower)
7098 {
7099 	return set_bound_multi_pw_aff(set, lower, &map_lower_bound_pw_aff);
7100 }
7101 
7102 /* Force the values of the set dimensions of "set"
7103  * to be no greater than the corresponding constant symbolic expressions
7104  * in "upper".
7105  */
isl_set_upper_bound_multi_pw_aff(__isl_take isl_set * set,__isl_take isl_multi_pw_aff * upper)7106 __isl_give isl_set *isl_set_upper_bound_multi_pw_aff(__isl_take isl_set *set,
7107 	__isl_take isl_multi_pw_aff *upper)
7108 {
7109 	return set_bound_multi_pw_aff(set, upper, &map_upper_bound_pw_aff);
7110 }
7111 
7112 /* Force the values of the output dimensions of "map"
7113  * to be no smaller than the corresponding constant symbolic expressions
7114  * in "lower".
7115  */
isl_map_lower_bound_multi_pw_aff(__isl_take isl_map * map,__isl_take isl_multi_pw_aff * lower)7116 __isl_give isl_map *isl_map_lower_bound_multi_pw_aff(__isl_take isl_map *map,
7117 	__isl_take isl_multi_pw_aff *lower)
7118 {
7119 	return map_bound_multi_pw_aff(map, lower, &map_lower_bound_pw_aff);
7120 }
7121 
7122 /* Force the values of the output dimensions of "map"
7123  * to be no greater than the corresponding constant symbolic expressions
7124  * in "upper".
7125  */
isl_map_upper_bound_multi_pw_aff(__isl_take isl_map * map,__isl_take isl_multi_pw_aff * upper)7126 __isl_give isl_map *isl_map_upper_bound_multi_pw_aff(__isl_take isl_map *map,
7127 	__isl_take isl_multi_pw_aff *upper)
7128 {
7129 	return map_bound_multi_pw_aff(map, upper, &map_upper_bound_pw_aff);
7130 }
7131 
7132 /* Bound the given variable of "bset" from below (or above is "upper"
7133  * is set) to "value".
7134  */
isl_basic_set_bound(__isl_take isl_basic_set * bset,enum isl_dim_type type,unsigned pos,isl_int value,int upper)7135 static __isl_give isl_basic_set *isl_basic_set_bound(
7136 	__isl_take isl_basic_set *bset, enum isl_dim_type type, unsigned pos,
7137 	isl_int value, int upper)
7138 {
7139 	return bset_from_bmap(basic_map_bound(bset_to_bmap(bset),
7140 						type, pos, value, upper));
7141 }
7142 
7143 /* Bound the given variable of "bset" from below (or above is "upper"
7144  * is set) to "value".
7145  */
isl_basic_set_bound_val(__isl_take isl_basic_set * bset,enum isl_dim_type type,unsigned pos,__isl_take isl_val * value,int upper)7146 static __isl_give isl_basic_set *isl_basic_set_bound_val(
7147 	__isl_take isl_basic_set *bset, enum isl_dim_type type, unsigned pos,
7148 	__isl_take isl_val *value, int upper)
7149 {
7150 	if (!value)
7151 		goto error;
7152 	if (!isl_val_is_int(value))
7153 		isl_die(isl_basic_set_get_ctx(bset), isl_error_invalid,
7154 			"expecting integer value", goto error);
7155 	bset = isl_basic_set_bound(bset, type, pos, value->n, upper);
7156 	isl_val_free(value);
7157 	return bset;
7158 error:
7159 	isl_val_free(value);
7160 	isl_basic_set_free(bset);
7161 	return NULL;
7162 }
7163 
7164 /* Bound the given variable of "bset" from below to "value".
7165  */
isl_basic_set_lower_bound_val(__isl_take isl_basic_set * bset,enum isl_dim_type type,unsigned pos,__isl_take isl_val * value)7166 __isl_give isl_basic_set *isl_basic_set_lower_bound_val(
7167 	__isl_take isl_basic_set *bset, enum isl_dim_type type, unsigned pos,
7168 	__isl_take isl_val *value)
7169 {
7170 	return isl_basic_set_bound_val(bset, type, pos, value, 0);
7171 }
7172 
7173 /* Bound the given variable of "bset" from above to "value".
7174  */
isl_basic_set_upper_bound_val(__isl_take isl_basic_set * bset,enum isl_dim_type type,unsigned pos,__isl_take isl_val * value)7175 __isl_give isl_basic_set *isl_basic_set_upper_bound_val(
7176 	__isl_take isl_basic_set *bset, enum isl_dim_type type, unsigned pos,
7177 	__isl_take isl_val *value)
7178 {
7179 	return isl_basic_set_bound_val(bset, type, pos, value, 1);
7180 }
7181 
isl_map_reverse(__isl_take isl_map * map)7182 __isl_give isl_map *isl_map_reverse(__isl_take isl_map *map)
7183 {
7184 	return isl_map_transform(map, &isl_space_reverse,
7185 					&isl_basic_map_reverse);
7186 }
7187 
7188 /* Given a map A -> (B -> C), return the corresponding map A -> (C -> B).
7189  */
isl_map_range_reverse(__isl_take isl_map * map)7190 __isl_give isl_map *isl_map_range_reverse(__isl_take isl_map *map)
7191 {
7192 	return isl_map_transform(map, &isl_space_range_reverse,
7193 					&isl_basic_map_range_reverse);
7194 }
7195 
7196 #undef TYPE
7197 #define TYPE	isl_pw_multi_aff
7198 #undef SUFFIX
7199 #define SUFFIX	_pw_multi_aff
7200 #undef EMPTY
7201 #define EMPTY	isl_pw_multi_aff_empty
7202 #undef ADD
7203 #define ADD	isl_pw_multi_aff_union_add
7204 #include "isl_map_lexopt_templ.c"
7205 
7206 /* Given a map "map", compute the lexicographically minimal
7207  * (or maximal) image element for each domain element in dom,
7208  * in the form of an isl_pw_multi_aff.
7209  * If "empty" is not NULL, then set *empty to those elements in dom that
7210  * do not have an image element.
7211  * If "flags" includes ISL_OPT_FULL, then "dom" is NULL and the optimum
7212  * should be computed over the domain of "map".  "empty" is also NULL
7213  * in this case.
7214  *
7215  * We first compute the lexicographically minimal or maximal element
7216  * in the first basic map.  This results in a partial solution "res"
7217  * and a subset "todo" of dom that still need to be handled.
7218  * We then consider each of the remaining maps in "map" and successively
7219  * update both "res" and "todo".
7220  * If "empty" is NULL, then the todo sets are not needed and therefore
7221  * also not computed.
7222  */
isl_map_partial_lexopt_aligned_pw_multi_aff(__isl_take isl_map * map,__isl_take isl_set * dom,__isl_give isl_set ** empty,unsigned flags)7223 static __isl_give isl_pw_multi_aff *isl_map_partial_lexopt_aligned_pw_multi_aff(
7224 	__isl_take isl_map *map, __isl_take isl_set *dom,
7225 	__isl_give isl_set **empty, unsigned flags)
7226 {
7227 	int i;
7228 	int full;
7229 	isl_pw_multi_aff *res;
7230 	isl_set *todo;
7231 
7232 	full = ISL_FL_ISSET(flags, ISL_OPT_FULL);
7233 	if (!map || (!full && !dom))
7234 		goto error;
7235 
7236 	if (isl_map_plain_is_empty(map)) {
7237 		if (empty)
7238 			*empty = dom;
7239 		else
7240 			isl_set_free(dom);
7241 		return isl_pw_multi_aff_from_map(map);
7242 	}
7243 
7244 	res = basic_map_partial_lexopt_pw_multi_aff(
7245 					    isl_basic_map_copy(map->p[0]),
7246 					    isl_set_copy(dom), empty, flags);
7247 
7248 	if (empty)
7249 		todo = *empty;
7250 	for (i = 1; i < map->n; ++i) {
7251 		isl_pw_multi_aff *res_i;
7252 
7253 		res_i = basic_map_partial_lexopt_pw_multi_aff(
7254 					    isl_basic_map_copy(map->p[i]),
7255 					    isl_set_copy(dom), empty, flags);
7256 
7257 		if (ISL_FL_ISSET(flags, ISL_OPT_MAX))
7258 			res = isl_pw_multi_aff_union_lexmax(res, res_i);
7259 		else
7260 			res = isl_pw_multi_aff_union_lexmin(res, res_i);
7261 
7262 		if (empty)
7263 			todo = isl_set_intersect(todo, *empty);
7264 	}
7265 
7266 	isl_set_free(dom);
7267 	isl_map_free(map);
7268 
7269 	if (empty)
7270 		*empty = todo;
7271 
7272 	return res;
7273 error:
7274 	if (empty)
7275 		*empty = NULL;
7276 	isl_set_free(dom);
7277 	isl_map_free(map);
7278 	return NULL;
7279 }
7280 
7281 #undef TYPE
7282 #define TYPE	isl_map
7283 #undef SUFFIX
7284 #define SUFFIX
7285 #undef EMPTY
7286 #define EMPTY	isl_map_empty
7287 #undef ADD
7288 #define ADD	isl_map_union_disjoint
7289 #include "isl_map_lexopt_templ.c"
7290 
7291 /* Given a map "map", compute the lexicographically minimal
7292  * (or maximal) image element for each domain element in "dom",
7293  * in the form of an isl_map.
7294  * If "empty" is not NULL, then set *empty to those elements in "dom" that
7295  * do not have an image element.
7296  * If "flags" includes ISL_OPT_FULL, then "dom" is NULL and the optimum
7297  * should be computed over the domain of "map".  "empty" is also NULL
7298  * in this case.
7299  *
7300  * If the input consists of more than one disjunct, then first
7301  * compute the desired result in the form of an isl_pw_multi_aff and
7302  * then convert that into an isl_map.
7303  *
7304  * This function used to have an explicit implementation in terms
7305  * of isl_maps, but it would continually intersect the domains of
7306  * partial results with the complement of the domain of the next
7307  * partial solution, potentially leading to an explosion in the number
7308  * of disjuncts if there are several disjuncts in the input.
7309  * An even earlier implementation of this function would look for
7310  * better results in the domain of the partial result and for extra
7311  * results in the complement of this domain, which would lead to
7312  * even more splintering.
7313  */
isl_map_partial_lexopt_aligned(__isl_take isl_map * map,__isl_take isl_set * dom,__isl_give isl_set ** empty,unsigned flags)7314 static __isl_give isl_map *isl_map_partial_lexopt_aligned(
7315 	__isl_take isl_map *map, __isl_take isl_set *dom,
7316 	__isl_give isl_set **empty, unsigned flags)
7317 {
7318 	int full;
7319 	struct isl_map *res;
7320 	isl_pw_multi_aff *pma;
7321 
7322 	full = ISL_FL_ISSET(flags, ISL_OPT_FULL);
7323 	if (!map || (!full && !dom))
7324 		goto error;
7325 
7326 	if (isl_map_plain_is_empty(map)) {
7327 		if (empty)
7328 			*empty = dom;
7329 		else
7330 			isl_set_free(dom);
7331 		return map;
7332 	}
7333 
7334 	if (map->n == 1) {
7335 		res = basic_map_partial_lexopt(isl_basic_map_copy(map->p[0]),
7336 						dom, empty, flags);
7337 		isl_map_free(map);
7338 		return res;
7339 	}
7340 
7341 	pma = isl_map_partial_lexopt_aligned_pw_multi_aff(map, dom, empty,
7342 							flags);
7343 	return isl_map_from_pw_multi_aff_internal(pma);
7344 error:
7345 	if (empty)
7346 		*empty = NULL;
7347 	isl_set_free(dom);
7348 	isl_map_free(map);
7349 	return NULL;
7350 }
7351 
isl_map_partial_lexmax(__isl_take isl_map * map,__isl_take isl_set * dom,__isl_give isl_set ** empty)7352 __isl_give isl_map *isl_map_partial_lexmax(
7353 		__isl_take isl_map *map, __isl_take isl_set *dom,
7354 		__isl_give isl_set **empty)
7355 {
7356 	return isl_map_partial_lexopt(map, dom, empty, ISL_OPT_MAX);
7357 }
7358 
isl_map_partial_lexmin(__isl_take isl_map * map,__isl_take isl_set * dom,__isl_give isl_set ** empty)7359 __isl_give isl_map *isl_map_partial_lexmin(
7360 		__isl_take isl_map *map, __isl_take isl_set *dom,
7361 		__isl_give isl_set **empty)
7362 {
7363 	return isl_map_partial_lexopt(map, dom, empty, 0);
7364 }
7365 
isl_set_partial_lexmin(__isl_take isl_set * set,__isl_take isl_set * dom,__isl_give isl_set ** empty)7366 __isl_give isl_set *isl_set_partial_lexmin(
7367 		__isl_take isl_set *set, __isl_take isl_set *dom,
7368 		__isl_give isl_set **empty)
7369 {
7370 	return set_from_map(isl_map_partial_lexmin(set_to_map(set),
7371 						    dom, empty));
7372 }
7373 
isl_set_partial_lexmax(__isl_take isl_set * set,__isl_take isl_set * dom,__isl_give isl_set ** empty)7374 __isl_give isl_set *isl_set_partial_lexmax(
7375 		__isl_take isl_set *set, __isl_take isl_set *dom,
7376 		__isl_give isl_set **empty)
7377 {
7378 	return set_from_map(isl_map_partial_lexmax(set_to_map(set),
7379 						    dom, empty));
7380 }
7381 
7382 /* Compute the lexicographic minimum (or maximum if "flags" includes
7383  * ISL_OPT_MAX) of "bset" over its parametric domain.
7384  */
isl_basic_set_lexopt(__isl_take isl_basic_set * bset,unsigned flags)7385 __isl_give isl_set *isl_basic_set_lexopt(__isl_take isl_basic_set *bset,
7386 	unsigned flags)
7387 {
7388 	return isl_basic_map_lexopt(bset, flags);
7389 }
7390 
isl_basic_map_lexmax(__isl_take isl_basic_map * bmap)7391 __isl_give isl_map *isl_basic_map_lexmax(__isl_take isl_basic_map *bmap)
7392 {
7393 	return isl_basic_map_lexopt(bmap, ISL_OPT_MAX);
7394 }
7395 
isl_basic_set_lexmin(__isl_take isl_basic_set * bset)7396 __isl_give isl_set *isl_basic_set_lexmin(__isl_take isl_basic_set *bset)
7397 {
7398 	return set_from_map(isl_basic_map_lexmin(bset_to_bmap(bset)));
7399 }
7400 
isl_basic_set_lexmax(__isl_take isl_basic_set * bset)7401 __isl_give isl_set *isl_basic_set_lexmax(__isl_take isl_basic_set *bset)
7402 {
7403 	return set_from_map(isl_basic_map_lexmax(bset_to_bmap(bset)));
7404 }
7405 
7406 /* Compute the lexicographic minimum of "bset" over its parametric domain
7407  * for the purpose of quantifier elimination.
7408  * That is, find an explicit representation for all the existentially
7409  * quantified variables in "bset" by computing their lexicographic
7410  * minimum.
7411  */
isl_basic_set_lexmin_compute_divs(__isl_take isl_basic_set * bset)7412 static __isl_give isl_set *isl_basic_set_lexmin_compute_divs(
7413 	__isl_take isl_basic_set *bset)
7414 {
7415 	return isl_basic_set_lexopt(bset, ISL_OPT_QE);
7416 }
7417 
7418 /* Given a basic map with one output dimension, compute the minimum or
7419  * maximum of that dimension as an isl_pw_aff.
7420  *
7421  * Compute the optimum as a lexicographic optimum over the single
7422  * output dimension and extract the single isl_pw_aff from the result.
7423  */
basic_map_dim_opt(__isl_keep isl_basic_map * bmap,int max)7424 static __isl_give isl_pw_aff *basic_map_dim_opt(__isl_keep isl_basic_map *bmap,
7425 	int max)
7426 {
7427 	isl_pw_multi_aff *pma;
7428 	isl_pw_aff *pwaff;
7429 
7430 	bmap = isl_basic_map_copy(bmap);
7431 	pma = isl_basic_map_lexopt_pw_multi_aff(bmap, max ? ISL_OPT_MAX : 0);
7432 	pwaff = isl_pw_multi_aff_get_pw_aff(pma, 0);
7433 	isl_pw_multi_aff_free(pma);
7434 
7435 	return pwaff;
7436 }
7437 
7438 /* Compute the minimum or maximum of the given output dimension
7439  * as a function of the parameters and the input dimensions,
7440  * but independently of the other output dimensions.
7441  *
7442  * We first project out the other output dimension and then compute
7443  * the "lexicographic" maximum in each basic map, combining the results
7444  * using isl_pw_aff_union_max.
7445  */
map_dim_opt(__isl_take isl_map * map,int pos,int max)7446 static __isl_give isl_pw_aff *map_dim_opt(__isl_take isl_map *map, int pos,
7447 	int max)
7448 {
7449 	int i;
7450 	isl_pw_aff *pwaff;
7451 	isl_size n_out;
7452 
7453 	n_out = isl_map_dim(map, isl_dim_out);
7454 	if (n_out < 0)
7455 		map = isl_map_free(map);
7456 	map = isl_map_project_out(map, isl_dim_out, pos + 1, n_out - (pos + 1));
7457 	map = isl_map_project_out(map, isl_dim_out, 0, pos);
7458 	if (!map)
7459 		return NULL;
7460 
7461 	if (map->n == 0) {
7462 		isl_space *space = isl_map_get_space(map);
7463 		isl_map_free(map);
7464 		return isl_pw_aff_empty(space);
7465 	}
7466 
7467 	pwaff = basic_map_dim_opt(map->p[0], max);
7468 	for (i = 1; i < map->n; ++i) {
7469 		isl_pw_aff *pwaff_i;
7470 
7471 		pwaff_i = basic_map_dim_opt(map->p[i], max);
7472 		pwaff = isl_pw_aff_union_opt(pwaff, pwaff_i, max);
7473 	}
7474 
7475 	isl_map_free(map);
7476 
7477 	return pwaff;
7478 }
7479 
7480 /* Compute the minimum of the given output dimension as a function of the
7481  * parameters and input dimensions, but independently of
7482  * the other output dimensions.
7483  */
isl_map_dim_min(__isl_take isl_map * map,int pos)7484 __isl_give isl_pw_aff *isl_map_dim_min(__isl_take isl_map *map, int pos)
7485 {
7486 	return map_dim_opt(map, pos, 0);
7487 }
7488 
7489 /* Compute the maximum of the given output dimension as a function of the
7490  * parameters and input dimensions, but independently of
7491  * the other output dimensions.
7492  */
isl_map_dim_max(__isl_take isl_map * map,int pos)7493 __isl_give isl_pw_aff *isl_map_dim_max(__isl_take isl_map *map, int pos)
7494 {
7495 	return map_dim_opt(map, pos, 1);
7496 }
7497 
7498 /* Compute the minimum or maximum of the given set dimension
7499  * as a function of the parameters,
7500  * but independently of the other set dimensions.
7501  */
set_dim_opt(__isl_take isl_set * set,int pos,int max)7502 static __isl_give isl_pw_aff *set_dim_opt(__isl_take isl_set *set, int pos,
7503 	int max)
7504 {
7505 	return map_dim_opt(set, pos, max);
7506 }
7507 
7508 /* Compute the maximum of the given set dimension as a function of the
7509  * parameters, but independently of the other set dimensions.
7510  */
isl_set_dim_max(__isl_take isl_set * set,int pos)7511 __isl_give isl_pw_aff *isl_set_dim_max(__isl_take isl_set *set, int pos)
7512 {
7513 	return set_dim_opt(set, pos, 1);
7514 }
7515 
7516 /* Compute the minimum of the given set dimension as a function of the
7517  * parameters, but independently of the other set dimensions.
7518  */
isl_set_dim_min(__isl_take isl_set * set,int pos)7519 __isl_give isl_pw_aff *isl_set_dim_min(__isl_take isl_set *set, int pos)
7520 {
7521 	return set_dim_opt(set, pos, 0);
7522 }
7523 
7524 /* Apply a preimage specified by "mat" on the parameters of "bset".
7525  * bset is assumed to have only parameters and divs.
7526  */
basic_set_parameter_preimage(__isl_take isl_basic_set * bset,__isl_take isl_mat * mat)7527 static __isl_give isl_basic_set *basic_set_parameter_preimage(
7528 	__isl_take isl_basic_set *bset, __isl_take isl_mat *mat)
7529 {
7530 	isl_size nparam;
7531 
7532 	nparam = isl_basic_set_dim(bset, isl_dim_param);
7533 	if (nparam < 0 || !mat)
7534 		goto error;
7535 
7536 	bset->dim = isl_space_cow(bset->dim);
7537 	if (!bset->dim)
7538 		goto error;
7539 
7540 	isl_assert(bset->ctx, mat->n_row == 1 + nparam, goto error);
7541 
7542 	bset->dim->nparam = 0;
7543 	bset->dim->n_out = nparam;
7544 	bset = isl_basic_set_preimage(bset, mat);
7545 	if (bset) {
7546 		bset->dim->nparam = bset->dim->n_out;
7547 		bset->dim->n_out = 0;
7548 	}
7549 	return bset;
7550 error:
7551 	isl_mat_free(mat);
7552 	isl_basic_set_free(bset);
7553 	return NULL;
7554 }
7555 
7556 /* Apply a preimage specified by "mat" on the parameters of "set".
7557  * set is assumed to have only parameters and divs.
7558  */
set_parameter_preimage(__isl_take isl_set * set,__isl_take isl_mat * mat)7559 static __isl_give isl_set *set_parameter_preimage(__isl_take isl_set *set,
7560 	__isl_take isl_mat *mat)
7561 {
7562 	isl_space *space;
7563 	isl_size nparam;
7564 
7565 	nparam = isl_set_dim(set, isl_dim_param);
7566 	if (nparam < 0 || !mat)
7567 		goto error;
7568 
7569 	if (mat->n_row != 1 + nparam)
7570 		isl_die(isl_set_get_ctx(set), isl_error_internal,
7571 			"unexpected number of rows", goto error);
7572 
7573 	space = isl_set_get_space(set);
7574 	space = isl_space_move_dims(space, isl_dim_set, 0,
7575 				    isl_dim_param, 0, nparam);
7576 	set = isl_set_reset_space(set, space);
7577 	set = isl_set_preimage(set, mat);
7578 	nparam = isl_set_dim(set, isl_dim_out);
7579 	if (nparam < 0)
7580 		set = isl_set_free(set);
7581 	space = isl_set_get_space(set);
7582 	space = isl_space_move_dims(space, isl_dim_param, 0,
7583 				    isl_dim_out, 0, nparam);
7584 	set = isl_set_reset_space(set, space);
7585 	return set;
7586 error:
7587 	isl_mat_free(mat);
7588 	isl_set_free(set);
7589 	return NULL;
7590 }
7591 
7592 /* Intersect the basic set "bset" with the affine space specified by the
7593  * equalities in "eq".
7594  */
basic_set_append_equalities(__isl_take isl_basic_set * bset,__isl_take isl_mat * eq)7595 static __isl_give isl_basic_set *basic_set_append_equalities(
7596 	__isl_take isl_basic_set *bset, __isl_take isl_mat *eq)
7597 {
7598 	int i, k;
7599 	unsigned len;
7600 
7601 	if (!bset || !eq)
7602 		goto error;
7603 
7604 	bset = isl_basic_set_extend(bset, 0, eq->n_row, 0);
7605 	if (!bset)
7606 		goto error;
7607 
7608 	len = isl_basic_set_offset(bset, isl_dim_div) + bset->extra;
7609 	for (i = 0; i < eq->n_row; ++i) {
7610 		k = isl_basic_set_alloc_equality(bset);
7611 		if (k < 0)
7612 			goto error;
7613 		isl_seq_cpy(bset->eq[k], eq->row[i], eq->n_col);
7614 		isl_seq_clr(bset->eq[k] + eq->n_col, len - eq->n_col);
7615 	}
7616 	isl_mat_free(eq);
7617 
7618 	bset = isl_basic_set_gauss(bset, NULL);
7619 	bset = isl_basic_set_finalize(bset);
7620 
7621 	return bset;
7622 error:
7623 	isl_mat_free(eq);
7624 	isl_basic_set_free(bset);
7625 	return NULL;
7626 }
7627 
7628 /* Intersect the set "set" with the affine space specified by the
7629  * equalities in "eq".
7630  */
set_append_equalities(__isl_take isl_set * set,__isl_take isl_mat * eq)7631 static __isl_give isl_set *set_append_equalities(__isl_take isl_set *set,
7632 	__isl_take isl_mat *eq)
7633 {
7634 	int i;
7635 
7636 	if (!set || !eq)
7637 		goto error;
7638 
7639 	for (i = 0; i < set->n; ++i) {
7640 		set->p[i] = basic_set_append_equalities(set->p[i],
7641 					isl_mat_copy(eq));
7642 		if (!set->p[i])
7643 			goto error;
7644 	}
7645 	isl_mat_free(eq);
7646 	return set;
7647 error:
7648 	isl_mat_free(eq);
7649 	isl_set_free(set);
7650 	return NULL;
7651 }
7652 
7653 /* Given a basic set "bset" that only involves parameters and existentially
7654  * quantified variables, return the index of the first equality
7655  * that only involves parameters.  If there is no such equality then
7656  * return bset->n_eq.
7657  *
7658  * This function assumes that isl_basic_set_gauss has been called on "bset".
7659  */
first_parameter_equality(__isl_keep isl_basic_set * bset)7660 static int first_parameter_equality(__isl_keep isl_basic_set *bset)
7661 {
7662 	int i, j;
7663 	isl_size nparam, n_div;
7664 
7665 	nparam = isl_basic_set_dim(bset, isl_dim_param);
7666 	n_div = isl_basic_set_dim(bset, isl_dim_div);
7667 	if (nparam < 0 || n_div < 0)
7668 		return -1;
7669 
7670 	for (i = 0, j = n_div - 1; i < bset->n_eq && j >= 0; --j) {
7671 		if (!isl_int_is_zero(bset->eq[i][1 + nparam + j]))
7672 			++i;
7673 	}
7674 
7675 	return i;
7676 }
7677 
7678 /* Compute an explicit representation for the existentially quantified
7679  * variables in "bset" by computing the "minimal value" of the set
7680  * variables.  Since there are no set variables, the computation of
7681  * the minimal value essentially computes an explicit representation
7682  * of the non-empty part(s) of "bset".
7683  *
7684  * The input only involves parameters and existentially quantified variables.
7685  * All equalities among parameters have been removed.
7686  *
7687  * Since the existentially quantified variables in the result are in general
7688  * going to be different from those in the input, we first replace
7689  * them by the minimal number of variables based on their equalities.
7690  * This should simplify the parametric integer programming.
7691  */
base_compute_divs(__isl_take isl_basic_set * bset)7692 static __isl_give isl_set *base_compute_divs(__isl_take isl_basic_set *bset)
7693 {
7694 	isl_morph *morph1, *morph2;
7695 	isl_set *set;
7696 	isl_size n;
7697 
7698 	if (!bset)
7699 		return NULL;
7700 	if (bset->n_eq == 0)
7701 		return isl_basic_set_lexmin_compute_divs(bset);
7702 
7703 	morph1 = isl_basic_set_parameter_compression(bset);
7704 	bset = isl_morph_basic_set(isl_morph_copy(morph1), bset);
7705 	bset = isl_basic_set_lift(bset);
7706 	morph2 = isl_basic_set_variable_compression(bset, isl_dim_set);
7707 	bset = isl_morph_basic_set(morph2, bset);
7708 	n = isl_basic_set_dim(bset, isl_dim_set);
7709 	if (n < 0)
7710 		bset = isl_basic_set_free(bset);
7711 	bset = isl_basic_set_project_out(bset, isl_dim_set, 0, n);
7712 
7713 	set = isl_basic_set_lexmin_compute_divs(bset);
7714 
7715 	set = isl_morph_set(isl_morph_inverse(morph1), set);
7716 
7717 	return set;
7718 }
7719 
7720 /* Project the given basic set onto its parameter domain, possibly introducing
7721  * new, explicit, existential variables in the constraints.
7722  * The input has parameters and (possibly implicit) existential variables.
7723  * The output has the same parameters, but only
7724  * explicit existentially quantified variables.
7725  *
7726  * The actual projection is performed by pip, but pip doesn't seem
7727  * to like equalities very much, so we first remove the equalities
7728  * among the parameters by performing a variable compression on
7729  * the parameters.  Afterward, an inverse transformation is performed
7730  * and the equalities among the parameters are inserted back in.
7731  *
7732  * The variable compression on the parameters may uncover additional
7733  * equalities that were only implicit before.  We therefore check
7734  * if there are any new parameter equalities in the result and
7735  * if so recurse.  The removal of parameter equalities is required
7736  * for the parameter compression performed by base_compute_divs.
7737  */
parameter_compute_divs(__isl_take isl_basic_set * bset)7738 static __isl_give isl_set *parameter_compute_divs(
7739 	__isl_take isl_basic_set *bset)
7740 {
7741 	int i;
7742 	struct isl_mat *eq;
7743 	struct isl_mat *T, *T2;
7744 	struct isl_set *set;
7745 	isl_size nparam;
7746 
7747 	bset = isl_basic_set_cow(bset);
7748 	if (!bset)
7749 		return NULL;
7750 
7751 	if (bset->n_eq == 0)
7752 		return base_compute_divs(bset);
7753 
7754 	bset = isl_basic_set_gauss(bset, NULL);
7755 	if (!bset)
7756 		return NULL;
7757 	if (isl_basic_set_plain_is_empty(bset))
7758 		return isl_set_from_basic_set(bset);
7759 
7760 	i = first_parameter_equality(bset);
7761 	if (i == bset->n_eq)
7762 		return base_compute_divs(bset);
7763 
7764 	nparam = isl_basic_set_dim(bset, isl_dim_param);
7765 	if (nparam < 0)
7766 		return isl_set_from_basic_set(isl_basic_set_free(bset));
7767 	eq = isl_mat_sub_alloc6(bset->ctx, bset->eq, i, bset->n_eq - i,
7768 		0, 1 + nparam);
7769 	eq = isl_mat_cow(eq);
7770 	T = isl_mat_variable_compression(isl_mat_copy(eq), &T2);
7771 	if (T && T->n_col == 0) {
7772 		isl_mat_free(T);
7773 		isl_mat_free(T2);
7774 		isl_mat_free(eq);
7775 		bset = isl_basic_set_set_to_empty(bset);
7776 		return isl_set_from_basic_set(bset);
7777 	}
7778 	bset = basic_set_parameter_preimage(bset, T);
7779 
7780 	i = first_parameter_equality(bset);
7781 	if (!bset)
7782 		set = NULL;
7783 	else if (i == bset->n_eq)
7784 		set = base_compute_divs(bset);
7785 	else
7786 		set = parameter_compute_divs(bset);
7787 	set = set_parameter_preimage(set, T2);
7788 	set = set_append_equalities(set, eq);
7789 	return set;
7790 }
7791 
7792 /* Insert the divs from "ls" before those of "bmap".
7793  *
7794  * The number of columns is not changed, which means that the last
7795  * dimensions of "bmap" are being reintepreted as the divs from "ls".
7796  * The caller is responsible for removing the same number of dimensions
7797  * from the space of "bmap".
7798  */
insert_divs_from_local_space(__isl_take isl_basic_map * bmap,__isl_keep isl_local_space * ls)7799 static __isl_give isl_basic_map *insert_divs_from_local_space(
7800 	__isl_take isl_basic_map *bmap, __isl_keep isl_local_space *ls)
7801 {
7802 	int i;
7803 	isl_size n_div;
7804 	int old_n_div;
7805 
7806 	n_div = isl_local_space_dim(ls, isl_dim_div);
7807 	if (n_div < 0)
7808 		return isl_basic_map_free(bmap);
7809 	if (n_div == 0)
7810 		return bmap;
7811 
7812 	old_n_div = bmap->n_div;
7813 	bmap = insert_div_rows(bmap, n_div);
7814 	if (!bmap)
7815 		return NULL;
7816 
7817 	for (i = 0; i < n_div; ++i) {
7818 		isl_seq_cpy(bmap->div[i], ls->div->row[i], ls->div->n_col);
7819 		isl_seq_clr(bmap->div[i] + ls->div->n_col, old_n_div);
7820 	}
7821 
7822 	return bmap;
7823 }
7824 
7825 /* Replace the space of "bmap" by the space and divs of "ls".
7826  *
7827  * If "ls" has any divs, then we simplify the result since we may
7828  * have discovered some additional equalities that could simplify
7829  * the div expressions.
7830  */
basic_replace_space_by_local_space(__isl_take isl_basic_map * bmap,__isl_take isl_local_space * ls)7831 static __isl_give isl_basic_map *basic_replace_space_by_local_space(
7832 	__isl_take isl_basic_map *bmap, __isl_take isl_local_space *ls)
7833 {
7834 	isl_size n_div;
7835 
7836 	bmap = isl_basic_map_cow(bmap);
7837 	n_div = isl_local_space_dim(ls, isl_dim_div);
7838 	if (!bmap || n_div < 0)
7839 		goto error;
7840 
7841 	bmap = insert_divs_from_local_space(bmap, ls);
7842 	if (!bmap)
7843 		goto error;
7844 
7845 	isl_space_free(bmap->dim);
7846 	bmap->dim = isl_local_space_get_space(ls);
7847 	if (!bmap->dim)
7848 		goto error;
7849 
7850 	isl_local_space_free(ls);
7851 	if (n_div > 0)
7852 		bmap = isl_basic_map_simplify(bmap);
7853 	bmap = isl_basic_map_finalize(bmap);
7854 	return bmap;
7855 error:
7856 	isl_basic_map_free(bmap);
7857 	isl_local_space_free(ls);
7858 	return NULL;
7859 }
7860 
7861 /* Replace the space of "map" by the space and divs of "ls".
7862  */
replace_space_by_local_space(__isl_take isl_map * map,__isl_take isl_local_space * ls)7863 static __isl_give isl_map *replace_space_by_local_space(__isl_take isl_map *map,
7864 	__isl_take isl_local_space *ls)
7865 {
7866 	int i;
7867 
7868 	map = isl_map_cow(map);
7869 	if (!map || !ls)
7870 		goto error;
7871 
7872 	for (i = 0; i < map->n; ++i) {
7873 		map->p[i] = basic_replace_space_by_local_space(map->p[i],
7874 						    isl_local_space_copy(ls));
7875 		if (!map->p[i])
7876 			goto error;
7877 	}
7878 	isl_space_free(isl_map_take_space(map));
7879 	map = isl_map_restore_space(map, isl_local_space_get_space(ls));
7880 
7881 	isl_local_space_free(ls);
7882 	return map;
7883 error:
7884 	isl_local_space_free(ls);
7885 	isl_map_free(map);
7886 	return NULL;
7887 }
7888 
7889 /* Compute an explicit representation for the existentially
7890  * quantified variables for which do not know any explicit representation yet.
7891  *
7892  * We first sort the existentially quantified variables so that the
7893  * existentially quantified variables for which we already have an explicit
7894  * representation are placed before those for which we do not.
7895  * The input dimensions, the output dimensions and the existentially
7896  * quantified variables for which we already have an explicit
7897  * representation are then turned into parameters.
7898  * compute_divs returns a map with the same parameters and
7899  * no input or output dimensions and the dimension specification
7900  * is reset to that of the input, including the existentially quantified
7901  * variables for which we already had an explicit representation.
7902  */
compute_divs(__isl_take isl_basic_map * bmap)7903 static __isl_give isl_map *compute_divs(__isl_take isl_basic_map *bmap)
7904 {
7905 	struct isl_basic_set *bset;
7906 	struct isl_set *set;
7907 	struct isl_map *map;
7908 	isl_space *space;
7909 	isl_local_space *ls;
7910 	isl_size nparam;
7911 	isl_size n_in;
7912 	isl_size n_out;
7913 	int n_known;
7914 	int i;
7915 
7916 	bmap = isl_basic_map_sort_divs(bmap);
7917 	bmap = isl_basic_map_cow(bmap);
7918 	if (!bmap)
7919 		return NULL;
7920 
7921 	n_known = isl_basic_map_first_unknown_div(bmap);
7922 	nparam = isl_basic_map_dim(bmap, isl_dim_param);
7923 	n_in = isl_basic_map_dim(bmap, isl_dim_in);
7924 	n_out = isl_basic_map_dim(bmap, isl_dim_out);
7925 	if (n_known < 0 || nparam < 0 || n_in < 0 || n_out < 0)
7926 		return isl_map_from_basic_map(isl_basic_map_free(bmap));
7927 
7928 	space = isl_space_set_alloc(bmap->ctx,
7929 				    nparam + n_in + n_out + n_known, 0);
7930 	if (!space)
7931 		goto error;
7932 
7933 	ls = isl_basic_map_get_local_space(bmap);
7934 	ls = isl_local_space_drop_dims(ls, isl_dim_div,
7935 					n_known, bmap->n_div - n_known);
7936 	if (n_known > 0) {
7937 		for (i = n_known; i < bmap->n_div; ++i)
7938 			swap_div(bmap, i - n_known, i);
7939 		bmap->n_div -= n_known;
7940 		bmap->extra -= n_known;
7941 	}
7942 	bmap = isl_basic_map_reset_space(bmap, space);
7943 	bset = bset_from_bmap(bmap);
7944 
7945 	set = parameter_compute_divs(bset);
7946 	map = set_to_map(set);
7947 	map = replace_space_by_local_space(map, ls);
7948 
7949 	return map;
7950 error:
7951 	isl_basic_map_free(bmap);
7952 	return NULL;
7953 }
7954 
7955 /* Remove the explicit representation of local variable "div",
7956  * if there is any.
7957  */
isl_basic_map_mark_div_unknown(__isl_take isl_basic_map * bmap,int div)7958 __isl_give isl_basic_map *isl_basic_map_mark_div_unknown(
7959 	__isl_take isl_basic_map *bmap, int div)
7960 {
7961 	isl_bool unknown;
7962 
7963 	unknown = isl_basic_map_div_is_marked_unknown(bmap, div);
7964 	if (unknown < 0)
7965 		return isl_basic_map_free(bmap);
7966 	if (unknown)
7967 		return bmap;
7968 
7969 	bmap = isl_basic_map_cow(bmap);
7970 	if (!bmap)
7971 		return NULL;
7972 	isl_int_set_si(bmap->div[div][0], 0);
7973 	return bmap;
7974 }
7975 
7976 /* Is local variable "div" of "bmap" marked as not having an explicit
7977  * representation?
7978  * Note that even if "div" is not marked in this way and therefore
7979  * has an explicit representation, this representation may still
7980  * depend (indirectly) on other local variables that do not
7981  * have an explicit representation.
7982  */
isl_basic_map_div_is_marked_unknown(__isl_keep isl_basic_map * bmap,int div)7983 isl_bool isl_basic_map_div_is_marked_unknown(__isl_keep isl_basic_map *bmap,
7984 	int div)
7985 {
7986 	if (isl_basic_map_check_range(bmap, isl_dim_div, div, 1) < 0)
7987 		return isl_bool_error;
7988 	return isl_int_is_zero(bmap->div[div][0]);
7989 }
7990 
7991 /* Return the position of the first local variable that does not
7992  * have an explicit representation.
7993  * Return the total number of local variables if they all have
7994  * an explicit representation.
7995  * Return -1 on error.
7996  */
isl_basic_map_first_unknown_div(__isl_keep isl_basic_map * bmap)7997 int isl_basic_map_first_unknown_div(__isl_keep isl_basic_map *bmap)
7998 {
7999 	int i;
8000 
8001 	if (!bmap)
8002 		return -1;
8003 
8004 	for (i = 0; i < bmap->n_div; ++i) {
8005 		if (!isl_basic_map_div_is_known(bmap, i))
8006 			return i;
8007 	}
8008 	return bmap->n_div;
8009 }
8010 
8011 /* Return the position of the first local variable that does not
8012  * have an explicit representation.
8013  * Return the total number of local variables if they all have
8014  * an explicit representation.
8015  * Return -1 on error.
8016  */
isl_basic_set_first_unknown_div(__isl_keep isl_basic_set * bset)8017 int isl_basic_set_first_unknown_div(__isl_keep isl_basic_set *bset)
8018 {
8019 	return isl_basic_map_first_unknown_div(bset);
8020 }
8021 
8022 /* Does "bmap" have an explicit representation for all local variables?
8023  */
isl_basic_map_divs_known(__isl_keep isl_basic_map * bmap)8024 isl_bool isl_basic_map_divs_known(__isl_keep isl_basic_map *bmap)
8025 {
8026 	int first;
8027 	isl_size n;
8028 
8029 	n = isl_basic_map_dim(bmap, isl_dim_div);
8030 	first = isl_basic_map_first_unknown_div(bmap);
8031 	if (n < 0 || first < 0)
8032 		return isl_bool_error;
8033 	return first == n;
8034 }
8035 
8036 /* Do all basic maps in "map" have an explicit representation
8037  * for all local variables?
8038  */
isl_map_divs_known(__isl_keep isl_map * map)8039 isl_bool isl_map_divs_known(__isl_keep isl_map *map)
8040 {
8041 	int i;
8042 
8043 	if (!map)
8044 		return isl_bool_error;
8045 
8046 	for (i = 0; i < map->n; ++i) {
8047 		int known = isl_basic_map_divs_known(map->p[i]);
8048 		if (known <= 0)
8049 			return known;
8050 	}
8051 
8052 	return isl_bool_true;
8053 }
8054 
8055 /* If bmap contains any unknown divs, then compute explicit
8056  * expressions for them.  However, this computation may be
8057  * quite expensive, so first try to remove divs that aren't
8058  * strictly needed.
8059  */
isl_basic_map_compute_divs(__isl_take isl_basic_map * bmap)8060 __isl_give isl_map *isl_basic_map_compute_divs(__isl_take isl_basic_map *bmap)
8061 {
8062 	int known;
8063 	struct isl_map *map;
8064 
8065 	known = isl_basic_map_divs_known(bmap);
8066 	if (known < 0)
8067 		goto error;
8068 	if (known)
8069 		return isl_map_from_basic_map(bmap);
8070 
8071 	bmap = isl_basic_map_drop_redundant_divs(bmap);
8072 
8073 	known = isl_basic_map_divs_known(bmap);
8074 	if (known < 0)
8075 		goto error;
8076 	if (known)
8077 		return isl_map_from_basic_map(bmap);
8078 
8079 	map = compute_divs(bmap);
8080 	return map;
8081 error:
8082 	isl_basic_map_free(bmap);
8083 	return NULL;
8084 }
8085 
isl_map_compute_divs(__isl_take isl_map * map)8086 __isl_give isl_map *isl_map_compute_divs(__isl_take isl_map *map)
8087 {
8088 	int i;
8089 	int known;
8090 	struct isl_map *res;
8091 
8092 	if (!map)
8093 		return NULL;
8094 	if (map->n == 0)
8095 		return map;
8096 
8097 	known = isl_map_divs_known(map);
8098 	if (known < 0) {
8099 		isl_map_free(map);
8100 		return NULL;
8101 	}
8102 	if (known)
8103 		return map;
8104 
8105 	res = isl_basic_map_compute_divs(isl_basic_map_copy(map->p[0]));
8106 	for (i = 1 ; i < map->n; ++i) {
8107 		struct isl_map *r2;
8108 		r2 = isl_basic_map_compute_divs(isl_basic_map_copy(map->p[i]));
8109 		if (ISL_F_ISSET(map, ISL_MAP_DISJOINT))
8110 			res = isl_map_union_disjoint(res, r2);
8111 		else
8112 			res = isl_map_union(res, r2);
8113 	}
8114 	isl_map_free(map);
8115 
8116 	return res;
8117 }
8118 
isl_basic_set_compute_divs(__isl_take isl_basic_set * bset)8119 __isl_give isl_set *isl_basic_set_compute_divs(__isl_take isl_basic_set *bset)
8120 {
8121 	return set_from_map(isl_basic_map_compute_divs(bset_to_bmap(bset)));
8122 }
8123 
isl_set_compute_divs(__isl_take isl_set * set)8124 __isl_give isl_set *isl_set_compute_divs(__isl_take isl_set *set)
8125 {
8126 	return set_from_map(isl_map_compute_divs(set_to_map(set)));
8127 }
8128 
isl_map_domain(__isl_take isl_map * map)8129 __isl_give isl_set *isl_map_domain(__isl_take isl_map *map)
8130 {
8131 	isl_space *space;
8132 	isl_size n_out;
8133 
8134 	n_out = isl_map_dim(map, isl_dim_out);
8135 	if (n_out < 0)
8136 		return set_from_map(isl_map_free(map));
8137 	space = isl_space_domain(isl_map_get_space(map));
8138 
8139 	map = isl_map_project_out(map, isl_dim_out, 0, n_out);
8140 
8141 	return set_from_map(isl_map_reset_space(map, space));
8142 }
8143 
8144 /* Return the union of "map1" and "map2", where we assume for now that
8145  * "map1" and "map2" are disjoint.  Note that the basic maps inside
8146  * "map1" or "map2" may not be disjoint from each other.
8147  * Also note that this function is also called from isl_map_union,
8148  * which takes care of handling the situation where "map1" and "map2"
8149  * may not be disjoint.
8150  *
8151  * If one of the inputs is empty, we can simply return the other input.
8152  * Similarly, if one of the inputs is universal, then it is equal to the union.
8153  */
map_union_disjoint(__isl_take isl_map * map1,__isl_take isl_map * map2)8154 static __isl_give isl_map *map_union_disjoint(__isl_take isl_map *map1,
8155 	__isl_take isl_map *map2)
8156 {
8157 	int i;
8158 	unsigned flags = 0;
8159 	struct isl_map *map = NULL;
8160 	int is_universe;
8161 
8162 	if (isl_map_check_equal_space(map1, map2) < 0)
8163 		goto error;
8164 
8165 	if (map1->n == 0) {
8166 		isl_map_free(map1);
8167 		return map2;
8168 	}
8169 	if (map2->n == 0) {
8170 		isl_map_free(map2);
8171 		return map1;
8172 	}
8173 
8174 	is_universe = isl_map_plain_is_universe(map1);
8175 	if (is_universe < 0)
8176 		goto error;
8177 	if (is_universe) {
8178 		isl_map_free(map2);
8179 		return map1;
8180 	}
8181 
8182 	is_universe = isl_map_plain_is_universe(map2);
8183 	if (is_universe < 0)
8184 		goto error;
8185 	if (is_universe) {
8186 		isl_map_free(map1);
8187 		return map2;
8188 	}
8189 
8190 	if (ISL_F_ISSET(map1, ISL_MAP_DISJOINT) &&
8191 	    ISL_F_ISSET(map2, ISL_MAP_DISJOINT))
8192 		ISL_FL_SET(flags, ISL_MAP_DISJOINT);
8193 
8194 	map = isl_map_alloc_space(isl_space_copy(map1->dim),
8195 				map1->n + map2->n, flags);
8196 	if (!map)
8197 		goto error;
8198 	for (i = 0; i < map1->n; ++i) {
8199 		map = isl_map_add_basic_map(map,
8200 				  isl_basic_map_copy(map1->p[i]));
8201 		if (!map)
8202 			goto error;
8203 	}
8204 	for (i = 0; i < map2->n; ++i) {
8205 		map = isl_map_add_basic_map(map,
8206 				  isl_basic_map_copy(map2->p[i]));
8207 		if (!map)
8208 			goto error;
8209 	}
8210 	isl_map_free(map1);
8211 	isl_map_free(map2);
8212 	return map;
8213 error:
8214 	isl_map_free(map);
8215 	isl_map_free(map1);
8216 	isl_map_free(map2);
8217 	return NULL;
8218 }
8219 
8220 /* Return the union of "map1" and "map2", where "map1" and "map2" are
8221  * guaranteed to be disjoint by the caller.
8222  *
8223  * Note that this functions is called from within isl_map_make_disjoint,
8224  * so we have to be careful not to touch the constraints of the inputs
8225  * in any way.
8226  */
isl_map_union_disjoint(__isl_take isl_map * map1,__isl_take isl_map * map2)8227 __isl_give isl_map *isl_map_union_disjoint(__isl_take isl_map *map1,
8228 	__isl_take isl_map *map2)
8229 {
8230 	isl_map_align_params_bin(&map1, &map2);
8231 	return map_union_disjoint(map1, map2);
8232 }
8233 
8234 /* Return the union of "map1" and "map2", where "map1" and "map2" may
8235  * not be disjoint.
8236  *
8237  * We currently simply call map_union_disjoint, the internal operation
8238  * of which does not really depend on the inputs being disjoint.
8239  * If the result contains more than one basic map, then we clear
8240  * the disjoint flag since the result may contain basic maps from
8241  * both inputs and these are not guaranteed to be disjoint.
8242  *
8243  * As a special case, if "map1" and "map2" are obviously equal,
8244  * then we simply return "map1".
8245  */
isl_map_union(__isl_take isl_map * map1,__isl_take isl_map * map2)8246 __isl_give isl_map *isl_map_union(__isl_take isl_map *map1,
8247 	__isl_take isl_map *map2)
8248 {
8249 	int equal;
8250 
8251 	if (isl_map_align_params_bin(&map1, &map2) < 0)
8252 		goto error;
8253 
8254 	equal = isl_map_plain_is_equal(map1, map2);
8255 	if (equal < 0)
8256 		goto error;
8257 	if (equal) {
8258 		isl_map_free(map2);
8259 		return map1;
8260 	}
8261 
8262 	map1 = map_union_disjoint(map1, map2);
8263 	if (!map1)
8264 		return NULL;
8265 	if (map1->n > 1)
8266 		ISL_F_CLR(map1, ISL_MAP_DISJOINT);
8267 	return map1;
8268 error:
8269 	isl_map_free(map1);
8270 	isl_map_free(map2);
8271 	return NULL;
8272 }
8273 
isl_set_union_disjoint(__isl_take isl_set * set1,__isl_take isl_set * set2)8274 __isl_give isl_set *isl_set_union_disjoint(
8275 	__isl_take isl_set *set1, __isl_take isl_set *set2)
8276 {
8277 	return set_from_map(isl_map_union_disjoint(set_to_map(set1),
8278 						    set_to_map(set2)));
8279 }
8280 
isl_set_union(__isl_take isl_set * set1,__isl_take isl_set * set2)8281 __isl_give isl_set *isl_set_union(__isl_take isl_set *set1,
8282 	__isl_take isl_set *set2)
8283 {
8284 	return set_from_map(isl_map_union(set_to_map(set1), set_to_map(set2)));
8285 }
8286 
8287 /* Apply "fn" to pairs of elements from "map" and "set" and collect
8288  * the results in a map living in "space".
8289  *
8290  * "map" and "set" are assumed to be compatible and non-NULL.
8291  */
map_intersect_set(__isl_take isl_map * map,__isl_take isl_space * space,__isl_take isl_set * set,__isl_give isl_basic_map * fn (__isl_take isl_basic_map * bmap,__isl_take isl_basic_set * bset))8292 static __isl_give isl_map *map_intersect_set(__isl_take isl_map *map,
8293 	__isl_take isl_space *space, __isl_take isl_set *set,
8294 	__isl_give isl_basic_map *fn(__isl_take isl_basic_map *bmap,
8295 		__isl_take isl_basic_set *bset))
8296 {
8297 	unsigned flags = 0;
8298 	struct isl_map *result;
8299 	int i, j;
8300 
8301 	if (isl_set_plain_is_universe(set)) {
8302 		isl_set_free(set);
8303 		return isl_map_reset_equal_dim_space(map, space);
8304 	}
8305 
8306 	if (ISL_F_ISSET(map, ISL_MAP_DISJOINT) &&
8307 	    ISL_F_ISSET(set, ISL_MAP_DISJOINT))
8308 		ISL_FL_SET(flags, ISL_MAP_DISJOINT);
8309 
8310 	result = isl_map_alloc_space(space, map->n * set->n, flags);
8311 	for (i = 0; result && i < map->n; ++i)
8312 		for (j = 0; j < set->n; ++j) {
8313 			result = isl_map_add_basic_map(result,
8314 					fn(isl_basic_map_copy(map->p[i]),
8315 					    isl_basic_set_copy(set->p[j])));
8316 			if (!result)
8317 				break;
8318 		}
8319 
8320 	isl_map_free(map);
8321 	isl_set_free(set);
8322 	return result;
8323 }
8324 
isl_map_intersect_range(__isl_take isl_map * map,__isl_take isl_set * set)8325 __isl_give isl_map *isl_map_intersect_range(__isl_take isl_map *map,
8326 	__isl_take isl_set *set)
8327 {
8328 	isl_bool ok;
8329 	isl_space *space;
8330 
8331 	isl_map_align_params_set(&map, &set);
8332 	ok = isl_map_compatible_range(map, set);
8333 	if (ok < 0)
8334 		goto error;
8335 	if (!ok)
8336 		isl_die(set->ctx, isl_error_invalid,
8337 			"incompatible spaces", goto error);
8338 
8339 	space = isl_map_get_space(map);
8340 	return map_intersect_set(map, space, set,
8341 				&isl_basic_map_intersect_range);
8342 error:
8343 	isl_map_free(map);
8344 	isl_set_free(set);
8345 	return NULL;
8346 }
8347 
8348 /* Intersect the domain of "map" with "set".
8349  *
8350  * If the domain dimensions of "map" do not have any identifiers,
8351  * then copy them over from "set".
8352  */
isl_map_intersect_domain(__isl_take isl_map * map,__isl_take isl_set * set)8353 __isl_give isl_map *isl_map_intersect_domain(__isl_take isl_map *map,
8354 	__isl_take isl_set *set)
8355 {
8356 	isl_bool ok;
8357 	isl_space *space;
8358 
8359 	isl_map_align_params_set(&map, &set);
8360 	ok = isl_map_compatible_domain(map, set);
8361 	if (ok < 0)
8362 		goto error;
8363 	if (!ok)
8364 		isl_die(set->ctx, isl_error_invalid,
8365 			"incompatible spaces", goto error);
8366 
8367 	space = isl_map_get_space(map);
8368 	space = isl_space_copy_ids_if_unset(space, isl_dim_in,
8369 					isl_set_peek_space(set), isl_dim_set);
8370 	return map_intersect_set(map, space, set,
8371 				&isl_basic_map_intersect_domain);
8372 error:
8373 	isl_map_free(map);
8374 	isl_set_free(set);
8375 	return NULL;
8376 }
8377 
8378 #undef TYPE
8379 #define TYPE isl_map
8380 static
8381 #include "isl_copy_tuple_id_templ.c"
8382 
8383 /* Data structure that specifies how isl_map_intersect_factor
8384  * should operate.
8385  *
8386  * "preserve_type" is the tuple where the factor differs from
8387  * the input map and of which the identifiers needs
8388  * to be preserved explicitly.
8389  * "other_factor" is used to extract the space of the other factor
8390  * from the space of the product ("map").
8391  * "product" is used to combine the given factor and a universe map
8392  * in the space returned by "other_factor" to produce a map
8393  * that lives in the same space as the input map.
8394  */
8395 struct isl_intersect_factor_control {
8396 	enum isl_dim_type preserve_type;
8397 	__isl_give isl_space *(*other_factor)(__isl_take isl_space *space);
8398 	__isl_give isl_map *(*product)(__isl_take isl_map *factor,
8399 		__isl_take isl_map *other);
8400 };
8401 
8402 /* Given a map "map" in some product space and a map "factor"
8403  * living in some factor space, return the intersection.
8404  *
8405  * After aligning the parameters,
8406  * the map "factor" is first extended to a map living in the same space
8407  * as "map" and then a regular intersection is computed.
8408  *
8409  * Note that the extension is computed as a product, which is anonymous
8410  * by default.  If "map" has an identifier on the corresponding tuple,
8411  * then this identifier needs to be set on the product
8412  * before the intersection is computed.
8413  */
isl_map_intersect_factor(__isl_take isl_map * map,__isl_take isl_map * factor,struct isl_intersect_factor_control * control)8414 static __isl_give isl_map *isl_map_intersect_factor(
8415 	__isl_take isl_map *map, __isl_take isl_map *factor,
8416 	struct isl_intersect_factor_control *control)
8417 {
8418 	isl_bool equal;
8419 	isl_space *space;
8420 	isl_map *other, *product;
8421 
8422 	equal = isl_map_has_equal_params(map, factor);
8423 	if (equal < 0)
8424 		goto error;
8425 	if (!equal) {
8426 		map = isl_map_align_params(map, isl_map_get_space(factor));
8427 		factor = isl_map_align_params(factor, isl_map_get_space(map));
8428 	}
8429 
8430 	space = isl_map_get_space(map);
8431 	other = isl_map_universe(control->other_factor(space));
8432 	product = control->product(factor, other);
8433 
8434 	space = isl_map_peek_space(map);
8435 	product = isl_map_copy_tuple_id(product, control->preserve_type,
8436 					space, control->preserve_type);
8437 	return map_intersect(map, product);
8438 error:
8439 	isl_map_free(map);
8440 	isl_map_free(factor);
8441 	return NULL;
8442 }
8443 
8444 /* Return the domain product of "map2" and "map1".
8445  */
isl_map_reverse_domain_product(__isl_take isl_map * map1,__isl_take isl_map * map2)8446 static __isl_give isl_map *isl_map_reverse_domain_product(
8447 	__isl_take isl_map *map1, __isl_take isl_map *map2)
8448 {
8449 	return isl_map_domain_product(map2, map1);
8450 }
8451 
8452 /* Return the range product of "map2" and "map1".
8453  */
isl_map_reverse_range_product(__isl_take isl_map * map1,__isl_take isl_map * map2)8454 static __isl_give isl_map *isl_map_reverse_range_product(
8455 	__isl_take isl_map *map1, __isl_take isl_map *map2)
8456 {
8457 	return isl_map_range_product(map2, map1);
8458 }
8459 
8460 /* Given a map "map" in a space [A -> B] -> C and a map "factor"
8461  * in the space A -> C, return the intersection.
8462  */
isl_map_intersect_domain_factor_domain(__isl_take isl_map * map,__isl_take isl_map * factor)8463 __isl_give isl_map *isl_map_intersect_domain_factor_domain(
8464 	__isl_take isl_map *map, __isl_take isl_map *factor)
8465 {
8466 	struct isl_intersect_factor_control control = {
8467 		.preserve_type = isl_dim_in,
8468 		.other_factor = isl_space_domain_factor_range,
8469 		.product = isl_map_domain_product,
8470 	};
8471 
8472 	return isl_map_intersect_factor(map, factor, &control);
8473 }
8474 
8475 /* Given a map "map" in a space [A -> B] -> C and a map "factor"
8476  * in the space B -> C, return the intersection.
8477  */
isl_map_intersect_domain_factor_range(__isl_take isl_map * map,__isl_take isl_map * factor)8478 __isl_give isl_map *isl_map_intersect_domain_factor_range(
8479 	__isl_take isl_map *map, __isl_take isl_map *factor)
8480 {
8481 	struct isl_intersect_factor_control control = {
8482 		.preserve_type = isl_dim_in,
8483 		.other_factor = isl_space_domain_factor_domain,
8484 		.product = isl_map_reverse_domain_product,
8485 	};
8486 
8487 	return isl_map_intersect_factor(map, factor, &control);
8488 }
8489 
8490 /* Given a map "map" in a space A -> [B -> C] and a map "factor"
8491  * in the space A -> B, return the intersection.
8492  */
isl_map_intersect_range_factor_domain(__isl_take isl_map * map,__isl_take isl_map * factor)8493 __isl_give isl_map *isl_map_intersect_range_factor_domain(
8494 	__isl_take isl_map *map, __isl_take isl_map *factor)
8495 {
8496 	struct isl_intersect_factor_control control = {
8497 		.preserve_type = isl_dim_out,
8498 		.other_factor = isl_space_range_factor_range,
8499 		.product = isl_map_range_product,
8500 	};
8501 
8502 	return isl_map_intersect_factor(map, factor, &control);
8503 }
8504 
8505 /* Given a map "map" in a space A -> [B -> C] and a map "factor"
8506  * in the space A -> C, return the intersection.
8507  */
isl_map_intersect_range_factor_range(__isl_take isl_map * map,__isl_take isl_map * factor)8508 __isl_give isl_map *isl_map_intersect_range_factor_range(
8509 	__isl_take isl_map *map, __isl_take isl_map *factor)
8510 {
8511 	struct isl_intersect_factor_control control = {
8512 		.preserve_type = isl_dim_out,
8513 		.other_factor = isl_space_range_factor_domain,
8514 		.product = isl_map_reverse_range_product,
8515 	};
8516 
8517 	return isl_map_intersect_factor(map, factor, &control);
8518 }
8519 
8520 /* Given a set "set" in a space [A -> B] and a set "domain"
8521  * in the space A, return the intersection.
8522  *
8523  * The set "domain" is first extended to a set living in the space
8524  * [A -> B] and then a regular intersection is computed.
8525  */
isl_set_intersect_factor_domain(__isl_take isl_set * set,__isl_take isl_set * domain)8526 __isl_give isl_set *isl_set_intersect_factor_domain(__isl_take isl_set *set,
8527 	__isl_take isl_set *domain)
8528 {
8529 	struct isl_intersect_factor_control control = {
8530 		.preserve_type = isl_dim_set,
8531 		.other_factor = isl_space_factor_range,
8532 		.product = isl_map_range_product,
8533 	};
8534 
8535 	return set_from_map(isl_map_intersect_factor(set_to_map(set),
8536 						set_to_map(domain), &control));
8537 }
8538 
8539 /* Given a set "set" in a space [A -> B] and a set "range"
8540  * in the space B, return the intersection.
8541  *
8542  * The set "range" is first extended to a set living in the space
8543  * [A -> B] and then a regular intersection is computed.
8544  */
isl_set_intersect_factor_range(__isl_take isl_set * set,__isl_take isl_set * range)8545 __isl_give isl_set *isl_set_intersect_factor_range(__isl_take isl_set *set,
8546 	__isl_take isl_set *range)
8547 {
8548 	struct isl_intersect_factor_control control = {
8549 		.preserve_type = isl_dim_set,
8550 		.other_factor = isl_space_factor_domain,
8551 		.product = isl_map_reverse_range_product,
8552 	};
8553 
8554 	return set_from_map(isl_map_intersect_factor(set_to_map(set),
8555 						set_to_map(range), &control));
8556 }
8557 
8558 /* Given a map "map" in a space [A -> B] -> C and a set "domain"
8559  * in the space A, return the intersection.
8560  *
8561  * The set "domain" is extended to a set living in the space [A -> B] and
8562  * the domain of "map" is intersected with this set.
8563  */
isl_map_intersect_domain_wrapped_domain(__isl_take isl_map * map,__isl_take isl_set * domain)8564 __isl_give isl_map *isl_map_intersect_domain_wrapped_domain(
8565 	__isl_take isl_map *map, __isl_take isl_set *domain)
8566 {
8567 	isl_space *space;
8568 	isl_set *factor;
8569 
8570 	isl_map_align_params_set(&map, &domain);
8571 	space = isl_map_get_space(map);
8572 	space = isl_space_domain_wrapped_range(space);
8573 	factor = isl_set_universe(space);
8574 	domain = isl_set_product(domain, factor);
8575 	return isl_map_intersect_domain(map, domain);
8576 }
8577 
8578 /* Given a map "map" in a space A -> [B -> C] and a set "domain"
8579  * in the space B, return the intersection.
8580  *
8581  * The set "domain" is extended to a set living in the space [B -> C] and
8582  * the range of "map" is intersected with this set.
8583  */
isl_map_intersect_range_wrapped_domain(__isl_take isl_map * map,__isl_take isl_set * domain)8584 __isl_give isl_map *isl_map_intersect_range_wrapped_domain(
8585 	__isl_take isl_map *map, __isl_take isl_set *domain)
8586 {
8587 	isl_space *space;
8588 	isl_set *factor;
8589 
8590 	isl_map_align_params_set(&map, &domain);
8591 	space = isl_map_get_space(map);
8592 	space = isl_space_range_wrapped_range(space);
8593 	factor = isl_set_universe(space);
8594 	domain = isl_set_product(domain, factor);
8595 	return isl_map_intersect_range(map, domain);
8596 }
8597 
isl_map_apply_domain(__isl_take isl_map * map1,__isl_take isl_map * map2)8598 __isl_give isl_map *isl_map_apply_domain(__isl_take isl_map *map1,
8599 	__isl_take isl_map *map2)
8600 {
8601 	if (isl_map_align_params_bin(&map1, &map2) < 0)
8602 		goto error;
8603 	map1 = isl_map_reverse(map1);
8604 	map1 = isl_map_apply_range(map1, map2);
8605 	return isl_map_reverse(map1);
8606 error:
8607 	isl_map_free(map1);
8608 	isl_map_free(map2);
8609 	return NULL;
8610 }
8611 
isl_map_apply_range(__isl_take isl_map * map1,__isl_take isl_map * map2)8612 __isl_give isl_map *isl_map_apply_range(__isl_take isl_map *map1,
8613 	__isl_take isl_map *map2)
8614 {
8615 	isl_space *space;
8616 	struct isl_map *result;
8617 	int i, j;
8618 
8619 	if (isl_map_align_params_bin(&map1, &map2) < 0)
8620 		goto error;
8621 
8622 	space = isl_space_join(isl_space_copy(map1->dim),
8623 				  isl_space_copy(map2->dim));
8624 
8625 	result = isl_map_alloc_space(space, map1->n * map2->n, 0);
8626 	if (!result)
8627 		goto error;
8628 	for (i = 0; i < map1->n; ++i)
8629 		for (j = 0; j < map2->n; ++j) {
8630 			result = isl_map_add_basic_map(result,
8631 			    isl_basic_map_apply_range(
8632 				isl_basic_map_copy(map1->p[i]),
8633 				isl_basic_map_copy(map2->p[j])));
8634 			if (!result)
8635 				goto error;
8636 		}
8637 	isl_map_free(map1);
8638 	isl_map_free(map2);
8639 	if (result && result->n <= 1)
8640 		ISL_F_SET(result, ISL_MAP_DISJOINT);
8641 	return result;
8642 error:
8643 	isl_map_free(map1);
8644 	isl_map_free(map2);
8645 	return NULL;
8646 }
8647 
8648 /* Is "bmap" a transformation, i.e.,
8649  * does it relate elements from the same space.
8650  */
isl_basic_map_is_transformation(__isl_keep isl_basic_map * bmap)8651 isl_bool isl_basic_map_is_transformation(__isl_keep isl_basic_map *bmap)
8652 {
8653 	isl_space *space;
8654 
8655 	space = isl_basic_map_peek_space(bmap);
8656 	return isl_space_tuple_is_equal(space, isl_dim_in, space, isl_dim_out);
8657 }
8658 
8659 /* Check that "bmap" is a transformation, i.e.,
8660  * that it relates elements from the same space.
8661  */
isl_basic_map_check_transformation(__isl_keep isl_basic_map * bmap)8662 static isl_stat isl_basic_map_check_transformation(
8663 	__isl_keep isl_basic_map *bmap)
8664 {
8665 	isl_bool equal;
8666 
8667 	equal = isl_basic_map_is_transformation(bmap);
8668 	if (equal < 0)
8669 		return isl_stat_error;
8670 	if (!equal)
8671 		isl_die(isl_basic_map_get_ctx(bmap), isl_error_invalid,
8672 			"domain and range don't match", return isl_stat_error);
8673 	return isl_stat_ok;
8674 }
8675 
8676 /*
8677  * returns range - domain
8678  */
isl_basic_map_deltas(__isl_take isl_basic_map * bmap)8679 __isl_give isl_basic_set *isl_basic_map_deltas(__isl_take isl_basic_map *bmap)
8680 {
8681 	isl_space *target_space;
8682 	struct isl_basic_set *bset;
8683 	isl_size dim;
8684 	isl_size nparam;
8685 	isl_size total;
8686 	int i;
8687 
8688 	if (isl_basic_map_check_transformation(bmap) < 0)
8689 		return isl_basic_map_free(bmap);
8690 	dim = isl_basic_map_dim(bmap, isl_dim_in);
8691 	nparam = isl_basic_map_dim(bmap, isl_dim_param);
8692 	if (dim < 0 || nparam < 0)
8693 		goto error;
8694 	target_space = isl_space_domain(isl_basic_map_get_space(bmap));
8695 	bmap = isl_basic_map_from_range(isl_basic_map_wrap(bmap));
8696 	bmap = isl_basic_map_add_dims(bmap, isl_dim_in, dim);
8697 	total = isl_basic_map_dim(bmap, isl_dim_all);
8698 	if (total < 0)
8699 		bmap = isl_basic_map_free(bmap);
8700 	bmap = isl_basic_map_extend_constraints(bmap, dim, 0);
8701 	for (i = 0; i < dim; ++i) {
8702 		int j = isl_basic_map_alloc_equality(bmap);
8703 		if (j < 0) {
8704 			bmap = isl_basic_map_free(bmap);
8705 			break;
8706 		}
8707 		isl_seq_clr(bmap->eq[j], 1 + total);
8708 		isl_int_set_si(bmap->eq[j][1+nparam+i], 1);
8709 		isl_int_set_si(bmap->eq[j][1+nparam+dim+i], 1);
8710 		isl_int_set_si(bmap->eq[j][1+nparam+2*dim+i], -1);
8711 	}
8712 	bset = isl_basic_map_domain(bmap);
8713 	bset = isl_basic_set_reset_space(bset, target_space);
8714 	return bset;
8715 error:
8716 	isl_basic_map_free(bmap);
8717 	return NULL;
8718 }
8719 
8720 /* Is the tuple of type "type1" of "map" the same as
8721  * the tuple of type "type2" of "space"?
8722  */
isl_map_space_tuple_is_equal(__isl_keep isl_map * map,enum isl_dim_type type1,__isl_keep isl_space * space,enum isl_dim_type type2)8723 isl_bool isl_map_space_tuple_is_equal(__isl_keep isl_map *map,
8724 	enum isl_dim_type type1, __isl_keep isl_space *space,
8725 	enum isl_dim_type type2)
8726 {
8727 	isl_space *map_space;
8728 
8729 	map_space = isl_map_peek_space(map);
8730 	return isl_space_tuple_is_equal(map_space, type1, space, type2);
8731 }
8732 
8733 /* Is the tuple of type "type1" of "map1" the same as
8734  * the tuple of type "type2" of "map2"?
8735  */
isl_map_tuple_is_equal(__isl_keep isl_map * map1,enum isl_dim_type type1,__isl_keep isl_map * map2,enum isl_dim_type type2)8736 isl_bool isl_map_tuple_is_equal(__isl_keep isl_map *map1,
8737 	enum isl_dim_type type1, __isl_keep isl_map *map2,
8738 	enum isl_dim_type type2)
8739 {
8740 	isl_space *space1, *space2;
8741 
8742 	space1 = isl_map_peek_space(map1);
8743 	space2 = isl_map_peek_space(map2);
8744 	return isl_space_tuple_is_equal(space1, type1, space2, type2);
8745 }
8746 
8747 /* Is the space of "obj" equal to "space", ignoring parameters?
8748  */
isl_map_has_space_tuples(__isl_keep isl_map * map,__isl_keep isl_space * space)8749 isl_bool isl_map_has_space_tuples(__isl_keep isl_map *map,
8750 	__isl_keep isl_space *space)
8751 {
8752 	isl_space *map_space;
8753 
8754 	map_space = isl_map_peek_space(map);
8755 	return isl_space_has_equal_tuples(map_space, space);
8756 }
8757 
8758 /* Check that "map" is a transformation, i.e.,
8759  * that it relates elements from the same space.
8760  */
isl_map_check_transformation(__isl_keep isl_map * map)8761 isl_stat isl_map_check_transformation(__isl_keep isl_map *map)
8762 {
8763 	isl_bool equal;
8764 
8765 	equal = isl_map_tuple_is_equal(map, isl_dim_in, map, isl_dim_out);
8766 	if (equal < 0)
8767 		return isl_stat_error;
8768 	if (!equal)
8769 		isl_die(isl_map_get_ctx(map), isl_error_invalid,
8770 			"domain and range don't match", return isl_stat_error);
8771 	return isl_stat_ok;
8772 }
8773 
8774 /*
8775  * returns range - domain
8776  */
isl_map_deltas(__isl_take isl_map * map)8777 __isl_give isl_set *isl_map_deltas(__isl_take isl_map *map)
8778 {
8779 	int i;
8780 	isl_space *space;
8781 	struct isl_set *result;
8782 
8783 	if (isl_map_check_transformation(map) < 0)
8784 		goto error;
8785 	space = isl_map_get_space(map);
8786 	space = isl_space_domain(space);
8787 	result = isl_set_alloc_space(space, map->n, 0);
8788 	if (!result)
8789 		goto error;
8790 	for (i = 0; i < map->n; ++i)
8791 		result = isl_set_add_basic_set(result,
8792 			  isl_basic_map_deltas(isl_basic_map_copy(map->p[i])));
8793 	isl_map_free(map);
8794 	return result;
8795 error:
8796 	isl_map_free(map);
8797 	return NULL;
8798 }
8799 
8800 /*
8801  * returns [domain -> range] -> range - domain
8802  */
isl_basic_map_deltas_map(__isl_take isl_basic_map * bmap)8803 __isl_give isl_basic_map *isl_basic_map_deltas_map(
8804 	__isl_take isl_basic_map *bmap)
8805 {
8806 	int i, k;
8807 	isl_space *space;
8808 	isl_basic_map *domain;
8809 	isl_size nparam, n;
8810 	isl_size total;
8811 
8812 	if (isl_basic_map_check_transformation(bmap) < 0)
8813 		return isl_basic_map_free(bmap);
8814 
8815 	nparam = isl_basic_map_dim(bmap, isl_dim_param);
8816 	n = isl_basic_map_dim(bmap, isl_dim_in);
8817 	if (nparam < 0 || n < 0)
8818 		return isl_basic_map_free(bmap);
8819 
8820 	space = isl_basic_map_get_space(bmap);
8821 	space = isl_space_from_range(isl_space_domain(space));
8822 	domain = isl_basic_map_universe(space);
8823 
8824 	bmap = isl_basic_map_from_domain(isl_basic_map_wrap(bmap));
8825 	bmap = isl_basic_map_apply_range(bmap, domain);
8826 	bmap = isl_basic_map_extend_constraints(bmap, n, 0);
8827 
8828 	total = isl_basic_map_dim(bmap, isl_dim_all);
8829 	if (total < 0)
8830 		return isl_basic_map_free(bmap);
8831 
8832 	for (i = 0; i < n; ++i) {
8833 		k = isl_basic_map_alloc_equality(bmap);
8834 		if (k < 0)
8835 			goto error;
8836 		isl_seq_clr(bmap->eq[k], 1 + total);
8837 		isl_int_set_si(bmap->eq[k][1 + nparam + i], 1);
8838 		isl_int_set_si(bmap->eq[k][1 + nparam + n + i], -1);
8839 		isl_int_set_si(bmap->eq[k][1 + nparam + n + n + i], 1);
8840 	}
8841 
8842 	bmap = isl_basic_map_gauss(bmap, NULL);
8843 	return isl_basic_map_finalize(bmap);
8844 error:
8845 	isl_basic_map_free(bmap);
8846 	return NULL;
8847 }
8848 
8849 /*
8850  * returns [domain -> range] -> range - domain
8851  */
isl_map_deltas_map(__isl_take isl_map * map)8852 __isl_give isl_map *isl_map_deltas_map(__isl_take isl_map *map)
8853 {
8854 	if (isl_map_check_transformation(map) < 0)
8855 		return isl_map_free(map);
8856 
8857 	return isl_map_transform(map, &isl_space_range_map,
8858 					&isl_basic_map_deltas_map);
8859 }
8860 
8861 /* Return pairs of elements { x -> y } such that y - x is in "deltas".
8862  */
isl_set_translation(__isl_take isl_set * deltas)8863 __isl_give isl_map *isl_set_translation(__isl_take isl_set *deltas)
8864 {
8865 	isl_space *space;
8866 	isl_map *map;
8867 
8868 	space = isl_space_map_from_set(isl_set_get_space(deltas));
8869 	map = isl_map_deltas_map(isl_map_universe(space));
8870 	map = isl_map_intersect_range(map, deltas);
8871 
8872 	return isl_set_unwrap(isl_map_domain(map));
8873 }
8874 
isl_basic_map_identity(__isl_take isl_space * space)8875 __isl_give isl_basic_map *isl_basic_map_identity(__isl_take isl_space *space)
8876 {
8877 	isl_size n_in, n_out;
8878 
8879 	n_in = isl_space_dim(space, isl_dim_in);
8880 	n_out = isl_space_dim(space, isl_dim_out);
8881 	if (n_in < 0 || n_out < 0)
8882 		goto error;
8883 	if (n_in != n_out)
8884 		isl_die(space->ctx, isl_error_invalid,
8885 			"number of input and output dimensions needs to be "
8886 			"the same", goto error);
8887 	return isl_basic_map_equal(space, n_in);
8888 error:
8889 	isl_space_free(space);
8890 	return NULL;
8891 }
8892 
isl_map_identity(__isl_take isl_space * space)8893 __isl_give isl_map *isl_map_identity(__isl_take isl_space *space)
8894 {
8895 	return isl_map_from_basic_map(isl_basic_map_identity(space));
8896 }
8897 
isl_set_identity(__isl_take isl_set * set)8898 __isl_give isl_map *isl_set_identity(__isl_take isl_set *set)
8899 {
8900 	isl_space *space = isl_set_get_space(set);
8901 	isl_map *id;
8902 	id = isl_map_identity(isl_space_map_from_set(space));
8903 	return isl_map_intersect_range(id, set);
8904 }
8905 
8906 /* Construct a basic set with all set dimensions having only non-negative
8907  * values.
8908  */
isl_basic_set_positive_orthant(__isl_take isl_space * space)8909 __isl_give isl_basic_set *isl_basic_set_positive_orthant(
8910 	__isl_take isl_space *space)
8911 {
8912 	int i;
8913 	isl_size nparam;
8914 	isl_size dim;
8915 	isl_size total;
8916 	struct isl_basic_set *bset;
8917 
8918 	nparam = isl_space_dim(space, isl_dim_param);
8919 	dim = isl_space_dim(space, isl_dim_set);
8920 	total = isl_space_dim(space, isl_dim_all);
8921 	if (nparam < 0 || dim < 0 || total < 0)
8922 		space = isl_space_free(space);
8923 	bset = isl_basic_set_alloc_space(space, 0, 0, dim);
8924 	if (!bset)
8925 		return NULL;
8926 	for (i = 0; i < dim; ++i) {
8927 		int k = isl_basic_set_alloc_inequality(bset);
8928 		if (k < 0)
8929 			goto error;
8930 		isl_seq_clr(bset->ineq[k], 1 + total);
8931 		isl_int_set_si(bset->ineq[k][1 + nparam + i], 1);
8932 	}
8933 	return bset;
8934 error:
8935 	isl_basic_set_free(bset);
8936 	return NULL;
8937 }
8938 
8939 /* Construct the half-space x_pos >= 0.
8940  */
nonneg_halfspace(__isl_take isl_space * space,int pos)8941 static __isl_give isl_basic_set *nonneg_halfspace(__isl_take isl_space *space,
8942 	int pos)
8943 {
8944 	int k;
8945 	isl_size total;
8946 	isl_basic_set *nonneg;
8947 
8948 	total = isl_space_dim(space, isl_dim_all);
8949 	if (total < 0)
8950 		space = isl_space_free(space);
8951 	nonneg = isl_basic_set_alloc_space(space, 0, 0, 1);
8952 	k = isl_basic_set_alloc_inequality(nonneg);
8953 	if (k < 0)
8954 		goto error;
8955 	isl_seq_clr(nonneg->ineq[k], 1 + total);
8956 	isl_int_set_si(nonneg->ineq[k][pos], 1);
8957 
8958 	return isl_basic_set_finalize(nonneg);
8959 error:
8960 	isl_basic_set_free(nonneg);
8961 	return NULL;
8962 }
8963 
8964 /* Construct the half-space x_pos <= -1.
8965  */
neg_halfspace(__isl_take isl_space * space,int pos)8966 static __isl_give isl_basic_set *neg_halfspace(__isl_take isl_space *space,
8967 	int pos)
8968 {
8969 	int k;
8970 	isl_size total;
8971 	isl_basic_set *neg;
8972 
8973 	total = isl_space_dim(space, isl_dim_all);
8974 	if (total < 0)
8975 		space = isl_space_free(space);
8976 	neg = isl_basic_set_alloc_space(space, 0, 0, 1);
8977 	k = isl_basic_set_alloc_inequality(neg);
8978 	if (k < 0)
8979 		goto error;
8980 	isl_seq_clr(neg->ineq[k], 1 + total);
8981 	isl_int_set_si(neg->ineq[k][0], -1);
8982 	isl_int_set_si(neg->ineq[k][pos], -1);
8983 
8984 	return isl_basic_set_finalize(neg);
8985 error:
8986 	isl_basic_set_free(neg);
8987 	return NULL;
8988 }
8989 
isl_set_split_dims(__isl_take isl_set * set,enum isl_dim_type type,unsigned first,unsigned n)8990 __isl_give isl_set *isl_set_split_dims(__isl_take isl_set *set,
8991 	enum isl_dim_type type, unsigned first, unsigned n)
8992 {
8993 	int i;
8994 	unsigned offset;
8995 	isl_basic_set *nonneg;
8996 	isl_basic_set *neg;
8997 
8998 	if (n == 0)
8999 		return set;
9000 
9001 	if (isl_set_check_range(set, type, first, n) < 0)
9002 		return isl_set_free(set);
9003 
9004 	offset = pos(set->dim, type);
9005 	for (i = 0; i < n; ++i) {
9006 		nonneg = nonneg_halfspace(isl_set_get_space(set),
9007 					  offset + first + i);
9008 		neg = neg_halfspace(isl_set_get_space(set), offset + first + i);
9009 
9010 		set = isl_set_intersect(set, isl_basic_set_union(nonneg, neg));
9011 	}
9012 
9013 	return set;
9014 }
9015 
foreach_orthant(__isl_take isl_set * set,int * signs,int first,int len,isl_stat (* fn)(__isl_take isl_set * orthant,int * signs,void * user),void * user)9016 static isl_stat foreach_orthant(__isl_take isl_set *set, int *signs, int first,
9017 	int len,
9018 	isl_stat (*fn)(__isl_take isl_set *orthant, int *signs, void *user),
9019 	void *user)
9020 {
9021 	isl_set *half;
9022 
9023 	if (!set)
9024 		return isl_stat_error;
9025 	if (isl_set_plain_is_empty(set)) {
9026 		isl_set_free(set);
9027 		return isl_stat_ok;
9028 	}
9029 	if (first == len)
9030 		return fn(set, signs, user);
9031 
9032 	signs[first] = 1;
9033 	half = isl_set_from_basic_set(nonneg_halfspace(isl_set_get_space(set),
9034 							1 + first));
9035 	half = isl_set_intersect(half, isl_set_copy(set));
9036 	if (foreach_orthant(half, signs, first + 1, len, fn, user) < 0)
9037 		goto error;
9038 
9039 	signs[first] = -1;
9040 	half = isl_set_from_basic_set(neg_halfspace(isl_set_get_space(set),
9041 							1 + first));
9042 	half = isl_set_intersect(half, set);
9043 	return foreach_orthant(half, signs, first + 1, len, fn, user);
9044 error:
9045 	isl_set_free(set);
9046 	return isl_stat_error;
9047 }
9048 
9049 /* Call "fn" on the intersections of "set" with each of the orthants
9050  * (except for obviously empty intersections).  The orthant is identified
9051  * by the signs array, with each entry having value 1 or -1 according
9052  * to the sign of the corresponding variable.
9053  */
isl_set_foreach_orthant(__isl_keep isl_set * set,isl_stat (* fn)(__isl_take isl_set * orthant,int * signs,void * user),void * user)9054 isl_stat isl_set_foreach_orthant(__isl_keep isl_set *set,
9055 	isl_stat (*fn)(__isl_take isl_set *orthant, int *signs, void *user),
9056 	void *user)
9057 {
9058 	isl_size nparam;
9059 	isl_size nvar;
9060 	int *signs;
9061 	isl_stat r;
9062 
9063 	if (!set)
9064 		return isl_stat_error;
9065 	if (isl_set_plain_is_empty(set))
9066 		return isl_stat_ok;
9067 
9068 	nparam = isl_set_dim(set, isl_dim_param);
9069 	nvar = isl_set_dim(set, isl_dim_set);
9070 	if (nparam < 0 || nvar < 0)
9071 		return isl_stat_error;
9072 
9073 	signs = isl_alloc_array(set->ctx, int, nparam + nvar);
9074 
9075 	r = foreach_orthant(isl_set_copy(set), signs, 0, nparam + nvar,
9076 			    fn, user);
9077 
9078 	free(signs);
9079 
9080 	return r;
9081 }
9082 
isl_set_is_equal(__isl_keep isl_set * set1,__isl_keep isl_set * set2)9083 isl_bool isl_set_is_equal(__isl_keep isl_set *set1, __isl_keep isl_set *set2)
9084 {
9085 	return isl_map_is_equal(set_to_map(set1), set_to_map(set2));
9086 }
9087 
isl_basic_map_is_subset(__isl_keep isl_basic_map * bmap1,__isl_keep isl_basic_map * bmap2)9088 isl_bool isl_basic_map_is_subset(__isl_keep isl_basic_map *bmap1,
9089 	__isl_keep isl_basic_map *bmap2)
9090 {
9091 	isl_bool is_subset;
9092 	struct isl_map *map1;
9093 	struct isl_map *map2;
9094 
9095 	if (!bmap1 || !bmap2)
9096 		return isl_bool_error;
9097 
9098 	map1 = isl_map_from_basic_map(isl_basic_map_copy(bmap1));
9099 	map2 = isl_map_from_basic_map(isl_basic_map_copy(bmap2));
9100 
9101 	is_subset = isl_map_is_subset(map1, map2);
9102 
9103 	isl_map_free(map1);
9104 	isl_map_free(map2);
9105 
9106 	return is_subset;
9107 }
9108 
isl_basic_set_is_subset(__isl_keep isl_basic_set * bset1,__isl_keep isl_basic_set * bset2)9109 isl_bool isl_basic_set_is_subset(__isl_keep isl_basic_set *bset1,
9110 	__isl_keep isl_basic_set *bset2)
9111 {
9112 	return isl_basic_map_is_subset(bset1, bset2);
9113 }
9114 
isl_basic_map_is_equal(__isl_keep isl_basic_map * bmap1,__isl_keep isl_basic_map * bmap2)9115 isl_bool isl_basic_map_is_equal(__isl_keep isl_basic_map *bmap1,
9116 	__isl_keep isl_basic_map *bmap2)
9117 {
9118 	isl_bool is_subset;
9119 
9120 	if (!bmap1 || !bmap2)
9121 		return isl_bool_error;
9122 	is_subset = isl_basic_map_is_subset(bmap1, bmap2);
9123 	if (is_subset != isl_bool_true)
9124 		return is_subset;
9125 	is_subset = isl_basic_map_is_subset(bmap2, bmap1);
9126 	return is_subset;
9127 }
9128 
isl_basic_set_is_equal(__isl_keep isl_basic_set * bset1,__isl_keep isl_basic_set * bset2)9129 isl_bool isl_basic_set_is_equal(__isl_keep isl_basic_set *bset1,
9130 	__isl_keep isl_basic_set *bset2)
9131 {
9132 	return isl_basic_map_is_equal(
9133 		bset_to_bmap(bset1), bset_to_bmap(bset2));
9134 }
9135 
isl_map_is_empty(__isl_keep isl_map * map)9136 isl_bool isl_map_is_empty(__isl_keep isl_map *map)
9137 {
9138 	int i;
9139 	int is_empty;
9140 
9141 	if (!map)
9142 		return isl_bool_error;
9143 	for (i = 0; i < map->n; ++i) {
9144 		is_empty = isl_basic_map_is_empty(map->p[i]);
9145 		if (is_empty < 0)
9146 			return isl_bool_error;
9147 		if (!is_empty)
9148 			return isl_bool_false;
9149 	}
9150 	return isl_bool_true;
9151 }
9152 
isl_map_plain_is_empty(__isl_keep isl_map * map)9153 isl_bool isl_map_plain_is_empty(__isl_keep isl_map *map)
9154 {
9155 	return map ? map->n == 0 : isl_bool_error;
9156 }
9157 
isl_set_plain_is_empty(__isl_keep isl_set * set)9158 isl_bool isl_set_plain_is_empty(__isl_keep isl_set *set)
9159 {
9160 	return set ? set->n == 0 : isl_bool_error;
9161 }
9162 
isl_set_is_empty(__isl_keep isl_set * set)9163 isl_bool isl_set_is_empty(__isl_keep isl_set *set)
9164 {
9165 	return isl_map_is_empty(set_to_map(set));
9166 }
9167 
9168 #undef TYPE
9169 #define TYPE	isl_basic_map
9170 
9171 static
9172 #include "isl_type_has_equal_space_bin_templ.c"
9173 #include "isl_type_check_equal_space_templ.c"
9174 
9175 /* Check that "bset1" and "bset2" live in the same space,
9176  * reporting an error if they do not.
9177  */
isl_basic_set_check_equal_space(__isl_keep isl_basic_set * bset1,__isl_keep isl_basic_set * bset2)9178 isl_stat isl_basic_set_check_equal_space(__isl_keep isl_basic_set *bset1,
9179 	__isl_keep isl_basic_set *bset2)
9180 {
9181 	return isl_basic_map_check_equal_space(bset_to_bmap(bset1),
9182 						bset_to_bmap(bset1));
9183 }
9184 
9185 #undef TYPE
9186 #define TYPE	isl_map
9187 
9188 #include "isl_type_has_equal_space_bin_templ.c"
9189 #include "isl_type_check_equal_space_templ.c"
9190 #include "isl_type_has_space_templ.c"
9191 
isl_set_has_equal_space(__isl_keep isl_set * set1,__isl_keep isl_set * set2)9192 isl_bool isl_set_has_equal_space(__isl_keep isl_set *set1,
9193 	__isl_keep isl_set *set2)
9194 {
9195 	return isl_map_has_equal_space(set_to_map(set1), set_to_map(set2));
9196 }
9197 
9198 #undef TYPE1
9199 #define TYPE1		isl_map
9200 #undef TYPE2
9201 #define TYPE2		isl_basic_map
9202 #undef TYPE_PAIR
9203 #define TYPE_PAIR	isl_map_basic_map
9204 
9205 static
9206 #include "isl_type_has_equal_space_templ.c"
9207 #include "isl_type_check_equal_space_templ.c"
9208 
9209 /* Check that "set" and "bset" live in the same space,
9210  * reporting an error if they do not.
9211  */
isl_set_basic_set_check_equal_space(__isl_keep isl_set * set,__isl_keep isl_basic_set * bset)9212 isl_stat isl_set_basic_set_check_equal_space(__isl_keep isl_set *set,
9213 	__isl_keep isl_basic_set *bset)
9214 {
9215 	return isl_map_basic_map_check_equal_space(set_to_map(set),
9216 						    bset_to_bmap(bset));
9217 }
9218 
map_is_equal(__isl_keep isl_map * map1,__isl_keep isl_map * map2)9219 static isl_bool map_is_equal(__isl_keep isl_map *map1, __isl_keep isl_map *map2)
9220 {
9221 	isl_bool is_subset;
9222 
9223 	if (!map1 || !map2)
9224 		return isl_bool_error;
9225 	is_subset = isl_map_is_subset(map1, map2);
9226 	if (is_subset != isl_bool_true)
9227 		return is_subset;
9228 	is_subset = isl_map_is_subset(map2, map1);
9229 	return is_subset;
9230 }
9231 
9232 /* Is "map1" equal to "map2"?
9233  *
9234  * First check if they are obviously equal.
9235  * If not, then perform a more detailed analysis.
9236  */
isl_map_is_equal(__isl_keep isl_map * map1,__isl_keep isl_map * map2)9237 isl_bool isl_map_is_equal(__isl_keep isl_map *map1, __isl_keep isl_map *map2)
9238 {
9239 	isl_bool equal;
9240 
9241 	equal = isl_map_plain_is_equal(map1, map2);
9242 	if (equal < 0 || equal)
9243 		return equal;
9244 	return isl_map_align_params_map_map_and_test(map1, map2, &map_is_equal);
9245 }
9246 
isl_basic_map_is_strict_subset(__isl_keep isl_basic_map * bmap1,__isl_keep isl_basic_map * bmap2)9247 isl_bool isl_basic_map_is_strict_subset(__isl_keep isl_basic_map *bmap1,
9248 	__isl_keep isl_basic_map *bmap2)
9249 {
9250 	isl_bool is_subset;
9251 
9252 	if (!bmap1 || !bmap2)
9253 		return isl_bool_error;
9254 	is_subset = isl_basic_map_is_subset(bmap1, bmap2);
9255 	if (is_subset != isl_bool_true)
9256 		return is_subset;
9257 	is_subset = isl_basic_map_is_subset(bmap2, bmap1);
9258 	return isl_bool_not(is_subset);
9259 }
9260 
isl_map_is_strict_subset(__isl_keep isl_map * map1,__isl_keep isl_map * map2)9261 isl_bool isl_map_is_strict_subset(__isl_keep isl_map *map1,
9262 	__isl_keep isl_map *map2)
9263 {
9264 	isl_bool is_subset;
9265 
9266 	if (!map1 || !map2)
9267 		return isl_bool_error;
9268 	is_subset = isl_map_is_subset(map1, map2);
9269 	if (is_subset != isl_bool_true)
9270 		return is_subset;
9271 	is_subset = isl_map_is_subset(map2, map1);
9272 	return isl_bool_not(is_subset);
9273 }
9274 
isl_set_is_strict_subset(__isl_keep isl_set * set1,__isl_keep isl_set * set2)9275 isl_bool isl_set_is_strict_subset(__isl_keep isl_set *set1,
9276 	__isl_keep isl_set *set2)
9277 {
9278 	return isl_map_is_strict_subset(set_to_map(set1), set_to_map(set2));
9279 }
9280 
9281 /* Is "bmap" obviously equal to the universe with the same space?
9282  *
9283  * That is, does it not have any constraints?
9284  */
isl_basic_map_plain_is_universe(__isl_keep isl_basic_map * bmap)9285 isl_bool isl_basic_map_plain_is_universe(__isl_keep isl_basic_map *bmap)
9286 {
9287 	if (!bmap)
9288 		return isl_bool_error;
9289 	return bmap->n_eq == 0 && bmap->n_ineq == 0;
9290 }
9291 
9292 /* Is "bset" obviously equal to the universe with the same space?
9293  */
isl_basic_set_plain_is_universe(__isl_keep isl_basic_set * bset)9294 isl_bool isl_basic_set_plain_is_universe(__isl_keep isl_basic_set *bset)
9295 {
9296 	return isl_basic_map_plain_is_universe(bset);
9297 }
9298 
9299 /* If "c" does not involve any existentially quantified variables,
9300  * then set *univ to false and abort
9301  */
involves_divs(__isl_take isl_constraint * c,void * user)9302 static isl_stat involves_divs(__isl_take isl_constraint *c, void *user)
9303 {
9304 	isl_bool *univ = user;
9305 	isl_size n;
9306 
9307 	n = isl_constraint_dim(c, isl_dim_div);
9308 	if (n < 0)
9309 		c = isl_constraint_free(c);
9310 	*univ = isl_constraint_involves_dims(c, isl_dim_div, 0, n);
9311 	isl_constraint_free(c);
9312 	if (*univ < 0 || !*univ)
9313 		return isl_stat_error;
9314 	return isl_stat_ok;
9315 }
9316 
9317 /* Is "bmap" equal to the universe with the same space?
9318  *
9319  * First check if it is obviously equal to the universe.
9320  * If not and if there are any constraints not involving
9321  * existentially quantified variables, then it is certainly
9322  * not equal to the universe.
9323  * Otherwise, check if the universe is a subset of "bmap".
9324  */
isl_basic_map_is_universe(__isl_keep isl_basic_map * bmap)9325 isl_bool isl_basic_map_is_universe(__isl_keep isl_basic_map *bmap)
9326 {
9327 	isl_size n_div;
9328 	isl_bool univ;
9329 	isl_basic_map *test;
9330 
9331 	univ = isl_basic_map_plain_is_universe(bmap);
9332 	if (univ < 0 || univ)
9333 		return univ;
9334 	n_div = isl_basic_map_dim(bmap, isl_dim_div);
9335 	if (n_div < 0)
9336 		return isl_bool_error;
9337 	if (n_div == 0)
9338 		return isl_bool_false;
9339 	univ = isl_bool_true;
9340 	if (isl_basic_map_foreach_constraint(bmap, &involves_divs, &univ) < 0 &&
9341 	    univ)
9342 		return isl_bool_error;
9343 	if (univ < 0 || !univ)
9344 		return univ;
9345 	test = isl_basic_map_universe(isl_basic_map_get_space(bmap));
9346 	univ = isl_basic_map_is_subset(test, bmap);
9347 	isl_basic_map_free(test);
9348 	return univ;
9349 }
9350 
9351 /* Is "bset" equal to the universe with the same space?
9352  */
isl_basic_set_is_universe(__isl_keep isl_basic_set * bset)9353 isl_bool isl_basic_set_is_universe(__isl_keep isl_basic_set *bset)
9354 {
9355 	return isl_basic_map_is_universe(bset);
9356 }
9357 
isl_map_plain_is_universe(__isl_keep isl_map * map)9358 isl_bool isl_map_plain_is_universe(__isl_keep isl_map *map)
9359 {
9360 	int i;
9361 
9362 	if (!map)
9363 		return isl_bool_error;
9364 
9365 	for (i = 0; i < map->n; ++i) {
9366 		isl_bool r = isl_basic_map_plain_is_universe(map->p[i]);
9367 		if (r < 0 || r)
9368 			return r;
9369 	}
9370 
9371 	return isl_bool_false;
9372 }
9373 
isl_set_plain_is_universe(__isl_keep isl_set * set)9374 isl_bool isl_set_plain_is_universe(__isl_keep isl_set *set)
9375 {
9376 	return isl_map_plain_is_universe(set_to_map(set));
9377 }
9378 
isl_basic_map_is_empty(__isl_keep isl_basic_map * bmap)9379 isl_bool isl_basic_map_is_empty(__isl_keep isl_basic_map *bmap)
9380 {
9381 	struct isl_basic_set *bset = NULL;
9382 	struct isl_vec *sample = NULL;
9383 	isl_bool empty, non_empty;
9384 
9385 	if (!bmap)
9386 		return isl_bool_error;
9387 
9388 	if (ISL_F_ISSET(bmap, ISL_BASIC_MAP_EMPTY))
9389 		return isl_bool_true;
9390 
9391 	if (isl_basic_map_plain_is_universe(bmap))
9392 		return isl_bool_false;
9393 
9394 	if (ISL_F_ISSET(bmap, ISL_BASIC_MAP_RATIONAL)) {
9395 		struct isl_basic_map *copy = isl_basic_map_copy(bmap);
9396 		copy = isl_basic_map_remove_redundancies(copy);
9397 		empty = isl_basic_map_plain_is_empty(copy);
9398 		isl_basic_map_free(copy);
9399 		return empty;
9400 	}
9401 
9402 	non_empty = isl_basic_map_plain_is_non_empty(bmap);
9403 	if (non_empty < 0)
9404 		return isl_bool_error;
9405 	if (non_empty)
9406 		return isl_bool_false;
9407 	isl_vec_free(bmap->sample);
9408 	bmap->sample = NULL;
9409 	bset = isl_basic_map_underlying_set(isl_basic_map_copy(bmap));
9410 	if (!bset)
9411 		return isl_bool_error;
9412 	sample = isl_basic_set_sample_vec(bset);
9413 	if (!sample)
9414 		return isl_bool_error;
9415 	empty = sample->size == 0;
9416 	isl_vec_free(bmap->sample);
9417 	bmap->sample = sample;
9418 	if (empty)
9419 		ISL_F_SET(bmap, ISL_BASIC_MAP_EMPTY);
9420 
9421 	return empty;
9422 }
9423 
isl_basic_map_plain_is_empty(__isl_keep isl_basic_map * bmap)9424 isl_bool isl_basic_map_plain_is_empty(__isl_keep isl_basic_map *bmap)
9425 {
9426 	if (!bmap)
9427 		return isl_bool_error;
9428 	return ISL_F_ISSET(bmap, ISL_BASIC_MAP_EMPTY);
9429 }
9430 
isl_basic_set_plain_is_empty(__isl_keep isl_basic_set * bset)9431 isl_bool isl_basic_set_plain_is_empty(__isl_keep isl_basic_set *bset)
9432 {
9433 	if (!bset)
9434 		return isl_bool_error;
9435 	return ISL_F_ISSET(bset, ISL_BASIC_SET_EMPTY);
9436 }
9437 
9438 /* Is "bmap" known to be non-empty?
9439  *
9440  * That is, is the cached sample still valid?
9441  */
isl_basic_map_plain_is_non_empty(__isl_keep isl_basic_map * bmap)9442 isl_bool isl_basic_map_plain_is_non_empty(__isl_keep isl_basic_map *bmap)
9443 {
9444 	isl_size total;
9445 
9446 	if (!bmap)
9447 		return isl_bool_error;
9448 	if (!bmap->sample)
9449 		return isl_bool_false;
9450 	total = isl_basic_map_dim(bmap, isl_dim_all);
9451 	if (total < 0)
9452 		return isl_bool_error;
9453 	if (bmap->sample->size != 1 + total)
9454 		return isl_bool_false;
9455 	return isl_basic_map_contains(bmap, bmap->sample);
9456 }
9457 
isl_basic_set_is_empty(__isl_keep isl_basic_set * bset)9458 isl_bool isl_basic_set_is_empty(__isl_keep isl_basic_set *bset)
9459 {
9460 	return isl_basic_map_is_empty(bset_to_bmap(bset));
9461 }
9462 
isl_basic_map_union(__isl_take isl_basic_map * bmap1,__isl_take isl_basic_map * bmap2)9463 __isl_give isl_map *isl_basic_map_union(__isl_take isl_basic_map *bmap1,
9464 	__isl_take isl_basic_map *bmap2)
9465 {
9466 	struct isl_map *map;
9467 
9468 	if (isl_basic_map_check_equal_space(bmap1, bmap2) < 0)
9469 		goto error;
9470 
9471 	map = isl_map_alloc_space(isl_space_copy(bmap1->dim), 2, 0);
9472 	if (!map)
9473 		goto error;
9474 	map = isl_map_add_basic_map(map, bmap1);
9475 	map = isl_map_add_basic_map(map, bmap2);
9476 	return map;
9477 error:
9478 	isl_basic_map_free(bmap1);
9479 	isl_basic_map_free(bmap2);
9480 	return NULL;
9481 }
9482 
isl_basic_set_union(__isl_take isl_basic_set * bset1,__isl_take isl_basic_set * bset2)9483 __isl_give isl_set *isl_basic_set_union(__isl_take isl_basic_set *bset1,
9484 	__isl_take isl_basic_set *bset2)
9485 {
9486 	return set_from_map(isl_basic_map_union(bset_to_bmap(bset1),
9487 						bset_to_bmap(bset2)));
9488 }
9489 
9490 /* Order divs such that any div only depends on previous divs */
isl_basic_map_order_divs(__isl_take isl_basic_map * bmap)9491 __isl_give isl_basic_map *isl_basic_map_order_divs(
9492 	__isl_take isl_basic_map *bmap)
9493 {
9494 	int i;
9495 	isl_size off;
9496 
9497 	off = isl_basic_map_var_offset(bmap, isl_dim_div);
9498 	if (off < 0)
9499 		return isl_basic_map_free(bmap);
9500 
9501 	for (i = 0; i < bmap->n_div; ++i) {
9502 		int pos;
9503 		if (isl_int_is_zero(bmap->div[i][0]))
9504 			continue;
9505 		pos = isl_seq_first_non_zero(bmap->div[i]+1+1+off+i,
9506 							    bmap->n_div-i);
9507 		if (pos == -1)
9508 			continue;
9509 		if (pos == 0)
9510 			isl_die(isl_basic_map_get_ctx(bmap), isl_error_internal,
9511 				"integer division depends on itself",
9512 				return isl_basic_map_free(bmap));
9513 		bmap = isl_basic_map_swap_div(bmap, i, i + pos);
9514 		if (!bmap)
9515 			return NULL;
9516 		--i;
9517 	}
9518 	return bmap;
9519 }
9520 
isl_map_order_divs(__isl_take isl_map * map)9521 __isl_give isl_map *isl_map_order_divs(__isl_take isl_map *map)
9522 {
9523 	int i;
9524 
9525 	if (!map)
9526 		return 0;
9527 
9528 	for (i = 0; i < map->n; ++i) {
9529 		map->p[i] = isl_basic_map_order_divs(map->p[i]);
9530 		if (!map->p[i])
9531 			goto error;
9532 	}
9533 
9534 	return map;
9535 error:
9536 	isl_map_free(map);
9537 	return NULL;
9538 }
9539 
9540 /* Sort the local variables of "bset".
9541  */
isl_basic_set_sort_divs(__isl_take isl_basic_set * bset)9542 __isl_give isl_basic_set *isl_basic_set_sort_divs(
9543 	__isl_take isl_basic_set *bset)
9544 {
9545 	return bset_from_bmap(isl_basic_map_sort_divs(bset_to_bmap(bset)));
9546 }
9547 
9548 /* Apply the expansion computed by isl_merge_divs.
9549  * The expansion itself is given by "exp" while the resulting
9550  * list of divs is given by "div".
9551  *
9552  * Move the integer divisions of "bmap" into the right position
9553  * according to "exp" and then introduce the additional integer
9554  * divisions, adding div constraints.
9555  * The moving should be done first to avoid moving coefficients
9556  * in the definitions of the extra integer divisions.
9557  */
isl_basic_map_expand_divs(__isl_take isl_basic_map * bmap,__isl_take isl_mat * div,int * exp)9558 __isl_give isl_basic_map *isl_basic_map_expand_divs(
9559 	__isl_take isl_basic_map *bmap, __isl_take isl_mat *div, int *exp)
9560 {
9561 	int i, j;
9562 	int n_div;
9563 
9564 	bmap = isl_basic_map_cow(bmap);
9565 	if (!bmap || !div)
9566 		goto error;
9567 
9568 	if (div->n_row < bmap->n_div)
9569 		isl_die(isl_mat_get_ctx(div), isl_error_invalid,
9570 			"not an expansion", goto error);
9571 
9572 	n_div = bmap->n_div;
9573 	bmap = isl_basic_map_extend(bmap, div->n_row - n_div, 0,
9574 					    2 * (div->n_row - n_div));
9575 
9576 	for (i = n_div; i < div->n_row; ++i)
9577 		if (isl_basic_map_alloc_div(bmap) < 0)
9578 			goto error;
9579 
9580 	for (j = n_div - 1; j >= 0; --j) {
9581 		if (exp[j] == j)
9582 			break;
9583 		bmap = isl_basic_map_swap_div(bmap, j, exp[j]);
9584 		if (!bmap)
9585 			goto error;
9586 	}
9587 	j = 0;
9588 	for (i = 0; i < div->n_row; ++i) {
9589 		if (j < n_div && exp[j] == i) {
9590 			j++;
9591 		} else {
9592 			isl_seq_cpy(bmap->div[i], div->row[i], div->n_col);
9593 			if (isl_basic_map_div_is_marked_unknown(bmap, i))
9594 				continue;
9595 			bmap = isl_basic_map_add_div_constraints(bmap, i);
9596 			if (!bmap)
9597 				goto error;
9598 		}
9599 	}
9600 
9601 	isl_mat_free(div);
9602 	return bmap;
9603 error:
9604 	isl_basic_map_free(bmap);
9605 	isl_mat_free(div);
9606 	return NULL;
9607 }
9608 
9609 /* Apply the expansion computed by isl_merge_divs.
9610  * The expansion itself is given by "exp" while the resulting
9611  * list of divs is given by "div".
9612  */
isl_basic_set_expand_divs(__isl_take isl_basic_set * bset,__isl_take isl_mat * div,int * exp)9613 __isl_give isl_basic_set *isl_basic_set_expand_divs(
9614 	__isl_take isl_basic_set *bset, __isl_take isl_mat *div, int *exp)
9615 {
9616 	return isl_basic_map_expand_divs(bset, div, exp);
9617 }
9618 
9619 /* Look for a div in dst that corresponds to the div "div" in src.
9620  * The divs before "div" in src and dst are assumed to be the same.
9621  *
9622  * Return the position of the corresponding div in dst
9623  * if there is one.  Otherwise, return a position beyond the integer divisions.
9624  * Return -1 on error.
9625  */
find_div(__isl_keep isl_basic_map * dst,__isl_keep isl_basic_map * src,unsigned div)9626 static int find_div(__isl_keep isl_basic_map *dst,
9627 	__isl_keep isl_basic_map *src, unsigned div)
9628 {
9629 	int i;
9630 	isl_size n_div;
9631 	isl_size v_div;
9632 
9633 	v_div = isl_basic_map_var_offset(src, isl_dim_div);
9634 	n_div = isl_basic_map_dim(dst, isl_dim_div);
9635 	if (n_div < 0 || v_div < 0)
9636 		return -1;
9637 	isl_assert(dst->ctx, div <= n_div, return -1);
9638 	for (i = div; i < n_div; ++i)
9639 		if (isl_seq_eq(dst->div[i], src->div[div], 1+1+v_div+div) &&
9640 		    isl_seq_first_non_zero(dst->div[i] + 1 + 1 + v_div + div,
9641 						n_div - div) == -1)
9642 			return i;
9643 	return n_div;
9644 }
9645 
9646 /* Align the divs of "dst" to those of "src", adding divs from "src"
9647  * if needed.  That is, make sure that the first src->n_div divs
9648  * of the result are equal to those of src.
9649  * The integer division of "src" are assumed to be ordered.
9650  *
9651  * The integer divisions are swapped into the right position
9652  * (possibly after adding them first).  This may result
9653  * in the remaining integer divisions appearing in the wrong order,
9654  * i.e., with some integer division appearing before
9655  * some other integer division on which it depends.
9656  * The integer divisions therefore need to be ordered.
9657  * This will not affect the integer divisions aligned to those of "src",
9658  * since "src" is assumed to have ordered integer divisions.
9659  *
9660  * The result is not finalized as by design it will have redundant
9661  * divs if any divs from "src" were copied.
9662  */
isl_basic_map_align_divs(__isl_take isl_basic_map * dst,__isl_keep isl_basic_map * src)9663 __isl_give isl_basic_map *isl_basic_map_align_divs(
9664 	__isl_take isl_basic_map *dst, __isl_keep isl_basic_map *src)
9665 {
9666 	int i;
9667 	isl_bool known;
9668 	int extended;
9669 	isl_size v_div;
9670 	isl_size dst_n_div;
9671 
9672 	if (!dst || !src)
9673 		return isl_basic_map_free(dst);
9674 
9675 	if (src->n_div == 0)
9676 		return dst;
9677 
9678 	known = isl_basic_map_divs_known(src);
9679 	if (known < 0)
9680 		return isl_basic_map_free(dst);
9681 	if (!known)
9682 		isl_die(isl_basic_map_get_ctx(src), isl_error_invalid,
9683 			"some src divs are unknown",
9684 			return isl_basic_map_free(dst));
9685 
9686 	v_div = isl_basic_map_var_offset(src, isl_dim_div);
9687 	if (v_div < 0)
9688 		return isl_basic_map_free(dst);
9689 
9690 	extended = 0;
9691 	dst_n_div = isl_basic_map_dim(dst, isl_dim_div);
9692 	if (dst_n_div < 0)
9693 		dst = isl_basic_map_free(dst);
9694 	for (i = 0; i < src->n_div; ++i) {
9695 		int j = find_div(dst, src, i);
9696 		if (j < 0)
9697 			dst = isl_basic_map_free(dst);
9698 		if (j == dst_n_div) {
9699 			if (!extended) {
9700 				int extra = src->n_div - i;
9701 				dst = isl_basic_map_cow(dst);
9702 				if (!dst)
9703 					return isl_basic_map_free(dst);
9704 				dst = isl_basic_map_extend(dst,
9705 						extra, 0, 2 * extra);
9706 				extended = 1;
9707 			}
9708 			j = isl_basic_map_alloc_div(dst);
9709 			if (j < 0)
9710 				return isl_basic_map_free(dst);
9711 			isl_seq_cpy(dst->div[j], src->div[i], 1+1+v_div+i);
9712 			isl_seq_clr(dst->div[j]+1+1+v_div+i, dst->n_div - i);
9713 			dst_n_div++;
9714 			dst = isl_basic_map_add_div_constraints(dst, j);
9715 			if (!dst)
9716 				return isl_basic_map_free(dst);
9717 		}
9718 		if (j != i)
9719 			dst = isl_basic_map_swap_div(dst, i, j);
9720 		if (!dst)
9721 			return isl_basic_map_free(dst);
9722 	}
9723 	return isl_basic_map_order_divs(dst);
9724 }
9725 
isl_map_align_divs_internal(__isl_take isl_map * map)9726 __isl_give isl_map *isl_map_align_divs_internal(__isl_take isl_map *map)
9727 {
9728 	int i;
9729 
9730 	if (!map)
9731 		return NULL;
9732 	if (map->n == 0)
9733 		return map;
9734 	map = isl_map_compute_divs(map);
9735 	map = isl_map_order_divs(map);
9736 	map = isl_map_cow(map);
9737 	if (!map)
9738 		return NULL;
9739 
9740 	for (i = 1; i < map->n; ++i)
9741 		map->p[0] = isl_basic_map_align_divs(map->p[0], map->p[i]);
9742 	for (i = 1; i < map->n; ++i) {
9743 		map->p[i] = isl_basic_map_align_divs(map->p[i], map->p[0]);
9744 		if (!map->p[i])
9745 			return isl_map_free(map);
9746 	}
9747 
9748 	map = isl_map_unmark_normalized(map);
9749 	return map;
9750 }
9751 
isl_map_align_divs(__isl_take isl_map * map)9752 __isl_give isl_map *isl_map_align_divs(__isl_take isl_map *map)
9753 {
9754 	return isl_map_align_divs_internal(map);
9755 }
9756 
isl_set_align_divs(__isl_take isl_set * set)9757 __isl_give isl_set *isl_set_align_divs(__isl_take isl_set *set)
9758 {
9759 	return set_from_map(isl_map_align_divs_internal(set_to_map(set)));
9760 }
9761 
9762 /* Align the divs of the basic maps in "map" to those
9763  * of the basic maps in "list", as well as to the other basic maps in "map".
9764  * The elements in "list" are assumed to have known divs.
9765  */
isl_map_align_divs_to_basic_map_list(__isl_take isl_map * map,__isl_keep isl_basic_map_list * list)9766 __isl_give isl_map *isl_map_align_divs_to_basic_map_list(
9767 	__isl_take isl_map *map, __isl_keep isl_basic_map_list *list)
9768 {
9769 	int i;
9770 	isl_size n;
9771 
9772 	n = isl_basic_map_list_n_basic_map(list);
9773 	map = isl_map_compute_divs(map);
9774 	map = isl_map_cow(map);
9775 	if (!map || n < 0)
9776 		return isl_map_free(map);
9777 	if (map->n == 0)
9778 		return map;
9779 
9780 	for (i = 0; i < n; ++i) {
9781 		isl_basic_map *bmap;
9782 
9783 		bmap = isl_basic_map_list_get_basic_map(list, i);
9784 		bmap = isl_basic_map_order_divs(bmap);
9785 		map->p[0] = isl_basic_map_align_divs(map->p[0], bmap);
9786 		isl_basic_map_free(bmap);
9787 	}
9788 	if (!map->p[0])
9789 		return isl_map_free(map);
9790 
9791 	return isl_map_align_divs_internal(map);
9792 }
9793 
9794 /* Align the divs of each element of "list" to those of "bmap".
9795  * Both "bmap" and the elements of "list" are assumed to have known divs.
9796  */
isl_basic_map_list_align_divs_to_basic_map(__isl_take isl_basic_map_list * list,__isl_keep isl_basic_map * bmap)9797 __isl_give isl_basic_map_list *isl_basic_map_list_align_divs_to_basic_map(
9798 	__isl_take isl_basic_map_list *list, __isl_keep isl_basic_map *bmap)
9799 {
9800 	int i;
9801 	isl_size n;
9802 
9803 	n = isl_basic_map_list_n_basic_map(list);
9804 	if (n < 0 || !bmap)
9805 		return isl_basic_map_list_free(list);
9806 
9807 	for (i = 0; i < n; ++i) {
9808 		isl_basic_map *bmap_i;
9809 
9810 		bmap_i = isl_basic_map_list_get_basic_map(list, i);
9811 		bmap_i = isl_basic_map_align_divs(bmap_i, bmap);
9812 		list = isl_basic_map_list_set_basic_map(list, i, bmap_i);
9813 	}
9814 
9815 	return list;
9816 }
9817 
isl_set_apply(__isl_take isl_set * set,__isl_take isl_map * map)9818 __isl_give isl_set *isl_set_apply( __isl_take isl_set *set,
9819 	__isl_take isl_map *map)
9820 {
9821 	isl_bool ok;
9822 
9823 	isl_map_align_params_set(&map, &set);
9824 	ok = isl_map_compatible_domain(map, set);
9825 	if (ok < 0)
9826 		goto error;
9827 	if (!ok)
9828 		isl_die(isl_set_get_ctx(set), isl_error_invalid,
9829 			"incompatible spaces", goto error);
9830 	map = isl_map_intersect_domain(map, set);
9831 	set = isl_map_range(map);
9832 	return set;
9833 error:
9834 	isl_set_free(set);
9835 	isl_map_free(map);
9836 	return NULL;
9837 }
9838 
9839 /* There is no need to cow as removing empty parts doesn't change
9840  * the meaning of the set.
9841  */
isl_map_remove_empty_parts(__isl_take isl_map * map)9842 __isl_give isl_map *isl_map_remove_empty_parts(__isl_take isl_map *map)
9843 {
9844 	int i;
9845 
9846 	if (!map)
9847 		return NULL;
9848 
9849 	for (i = map->n - 1; i >= 0; --i)
9850 		map = remove_if_empty(map, i);
9851 
9852 	return map;
9853 }
9854 
isl_set_remove_empty_parts(__isl_take isl_set * set)9855 __isl_give isl_set *isl_set_remove_empty_parts(__isl_take isl_set *set)
9856 {
9857 	return set_from_map(isl_map_remove_empty_parts(set_to_map(set)));
9858 }
9859 
9860 /* Create a binary relation that maps the shared initial "pos" dimensions
9861  * of "bset1" and "bset2" to the remaining dimensions of "bset1" and "bset2".
9862  */
join_initial(__isl_keep isl_basic_set * bset1,__isl_keep isl_basic_set * bset2,int pos)9863 static __isl_give isl_basic_map *join_initial(__isl_keep isl_basic_set *bset1,
9864 	__isl_keep isl_basic_set *bset2, int pos)
9865 {
9866 	isl_basic_map *bmap1;
9867 	isl_basic_map *bmap2;
9868 
9869 	bmap1 = isl_basic_map_from_range(isl_basic_set_copy(bset1));
9870 	bmap2 = isl_basic_map_from_range(isl_basic_set_copy(bset2));
9871 	bmap1 = isl_basic_map_move_dims(bmap1, isl_dim_in, 0,
9872 					isl_dim_out, 0, pos);
9873 	bmap2 = isl_basic_map_move_dims(bmap2, isl_dim_in, 0,
9874 					isl_dim_out, 0, pos);
9875 	return isl_basic_map_range_product(bmap1, bmap2);
9876 }
9877 
9878 /* Given two basic sets bset1 and bset2, compute the maximal difference
9879  * between the values of dimension pos in bset1 and those in bset2
9880  * for any common value of the parameters and dimensions preceding pos.
9881  */
basic_set_maximal_difference_at(__isl_keep isl_basic_set * bset1,__isl_keep isl_basic_set * bset2,int pos,isl_int * opt)9882 static enum isl_lp_result basic_set_maximal_difference_at(
9883 	__isl_keep isl_basic_set *bset1, __isl_keep isl_basic_set *bset2,
9884 	int pos, isl_int *opt)
9885 {
9886 	isl_basic_map *bmap1;
9887 	struct isl_ctx *ctx;
9888 	struct isl_vec *obj;
9889 	isl_size total;
9890 	isl_size nparam;
9891 	isl_size dim1;
9892 	enum isl_lp_result res;
9893 
9894 	nparam = isl_basic_set_dim(bset1, isl_dim_param);
9895 	dim1 = isl_basic_set_dim(bset1, isl_dim_set);
9896 	if (nparam < 0 || dim1 < 0 || !bset2)
9897 		return isl_lp_error;
9898 
9899 	bmap1 = join_initial(bset1, bset2, pos);
9900 	total = isl_basic_map_dim(bmap1, isl_dim_all);
9901 	if (total < 0)
9902 		return isl_lp_error;
9903 
9904 	ctx = bmap1->ctx;
9905 	obj = isl_vec_alloc(ctx, 1 + total);
9906 	if (!obj)
9907 		goto error;
9908 	isl_seq_clr(obj->block.data, 1 + total);
9909 	isl_int_set_si(obj->block.data[1+nparam+pos], 1);
9910 	isl_int_set_si(obj->block.data[1+nparam+pos+(dim1-pos)], -1);
9911 	res = isl_basic_map_solve_lp(bmap1, 1, obj->block.data, ctx->one,
9912 					opt, NULL, NULL);
9913 	isl_basic_map_free(bmap1);
9914 	isl_vec_free(obj);
9915 	return res;
9916 error:
9917 	isl_basic_map_free(bmap1);
9918 	return isl_lp_error;
9919 }
9920 
9921 /* Given two _disjoint_ basic sets bset1 and bset2, check whether
9922  * for any common value of the parameters and dimensions preceding pos
9923  * in both basic sets, the values of dimension pos in bset1 are
9924  * smaller or larger than those in bset2.
9925  *
9926  * Returns
9927  *	 1 if bset1 follows bset2
9928  *	-1 if bset1 precedes bset2
9929  *	 0 if bset1 and bset2 are incomparable
9930  *	-2 if some error occurred.
9931  */
isl_basic_set_compare_at(__isl_keep isl_basic_set * bset1,__isl_keep isl_basic_set * bset2,int pos)9932 int isl_basic_set_compare_at(__isl_keep isl_basic_set *bset1,
9933 	__isl_keep isl_basic_set *bset2, int pos)
9934 {
9935 	isl_int opt;
9936 	enum isl_lp_result res;
9937 	int cmp;
9938 
9939 	isl_int_init(opt);
9940 
9941 	res = basic_set_maximal_difference_at(bset1, bset2, pos, &opt);
9942 
9943 	if (res == isl_lp_empty)
9944 		cmp = 0;
9945 	else if ((res == isl_lp_ok && isl_int_is_pos(opt)) ||
9946 		  res == isl_lp_unbounded)
9947 		cmp = 1;
9948 	else if (res == isl_lp_ok && isl_int_is_neg(opt))
9949 		cmp = -1;
9950 	else
9951 		cmp = -2;
9952 
9953 	isl_int_clear(opt);
9954 	return cmp;
9955 }
9956 
9957 /* Given two basic sets bset1 and bset2, check whether
9958  * for any common value of the parameters and dimensions preceding pos
9959  * there is a value of dimension pos in bset1 that is larger
9960  * than a value of the same dimension in bset2.
9961  *
9962  * Return
9963  *	 1 if there exists such a pair
9964  *	 0 if there is no such pair, but there is a pair of equal values
9965  *	-1 otherwise
9966  *	-2 if some error occurred.
9967  */
isl_basic_set_follows_at(__isl_keep isl_basic_set * bset1,__isl_keep isl_basic_set * bset2,int pos)9968 int isl_basic_set_follows_at(__isl_keep isl_basic_set *bset1,
9969 	__isl_keep isl_basic_set *bset2, int pos)
9970 {
9971 	isl_bool empty;
9972 	isl_basic_map *bmap;
9973 	isl_size dim1;
9974 
9975 	dim1 = isl_basic_set_dim(bset1, isl_dim_set);
9976 	if (dim1 < 0)
9977 		return -2;
9978 	bmap = join_initial(bset1, bset2, pos);
9979 	bmap = isl_basic_map_order_ge(bmap, isl_dim_out, 0,
9980 					    isl_dim_out, dim1 - pos);
9981 	empty = isl_basic_map_is_empty(bmap);
9982 	if (empty < 0)
9983 		goto error;
9984 	if (empty) {
9985 		isl_basic_map_free(bmap);
9986 		return -1;
9987 	}
9988 	bmap = isl_basic_map_order_gt(bmap, isl_dim_out, 0,
9989 					    isl_dim_out, dim1 - pos);
9990 	empty = isl_basic_map_is_empty(bmap);
9991 	if (empty < 0)
9992 		goto error;
9993 	isl_basic_map_free(bmap);
9994 	if (empty)
9995 		return 0;
9996 	return 1;
9997 error:
9998 	isl_basic_map_free(bmap);
9999 	return -2;
10000 }
10001 
10002 /* Given two sets set1 and set2, check whether
10003  * for any common value of the parameters and dimensions preceding pos
10004  * there is a value of dimension pos in set1 that is larger
10005  * than a value of the same dimension in set2.
10006  *
10007  * Return
10008  *	 1 if there exists such a pair
10009  *	 0 if there is no such pair, but there is a pair of equal values
10010  *	-1 otherwise
10011  *	-2 if some error occurred.
10012  */
isl_set_follows_at(__isl_keep isl_set * set1,__isl_keep isl_set * set2,int pos)10013 int isl_set_follows_at(__isl_keep isl_set *set1,
10014 	__isl_keep isl_set *set2, int pos)
10015 {
10016 	int i, j;
10017 	int follows = -1;
10018 
10019 	if (!set1 || !set2)
10020 		return -2;
10021 
10022 	for (i = 0; i < set1->n; ++i)
10023 		for (j = 0; j < set2->n; ++j) {
10024 			int f;
10025 			f = isl_basic_set_follows_at(set1->p[i], set2->p[j], pos);
10026 			if (f == 1 || f == -2)
10027 				return f;
10028 			if (f > follows)
10029 				follows = f;
10030 		}
10031 
10032 	return follows;
10033 }
10034 
isl_basic_map_plain_has_fixed_var(__isl_keep isl_basic_map * bmap,unsigned pos,isl_int * val)10035 static isl_bool isl_basic_map_plain_has_fixed_var(
10036 	__isl_keep isl_basic_map *bmap, unsigned pos, isl_int *val)
10037 {
10038 	int i;
10039 	int d;
10040 	isl_size total;
10041 
10042 	total = isl_basic_map_dim(bmap, isl_dim_all);
10043 	if (total < 0)
10044 		return isl_bool_error;
10045 	for (i = 0, d = total-1; i < bmap->n_eq && d+1 > pos; ++i) {
10046 		for (; d+1 > pos; --d)
10047 			if (!isl_int_is_zero(bmap->eq[i][1+d]))
10048 				break;
10049 		if (d != pos)
10050 			continue;
10051 		if (isl_seq_first_non_zero(bmap->eq[i]+1, d) != -1)
10052 			return isl_bool_false;
10053 		if (isl_seq_first_non_zero(bmap->eq[i]+1+d+1, total-d-1) != -1)
10054 			return isl_bool_false;
10055 		if (!isl_int_is_one(bmap->eq[i][1+d]))
10056 			return isl_bool_false;
10057 		if (val)
10058 			isl_int_neg(*val, bmap->eq[i][0]);
10059 		return isl_bool_true;
10060 	}
10061 	return isl_bool_false;
10062 }
10063 
isl_map_plain_has_fixed_var(__isl_keep isl_map * map,unsigned pos,isl_int * val)10064 static isl_bool isl_map_plain_has_fixed_var(__isl_keep isl_map *map,
10065 	unsigned pos, isl_int *val)
10066 {
10067 	int i;
10068 	isl_int v;
10069 	isl_int tmp;
10070 	isl_bool fixed;
10071 
10072 	if (!map)
10073 		return isl_bool_error;
10074 	if (map->n == 0)
10075 		return isl_bool_false;
10076 	if (map->n == 1)
10077 		return isl_basic_map_plain_has_fixed_var(map->p[0], pos, val);
10078 	isl_int_init(v);
10079 	isl_int_init(tmp);
10080 	fixed = isl_basic_map_plain_has_fixed_var(map->p[0], pos, &v);
10081 	for (i = 1; fixed == isl_bool_true && i < map->n; ++i) {
10082 		fixed = isl_basic_map_plain_has_fixed_var(map->p[i], pos, &tmp);
10083 		if (fixed == isl_bool_true && isl_int_ne(tmp, v))
10084 			fixed = isl_bool_false;
10085 	}
10086 	if (val)
10087 		isl_int_set(*val, v);
10088 	isl_int_clear(tmp);
10089 	isl_int_clear(v);
10090 	return fixed;
10091 }
10092 
isl_basic_set_plain_has_fixed_var(__isl_keep isl_basic_set * bset,unsigned pos,isl_int * val)10093 static isl_bool isl_basic_set_plain_has_fixed_var(
10094 	__isl_keep isl_basic_set *bset, unsigned pos, isl_int *val)
10095 {
10096 	return isl_basic_map_plain_has_fixed_var(bset_to_bmap(bset),
10097 						pos, val);
10098 }
10099 
isl_basic_map_plain_is_fixed(__isl_keep isl_basic_map * bmap,enum isl_dim_type type,unsigned pos,isl_int * val)10100 isl_bool isl_basic_map_plain_is_fixed(__isl_keep isl_basic_map *bmap,
10101 	enum isl_dim_type type, unsigned pos, isl_int *val)
10102 {
10103 	if (isl_basic_map_check_range(bmap, type, pos, 1) < 0)
10104 		return isl_bool_error;
10105 	return isl_basic_map_plain_has_fixed_var(bmap,
10106 		isl_basic_map_offset(bmap, type) - 1 + pos, val);
10107 }
10108 
10109 /* If "bmap" obviously lies on a hyperplane where the given dimension
10110  * has a fixed value, then return that value.
10111  * Otherwise return NaN.
10112  */
isl_basic_map_plain_get_val_if_fixed(__isl_keep isl_basic_map * bmap,enum isl_dim_type type,unsigned pos)10113 __isl_give isl_val *isl_basic_map_plain_get_val_if_fixed(
10114 	__isl_keep isl_basic_map *bmap,
10115 	enum isl_dim_type type, unsigned pos)
10116 {
10117 	isl_ctx *ctx;
10118 	isl_val *v;
10119 	isl_bool fixed;
10120 
10121 	if (!bmap)
10122 		return NULL;
10123 	ctx = isl_basic_map_get_ctx(bmap);
10124 	v = isl_val_alloc(ctx);
10125 	if (!v)
10126 		return NULL;
10127 	fixed = isl_basic_map_plain_is_fixed(bmap, type, pos, &v->n);
10128 	if (fixed < 0)
10129 		return isl_val_free(v);
10130 	if (fixed) {
10131 		isl_int_set_si(v->d, 1);
10132 		return v;
10133 	}
10134 	isl_val_free(v);
10135 	return isl_val_nan(ctx);
10136 }
10137 
isl_map_plain_is_fixed(__isl_keep isl_map * map,enum isl_dim_type type,unsigned pos,isl_int * val)10138 isl_bool isl_map_plain_is_fixed(__isl_keep isl_map *map,
10139 	enum isl_dim_type type, unsigned pos, isl_int *val)
10140 {
10141 	if (isl_map_check_range(map, type, pos, 1) < 0)
10142 		return isl_bool_error;
10143 	return isl_map_plain_has_fixed_var(map,
10144 		map_offset(map, type) - 1 + pos, val);
10145 }
10146 
10147 /* If "map" obviously lies on a hyperplane where the given dimension
10148  * has a fixed value, then return that value.
10149  * Otherwise return NaN.
10150  */
isl_map_plain_get_val_if_fixed(__isl_keep isl_map * map,enum isl_dim_type type,unsigned pos)10151 __isl_give isl_val *isl_map_plain_get_val_if_fixed(__isl_keep isl_map *map,
10152 	enum isl_dim_type type, unsigned pos)
10153 {
10154 	isl_ctx *ctx;
10155 	isl_val *v;
10156 	isl_bool fixed;
10157 
10158 	if (!map)
10159 		return NULL;
10160 	ctx = isl_map_get_ctx(map);
10161 	v = isl_val_alloc(ctx);
10162 	if (!v)
10163 		return NULL;
10164 	fixed = isl_map_plain_is_fixed(map, type, pos, &v->n);
10165 	if (fixed < 0)
10166 		return isl_val_free(v);
10167 	if (fixed) {
10168 		isl_int_set_si(v->d, 1);
10169 		return v;
10170 	}
10171 	isl_val_free(v);
10172 	return isl_val_nan(ctx);
10173 }
10174 
10175 /* If "set" obviously lies on a hyperplane where the given dimension
10176  * has a fixed value, then return that value.
10177  * Otherwise return NaN.
10178  */
isl_set_plain_get_val_if_fixed(__isl_keep isl_set * set,enum isl_dim_type type,unsigned pos)10179 __isl_give isl_val *isl_set_plain_get_val_if_fixed(__isl_keep isl_set *set,
10180 	enum isl_dim_type type, unsigned pos)
10181 {
10182 	return isl_map_plain_get_val_if_fixed(set, type, pos);
10183 }
10184 
10185 /* Return a sequence of values in the same space as "set"
10186  * that are equal to the corresponding set dimensions of "set"
10187  * for those set dimensions that obviously lie on a hyperplane
10188  * where the dimension has a fixed value.
10189  * The other elements are set to NaN.
10190  */
isl_set_get_plain_multi_val_if_fixed(__isl_keep isl_set * set)10191 __isl_give isl_multi_val *isl_set_get_plain_multi_val_if_fixed(
10192 	__isl_keep isl_set *set)
10193 {
10194 	int i;
10195 	isl_size n;
10196 	isl_space *space;
10197 	isl_multi_val *mv;
10198 
10199 	space = isl_space_drop_all_params(isl_set_get_space(set));
10200 	mv = isl_multi_val_alloc(space);
10201 	n = isl_multi_val_size(mv);
10202 	if (n < 0)
10203 		return isl_multi_val_free(mv);
10204 
10205 	for (i = 0; i < n; ++i) {
10206 		isl_val *v;
10207 
10208 		v = isl_set_plain_get_val_if_fixed(set, isl_dim_set, i);
10209 		mv = isl_multi_val_set_val(mv, i, v);
10210 	}
10211 
10212 	return mv;
10213 }
10214 
10215 /* Check if dimension dim has fixed value and if so and if val is not NULL,
10216  * then return this fixed value in *val.
10217  */
isl_basic_set_plain_dim_is_fixed(__isl_keep isl_basic_set * bset,unsigned dim,isl_int * val)10218 isl_bool isl_basic_set_plain_dim_is_fixed(__isl_keep isl_basic_set *bset,
10219 	unsigned dim, isl_int *val)
10220 {
10221 	isl_size nparam;
10222 
10223 	nparam = isl_basic_set_dim(bset, isl_dim_param);
10224 	if (nparam < 0)
10225 		return isl_bool_error;
10226 	return isl_basic_set_plain_has_fixed_var(bset, nparam + dim, val);
10227 }
10228 
10229 /* Return -1 if the constraint "c1" should be sorted before "c2"
10230  * and 1 if it should be sorted after "c2".
10231  * Return 0 if the two constraints are the same (up to the constant term).
10232  *
10233  * In particular, if a constraint involves later variables than another
10234  * then it is sorted after this other constraint.
10235  * uset_gist depends on constraints without existentially quantified
10236  * variables sorting first.
10237  *
10238  * For constraints that have the same latest variable, those
10239  * with the same coefficient for this latest variable (first in absolute value
10240  * and then in actual value) are grouped together.
10241  * This is useful for detecting pairs of constraints that can
10242  * be chained in their printed representation.
10243  *
10244  * Finally, within a group, constraints are sorted according to
10245  * their coefficients (excluding the constant term).
10246  */
sort_constraint_cmp(const void * p1,const void * p2,void * arg)10247 static int sort_constraint_cmp(const void *p1, const void *p2, void *arg)
10248 {
10249 	isl_int **c1 = (isl_int **) p1;
10250 	isl_int **c2 = (isl_int **) p2;
10251 	int l1, l2;
10252 	unsigned size = *(unsigned *) arg;
10253 	int cmp;
10254 
10255 	l1 = isl_seq_last_non_zero(*c1 + 1, size);
10256 	l2 = isl_seq_last_non_zero(*c2 + 1, size);
10257 
10258 	if (l1 != l2)
10259 		return l1 - l2;
10260 
10261 	cmp = isl_int_abs_cmp((*c1)[1 + l1], (*c2)[1 + l1]);
10262 	if (cmp != 0)
10263 		return cmp;
10264 	cmp = isl_int_cmp((*c1)[1 + l1], (*c2)[1 + l1]);
10265 	if (cmp != 0)
10266 		return -cmp;
10267 
10268 	return isl_seq_cmp(*c1 + 1, *c2 + 1, size);
10269 }
10270 
10271 /* Return -1 if the constraint "c1" of "bmap" is sorted before "c2"
10272  * by isl_basic_map_sort_constraints, 1 if it is sorted after "c2"
10273  * and 0 if the two constraints are the same (up to the constant term).
10274  */
isl_basic_map_constraint_cmp(__isl_keep isl_basic_map * bmap,isl_int * c1,isl_int * c2)10275 int isl_basic_map_constraint_cmp(__isl_keep isl_basic_map *bmap,
10276 	isl_int *c1, isl_int *c2)
10277 {
10278 	isl_size total;
10279 	unsigned size;
10280 
10281 	total = isl_basic_map_dim(bmap, isl_dim_all);
10282 	if (total < 0)
10283 		return -2;
10284 	size = total;
10285 	return sort_constraint_cmp(&c1, &c2, &size);
10286 }
10287 
isl_basic_map_sort_constraints(__isl_take isl_basic_map * bmap)10288 __isl_give isl_basic_map *isl_basic_map_sort_constraints(
10289 	__isl_take isl_basic_map *bmap)
10290 {
10291 	isl_size total;
10292 	unsigned size;
10293 
10294 	if (!bmap)
10295 		return NULL;
10296 	if (bmap->n_ineq == 0)
10297 		return bmap;
10298 	if (ISL_F_ISSET(bmap, ISL_BASIC_MAP_SORTED))
10299 		return bmap;
10300 	total = isl_basic_map_dim(bmap, isl_dim_all);
10301 	if (total < 0)
10302 		return isl_basic_map_free(bmap);
10303 	size = total;
10304 	if (isl_sort(bmap->ineq, bmap->n_ineq, sizeof(isl_int *),
10305 		    &sort_constraint_cmp, &size) < 0)
10306 		return isl_basic_map_free(bmap);
10307 	ISL_F_SET(bmap, ISL_BASIC_MAP_SORTED);
10308 	return bmap;
10309 }
10310 
isl_basic_set_sort_constraints(__isl_take isl_basic_set * bset)10311 __isl_give isl_basic_set *isl_basic_set_sort_constraints(
10312 	__isl_take isl_basic_set *bset)
10313 {
10314 	isl_basic_map *bmap = bset_to_bmap(bset);
10315 	return bset_from_bmap(isl_basic_map_sort_constraints(bmap));
10316 }
10317 
isl_basic_map_normalize(__isl_take isl_basic_map * bmap)10318 __isl_give isl_basic_map *isl_basic_map_normalize(
10319 	__isl_take isl_basic_map *bmap)
10320 {
10321 	bmap = isl_basic_map_remove_redundancies(bmap);
10322 	bmap = isl_basic_map_sort_constraints(bmap);
10323 	return bmap;
10324 }
isl_basic_map_plain_cmp(__isl_keep isl_basic_map * bmap1,__isl_keep isl_basic_map * bmap2)10325 int isl_basic_map_plain_cmp(__isl_keep isl_basic_map *bmap1,
10326 	__isl_keep isl_basic_map *bmap2)
10327 {
10328 	int i, cmp;
10329 	isl_size total;
10330 	isl_space *space1, *space2;
10331 
10332 	if (!bmap1 || !bmap2)
10333 		return -1;
10334 
10335 	if (bmap1 == bmap2)
10336 		return 0;
10337 	space1 = isl_basic_map_peek_space(bmap1);
10338 	space2 = isl_basic_map_peek_space(bmap2);
10339 	cmp = isl_space_cmp(space1, space2);
10340 	if (cmp)
10341 		return cmp;
10342 	if (ISL_F_ISSET(bmap1, ISL_BASIC_MAP_RATIONAL) !=
10343 	    ISL_F_ISSET(bmap2, ISL_BASIC_MAP_RATIONAL))
10344 		return ISL_F_ISSET(bmap1, ISL_BASIC_MAP_RATIONAL) ? -1 : 1;
10345 	if (ISL_F_ISSET(bmap1, ISL_BASIC_MAP_EMPTY) &&
10346 	    ISL_F_ISSET(bmap2, ISL_BASIC_MAP_EMPTY))
10347 		return 0;
10348 	if (ISL_F_ISSET(bmap1, ISL_BASIC_MAP_EMPTY))
10349 		return 1;
10350 	if (ISL_F_ISSET(bmap2, ISL_BASIC_MAP_EMPTY))
10351 		return -1;
10352 	if (bmap1->n_eq != bmap2->n_eq)
10353 		return bmap1->n_eq - bmap2->n_eq;
10354 	if (bmap1->n_ineq != bmap2->n_ineq)
10355 		return bmap1->n_ineq - bmap2->n_ineq;
10356 	if (bmap1->n_div != bmap2->n_div)
10357 		return bmap1->n_div - bmap2->n_div;
10358 	total = isl_basic_map_dim(bmap1, isl_dim_all);
10359 	if (total < 0)
10360 		return -1;
10361 	for (i = 0; i < bmap1->n_eq; ++i) {
10362 		cmp = isl_seq_cmp(bmap1->eq[i], bmap2->eq[i], 1+total);
10363 		if (cmp)
10364 			return cmp;
10365 	}
10366 	for (i = 0; i < bmap1->n_ineq; ++i) {
10367 		cmp = isl_seq_cmp(bmap1->ineq[i], bmap2->ineq[i], 1+total);
10368 		if (cmp)
10369 			return cmp;
10370 	}
10371 	for (i = 0; i < bmap1->n_div; ++i) {
10372 		isl_bool unknown1, unknown2;
10373 
10374 		unknown1 = isl_basic_map_div_is_marked_unknown(bmap1, i);
10375 		unknown2 = isl_basic_map_div_is_marked_unknown(bmap2, i);
10376 		if (unknown1 < 0 || unknown2 < 0)
10377 			return -1;
10378 		if (unknown1 && unknown2)
10379 			continue;
10380 		if (unknown1)
10381 			return 1;
10382 		if (unknown2)
10383 			return -1;
10384 		cmp = isl_seq_cmp(bmap1->div[i], bmap2->div[i], 1+1+total);
10385 		if (cmp)
10386 			return cmp;
10387 	}
10388 	return 0;
10389 }
10390 
isl_basic_set_plain_cmp(__isl_keep isl_basic_set * bset1,__isl_keep isl_basic_set * bset2)10391 int isl_basic_set_plain_cmp(__isl_keep isl_basic_set *bset1,
10392 	__isl_keep isl_basic_set *bset2)
10393 {
10394 	return isl_basic_map_plain_cmp(bset1, bset2);
10395 }
10396 
isl_set_plain_cmp(__isl_keep isl_set * set1,__isl_keep isl_set * set2)10397 int isl_set_plain_cmp(__isl_keep isl_set *set1, __isl_keep isl_set *set2)
10398 {
10399 	int i, cmp;
10400 
10401 	if (set1 == set2)
10402 		return 0;
10403 	if (set1->n != set2->n)
10404 		return set1->n - set2->n;
10405 
10406 	for (i = 0; i < set1->n; ++i) {
10407 		cmp = isl_basic_set_plain_cmp(set1->p[i], set2->p[i]);
10408 		if (cmp)
10409 			return cmp;
10410 	}
10411 
10412 	return 0;
10413 }
10414 
isl_basic_map_plain_is_equal(__isl_keep isl_basic_map * bmap1,__isl_keep isl_basic_map * bmap2)10415 isl_bool isl_basic_map_plain_is_equal(__isl_keep isl_basic_map *bmap1,
10416 	__isl_keep isl_basic_map *bmap2)
10417 {
10418 	if (!bmap1 || !bmap2)
10419 		return isl_bool_error;
10420 	return isl_basic_map_plain_cmp(bmap1, bmap2) == 0;
10421 }
10422 
isl_basic_set_plain_is_equal(__isl_keep isl_basic_set * bset1,__isl_keep isl_basic_set * bset2)10423 isl_bool isl_basic_set_plain_is_equal(__isl_keep isl_basic_set *bset1,
10424 	__isl_keep isl_basic_set *bset2)
10425 {
10426 	return isl_basic_map_plain_is_equal(bset_to_bmap(bset1),
10427 					    bset_to_bmap(bset2));
10428 }
10429 
qsort_bmap_cmp(const void * p1,const void * p2)10430 static int qsort_bmap_cmp(const void *p1, const void *p2)
10431 {
10432 	isl_basic_map *bmap1 = *(isl_basic_map **) p1;
10433 	isl_basic_map *bmap2 = *(isl_basic_map **) p2;
10434 
10435 	return isl_basic_map_plain_cmp(bmap1, bmap2);
10436 }
10437 
10438 /* Sort the basic maps of "map" and remove duplicate basic maps.
10439  *
10440  * While removing basic maps, we make sure that the basic maps remain
10441  * sorted because isl_map_normalize expects the basic maps of the result
10442  * to be sorted.
10443  */
sort_and_remove_duplicates(__isl_take isl_map * map)10444 static __isl_give isl_map *sort_and_remove_duplicates(__isl_take isl_map *map)
10445 {
10446 	int i, j;
10447 
10448 	map = isl_map_remove_empty_parts(map);
10449 	if (!map)
10450 		return NULL;
10451 	qsort(map->p, map->n, sizeof(struct isl_basic_map *), qsort_bmap_cmp);
10452 	for (i = map->n - 1; i >= 1; --i) {
10453 		if (!isl_basic_map_plain_is_equal(map->p[i - 1], map->p[i]))
10454 			continue;
10455 		isl_basic_map_free(map->p[i-1]);
10456 		for (j = i; j < map->n; ++j)
10457 			map->p[j - 1] = map->p[j];
10458 		map->n--;
10459 	}
10460 
10461 	return map;
10462 }
10463 
10464 /* Remove obvious duplicates among the basic maps of "map".
10465  *
10466  * Unlike isl_map_normalize, this function does not remove redundant
10467  * constraints and only removes duplicates that have exactly the same
10468  * constraints in the input.  It does sort the constraints and
10469  * the basic maps to ease the detection of duplicates.
10470  *
10471  * If "map" has already been normalized or if the basic maps are
10472  * disjoint, then there can be no duplicates.
10473  */
isl_map_remove_obvious_duplicates(__isl_take isl_map * map)10474 __isl_give isl_map *isl_map_remove_obvious_duplicates(__isl_take isl_map *map)
10475 {
10476 	int i;
10477 	isl_basic_map *bmap;
10478 
10479 	if (!map)
10480 		return NULL;
10481 	if (map->n <= 1)
10482 		return map;
10483 	if (ISL_F_ISSET(map, ISL_MAP_NORMALIZED | ISL_MAP_DISJOINT))
10484 		return map;
10485 	for (i = 0; i < map->n; ++i) {
10486 		bmap = isl_basic_map_copy(map->p[i]);
10487 		bmap = isl_basic_map_sort_constraints(bmap);
10488 		if (!bmap)
10489 			return isl_map_free(map);
10490 		isl_basic_map_free(map->p[i]);
10491 		map->p[i] = bmap;
10492 	}
10493 
10494 	map = sort_and_remove_duplicates(map);
10495 	return map;
10496 }
10497 
10498 /* We normalize in place, but if anything goes wrong we need
10499  * to return NULL, so we need to make sure we don't change the
10500  * meaning of any possible other copies of map.
10501  */
isl_map_normalize(__isl_take isl_map * map)10502 __isl_give isl_map *isl_map_normalize(__isl_take isl_map *map)
10503 {
10504 	int i;
10505 	struct isl_basic_map *bmap;
10506 
10507 	if (!map)
10508 		return NULL;
10509 	if (ISL_F_ISSET(map, ISL_MAP_NORMALIZED))
10510 		return map;
10511 	for (i = 0; i < map->n; ++i) {
10512 		bmap = isl_basic_map_normalize(isl_basic_map_copy(map->p[i]));
10513 		if (!bmap)
10514 			goto error;
10515 		isl_basic_map_free(map->p[i]);
10516 		map->p[i] = bmap;
10517 	}
10518 
10519 	map = sort_and_remove_duplicates(map);
10520 	if (map)
10521 		ISL_F_SET(map, ISL_MAP_NORMALIZED);
10522 	return map;
10523 error:
10524 	isl_map_free(map);
10525 	return NULL;
10526 }
10527 
isl_set_normalize(__isl_take isl_set * set)10528 __isl_give isl_set *isl_set_normalize(__isl_take isl_set *set)
10529 {
10530 	return set_from_map(isl_map_normalize(set_to_map(set)));
10531 }
10532 
isl_map_plain_is_equal(__isl_keep isl_map * map1,__isl_keep isl_map * map2)10533 isl_bool isl_map_plain_is_equal(__isl_keep isl_map *map1,
10534 	__isl_keep isl_map *map2)
10535 {
10536 	int i;
10537 	isl_bool equal;
10538 
10539 	if (!map1 || !map2)
10540 		return isl_bool_error;
10541 
10542 	if (map1 == map2)
10543 		return isl_bool_true;
10544 	equal = isl_map_has_equal_space(map1, map2);
10545 	if (equal < 0 || !equal)
10546 		return equal;
10547 
10548 	map1 = isl_map_copy(map1);
10549 	map2 = isl_map_copy(map2);
10550 	map1 = isl_map_normalize(map1);
10551 	map2 = isl_map_normalize(map2);
10552 	if (!map1 || !map2)
10553 		goto error;
10554 	equal = map1->n == map2->n;
10555 	for (i = 0; equal && i < map1->n; ++i) {
10556 		equal = isl_basic_map_plain_is_equal(map1->p[i], map2->p[i]);
10557 		if (equal < 0)
10558 			goto error;
10559 	}
10560 	isl_map_free(map1);
10561 	isl_map_free(map2);
10562 	return equal;
10563 error:
10564 	isl_map_free(map1);
10565 	isl_map_free(map2);
10566 	return isl_bool_error;
10567 }
10568 
isl_set_plain_is_equal(__isl_keep isl_set * set1,__isl_keep isl_set * set2)10569 isl_bool isl_set_plain_is_equal(__isl_keep isl_set *set1,
10570 	__isl_keep isl_set *set2)
10571 {
10572 	return isl_map_plain_is_equal(set_to_map(set1), set_to_map(set2));
10573 }
10574 
10575 /* Return the basic maps in "map" as a list.
10576  */
isl_map_get_basic_map_list(__isl_keep isl_map * map)10577 __isl_give isl_basic_map_list *isl_map_get_basic_map_list(
10578 	__isl_keep isl_map *map)
10579 {
10580 	int i;
10581 	isl_ctx *ctx;
10582 	isl_basic_map_list *list;
10583 
10584 	if (!map)
10585 		return NULL;
10586 	ctx = isl_map_get_ctx(map);
10587 	list = isl_basic_map_list_alloc(ctx, map->n);
10588 
10589 	for (i = 0; i < map->n; ++i) {
10590 		isl_basic_map *bmap;
10591 
10592 		bmap = isl_basic_map_copy(map->p[i]);
10593 		list = isl_basic_map_list_add(list, bmap);
10594 	}
10595 
10596 	return list;
10597 }
10598 
10599 /* Return the intersection of the elements in the non-empty list "list".
10600  * All elements are assumed to live in the same space.
10601  */
isl_basic_map_list_intersect(__isl_take isl_basic_map_list * list)10602 __isl_give isl_basic_map *isl_basic_map_list_intersect(
10603 	__isl_take isl_basic_map_list *list)
10604 {
10605 	int i;
10606 	isl_size n;
10607 	isl_basic_map *bmap;
10608 
10609 	n = isl_basic_map_list_n_basic_map(list);
10610 	if (n < 0)
10611 		goto error;
10612 	if (n < 1)
10613 		isl_die(isl_basic_map_list_get_ctx(list), isl_error_invalid,
10614 			"expecting non-empty list", goto error);
10615 
10616 	bmap = isl_basic_map_list_get_basic_map(list, 0);
10617 	for (i = 1; i < n; ++i) {
10618 		isl_basic_map *bmap_i;
10619 
10620 		bmap_i = isl_basic_map_list_get_basic_map(list, i);
10621 		bmap = isl_basic_map_intersect(bmap, bmap_i);
10622 	}
10623 
10624 	isl_basic_map_list_free(list);
10625 	return bmap;
10626 error:
10627 	isl_basic_map_list_free(list);
10628 	return NULL;
10629 }
10630 
10631 /* Return the intersection of the elements in the non-empty list "list".
10632  * All elements are assumed to live in the same space.
10633  */
isl_basic_set_list_intersect(__isl_take isl_basic_set_list * list)10634 __isl_give isl_basic_set *isl_basic_set_list_intersect(
10635 	__isl_take isl_basic_set_list *list)
10636 {
10637 	return isl_basic_map_list_intersect(list);
10638 }
10639 
10640 /* Return the union of the elements of "list".
10641  * The list is required to have at least one element.
10642  */
isl_basic_set_list_union(__isl_take isl_basic_set_list * list)10643 __isl_give isl_set *isl_basic_set_list_union(
10644 	__isl_take isl_basic_set_list *list)
10645 {
10646 	int i;
10647 	isl_size n;
10648 	isl_space *space;
10649 	isl_basic_set *bset;
10650 	isl_set *set;
10651 
10652 	n = isl_basic_set_list_n_basic_set(list);
10653 	if (n < 0)
10654 		goto error;
10655 	if (n < 1)
10656 		isl_die(isl_basic_set_list_get_ctx(list), isl_error_invalid,
10657 			"expecting non-empty list", goto error);
10658 
10659 	bset = isl_basic_set_list_get_basic_set(list, 0);
10660 	space = isl_basic_set_get_space(bset);
10661 	isl_basic_set_free(bset);
10662 
10663 	set = isl_set_alloc_space(space, n, 0);
10664 	for (i = 0; i < n; ++i) {
10665 		bset = isl_basic_set_list_get_basic_set(list, i);
10666 		set = isl_set_add_basic_set(set, bset);
10667 	}
10668 
10669 	isl_basic_set_list_free(list);
10670 	return set;
10671 error:
10672 	isl_basic_set_list_free(list);
10673 	return NULL;
10674 }
10675 
10676 /* Return the union of the elements in the non-empty list "list".
10677  * All elements are assumed to live in the same space.
10678  */
isl_set_list_union(__isl_take isl_set_list * list)10679 __isl_give isl_set *isl_set_list_union(__isl_take isl_set_list *list)
10680 {
10681 	int i;
10682 	isl_size n;
10683 	isl_set *set;
10684 
10685 	n = isl_set_list_n_set(list);
10686 	if (n < 0)
10687 		goto error;
10688 	if (n < 1)
10689 		isl_die(isl_set_list_get_ctx(list), isl_error_invalid,
10690 			"expecting non-empty list", goto error);
10691 
10692 	set = isl_set_list_get_set(list, 0);
10693 	for (i = 1; i < n; ++i) {
10694 		isl_set *set_i;
10695 
10696 		set_i = isl_set_list_get_set(list, i);
10697 		set = isl_set_union(set, set_i);
10698 	}
10699 
10700 	isl_set_list_free(list);
10701 	return set;
10702 error:
10703 	isl_set_list_free(list);
10704 	return NULL;
10705 }
10706 
isl_basic_map_product(__isl_take isl_basic_map * bmap1,__isl_take isl_basic_map * bmap2)10707 __isl_give isl_basic_map *isl_basic_map_product(
10708 	__isl_take isl_basic_map *bmap1, __isl_take isl_basic_map *bmap2)
10709 {
10710 	isl_space *space_result = NULL;
10711 	struct isl_basic_map *bmap;
10712 	unsigned in1, in2, out1, out2, nparam, total, pos;
10713 	struct isl_dim_map *dim_map1, *dim_map2;
10714 
10715 	if (isl_basic_map_check_equal_params(bmap1, bmap2) < 0)
10716 		goto error;
10717 	space_result = isl_space_product(isl_space_copy(bmap1->dim),
10718 						   isl_space_copy(bmap2->dim));
10719 
10720 	in1 = isl_basic_map_dim(bmap1, isl_dim_in);
10721 	in2 = isl_basic_map_dim(bmap2, isl_dim_in);
10722 	out1 = isl_basic_map_dim(bmap1, isl_dim_out);
10723 	out2 = isl_basic_map_dim(bmap2, isl_dim_out);
10724 	nparam = isl_basic_map_dim(bmap1, isl_dim_param);
10725 
10726 	total = nparam + in1 + in2 + out1 + out2 + bmap1->n_div + bmap2->n_div;
10727 	dim_map1 = isl_dim_map_alloc(bmap1->ctx, total);
10728 	dim_map2 = isl_dim_map_alloc(bmap1->ctx, total);
10729 	isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_param, pos = 0);
10730 	isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_param, pos = 0);
10731 	isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_in, pos += nparam);
10732 	isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_in, pos += in1);
10733 	isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_out, pos += in2);
10734 	isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_out, pos += out1);
10735 	isl_dim_map_div(dim_map1, bmap1, pos += out2);
10736 	isl_dim_map_div(dim_map2, bmap2, pos += bmap1->n_div);
10737 
10738 	bmap = isl_basic_map_alloc_space(space_result,
10739 			bmap1->n_div + bmap2->n_div,
10740 			bmap1->n_eq + bmap2->n_eq,
10741 			bmap1->n_ineq + bmap2->n_ineq);
10742 	bmap = isl_basic_map_add_constraints_dim_map(bmap, bmap1, dim_map1);
10743 	bmap = isl_basic_map_add_constraints_dim_map(bmap, bmap2, dim_map2);
10744 	bmap = isl_basic_map_simplify(bmap);
10745 	return isl_basic_map_finalize(bmap);
10746 error:
10747 	isl_basic_map_free(bmap1);
10748 	isl_basic_map_free(bmap2);
10749 	return NULL;
10750 }
10751 
isl_basic_map_flat_product(__isl_take isl_basic_map * bmap1,__isl_take isl_basic_map * bmap2)10752 __isl_give isl_basic_map *isl_basic_map_flat_product(
10753 	__isl_take isl_basic_map *bmap1, __isl_take isl_basic_map *bmap2)
10754 {
10755 	isl_basic_map *prod;
10756 
10757 	prod = isl_basic_map_product(bmap1, bmap2);
10758 	prod = isl_basic_map_flatten(prod);
10759 	return prod;
10760 }
10761 
isl_basic_set_flat_product(__isl_take isl_basic_set * bset1,__isl_take isl_basic_set * bset2)10762 __isl_give isl_basic_set *isl_basic_set_flat_product(
10763 	__isl_take isl_basic_set *bset1, __isl_take isl_basic_set *bset2)
10764 {
10765 	return isl_basic_map_flat_range_product(bset1, bset2);
10766 }
10767 
isl_basic_map_domain_product(__isl_take isl_basic_map * bmap1,__isl_take isl_basic_map * bmap2)10768 __isl_give isl_basic_map *isl_basic_map_domain_product(
10769 	__isl_take isl_basic_map *bmap1, __isl_take isl_basic_map *bmap2)
10770 {
10771 	isl_space *space1, *space2;
10772 	isl_space *space_result = NULL;
10773 	isl_basic_map *bmap;
10774 	isl_size in1, in2, out, nparam;
10775 	unsigned total, pos;
10776 	struct isl_dim_map *dim_map1, *dim_map2;
10777 
10778 	in1 = isl_basic_map_dim(bmap1, isl_dim_in);
10779 	in2 = isl_basic_map_dim(bmap2, isl_dim_in);
10780 	out = isl_basic_map_dim(bmap1, isl_dim_out);
10781 	nparam = isl_basic_map_dim(bmap1, isl_dim_param);
10782 	if (in1 < 0 || in2 < 0 || out < 0 || nparam < 0)
10783 		goto error;
10784 
10785 	space1 = isl_basic_map_get_space(bmap1);
10786 	space2 = isl_basic_map_get_space(bmap2);
10787 	space_result = isl_space_domain_product(space1, space2);
10788 
10789 	total = nparam + in1 + in2 + out + bmap1->n_div + bmap2->n_div;
10790 	dim_map1 = isl_dim_map_alloc(bmap1->ctx, total);
10791 	dim_map2 = isl_dim_map_alloc(bmap1->ctx, total);
10792 	isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_param, pos = 0);
10793 	isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_param, pos = 0);
10794 	isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_in, pos += nparam);
10795 	isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_in, pos += in1);
10796 	isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_out, pos += in2);
10797 	isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_out, pos);
10798 	isl_dim_map_div(dim_map1, bmap1, pos += out);
10799 	isl_dim_map_div(dim_map2, bmap2, pos += bmap1->n_div);
10800 
10801 	bmap = isl_basic_map_alloc_space(space_result,
10802 			bmap1->n_div + bmap2->n_div,
10803 			bmap1->n_eq + bmap2->n_eq,
10804 			bmap1->n_ineq + bmap2->n_ineq);
10805 	bmap = isl_basic_map_add_constraints_dim_map(bmap, bmap1, dim_map1);
10806 	bmap = isl_basic_map_add_constraints_dim_map(bmap, bmap2, dim_map2);
10807 	bmap = isl_basic_map_simplify(bmap);
10808 	return isl_basic_map_finalize(bmap);
10809 error:
10810 	isl_basic_map_free(bmap1);
10811 	isl_basic_map_free(bmap2);
10812 	return NULL;
10813 }
10814 
isl_basic_map_range_product(__isl_take isl_basic_map * bmap1,__isl_take isl_basic_map * bmap2)10815 __isl_give isl_basic_map *isl_basic_map_range_product(
10816 	__isl_take isl_basic_map *bmap1, __isl_take isl_basic_map *bmap2)
10817 {
10818 	isl_bool rational;
10819 	isl_space *space_result = NULL;
10820 	isl_basic_map *bmap;
10821 	isl_size in, out1, out2, nparam;
10822 	unsigned total, pos;
10823 	struct isl_dim_map *dim_map1, *dim_map2;
10824 
10825 	rational = isl_basic_map_is_rational(bmap1);
10826 	if (rational >= 0 && rational)
10827 		rational = isl_basic_map_is_rational(bmap2);
10828 	in = isl_basic_map_dim(bmap1, isl_dim_in);
10829 	out1 = isl_basic_map_dim(bmap1, isl_dim_out);
10830 	out2 = isl_basic_map_dim(bmap2, isl_dim_out);
10831 	nparam = isl_basic_map_dim(bmap1, isl_dim_param);
10832 	if (in < 0 || out1 < 0 || out2 < 0 || nparam < 0 || rational < 0)
10833 		goto error;
10834 
10835 	if (isl_basic_map_check_equal_params(bmap1, bmap2) < 0)
10836 		goto error;
10837 
10838 	space_result = isl_space_range_product(isl_space_copy(bmap1->dim),
10839 					   isl_space_copy(bmap2->dim));
10840 
10841 	total = nparam + in + out1 + out2 + bmap1->n_div + bmap2->n_div;
10842 	dim_map1 = isl_dim_map_alloc(bmap1->ctx, total);
10843 	dim_map2 = isl_dim_map_alloc(bmap1->ctx, total);
10844 	isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_param, pos = 0);
10845 	isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_param, pos = 0);
10846 	isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_in, pos += nparam);
10847 	isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_in, pos);
10848 	isl_dim_map_dim(dim_map1, bmap1->dim, isl_dim_out, pos += in);
10849 	isl_dim_map_dim(dim_map2, bmap2->dim, isl_dim_out, pos += out1);
10850 	isl_dim_map_div(dim_map1, bmap1, pos += out2);
10851 	isl_dim_map_div(dim_map2, bmap2, pos += bmap1->n_div);
10852 
10853 	bmap = isl_basic_map_alloc_space(space_result,
10854 			bmap1->n_div + bmap2->n_div,
10855 			bmap1->n_eq + bmap2->n_eq,
10856 			bmap1->n_ineq + bmap2->n_ineq);
10857 	bmap = isl_basic_map_add_constraints_dim_map(bmap, bmap1, dim_map1);
10858 	bmap = isl_basic_map_add_constraints_dim_map(bmap, bmap2, dim_map2);
10859 	if (rational)
10860 		bmap = isl_basic_map_set_rational(bmap);
10861 	bmap = isl_basic_map_simplify(bmap);
10862 	return isl_basic_map_finalize(bmap);
10863 error:
10864 	isl_basic_map_free(bmap1);
10865 	isl_basic_map_free(bmap2);
10866 	return NULL;
10867 }
10868 
isl_basic_map_flat_range_product(__isl_take isl_basic_map * bmap1,__isl_take isl_basic_map * bmap2)10869 __isl_give isl_basic_map *isl_basic_map_flat_range_product(
10870 	__isl_take isl_basic_map *bmap1, __isl_take isl_basic_map *bmap2)
10871 {
10872 	isl_basic_map *prod;
10873 
10874 	prod = isl_basic_map_range_product(bmap1, bmap2);
10875 	prod = isl_basic_map_flatten_range(prod);
10876 	return prod;
10877 }
10878 
10879 /* Apply "basic_map_product" to each pair of basic maps in "map1" and "map2"
10880  * and collect the results.
10881  * The result live in the space obtained by calling "space_product"
10882  * on the spaces of "map1" and "map2".
10883  * If "remove_duplicates" is set then the result may contain duplicates
10884  * (even if the inputs do not) and so we try and remove the obvious
10885  * duplicates.
10886  */
map_product(__isl_take isl_map * map1,__isl_take isl_map * map2,__isl_give isl_space * (* space_product)(__isl_take isl_space * left,__isl_take isl_space * right),__isl_give isl_basic_map * (* basic_map_product)(__isl_take isl_basic_map * left,__isl_take isl_basic_map * right),int remove_duplicates)10887 static __isl_give isl_map *map_product(__isl_take isl_map *map1,
10888 	__isl_take isl_map *map2,
10889 	__isl_give isl_space *(*space_product)(__isl_take isl_space *left,
10890 					   __isl_take isl_space *right),
10891 	__isl_give isl_basic_map *(*basic_map_product)(
10892 		__isl_take isl_basic_map *left,
10893 		__isl_take isl_basic_map *right),
10894 	int remove_duplicates)
10895 {
10896 	unsigned flags = 0;
10897 	struct isl_map *result;
10898 	int i, j;
10899 	isl_bool m;
10900 
10901 	m = isl_map_has_equal_params(map1, map2);
10902 	if (m < 0)
10903 		goto error;
10904 	if (!m)
10905 		isl_die(isl_map_get_ctx(map1), isl_error_invalid,
10906 			"parameters don't match", goto error);
10907 
10908 	if (ISL_F_ISSET(map1, ISL_MAP_DISJOINT) &&
10909 	    ISL_F_ISSET(map2, ISL_MAP_DISJOINT))
10910 		ISL_FL_SET(flags, ISL_MAP_DISJOINT);
10911 
10912 	result = isl_map_alloc_space(space_product(isl_space_copy(map1->dim),
10913 					       isl_space_copy(map2->dim)),
10914 				map1->n * map2->n, flags);
10915 	if (!result)
10916 		goto error;
10917 	for (i = 0; i < map1->n; ++i)
10918 		for (j = 0; j < map2->n; ++j) {
10919 			struct isl_basic_map *part;
10920 			part = basic_map_product(isl_basic_map_copy(map1->p[i]),
10921 						 isl_basic_map_copy(map2->p[j]));
10922 			if (isl_basic_map_is_empty(part))
10923 				isl_basic_map_free(part);
10924 			else
10925 				result = isl_map_add_basic_map(result, part);
10926 			if (!result)
10927 				goto error;
10928 		}
10929 	if (remove_duplicates)
10930 		result = isl_map_remove_obvious_duplicates(result);
10931 	isl_map_free(map1);
10932 	isl_map_free(map2);
10933 	return result;
10934 error:
10935 	isl_map_free(map1);
10936 	isl_map_free(map2);
10937 	return NULL;
10938 }
10939 
10940 /* Given two maps A -> B and C -> D, construct a map [A -> C] -> [B -> D]
10941  */
isl_map_product(__isl_take isl_map * map1,__isl_take isl_map * map2)10942 __isl_give isl_map *isl_map_product(__isl_take isl_map *map1,
10943 	__isl_take isl_map *map2)
10944 {
10945 	isl_map_align_params_bin(&map1, &map2);
10946 	return map_product(map1, map2, &isl_space_product,
10947 			&isl_basic_map_product, 0);
10948 }
10949 
10950 /* Given two maps A -> B and C -> D, construct a map (A, C) -> (B, D)
10951  */
isl_map_flat_product(__isl_take isl_map * map1,__isl_take isl_map * map2)10952 __isl_give isl_map *isl_map_flat_product(__isl_take isl_map *map1,
10953 	__isl_take isl_map *map2)
10954 {
10955 	isl_map *prod;
10956 
10957 	prod = isl_map_product(map1, map2);
10958 	prod = isl_map_flatten(prod);
10959 	return prod;
10960 }
10961 
10962 /* Given two set A and B, construct its Cartesian product A x B.
10963  */
isl_set_product(__isl_take isl_set * set1,__isl_take isl_set * set2)10964 __isl_give isl_set *isl_set_product(__isl_take isl_set *set1,
10965 	__isl_take isl_set *set2)
10966 {
10967 	return isl_map_range_product(set1, set2);
10968 }
10969 
isl_set_flat_product(__isl_take isl_set * set1,__isl_take isl_set * set2)10970 __isl_give isl_set *isl_set_flat_product(__isl_take isl_set *set1,
10971 	__isl_take isl_set *set2)
10972 {
10973 	return isl_map_flat_range_product(set1, set2);
10974 }
10975 
10976 /* Given two maps A -> B and C -> D, construct a map [A -> C] -> (B * D)
10977  */
isl_map_domain_product(__isl_take isl_map * map1,__isl_take isl_map * map2)10978 __isl_give isl_map *isl_map_domain_product(__isl_take isl_map *map1,
10979 	__isl_take isl_map *map2)
10980 {
10981 	isl_map_align_params_bin(&map1, &map2);
10982 	return map_product(map1, map2, &isl_space_domain_product,
10983 				&isl_basic_map_domain_product, 1);
10984 }
10985 
10986 /* Given two maps A -> B and C -> D, construct a map (A * C) -> [B -> D]
10987  */
isl_map_range_product(__isl_take isl_map * map1,__isl_take isl_map * map2)10988 __isl_give isl_map *isl_map_range_product(__isl_take isl_map *map1,
10989 	__isl_take isl_map *map2)
10990 {
10991 	isl_map_align_params_bin(&map1, &map2);
10992 	return map_product(map1, map2, &isl_space_range_product,
10993 				&isl_basic_map_range_product, 1);
10994 }
10995 
10996 /* Given a map of the form [A -> B] -> [C -> D], return the map A -> C.
10997  */
isl_map_factor_domain(__isl_take isl_map * map)10998 __isl_give isl_map *isl_map_factor_domain(__isl_take isl_map *map)
10999 {
11000 	isl_space *space;
11001 	isl_size total1, keep1, total2, keep2;
11002 
11003 	total1 = isl_map_dim(map, isl_dim_in);
11004 	total2 = isl_map_dim(map, isl_dim_out);
11005 	if (total1 < 0 || total2 < 0)
11006 		return isl_map_free(map);
11007 	if (!isl_space_domain_is_wrapping(map->dim) ||
11008 	    !isl_space_range_is_wrapping(map->dim))
11009 		isl_die(isl_map_get_ctx(map), isl_error_invalid,
11010 			"not a product", return isl_map_free(map));
11011 
11012 	space = isl_map_get_space(map);
11013 	space = isl_space_factor_domain(space);
11014 	keep1 = isl_space_dim(space, isl_dim_in);
11015 	keep2 = isl_space_dim(space, isl_dim_out);
11016 	if (keep1 < 0 || keep2 < 0)
11017 		map = isl_map_free(map);
11018 	map = isl_map_project_out(map, isl_dim_in, keep1, total1 - keep1);
11019 	map = isl_map_project_out(map, isl_dim_out, keep2, total2 - keep2);
11020 	map = isl_map_reset_space(map, space);
11021 
11022 	return map;
11023 }
11024 
11025 /* Given a map of the form [A -> B] -> [C -> D], return the map B -> D.
11026  */
isl_map_factor_range(__isl_take isl_map * map)11027 __isl_give isl_map *isl_map_factor_range(__isl_take isl_map *map)
11028 {
11029 	isl_space *space;
11030 	isl_size total1, keep1, total2, keep2;
11031 
11032 	total1 = isl_map_dim(map, isl_dim_in);
11033 	total2 = isl_map_dim(map, isl_dim_out);
11034 	if (total1 < 0 || total2 < 0)
11035 		return isl_map_free(map);
11036 	if (!isl_space_domain_is_wrapping(map->dim) ||
11037 	    !isl_space_range_is_wrapping(map->dim))
11038 		isl_die(isl_map_get_ctx(map), isl_error_invalid,
11039 			"not a product", return isl_map_free(map));
11040 
11041 	space = isl_map_get_space(map);
11042 	space = isl_space_factor_range(space);
11043 	keep1 = isl_space_dim(space, isl_dim_in);
11044 	keep2 = isl_space_dim(space, isl_dim_out);
11045 	if (keep1 < 0 || keep2 < 0)
11046 		map = isl_map_free(map);
11047 	map = isl_map_project_out(map, isl_dim_in, 0, total1 - keep1);
11048 	map = isl_map_project_out(map, isl_dim_out, 0, total2 - keep2);
11049 	map = isl_map_reset_space(map, space);
11050 
11051 	return map;
11052 }
11053 
11054 /* Given a map of the form [A -> B] -> C, return the map A -> C.
11055  */
isl_map_domain_factor_domain(__isl_take isl_map * map)11056 __isl_give isl_map *isl_map_domain_factor_domain(__isl_take isl_map *map)
11057 {
11058 	isl_space *space;
11059 	isl_size total, keep;
11060 
11061 	total = isl_map_dim(map, isl_dim_in);
11062 	if (total < 0)
11063 		return isl_map_free(map);
11064 	if (!isl_space_domain_is_wrapping(map->dim))
11065 		isl_die(isl_map_get_ctx(map), isl_error_invalid,
11066 			"domain is not a product", return isl_map_free(map));
11067 
11068 	space = isl_map_get_space(map);
11069 	space = isl_space_domain_factor_domain(space);
11070 	keep = isl_space_dim(space, isl_dim_in);
11071 	if (keep < 0)
11072 		map = isl_map_free(map);
11073 	map = isl_map_project_out(map, isl_dim_in, keep, total - keep);
11074 	map = isl_map_reset_space(map, space);
11075 
11076 	return map;
11077 }
11078 
11079 /* Given a map of the form [A -> B] -> C, return the map B -> C.
11080  */
isl_map_domain_factor_range(__isl_take isl_map * map)11081 __isl_give isl_map *isl_map_domain_factor_range(__isl_take isl_map *map)
11082 {
11083 	isl_space *space;
11084 	isl_size total, keep;
11085 
11086 	total = isl_map_dim(map, isl_dim_in);
11087 	if (total < 0)
11088 		return isl_map_free(map);
11089 	if (!isl_space_domain_is_wrapping(map->dim))
11090 		isl_die(isl_map_get_ctx(map), isl_error_invalid,
11091 			"domain is not a product", return isl_map_free(map));
11092 
11093 	space = isl_map_get_space(map);
11094 	space = isl_space_domain_factor_range(space);
11095 	keep = isl_space_dim(space, isl_dim_in);
11096 	if (keep < 0)
11097 		map = isl_map_free(map);
11098 	map = isl_map_project_out(map, isl_dim_in, 0, total - keep);
11099 	map = isl_map_reset_space(map, space);
11100 
11101 	return map;
11102 }
11103 
11104 /* Given a map A -> [B -> C], extract the map A -> B.
11105  */
isl_map_range_factor_domain(__isl_take isl_map * map)11106 __isl_give isl_map *isl_map_range_factor_domain(__isl_take isl_map *map)
11107 {
11108 	isl_space *space;
11109 	isl_size total, keep;
11110 
11111 	total = isl_map_dim(map, isl_dim_out);
11112 	if (total < 0)
11113 		return isl_map_free(map);
11114 	if (!isl_space_range_is_wrapping(map->dim))
11115 		isl_die(isl_map_get_ctx(map), isl_error_invalid,
11116 			"range is not a product", return isl_map_free(map));
11117 
11118 	space = isl_map_get_space(map);
11119 	space = isl_space_range_factor_domain(space);
11120 	keep = isl_space_dim(space, isl_dim_out);
11121 	if (keep < 0)
11122 		map = isl_map_free(map);
11123 	map = isl_map_project_out(map, isl_dim_out, keep, total - keep);
11124 	map = isl_map_reset_space(map, space);
11125 
11126 	return map;
11127 }
11128 
11129 /* Given a map A -> [B -> C], extract the map A -> C.
11130  */
isl_map_range_factor_range(__isl_take isl_map * map)11131 __isl_give isl_map *isl_map_range_factor_range(__isl_take isl_map *map)
11132 {
11133 	isl_space *space;
11134 	isl_size total, keep;
11135 
11136 	total = isl_map_dim(map, isl_dim_out);
11137 	if (total < 0)
11138 		return isl_map_free(map);
11139 	if (!isl_space_range_is_wrapping(map->dim))
11140 		isl_die(isl_map_get_ctx(map), isl_error_invalid,
11141 			"range is not a product", return isl_map_free(map));
11142 
11143 	space = isl_map_get_space(map);
11144 	space = isl_space_range_factor_range(space);
11145 	keep = isl_space_dim(space, isl_dim_out);
11146 	if (keep < 0)
11147 		map = isl_map_free(map);
11148 	map = isl_map_project_out(map, isl_dim_out, 0, total - keep);
11149 	map = isl_map_reset_space(map, space);
11150 
11151 	return map;
11152 }
11153 
11154 /* Given two maps A -> B and C -> D, construct a map (A, C) -> (B * D)
11155  */
isl_map_flat_domain_product(__isl_take isl_map * map1,__isl_take isl_map * map2)11156 __isl_give isl_map *isl_map_flat_domain_product(__isl_take isl_map *map1,
11157 	__isl_take isl_map *map2)
11158 {
11159 	isl_map *prod;
11160 
11161 	prod = isl_map_domain_product(map1, map2);
11162 	prod = isl_map_flatten_domain(prod);
11163 	return prod;
11164 }
11165 
11166 /* Given two maps A -> B and C -> D, construct a map (A * C) -> (B, D)
11167  */
isl_map_flat_range_product(__isl_take isl_map * map1,__isl_take isl_map * map2)11168 __isl_give isl_map *isl_map_flat_range_product(__isl_take isl_map *map1,
11169 	__isl_take isl_map *map2)
11170 {
11171 	isl_map *prod;
11172 
11173 	prod = isl_map_range_product(map1, map2);
11174 	prod = isl_map_flatten_range(prod);
11175 	return prod;
11176 }
11177 
isl_basic_map_get_hash(__isl_keep isl_basic_map * bmap)11178 uint32_t isl_basic_map_get_hash(__isl_keep isl_basic_map *bmap)
11179 {
11180 	int i;
11181 	uint32_t hash = isl_hash_init();
11182 	isl_size total;
11183 
11184 	if (!bmap)
11185 		return 0;
11186 	bmap = isl_basic_map_copy(bmap);
11187 	bmap = isl_basic_map_normalize(bmap);
11188 	total = isl_basic_map_dim(bmap, isl_dim_all);
11189 	if (total < 0)
11190 		return 0;
11191 	isl_hash_byte(hash, bmap->n_eq & 0xFF);
11192 	for (i = 0; i < bmap->n_eq; ++i) {
11193 		uint32_t c_hash;
11194 		c_hash = isl_seq_get_hash(bmap->eq[i], 1 + total);
11195 		isl_hash_hash(hash, c_hash);
11196 	}
11197 	isl_hash_byte(hash, bmap->n_ineq & 0xFF);
11198 	for (i = 0; i < bmap->n_ineq; ++i) {
11199 		uint32_t c_hash;
11200 		c_hash = isl_seq_get_hash(bmap->ineq[i], 1 + total);
11201 		isl_hash_hash(hash, c_hash);
11202 	}
11203 	isl_hash_byte(hash, bmap->n_div & 0xFF);
11204 	for (i = 0; i < bmap->n_div; ++i) {
11205 		uint32_t c_hash;
11206 		if (isl_int_is_zero(bmap->div[i][0]))
11207 			continue;
11208 		isl_hash_byte(hash, i & 0xFF);
11209 		c_hash = isl_seq_get_hash(bmap->div[i], 1 + 1 + total);
11210 		isl_hash_hash(hash, c_hash);
11211 	}
11212 	isl_basic_map_free(bmap);
11213 	return hash;
11214 }
11215 
isl_basic_set_get_hash(__isl_keep isl_basic_set * bset)11216 uint32_t isl_basic_set_get_hash(__isl_keep isl_basic_set *bset)
11217 {
11218 	return isl_basic_map_get_hash(bset_to_bmap(bset));
11219 }
11220 
isl_map_get_hash(__isl_keep isl_map * map)11221 uint32_t isl_map_get_hash(__isl_keep isl_map *map)
11222 {
11223 	int i;
11224 	uint32_t hash;
11225 
11226 	if (!map)
11227 		return 0;
11228 	map = isl_map_copy(map);
11229 	map = isl_map_normalize(map);
11230 	if (!map)
11231 		return 0;
11232 
11233 	hash = isl_hash_init();
11234 	for (i = 0; i < map->n; ++i) {
11235 		uint32_t bmap_hash;
11236 		bmap_hash = isl_basic_map_get_hash(map->p[i]);
11237 		isl_hash_hash(hash, bmap_hash);
11238 	}
11239 
11240 	isl_map_free(map);
11241 
11242 	return hash;
11243 }
11244 
isl_set_get_hash(__isl_keep isl_set * set)11245 uint32_t isl_set_get_hash(__isl_keep isl_set *set)
11246 {
11247 	return isl_map_get_hash(set_to_map(set));
11248 }
11249 
11250 /* Return the number of basic maps in the (current) representation of "map".
11251  */
isl_map_n_basic_map(__isl_keep isl_map * map)11252 isl_size isl_map_n_basic_map(__isl_keep isl_map *map)
11253 {
11254 	return map ? map->n : isl_size_error;
11255 }
11256 
isl_set_n_basic_set(__isl_keep isl_set * set)11257 isl_size isl_set_n_basic_set(__isl_keep isl_set *set)
11258 {
11259 	return set ? set->n : isl_size_error;
11260 }
11261 
isl_map_foreach_basic_map(__isl_keep isl_map * map,isl_stat (* fn)(__isl_take isl_basic_map * bmap,void * user),void * user)11262 isl_stat isl_map_foreach_basic_map(__isl_keep isl_map *map,
11263 	isl_stat (*fn)(__isl_take isl_basic_map *bmap, void *user), void *user)
11264 {
11265 	int i;
11266 
11267 	if (!map)
11268 		return isl_stat_error;
11269 
11270 	for (i = 0; i < map->n; ++i)
11271 		if (fn(isl_basic_map_copy(map->p[i]), user) < 0)
11272 			return isl_stat_error;
11273 
11274 	return isl_stat_ok;
11275 }
11276 
isl_set_foreach_basic_set(__isl_keep isl_set * set,isl_stat (* fn)(__isl_take isl_basic_set * bset,void * user),void * user)11277 isl_stat isl_set_foreach_basic_set(__isl_keep isl_set *set,
11278 	isl_stat (*fn)(__isl_take isl_basic_set *bset, void *user), void *user)
11279 {
11280 	int i;
11281 
11282 	if (!set)
11283 		return isl_stat_error;
11284 
11285 	for (i = 0; i < set->n; ++i)
11286 		if (fn(isl_basic_set_copy(set->p[i]), user) < 0)
11287 			return isl_stat_error;
11288 
11289 	return isl_stat_ok;
11290 }
11291 
11292 /* Does "test" succeed on every basic set in "set"?
11293  */
isl_set_every_basic_set(__isl_keep isl_set * set,isl_bool (* test)(__isl_keep isl_basic_set * bset,void * user),void * user)11294 isl_bool isl_set_every_basic_set(__isl_keep isl_set *set,
11295 	isl_bool (*test)(__isl_keep isl_basic_set *bset, void *user),
11296 	void *user)
11297 {
11298 	int i;
11299 
11300 	if (!set)
11301 		return isl_bool_error;
11302 
11303 	for (i = 0; i < set->n; ++i) {
11304 		isl_bool r;
11305 
11306 		r = test(set->p[i], user);
11307 		if (r < 0 || !r)
11308 			return r;
11309 	}
11310 
11311 	return isl_bool_true;
11312 }
11313 
11314 /* Return a list of basic sets, the union of which is equal to "set".
11315  */
isl_set_get_basic_set_list(__isl_keep isl_set * set)11316 __isl_give isl_basic_set_list *isl_set_get_basic_set_list(
11317 	__isl_keep isl_set *set)
11318 {
11319 	int i;
11320 	isl_basic_set_list *list;
11321 
11322 	if (!set)
11323 		return NULL;
11324 
11325 	list = isl_basic_set_list_alloc(isl_set_get_ctx(set), set->n);
11326 	for (i = 0; i < set->n; ++i) {
11327 		isl_basic_set *bset;
11328 
11329 		bset = isl_basic_set_copy(set->p[i]);
11330 		list = isl_basic_set_list_add(list, bset);
11331 	}
11332 
11333 	return list;
11334 }
11335 
isl_basic_set_lift(__isl_take isl_basic_set * bset)11336 __isl_give isl_basic_set *isl_basic_set_lift(__isl_take isl_basic_set *bset)
11337 {
11338 	isl_space *space;
11339 
11340 	if (!bset)
11341 		return NULL;
11342 
11343 	bset = isl_basic_set_cow(bset);
11344 	if (!bset)
11345 		return NULL;
11346 
11347 	space = isl_basic_set_get_space(bset);
11348 	space = isl_space_lift(space, bset->n_div);
11349 	if (!space)
11350 		goto error;
11351 	isl_space_free(bset->dim);
11352 	bset->dim = space;
11353 	bset->extra -= bset->n_div;
11354 	bset->n_div = 0;
11355 
11356 	bset = isl_basic_set_finalize(bset);
11357 
11358 	return bset;
11359 error:
11360 	isl_basic_set_free(bset);
11361 	return NULL;
11362 }
11363 
isl_set_lift(__isl_take isl_set * set)11364 __isl_give isl_set *isl_set_lift(__isl_take isl_set *set)
11365 {
11366 	int i;
11367 	isl_space *space;
11368 	unsigned n_div;
11369 
11370 	set = set_from_map(isl_map_align_divs_internal(set_to_map(set)));
11371 
11372 	if (!set)
11373 		return NULL;
11374 
11375 	set = isl_set_cow(set);
11376 	if (!set)
11377 		return NULL;
11378 
11379 	n_div = set->p[0]->n_div;
11380 	space = isl_set_get_space(set);
11381 	space = isl_space_lift(space, n_div);
11382 	if (!space)
11383 		goto error;
11384 	isl_space_free(set->dim);
11385 	set->dim = space;
11386 
11387 	for (i = 0; i < set->n; ++i) {
11388 		set->p[i] = isl_basic_set_lift(set->p[i]);
11389 		if (!set->p[i])
11390 			goto error;
11391 	}
11392 
11393 	return set;
11394 error:
11395 	isl_set_free(set);
11396 	return NULL;
11397 }
11398 
isl_basic_set_size(__isl_keep isl_basic_set * bset)11399 int isl_basic_set_size(__isl_keep isl_basic_set *bset)
11400 {
11401 	isl_size dim;
11402 	int size = 0;
11403 
11404 	dim = isl_basic_set_dim(bset, isl_dim_all);
11405 	if (dim < 0)
11406 		return -1;
11407 	size += bset->n_eq * (1 + dim);
11408 	size += bset->n_ineq * (1 + dim);
11409 	size += bset->n_div * (2 + dim);
11410 
11411 	return size;
11412 }
11413 
isl_set_size(__isl_keep isl_set * set)11414 int isl_set_size(__isl_keep isl_set *set)
11415 {
11416 	int i;
11417 	int size = 0;
11418 
11419 	if (!set)
11420 		return -1;
11421 
11422 	for (i = 0; i < set->n; ++i)
11423 		size += isl_basic_set_size(set->p[i]);
11424 
11425 	return size;
11426 }
11427 
11428 /* Check if there is any lower bound (if lower == 0) and/or upper
11429  * bound (if upper == 0) on the specified dim.
11430  */
basic_map_dim_is_bounded(__isl_keep isl_basic_map * bmap,enum isl_dim_type type,unsigned pos,int lower,int upper)11431 static isl_bool basic_map_dim_is_bounded(__isl_keep isl_basic_map *bmap,
11432 	enum isl_dim_type type, unsigned pos, int lower, int upper)
11433 {
11434 	int i;
11435 
11436 	if (isl_basic_map_check_range(bmap, type, pos, 1) < 0)
11437 		return isl_bool_error;
11438 
11439 	pos += isl_basic_map_offset(bmap, type);
11440 
11441 	for (i = 0; i < bmap->n_div; ++i) {
11442 		if (isl_int_is_zero(bmap->div[i][0]))
11443 			continue;
11444 		if (!isl_int_is_zero(bmap->div[i][1 + pos]))
11445 			return isl_bool_true;
11446 	}
11447 
11448 	for (i = 0; i < bmap->n_eq; ++i)
11449 		if (!isl_int_is_zero(bmap->eq[i][pos]))
11450 			return isl_bool_true;
11451 
11452 	for (i = 0; i < bmap->n_ineq; ++i) {
11453 		int sgn = isl_int_sgn(bmap->ineq[i][pos]);
11454 		if (sgn > 0)
11455 			lower = 1;
11456 		if (sgn < 0)
11457 			upper = 1;
11458 	}
11459 
11460 	return lower && upper;
11461 }
11462 
isl_basic_map_dim_is_bounded(__isl_keep isl_basic_map * bmap,enum isl_dim_type type,unsigned pos)11463 isl_bool isl_basic_map_dim_is_bounded(__isl_keep isl_basic_map *bmap,
11464 	enum isl_dim_type type, unsigned pos)
11465 {
11466 	return basic_map_dim_is_bounded(bmap, type, pos, 0, 0);
11467 }
11468 
isl_basic_map_dim_has_lower_bound(__isl_keep isl_basic_map * bmap,enum isl_dim_type type,unsigned pos)11469 isl_bool isl_basic_map_dim_has_lower_bound(__isl_keep isl_basic_map *bmap,
11470 	enum isl_dim_type type, unsigned pos)
11471 {
11472 	return basic_map_dim_is_bounded(bmap, type, pos, 0, 1);
11473 }
11474 
isl_basic_map_dim_has_upper_bound(__isl_keep isl_basic_map * bmap,enum isl_dim_type type,unsigned pos)11475 isl_bool isl_basic_map_dim_has_upper_bound(__isl_keep isl_basic_map *bmap,
11476 	enum isl_dim_type type, unsigned pos)
11477 {
11478 	return basic_map_dim_is_bounded(bmap, type, pos, 1, 0);
11479 }
11480 
isl_map_dim_is_bounded(__isl_keep isl_map * map,enum isl_dim_type type,unsigned pos)11481 isl_bool isl_map_dim_is_bounded(__isl_keep isl_map *map,
11482 	enum isl_dim_type type, unsigned pos)
11483 {
11484 	int i;
11485 
11486 	if (!map)
11487 		return isl_bool_error;
11488 
11489 	for (i = 0; i < map->n; ++i) {
11490 		isl_bool bounded;
11491 		bounded = isl_basic_map_dim_is_bounded(map->p[i], type, pos);
11492 		if (bounded < 0 || !bounded)
11493 			return bounded;
11494 	}
11495 
11496 	return isl_bool_true;
11497 }
11498 
11499 /* Return true if the specified dim is involved in both an upper bound
11500  * and a lower bound.
11501  */
isl_set_dim_is_bounded(__isl_keep isl_set * set,enum isl_dim_type type,unsigned pos)11502 isl_bool isl_set_dim_is_bounded(__isl_keep isl_set *set,
11503 	enum isl_dim_type type, unsigned pos)
11504 {
11505 	return isl_map_dim_is_bounded(set_to_map(set), type, pos);
11506 }
11507 
11508 /* Does "map" have a bound (according to "fn") for any of its basic maps?
11509  */
has_any_bound(__isl_keep isl_map * map,enum isl_dim_type type,unsigned pos,isl_bool (* fn)(__isl_keep isl_basic_map * bmap,enum isl_dim_type type,unsigned pos))11510 static isl_bool has_any_bound(__isl_keep isl_map *map,
11511 	enum isl_dim_type type, unsigned pos,
11512 	isl_bool (*fn)(__isl_keep isl_basic_map *bmap,
11513 		  enum isl_dim_type type, unsigned pos))
11514 {
11515 	int i;
11516 
11517 	if (!map)
11518 		return isl_bool_error;
11519 
11520 	for (i = 0; i < map->n; ++i) {
11521 		isl_bool bounded;
11522 		bounded = fn(map->p[i], type, pos);
11523 		if (bounded < 0 || bounded)
11524 			return bounded;
11525 	}
11526 
11527 	return isl_bool_false;
11528 }
11529 
11530 /* Return 1 if the specified dim is involved in any lower bound.
11531  */
isl_set_dim_has_any_lower_bound(__isl_keep isl_set * set,enum isl_dim_type type,unsigned pos)11532 isl_bool isl_set_dim_has_any_lower_bound(__isl_keep isl_set *set,
11533 	enum isl_dim_type type, unsigned pos)
11534 {
11535 	return has_any_bound(set, type, pos,
11536 				&isl_basic_map_dim_has_lower_bound);
11537 }
11538 
11539 /* Return 1 if the specified dim is involved in any upper bound.
11540  */
isl_set_dim_has_any_upper_bound(__isl_keep isl_set * set,enum isl_dim_type type,unsigned pos)11541 isl_bool isl_set_dim_has_any_upper_bound(__isl_keep isl_set *set,
11542 	enum isl_dim_type type, unsigned pos)
11543 {
11544 	return has_any_bound(set, type, pos,
11545 				&isl_basic_map_dim_has_upper_bound);
11546 }
11547 
11548 /* Does "map" have a bound (according to "fn") for all of its basic maps?
11549  */
has_bound(__isl_keep isl_map * map,enum isl_dim_type type,unsigned pos,isl_bool (* fn)(__isl_keep isl_basic_map * bmap,enum isl_dim_type type,unsigned pos))11550 static isl_bool has_bound(__isl_keep isl_map *map,
11551 	enum isl_dim_type type, unsigned pos,
11552 	isl_bool (*fn)(__isl_keep isl_basic_map *bmap,
11553 		  enum isl_dim_type type, unsigned pos))
11554 {
11555 	int i;
11556 
11557 	if (!map)
11558 		return isl_bool_error;
11559 
11560 	for (i = 0; i < map->n; ++i) {
11561 		isl_bool bounded;
11562 		bounded = fn(map->p[i], type, pos);
11563 		if (bounded < 0 || !bounded)
11564 			return bounded;
11565 	}
11566 
11567 	return isl_bool_true;
11568 }
11569 
11570 /* Return 1 if the specified dim has a lower bound (in each of its basic sets).
11571  */
isl_set_dim_has_lower_bound(__isl_keep isl_set * set,enum isl_dim_type type,unsigned pos)11572 isl_bool isl_set_dim_has_lower_bound(__isl_keep isl_set *set,
11573 	enum isl_dim_type type, unsigned pos)
11574 {
11575 	return has_bound(set, type, pos, &isl_basic_map_dim_has_lower_bound);
11576 }
11577 
11578 /* Return 1 if the specified dim has an upper bound (in each of its basic sets).
11579  */
isl_set_dim_has_upper_bound(__isl_keep isl_set * set,enum isl_dim_type type,unsigned pos)11580 isl_bool isl_set_dim_has_upper_bound(__isl_keep isl_set *set,
11581 	enum isl_dim_type type, unsigned pos)
11582 {
11583 	return has_bound(set, type, pos, &isl_basic_map_dim_has_upper_bound);
11584 }
11585 
11586 /* For each of the "n" variables starting at "first", determine
11587  * the sign of the variable and put the results in the first "n"
11588  * elements of the array "signs".
11589  * Sign
11590  *	1 means that the variable is non-negative
11591  *	-1 means that the variable is non-positive
11592  *	0 means the variable attains both positive and negative values.
11593  */
isl_basic_set_vars_get_sign(__isl_keep isl_basic_set * bset,unsigned first,unsigned n,int * signs)11594 isl_stat isl_basic_set_vars_get_sign(__isl_keep isl_basic_set *bset,
11595 	unsigned first, unsigned n, int *signs)
11596 {
11597 	isl_vec *bound = NULL;
11598 	struct isl_tab *tab = NULL;
11599 	struct isl_tab_undo *snap;
11600 	int i;
11601 	isl_size total;
11602 
11603 	total = isl_basic_set_dim(bset, isl_dim_all);
11604 	if (total < 0 || !signs)
11605 		return isl_stat_error;
11606 
11607 	bound = isl_vec_alloc(bset->ctx, 1 + total);
11608 	tab = isl_tab_from_basic_set(bset, 0);
11609 	if (!bound || !tab)
11610 		goto error;
11611 
11612 	isl_seq_clr(bound->el, bound->size);
11613 	isl_int_set_si(bound->el[0], -1);
11614 
11615 	snap = isl_tab_snap(tab);
11616 	for (i = 0; i < n; ++i) {
11617 		int empty;
11618 
11619 		isl_int_set_si(bound->el[1 + first + i], -1);
11620 		if (isl_tab_add_ineq(tab, bound->el) < 0)
11621 			goto error;
11622 		empty = tab->empty;
11623 		isl_int_set_si(bound->el[1 + first + i], 0);
11624 		if (isl_tab_rollback(tab, snap) < 0)
11625 			goto error;
11626 
11627 		if (empty) {
11628 			signs[i] = 1;
11629 			continue;
11630 		}
11631 
11632 		isl_int_set_si(bound->el[1 + first + i], 1);
11633 		if (isl_tab_add_ineq(tab, bound->el) < 0)
11634 			goto error;
11635 		empty = tab->empty;
11636 		isl_int_set_si(bound->el[1 + first + i], 0);
11637 		if (isl_tab_rollback(tab, snap) < 0)
11638 			goto error;
11639 
11640 		signs[i] = empty ? -1 : 0;
11641 	}
11642 
11643 	isl_tab_free(tab);
11644 	isl_vec_free(bound);
11645 	return isl_stat_ok;
11646 error:
11647 	isl_tab_free(tab);
11648 	isl_vec_free(bound);
11649 	return isl_stat_error;
11650 }
11651 
isl_basic_set_dims_get_sign(__isl_keep isl_basic_set * bset,enum isl_dim_type type,unsigned first,unsigned n,int * signs)11652 isl_stat isl_basic_set_dims_get_sign(__isl_keep isl_basic_set *bset,
11653 	enum isl_dim_type type, unsigned first, unsigned n, int *signs)
11654 {
11655 	if (!bset || !signs)
11656 		return isl_stat_error;
11657 	if (isl_basic_set_check_range(bset, type, first, n) < 0)
11658 		return isl_stat_error;
11659 
11660 	first += pos(bset->dim, type) - 1;
11661 	return isl_basic_set_vars_get_sign(bset, first, n, signs);
11662 }
11663 
11664 /* Is it possible for the integer division "div" to depend (possibly
11665  * indirectly) on any output dimensions?
11666  *
11667  * If the div is undefined, then we conservatively assume that it
11668  * may depend on them.
11669  * Otherwise, we check if it actually depends on them or on any integer
11670  * divisions that may depend on them.
11671  */
div_may_involve_output(__isl_keep isl_basic_map * bmap,int div)11672 static isl_bool div_may_involve_output(__isl_keep isl_basic_map *bmap, int div)
11673 {
11674 	int i;
11675 	isl_size n_out, n_div;
11676 	unsigned o_out, o_div;
11677 
11678 	if (isl_int_is_zero(bmap->div[div][0]))
11679 		return isl_bool_true;
11680 
11681 	n_out = isl_basic_map_dim(bmap, isl_dim_out);
11682 	if (n_out < 0)
11683 		return isl_bool_error;
11684 	o_out = isl_basic_map_offset(bmap, isl_dim_out);
11685 
11686 	if (isl_seq_first_non_zero(bmap->div[div] + 1 + o_out, n_out) != -1)
11687 		return isl_bool_true;
11688 
11689 	n_div = isl_basic_map_dim(bmap, isl_dim_div);
11690 	if (n_div < 0)
11691 		return isl_bool_error;
11692 	o_div = isl_basic_map_offset(bmap, isl_dim_div);
11693 
11694 	for (i = 0; i < n_div; ++i) {
11695 		isl_bool may_involve;
11696 
11697 		if (isl_int_is_zero(bmap->div[div][1 + o_div + i]))
11698 			continue;
11699 		may_involve = div_may_involve_output(bmap, i);
11700 		if (may_involve < 0 || may_involve)
11701 			return may_involve;
11702 	}
11703 
11704 	return isl_bool_false;
11705 }
11706 
11707 /* Return the first integer division of "bmap" in the range
11708  * [first, first + n[ that may depend on any output dimensions and
11709  * that has a non-zero coefficient in "c" (where the first coefficient
11710  * in "c" corresponds to integer division "first").
11711  */
first_div_may_involve_output(__isl_keep isl_basic_map * bmap,isl_int * c,int first,int n)11712 static int first_div_may_involve_output(__isl_keep isl_basic_map *bmap,
11713 	isl_int *c, int first, int n)
11714 {
11715 	int k;
11716 
11717 	if (!bmap)
11718 		return -1;
11719 
11720 	for (k = first; k < first + n; ++k) {
11721 		isl_bool may_involve;
11722 
11723 		if (isl_int_is_zero(c[k]))
11724 			continue;
11725 		may_involve = div_may_involve_output(bmap, k);
11726 		if (may_involve < 0)
11727 			return -1;
11728 		if (may_involve)
11729 			return k;
11730 	}
11731 
11732 	return first + n;
11733 }
11734 
11735 /* Look for a pair of inequality constraints in "bmap" of the form
11736  *
11737  *	-l + i >= 0		or		i >= l
11738  * and
11739  *	n + l - i >= 0		or		i <= l + n
11740  *
11741  * with n < "m" and i the output dimension at position "pos".
11742  * (Note that n >= 0 as otherwise the two constraints would conflict.)
11743  * Furthermore, "l" is only allowed to involve parameters, input dimensions
11744  * and earlier output dimensions, as well as integer divisions that do
11745  * not involve any of the output dimensions.
11746  *
11747  * Return the index of the first inequality constraint or bmap->n_ineq
11748  * if no such pair can be found.
11749  */
find_modulo_constraint_pair(__isl_keep isl_basic_map * bmap,int pos,isl_int m)11750 static int find_modulo_constraint_pair(__isl_keep isl_basic_map *bmap,
11751 	int pos, isl_int m)
11752 {
11753 	int i, j;
11754 	isl_ctx *ctx;
11755 	isl_size total;
11756 	isl_size n_div, n_out;
11757 	unsigned o_div, o_out;
11758 	int less;
11759 
11760 	total = isl_basic_map_dim(bmap, isl_dim_all);
11761 	n_out = isl_basic_map_dim(bmap, isl_dim_out);
11762 	n_div = isl_basic_map_dim(bmap, isl_dim_div);
11763 	if (total < 0 || n_out < 0 || n_div < 0)
11764 		return -1;
11765 
11766 	ctx = isl_basic_map_get_ctx(bmap);
11767 	o_out = isl_basic_map_offset(bmap, isl_dim_out);
11768 	o_div = isl_basic_map_offset(bmap, isl_dim_div);
11769 	for (i = 0; i < bmap->n_ineq; ++i) {
11770 		if (!isl_int_abs_eq(bmap->ineq[i][o_out + pos], ctx->one))
11771 			continue;
11772 		if (isl_seq_first_non_zero(bmap->ineq[i] + o_out + pos + 1,
11773 					n_out - (pos + 1)) != -1)
11774 			continue;
11775 		if (first_div_may_involve_output(bmap, bmap->ineq[i] + o_div,
11776 						0, n_div) < n_div)
11777 			continue;
11778 		for (j = i + 1; j < bmap->n_ineq; ++j) {
11779 			if (!isl_int_abs_eq(bmap->ineq[j][o_out + pos],
11780 					    ctx->one))
11781 				continue;
11782 			if (!isl_seq_is_neg(bmap->ineq[i] + 1,
11783 					    bmap->ineq[j] + 1, total))
11784 				continue;
11785 			break;
11786 		}
11787 		if (j >= bmap->n_ineq)
11788 			continue;
11789 		isl_int_add(bmap->ineq[i][0],
11790 			    bmap->ineq[i][0], bmap->ineq[j][0]);
11791 		less = isl_int_abs_lt(bmap->ineq[i][0], m);
11792 		isl_int_sub(bmap->ineq[i][0],
11793 			    bmap->ineq[i][0], bmap->ineq[j][0]);
11794 		if (!less)
11795 			continue;
11796 		if (isl_int_is_one(bmap->ineq[i][o_out + pos]))
11797 			return i;
11798 		else
11799 			return j;
11800 	}
11801 
11802 	return bmap->n_ineq;
11803 }
11804 
11805 /* Return the index of the equality of "bmap" that defines
11806  * the output dimension "pos" in terms of earlier dimensions.
11807  * The equality may also involve integer divisions, as long
11808  * as those integer divisions are defined in terms of
11809  * parameters or input dimensions.
11810  * In this case, *div is set to the number of integer divisions and
11811  * *ineq is set to the number of inequality constraints (provided
11812  * div and ineq are not NULL).
11813  *
11814  * The equality may also involve a single integer division involving
11815  * the output dimensions (typically only output dimension "pos") as
11816  * long as the coefficient of output dimension "pos" is 1 or -1 and
11817  * there is a pair of constraints i >= l and i <= l + n, with i referring
11818  * to output dimension "pos", l an expression involving only earlier
11819  * dimensions and n smaller than the coefficient of the integer division
11820  * in the equality.  In this case, the output dimension can be defined
11821  * in terms of a modulo expression that does not involve the integer division.
11822  * *div is then set to this single integer division and
11823  * *ineq is set to the index of constraint i >= l.
11824  *
11825  * Return bmap->n_eq if there is no such equality.
11826  * Return -1 on error.
11827  */
isl_basic_map_output_defining_equality(__isl_keep isl_basic_map * bmap,int pos,int * div,int * ineq)11828 int isl_basic_map_output_defining_equality(__isl_keep isl_basic_map *bmap,
11829 	int pos, int *div, int *ineq)
11830 {
11831 	int j, k, l;
11832 	isl_size n_div, n_out;
11833 	unsigned o_div, o_out;
11834 
11835 	n_out = isl_basic_map_dim(bmap, isl_dim_out);
11836 	n_div = isl_basic_map_dim(bmap, isl_dim_div);
11837 	if (n_out < 0 || n_div < 0)
11838 		return -1;
11839 
11840 	o_out = isl_basic_map_offset(bmap, isl_dim_out);
11841 	o_div = isl_basic_map_offset(bmap, isl_dim_div);
11842 
11843 	if (ineq)
11844 		*ineq = bmap->n_ineq;
11845 	if (div)
11846 		*div = n_div;
11847 	for (j = 0; j < bmap->n_eq; ++j) {
11848 		if (isl_int_is_zero(bmap->eq[j][o_out + pos]))
11849 			continue;
11850 		if (isl_seq_first_non_zero(bmap->eq[j] + o_out + pos + 1,
11851 					n_out - (pos + 1)) != -1)
11852 			continue;
11853 		k = first_div_may_involve_output(bmap, bmap->eq[j] + o_div,
11854 						0, n_div);
11855 		if (k >= n_div)
11856 			return j;
11857 		if (!isl_int_is_one(bmap->eq[j][o_out + pos]) &&
11858 		    !isl_int_is_negone(bmap->eq[j][o_out + pos]))
11859 			continue;
11860 		if (first_div_may_involve_output(bmap, bmap->eq[j] + o_div,
11861 						k + 1, n_div - (k+1)) < n_div)
11862 			continue;
11863 		l = find_modulo_constraint_pair(bmap, pos,
11864 						bmap->eq[j][o_div + k]);
11865 		if (l < 0)
11866 			return -1;
11867 		if (l >= bmap->n_ineq)
11868 			continue;
11869 		if (div)
11870 			*div = k;
11871 		if (ineq)
11872 			*ineq = l;
11873 		return j;
11874 	}
11875 
11876 	return bmap->n_eq;
11877 }
11878 
11879 /* Check if the given basic map is obviously single-valued.
11880  * In particular, for each output dimension, check that there is
11881  * an equality that defines the output dimension in terms of
11882  * earlier dimensions.
11883  */
isl_basic_map_plain_is_single_valued(__isl_keep isl_basic_map * bmap)11884 isl_bool isl_basic_map_plain_is_single_valued(__isl_keep isl_basic_map *bmap)
11885 {
11886 	int i;
11887 	isl_size n_out;
11888 
11889 	n_out = isl_basic_map_dim(bmap, isl_dim_out);
11890 	if (n_out < 0)
11891 		return isl_bool_error;
11892 
11893 	for (i = 0; i < n_out; ++i) {
11894 		int eq;
11895 
11896 		eq = isl_basic_map_output_defining_equality(bmap, i,
11897 							    NULL, NULL);
11898 		if (eq < 0)
11899 			return isl_bool_error;
11900 		if (eq >= bmap->n_eq)
11901 			return isl_bool_false;
11902 	}
11903 
11904 	return isl_bool_true;
11905 }
11906 
11907 /* Check if the given basic map is single-valued.
11908  * We simply compute
11909  *
11910  *	M \circ M^-1
11911  *
11912  * and check if the result is a subset of the identity mapping.
11913  */
isl_basic_map_is_single_valued(__isl_keep isl_basic_map * bmap)11914 isl_bool isl_basic_map_is_single_valued(__isl_keep isl_basic_map *bmap)
11915 {
11916 	isl_space *space;
11917 	isl_basic_map *test;
11918 	isl_basic_map *id;
11919 	isl_bool sv;
11920 
11921 	sv = isl_basic_map_plain_is_single_valued(bmap);
11922 	if (sv < 0 || sv)
11923 		return sv;
11924 
11925 	test = isl_basic_map_reverse(isl_basic_map_copy(bmap));
11926 	test = isl_basic_map_apply_range(test, isl_basic_map_copy(bmap));
11927 
11928 	space = isl_basic_map_get_space(bmap);
11929 	space = isl_space_map_from_set(isl_space_range(space));
11930 	id = isl_basic_map_identity(space);
11931 
11932 	sv = isl_basic_map_is_subset(test, id);
11933 
11934 	isl_basic_map_free(test);
11935 	isl_basic_map_free(id);
11936 
11937 	return sv;
11938 }
11939 
11940 /* Check if the given map is obviously single-valued.
11941  */
isl_map_plain_is_single_valued(__isl_keep isl_map * map)11942 isl_bool isl_map_plain_is_single_valued(__isl_keep isl_map *map)
11943 {
11944 	if (!map)
11945 		return isl_bool_error;
11946 	if (map->n == 0)
11947 		return isl_bool_true;
11948 	if (map->n >= 2)
11949 		return isl_bool_false;
11950 
11951 	return isl_basic_map_plain_is_single_valued(map->p[0]);
11952 }
11953 
11954 /* Check if the given map is single-valued.
11955  * We simply compute
11956  *
11957  *	M \circ M^-1
11958  *
11959  * and check if the result is a subset of the identity mapping.
11960  */
isl_map_is_single_valued(__isl_keep isl_map * map)11961 isl_bool isl_map_is_single_valued(__isl_keep isl_map *map)
11962 {
11963 	isl_space *space;
11964 	isl_map *test;
11965 	isl_map *id;
11966 	isl_bool sv;
11967 
11968 	sv = isl_map_plain_is_single_valued(map);
11969 	if (sv < 0 || sv)
11970 		return sv;
11971 
11972 	test = isl_map_reverse(isl_map_copy(map));
11973 	test = isl_map_apply_range(test, isl_map_copy(map));
11974 
11975 	space = isl_space_map_from_set(isl_space_range(isl_map_get_space(map)));
11976 	id = isl_map_identity(space);
11977 
11978 	sv = isl_map_is_subset(test, id);
11979 
11980 	isl_map_free(test);
11981 	isl_map_free(id);
11982 
11983 	return sv;
11984 }
11985 
isl_map_is_injective(__isl_keep isl_map * map)11986 isl_bool isl_map_is_injective(__isl_keep isl_map *map)
11987 {
11988 	isl_bool in;
11989 
11990 	map = isl_map_copy(map);
11991 	map = isl_map_reverse(map);
11992 	in = isl_map_is_single_valued(map);
11993 	isl_map_free(map);
11994 
11995 	return in;
11996 }
11997 
11998 /* Check if the given map is obviously injective.
11999  */
isl_map_plain_is_injective(__isl_keep isl_map * map)12000 isl_bool isl_map_plain_is_injective(__isl_keep isl_map *map)
12001 {
12002 	isl_bool in;
12003 
12004 	map = isl_map_copy(map);
12005 	map = isl_map_reverse(map);
12006 	in = isl_map_plain_is_single_valued(map);
12007 	isl_map_free(map);
12008 
12009 	return in;
12010 }
12011 
isl_map_is_bijective(__isl_keep isl_map * map)12012 isl_bool isl_map_is_bijective(__isl_keep isl_map *map)
12013 {
12014 	isl_bool sv;
12015 
12016 	sv = isl_map_is_single_valued(map);
12017 	if (sv < 0 || !sv)
12018 		return sv;
12019 
12020 	return isl_map_is_injective(map);
12021 }
12022 
isl_set_is_singleton(__isl_keep isl_set * set)12023 isl_bool isl_set_is_singleton(__isl_keep isl_set *set)
12024 {
12025 	return isl_map_is_single_valued(set_to_map(set));
12026 }
12027 
12028 /* Does "map" only map elements to themselves?
12029  *
12030  * If the domain and range spaces are different, then "map"
12031  * is considered not to be an identity relation, even if it is empty.
12032  * Otherwise, construct the maximal identity relation and
12033  * check whether "map" is a subset of this relation.
12034  */
isl_map_is_identity(__isl_keep isl_map * map)12035 isl_bool isl_map_is_identity(__isl_keep isl_map *map)
12036 {
12037 	isl_map *id;
12038 	isl_bool equal, is_identity;
12039 
12040 	equal = isl_map_tuple_is_equal(map, isl_dim_in, map, isl_dim_out);
12041 	if (equal < 0 || !equal)
12042 		return equal;
12043 
12044 	id = isl_map_identity(isl_map_get_space(map));
12045 	is_identity = isl_map_is_subset(map, id);
12046 	isl_map_free(id);
12047 
12048 	return is_identity;
12049 }
12050 
isl_map_is_translation(__isl_keep isl_map * map)12051 int isl_map_is_translation(__isl_keep isl_map *map)
12052 {
12053 	int ok;
12054 	isl_set *delta;
12055 
12056 	delta = isl_map_deltas(isl_map_copy(map));
12057 	ok = isl_set_is_singleton(delta);
12058 	isl_set_free(delta);
12059 
12060 	return ok;
12061 }
12062 
unique(isl_int * p,unsigned pos,unsigned len)12063 static int unique(isl_int *p, unsigned pos, unsigned len)
12064 {
12065 	if (isl_seq_first_non_zero(p, pos) != -1)
12066 		return 0;
12067 	if (isl_seq_first_non_zero(p + pos + 1, len - pos - 1) != -1)
12068 		return 0;
12069 	return 1;
12070 }
12071 
isl_basic_set_is_box(__isl_keep isl_basic_set * bset)12072 isl_bool isl_basic_set_is_box(__isl_keep isl_basic_set *bset)
12073 {
12074 	int i, j;
12075 	isl_size nvar, n_div;
12076 	unsigned ovar;
12077 
12078 	n_div = isl_basic_set_dim(bset, isl_dim_div);
12079 	if (n_div < 0)
12080 		return isl_bool_error;
12081 	if (n_div != 0)
12082 		return isl_bool_false;
12083 
12084 	nvar = isl_basic_set_dim(bset, isl_dim_set);
12085 	if (nvar < 0)
12086 		return isl_bool_error;
12087 	ovar = isl_space_offset(bset->dim, isl_dim_set);
12088 	for (j = 0; j < nvar; ++j) {
12089 		int lower = 0, upper = 0;
12090 		for (i = 0; i < bset->n_eq; ++i) {
12091 			if (isl_int_is_zero(bset->eq[i][1 + ovar + j]))
12092 				continue;
12093 			if (!unique(bset->eq[i] + 1 + ovar, j, nvar))
12094 				return isl_bool_false;
12095 			break;
12096 		}
12097 		if (i < bset->n_eq)
12098 			continue;
12099 		for (i = 0; i < bset->n_ineq; ++i) {
12100 			if (isl_int_is_zero(bset->ineq[i][1 + ovar + j]))
12101 				continue;
12102 			if (!unique(bset->ineq[i] + 1 + ovar, j, nvar))
12103 				return isl_bool_false;
12104 			if (isl_int_is_pos(bset->ineq[i][1 + ovar + j]))
12105 				lower = 1;
12106 			else
12107 				upper = 1;
12108 		}
12109 		if (!lower || !upper)
12110 			return isl_bool_false;
12111 	}
12112 
12113 	return isl_bool_true;
12114 }
12115 
isl_set_is_box(__isl_keep isl_set * set)12116 isl_bool isl_set_is_box(__isl_keep isl_set *set)
12117 {
12118 	if (!set)
12119 		return isl_bool_error;
12120 	if (set->n != 1)
12121 		return isl_bool_false;
12122 
12123 	return isl_basic_set_is_box(set->p[0]);
12124 }
12125 
isl_basic_set_is_wrapping(__isl_keep isl_basic_set * bset)12126 isl_bool isl_basic_set_is_wrapping(__isl_keep isl_basic_set *bset)
12127 {
12128 	if (!bset)
12129 		return isl_bool_error;
12130 
12131 	return isl_space_is_wrapping(bset->dim);
12132 }
12133 
isl_set_is_wrapping(__isl_keep isl_set * set)12134 isl_bool isl_set_is_wrapping(__isl_keep isl_set *set)
12135 {
12136 	if (!set)
12137 		return isl_bool_error;
12138 
12139 	return isl_space_is_wrapping(set->dim);
12140 }
12141 
12142 /* Modify the space of "map" through a call to "change".
12143  * If "can_change" is set (not NULL), then first call it to check
12144  * if the modification is allowed, printing the error message "cannot_change"
12145  * if it is not.
12146  */
isl_map_change_space(__isl_take isl_map * map,isl_bool (* can_change)(__isl_keep isl_map * map),const char * cannot_change,__isl_give isl_space * (* change)(__isl_take isl_space * space))12147 static __isl_give isl_map *isl_map_change_space(__isl_take isl_map *map,
12148 	isl_bool (*can_change)(__isl_keep isl_map *map),
12149 	const char *cannot_change,
12150 	__isl_give isl_space *(*change)(__isl_take isl_space *space))
12151 {
12152 	isl_bool ok;
12153 	isl_space *space;
12154 
12155 	if (!map)
12156 		return NULL;
12157 
12158 	ok = can_change ? can_change(map) : isl_bool_true;
12159 	if (ok < 0)
12160 		return isl_map_free(map);
12161 	if (!ok)
12162 		isl_die(isl_map_get_ctx(map), isl_error_invalid, cannot_change,
12163 			return isl_map_free(map));
12164 
12165 	space = change(isl_map_get_space(map));
12166 	map = isl_map_reset_space(map, space);
12167 
12168 	return map;
12169 }
12170 
12171 /* Is the domain of "map" a wrapped relation?
12172  */
isl_map_domain_is_wrapping(__isl_keep isl_map * map)12173 isl_bool isl_map_domain_is_wrapping(__isl_keep isl_map *map)
12174 {
12175 	if (!map)
12176 		return isl_bool_error;
12177 
12178 	return isl_space_domain_is_wrapping(map->dim);
12179 }
12180 
12181 /* Does "map" have a wrapped relation in both domain and range?
12182  */
isl_map_is_product(__isl_keep isl_map * map)12183 isl_bool isl_map_is_product(__isl_keep isl_map *map)
12184 {
12185 	return isl_space_is_product(isl_map_peek_space(map));
12186 }
12187 
12188 /* Is the range of "map" a wrapped relation?
12189  */
isl_map_range_is_wrapping(__isl_keep isl_map * map)12190 isl_bool isl_map_range_is_wrapping(__isl_keep isl_map *map)
12191 {
12192 	if (!map)
12193 		return isl_bool_error;
12194 
12195 	return isl_space_range_is_wrapping(map->dim);
12196 }
12197 
isl_basic_map_wrap(__isl_take isl_basic_map * bmap)12198 __isl_give isl_basic_set *isl_basic_map_wrap(__isl_take isl_basic_map *bmap)
12199 {
12200 	isl_space *space;
12201 
12202 	space = isl_basic_map_take_space(bmap);
12203 	space = isl_space_wrap(space);
12204 	bmap = isl_basic_map_restore_space(bmap, space);
12205 
12206 	bmap = isl_basic_map_finalize(bmap);
12207 
12208 	return bset_from_bmap(bmap);
12209 }
12210 
12211 /* Given a map A -> B, return the set (A -> B).
12212  */
isl_map_wrap(__isl_take isl_map * map)12213 __isl_give isl_set *isl_map_wrap(__isl_take isl_map *map)
12214 {
12215 	return isl_map_change_space(map, NULL, NULL, &isl_space_wrap);
12216 }
12217 
isl_basic_set_unwrap(__isl_take isl_basic_set * bset)12218 __isl_give isl_basic_map *isl_basic_set_unwrap(__isl_take isl_basic_set *bset)
12219 {
12220 	bset = isl_basic_set_cow(bset);
12221 	if (!bset)
12222 		return NULL;
12223 
12224 	bset->dim = isl_space_unwrap(bset->dim);
12225 	if (!bset->dim)
12226 		goto error;
12227 
12228 	bset = isl_basic_set_finalize(bset);
12229 
12230 	return bset_to_bmap(bset);
12231 error:
12232 	isl_basic_set_free(bset);
12233 	return NULL;
12234 }
12235 
12236 /* Given a set (A -> B), return the map A -> B.
12237  * Error out if "set" is not of the form (A -> B).
12238  */
isl_set_unwrap(__isl_take isl_set * set)12239 __isl_give isl_map *isl_set_unwrap(__isl_take isl_set *set)
12240 {
12241 	return isl_map_change_space(set, &isl_set_is_wrapping,
12242 				    "not a wrapping set", &isl_space_unwrap);
12243 }
12244 
isl_basic_map_reset(__isl_take isl_basic_map * bmap,enum isl_dim_type type)12245 __isl_give isl_basic_map *isl_basic_map_reset(__isl_take isl_basic_map *bmap,
12246 	enum isl_dim_type type)
12247 {
12248 	isl_space *space;
12249 
12250 	space = isl_basic_map_take_space(bmap);
12251 	space = isl_space_reset(space, type);
12252 	bmap = isl_basic_map_restore_space(bmap, space);
12253 
12254 	bmap = isl_basic_map_mark_final(bmap);
12255 
12256 	return bmap;
12257 }
12258 
isl_map_reset(__isl_take isl_map * map,enum isl_dim_type type)12259 __isl_give isl_map *isl_map_reset(__isl_take isl_map *map,
12260 	enum isl_dim_type type)
12261 {
12262 	int i;
12263 	isl_space *space;
12264 
12265 	if (!map)
12266 		return NULL;
12267 
12268 	if (!isl_space_is_named_or_nested(map->dim, type))
12269 		return map;
12270 
12271 	map = isl_map_cow(map);
12272 	if (!map)
12273 		return NULL;
12274 
12275 	for (i = 0; i < map->n; ++i) {
12276 		map->p[i] = isl_basic_map_reset(map->p[i], type);
12277 		if (!map->p[i])
12278 			goto error;
12279 	}
12280 
12281 	space = isl_map_take_space(map);
12282 	space = isl_space_reset(space, type);
12283 	map = isl_map_restore_space(map, space);
12284 
12285 	return map;
12286 error:
12287 	isl_map_free(map);
12288 	return NULL;
12289 }
12290 
isl_basic_map_flatten(__isl_take isl_basic_map * bmap)12291 __isl_give isl_basic_map *isl_basic_map_flatten(__isl_take isl_basic_map *bmap)
12292 {
12293 	isl_space *space;
12294 
12295 	space = isl_basic_map_take_space(bmap);
12296 	space = isl_space_flatten(space);
12297 	bmap = isl_basic_map_restore_space(bmap, space);
12298 
12299 	bmap = isl_basic_map_mark_final(bmap);
12300 
12301 	return bmap;
12302 }
12303 
isl_basic_set_flatten(__isl_take isl_basic_set * bset)12304 __isl_give isl_basic_set *isl_basic_set_flatten(__isl_take isl_basic_set *bset)
12305 {
12306 	return bset_from_bmap(isl_basic_map_flatten(bset_to_bmap(bset)));
12307 }
12308 
isl_basic_map_flatten_domain(__isl_take isl_basic_map * bmap)12309 __isl_give isl_basic_map *isl_basic_map_flatten_domain(
12310 	__isl_take isl_basic_map *bmap)
12311 {
12312 	isl_space *space;
12313 
12314 	space = isl_basic_map_take_space(bmap);
12315 	space = isl_space_flatten_domain(space);
12316 	bmap = isl_basic_map_restore_space(bmap, space);
12317 
12318 	bmap = isl_basic_map_mark_final(bmap);
12319 
12320 	return bmap;
12321 }
12322 
isl_basic_map_flatten_range(__isl_take isl_basic_map * bmap)12323 __isl_give isl_basic_map *isl_basic_map_flatten_range(
12324 	__isl_take isl_basic_map *bmap)
12325 {
12326 	isl_space *space;
12327 
12328 	space = isl_basic_map_take_space(bmap);
12329 	space = isl_space_flatten_range(space);
12330 	bmap = isl_basic_map_restore_space(bmap, space);
12331 
12332 	bmap = isl_basic_map_mark_final(bmap);
12333 
12334 	return bmap;
12335 }
12336 
12337 /* Remove any internal structure from the spaces of domain and range of "map".
12338  */
isl_map_flatten(__isl_take isl_map * map)12339 __isl_give isl_map *isl_map_flatten(__isl_take isl_map *map)
12340 {
12341 	if (!map)
12342 		return NULL;
12343 
12344 	if (!map->dim->nested[0] && !map->dim->nested[1])
12345 		return map;
12346 
12347 	return isl_map_change_space(map, NULL, NULL, &isl_space_flatten);
12348 }
12349 
isl_set_flatten(__isl_take isl_set * set)12350 __isl_give isl_set *isl_set_flatten(__isl_take isl_set *set)
12351 {
12352 	return set_from_map(isl_map_flatten(set_to_map(set)));
12353 }
12354 
isl_set_flatten_map(__isl_take isl_set * set)12355 __isl_give isl_map *isl_set_flatten_map(__isl_take isl_set *set)
12356 {
12357 	isl_space *space, *flat_space;
12358 	isl_map *map;
12359 
12360 	space = isl_set_get_space(set);
12361 	flat_space = isl_space_flatten(isl_space_copy(space));
12362 	map = isl_map_identity(isl_space_join(isl_space_reverse(space),
12363 						flat_space));
12364 	map = isl_map_intersect_domain(map, set);
12365 
12366 	return map;
12367 }
12368 
12369 /* Remove any internal structure from the space of the domain of "map".
12370  */
isl_map_flatten_domain(__isl_take isl_map * map)12371 __isl_give isl_map *isl_map_flatten_domain(__isl_take isl_map *map)
12372 {
12373 	if (!map)
12374 		return NULL;
12375 
12376 	if (!map->dim->nested[0])
12377 		return map;
12378 
12379 	return isl_map_change_space(map, NULL, NULL, &isl_space_flatten_domain);
12380 }
12381 
12382 /* Remove any internal structure from the space of the range of "map".
12383  */
isl_map_flatten_range(__isl_take isl_map * map)12384 __isl_give isl_map *isl_map_flatten_range(__isl_take isl_map *map)
12385 {
12386 	if (!map)
12387 		return NULL;
12388 
12389 	if (!map->dim->nested[1])
12390 		return map;
12391 
12392 	return isl_map_change_space(map, NULL, NULL, &isl_space_flatten_range);
12393 }
12394 
12395 /* Reorder the dimensions of "bmap" according to the given dim_map
12396  * and set the dimension specification to "space" and
12397  * perform Gaussian elimination on the result.
12398  */
isl_basic_map_realign(__isl_take isl_basic_map * bmap,__isl_take isl_space * space,__isl_take struct isl_dim_map * dim_map)12399 __isl_give isl_basic_map *isl_basic_map_realign(__isl_take isl_basic_map *bmap,
12400 	__isl_take isl_space *space, __isl_take struct isl_dim_map *dim_map)
12401 {
12402 	isl_basic_map *res;
12403 	unsigned flags;
12404 	isl_size n_div;
12405 
12406 	n_div = isl_basic_map_dim(bmap, isl_dim_div);
12407 	if (n_div < 0 || !space || !dim_map)
12408 		goto error;
12409 
12410 	flags = bmap->flags;
12411 	ISL_FL_CLR(flags, ISL_BASIC_MAP_FINAL);
12412 	ISL_FL_CLR(flags, ISL_BASIC_MAP_SORTED);
12413 	ISL_FL_CLR(flags, ISL_BASIC_MAP_NORMALIZED_DIVS);
12414 	res = isl_basic_map_alloc_space(space, n_div, bmap->n_eq, bmap->n_ineq);
12415 	res = isl_basic_map_add_constraints_dim_map(res, bmap, dim_map);
12416 	if (res)
12417 		res->flags = flags;
12418 	res = isl_basic_map_gauss(res, NULL);
12419 	res = isl_basic_map_finalize(res);
12420 	return res;
12421 error:
12422 	isl_dim_map_free(dim_map);
12423 	isl_basic_map_free(bmap);
12424 	isl_space_free(space);
12425 	return NULL;
12426 }
12427 
12428 /* Reorder the dimensions of "map" according to given reordering.
12429  */
isl_map_realign(__isl_take isl_map * map,__isl_take isl_reordering * r)12430 __isl_give isl_map *isl_map_realign(__isl_take isl_map *map,
12431 	__isl_take isl_reordering *r)
12432 {
12433 	int i;
12434 	struct isl_dim_map *dim_map;
12435 
12436 	map = isl_map_cow(map);
12437 	dim_map = isl_dim_map_from_reordering(r);
12438 	if (!map || !r || !dim_map)
12439 		goto error;
12440 
12441 	for (i = 0; i < map->n; ++i) {
12442 		struct isl_dim_map *dim_map_i;
12443 		isl_space *space;
12444 
12445 		dim_map_i = isl_dim_map_extend(dim_map, map->p[i]);
12446 
12447 		space = isl_reordering_get_space(r);
12448 		map->p[i] = isl_basic_map_realign(map->p[i], space, dim_map_i);
12449 
12450 		if (!map->p[i])
12451 			goto error;
12452 	}
12453 
12454 	map = isl_map_reset_space(map, isl_reordering_get_space(r));
12455 	map = isl_map_unmark_normalized(map);
12456 
12457 	isl_reordering_free(r);
12458 	isl_dim_map_free(dim_map);
12459 	return map;
12460 error:
12461 	isl_dim_map_free(dim_map);
12462 	isl_map_free(map);
12463 	isl_reordering_free(r);
12464 	return NULL;
12465 }
12466 
isl_set_realign(__isl_take isl_set * set,__isl_take isl_reordering * r)12467 __isl_give isl_set *isl_set_realign(__isl_take isl_set *set,
12468 	__isl_take isl_reordering *r)
12469 {
12470 	return set_from_map(isl_map_realign(set_to_map(set), r));
12471 }
12472 
isl_map_align_params(__isl_take isl_map * map,__isl_take isl_space * model)12473 __isl_give isl_map *isl_map_align_params(__isl_take isl_map *map,
12474 	__isl_take isl_space *model)
12475 {
12476 	isl_ctx *ctx;
12477 	isl_bool aligned;
12478 
12479 	if (!map || !model)
12480 		goto error;
12481 
12482 	ctx = isl_space_get_ctx(model);
12483 	if (!isl_space_has_named_params(model))
12484 		isl_die(ctx, isl_error_invalid,
12485 			"model has unnamed parameters", goto error);
12486 	if (isl_map_check_named_params(map) < 0)
12487 		goto error;
12488 	aligned = isl_map_space_has_equal_params(map, model);
12489 	if (aligned < 0)
12490 		goto error;
12491 	if (!aligned) {
12492 		isl_space *space;
12493 		isl_reordering *exp;
12494 
12495 		space = isl_map_peek_space(map);
12496 		exp = isl_parameter_alignment_reordering(space, model);
12497 		map = isl_map_realign(map, exp);
12498 	}
12499 
12500 	isl_space_free(model);
12501 	return map;
12502 error:
12503 	isl_space_free(model);
12504 	isl_map_free(map);
12505 	return NULL;
12506 }
12507 
isl_set_align_params(__isl_take isl_set * set,__isl_take isl_space * model)12508 __isl_give isl_set *isl_set_align_params(__isl_take isl_set *set,
12509 	__isl_take isl_space *model)
12510 {
12511 	return isl_map_align_params(set, model);
12512 }
12513 
12514 /* Align the parameters of "bmap" to those of "model", introducing
12515  * additional parameters if needed.
12516  */
isl_basic_map_align_params(__isl_take isl_basic_map * bmap,__isl_take isl_space * model)12517 __isl_give isl_basic_map *isl_basic_map_align_params(
12518 	__isl_take isl_basic_map *bmap, __isl_take isl_space *model)
12519 {
12520 	isl_ctx *ctx;
12521 	isl_bool equal_params;
12522 	isl_space *bmap_space;
12523 
12524 	if (!bmap || !model)
12525 		goto error;
12526 
12527 	ctx = isl_space_get_ctx(model);
12528 	if (!isl_space_has_named_params(model))
12529 		isl_die(ctx, isl_error_invalid,
12530 			"model has unnamed parameters", goto error);
12531 	if (isl_basic_map_check_named_params(bmap) < 0)
12532 		goto error;
12533 	bmap_space = isl_basic_map_peek_space(bmap);
12534 	equal_params = isl_space_has_equal_params(bmap_space, model);
12535 	if (equal_params < 0)
12536 		goto error;
12537 	if (!equal_params) {
12538 		isl_reordering *exp;
12539 		struct isl_dim_map *dim_map;
12540 
12541 		exp = isl_parameter_alignment_reordering(bmap_space, model);
12542 		dim_map = isl_dim_map_from_reordering(exp);
12543 		bmap = isl_basic_map_realign(bmap,
12544 				    isl_reordering_get_space(exp),
12545 				    isl_dim_map_extend(dim_map, bmap));
12546 		isl_reordering_free(exp);
12547 		isl_dim_map_free(dim_map);
12548 	}
12549 
12550 	isl_space_free(model);
12551 	return bmap;
12552 error:
12553 	isl_space_free(model);
12554 	isl_basic_map_free(bmap);
12555 	return NULL;
12556 }
12557 
12558 /* Do "bset" and "space" have the same parameters?
12559  */
isl_basic_set_space_has_equal_params(__isl_keep isl_basic_set * bset,__isl_keep isl_space * space)12560 isl_bool isl_basic_set_space_has_equal_params(__isl_keep isl_basic_set *bset,
12561 	__isl_keep isl_space *space)
12562 {
12563 	isl_space *bset_space;
12564 
12565 	bset_space = isl_basic_set_peek_space(bset);
12566 	return isl_space_has_equal_params(bset_space, space);
12567 }
12568 
12569 /* Do "map" and "space" have the same parameters?
12570  */
isl_map_space_has_equal_params(__isl_keep isl_map * map,__isl_keep isl_space * space)12571 isl_bool isl_map_space_has_equal_params(__isl_keep isl_map *map,
12572 	__isl_keep isl_space *space)
12573 {
12574 	isl_space *map_space;
12575 
12576 	map_space = isl_map_peek_space(map);
12577 	return isl_space_has_equal_params(map_space, space);
12578 }
12579 
12580 /* Do "set" and "space" have the same parameters?
12581  */
isl_set_space_has_equal_params(__isl_keep isl_set * set,__isl_keep isl_space * space)12582 isl_bool isl_set_space_has_equal_params(__isl_keep isl_set *set,
12583 	__isl_keep isl_space *space)
12584 {
12585 	return isl_map_space_has_equal_params(set_to_map(set), space);
12586 }
12587 
12588 /* Align the parameters of "bset" to those of "model", introducing
12589  * additional parameters if needed.
12590  */
isl_basic_set_align_params(__isl_take isl_basic_set * bset,__isl_take isl_space * model)12591 __isl_give isl_basic_set *isl_basic_set_align_params(
12592 	__isl_take isl_basic_set *bset, __isl_take isl_space *model)
12593 {
12594 	return isl_basic_map_align_params(bset, model);
12595 }
12596 
12597 /* Drop all parameters not referenced by "map".
12598  */
isl_map_drop_unused_params(__isl_take isl_map * map)12599 __isl_give isl_map *isl_map_drop_unused_params(__isl_take isl_map *map)
12600 {
12601 	int i;
12602 	isl_size n;
12603 
12604 	n = isl_map_dim(map, isl_dim_param);
12605 	if (isl_map_check_named_params(map) < 0 || n < 0)
12606 		return isl_map_free(map);
12607 
12608 	for (i = n - 1; i >= 0; i--) {
12609 		isl_bool involves;
12610 
12611 		involves = isl_map_involves_dims(map, isl_dim_param, i, 1);
12612 		if (involves < 0)
12613 			return isl_map_free(map);
12614 		if (!involves)
12615 			map = isl_map_project_out(map, isl_dim_param, i, 1);
12616 	}
12617 
12618 	return map;
12619 }
12620 
12621 /* Drop all parameters not referenced by "set".
12622  */
isl_set_drop_unused_params(__isl_take isl_set * set)12623 __isl_give isl_set *isl_set_drop_unused_params(
12624 	__isl_take isl_set *set)
12625 {
12626 	return set_from_map(isl_map_drop_unused_params(set_to_map(set)));
12627 }
12628 
12629 /* Drop all parameters not referenced by "bmap".
12630  */
isl_basic_map_drop_unused_params(__isl_take isl_basic_map * bmap)12631 __isl_give isl_basic_map *isl_basic_map_drop_unused_params(
12632 	__isl_take isl_basic_map *bmap)
12633 {
12634 	isl_size nparam;
12635 	int i;
12636 
12637 	nparam = isl_basic_map_dim(bmap, isl_dim_param);
12638 	if (nparam < 0 || isl_basic_map_check_named_params(bmap) < 0)
12639 		return isl_basic_map_free(bmap);
12640 
12641 	for (i = nparam - 1; i >= 0; i--) {
12642 		isl_bool involves;
12643 
12644 		involves = isl_basic_map_involves_dims(bmap,
12645 							isl_dim_param, i, 1);
12646 		if (involves < 0)
12647 			return isl_basic_map_free(bmap);
12648 		if (!involves)
12649 			bmap = isl_basic_map_drop(bmap, isl_dim_param, i, 1);
12650 	}
12651 
12652 	return bmap;
12653 }
12654 
12655 /* Drop all parameters not referenced by "bset".
12656  */
isl_basic_set_drop_unused_params(__isl_take isl_basic_set * bset)12657 __isl_give isl_basic_set *isl_basic_set_drop_unused_params(
12658 	__isl_take isl_basic_set *bset)
12659 {
12660 	return bset_from_bmap(isl_basic_map_drop_unused_params(
12661 							bset_to_bmap(bset)));
12662 }
12663 
12664 /* Given a tuple of identifiers "tuple" in a space that corresponds
12665  * to that of "set", if any of those identifiers appear as parameters
12666  * in "set", then equate those parameters with the corresponding
12667  * set dimensions and project out the parameters.
12668  * The result therefore has no such parameters.
12669  */
equate_params(__isl_take isl_set * set,__isl_keep isl_multi_id * tuple)12670 static __isl_give isl_set *equate_params(__isl_take isl_set *set,
12671 	__isl_keep isl_multi_id *tuple)
12672 {
12673 	int i;
12674 	isl_size n;
12675 	isl_space *set_space, *tuple_space;
12676 
12677 	set_space = isl_set_peek_space(set);
12678 	tuple_space = isl_multi_id_peek_space(tuple);
12679 	if (isl_space_check_equal_tuples(tuple_space, set_space) < 0)
12680 		return isl_set_free(set);
12681 	n = isl_multi_id_size(tuple);
12682 	if (n < 0)
12683 		return isl_set_free(set);
12684 	for (i = 0; i < n; ++i) {
12685 		isl_id *id;
12686 		int pos;
12687 
12688 		id = isl_multi_id_get_at(tuple, i);
12689 		if (!id)
12690 			return isl_set_free(set);
12691 		pos = isl_set_find_dim_by_id(set, isl_dim_param, id);
12692 		isl_id_free(id);
12693 		if (pos < 0)
12694 			continue;
12695 		set = isl_set_equate(set, isl_dim_param, pos, isl_dim_set, i);
12696 		set = isl_set_project_out(set, isl_dim_param, pos, 1);
12697 	}
12698 	return set;
12699 }
12700 
12701 /* Bind the set dimensions of "set" to parameters with identifiers
12702  * specified by "tuple", living in the same space as "set".
12703  *
12704  * If no parameters with these identifiers appear in "set" already,
12705  * then the set dimensions are simply reinterpreted as parameters.
12706  * Otherwise, the parameters are first equated to the corresponding
12707  * set dimensions.
12708  */
isl_set_bind(__isl_take isl_set * set,__isl_take isl_multi_id * tuple)12709 __isl_give isl_set *isl_set_bind(__isl_take isl_set *set,
12710 	__isl_take isl_multi_id *tuple)
12711 {
12712 	isl_space *space;
12713 
12714 	set = equate_params(set, tuple);
12715 	space = isl_set_get_space(set);
12716 	space = isl_space_bind_set(space, tuple);
12717 	isl_multi_id_free(tuple);
12718 	set = isl_set_reset_space(set, space);
12719 
12720 	return set;
12721 }
12722 
12723 /* Given a tuple of identifiers "tuple" in a space that corresponds
12724  * to the domain of "map", if any of those identifiers appear as parameters
12725  * in "map", then equate those parameters with the corresponding
12726  * input dimensions and project out the parameters.
12727  * The result therefore has no such parameters.
12728  */
map_equate_params(__isl_take isl_map * map,__isl_keep isl_multi_id * tuple)12729 static __isl_give isl_map *map_equate_params(__isl_take isl_map *map,
12730 	__isl_keep isl_multi_id *tuple)
12731 {
12732 	int i;
12733 	isl_size n;
12734 	isl_space *map_space, *tuple_space;
12735 
12736 	map_space = isl_map_peek_space(map);
12737 	tuple_space = isl_multi_id_peek_space(tuple);
12738 	if (isl_space_check_domain_tuples(tuple_space, map_space) < 0)
12739 		return isl_map_free(map);
12740 	n = isl_multi_id_size(tuple);
12741 	if (n < 0)
12742 		return isl_map_free(map);
12743 	for (i = 0; i < n; ++i) {
12744 		isl_id *id;
12745 		int pos;
12746 
12747 		id = isl_multi_id_get_at(tuple, i);
12748 		if (!id)
12749 			return isl_map_free(map);
12750 		pos = isl_map_find_dim_by_id(map, isl_dim_param, id);
12751 		isl_id_free(id);
12752 		if (pos < 0)
12753 			continue;
12754 		map = isl_map_equate(map, isl_dim_param, pos, isl_dim_in, i);
12755 		map = isl_map_project_out(map, isl_dim_param, pos, 1);
12756 	}
12757 	return map;
12758 }
12759 
12760 /* Bind the input dimensions of "map" to parameters with identifiers
12761  * specified by "tuple", living in the domain space of "map".
12762  *
12763  * If no parameters with these identifiers appear in "map" already,
12764  * then the input dimensions are simply reinterpreted as parameters.
12765  * Otherwise, the parameters are first equated to the corresponding
12766  * input dimensions.
12767  */
isl_map_bind_domain(__isl_take isl_map * map,__isl_take isl_multi_id * tuple)12768 __isl_give isl_set *isl_map_bind_domain(__isl_take isl_map *map,
12769 	__isl_take isl_multi_id *tuple)
12770 {
12771 	isl_space *space;
12772 	isl_set *set;
12773 
12774 	map = map_equate_params(map, tuple);
12775 	space = isl_map_get_space(map);
12776 	space = isl_space_bind_map_domain(space, tuple);
12777 	isl_multi_id_free(tuple);
12778 	set = set_from_map(isl_map_reset_space(map, space));
12779 
12780 	return set;
12781 }
12782 
12783 /* Bind the output dimensions of "map" to parameters with identifiers
12784  * specified by "tuple", living in the range space of "map".
12785  *
12786  * Since binding is more easily implemented on the domain,
12787  * bind the input dimensions of the inverse of "map".
12788  */
isl_map_bind_range(__isl_take isl_map * map,__isl_take isl_multi_id * tuple)12789 __isl_give isl_set *isl_map_bind_range(__isl_take isl_map *map,
12790 	__isl_take isl_multi_id *tuple)
12791 {
12792 	return isl_map_bind_domain(isl_map_reverse(map), tuple);
12793 }
12794 
12795 /* Insert a domain corresponding to "tuple"
12796  * into the nullary or unary relation "set".
12797  * The result has an extra initial tuple and is therefore
12798  * either a unary or binary relation.
12799  * Any parameters with identifiers in "tuple" are reinterpreted
12800  * as the corresponding domain dimensions.
12801  */
unbind_params_insert_domain(__isl_take isl_set * set,__isl_take isl_multi_id * tuple)12802 static __isl_give isl_map *unbind_params_insert_domain(
12803 	__isl_take isl_set *set, __isl_take isl_multi_id *tuple)
12804 {
12805 	isl_space *space;
12806 	isl_reordering *r;
12807 
12808 	space = isl_set_peek_space(set);
12809 	r = isl_reordering_unbind_params_insert_domain(space, tuple);
12810 	isl_multi_id_free(tuple);
12811 
12812 	return isl_map_realign(set_to_map(set), r);
12813 }
12814 
12815 /* Construct a set with "tuple" as domain from the parameter domain "set".
12816  * Any parameters with identifiers in "tuple" are reinterpreted
12817  * as the corresponding set dimensions.
12818  */
isl_set_unbind_params(__isl_take isl_set * set,__isl_take isl_multi_id * tuple)12819 __isl_give isl_set *isl_set_unbind_params(__isl_take isl_set *set,
12820 	__isl_take isl_multi_id *tuple)
12821 {
12822 	isl_bool is_params;
12823 
12824 	is_params = isl_set_is_params(set);
12825 	if (is_params < 0)
12826 		set = isl_set_free(set);
12827 	else if (!is_params)
12828 		isl_die(isl_set_get_ctx(set), isl_error_invalid,
12829 			"expecting parameter domain", set = isl_set_free(set));
12830 	return set_from_map(unbind_params_insert_domain(set, tuple));
12831 }
12832 
12833 /* Check that "set" is a proper set, i.e., that it is not a parameter domain.
12834  */
isl_set_check_is_set(__isl_keep isl_set * set)12835 static isl_stat isl_set_check_is_set(__isl_keep isl_set *set)
12836 {
12837 	isl_bool is_params;
12838 
12839 	is_params = isl_set_is_params(set);
12840 	if (is_params < 0)
12841 		return isl_stat_error;
12842 	else if (is_params)
12843 		isl_die(isl_set_get_ctx(set), isl_error_invalid,
12844 			"expecting proper set", return isl_stat_error);
12845 
12846 	return isl_stat_ok;
12847 }
12848 
12849 /* Construct a map with "domain" as domain and "set" as range.
12850  * Any parameters with identifiers in "domain" are reinterpreted
12851  * as the corresponding domain dimensions.
12852  */
isl_set_unbind_params_insert_domain(__isl_take isl_set * set,__isl_take isl_multi_id * domain)12853 __isl_give isl_map *isl_set_unbind_params_insert_domain(
12854 	__isl_take isl_set *set, __isl_take isl_multi_id *domain)
12855 {
12856 	if (isl_set_check_is_set(set) < 0)
12857 		set = isl_set_free(set);
12858 	return unbind_params_insert_domain(set, domain);
12859 }
12860 
12861 /* Construct a map with "domain" as domain and "set" as range.
12862  */
isl_set_insert_domain(__isl_take isl_set * set,__isl_take isl_space * domain)12863 __isl_give isl_map *isl_set_insert_domain(__isl_take isl_set *set,
12864 	__isl_take isl_space *domain)
12865 {
12866 	isl_size dim;
12867 	isl_space *space;
12868 	isl_map *map;
12869 
12870 	if (isl_set_check_is_set(set) < 0 || isl_space_check_is_set(domain) < 0)
12871 		domain = isl_space_free(domain);
12872 	dim = isl_space_dim(domain, isl_dim_set);
12873 	if (dim < 0)
12874 		domain = isl_space_free(domain);
12875 	space = isl_set_get_space(set);
12876 	domain = isl_space_replace_params(domain, space);
12877 	space = isl_space_map_from_domain_and_range(domain, space);
12878 
12879 	map = isl_map_from_range(set);
12880 	map = isl_map_add_dims(map, isl_dim_in, dim);
12881 	map = isl_map_reset_space(map, space);
12882 
12883 	return map;
12884 }
12885 
isl_basic_map_equalities_matrix(__isl_keep isl_basic_map * bmap,enum isl_dim_type c1,enum isl_dim_type c2,enum isl_dim_type c3,enum isl_dim_type c4,enum isl_dim_type c5)12886 __isl_give isl_mat *isl_basic_map_equalities_matrix(
12887 		__isl_keep isl_basic_map *bmap, enum isl_dim_type c1,
12888 		enum isl_dim_type c2, enum isl_dim_type c3,
12889 		enum isl_dim_type c4, enum isl_dim_type c5)
12890 {
12891 	enum isl_dim_type c[5] = { c1, c2, c3, c4, c5 };
12892 	struct isl_mat *mat;
12893 	int i, j, k;
12894 	int pos;
12895 	isl_size total;
12896 
12897 	total = isl_basic_map_dim(bmap, isl_dim_all);
12898 	if (total < 0)
12899 		return NULL;
12900 	mat = isl_mat_alloc(bmap->ctx, bmap->n_eq, total + 1);
12901 	if (!mat)
12902 		return NULL;
12903 	for (i = 0; i < bmap->n_eq; ++i)
12904 		for (j = 0, pos = 0; j < 5; ++j) {
12905 			int off = isl_basic_map_offset(bmap, c[j]);
12906 			isl_size dim = isl_basic_map_dim(bmap, c[j]);
12907 			if (dim < 0)
12908 				return isl_mat_free(mat);
12909 			for (k = 0; k < dim; ++k) {
12910 				isl_int_set(mat->row[i][pos],
12911 					    bmap->eq[i][off + k]);
12912 				++pos;
12913 			}
12914 		}
12915 
12916 	return mat;
12917 }
12918 
isl_basic_map_inequalities_matrix(__isl_keep isl_basic_map * bmap,enum isl_dim_type c1,enum isl_dim_type c2,enum isl_dim_type c3,enum isl_dim_type c4,enum isl_dim_type c5)12919 __isl_give isl_mat *isl_basic_map_inequalities_matrix(
12920 		__isl_keep isl_basic_map *bmap, enum isl_dim_type c1,
12921 		enum isl_dim_type c2, enum isl_dim_type c3,
12922 		enum isl_dim_type c4, enum isl_dim_type c5)
12923 {
12924 	enum isl_dim_type c[5] = { c1, c2, c3, c4, c5 };
12925 	struct isl_mat *mat;
12926 	int i, j, k;
12927 	int pos;
12928 	isl_size total;
12929 
12930 	total = isl_basic_map_dim(bmap, isl_dim_all);
12931 	if (total < 0)
12932 		return NULL;
12933 	mat = isl_mat_alloc(bmap->ctx, bmap->n_ineq, total + 1);
12934 	if (!mat)
12935 		return NULL;
12936 	for (i = 0; i < bmap->n_ineq; ++i)
12937 		for (j = 0, pos = 0; j < 5; ++j) {
12938 			int off = isl_basic_map_offset(bmap, c[j]);
12939 			isl_size dim = isl_basic_map_dim(bmap, c[j]);
12940 			if (dim < 0)
12941 				return isl_mat_free(mat);
12942 			for (k = 0; k < dim; ++k) {
12943 				isl_int_set(mat->row[i][pos],
12944 					    bmap->ineq[i][off + k]);
12945 				++pos;
12946 			}
12947 		}
12948 
12949 	return mat;
12950 }
12951 
isl_basic_map_from_constraint_matrices(__isl_take isl_space * space,__isl_take isl_mat * eq,__isl_take isl_mat * ineq,enum isl_dim_type c1,enum isl_dim_type c2,enum isl_dim_type c3,enum isl_dim_type c4,enum isl_dim_type c5)12952 __isl_give isl_basic_map *isl_basic_map_from_constraint_matrices(
12953 	__isl_take isl_space *space,
12954 	__isl_take isl_mat *eq, __isl_take isl_mat *ineq, enum isl_dim_type c1,
12955 	enum isl_dim_type c2, enum isl_dim_type c3,
12956 	enum isl_dim_type c4, enum isl_dim_type c5)
12957 {
12958 	enum isl_dim_type c[5] = { c1, c2, c3, c4, c5 };
12959 	isl_basic_map *bmap = NULL;
12960 	isl_size dim;
12961 	unsigned total;
12962 	unsigned extra;
12963 	int i, j, k, l;
12964 	int pos;
12965 
12966 	dim = isl_space_dim(space, isl_dim_all);
12967 	if (dim < 0 || !eq || !ineq)
12968 		goto error;
12969 
12970 	if (eq->n_col != ineq->n_col)
12971 		isl_die(space->ctx, isl_error_invalid,
12972 			"equalities and inequalities matrices should have "
12973 			"same number of columns", goto error);
12974 
12975 	total = 1 + dim;
12976 
12977 	if (eq->n_col < total)
12978 		isl_die(space->ctx, isl_error_invalid,
12979 			"number of columns too small", goto error);
12980 
12981 	extra = eq->n_col - total;
12982 
12983 	bmap = isl_basic_map_alloc_space(isl_space_copy(space), extra,
12984 				       eq->n_row, ineq->n_row);
12985 	if (!bmap)
12986 		goto error;
12987 	for (i = 0; i < extra; ++i) {
12988 		k = isl_basic_map_alloc_div(bmap);
12989 		if (k < 0)
12990 			goto error;
12991 		isl_int_set_si(bmap->div[k][0], 0);
12992 	}
12993 	for (i = 0; i < eq->n_row; ++i) {
12994 		l = isl_basic_map_alloc_equality(bmap);
12995 		if (l < 0)
12996 			goto error;
12997 		for (j = 0, pos = 0; j < 5; ++j) {
12998 			int off = isl_basic_map_offset(bmap, c[j]);
12999 			isl_size dim = isl_basic_map_dim(bmap, c[j]);
13000 			if (dim < 0)
13001 				goto error;
13002 			for (k = 0; k < dim; ++k) {
13003 				isl_int_set(bmap->eq[l][off + k],
13004 					    eq->row[i][pos]);
13005 				++pos;
13006 			}
13007 		}
13008 	}
13009 	for (i = 0; i < ineq->n_row; ++i) {
13010 		l = isl_basic_map_alloc_inequality(bmap);
13011 		if (l < 0)
13012 			goto error;
13013 		for (j = 0, pos = 0; j < 5; ++j) {
13014 			int off = isl_basic_map_offset(bmap, c[j]);
13015 			isl_size dim = isl_basic_map_dim(bmap, c[j]);
13016 			if (dim < 0)
13017 				goto error;
13018 			for (k = 0; k < dim; ++k) {
13019 				isl_int_set(bmap->ineq[l][off + k],
13020 					    ineq->row[i][pos]);
13021 				++pos;
13022 			}
13023 		}
13024 	}
13025 
13026 	isl_space_free(space);
13027 	isl_mat_free(eq);
13028 	isl_mat_free(ineq);
13029 
13030 	bmap = isl_basic_map_simplify(bmap);
13031 	return isl_basic_map_finalize(bmap);
13032 error:
13033 	isl_space_free(space);
13034 	isl_mat_free(eq);
13035 	isl_mat_free(ineq);
13036 	isl_basic_map_free(bmap);
13037 	return NULL;
13038 }
13039 
isl_basic_set_equalities_matrix(__isl_keep isl_basic_set * bset,enum isl_dim_type c1,enum isl_dim_type c2,enum isl_dim_type c3,enum isl_dim_type c4)13040 __isl_give isl_mat *isl_basic_set_equalities_matrix(
13041 	__isl_keep isl_basic_set *bset, enum isl_dim_type c1,
13042 	enum isl_dim_type c2, enum isl_dim_type c3, enum isl_dim_type c4)
13043 {
13044 	return isl_basic_map_equalities_matrix(bset_to_bmap(bset),
13045 						c1, c2, c3, c4, isl_dim_in);
13046 }
13047 
isl_basic_set_inequalities_matrix(__isl_keep isl_basic_set * bset,enum isl_dim_type c1,enum isl_dim_type c2,enum isl_dim_type c3,enum isl_dim_type c4)13048 __isl_give isl_mat *isl_basic_set_inequalities_matrix(
13049 	__isl_keep isl_basic_set *bset, enum isl_dim_type c1,
13050 	enum isl_dim_type c2, enum isl_dim_type c3, enum isl_dim_type c4)
13051 {
13052 	return isl_basic_map_inequalities_matrix(bset_to_bmap(bset),
13053 						 c1, c2, c3, c4, isl_dim_in);
13054 }
13055 
isl_basic_set_from_constraint_matrices(__isl_take isl_space * space,__isl_take isl_mat * eq,__isl_take isl_mat * ineq,enum isl_dim_type c1,enum isl_dim_type c2,enum isl_dim_type c3,enum isl_dim_type c4)13056 __isl_give isl_basic_set *isl_basic_set_from_constraint_matrices(
13057 	__isl_take isl_space *space,
13058 	__isl_take isl_mat *eq, __isl_take isl_mat *ineq, enum isl_dim_type c1,
13059 	enum isl_dim_type c2, enum isl_dim_type c3, enum isl_dim_type c4)
13060 {
13061 	isl_basic_map *bmap;
13062 	bmap = isl_basic_map_from_constraint_matrices(space, eq, ineq,
13063 						   c1, c2, c3, c4, isl_dim_in);
13064 	return bset_from_bmap(bmap);
13065 }
13066 
isl_basic_map_can_zip(__isl_keep isl_basic_map * bmap)13067 isl_bool isl_basic_map_can_zip(__isl_keep isl_basic_map *bmap)
13068 {
13069 	if (!bmap)
13070 		return isl_bool_error;
13071 
13072 	return isl_space_can_zip(bmap->dim);
13073 }
13074 
isl_map_can_zip(__isl_keep isl_map * map)13075 isl_bool isl_map_can_zip(__isl_keep isl_map *map)
13076 {
13077 	if (!map)
13078 		return isl_bool_error;
13079 
13080 	return isl_space_can_zip(map->dim);
13081 }
13082 
13083 /* Given a basic map (A -> B) -> (C -> D), return the corresponding basic map
13084  * (A -> C) -> (B -> D).
13085  */
isl_basic_map_zip(__isl_take isl_basic_map * bmap)13086 __isl_give isl_basic_map *isl_basic_map_zip(__isl_take isl_basic_map *bmap)
13087 {
13088 	unsigned pos;
13089 	isl_size n_in;
13090 	isl_size n1;
13091 	isl_size n2;
13092 
13093 	if (!bmap)
13094 		return NULL;
13095 
13096 	if (!isl_basic_map_can_zip(bmap))
13097 		isl_die(bmap->ctx, isl_error_invalid,
13098 			"basic map cannot be zipped", goto error);
13099 	n_in = isl_space_dim(bmap->dim->nested[0], isl_dim_in);
13100 	n1 = isl_space_dim(bmap->dim->nested[0], isl_dim_out);
13101 	n2 = isl_space_dim(bmap->dim->nested[1], isl_dim_in);
13102 	if (n_in < 0 || n1 < 0 || n2 < 0)
13103 		return isl_basic_map_free(bmap);
13104 	pos = isl_basic_map_offset(bmap, isl_dim_in) + n_in;
13105 	bmap = isl_basic_map_cow(bmap);
13106 	bmap = isl_basic_map_swap_vars(bmap, pos, n1, n2);
13107 	if (!bmap)
13108 		return NULL;
13109 	bmap->dim = isl_space_zip(bmap->dim);
13110 	if (!bmap->dim)
13111 		goto error;
13112 	bmap = isl_basic_map_mark_final(bmap);
13113 	return bmap;
13114 error:
13115 	isl_basic_map_free(bmap);
13116 	return NULL;
13117 }
13118 
13119 /* Given a map (A -> B) -> (C -> D), return the corresponding map
13120  * (A -> C) -> (B -> D).
13121  */
isl_map_zip(__isl_take isl_map * map)13122 __isl_give isl_map *isl_map_zip(__isl_take isl_map *map)
13123 {
13124 	if (!map)
13125 		return NULL;
13126 
13127 	if (!isl_map_can_zip(map))
13128 		isl_die(map->ctx, isl_error_invalid, "map cannot be zipped",
13129 			goto error);
13130 
13131 	return isl_map_transform(map, &isl_space_zip, &isl_basic_map_zip);
13132 error:
13133 	isl_map_free(map);
13134 	return NULL;
13135 }
13136 
13137 /* Can we apply isl_basic_map_curry to "bmap"?
13138  * That is, does it have a nested relation in its domain?
13139  */
isl_basic_map_can_curry(__isl_keep isl_basic_map * bmap)13140 isl_bool isl_basic_map_can_curry(__isl_keep isl_basic_map *bmap)
13141 {
13142 	if (!bmap)
13143 		return isl_bool_error;
13144 
13145 	return isl_space_can_curry(bmap->dim);
13146 }
13147 
13148 /* Can we apply isl_map_curry to "map"?
13149  * That is, does it have a nested relation in its domain?
13150  */
isl_map_can_curry(__isl_keep isl_map * map)13151 isl_bool isl_map_can_curry(__isl_keep isl_map *map)
13152 {
13153 	if (!map)
13154 		return isl_bool_error;
13155 
13156 	return isl_space_can_curry(map->dim);
13157 }
13158 
13159 /* Given a basic map (A -> B) -> C, return the corresponding basic map
13160  * A -> (B -> C).
13161  */
isl_basic_map_curry(__isl_take isl_basic_map * bmap)13162 __isl_give isl_basic_map *isl_basic_map_curry(__isl_take isl_basic_map *bmap)
13163 {
13164 
13165 	if (!bmap)
13166 		return NULL;
13167 
13168 	if (!isl_basic_map_can_curry(bmap))
13169 		isl_die(bmap->ctx, isl_error_invalid,
13170 			"basic map cannot be curried", goto error);
13171 	bmap = isl_basic_map_cow(bmap);
13172 	if (!bmap)
13173 		return NULL;
13174 	bmap->dim = isl_space_curry(bmap->dim);
13175 	if (!bmap->dim)
13176 		goto error;
13177 	bmap = isl_basic_map_mark_final(bmap);
13178 	return bmap;
13179 error:
13180 	isl_basic_map_free(bmap);
13181 	return NULL;
13182 }
13183 
13184 /* Given a map (A -> B) -> C, return the corresponding map
13185  * A -> (B -> C).
13186  */
isl_map_curry(__isl_take isl_map * map)13187 __isl_give isl_map *isl_map_curry(__isl_take isl_map *map)
13188 {
13189 	return isl_map_change_space(map, &isl_map_can_curry,
13190 				    "map cannot be curried", &isl_space_curry);
13191 }
13192 
13193 /* Can isl_map_range_curry be applied to "map"?
13194  * That is, does it have a nested relation in its range,
13195  * the domain of which is itself a nested relation?
13196  */
isl_map_can_range_curry(__isl_keep isl_map * map)13197 isl_bool isl_map_can_range_curry(__isl_keep isl_map *map)
13198 {
13199 	if (!map)
13200 		return isl_bool_error;
13201 
13202 	return isl_space_can_range_curry(map->dim);
13203 }
13204 
13205 /* Given a map A -> ((B -> C) -> D), return the corresponding map
13206  * A -> (B -> (C -> D)).
13207  */
isl_map_range_curry(__isl_take isl_map * map)13208 __isl_give isl_map *isl_map_range_curry(__isl_take isl_map *map)
13209 {
13210 	return isl_map_change_space(map, &isl_map_can_range_curry,
13211 				    "map range cannot be curried",
13212 				    &isl_space_range_curry);
13213 }
13214 
13215 /* Can we apply isl_basic_map_uncurry to "bmap"?
13216  * That is, does it have a nested relation in its domain?
13217  */
isl_basic_map_can_uncurry(__isl_keep isl_basic_map * bmap)13218 isl_bool isl_basic_map_can_uncurry(__isl_keep isl_basic_map *bmap)
13219 {
13220 	if (!bmap)
13221 		return isl_bool_error;
13222 
13223 	return isl_space_can_uncurry(bmap->dim);
13224 }
13225 
13226 /* Can we apply isl_map_uncurry to "map"?
13227  * That is, does it have a nested relation in its domain?
13228  */
isl_map_can_uncurry(__isl_keep isl_map * map)13229 isl_bool isl_map_can_uncurry(__isl_keep isl_map *map)
13230 {
13231 	if (!map)
13232 		return isl_bool_error;
13233 
13234 	return isl_space_can_uncurry(map->dim);
13235 }
13236 
13237 /* Given a basic map A -> (B -> C), return the corresponding basic map
13238  * (A -> B) -> C.
13239  */
isl_basic_map_uncurry(__isl_take isl_basic_map * bmap)13240 __isl_give isl_basic_map *isl_basic_map_uncurry(__isl_take isl_basic_map *bmap)
13241 {
13242 
13243 	if (!bmap)
13244 		return NULL;
13245 
13246 	if (!isl_basic_map_can_uncurry(bmap))
13247 		isl_die(bmap->ctx, isl_error_invalid,
13248 			"basic map cannot be uncurried",
13249 			return isl_basic_map_free(bmap));
13250 	bmap = isl_basic_map_cow(bmap);
13251 	if (!bmap)
13252 		return NULL;
13253 	bmap->dim = isl_space_uncurry(bmap->dim);
13254 	if (!bmap->dim)
13255 		return isl_basic_map_free(bmap);
13256 	bmap = isl_basic_map_mark_final(bmap);
13257 	return bmap;
13258 }
13259 
13260 /* Given a map A -> (B -> C), return the corresponding map
13261  * (A -> B) -> C.
13262  */
isl_map_uncurry(__isl_take isl_map * map)13263 __isl_give isl_map *isl_map_uncurry(__isl_take isl_map *map)
13264 {
13265 	return isl_map_change_space(map, &isl_map_can_uncurry,
13266 				"map cannot be uncurried", &isl_space_uncurry);
13267 }
13268 
isl_set_equate(__isl_take isl_set * set,enum isl_dim_type type1,int pos1,enum isl_dim_type type2,int pos2)13269 __isl_give isl_set *isl_set_equate(__isl_take isl_set *set,
13270 	enum isl_dim_type type1, int pos1, enum isl_dim_type type2, int pos2)
13271 {
13272 	return isl_map_equate(set, type1, pos1, type2, pos2);
13273 }
13274 
13275 /* Construct a basic map where the given dimensions are equal to each other.
13276  */
equator(__isl_take isl_space * space,enum isl_dim_type type1,int pos1,enum isl_dim_type type2,int pos2)13277 static __isl_give isl_basic_map *equator(__isl_take isl_space *space,
13278 	enum isl_dim_type type1, int pos1, enum isl_dim_type type2, int pos2)
13279 {
13280 	isl_basic_map *bmap = NULL;
13281 	int i;
13282 	isl_size total;
13283 
13284 	total = isl_space_dim(space, isl_dim_all);
13285 	if (total < 0 ||
13286 	    isl_space_check_range(space, type1, pos1, 1) < 0 ||
13287 	    isl_space_check_range(space, type2, pos2, 1) < 0)
13288 		goto error;
13289 
13290 	if (type1 == type2 && pos1 == pos2)
13291 		return isl_basic_map_universe(space);
13292 
13293 	bmap = isl_basic_map_alloc_space(isl_space_copy(space), 0, 1, 0);
13294 	i = isl_basic_map_alloc_equality(bmap);
13295 	if (i < 0)
13296 		goto error;
13297 	isl_seq_clr(bmap->eq[i], 1 + total);
13298 	pos1 += isl_basic_map_offset(bmap, type1);
13299 	pos2 += isl_basic_map_offset(bmap, type2);
13300 	isl_int_set_si(bmap->eq[i][pos1], -1);
13301 	isl_int_set_si(bmap->eq[i][pos2], 1);
13302 	bmap = isl_basic_map_finalize(bmap);
13303 	isl_space_free(space);
13304 	return bmap;
13305 error:
13306 	isl_space_free(space);
13307 	isl_basic_map_free(bmap);
13308 	return NULL;
13309 }
13310 
13311 /* Add a constraint imposing that the given two dimensions are equal.
13312  */
isl_basic_map_equate(__isl_take isl_basic_map * bmap,enum isl_dim_type type1,int pos1,enum isl_dim_type type2,int pos2)13313 __isl_give isl_basic_map *isl_basic_map_equate(__isl_take isl_basic_map *bmap,
13314 	enum isl_dim_type type1, int pos1, enum isl_dim_type type2, int pos2)
13315 {
13316 	isl_basic_map *eq;
13317 
13318 	eq = equator(isl_basic_map_get_space(bmap), type1, pos1, type2, pos2);
13319 
13320 	bmap = isl_basic_map_intersect(bmap, eq);
13321 
13322 	return bmap;
13323 }
13324 
13325 /* Add a constraint imposing that the given two dimensions are equal.
13326  */
isl_map_equate(__isl_take isl_map * map,enum isl_dim_type type1,int pos1,enum isl_dim_type type2,int pos2)13327 __isl_give isl_map *isl_map_equate(__isl_take isl_map *map,
13328 	enum isl_dim_type type1, int pos1, enum isl_dim_type type2, int pos2)
13329 {
13330 	isl_basic_map *bmap;
13331 
13332 	bmap = equator(isl_map_get_space(map), type1, pos1, type2, pos2);
13333 
13334 	map = isl_map_intersect(map, isl_map_from_basic_map(bmap));
13335 
13336 	return map;
13337 }
13338 
13339 /* Add a constraint imposing that the given two dimensions have opposite values.
13340  */
isl_map_oppose(__isl_take isl_map * map,enum isl_dim_type type1,int pos1,enum isl_dim_type type2,int pos2)13341 __isl_give isl_map *isl_map_oppose(__isl_take isl_map *map,
13342 	enum isl_dim_type type1, int pos1, enum isl_dim_type type2, int pos2)
13343 {
13344 	isl_basic_map *bmap = NULL;
13345 	int i;
13346 	isl_size total;
13347 
13348 	if (isl_map_check_range(map, type1, pos1, 1) < 0)
13349 		return isl_map_free(map);
13350 	if (isl_map_check_range(map, type2, pos2, 1) < 0)
13351 		return isl_map_free(map);
13352 
13353 	total = isl_map_dim(map, isl_dim_all);
13354 	if (total < 0)
13355 		return isl_map_free(map);
13356 	bmap = isl_basic_map_alloc_space(isl_map_get_space(map), 0, 1, 0);
13357 	i = isl_basic_map_alloc_equality(bmap);
13358 	if (i < 0)
13359 		goto error;
13360 	isl_seq_clr(bmap->eq[i], 1 + total);
13361 	pos1 += isl_basic_map_offset(bmap, type1);
13362 	pos2 += isl_basic_map_offset(bmap, type2);
13363 	isl_int_set_si(bmap->eq[i][pos1], 1);
13364 	isl_int_set_si(bmap->eq[i][pos2], 1);
13365 	bmap = isl_basic_map_finalize(bmap);
13366 
13367 	map = isl_map_intersect(map, isl_map_from_basic_map(bmap));
13368 
13369 	return map;
13370 error:
13371 	isl_basic_map_free(bmap);
13372 	isl_map_free(map);
13373 	return NULL;
13374 }
13375 
13376 /* Construct a constraint imposing that the value of the first dimension is
13377  * greater than or equal to that of the second.
13378  */
constraint_order_ge(__isl_take isl_space * space,enum isl_dim_type type1,int pos1,enum isl_dim_type type2,int pos2)13379 static __isl_give isl_constraint *constraint_order_ge(
13380 	__isl_take isl_space *space, enum isl_dim_type type1, int pos1,
13381 	enum isl_dim_type type2, int pos2)
13382 {
13383 	isl_constraint *c;
13384 
13385 	if (isl_space_check_range(space, type1, pos1, 1) < 0 ||
13386 	    isl_space_check_range(space, type2, pos2, 1) < 0)
13387 		space = isl_space_free(space);
13388 	if (!space)
13389 		return NULL;
13390 
13391 	c = isl_constraint_alloc_inequality(isl_local_space_from_space(space));
13392 
13393 	if (type1 == type2 && pos1 == pos2)
13394 		return c;
13395 
13396 	c = isl_constraint_set_coefficient_si(c, type1, pos1, 1);
13397 	c = isl_constraint_set_coefficient_si(c, type2, pos2, -1);
13398 
13399 	return c;
13400 }
13401 
13402 /* Add a constraint imposing that the value of the first dimension is
13403  * greater than or equal to that of the second.
13404  */
isl_basic_map_order_ge(__isl_take isl_basic_map * bmap,enum isl_dim_type type1,int pos1,enum isl_dim_type type2,int pos2)13405 __isl_give isl_basic_map *isl_basic_map_order_ge(__isl_take isl_basic_map *bmap,
13406 	enum isl_dim_type type1, int pos1, enum isl_dim_type type2, int pos2)
13407 {
13408 	isl_constraint *c;
13409 	isl_space *space;
13410 
13411 	if (type1 == type2 && pos1 == pos2)
13412 		return bmap;
13413 	space = isl_basic_map_get_space(bmap);
13414 	c = constraint_order_ge(space, type1, pos1, type2, pos2);
13415 	bmap = isl_basic_map_add_constraint(bmap, c);
13416 
13417 	return bmap;
13418 }
13419 
13420 /* Add a constraint imposing that the value of the first dimension is
13421  * greater than or equal to that of the second.
13422  */
isl_map_order_ge(__isl_take isl_map * map,enum isl_dim_type type1,int pos1,enum isl_dim_type type2,int pos2)13423 __isl_give isl_map *isl_map_order_ge(__isl_take isl_map *map,
13424 	enum isl_dim_type type1, int pos1, enum isl_dim_type type2, int pos2)
13425 {
13426 	isl_constraint *c;
13427 	isl_space *space;
13428 
13429 	if (type1 == type2 && pos1 == pos2)
13430 		return map;
13431 	space = isl_map_get_space(map);
13432 	c = constraint_order_ge(space, type1, pos1, type2, pos2);
13433 	map = isl_map_add_constraint(map, c);
13434 
13435 	return map;
13436 }
13437 
13438 /* Add a constraint imposing that the value of the first dimension is
13439  * less than or equal to that of the second.
13440  */
isl_map_order_le(__isl_take isl_map * map,enum isl_dim_type type1,int pos1,enum isl_dim_type type2,int pos2)13441 __isl_give isl_map *isl_map_order_le(__isl_take isl_map *map,
13442 	enum isl_dim_type type1, int pos1, enum isl_dim_type type2, int pos2)
13443 {
13444 	return isl_map_order_ge(map, type2, pos2, type1, pos1);
13445 }
13446 
13447 /* Construct a basic map where the value of the first dimension is
13448  * greater than that of the second.
13449  */
greator(__isl_take isl_space * space,enum isl_dim_type type1,int pos1,enum isl_dim_type type2,int pos2)13450 static __isl_give isl_basic_map *greator(__isl_take isl_space *space,
13451 	enum isl_dim_type type1, int pos1, enum isl_dim_type type2, int pos2)
13452 {
13453 	isl_basic_map *bmap = NULL;
13454 	int i;
13455 	isl_size total;
13456 
13457 	if (isl_space_check_range(space, type1, pos1, 1) < 0 ||
13458 	    isl_space_check_range(space, type2, pos2, 1) < 0)
13459 		goto error;
13460 
13461 	if (type1 == type2 && pos1 == pos2)
13462 		return isl_basic_map_empty(space);
13463 
13464 	bmap = isl_basic_map_alloc_space(space, 0, 0, 1);
13465 	total = isl_basic_map_dim(bmap, isl_dim_all);
13466 	i = isl_basic_map_alloc_inequality(bmap);
13467 	if (total < 0 || i < 0)
13468 		return isl_basic_map_free(bmap);
13469 	isl_seq_clr(bmap->ineq[i], 1 + total);
13470 	pos1 += isl_basic_map_offset(bmap, type1);
13471 	pos2 += isl_basic_map_offset(bmap, type2);
13472 	isl_int_set_si(bmap->ineq[i][pos1], 1);
13473 	isl_int_set_si(bmap->ineq[i][pos2], -1);
13474 	isl_int_set_si(bmap->ineq[i][0], -1);
13475 	bmap = isl_basic_map_finalize(bmap);
13476 
13477 	return bmap;
13478 error:
13479 	isl_space_free(space);
13480 	isl_basic_map_free(bmap);
13481 	return NULL;
13482 }
13483 
13484 /* Add a constraint imposing that the value of the first dimension is
13485  * greater than that of the second.
13486  */
isl_basic_map_order_gt(__isl_take isl_basic_map * bmap,enum isl_dim_type type1,int pos1,enum isl_dim_type type2,int pos2)13487 __isl_give isl_basic_map *isl_basic_map_order_gt(__isl_take isl_basic_map *bmap,
13488 	enum isl_dim_type type1, int pos1, enum isl_dim_type type2, int pos2)
13489 {
13490 	isl_basic_map *gt;
13491 
13492 	gt = greator(isl_basic_map_get_space(bmap), type1, pos1, type2, pos2);
13493 
13494 	bmap = isl_basic_map_intersect(bmap, gt);
13495 
13496 	return bmap;
13497 }
13498 
13499 /* Add a constraint imposing that the value of the first dimension is
13500  * greater than that of the second.
13501  */
isl_map_order_gt(__isl_take isl_map * map,enum isl_dim_type type1,int pos1,enum isl_dim_type type2,int pos2)13502 __isl_give isl_map *isl_map_order_gt(__isl_take isl_map *map,
13503 	enum isl_dim_type type1, int pos1, enum isl_dim_type type2, int pos2)
13504 {
13505 	isl_basic_map *bmap;
13506 
13507 	bmap = greator(isl_map_get_space(map), type1, pos1, type2, pos2);
13508 
13509 	map = isl_map_intersect(map, isl_map_from_basic_map(bmap));
13510 
13511 	return map;
13512 }
13513 
13514 /* Add a constraint imposing that the value of the first dimension is
13515  * smaller than that of the second.
13516  */
isl_map_order_lt(__isl_take isl_map * map,enum isl_dim_type type1,int pos1,enum isl_dim_type type2,int pos2)13517 __isl_give isl_map *isl_map_order_lt(__isl_take isl_map *map,
13518 	enum isl_dim_type type1, int pos1, enum isl_dim_type type2, int pos2)
13519 {
13520 	return isl_map_order_gt(map, type2, pos2, type1, pos1);
13521 }
13522 
isl_basic_map_get_div(__isl_keep isl_basic_map * bmap,int pos)13523 __isl_give isl_aff *isl_basic_map_get_div(__isl_keep isl_basic_map *bmap,
13524 	int pos)
13525 {
13526 	isl_aff *div;
13527 	isl_local_space *ls;
13528 
13529 	if (!bmap)
13530 		return NULL;
13531 
13532 	if (!isl_basic_map_divs_known(bmap))
13533 		isl_die(isl_basic_map_get_ctx(bmap), isl_error_invalid,
13534 			"some divs are unknown", return NULL);
13535 
13536 	ls = isl_basic_map_get_local_space(bmap);
13537 	div = isl_local_space_get_div(ls, pos);
13538 	isl_local_space_free(ls);
13539 
13540 	return div;
13541 }
13542 
isl_basic_set_get_div(__isl_keep isl_basic_set * bset,int pos)13543 __isl_give isl_aff *isl_basic_set_get_div(__isl_keep isl_basic_set *bset,
13544 	int pos)
13545 {
13546 	return isl_basic_map_get_div(bset, pos);
13547 }
13548 
13549 /* Plug in "subs" for set dimension "pos" of "set".
13550  */
isl_set_substitute(__isl_take isl_set * set,unsigned pos,__isl_keep isl_aff * subs)13551 __isl_give isl_set *isl_set_substitute(__isl_take isl_set *set,
13552 	unsigned pos, __isl_keep isl_aff *subs)
13553 {
13554 	isl_multi_aff *ma;
13555 
13556 	if (set && isl_set_plain_is_empty(set))
13557 		return set;
13558 
13559 	ma = isl_multi_aff_identity_on_domain_space(isl_set_get_space(set));
13560 	ma = isl_multi_aff_set_aff(ma, pos, isl_aff_copy(subs));
13561 	return isl_set_preimage_multi_aff(set, ma);
13562 }
13563 
13564 /* Check if the range of "ma" is compatible with the domain or range
13565  * (depending on "type") of "bmap".
13566  */
check_basic_map_compatible_range_multi_aff(__isl_keep isl_basic_map * bmap,enum isl_dim_type type,__isl_keep isl_multi_aff * ma)13567 static isl_stat check_basic_map_compatible_range_multi_aff(
13568 	__isl_keep isl_basic_map *bmap, enum isl_dim_type type,
13569 	__isl_keep isl_multi_aff *ma)
13570 {
13571 	isl_bool m;
13572 	isl_space *ma_space;
13573 
13574 	ma_space = isl_multi_aff_get_space(ma);
13575 
13576 	m = isl_space_has_equal_params(bmap->dim, ma_space);
13577 	if (m < 0)
13578 		goto error;
13579 	if (!m)
13580 		isl_die(isl_basic_map_get_ctx(bmap), isl_error_invalid,
13581 			"parameters don't match", goto error);
13582 	m = isl_space_tuple_is_equal(bmap->dim, type, ma_space, isl_dim_out);
13583 	if (m < 0)
13584 		goto error;
13585 	if (!m)
13586 		isl_die(isl_basic_map_get_ctx(bmap), isl_error_invalid,
13587 			"spaces don't match", goto error);
13588 
13589 	isl_space_free(ma_space);
13590 	return isl_stat_ok;
13591 error:
13592 	isl_space_free(ma_space);
13593 	return isl_stat_error;
13594 }
13595 
13596 /* Copy the divs from "ma" to "bmap", adding zeros for the "n_before"
13597  * coefficients before the transformed range of dimensions,
13598  * the "n_after" coefficients after the transformed range of dimensions
13599  * and the coefficients of the other divs in "bmap".
13600  */
set_ma_divs(__isl_take isl_basic_map * bmap,__isl_keep isl_multi_aff * ma,int n_before,int n_after,int n_div)13601 static __isl_give isl_basic_map *set_ma_divs(__isl_take isl_basic_map *bmap,
13602 	__isl_keep isl_multi_aff *ma, int n_before, int n_after, int n_div)
13603 {
13604 	int i;
13605 	isl_size n_param;
13606 	isl_size n_set;
13607 	isl_local_space *ls;
13608 
13609 	if (n_div == 0)
13610 		return bmap;
13611 
13612 	ls = isl_aff_get_domain_local_space(ma->u.p[0]);
13613 	n_param = isl_local_space_dim(ls, isl_dim_param);
13614 	n_set = isl_local_space_dim(ls, isl_dim_set);
13615 	if (n_param < 0 || n_set < 0)
13616 		return isl_basic_map_free(bmap);
13617 
13618 	for (i = 0; i < n_div; ++i) {
13619 		int o_bmap = 0, o_ls = 0;
13620 
13621 		isl_seq_cpy(bmap->div[i], ls->div->row[i], 1 + 1 + n_param);
13622 		o_bmap += 1 + 1 + n_param;
13623 		o_ls += 1 + 1 + n_param;
13624 		isl_seq_clr(bmap->div[i] + o_bmap, n_before);
13625 		o_bmap += n_before;
13626 		isl_seq_cpy(bmap->div[i] + o_bmap,
13627 			    ls->div->row[i] + o_ls, n_set);
13628 		o_bmap += n_set;
13629 		o_ls += n_set;
13630 		isl_seq_clr(bmap->div[i] + o_bmap, n_after);
13631 		o_bmap += n_after;
13632 		isl_seq_cpy(bmap->div[i] + o_bmap,
13633 			    ls->div->row[i] + o_ls, n_div);
13634 		o_bmap += n_div;
13635 		o_ls += n_div;
13636 		isl_seq_clr(bmap->div[i] + o_bmap, bmap->n_div - n_div);
13637 		bmap = isl_basic_map_add_div_constraints(bmap, i);
13638 		if (!bmap)
13639 			goto error;
13640 	}
13641 
13642 	isl_local_space_free(ls);
13643 	return bmap;
13644 error:
13645 	isl_local_space_free(ls);
13646 	return isl_basic_map_free(bmap);
13647 }
13648 
13649 /* How many stride constraints does "ma" enforce?
13650  * That is, how many of the affine expressions have a denominator
13651  * different from one?
13652  */
multi_aff_strides(__isl_keep isl_multi_aff * ma)13653 static int multi_aff_strides(__isl_keep isl_multi_aff *ma)
13654 {
13655 	int i;
13656 	int strides = 0;
13657 
13658 	for (i = 0; i < ma->n; ++i)
13659 		if (!isl_int_is_one(ma->u.p[i]->v->el[0]))
13660 			strides++;
13661 
13662 	return strides;
13663 }
13664 
13665 /* For each affine expression in ma of the form
13666  *
13667  *	x_i = (f_i y + h_i)/m_i
13668  *
13669  * with m_i different from one, add a constraint to "bmap"
13670  * of the form
13671  *
13672  *	f_i y + h_i = m_i alpha_i
13673  *
13674  * with alpha_i an additional existentially quantified variable.
13675  *
13676  * The input variables of "ma" correspond to a subset of the variables
13677  * of "bmap".  There are "n_before" variables in "bmap" before this
13678  * subset and "n_after" variables after this subset.
13679  * The integer divisions of the affine expressions in "ma" are assumed
13680  * to have been aligned.  There are "n_div_ma" of them and
13681  * they appear first in "bmap", straight after the "n_after" variables.
13682  */
add_ma_strides(__isl_take isl_basic_map * bmap,__isl_keep isl_multi_aff * ma,int n_before,int n_after,int n_div_ma)13683 static __isl_give isl_basic_map *add_ma_strides(
13684 	__isl_take isl_basic_map *bmap, __isl_keep isl_multi_aff *ma,
13685 	int n_before, int n_after, int n_div_ma)
13686 {
13687 	int i, k;
13688 	int div;
13689 	isl_size total;
13690 	isl_size n_param;
13691 	isl_size n_in;
13692 
13693 	total = isl_basic_map_dim(bmap, isl_dim_all);
13694 	n_param = isl_multi_aff_dim(ma, isl_dim_param);
13695 	n_in = isl_multi_aff_dim(ma, isl_dim_in);
13696 	if (total < 0 || n_param < 0 || n_in < 0)
13697 		return isl_basic_map_free(bmap);
13698 	for (i = 0; i < ma->n; ++i) {
13699 		int o_bmap = 0, o_ma = 1;
13700 
13701 		if (isl_int_is_one(ma->u.p[i]->v->el[0]))
13702 			continue;
13703 		div = isl_basic_map_alloc_div(bmap);
13704 		k = isl_basic_map_alloc_equality(bmap);
13705 		if (div < 0 || k < 0)
13706 			goto error;
13707 		isl_int_set_si(bmap->div[div][0], 0);
13708 		isl_seq_cpy(bmap->eq[k] + o_bmap,
13709 			    ma->u.p[i]->v->el + o_ma, 1 + n_param);
13710 		o_bmap += 1 + n_param;
13711 		o_ma += 1 + n_param;
13712 		isl_seq_clr(bmap->eq[k] + o_bmap, n_before);
13713 		o_bmap += n_before;
13714 		isl_seq_cpy(bmap->eq[k] + o_bmap,
13715 			    ma->u.p[i]->v->el + o_ma, n_in);
13716 		o_bmap += n_in;
13717 		o_ma += n_in;
13718 		isl_seq_clr(bmap->eq[k] + o_bmap, n_after);
13719 		o_bmap += n_after;
13720 		isl_seq_cpy(bmap->eq[k] + o_bmap,
13721 			    ma->u.p[i]->v->el + o_ma, n_div_ma);
13722 		o_bmap += n_div_ma;
13723 		o_ma += n_div_ma;
13724 		isl_seq_clr(bmap->eq[k] + o_bmap, 1 + total - o_bmap);
13725 		isl_int_neg(bmap->eq[k][1 + total], ma->u.p[i]->v->el[0]);
13726 		total++;
13727 	}
13728 
13729 	return bmap;
13730 error:
13731 	isl_basic_map_free(bmap);
13732 	return NULL;
13733 }
13734 
13735 /* Replace the domain or range space (depending on "type) of "space" by "set".
13736  */
isl_space_set(__isl_take isl_space * space,enum isl_dim_type type,__isl_take isl_space * set)13737 static __isl_give isl_space *isl_space_set(__isl_take isl_space *space,
13738 	enum isl_dim_type type, __isl_take isl_space *set)
13739 {
13740 	if (type == isl_dim_in) {
13741 		space = isl_space_range(space);
13742 		space = isl_space_map_from_domain_and_range(set, space);
13743 	} else {
13744 		space = isl_space_domain(space);
13745 		space = isl_space_map_from_domain_and_range(space, set);
13746 	}
13747 
13748 	return space;
13749 }
13750 
13751 /* Compute the preimage of the domain or range (depending on "type")
13752  * of "bmap" under the function represented by "ma".
13753  * In other words, plug in "ma" in the domain or range of "bmap".
13754  * The result is a basic map that lives in the same space as "bmap"
13755  * except that the domain or range has been replaced by
13756  * the domain space of "ma".
13757  *
13758  * If bmap is represented by
13759  *
13760  *	A(p) + S u + B x + T v + C(divs) >= 0,
13761  *
13762  * where u and x are input and output dimensions if type == isl_dim_out
13763  * while x and v are input and output dimensions if type == isl_dim_in,
13764  * and ma is represented by
13765  *
13766  *	x = D(p) + F(y) + G(divs')
13767  *
13768  * then the result is
13769  *
13770  *	A(p) + B D(p) + S u + B F(y) + T v + B G(divs') + C(divs) >= 0
13771  *
13772  * The divs in the input set are similarly adjusted.
13773  * In particular
13774  *
13775  *	floor((a_i(p) + s u + b_i x + t v + c_i(divs))/n_i)
13776  *
13777  * becomes
13778  *
13779  *	floor((a_i(p) + b_i D(p) + s u + b_i F(y) + t v +
13780  *		B_i G(divs') + c_i(divs))/n_i)
13781  *
13782  * If bmap is not a rational map and if F(y) involves any denominators
13783  *
13784  *	x_i = (f_i y + h_i)/m_i
13785  *
13786  * then additional constraints are added to ensure that we only
13787  * map back integer points.  That is we enforce
13788  *
13789  *	f_i y + h_i = m_i alpha_i
13790  *
13791  * with alpha_i an additional existentially quantified variable.
13792  *
13793  * We first copy over the divs from "ma".
13794  * Then we add the modified constraints and divs from "bmap".
13795  * Finally, we add the stride constraints, if needed.
13796  */
isl_basic_map_preimage_multi_aff(__isl_take isl_basic_map * bmap,enum isl_dim_type type,__isl_take isl_multi_aff * ma)13797 __isl_give isl_basic_map *isl_basic_map_preimage_multi_aff(
13798 	__isl_take isl_basic_map *bmap, enum isl_dim_type type,
13799 	__isl_take isl_multi_aff *ma)
13800 {
13801 	int i, k;
13802 	isl_space *space;
13803 	isl_basic_map *res = NULL;
13804 	isl_size n_before, n_after, n_div_bmap, n_div_ma;
13805 	isl_int f, c1, c2, g;
13806 	isl_bool rational;
13807 	int strides;
13808 
13809 	isl_int_init(f);
13810 	isl_int_init(c1);
13811 	isl_int_init(c2);
13812 	isl_int_init(g);
13813 
13814 	ma = isl_multi_aff_align_divs(ma);
13815 	if (!bmap || !ma)
13816 		goto error;
13817 	if (check_basic_map_compatible_range_multi_aff(bmap, type, ma) < 0)
13818 		goto error;
13819 
13820 	if (type == isl_dim_in) {
13821 		n_before = 0;
13822 		n_after = isl_basic_map_dim(bmap, isl_dim_out);
13823 	} else {
13824 		n_before = isl_basic_map_dim(bmap, isl_dim_in);
13825 		n_after = 0;
13826 	}
13827 	n_div_bmap = isl_basic_map_dim(bmap, isl_dim_div);
13828 	n_div_ma = ma->n ? isl_aff_dim(ma->u.p[0], isl_dim_div) : 0;
13829 	if (n_before < 0 || n_after < 0 || n_div_bmap < 0 || n_div_ma < 0)
13830 		goto error;
13831 
13832 	space = isl_multi_aff_get_domain_space(ma);
13833 	space = isl_space_set(isl_basic_map_get_space(bmap), type, space);
13834 	rational = isl_basic_map_is_rational(bmap);
13835 	strides = rational ? 0 : multi_aff_strides(ma);
13836 	res = isl_basic_map_alloc_space(space, n_div_ma + n_div_bmap + strides,
13837 			    bmap->n_eq + strides, bmap->n_ineq + 2 * n_div_ma);
13838 	if (rational)
13839 		res = isl_basic_map_set_rational(res);
13840 
13841 	for (i = 0; i < n_div_ma + n_div_bmap; ++i)
13842 		if (isl_basic_map_alloc_div(res) < 0)
13843 			goto error;
13844 
13845 	res = set_ma_divs(res, ma, n_before, n_after, n_div_ma);
13846 	if (!res)
13847 		goto error;
13848 
13849 	for (i = 0; i < bmap->n_eq; ++i) {
13850 		k = isl_basic_map_alloc_equality(res);
13851 		if (k < 0)
13852 			goto error;
13853 		if (isl_seq_preimage(res->eq[k], bmap->eq[i], ma, n_before,
13854 				    n_after, n_div_ma, n_div_bmap,
13855 				    f, c1, c2, g, 0) < 0)
13856 			goto error;
13857 	}
13858 
13859 	for (i = 0; i < bmap->n_ineq; ++i) {
13860 		k = isl_basic_map_alloc_inequality(res);
13861 		if (k < 0)
13862 			goto error;
13863 		if (isl_seq_preimage(res->ineq[k], bmap->ineq[i], ma, n_before,
13864 				    n_after, n_div_ma, n_div_bmap,
13865 				    f, c1, c2, g, 0) < 0)
13866 			goto error;
13867 	}
13868 
13869 	for (i = 0; i < bmap->n_div; ++i) {
13870 		if (isl_int_is_zero(bmap->div[i][0])) {
13871 			isl_int_set_si(res->div[n_div_ma + i][0], 0);
13872 			continue;
13873 		}
13874 		if (isl_seq_preimage(res->div[n_div_ma + i], bmap->div[i], ma,
13875 				    n_before, n_after, n_div_ma, n_div_bmap,
13876 				    f, c1, c2, g, 1) < 0)
13877 			goto error;
13878 	}
13879 
13880 	if (strides)
13881 		res = add_ma_strides(res, ma, n_before, n_after, n_div_ma);
13882 
13883 	isl_int_clear(f);
13884 	isl_int_clear(c1);
13885 	isl_int_clear(c2);
13886 	isl_int_clear(g);
13887 	isl_basic_map_free(bmap);
13888 	isl_multi_aff_free(ma);
13889 	res = isl_basic_map_simplify(res);
13890 	return isl_basic_map_finalize(res);
13891 error:
13892 	isl_int_clear(f);
13893 	isl_int_clear(c1);
13894 	isl_int_clear(c2);
13895 	isl_int_clear(g);
13896 	isl_basic_map_free(bmap);
13897 	isl_multi_aff_free(ma);
13898 	isl_basic_map_free(res);
13899 	return NULL;
13900 }
13901 
13902 /* Compute the preimage of "bset" under the function represented by "ma".
13903  * In other words, plug in "ma" in "bset".  The result is a basic set
13904  * that lives in the domain space of "ma".
13905  */
isl_basic_set_preimage_multi_aff(__isl_take isl_basic_set * bset,__isl_take isl_multi_aff * ma)13906 __isl_give isl_basic_set *isl_basic_set_preimage_multi_aff(
13907 	__isl_take isl_basic_set *bset, __isl_take isl_multi_aff *ma)
13908 {
13909 	return isl_basic_map_preimage_multi_aff(bset, isl_dim_set, ma);
13910 }
13911 
13912 /* Compute the preimage of the domain of "bmap" under the function
13913  * represented by "ma".
13914  * In other words, plug in "ma" in the domain of "bmap".
13915  * The result is a basic map that lives in the same space as "bmap"
13916  * except that the domain has been replaced by the domain space of "ma".
13917  */
isl_basic_map_preimage_domain_multi_aff(__isl_take isl_basic_map * bmap,__isl_take isl_multi_aff * ma)13918 __isl_give isl_basic_map *isl_basic_map_preimage_domain_multi_aff(
13919 	__isl_take isl_basic_map *bmap, __isl_take isl_multi_aff *ma)
13920 {
13921 	return isl_basic_map_preimage_multi_aff(bmap, isl_dim_in, ma);
13922 }
13923 
13924 /* Compute the preimage of the range of "bmap" under the function
13925  * represented by "ma".
13926  * In other words, plug in "ma" in the range of "bmap".
13927  * The result is a basic map that lives in the same space as "bmap"
13928  * except that the range has been replaced by the domain space of "ma".
13929  */
isl_basic_map_preimage_range_multi_aff(__isl_take isl_basic_map * bmap,__isl_take isl_multi_aff * ma)13930 __isl_give isl_basic_map *isl_basic_map_preimage_range_multi_aff(
13931 	__isl_take isl_basic_map *bmap, __isl_take isl_multi_aff *ma)
13932 {
13933 	return isl_basic_map_preimage_multi_aff(bmap, isl_dim_out, ma);
13934 }
13935 
13936 /* Check if the range of "ma" is compatible with the domain or range
13937  * (depending on "type") of "map".
13938  * Return isl_stat_error if anything is wrong.
13939  */
check_map_compatible_range_multi_aff(__isl_keep isl_map * map,enum isl_dim_type type,__isl_keep isl_multi_aff * ma)13940 static isl_stat check_map_compatible_range_multi_aff(
13941 	__isl_keep isl_map *map, enum isl_dim_type type,
13942 	__isl_keep isl_multi_aff *ma)
13943 {
13944 	isl_bool m;
13945 	isl_space *ma_space;
13946 
13947 	ma_space = isl_multi_aff_get_space(ma);
13948 	m = isl_map_space_tuple_is_equal(map, type, ma_space, isl_dim_out);
13949 	isl_space_free(ma_space);
13950 	if (m < 0)
13951 		return isl_stat_error;
13952 	if (!m)
13953 		isl_die(isl_map_get_ctx(map), isl_error_invalid,
13954 			"spaces don't match", return isl_stat_error);
13955 	return isl_stat_ok;
13956 }
13957 
13958 /* Compute the preimage of the domain or range (depending on "type")
13959  * of "map" under the function represented by "ma".
13960  * In other words, plug in "ma" in the domain or range of "map".
13961  * The result is a map that lives in the same space as "map"
13962  * except that the domain or range has been replaced by
13963  * the domain space of "ma".
13964  *
13965  * The parameters are assumed to have been aligned.
13966  */
map_preimage_multi_aff(__isl_take isl_map * map,enum isl_dim_type type,__isl_take isl_multi_aff * ma)13967 static __isl_give isl_map *map_preimage_multi_aff(__isl_take isl_map *map,
13968 	enum isl_dim_type type, __isl_take isl_multi_aff *ma)
13969 {
13970 	int i;
13971 	isl_space *space;
13972 
13973 	map = isl_map_cow(map);
13974 	ma = isl_multi_aff_align_divs(ma);
13975 	if (!map || !ma)
13976 		goto error;
13977 	if (check_map_compatible_range_multi_aff(map, type, ma) < 0)
13978 		goto error;
13979 
13980 	for (i = 0; i < map->n; ++i) {
13981 		map->p[i] = isl_basic_map_preimage_multi_aff(map->p[i], type,
13982 							isl_multi_aff_copy(ma));
13983 		if (!map->p[i])
13984 			goto error;
13985 	}
13986 
13987 	space = isl_multi_aff_get_domain_space(ma);
13988 	space = isl_space_set(isl_map_get_space(map), type, space);
13989 
13990 	isl_space_free(isl_map_take_space(map));
13991 	map = isl_map_restore_space(map, space);
13992 	if (!map)
13993 		goto error;
13994 
13995 	isl_multi_aff_free(ma);
13996 	if (map->n > 1)
13997 		ISL_F_CLR(map, ISL_MAP_DISJOINT);
13998 	ISL_F_CLR(map, ISL_SET_NORMALIZED);
13999 	return map;
14000 error:
14001 	isl_multi_aff_free(ma);
14002 	isl_map_free(map);
14003 	return NULL;
14004 }
14005 
14006 /* Compute the preimage of the domain or range (depending on "type")
14007  * of "map" under the function represented by "ma".
14008  * In other words, plug in "ma" in the domain or range of "map".
14009  * The result is a map that lives in the same space as "map"
14010  * except that the domain or range has been replaced by
14011  * the domain space of "ma".
14012  */
isl_map_preimage_multi_aff(__isl_take isl_map * map,enum isl_dim_type type,__isl_take isl_multi_aff * ma)14013 __isl_give isl_map *isl_map_preimage_multi_aff(__isl_take isl_map *map,
14014 	enum isl_dim_type type, __isl_take isl_multi_aff *ma)
14015 {
14016 	isl_bool aligned;
14017 
14018 	if (!map || !ma)
14019 		goto error;
14020 
14021 	aligned = isl_map_space_has_equal_params(map, ma->space);
14022 	if (aligned < 0)
14023 		goto error;
14024 	if (aligned)
14025 		return map_preimage_multi_aff(map, type, ma);
14026 
14027 	if (isl_map_check_named_params(map) < 0)
14028 		goto error;
14029 	if (!isl_space_has_named_params(ma->space))
14030 		isl_die(map->ctx, isl_error_invalid,
14031 			"unaligned unnamed parameters", goto error);
14032 	map = isl_map_align_params(map, isl_multi_aff_get_space(ma));
14033 	ma = isl_multi_aff_align_params(ma, isl_map_get_space(map));
14034 
14035 	return map_preimage_multi_aff(map, type, ma);
14036 error:
14037 	isl_multi_aff_free(ma);
14038 	return isl_map_free(map);
14039 }
14040 
14041 /* Compute the preimage of "set" under the function represented by "ma".
14042  * In other words, plug in "ma" in "set".  The result is a set
14043  * that lives in the domain space of "ma".
14044  */
isl_set_preimage_multi_aff(__isl_take isl_set * set,__isl_take isl_multi_aff * ma)14045 __isl_give isl_set *isl_set_preimage_multi_aff(__isl_take isl_set *set,
14046 	__isl_take isl_multi_aff *ma)
14047 {
14048 	return isl_map_preimage_multi_aff(set, isl_dim_set, ma);
14049 }
14050 
14051 /* Compute the preimage of the domain of "map" under the function
14052  * represented by "ma".
14053  * In other words, plug in "ma" in the domain of "map".
14054  * The result is a map that lives in the same space as "map"
14055  * except that the domain has been replaced by the domain space of "ma".
14056  */
isl_map_preimage_domain_multi_aff(__isl_take isl_map * map,__isl_take isl_multi_aff * ma)14057 __isl_give isl_map *isl_map_preimage_domain_multi_aff(__isl_take isl_map *map,
14058 	__isl_take isl_multi_aff *ma)
14059 {
14060 	return isl_map_preimage_multi_aff(map, isl_dim_in, ma);
14061 }
14062 
14063 /* Compute the preimage of the range of "map" under the function
14064  * represented by "ma".
14065  * In other words, plug in "ma" in the range of "map".
14066  * The result is a map that lives in the same space as "map"
14067  * except that the range has been replaced by the domain space of "ma".
14068  */
isl_map_preimage_range_multi_aff(__isl_take isl_map * map,__isl_take isl_multi_aff * ma)14069 __isl_give isl_map *isl_map_preimage_range_multi_aff(__isl_take isl_map *map,
14070 	__isl_take isl_multi_aff *ma)
14071 {
14072 	return isl_map_preimage_multi_aff(map, isl_dim_out, ma);
14073 }
14074 
14075 /* Compute the preimage of "map" under the function represented by "pma".
14076  * In other words, plug in "pma" in the domain or range of "map".
14077  * The result is a map that lives in the same space as "map",
14078  * except that the space of type "type" has been replaced by
14079  * the domain space of "pma".
14080  *
14081  * The parameters of "map" and "pma" are assumed to have been aligned.
14082  */
isl_map_preimage_pw_multi_aff_aligned(__isl_take isl_map * map,enum isl_dim_type type,__isl_take isl_pw_multi_aff * pma)14083 static __isl_give isl_map *isl_map_preimage_pw_multi_aff_aligned(
14084 	__isl_take isl_map *map, enum isl_dim_type type,
14085 	__isl_take isl_pw_multi_aff *pma)
14086 {
14087 	int i;
14088 	isl_map *res;
14089 
14090 	if (!pma)
14091 		goto error;
14092 
14093 	if (pma->n == 0) {
14094 		isl_pw_multi_aff_free(pma);
14095 		res = isl_map_empty(isl_map_get_space(map));
14096 		isl_map_free(map);
14097 		return res;
14098 	}
14099 
14100 	res = isl_map_preimage_multi_aff(isl_map_copy(map), type,
14101 					isl_multi_aff_copy(pma->p[0].maff));
14102 	if (type == isl_dim_in)
14103 		res = isl_map_intersect_domain(res,
14104 						isl_map_copy(pma->p[0].set));
14105 	else
14106 		res = isl_map_intersect_range(res,
14107 						isl_map_copy(pma->p[0].set));
14108 
14109 	for (i = 1; i < pma->n; ++i) {
14110 		isl_map *res_i;
14111 
14112 		res_i = isl_map_preimage_multi_aff(isl_map_copy(map), type,
14113 					isl_multi_aff_copy(pma->p[i].maff));
14114 		if (type == isl_dim_in)
14115 			res_i = isl_map_intersect_domain(res_i,
14116 						isl_map_copy(pma->p[i].set));
14117 		else
14118 			res_i = isl_map_intersect_range(res_i,
14119 						isl_map_copy(pma->p[i].set));
14120 		res = isl_map_union(res, res_i);
14121 	}
14122 
14123 	isl_pw_multi_aff_free(pma);
14124 	isl_map_free(map);
14125 	return res;
14126 error:
14127 	isl_pw_multi_aff_free(pma);
14128 	isl_map_free(map);
14129 	return NULL;
14130 }
14131 
14132 /* Compute the preimage of "map" under the function represented by "pma".
14133  * In other words, plug in "pma" in the domain or range of "map".
14134  * The result is a map that lives in the same space as "map",
14135  * except that the space of type "type" has been replaced by
14136  * the domain space of "pma".
14137  */
isl_map_preimage_pw_multi_aff(__isl_take isl_map * map,enum isl_dim_type type,__isl_take isl_pw_multi_aff * pma)14138 __isl_give isl_map *isl_map_preimage_pw_multi_aff(__isl_take isl_map *map,
14139 	enum isl_dim_type type, __isl_take isl_pw_multi_aff *pma)
14140 {
14141 	isl_bool aligned;
14142 
14143 	if (!map || !pma)
14144 		goto error;
14145 
14146 	aligned = isl_map_space_has_equal_params(map, pma->dim);
14147 	if (aligned < 0)
14148 		goto error;
14149 	if (aligned)
14150 		return isl_map_preimage_pw_multi_aff_aligned(map, type, pma);
14151 
14152 	if (isl_map_check_named_params(map) < 0)
14153 		goto error;
14154 	if (isl_pw_multi_aff_check_named_params(pma) < 0)
14155 		goto error;
14156 	map = isl_map_align_params(map, isl_pw_multi_aff_get_space(pma));
14157 	pma = isl_pw_multi_aff_align_params(pma, isl_map_get_space(map));
14158 
14159 	return isl_map_preimage_pw_multi_aff_aligned(map, type, pma);
14160 error:
14161 	isl_pw_multi_aff_free(pma);
14162 	return isl_map_free(map);
14163 }
14164 
14165 /* Compute the preimage of "set" under the function represented by "pma".
14166  * In other words, plug in "pma" in "set".  The result is a set
14167  * that lives in the domain space of "pma".
14168  */
isl_set_preimage_pw_multi_aff(__isl_take isl_set * set,__isl_take isl_pw_multi_aff * pma)14169 __isl_give isl_set *isl_set_preimage_pw_multi_aff(__isl_take isl_set *set,
14170 	__isl_take isl_pw_multi_aff *pma)
14171 {
14172 	return isl_map_preimage_pw_multi_aff(set, isl_dim_set, pma);
14173 }
14174 
14175 /* Compute the preimage of the domain of "map" under the function
14176  * represented by "pma".
14177  * In other words, plug in "pma" in the domain of "map".
14178  * The result is a map that lives in the same space as "map",
14179  * except that domain space has been replaced by the domain space of "pma".
14180  */
isl_map_preimage_domain_pw_multi_aff(__isl_take isl_map * map,__isl_take isl_pw_multi_aff * pma)14181 __isl_give isl_map *isl_map_preimage_domain_pw_multi_aff(
14182 	__isl_take isl_map *map, __isl_take isl_pw_multi_aff *pma)
14183 {
14184 	return isl_map_preimage_pw_multi_aff(map, isl_dim_in, pma);
14185 }
14186 
14187 /* Compute the preimage of the range of "map" under the function
14188  * represented by "pma".
14189  * In other words, plug in "pma" in the range of "map".
14190  * The result is a map that lives in the same space as "map",
14191  * except that range space has been replaced by the domain space of "pma".
14192  */
isl_map_preimage_range_pw_multi_aff(__isl_take isl_map * map,__isl_take isl_pw_multi_aff * pma)14193 __isl_give isl_map *isl_map_preimage_range_pw_multi_aff(
14194 	__isl_take isl_map *map, __isl_take isl_pw_multi_aff *pma)
14195 {
14196 	return isl_map_preimage_pw_multi_aff(map, isl_dim_out, pma);
14197 }
14198 
14199 /* Compute the preimage of "map" under the function represented by "mpa".
14200  * In other words, plug in "mpa" in the domain or range of "map".
14201  * The result is a map that lives in the same space as "map",
14202  * except that the space of type "type" has been replaced by
14203  * the domain space of "mpa".
14204  *
14205  * If the map does not involve any constraints that refer to the
14206  * dimensions of the substituted space, then the only possible
14207  * effect of "mpa" on the map is to map the space to a different space.
14208  * We create a separate isl_multi_aff to effectuate this change
14209  * in order to avoid spurious splitting of the map along the pieces
14210  * of "mpa".
14211  * If "mpa" has a non-trivial explicit domain, however,
14212  * then the full substitution should be performed.
14213  */
isl_map_preimage_multi_pw_aff(__isl_take isl_map * map,enum isl_dim_type type,__isl_take isl_multi_pw_aff * mpa)14214 __isl_give isl_map *isl_map_preimage_multi_pw_aff(__isl_take isl_map *map,
14215 	enum isl_dim_type type, __isl_take isl_multi_pw_aff *mpa)
14216 {
14217 	isl_size n;
14218 	isl_bool full;
14219 	isl_pw_multi_aff *pma;
14220 
14221 	n = isl_map_dim(map, type);
14222 	if (n < 0 || !mpa)
14223 		goto error;
14224 
14225 	full = isl_map_involves_dims(map, type, 0, n);
14226 	if (full >= 0 && !full)
14227 		full = isl_multi_pw_aff_has_non_trivial_domain(mpa);
14228 	if (full < 0)
14229 		goto error;
14230 	if (!full) {
14231 		isl_space *space;
14232 		isl_multi_aff *ma;
14233 
14234 		space = isl_multi_pw_aff_get_space(mpa);
14235 		isl_multi_pw_aff_free(mpa);
14236 		ma = isl_multi_aff_zero(space);
14237 		return isl_map_preimage_multi_aff(map, type, ma);
14238 	}
14239 
14240 	pma = isl_pw_multi_aff_from_multi_pw_aff(mpa);
14241 	return isl_map_preimage_pw_multi_aff(map, type, pma);
14242 error:
14243 	isl_map_free(map);
14244 	isl_multi_pw_aff_free(mpa);
14245 	return NULL;
14246 }
14247 
14248 /* Compute the preimage of "map" under the function represented by "mpa".
14249  * In other words, plug in "mpa" in the domain "map".
14250  * The result is a map that lives in the same space as "map",
14251  * except that domain space has been replaced by the domain space of "mpa".
14252  */
isl_map_preimage_domain_multi_pw_aff(__isl_take isl_map * map,__isl_take isl_multi_pw_aff * mpa)14253 __isl_give isl_map *isl_map_preimage_domain_multi_pw_aff(
14254 	__isl_take isl_map *map, __isl_take isl_multi_pw_aff *mpa)
14255 {
14256 	return isl_map_preimage_multi_pw_aff(map, isl_dim_in, mpa);
14257 }
14258 
14259 /* Compute the preimage of "set" by the function represented by "mpa".
14260  * In other words, plug in "mpa" in "set".
14261  */
isl_set_preimage_multi_pw_aff(__isl_take isl_set * set,__isl_take isl_multi_pw_aff * mpa)14262 __isl_give isl_set *isl_set_preimage_multi_pw_aff(__isl_take isl_set *set,
14263 	__isl_take isl_multi_pw_aff *mpa)
14264 {
14265 	return isl_map_preimage_multi_pw_aff(set, isl_dim_set, mpa);
14266 }
14267 
14268 /* Return a copy of the equality constraints of "bset" as a matrix.
14269  */
isl_basic_set_extract_equalities(__isl_keep isl_basic_set * bset)14270 __isl_give isl_mat *isl_basic_set_extract_equalities(
14271 	__isl_keep isl_basic_set *bset)
14272 {
14273 	isl_ctx *ctx;
14274 	isl_size total;
14275 
14276 	total = isl_basic_set_dim(bset, isl_dim_all);
14277 	if (total < 0)
14278 		return NULL;
14279 
14280 	ctx = isl_basic_set_get_ctx(bset);
14281 	return isl_mat_sub_alloc6(ctx, bset->eq, 0, bset->n_eq, 0, 1 + total);
14282 }
14283 
14284 /* Are the "n" "coefficients" starting at "first" of the integer division
14285  * expressions at position "pos1" in "bmap1" and "pos2" in "bmap2" equal
14286  * to each other?
14287  * The "coefficient" at position 0 is the denominator.
14288  * The "coefficient" at position 1 is the constant term.
14289  */
isl_basic_map_equal_div_expr_part(__isl_keep isl_basic_map * bmap1,int pos1,__isl_keep isl_basic_map * bmap2,int pos2,unsigned first,unsigned n)14290 isl_bool isl_basic_map_equal_div_expr_part(__isl_keep isl_basic_map *bmap1,
14291 	int pos1, __isl_keep isl_basic_map *bmap2, int pos2,
14292 	unsigned first, unsigned n)
14293 {
14294 	if (isl_basic_map_check_range(bmap1, isl_dim_div, pos1, 1) < 0)
14295 		return isl_bool_error;
14296 	if (isl_basic_map_check_range(bmap2, isl_dim_div, pos2, 1) < 0)
14297 		return isl_bool_error;
14298 	return isl_seq_eq(bmap1->div[pos1] + first,
14299 			  bmap2->div[pos2] + first, n);
14300 }
14301 
14302 /* Are the integer division expressions at position "pos1" in "bmap1" and
14303  * "pos2" in "bmap2" equal to each other, except that the constant terms
14304  * are different?
14305  */
isl_basic_map_equal_div_expr_except_constant(__isl_keep isl_basic_map * bmap1,int pos1,__isl_keep isl_basic_map * bmap2,int pos2)14306 isl_bool isl_basic_map_equal_div_expr_except_constant(
14307 	__isl_keep isl_basic_map *bmap1, int pos1,
14308 	__isl_keep isl_basic_map *bmap2, int pos2)
14309 {
14310 	isl_bool equal;
14311 	isl_size total, total2;
14312 
14313 	total = isl_basic_map_dim(bmap1, isl_dim_all);
14314 	total2 = isl_basic_map_dim(bmap2, isl_dim_all);
14315 	if (total < 0 || total2 < 0)
14316 		return isl_bool_error;
14317 	if (total != total2)
14318 		isl_die(isl_basic_map_get_ctx(bmap1), isl_error_invalid,
14319 			"incomparable div expressions", return isl_bool_error);
14320 	equal = isl_basic_map_equal_div_expr_part(bmap1, pos1, bmap2, pos2,
14321 						0, 1);
14322 	if (equal < 0 || !equal)
14323 		return equal;
14324 	equal = isl_basic_map_equal_div_expr_part(bmap1, pos1, bmap2, pos2,
14325 						1, 1);
14326 	if (equal < 0 || equal)
14327 		return isl_bool_not(equal);
14328 	return isl_basic_map_equal_div_expr_part(bmap1, pos1, bmap2, pos2,
14329 						2, total);
14330 }
14331 
14332 /* Replace the numerator of the constant term of the integer division
14333  * expression at position "div" in "bmap" by "value".
14334  * The caller guarantees that this does not change the meaning
14335  * of the input.
14336  */
isl_basic_map_set_div_expr_constant_num_si_inplace(__isl_take isl_basic_map * bmap,int div,int value)14337 __isl_give isl_basic_map *isl_basic_map_set_div_expr_constant_num_si_inplace(
14338 	__isl_take isl_basic_map *bmap, int div, int value)
14339 {
14340 	if (isl_basic_map_check_range(bmap, isl_dim_div, div, 1) < 0)
14341 		return isl_basic_map_free(bmap);
14342 
14343 	isl_int_set_si(bmap->div[div][1], value);
14344 
14345 	return bmap;
14346 }
14347 
14348 /* Is the point "inner" internal to inequality constraint "ineq"
14349  * of "bset"?
14350  * The point is considered to be internal to the inequality constraint,
14351  * if it strictly lies on the positive side of the inequality constraint,
14352  * or if it lies on the constraint and the constraint is lexico-positive.
14353  */
is_internal(__isl_keep isl_vec * inner,__isl_keep isl_basic_set * bset,int ineq)14354 static isl_bool is_internal(__isl_keep isl_vec *inner,
14355 	__isl_keep isl_basic_set *bset, int ineq)
14356 {
14357 	isl_ctx *ctx;
14358 	int pos;
14359 	isl_size total;
14360 
14361 	if (!inner || !bset)
14362 		return isl_bool_error;
14363 
14364 	ctx = isl_basic_set_get_ctx(bset);
14365 	isl_seq_inner_product(inner->el, bset->ineq[ineq], inner->size,
14366 				&ctx->normalize_gcd);
14367 	if (!isl_int_is_zero(ctx->normalize_gcd))
14368 		return isl_int_is_nonneg(ctx->normalize_gcd);
14369 
14370 	total = isl_basic_set_dim(bset, isl_dim_all);
14371 	if (total < 0)
14372 		return isl_bool_error;
14373 	pos = isl_seq_first_non_zero(bset->ineq[ineq] + 1, total);
14374 	return isl_int_is_pos(bset->ineq[ineq][1 + pos]);
14375 }
14376 
14377 /* Tighten the inequality constraints of "bset" that are outward with respect
14378  * to the point "vec".
14379  * That is, tighten the constraints that are not satisfied by "vec".
14380  *
14381  * "vec" is a point internal to some superset S of "bset" that is used
14382  * to make the subsets of S disjoint, by tightening one half of the constraints
14383  * that separate two subsets.  In particular, the constraints of S
14384  * are all satisfied by "vec" and should not be tightened.
14385  * Of the internal constraints, those that have "vec" on the outside
14386  * are tightened.  The shared facet is included in the adjacent subset
14387  * with the opposite constraint.
14388  * For constraints that saturate "vec", this criterion cannot be used
14389  * to determine which of the two sides should be tightened.
14390  * Instead, the sign of the first non-zero coefficient is used
14391  * to make this choice.  Note that this second criterion is never used
14392  * on the constraints of S since "vec" is interior to "S".
14393  */
isl_basic_set_tighten_outward(__isl_take isl_basic_set * bset,__isl_keep isl_vec * vec)14394 __isl_give isl_basic_set *isl_basic_set_tighten_outward(
14395 	__isl_take isl_basic_set *bset, __isl_keep isl_vec *vec)
14396 {
14397 	int j;
14398 
14399 	bset = isl_basic_set_cow(bset);
14400 	if (!bset)
14401 		return NULL;
14402 	for (j = 0; j < bset->n_ineq; ++j) {
14403 		isl_bool internal;
14404 
14405 		internal = is_internal(vec, bset, j);
14406 		if (internal < 0)
14407 			return isl_basic_set_free(bset);
14408 		if (internal)
14409 			continue;
14410 		isl_int_sub_ui(bset->ineq[j][0], bset->ineq[j][0], 1);
14411 	}
14412 
14413 	return bset;
14414 }
14415 
14416 /* Replace the variables x of type "type" starting at "first" in "bmap"
14417  * by x' with x = M x' with M the matrix trans.
14418  * That is, replace the corresponding coefficients c by c M.
14419  *
14420  * The transformation matrix should be a square matrix.
14421  */
isl_basic_map_transform_dims(__isl_take isl_basic_map * bmap,enum isl_dim_type type,unsigned first,__isl_take isl_mat * trans)14422 __isl_give isl_basic_map *isl_basic_map_transform_dims(
14423 	__isl_take isl_basic_map *bmap, enum isl_dim_type type, unsigned first,
14424 	__isl_take isl_mat *trans)
14425 {
14426 	unsigned pos;
14427 
14428 	bmap = isl_basic_map_cow(bmap);
14429 	if (!bmap || !trans)
14430 		goto error;
14431 
14432 	if (trans->n_row != trans->n_col)
14433 		isl_die(trans->ctx, isl_error_invalid,
14434 			"expecting square transformation matrix", goto error);
14435 	if (isl_basic_map_check_range(bmap, type, first, trans->n_row) < 0)
14436 		goto error;
14437 
14438 	pos = isl_basic_map_offset(bmap, type) + first;
14439 
14440 	if (isl_mat_sub_transform(bmap->eq, bmap->n_eq, pos,
14441 			isl_mat_copy(trans)) < 0)
14442 		goto error;
14443 	if (isl_mat_sub_transform(bmap->ineq, bmap->n_ineq, pos,
14444 		      isl_mat_copy(trans)) < 0)
14445 		goto error;
14446 	if (isl_mat_sub_transform(bmap->div, bmap->n_div, 1 + pos,
14447 		      isl_mat_copy(trans)) < 0)
14448 		goto error;
14449 
14450 	ISL_F_CLR(bmap, ISL_BASIC_MAP_SORTED);
14451 	ISL_F_CLR(bmap, ISL_BASIC_MAP_NORMALIZED_DIVS);
14452 
14453 	isl_mat_free(trans);
14454 	return bmap;
14455 error:
14456 	isl_mat_free(trans);
14457 	isl_basic_map_free(bmap);
14458 	return NULL;
14459 }
14460 
14461 /* Replace the variables x of type "type" starting at "first" in "bset"
14462  * by x' with x = M x' with M the matrix trans.
14463  * That is, replace the corresponding coefficients c by c M.
14464  *
14465  * The transformation matrix should be a square matrix.
14466  */
isl_basic_set_transform_dims(__isl_take isl_basic_set * bset,enum isl_dim_type type,unsigned first,__isl_take isl_mat * trans)14467 __isl_give isl_basic_set *isl_basic_set_transform_dims(
14468 	__isl_take isl_basic_set *bset, enum isl_dim_type type, unsigned first,
14469 	__isl_take isl_mat *trans)
14470 {
14471 	return isl_basic_map_transform_dims(bset, type, first, trans);
14472 }
14473