1 /*
2  * Copyright 2008-2009 Katholieke Universiteit Leuven
3  * Copyright 2010      INRIA Saclay
4  * Copyright 2012-2013 Ecole Normale Superieure
5  * Copyright 2019      Cerebras Systems
6  *
7  * Use of this software is governed by the MIT license
8  *
9  * Written by Sven Verdoolaege, K.U.Leuven, Departement
10  * Computerwetenschappen, Celestijnenlaan 200A, B-3001 Leuven, Belgium
11  * and INRIA Saclay - Ile-de-France, Parc Club Orsay Universite,
12  * ZAC des vignes, 4 rue Jacques Monod, 91893 Orsay, France
13  * and Ecole Normale Superieure, 45 rue d'Ulm, 75230 Paris, France
14  * and Cerebras Systems, 175 S San Antonio Rd, Los Altos, CA, USA
15  */
16 
17 #include <stdlib.h>
18 #include <string.h>
19 #include <isl_ctx_private.h>
20 #include <isl_map_private.h>
21 #include <isl/set.h>
22 #include <isl_seq.h>
23 #include <isl_polynomial_private.h>
24 #include <isl_printer_private.h>
25 #include <isl_space_private.h>
26 #include <isl_mat_private.h>
27 #include <isl_vec_private.h>
28 #include <isl/union_set.h>
29 #include <isl/union_map.h>
30 #include <isl/constraint.h>
31 #include <isl_local.h>
32 #include <isl_local_space_private.h>
33 #include <isl_aff_private.h>
34 #include <isl_id_private.h>
35 #include <isl_val_private.h>
36 #include <isl_constraint_private.h>
37 #include <isl/ast_build.h>
38 #include <isl_sort.h>
39 #include <isl_output_private.h>
40 
41 #include <bset_to_bmap.c>
42 #include <set_to_map.c>
43 #include <uset_to_umap.c>
44 
45 static const char *s_to[2] = { " -> ", " \\to " };
46 static const char *s_and[2] = { " and ", " \\wedge " };
47 static const char *s_or[2] = { " or ", " \\vee " };
48 static const char *s_le[2] = { "<=", "\\le" };
49 static const char *s_ge[2] = { ">=", "\\ge" };
50 static const char *s_open_set[2] = { "{ ", "\\{\\, " };
51 static const char *s_close_set[2] = { " }", " \\,\\}" };
52 static const char *s_open_list[2] = { "[", "(" };
53 static const char *s_close_list[2] = { "]", ")" };
54 static const char *s_such_that[2] = { " : ", " \\mid " };
55 static const char *s_open_exists[2] = { "exists (", "\\exists \\, " };
56 static const char *s_close_exists[2] = { ")", "" };
57 static const char *s_div_prefix[2] = { "e", "\\alpha_" };
58 static const char *s_mod[2] = { "mod", "\\bmod" };
59 static const char *s_param_prefix[2] = { "p", "p_" };
60 static const char *s_input_prefix[2] = { "i", "i_" };
61 static const char *s_output_prefix[2] = { "o", "o_" };
62 
print_constraint_polylib(struct isl_basic_map * bmap,int ineq,int n,__isl_take isl_printer * p)63 static __isl_give isl_printer *print_constraint_polylib(
64 	struct isl_basic_map *bmap, int ineq, int n, __isl_take isl_printer *p)
65 {
66 	int i;
67 	isl_size n_in = isl_basic_map_dim(bmap, isl_dim_in);
68 	isl_size n_out = isl_basic_map_dim(bmap, isl_dim_out);
69 	isl_size nparam = isl_basic_map_dim(bmap, isl_dim_param);
70 	isl_int *c = ineq ? bmap->ineq[n] : bmap->eq[n];
71 
72 	if (n_in < 0 || n_out < 0 || nparam < 0)
73 		return isl_printer_free(p);
74 
75 	p = isl_printer_start_line(p);
76 	p = isl_printer_print_int(p, ineq);
77 	for (i = 0; i < n_out; ++i) {
78 		p = isl_printer_print_str(p, " ");
79 		p = isl_printer_print_isl_int(p, c[1+nparam+n_in+i]);
80 	}
81 	for (i = 0; i < n_in; ++i) {
82 		p = isl_printer_print_str(p, " ");
83 		p = isl_printer_print_isl_int(p, c[1+nparam+i]);
84 	}
85 	for (i = 0; i < bmap->n_div; ++i) {
86 		p = isl_printer_print_str(p, " ");
87 		p = isl_printer_print_isl_int(p, c[1+nparam+n_in+n_out+i]);
88 	}
89 	for (i = 0; i < nparam; ++i) {
90 		p = isl_printer_print_str(p, " ");
91 		p = isl_printer_print_isl_int(p, c[1+i]);
92 	}
93 	p = isl_printer_print_str(p, " ");
94 	p = isl_printer_print_isl_int(p, c[0]);
95 	p = isl_printer_end_line(p);
96 	return p;
97 }
98 
print_constraints_polylib(struct isl_basic_map * bmap,__isl_take isl_printer * p)99 static __isl_give isl_printer *print_constraints_polylib(
100 	struct isl_basic_map *bmap, __isl_take isl_printer *p)
101 {
102 	int i;
103 
104 	p = isl_printer_set_isl_int_width(p, 5);
105 
106 	for (i = 0; i < bmap->n_eq; ++i)
107 		p = print_constraint_polylib(bmap, 0, i, p);
108 	for (i = 0; i < bmap->n_ineq; ++i)
109 		p = print_constraint_polylib(bmap, 1, i, p);
110 
111 	return p;
112 }
113 
bset_print_constraints_polylib(struct isl_basic_set * bset,__isl_take isl_printer * p)114 static __isl_give isl_printer *bset_print_constraints_polylib(
115 	struct isl_basic_set *bset, __isl_take isl_printer *p)
116 {
117 	return print_constraints_polylib(bset_to_bmap(bset), p);
118 }
119 
isl_basic_map_print_polylib(__isl_keep isl_basic_map * bmap,__isl_take isl_printer * p,int ext)120 static __isl_give isl_printer *isl_basic_map_print_polylib(
121 	__isl_keep isl_basic_map *bmap, __isl_take isl_printer *p, int ext)
122 {
123 	isl_size total;
124 
125 	total = isl_basic_map_dim(bmap, isl_dim_all);
126 	if (total < 0)
127 		return isl_printer_free(p);
128 	p = isl_printer_start_line(p);
129 	p = isl_printer_print_int(p, bmap->n_eq + bmap->n_ineq);
130 	p = isl_printer_print_str(p, " ");
131 	p = isl_printer_print_int(p, 1 + total + 1);
132 	if (ext) {
133 		isl_size n_in = isl_basic_map_dim(bmap, isl_dim_in);
134 		isl_size n_out = isl_basic_map_dim(bmap, isl_dim_out);
135 		isl_size n_div = isl_basic_map_dim(bmap, isl_dim_div);
136 		isl_size nparam = isl_basic_map_dim(bmap, isl_dim_param);
137 
138 		if (n_in < 0 || n_out < 0 || n_div < 0 || nparam < 0)
139 			return isl_printer_free(p);
140 
141 		p = isl_printer_print_str(p, " ");
142 		p = isl_printer_print_int(p, n_out);
143 		p = isl_printer_print_str(p, " ");
144 		p = isl_printer_print_int(p, n_in);
145 		p = isl_printer_print_str(p, " ");
146 		p = isl_printer_print_int(p, n_div);
147 		p = isl_printer_print_str(p, " ");
148 		p = isl_printer_print_int(p, nparam);
149 	}
150 	p = isl_printer_end_line(p);
151 	return print_constraints_polylib(bmap, p);
152 }
153 
isl_basic_set_print_polylib(__isl_keep isl_basic_set * bset,__isl_take isl_printer * p,int ext)154 static __isl_give isl_printer *isl_basic_set_print_polylib(
155 	__isl_keep isl_basic_set *bset, __isl_take isl_printer *p, int ext)
156 {
157 	return isl_basic_map_print_polylib(bset_to_bmap(bset), p, ext);
158 }
159 
isl_map_print_polylib(__isl_keep isl_map * map,__isl_take isl_printer * p,int ext)160 static __isl_give isl_printer *isl_map_print_polylib(__isl_keep isl_map *map,
161 	__isl_take isl_printer *p, int ext)
162 {
163 	int i;
164 
165 	p = isl_printer_start_line(p);
166 	p = isl_printer_print_int(p, map->n);
167 	p = isl_printer_end_line(p);
168 	for (i = 0; i < map->n; ++i) {
169 		p = isl_printer_start_line(p);
170 		p = isl_printer_end_line(p);
171 		p = isl_basic_map_print_polylib(map->p[i], p, ext);
172 	}
173 	return p;
174 }
175 
isl_set_print_polylib(__isl_keep isl_set * set,__isl_take isl_printer * p,int ext)176 static __isl_give isl_printer *isl_set_print_polylib(__isl_keep isl_set *set,
177 	__isl_take isl_printer *p, int ext)
178 {
179 	return isl_map_print_polylib(set_to_map(set), p, ext);
180 }
181 
count_same_name(__isl_keep isl_space * space,enum isl_dim_type type,unsigned pos,const char * name)182 static isl_size count_same_name(__isl_keep isl_space *space,
183 	enum isl_dim_type type, unsigned pos, const char *name)
184 {
185 	enum isl_dim_type t;
186 	int p;
187 	isl_size s;
188 	int count = 0;
189 
190 	for (t = isl_dim_param; t <= type && t <= isl_dim_out; ++t) {
191 		s = t == type ? pos : isl_space_dim(space, t);
192 		if (s < 0)
193 			return isl_size_error;
194 		for (p = 0; p < s; ++p) {
195 			const char *n = isl_space_get_dim_name(space, t, p);
196 			if (n && !strcmp(n, name))
197 				count++;
198 		}
199 	}
200 	return count;
201 }
202 
203 /* Print the name of the variable of type "type" and position "pos"
204  * in "space" to "p".
205  */
print_name(__isl_keep isl_space * space,__isl_take isl_printer * p,enum isl_dim_type type,unsigned pos,int latex)206 static __isl_give isl_printer *print_name(__isl_keep isl_space *space,
207 	__isl_take isl_printer *p, enum isl_dim_type type, unsigned pos,
208 	int latex)
209 {
210 	const char *name;
211 	char buffer[20];
212 	isl_size primes;
213 
214 	name = type == isl_dim_div ? NULL
215 				   : isl_space_get_dim_name(space, type, pos);
216 
217 	if (!name) {
218 		const char *prefix;
219 		if (type == isl_dim_param)
220 			prefix = s_param_prefix[latex];
221 		else if (type == isl_dim_div)
222 			prefix = s_div_prefix[latex];
223 		else if (isl_space_is_set(space) || type == isl_dim_in)
224 			prefix = s_input_prefix[latex];
225 		else
226 			prefix = s_output_prefix[latex];
227 		snprintf(buffer, sizeof(buffer), "%s%d", prefix, pos);
228 		name = buffer;
229 	}
230 	primes = count_same_name(space, name == buffer ? isl_dim_div : type,
231 				 pos, name);
232 	if (primes < 0)
233 		return isl_printer_free(p);
234 	p = isl_printer_print_str(p, name);
235 	while (primes-- > 0)
236 		p = isl_printer_print_str(p, "'");
237 	return p;
238 }
239 
pos2type(__isl_keep isl_space * space,enum isl_dim_type * type,unsigned * pos)240 static isl_stat pos2type(__isl_keep isl_space *space,
241 	enum isl_dim_type *type, unsigned *pos)
242 {
243 	isl_size n_in = isl_space_dim(space, isl_dim_in);
244 	isl_size n_out = isl_space_dim(space, isl_dim_out);
245 	isl_size nparam = isl_space_dim(space, isl_dim_param);
246 
247 	if (n_in < 0 || n_out < 0 || nparam < 0)
248 		return isl_stat_error;
249 
250 	if (*pos < 1 + nparam) {
251 		*type = isl_dim_param;
252 		*pos -= 1;
253 	} else if (*pos < 1 + nparam + n_in) {
254 		*type = isl_dim_in;
255 		*pos -= 1 + nparam;
256 	} else if (*pos < 1 + nparam + n_in + n_out) {
257 		*type = isl_dim_out;
258 		*pos -= 1 + nparam + n_in;
259 	} else {
260 		*type = isl_dim_div;
261 		*pos -= 1 + nparam + n_in + n_out;
262 	}
263 
264 	return isl_stat_ok;
265 }
266 
267 /* Can the div expression of the integer division at position "row" of "div"
268  * be printed?
269  * In particular, are the div expressions available and does the selected
270  * variable have a known explicit representation?
271  * Furthermore, the Omega format does not allow any div expressions
272  * to be printed.
273  */
can_print_div_expr(__isl_keep isl_printer * p,__isl_keep isl_mat * div,int pos)274 static isl_bool can_print_div_expr(__isl_keep isl_printer *p,
275 	__isl_keep isl_mat *div, int pos)
276 {
277 	if (p->output_format == ISL_FORMAT_OMEGA)
278 		return isl_bool_false;
279 	if (!div)
280 		return isl_bool_false;
281 	return isl_bool_not(isl_local_div_is_marked_unknown(div, pos));
282 }
283 
284 static __isl_give isl_printer *print_div(__isl_keep isl_space *space,
285 	__isl_keep isl_mat *div, int pos, __isl_take isl_printer *p);
286 
print_term(__isl_keep isl_space * space,__isl_keep isl_mat * div,isl_int c,unsigned pos,__isl_take isl_printer * p,int latex)287 static __isl_give isl_printer *print_term(__isl_keep isl_space *space,
288 	__isl_keep isl_mat *div,
289 	isl_int c, unsigned pos, __isl_take isl_printer *p, int latex)
290 {
291 	enum isl_dim_type type;
292 	int print_div_def;
293 
294 	if (!p || !space)
295 		return isl_printer_free(p);
296 
297 	if (pos == 0)
298 		return isl_printer_print_isl_int(p, c);
299 
300 	if (pos2type(space, &type, &pos) < 0)
301 		return isl_printer_free(p);
302 	print_div_def = type == isl_dim_div && can_print_div_expr(p, div, pos);
303 
304 	if (isl_int_is_one(c))
305 		;
306 	else if (isl_int_is_negone(c))
307 		p = isl_printer_print_str(p, "-");
308 	else {
309 		p = isl_printer_print_isl_int(p, c);
310 		if (p->output_format == ISL_FORMAT_C || print_div_def)
311 			p = isl_printer_print_str(p, "*");
312 	}
313 	if (print_div_def)
314 		p = print_div(space, div, pos, p);
315 	else
316 		p = print_name(space, p, type, pos, latex);
317 	return p;
318 }
319 
print_affine_of_len(__isl_keep isl_space * space,__isl_keep isl_mat * div,__isl_take isl_printer * p,isl_int * c,int len)320 static __isl_give isl_printer *print_affine_of_len(__isl_keep isl_space *space,
321 	__isl_keep isl_mat *div,
322 	__isl_take isl_printer *p, isl_int *c, int len)
323 {
324 	int i;
325 	int first;
326 
327 	for (i = 0, first = 1; i < len; ++i) {
328 		int flip = 0;
329 		if (isl_int_is_zero(c[i]))
330 			continue;
331 		if (!first) {
332 			if (isl_int_is_neg(c[i])) {
333 				flip = 1;
334 				isl_int_neg(c[i], c[i]);
335 				p = isl_printer_print_str(p, " - ");
336 			} else
337 				p = isl_printer_print_str(p, " + ");
338 		}
339 		first = 0;
340 		p = print_term(space, div, c[i], i, p, 0);
341 		if (flip)
342 			isl_int_neg(c[i], c[i]);
343 	}
344 	if (first)
345 		p = isl_printer_print_str(p, "0");
346 	return p;
347 }
348 
349 /* Print an affine expression "c"
350  * to "p", with the variable names taken from "space" and
351  * the integer division definitions taken from "div".
352  */
print_affine(__isl_take isl_printer * p,__isl_keep isl_space * space,__isl_keep isl_mat * div,isl_int * c)353 static __isl_give isl_printer *print_affine(__isl_take isl_printer *p,
354 	__isl_keep isl_space *space, __isl_keep isl_mat *div, isl_int *c)
355 {
356 	isl_size n_div, total;
357 	unsigned len;
358 
359 	total = isl_space_dim(space, isl_dim_all);
360 	n_div = isl_mat_rows(div);
361 	if (total < 0 || n_div < 0)
362 		return isl_printer_free(p);
363 	len = 1 + total + n_div;
364 	return print_affine_of_len(space, div, p, c, len);
365 }
366 
367 /* offset is the offset of local_space inside data->type of data->space.
368  */
print_nested_var_list(__isl_take isl_printer * p,__isl_keep isl_space * local_space,enum isl_dim_type local_type,struct isl_print_space_data * data,int offset)369 static __isl_give isl_printer *print_nested_var_list(__isl_take isl_printer *p,
370 	__isl_keep isl_space *local_space, enum isl_dim_type local_type,
371 	struct isl_print_space_data *data, int offset)
372 {
373 	int i;
374 	isl_size dim;
375 
376 	if (data->space != local_space && local_type == isl_dim_out)
377 		offset += local_space->n_in;
378 
379 	dim = isl_space_dim(local_space, local_type);
380 	if (dim < 0)
381 		return isl_printer_free(p);
382 	for (i = 0; i < dim; ++i) {
383 		if (i)
384 			p = isl_printer_print_str(p, ", ");
385 		if (data->print_dim)
386 			p = data->print_dim(p, data, offset + i);
387 		else
388 			p = print_name(data->space, p, data->type, offset + i,
389 					data->latex);
390 	}
391 	return p;
392 }
393 
print_var_list(__isl_take isl_printer * p,__isl_keep isl_space * space,enum isl_dim_type type)394 static __isl_give isl_printer *print_var_list(__isl_take isl_printer *p,
395 	__isl_keep isl_space *space, enum isl_dim_type type)
396 {
397 	struct isl_print_space_data data = { .space = space, .type = type };
398 
399 	return print_nested_var_list(p, space, type, &data, 0);
400 }
401 
402 static __isl_give isl_printer *print_nested_map_dim(__isl_take isl_printer *p,
403 	__isl_keep isl_space *local_dim,
404 	struct isl_print_space_data *data, int offset);
405 
print_nested_tuple(__isl_take isl_printer * p,__isl_keep isl_space * local_space,enum isl_dim_type local_type,struct isl_print_space_data * data,int offset)406 static __isl_give isl_printer *print_nested_tuple(__isl_take isl_printer *p,
407 	__isl_keep isl_space *local_space, enum isl_dim_type local_type,
408 	struct isl_print_space_data *data, int offset)
409 {
410 	const char *name = NULL;
411 	isl_size n = isl_space_dim(local_space, local_type);
412 
413 	if (n < 0)
414 		return isl_printer_free(p);
415 	if ((local_type == isl_dim_in || local_type == isl_dim_out)) {
416 		name = isl_space_get_tuple_name(local_space, local_type);
417 		if (name) {
418 			if (data->latex)
419 				p = isl_printer_print_str(p, "\\mathrm{");
420 			p = isl_printer_print_str(p, name);
421 			if (data->latex)
422 				p = isl_printer_print_str(p, "}");
423 		}
424 	}
425 	if (!data->latex || n != 1 || name)
426 		p = isl_printer_print_str(p, s_open_list[data->latex]);
427 	if ((local_type == isl_dim_in || local_type == isl_dim_out) &&
428 	    local_space->nested[local_type - isl_dim_in]) {
429 		if (data->space != local_space && local_type == isl_dim_out)
430 			offset += local_space->n_in;
431 		p = print_nested_map_dim(p,
432 				local_space->nested[local_type - isl_dim_in],
433 				data, offset);
434 	} else
435 		p = print_nested_var_list(p, local_space, local_type, data,
436 					  offset);
437 	if (!data->latex || n != 1 || name)
438 		p = isl_printer_print_str(p, s_close_list[data->latex]);
439 	return p;
440 }
441 
print_tuple(__isl_keep isl_space * space,__isl_take isl_printer * p,enum isl_dim_type type,struct isl_print_space_data * data)442 static __isl_give isl_printer *print_tuple(__isl_keep isl_space *space,
443 	__isl_take isl_printer *p, enum isl_dim_type type,
444 	struct isl_print_space_data *data)
445 {
446 	data->space = space;
447 	data->type = type;
448 	return print_nested_tuple(p, space, type, data, 0);
449 }
450 
print_nested_map_dim(__isl_take isl_printer * p,__isl_keep isl_space * local_dim,struct isl_print_space_data * data,int offset)451 static __isl_give isl_printer *print_nested_map_dim(__isl_take isl_printer *p,
452 	__isl_keep isl_space *local_dim,
453 	struct isl_print_space_data *data, int offset)
454 {
455 	p = print_nested_tuple(p, local_dim, isl_dim_in, data, offset);
456 	p = isl_printer_print_str(p, s_to[data->latex]);
457 	p = print_nested_tuple(p, local_dim, isl_dim_out, data, offset);
458 
459 	return p;
460 }
461 
isl_print_space(__isl_keep isl_space * space,__isl_take isl_printer * p,int rational,struct isl_print_space_data * data)462 __isl_give isl_printer *isl_print_space(__isl_keep isl_space *space,
463 	__isl_take isl_printer *p, int rational,
464 	struct isl_print_space_data *data)
465 {
466 	if (rational && !data->latex)
467 		p = isl_printer_print_str(p, "rat: ");
468 	if (isl_space_is_params(space))
469 		;
470 	else if (isl_space_is_set(space))
471 		p = print_tuple(space, p, isl_dim_set, data);
472 	else {
473 		p = print_tuple(space, p, isl_dim_in, data);
474 		p = isl_printer_print_str(p, s_to[data->latex]);
475 		p = print_tuple(space, p, isl_dim_out, data);
476 	}
477 
478 	return p;
479 }
480 
print_omega_parameters(__isl_keep isl_space * space,__isl_take isl_printer * p)481 static __isl_give isl_printer *print_omega_parameters(
482 	__isl_keep isl_space *space, __isl_take isl_printer *p)
483 {
484 	isl_size nparam = isl_space_dim(space, isl_dim_param);
485 
486 	if (nparam < 0)
487 		return isl_printer_free(p);
488 	if (nparam == 0)
489 		return p;
490 
491 	p = isl_printer_start_line(p);
492 	p = isl_printer_print_str(p, "symbolic ");
493 	p = print_var_list(p, space, isl_dim_param);
494 	p = isl_printer_print_str(p, ";");
495 	p = isl_printer_end_line(p);
496 	return p;
497 }
498 
499 /* Does the inequality constraint following "i" in "bmap"
500  * have an opposite value for the same last coefficient?
501  * "last" is the position of the last coefficient of inequality "i".
502  * If the next constraint is a div constraint, then it is ignored
503  * since div constraints are not printed.
504  */
next_is_opposite(__isl_keep isl_basic_map * bmap,int i,int last)505 static isl_bool next_is_opposite(__isl_keep isl_basic_map *bmap, int i,
506 	int last)
507 {
508 	int r;
509 	isl_size total = isl_basic_map_dim(bmap, isl_dim_all);
510 	unsigned o_div = isl_basic_map_offset(bmap, isl_dim_div);
511 
512 	if (total < 0)
513 		return isl_bool_error;
514 	if (i + 1 >= bmap->n_ineq)
515 		return isl_bool_false;
516 	if (isl_seq_last_non_zero(bmap->ineq[i + 1], 1 + total) != last)
517 		return isl_bool_false;
518 	if (last >= o_div) {
519 		isl_bool is_div;
520 		is_div = isl_basic_map_is_div_constraint(bmap,
521 					    bmap->ineq[i + 1], last - o_div);
522 		if (is_div < 0)
523 			return isl_bool_error;
524 		if (is_div)
525 			return isl_bool_false;
526 	}
527 	r = isl_int_abs_eq(bmap->ineq[i][last], bmap->ineq[i + 1][last]) &&
528 	    !isl_int_eq(bmap->ineq[i][last], bmap->ineq[i + 1][last]);
529 	return isl_bool_ok(r);
530 }
531 
532 /* Return a string representation of the operator used when
533  * printing a constraint where the LHS is greater than or equal to the LHS
534  * (sign > 0) or smaller than or equal to the LHS (sign < 0).
535  * If "strict" is set, then return the strict version of the comparison
536  * operator.
537  */
constraint_op(int sign,int strict,int latex)538 static const char *constraint_op(int sign, int strict, int latex)
539 {
540 	if (strict)
541 		return sign < 0 ? "<" : ">";
542 	if (sign < 0)
543 		return s_le[latex];
544 	else
545 		return s_ge[latex];
546 }
547 
548 /* Print one side of a constraint "c" to "p", with
549  * the variable names taken from "space" and the integer division definitions
550  * taken from "div".
551  * "last" is the position of the last non-zero coefficient.
552  * Let c' be the result of zeroing out this coefficient, then
553  * the partial constraint
554  *
555  *	c' op
556  *
557  * is printed.
558  */
print_half_constraint(__isl_take isl_printer * p,__isl_keep isl_space * space,__isl_keep isl_mat * div,isl_int * c,int last,const char * op,int latex)559 static __isl_give isl_printer *print_half_constraint(__isl_take isl_printer *p,
560 	__isl_keep isl_space *space, __isl_keep isl_mat *div,
561 	isl_int *c, int last, const char *op, int latex)
562 {
563 	isl_int_set_si(c[last], 0);
564 	p = print_affine(p, space, div, c);
565 
566 	p = isl_printer_print_str(p, " ");
567 	p = isl_printer_print_str(p, op);
568 	p = isl_printer_print_str(p, " ");
569 
570 	return p;
571 }
572 
573 /* Print a constraint "c" to "p", with the variable names
574  * taken from "space" and the integer division definitions taken from "div".
575  * "last" is the position of the last non-zero coefficient, which is
576  * moreover assumed to be negative.
577  * Let c' be the result of zeroing out this coefficient, then
578  * the constraint is printed in the form
579  *
580  *	-c[last] op c'
581  */
print_constraint(__isl_take isl_printer * p,__isl_keep isl_space * space,__isl_keep isl_mat * div,isl_int * c,int last,const char * op,int latex)582 static __isl_give isl_printer *print_constraint(__isl_take isl_printer *p,
583 	__isl_keep isl_space *space, __isl_keep isl_mat *div,
584 	isl_int *c, int last, const char *op, int latex)
585 {
586 	isl_int_abs(c[last], c[last]);
587 
588 	p = print_term(space, div, c[last], last, p, latex);
589 
590 	p = isl_printer_print_str(p, " ");
591 	p = isl_printer_print_str(p, op);
592 	p = isl_printer_print_str(p, " ");
593 
594 	isl_int_set_si(c[last], 0);
595 	p = print_affine(p, space, div, c);
596 
597 	return p;
598 }
599 
600 /* Given an integer division
601  *
602  *	floor(f/m)
603  *
604  * at position "pos" in "div", print the corresponding modulo expression
605  *
606  *	(f) mod m
607  *
608  * to "p".  The variable names are taken from "space", while any
609  * nested integer division definitions are taken from "div".
610  */
print_mod(__isl_take isl_printer * p,__isl_keep isl_space * space,__isl_keep isl_mat * div,int pos,int latex)611 static __isl_give isl_printer *print_mod(__isl_take isl_printer *p,
612 	__isl_keep isl_space *space, __isl_keep isl_mat *div, int pos,
613 	int latex)
614 {
615 	if (!p || !div)
616 		return isl_printer_free(p);
617 
618 	p = isl_printer_print_str(p, "(");
619 	p = print_affine_of_len(space, div, p,
620 				div->row[pos] + 1, div->n_col - 1);
621 	p = isl_printer_print_str(p, ") ");
622 	p = isl_printer_print_str(p, s_mod[latex]);
623 	p = isl_printer_print_str(p, " ");
624 	p = isl_printer_print_isl_int(p, div->row[pos][0]);
625 	return p;
626 }
627 
628 /* Given an equality constraint with a non-zero coefficient "c"
629  * in position "pos", is this term of the form
630  *
631  *	a m floor(g/m),
632  *
633  * with c = a m?
634  * Return the position of the corresponding integer division if so.
635  * Return the number of integer divisions if not.
636  * Return isl_size_error on error.
637  *
638  * Modulo constraints are currently not printed in C format.
639  * Other than that, "pos" needs to correspond to an integer division
640  * with explicit representation and "c" needs to be a multiple
641  * of the denominator of the integer division.
642  */
print_as_modulo_pos(__isl_keep isl_printer * p,__isl_keep isl_space * space,__isl_keep isl_mat * div,unsigned pos,isl_int c)643 static isl_size print_as_modulo_pos(__isl_keep isl_printer *p,
644 	__isl_keep isl_space *space, __isl_keep isl_mat *div, unsigned pos,
645 	isl_int c)
646 {
647 	isl_bool can_print;
648 	isl_size n_div;
649 	enum isl_dim_type type;
650 
651 	n_div = isl_mat_rows(div);
652 	if (!p || !space || n_div < 0)
653 		return isl_size_error;
654 	if (p->output_format == ISL_FORMAT_C)
655 		return n_div;
656 	if (pos2type(space, &type, &pos) < 0)
657 		return isl_size_error;
658 	if (type != isl_dim_div)
659 		return n_div;
660 	can_print = can_print_div_expr(p, div, pos);
661 	if (can_print < 0)
662 		return isl_size_error;
663 	if (!can_print)
664 		return n_div;
665 	if (!isl_int_is_divisible_by(c, div->row[pos][0]))
666 		return n_div;
667 	return pos;
668 }
669 
670 /* Print equality constraint "c" to "p" as a modulo constraint,
671  * with the variable names taken from "space" and
672  * the integer division definitions taken from "div".
673  * "last" is the position of the last non-zero coefficient, which is
674  * moreover assumed to be negative and a multiple of the denominator
675  * of the corresponding integer division.  "div_pos" is the corresponding
676  * position in the sequence of integer divisions.
677  *
678  * The equality is of the form
679  *
680  *	f - a m floor(g/m) = 0.
681  *
682  * Print it as
683  *
684  *	a (g mod m) = -f + a g
685  */
print_eq_mod_constraint(__isl_take isl_printer * p,__isl_keep isl_space * space,__isl_keep isl_mat * div,unsigned div_pos,isl_int * c,int last,int latex)686 static __isl_give isl_printer *print_eq_mod_constraint(
687 	__isl_take isl_printer *p, __isl_keep isl_space *space,
688 	__isl_keep isl_mat *div, unsigned div_pos,
689 	isl_int *c, int last, int latex)
690 {
691 	isl_ctx *ctx;
692 	int multiple;
693 
694 	ctx = isl_printer_get_ctx(p);
695 	isl_int_divexact(c[last], c[last], div->row[div_pos][0]);
696 	isl_int_abs(c[last], c[last]);
697 	multiple = !isl_int_is_one(c[last]);
698 	if (multiple) {
699 		p = isl_printer_print_isl_int(p, c[last]);
700 		p = isl_printer_print_str(p, "*(");
701 	}
702 	p = print_mod(p, space, div, div_pos, latex);
703 	if (multiple)
704 		p = isl_printer_print_str(p, ")");
705 	p = isl_printer_print_str(p, " = ");
706 	isl_seq_combine(c, ctx->negone, c,
707 			    c[last], div->row[div_pos] + 1, last);
708 	isl_int_set_si(c[last], 0);
709 	p = print_affine(p, space, div, c);
710 	return p;
711 }
712 
713 /* Print equality constraint "c" to "p", with the variable names
714  * taken from "space" and the integer division definitions taken from "div".
715  * "last" is the position of the last non-zero coefficient, which is
716  * moreover assumed to be negative.
717  *
718  * If possible, print the equality constraint as a modulo constraint.
719  */
print_eq_constraint(__isl_take isl_printer * p,__isl_keep isl_space * space,__isl_keep isl_mat * div,isl_int * c,int last,int latex)720 static __isl_give isl_printer *print_eq_constraint(__isl_take isl_printer *p,
721 	__isl_keep isl_space *space, __isl_keep isl_mat *div, isl_int *c,
722 	int last, int latex)
723 {
724 	isl_size n_div;
725 	isl_size div_pos;
726 
727 	n_div = isl_mat_rows(div);
728 	div_pos = print_as_modulo_pos(p, space, div, last, c[last]);
729 	if (n_div < 0 || div_pos < 0)
730 		return isl_printer_free(p);
731 	if (div_pos < n_div)
732 		return print_eq_mod_constraint(p, space, div, div_pos,
733 						c, last, latex);
734 	return print_constraint(p, space, div, c, last, "=", latex);
735 }
736 
737 /* Print the constraints of "bmap" to "p".
738  * The names of the variables are taken from "space" and
739  * the integer division definitions are taken from "div".
740  * Div constraints are only printed in "dump" mode.
741  * The constraints are sorted prior to printing (except in "dump" mode).
742  *
743  * If x is the last variable with a non-zero coefficient,
744  * then a lower bound
745  *
746  *	f - a x >= 0
747  *
748  * is printed as
749  *
750  *	a x <= f
751  *
752  * while an upper bound
753  *
754  *	f + a x >= 0
755  *
756  * is printed as
757  *
758  *	a x >= -f
759  *
760  * If the next constraint has an opposite sign for the same last coefficient,
761  * then it is printed as
762  *
763  *	f >= a x
764  *
765  * or
766  *
767  *	-f <= a x
768  *
769  * instead.  In fact, the "a x" part is not printed explicitly, but
770  * reused from the next constraint, which is therefore treated as
771  * a first constraint in the conjunction.
772  *
773  * If the constant term of "f" is -1, then "f" is replaced by "f + 1" and
774  * the comparison operator is replaced by the strict variant.
775  * Essentially, ">= 1" is replaced by "> 0".
776  */
print_constraints(__isl_keep isl_basic_map * bmap,__isl_keep isl_space * space,__isl_keep isl_mat * div,__isl_take isl_printer * p,int latex)777 static __isl_give isl_printer *print_constraints(__isl_keep isl_basic_map *bmap,
778 	__isl_keep isl_space *space, __isl_keep isl_mat *div,
779 	__isl_take isl_printer *p, int latex)
780 {
781 	int i;
782 	isl_vec *c = NULL;
783 	int rational = ISL_F_ISSET(bmap, ISL_BASIC_MAP_RATIONAL);
784 	isl_size total = isl_basic_map_dim(bmap, isl_dim_all);
785 	unsigned o_div = isl_basic_map_offset(bmap, isl_dim_div);
786 	int first = 1;
787 	int dump;
788 
789 	if (total < 0 || !p)
790 		return isl_printer_free(p);
791 	bmap = isl_basic_map_copy(bmap);
792 	dump = p->dump;
793 	if (!dump)
794 		bmap = isl_basic_map_sort_constraints(bmap);
795 	if (!bmap)
796 		goto error;
797 
798 	c = isl_vec_alloc(bmap->ctx, 1 + total);
799 	if (!c)
800 		goto error;
801 
802 	for (i = bmap->n_eq - 1; i >= 0; --i) {
803 		int l = isl_seq_last_non_zero(bmap->eq[i], 1 + total);
804 		if (l < 0) {
805 			if (i != bmap->n_eq - 1)
806 				p = isl_printer_print_str(p, s_and[latex]);
807 			p = isl_printer_print_str(p, "0 = 0");
808 			continue;
809 		}
810 		if (!first)
811 			p = isl_printer_print_str(p, s_and[latex]);
812 		if (isl_int_is_neg(bmap->eq[i][l]))
813 			isl_seq_cpy(c->el, bmap->eq[i], 1 + total);
814 		else
815 			isl_seq_neg(c->el, bmap->eq[i], 1 + total);
816 		p = print_eq_constraint(p, space, div, c->el, l, latex);
817 		first = 0;
818 	}
819 	for (i = 0; i < bmap->n_ineq; ++i) {
820 		isl_bool combine;
821 		int l = isl_seq_last_non_zero(bmap->ineq[i], 1 + total);
822 		int strict;
823 		int s;
824 		const char *op;
825 		if (l < 0)
826 			continue;
827 		if (!dump && l >= o_div &&
828 		    can_print_div_expr(p, div, l - o_div)) {
829 			isl_bool is_div;
830 			is_div = isl_basic_map_is_div_constraint(bmap,
831 						    bmap->ineq[i], l - o_div);
832 			if (is_div < 0)
833 				goto error;
834 			if (is_div)
835 				continue;
836 		}
837 		if (!first)
838 			p = isl_printer_print_str(p, s_and[latex]);
839 		s = isl_int_sgn(bmap->ineq[i][l]);
840 		strict = !rational && isl_int_is_negone(bmap->ineq[i][0]);
841 		if (s < 0)
842 			isl_seq_cpy(c->el, bmap->ineq[i], 1 + total);
843 		else
844 			isl_seq_neg(c->el, bmap->ineq[i], 1 + total);
845 		if (strict)
846 			isl_int_set_si(c->el[0], 0);
847 		combine = dump ? isl_bool_false : next_is_opposite(bmap, i, l);
848 		if (combine < 0)
849 			goto error;
850 		if (combine) {
851 			op = constraint_op(-s, strict, latex);
852 			p = print_half_constraint(p, space, div, c->el, l,
853 						op, latex);
854 			first = 1;
855 		} else {
856 			op = constraint_op(s, strict, latex);
857 			p = print_constraint(p, space, div, c->el, l,
858 						op, latex);
859 			first = 0;
860 		}
861 	}
862 
863 	isl_basic_map_free(bmap);
864 	isl_vec_free(c);
865 
866 	return p;
867 error:
868 	isl_basic_map_free(bmap);
869 	isl_vec_free(c);
870 	isl_printer_free(p);
871 	return NULL;
872 }
873 
print_div(__isl_keep isl_space * space,__isl_keep isl_mat * div,int pos,__isl_take isl_printer * p)874 static __isl_give isl_printer *print_div(__isl_keep isl_space *space,
875 	__isl_keep isl_mat *div, int pos, __isl_take isl_printer *p)
876 {
877 	int c;
878 
879 	if (!p || !div)
880 		return isl_printer_free(p);
881 
882 	c = p->output_format == ISL_FORMAT_C;
883 	p = isl_printer_print_str(p, c ? "floord(" : "floor((");
884 	p = print_affine_of_len(space, div, p,
885 				div->row[pos] + 1, div->n_col - 1);
886 	p = isl_printer_print_str(p, c ? ", " : ")/");
887 	p = isl_printer_print_isl_int(p, div->row[pos][0]);
888 	p = isl_printer_print_str(p, ")");
889 	return p;
890 }
891 
892 /* Print a comma separated list of div names, except those that have
893  * a definition that can be printed.
894  * If "print_defined_divs" is set, then those div names are printed
895  * as well, along with their definitions.
896  */
print_div_list(__isl_take isl_printer * p,__isl_keep isl_space * space,__isl_keep isl_mat * div,int latex,int print_defined_divs)897 static __isl_give isl_printer *print_div_list(__isl_take isl_printer *p,
898 	__isl_keep isl_space *space, __isl_keep isl_mat *div, int latex,
899 	int print_defined_divs)
900 {
901 	int i;
902 	int first = 1;
903 	isl_size n_div;
904 
905 	n_div = isl_mat_rows(div);
906 	if (!p || !space || n_div < 0)
907 		return isl_printer_free(p);
908 
909 	for (i = 0; i < n_div; ++i) {
910 		if (!print_defined_divs && can_print_div_expr(p, div, i))
911 			continue;
912 		if (!first)
913 			p = isl_printer_print_str(p, ", ");
914 		p = print_name(space, p, isl_dim_div, i, latex);
915 		first = 0;
916 		if (!can_print_div_expr(p, div, i))
917 			continue;
918 		p = isl_printer_print_str(p, " = ");
919 		p = print_div(space, div, i, p);
920 	}
921 
922 	return p;
923 }
924 
925 /* Does printing an object with local variables described by "div"
926  * require an "exists" clause?
927  * That is, are there any local variables without an explicit representation?
928  * An exists clause is also needed in "dump" mode because
929  * explicit div representations are not printed inline in that case.
930  */
need_exists(__isl_keep isl_printer * p,__isl_keep isl_mat * div)931 static isl_bool need_exists(__isl_keep isl_printer *p, __isl_keep isl_mat *div)
932 {
933 	int i;
934 	isl_size n;
935 
936 	n = isl_mat_rows(div);
937 	if (!p || n < 0)
938 		return isl_bool_error;
939 	if (n == 0)
940 		return isl_bool_false;
941 	if (p->dump)
942 		return isl_bool_true;
943 	for (i = 0; i < n; ++i)
944 		if (!can_print_div_expr(p, div, i))
945 			return isl_bool_true;
946 	return isl_bool_false;
947 }
948 
949 /* Print the start of an exists clause, i.e.,
950  *
951  *	(exists variables:
952  *
953  * In dump mode, local variables with an explicit definition are printed
954  * as well because they will not be printed inline.
955  */
open_exists(__isl_take isl_printer * p,__isl_keep isl_space * space,__isl_keep isl_mat * div,int latex)956 static __isl_give isl_printer *open_exists(__isl_take isl_printer *p,
957 	__isl_keep isl_space *space, __isl_keep isl_mat *div, int latex)
958 {
959 	int dump;
960 
961 	if (!p)
962 		return NULL;
963 
964 	dump = p->dump;
965 	p = isl_printer_print_str(p, s_open_exists[latex]);
966 	p = print_div_list(p, space, div, latex, dump);
967 	p = isl_printer_print_str(p, ": ");
968 
969 	return p;
970 }
971 
972 /* Remove the explicit representations of all local variables in "div".
973  */
mark_all_unknown(__isl_take isl_mat * div)974 static __isl_give isl_mat *mark_all_unknown(__isl_take isl_mat *div)
975 {
976 	int i;
977 	isl_size n_div;
978 
979 	n_div = isl_mat_rows(div);
980 	if (n_div < 0)
981 		return isl_mat_free(div);
982 
983 	for (i = 0; i < n_div; ++i)
984 		div = isl_mat_set_element_si(div, i, 0, 0);
985 	return div;
986 }
987 
988 /* Print the constraints of "bmap" to "p".
989  * The names of the variables are taken from "space".
990  * "latex" is set if the constraints should be printed in LaTeX format.
991  * Do not print inline explicit div representations in "dump" mode.
992  */
print_disjunct(__isl_keep isl_basic_map * bmap,__isl_keep isl_space * space,__isl_take isl_printer * p,int latex)993 static __isl_give isl_printer *print_disjunct(__isl_keep isl_basic_map *bmap,
994 	__isl_keep isl_space *space, __isl_take isl_printer *p, int latex)
995 {
996 	int dump;
997 	isl_mat *div;
998 	isl_bool exists;
999 
1000 	if (!p)
1001 		return NULL;
1002 	dump = p->dump;
1003 	div = isl_basic_map_get_divs(bmap);
1004 	exists = need_exists(p, div);
1005 	if (exists >= 0 && exists)
1006 		p = open_exists(p, space, div, latex);
1007 
1008 	if (dump)
1009 		div = mark_all_unknown(div);
1010 	p = print_constraints(bmap, space, div, p, latex);
1011 	isl_mat_free(div);
1012 
1013 	if (exists >= 0 && exists)
1014 		p = isl_printer_print_str(p, s_close_exists[latex]);
1015 	return p;
1016 }
1017 
1018 /* Print a colon followed by the constraints of "bmap"
1019  * to "p", provided there are any constraints.
1020  * The names of the variables are taken from "space".
1021  * "latex" is set if the constraints should be printed in LaTeX format.
1022  */
print_optional_disjunct(__isl_keep isl_basic_map * bmap,__isl_keep isl_space * space,__isl_take isl_printer * p,int latex)1023 static __isl_give isl_printer *print_optional_disjunct(
1024 	__isl_keep isl_basic_map *bmap, __isl_keep isl_space *space,
1025 	__isl_take isl_printer *p, int latex)
1026 {
1027 	if (isl_basic_map_plain_is_universe(bmap))
1028 		return p;
1029 
1030 	p = isl_printer_print_str(p, ": ");
1031 	p = print_disjunct(bmap, space, p, latex);
1032 
1033 	return p;
1034 }
1035 
basic_map_print_omega(__isl_keep isl_basic_map * bmap,__isl_take isl_printer * p)1036 static __isl_give isl_printer *basic_map_print_omega(
1037 	__isl_keep isl_basic_map *bmap, __isl_take isl_printer *p)
1038 {
1039 	p = isl_printer_print_str(p, "{ [");
1040 	p = print_var_list(p, bmap->dim, isl_dim_in);
1041 	p = isl_printer_print_str(p, "] -> [");
1042 	p = print_var_list(p, bmap->dim, isl_dim_out);
1043 	p = isl_printer_print_str(p, "] ");
1044 	p = print_optional_disjunct(bmap, bmap->dim, p, 0);
1045 	p = isl_printer_print_str(p, " }");
1046 	return p;
1047 }
1048 
basic_set_print_omega(__isl_keep isl_basic_set * bset,__isl_take isl_printer * p)1049 static __isl_give isl_printer *basic_set_print_omega(
1050 	__isl_keep isl_basic_set *bset, __isl_take isl_printer *p)
1051 {
1052 	p = isl_printer_print_str(p, "{ [");
1053 	p = print_var_list(p, bset->dim, isl_dim_set);
1054 	p = isl_printer_print_str(p, "] ");
1055 	p = print_optional_disjunct(bset, bset->dim, p, 0);
1056 	p = isl_printer_print_str(p, " }");
1057 	return p;
1058 }
1059 
isl_map_print_omega(__isl_keep isl_map * map,__isl_take isl_printer * p)1060 static __isl_give isl_printer *isl_map_print_omega(__isl_keep isl_map *map,
1061 	__isl_take isl_printer *p)
1062 {
1063 	int i;
1064 
1065 	for (i = 0; i < map->n; ++i) {
1066 		if (i)
1067 			p = isl_printer_print_str(p, " union ");
1068 		p = basic_map_print_omega(map->p[i], p);
1069 	}
1070 	return p;
1071 }
1072 
isl_set_print_omega(__isl_keep isl_set * set,__isl_take isl_printer * p)1073 static __isl_give isl_printer *isl_set_print_omega(__isl_keep isl_set *set,
1074 	__isl_take isl_printer *p)
1075 {
1076 	int i;
1077 
1078 	for (i = 0; i < set->n; ++i) {
1079 		if (i)
1080 			p = isl_printer_print_str(p, " union ");
1081 		p = basic_set_print_omega(set->p[i], p);
1082 	}
1083 	return p;
1084 }
1085 
1086 /* Print the list of parameters in "space", followed by an arrow, to "p",
1087  * if there are any parameters.
1088  */
print_param_tuple(__isl_take isl_printer * p,__isl_keep isl_space * space,struct isl_print_space_data * data)1089 static __isl_give isl_printer *print_param_tuple(__isl_take isl_printer *p,
1090 	__isl_keep isl_space *space, struct isl_print_space_data *data)
1091 {
1092 	isl_size nparam;
1093 
1094 	nparam = isl_space_dim(space, isl_dim_param);
1095 	if (!p || nparam < 0)
1096 		return isl_printer_free(p);
1097 	if (nparam == 0)
1098 		return p;
1099 
1100 	p = print_tuple(space, p, isl_dim_param, data);
1101 	p = isl_printer_print_str(p, s_to[data->latex]);
1102 
1103 	return p;
1104 }
1105 
isl_basic_map_print_isl(__isl_keep isl_basic_map * bmap,__isl_take isl_printer * p,int latex)1106 static __isl_give isl_printer *isl_basic_map_print_isl(
1107 	__isl_keep isl_basic_map *bmap, __isl_take isl_printer *p,
1108 	int latex)
1109 {
1110 	struct isl_print_space_data data = { .latex = latex };
1111 	int rational = ISL_F_ISSET(bmap, ISL_BASIC_MAP_RATIONAL);
1112 
1113 	p = print_param_tuple(p, bmap->dim, &data);
1114 	p = isl_printer_print_str(p, "{ ");
1115 	p = isl_print_space(bmap->dim, p, rational, &data);
1116 	p = isl_printer_print_str(p, " : ");
1117 	p = print_disjunct(bmap, bmap->dim, p, latex);
1118 	p = isl_printer_print_str(p, " }");
1119 	return p;
1120 }
1121 
1122 /* Print the disjuncts of a map (or set) "map" to "p".
1123  * The names of the variables are taken from "space".
1124  * "latex" is set if the constraints should be printed in LaTeX format.
1125  */
print_disjuncts_core(__isl_keep isl_map * map,__isl_keep isl_space * space,__isl_take isl_printer * p,int latex)1126 static __isl_give isl_printer *print_disjuncts_core(__isl_keep isl_map *map,
1127 	__isl_keep isl_space *space, __isl_take isl_printer *p, int latex)
1128 {
1129 	int i;
1130 
1131 	if (map->n == 0)
1132 		p = isl_printer_print_str(p, "false");
1133 	for (i = 0; i < map->n; ++i) {
1134 		if (i)
1135 			p = isl_printer_print_str(p, s_or[latex]);
1136 		if (map->n > 1 && map->p[i]->n_eq + map->p[i]->n_ineq > 1)
1137 			p = isl_printer_print_str(p, "(");
1138 		p = print_disjunct(map->p[i], space, p, latex);
1139 		if (map->n > 1 && map->p[i]->n_eq + map->p[i]->n_ineq > 1)
1140 			p = isl_printer_print_str(p, ")");
1141 	}
1142 	return p;
1143 }
1144 
1145 /* Print the disjuncts of a map (or set) "map" to "p".
1146  * The names of the variables are taken from "space".
1147  * "hull" describes constraints shared by all disjuncts of "map".
1148  * "latex" is set if the constraints should be printed in LaTeX format.
1149  *
1150  * Print the disjuncts as a conjunction of "hull" and
1151  * the result of removing the constraints of "hull" from "map".
1152  * If this result turns out to be the universe, then simply print "hull".
1153  */
print_disjuncts_in_hull(__isl_keep isl_map * map,__isl_keep isl_space * space,__isl_take isl_basic_map * hull,__isl_take isl_printer * p,int latex)1154 static __isl_give isl_printer *print_disjuncts_in_hull(__isl_keep isl_map *map,
1155 	__isl_keep isl_space *space, __isl_take isl_basic_map *hull,
1156 	__isl_take isl_printer *p, int latex)
1157 {
1158 	isl_bool is_universe;
1159 
1160 	p = print_disjunct(hull, space, p, latex);
1161 	map = isl_map_plain_gist_basic_map(isl_map_copy(map), hull);
1162 	is_universe = isl_map_plain_is_universe(map);
1163 	if (is_universe < 0)
1164 		goto error;
1165 	if (!is_universe) {
1166 		p = isl_printer_print_str(p, s_and[latex]);
1167 		p = isl_printer_print_str(p, "(");
1168 		p = print_disjuncts_core(map, space, p, latex);
1169 		p = isl_printer_print_str(p, ")");
1170 	}
1171 	isl_map_free(map);
1172 
1173 	return p;
1174 error:
1175 	isl_map_free(map);
1176 	isl_printer_free(p);
1177 	return NULL;
1178 }
1179 
1180 /* Print the disjuncts of a map (or set) "map" to "p".
1181  * The names of the variables are taken from "space".
1182  * "latex" is set if the constraints should be printed in LaTeX format.
1183  *
1184  * If there are at least two disjuncts and "dump" mode is not turned out,
1185  * check for any shared constraints among all disjuncts.
1186  * If there are any, then print them separately in print_disjuncts_in_hull.
1187  */
print_disjuncts(__isl_keep isl_map * map,__isl_keep isl_space * space,__isl_take isl_printer * p,int latex)1188 static __isl_give isl_printer *print_disjuncts(__isl_keep isl_map *map,
1189 	__isl_keep isl_space *space, __isl_take isl_printer *p, int latex)
1190 {
1191 	if (isl_map_plain_is_universe(map))
1192 		return p;
1193 
1194 	p = isl_printer_print_str(p, s_such_that[latex]);
1195 	if (!p)
1196 		return NULL;
1197 
1198 	if (!p->dump && map->n >= 2) {
1199 		isl_basic_map *hull;
1200 		isl_bool is_universe;
1201 
1202 		hull = isl_map_plain_unshifted_simple_hull(isl_map_copy(map));
1203 		is_universe = isl_basic_map_plain_is_universe(hull);
1204 		if (is_universe < 0)
1205 			p = isl_printer_free(p);
1206 		else if (!is_universe)
1207 			return print_disjuncts_in_hull(map, space, hull,
1208 							p, latex);
1209 		isl_basic_map_free(hull);
1210 	}
1211 
1212 	return print_disjuncts_core(map, space, p, latex);
1213 }
1214 
1215 /* Print the disjuncts of a map (or set).
1216  * The names of the variables are taken from "space".
1217  * "latex" is set if the constraints should be printed in LaTeX format.
1218  *
1219  * If the map turns out to be a universal parameter domain, then
1220  * we need to print the colon.  Otherwise, the output looks identical
1221  * to the empty set.
1222  */
print_disjuncts_map(__isl_keep isl_map * map,__isl_keep isl_space * space,__isl_take isl_printer * p,int latex)1223 static __isl_give isl_printer *print_disjuncts_map(__isl_keep isl_map *map,
1224 	__isl_keep isl_space *space, __isl_take isl_printer *p, int latex)
1225 {
1226 	if (isl_map_plain_is_universe(map) && isl_space_is_params(map->dim))
1227 		return isl_printer_print_str(p, s_such_that[latex]);
1228 	else
1229 		return print_disjuncts(map, space, p, latex);
1230 }
1231 
1232 /* Print the disjuncts of a set.
1233  * The names of the variables are taken from "space".
1234  * "latex" is set if the constraints should be printed in LaTeX format.
1235  */
print_disjuncts_set(__isl_keep isl_set * set,__isl_keep isl_space * space,__isl_take isl_printer * p,int latex)1236 static __isl_give isl_printer *print_disjuncts_set(__isl_keep isl_set *set,
1237 	__isl_keep isl_space *space, __isl_take isl_printer *p, int latex)
1238 {
1239 	return print_disjuncts_map(set_to_map(set), space, p, latex);
1240 }
1241 
1242 struct isl_aff_split {
1243 	isl_basic_map *aff;
1244 	isl_map *map;
1245 };
1246 
free_split(__isl_take struct isl_aff_split * split,int n)1247 static void free_split(__isl_take struct isl_aff_split *split, int n)
1248 {
1249 	int i;
1250 
1251 	if (!split)
1252 		return;
1253 
1254 	for (i = 0; i < n; ++i) {
1255 		isl_basic_map_free(split[i].aff);
1256 		isl_map_free(split[i].map);
1257 	}
1258 
1259 	free(split);
1260 }
1261 
get_aff(__isl_take isl_basic_map * bmap)1262 static __isl_give isl_basic_map *get_aff(__isl_take isl_basic_map *bmap)
1263 {
1264 	int i, j;
1265 	isl_size nparam, n_in, n_out, total;
1266 
1267 	bmap = isl_basic_map_cow(bmap);
1268 	if (!bmap)
1269 		return NULL;
1270 	bmap = isl_basic_map_free_inequality(bmap, bmap->n_ineq);
1271 
1272 	nparam = isl_basic_map_dim(bmap, isl_dim_param);
1273 	n_in = isl_basic_map_dim(bmap, isl_dim_in);
1274 	n_out = isl_basic_map_dim(bmap, isl_dim_out);
1275 	total = isl_basic_map_dim(bmap, isl_dim_all);
1276 	if (n_in < 0 || n_out < 0 || nparam < 0 || total < 0)
1277 		return isl_basic_map_free(bmap);
1278 
1279 	for (i = bmap->n_eq - 1; i >= 0; --i) {
1280 		j = isl_seq_last_non_zero(bmap->eq[i] + 1, total);
1281 		if (j >= nparam && j < nparam + n_in + n_out &&
1282 		    (isl_int_is_one(bmap->eq[i][1 + j]) ||
1283 		     isl_int_is_negone(bmap->eq[i][1 + j])))
1284 			continue;
1285 		if (isl_basic_map_drop_equality(bmap, i) < 0)
1286 			goto error;
1287 	}
1288 
1289 	bmap = isl_basic_map_finalize(bmap);
1290 
1291 	return bmap;
1292 error:
1293 	isl_basic_map_free(bmap);
1294 	return NULL;
1295 }
1296 
aff_split_cmp(const void * p1,const void * p2,void * user)1297 static int aff_split_cmp(const void *p1, const void *p2, void *user)
1298 {
1299 	const struct isl_aff_split *s1, *s2;
1300 	s1 = (const struct isl_aff_split *) p1;
1301 	s2 = (const struct isl_aff_split *) p2;
1302 
1303 	return isl_basic_map_plain_cmp(s1->aff, s2->aff);
1304 }
1305 
drop_aff(__isl_take isl_basic_map * bmap,__isl_keep isl_basic_map * aff)1306 static __isl_give isl_basic_map *drop_aff(__isl_take isl_basic_map *bmap,
1307 	__isl_keep isl_basic_map *aff)
1308 {
1309 	int i, j;
1310 	isl_size v_div;
1311 
1312 	v_div = isl_basic_map_var_offset(bmap, isl_dim_div);
1313 	if (v_div < 0 || !aff)
1314 		goto error;
1315 
1316 	for (i = bmap->n_eq - 1; i >= 0; --i) {
1317 		if (isl_seq_first_non_zero(bmap->eq[i] + 1 + v_div,
1318 					    bmap->n_div) != -1)
1319 			continue;
1320 		for (j = 0; j < aff->n_eq; ++j) {
1321 			if (!isl_seq_eq(bmap->eq[i], aff->eq[j], 1 + v_div) &&
1322 			    !isl_seq_is_neg(bmap->eq[i], aff->eq[j], 1 + v_div))
1323 				continue;
1324 			if (isl_basic_map_drop_equality(bmap, i) < 0)
1325 				goto error;
1326 			break;
1327 		}
1328 	}
1329 
1330 	return bmap;
1331 error:
1332 	isl_basic_map_free(bmap);
1333 	return NULL;
1334 }
1335 
split_aff(__isl_keep isl_map * map)1336 static __isl_give struct isl_aff_split *split_aff(__isl_keep isl_map *map)
1337 {
1338 	int i, n;
1339 	struct isl_aff_split *split;
1340 	isl_ctx *ctx;
1341 
1342 	ctx = isl_map_get_ctx(map);
1343 	split = isl_calloc_array(ctx, struct isl_aff_split, map->n);
1344 	if (!split)
1345 		return NULL;
1346 
1347 	for (i = 0; i < map->n; ++i) {
1348 		isl_basic_map *bmap;
1349 		split[i].aff = get_aff(isl_basic_map_copy(map->p[i]));
1350 		bmap = isl_basic_map_copy(map->p[i]);
1351 		bmap = isl_basic_map_cow(bmap);
1352 		bmap = drop_aff(bmap, split[i].aff);
1353 		split[i].map = isl_map_from_basic_map(bmap);
1354 		if (!split[i].aff || !split[i].map)
1355 			goto error;
1356 	}
1357 
1358 	if (isl_sort(split, map->n, sizeof(struct isl_aff_split),
1359 			&aff_split_cmp, NULL) < 0)
1360 		goto error;
1361 
1362 	n = map->n;
1363 	for (i = n - 1; i >= 1; --i) {
1364 		if (!isl_basic_map_plain_is_equal(split[i - 1].aff,
1365 						 split[i].aff))
1366 			continue;
1367 		isl_basic_map_free(split[i].aff);
1368 		split[i - 1].map = isl_map_union(split[i - 1].map,
1369 						 split[i].map);
1370 		if (i != n - 1)
1371 			split[i] = split[n - 1];
1372 		split[n - 1].aff = NULL;
1373 		split[n - 1].map = NULL;
1374 		--n;
1375 	}
1376 
1377 	return split;
1378 error:
1379 	free_split(split, map->n);
1380 	return NULL;
1381 }
1382 
defining_equality(__isl_keep isl_basic_map * eq,__isl_keep isl_space * space,enum isl_dim_type type,int pos)1383 static int defining_equality(__isl_keep isl_basic_map *eq,
1384 	__isl_keep isl_space *space, enum isl_dim_type type, int pos)
1385 {
1386 	int i;
1387 	isl_size total;
1388 
1389 	total = isl_basic_map_dim(eq, isl_dim_all);
1390 	if (total < 0)
1391 		return -1;
1392 
1393 	pos += isl_space_offset(space, type);
1394 
1395 	for (i = 0; i < eq->n_eq; ++i) {
1396 		if (isl_seq_last_non_zero(eq->eq[i] + 1, total) != pos)
1397 			continue;
1398 		if (isl_int_is_one(eq->eq[i][1 + pos]))
1399 			isl_seq_neg(eq->eq[i], eq->eq[i], 1 + total);
1400 		return i;
1401 	}
1402 
1403 	return -1;
1404 }
1405 
1406 /* Print dimension "pos" of data->space to "p".
1407  *
1408  * data->user is assumed to be an isl_basic_map keeping track of equalities.
1409  *
1410  * If the current dimension is defined by these equalities, then print
1411  * the corresponding expression, assigned to the name of the dimension
1412  * if there is any.  Otherwise, print the name of the dimension.
1413  */
print_dim_eq(__isl_take isl_printer * p,struct isl_print_space_data * data,unsigned pos)1414 static __isl_give isl_printer *print_dim_eq(__isl_take isl_printer *p,
1415 	struct isl_print_space_data *data, unsigned pos)
1416 {
1417 	isl_basic_map *eq = data->user;
1418 	int j;
1419 
1420 	j = defining_equality(eq, data->space, data->type, pos);
1421 	if (j >= 0) {
1422 		if (isl_space_has_dim_name(data->space, data->type, pos)) {
1423 			p = print_name(data->space, p, data->type, pos,
1424 					data->latex);
1425 			p = isl_printer_print_str(p, " = ");
1426 		}
1427 		pos += 1 + isl_space_offset(data->space, data->type);
1428 		p = print_affine_of_len(data->space, NULL, p, eq->eq[j], pos);
1429 	} else {
1430 		p = print_name(data->space, p, data->type, pos, data->latex);
1431 	}
1432 
1433 	return p;
1434 }
1435 
print_split_map(__isl_take isl_printer * p,struct isl_aff_split * split,int n,__isl_keep isl_space * space)1436 static __isl_give isl_printer *print_split_map(__isl_take isl_printer *p,
1437 	struct isl_aff_split *split, int n, __isl_keep isl_space *space)
1438 {
1439 	struct isl_print_space_data data = { 0 };
1440 	int i;
1441 	int rational;
1442 
1443 	data.print_dim = &print_dim_eq;
1444 	for (i = 0; i < n; ++i) {
1445 		if (!split[i].map)
1446 			break;
1447 		rational = split[i].map->n > 0 &&
1448 		    ISL_F_ISSET(split[i].map->p[0], ISL_BASIC_MAP_RATIONAL);
1449 		if (i)
1450 			p = isl_printer_print_str(p, "; ");
1451 		data.user = split[i].aff;
1452 		p = isl_print_space(space, p, rational, &data);
1453 		p = print_disjuncts_map(split[i].map, space, p, 0);
1454 	}
1455 
1456 	return p;
1457 }
1458 
isl_map_print_isl_body(__isl_keep isl_map * map,__isl_take isl_printer * p)1459 static __isl_give isl_printer *isl_map_print_isl_body(__isl_keep isl_map *map,
1460 	__isl_take isl_printer *p)
1461 {
1462 	struct isl_print_space_data data = { 0 };
1463 	struct isl_aff_split *split = NULL;
1464 	int rational;
1465 
1466 	if (!p || !map)
1467 		return isl_printer_free(p);
1468 	if (!p->dump && map->n > 0)
1469 		split = split_aff(map);
1470 	if (split) {
1471 		p = print_split_map(p, split, map->n, map->dim);
1472 	} else {
1473 		rational = map->n > 0 &&
1474 		    ISL_F_ISSET(map->p[0], ISL_BASIC_MAP_RATIONAL);
1475 		p = isl_print_space(map->dim, p, rational, &data);
1476 		p = print_disjuncts_map(map, map->dim, p, 0);
1477 	}
1478 	free_split(split, map->n);
1479 	return p;
1480 }
1481 
isl_map_print_isl(__isl_keep isl_map * map,__isl_take isl_printer * p)1482 static __isl_give isl_printer *isl_map_print_isl(__isl_keep isl_map *map,
1483 	__isl_take isl_printer *p)
1484 {
1485 	struct isl_print_space_data data = { 0 };
1486 
1487 	p = print_param_tuple(p, map->dim, &data);
1488 	p = isl_printer_print_str(p, s_open_set[0]);
1489 	p = isl_map_print_isl_body(map, p);
1490 	p = isl_printer_print_str(p, s_close_set[0]);
1491 	return p;
1492 }
1493 
print_latex_map(__isl_keep isl_map * map,__isl_take isl_printer * p,__isl_keep isl_basic_map * aff)1494 static __isl_give isl_printer *print_latex_map(__isl_keep isl_map *map,
1495 	__isl_take isl_printer *p, __isl_keep isl_basic_map *aff)
1496 {
1497 	struct isl_print_space_data data = { 0 };
1498 
1499 	data.latex = 1;
1500 	p = print_param_tuple(p, map->dim, &data);
1501 	p = isl_printer_print_str(p, s_open_set[1]);
1502 	data.print_dim = &print_dim_eq;
1503 	data.user = aff;
1504 	p = isl_print_space(map->dim, p, 0, &data);
1505 	p = print_disjuncts_map(map, map->dim, p, 1);
1506 	p = isl_printer_print_str(p, s_close_set[1]);
1507 
1508 	return p;
1509 }
1510 
isl_map_print_latex(__isl_keep isl_map * map,__isl_take isl_printer * p)1511 static __isl_give isl_printer *isl_map_print_latex(__isl_keep isl_map *map,
1512 	__isl_take isl_printer *p)
1513 {
1514 	int i;
1515 	struct isl_aff_split *split = NULL;
1516 
1517 	if (map->n > 0)
1518 		split = split_aff(map);
1519 
1520 	if (!split)
1521 		return print_latex_map(map, p, NULL);
1522 
1523 	for (i = 0; i < map->n; ++i) {
1524 		if (!split[i].map)
1525 			break;
1526 		if (i)
1527 			p = isl_printer_print_str(p, " \\cup ");
1528 		p = print_latex_map(split[i].map, p, split[i].aff);
1529 	}
1530 
1531 	free_split(split, map->n);
1532 	return p;
1533 }
1534 
isl_printer_print_basic_map(__isl_take isl_printer * p,__isl_keep isl_basic_map * bmap)1535 __isl_give isl_printer *isl_printer_print_basic_map(__isl_take isl_printer *p,
1536 	__isl_keep isl_basic_map *bmap)
1537 {
1538 	if (!p || !bmap)
1539 		goto error;
1540 	if (p->output_format == ISL_FORMAT_ISL)
1541 		return isl_basic_map_print_isl(bmap, p, 0);
1542 	else if (p->output_format == ISL_FORMAT_OMEGA)
1543 		return basic_map_print_omega(bmap, p);
1544 	isl_assert(bmap->ctx, 0, goto error);
1545 error:
1546 	isl_printer_free(p);
1547 	return NULL;
1548 }
1549 
isl_printer_print_basic_set(__isl_take isl_printer * p,__isl_keep isl_basic_set * bset)1550 __isl_give isl_printer *isl_printer_print_basic_set(__isl_take isl_printer *p,
1551 	__isl_keep isl_basic_set *bset)
1552 {
1553 	if (!p || !bset)
1554 		goto error;
1555 
1556 	if (p->output_format == ISL_FORMAT_ISL)
1557 		return isl_basic_map_print_isl(bset, p, 0);
1558 	else if (p->output_format == ISL_FORMAT_POLYLIB)
1559 		return isl_basic_set_print_polylib(bset, p, 0);
1560 	else if (p->output_format == ISL_FORMAT_EXT_POLYLIB)
1561 		return isl_basic_set_print_polylib(bset, p, 1);
1562 	else if (p->output_format == ISL_FORMAT_POLYLIB_CONSTRAINTS)
1563 		return bset_print_constraints_polylib(bset, p);
1564 	else if (p->output_format == ISL_FORMAT_OMEGA)
1565 		return basic_set_print_omega(bset, p);
1566 	isl_assert(p->ctx, 0, goto error);
1567 error:
1568 	isl_printer_free(p);
1569 	return NULL;
1570 }
1571 
isl_printer_print_set(__isl_take isl_printer * p,__isl_keep isl_set * set)1572 __isl_give isl_printer *isl_printer_print_set(__isl_take isl_printer *p,
1573 	__isl_keep isl_set *set)
1574 {
1575 	if (!p || !set)
1576 		goto error;
1577 	if (p->output_format == ISL_FORMAT_ISL)
1578 		return isl_map_print_isl(set_to_map(set), p);
1579 	else if (p->output_format == ISL_FORMAT_POLYLIB)
1580 		return isl_set_print_polylib(set, p, 0);
1581 	else if (p->output_format == ISL_FORMAT_EXT_POLYLIB)
1582 		return isl_set_print_polylib(set, p, 1);
1583 	else if (p->output_format == ISL_FORMAT_OMEGA)
1584 		return isl_set_print_omega(set, p);
1585 	else if (p->output_format == ISL_FORMAT_LATEX)
1586 		return isl_map_print_latex(set_to_map(set), p);
1587 	isl_assert(set->ctx, 0, goto error);
1588 error:
1589 	isl_printer_free(p);
1590 	return NULL;
1591 }
1592 
isl_printer_print_map(__isl_take isl_printer * p,__isl_keep isl_map * map)1593 __isl_give isl_printer *isl_printer_print_map(__isl_take isl_printer *p,
1594 	__isl_keep isl_map *map)
1595 {
1596 	if (!p || !map)
1597 		goto error;
1598 
1599 	if (p->output_format == ISL_FORMAT_ISL)
1600 		return isl_map_print_isl(map, p);
1601 	else if (p->output_format == ISL_FORMAT_POLYLIB)
1602 		return isl_map_print_polylib(map, p, 0);
1603 	else if (p->output_format == ISL_FORMAT_EXT_POLYLIB)
1604 		return isl_map_print_polylib(map, p, 1);
1605 	else if (p->output_format == ISL_FORMAT_OMEGA)
1606 		return isl_map_print_omega(map, p);
1607 	else if (p->output_format == ISL_FORMAT_LATEX)
1608 		return isl_map_print_latex(map, p);
1609 	isl_assert(map->ctx, 0, goto error);
1610 error:
1611 	isl_printer_free(p);
1612 	return NULL;
1613 }
1614 
1615 struct isl_union_print_data {
1616 	isl_printer *p;
1617 	int first;
1618 };
1619 
print_map_body(__isl_take isl_map * map,void * user)1620 static isl_stat print_map_body(__isl_take isl_map *map, void *user)
1621 {
1622 	struct isl_union_print_data *data;
1623 	data = (struct isl_union_print_data *)user;
1624 
1625 	if (!data->first)
1626 		data->p = isl_printer_print_str(data->p, "; ");
1627 	data->first = 0;
1628 
1629 	data->p = isl_map_print_isl_body(map, data->p);
1630 	isl_map_free(map);
1631 
1632 	return isl_stat_ok;
1633 }
1634 
1635 /* Print the body of "umap" (everything except the parameter declarations)
1636  * to "p" in isl format.
1637  */
isl_printer_print_union_map_isl_body(__isl_take isl_printer * p,__isl_keep isl_union_map * umap)1638 static __isl_give isl_printer *isl_printer_print_union_map_isl_body(
1639 	__isl_take isl_printer *p, __isl_keep isl_union_map *umap)
1640 {
1641 	struct isl_union_print_data data;
1642 
1643 	p = isl_printer_print_str(p, s_open_set[0]);
1644 	data.p = p;
1645 	data.first = 1;
1646 	isl_union_map_foreach_map(umap, &print_map_body, &data);
1647 	p = data.p;
1648 	p = isl_printer_print_str(p, s_close_set[0]);
1649 	return p;
1650 }
1651 
1652 /* Print the body of "uset" (everything except the parameter declarations)
1653  * to "p" in isl format.
1654  */
isl_printer_print_union_set_isl_body(__isl_take isl_printer * p,__isl_keep isl_union_set * uset)1655 static __isl_give isl_printer *isl_printer_print_union_set_isl_body(
1656 	__isl_take isl_printer *p, __isl_keep isl_union_set *uset)
1657 {
1658 	return isl_printer_print_union_map_isl_body(p, uset_to_umap(uset));
1659 }
1660 
1661 /* Print the isl_union_map "umap" to "p" in isl format.
1662  */
isl_union_map_print_isl(__isl_keep isl_union_map * umap,__isl_take isl_printer * p)1663 static __isl_give isl_printer *isl_union_map_print_isl(
1664 	__isl_keep isl_union_map *umap, __isl_take isl_printer *p)
1665 {
1666 	struct isl_print_space_data space_data = { 0 };
1667 	isl_space *space;
1668 
1669 	space = isl_union_map_get_space(umap);
1670 	p = print_param_tuple(p, space, &space_data);
1671 	isl_space_free(space);
1672 
1673 	p = isl_printer_print_union_map_isl_body(p, umap);
1674 
1675 	return p;
1676 }
1677 
print_latex_map_body(__isl_take isl_map * map,void * user)1678 static isl_stat print_latex_map_body(__isl_take isl_map *map, void *user)
1679 {
1680 	struct isl_union_print_data *data;
1681 	data = (struct isl_union_print_data *)user;
1682 
1683 	if (!data->first)
1684 		data->p = isl_printer_print_str(data->p, " \\cup ");
1685 	data->first = 0;
1686 
1687 	data->p = isl_map_print_latex(map, data->p);
1688 	isl_map_free(map);
1689 
1690 	return isl_stat_ok;
1691 }
1692 
isl_union_map_print_latex(__isl_keep isl_union_map * umap,__isl_take isl_printer * p)1693 static __isl_give isl_printer *isl_union_map_print_latex(
1694 	__isl_keep isl_union_map *umap, __isl_take isl_printer *p)
1695 {
1696 	struct isl_union_print_data data = { p, 1 };
1697 	isl_union_map_foreach_map(umap, &print_latex_map_body, &data);
1698 	p = data.p;
1699 	return p;
1700 }
1701 
isl_printer_print_union_map(__isl_take isl_printer * p,__isl_keep isl_union_map * umap)1702 __isl_give isl_printer *isl_printer_print_union_map(__isl_take isl_printer *p,
1703 	__isl_keep isl_union_map *umap)
1704 {
1705 	if (!p || !umap)
1706 		goto error;
1707 
1708 	if (p->output_format == ISL_FORMAT_ISL)
1709 		return isl_union_map_print_isl(umap, p);
1710 	if (p->output_format == ISL_FORMAT_LATEX)
1711 		return isl_union_map_print_latex(umap, p);
1712 
1713 	isl_die(p->ctx, isl_error_invalid,
1714 		"invalid output format for isl_union_map", goto error);
1715 error:
1716 	isl_printer_free(p);
1717 	return NULL;
1718 }
1719 
isl_printer_print_union_set(__isl_take isl_printer * p,__isl_keep isl_union_set * uset)1720 __isl_give isl_printer *isl_printer_print_union_set(__isl_take isl_printer *p,
1721 	__isl_keep isl_union_set *uset)
1722 {
1723 	if (!p || !uset)
1724 		goto error;
1725 
1726 	if (p->output_format == ISL_FORMAT_ISL)
1727 		return isl_union_map_print_isl(uset_to_umap(uset), p);
1728 	if (p->output_format == ISL_FORMAT_LATEX)
1729 		return isl_union_map_print_latex(uset_to_umap(uset), p);
1730 
1731 	isl_die(p->ctx, isl_error_invalid,
1732 		"invalid output format for isl_union_set", goto error);
1733 error:
1734 	isl_printer_free(p);
1735 	return NULL;
1736 }
1737 
poly_rec_n_non_zero(__isl_keep isl_poly_rec * rec)1738 static isl_size poly_rec_n_non_zero(__isl_keep isl_poly_rec *rec)
1739 {
1740 	int i;
1741 	int n;
1742 
1743 	if (!rec)
1744 		return isl_size_error;
1745 
1746 	for (i = 0, n = 0; i < rec->n; ++i) {
1747 		isl_bool is_zero = isl_poly_is_zero(rec->p[i]);
1748 
1749 		if (is_zero < 0)
1750 			return isl_size_error;
1751 		if (!is_zero)
1752 			++n;
1753 	}
1754 
1755 	return n;
1756 }
1757 
poly_print_cst(__isl_keep isl_poly * poly,__isl_take isl_printer * p,int first)1758 static __isl_give isl_printer *poly_print_cst(__isl_keep isl_poly *poly,
1759 	__isl_take isl_printer *p, int first)
1760 {
1761 	isl_poly_cst *cst;
1762 	int neg;
1763 
1764 	cst = isl_poly_as_cst(poly);
1765 	if (!cst)
1766 		goto error;
1767 	neg = !first && isl_int_is_neg(cst->n);
1768 	if (!first)
1769 		p = isl_printer_print_str(p, neg ? " - " :  " + ");
1770 	if (neg)
1771 		isl_int_neg(cst->n, cst->n);
1772 	if (isl_int_is_zero(cst->d)) {
1773 		int sgn = isl_int_sgn(cst->n);
1774 		p = isl_printer_print_str(p, sgn < 0 ? "-infty" :
1775 					    sgn == 0 ? "NaN" : "infty");
1776 	} else
1777 		p = isl_printer_print_isl_int(p, cst->n);
1778 	if (neg)
1779 		isl_int_neg(cst->n, cst->n);
1780 	if (!isl_int_is_zero(cst->d) && !isl_int_is_one(cst->d)) {
1781 		p = isl_printer_print_str(p, "/");
1782 		p = isl_printer_print_isl_int(p, cst->d);
1783 	}
1784 	return p;
1785 error:
1786 	isl_printer_free(p);
1787 	return NULL;
1788 }
1789 
print_base(__isl_take isl_printer * p,__isl_keep isl_space * space,__isl_keep isl_mat * div,int var)1790 static __isl_give isl_printer *print_base(__isl_take isl_printer *p,
1791 	__isl_keep isl_space *space, __isl_keep isl_mat *div, int var)
1792 {
1793 	isl_size total;
1794 
1795 	total = isl_space_dim(space, isl_dim_all);
1796 	if (total < 0)
1797 		return isl_printer_free(p);
1798 	if (var < total)
1799 		p = print_term(space, NULL, space->ctx->one, 1 + var, p, 0);
1800 	else
1801 		p = print_div(space, div, var - total, p);
1802 	return p;
1803 }
1804 
print_pow(__isl_take isl_printer * p,__isl_keep isl_space * space,__isl_keep isl_mat * div,int var,int exp)1805 static __isl_give isl_printer *print_pow(__isl_take isl_printer *p,
1806 	__isl_keep isl_space *space, __isl_keep isl_mat *div, int var, int exp)
1807 {
1808 	p = print_base(p, space, div, var);
1809 	if (exp == 1)
1810 		return p;
1811 	if (p->output_format == ISL_FORMAT_C) {
1812 		int i;
1813 		for (i = 1; i < exp; ++i) {
1814 			p = isl_printer_print_str(p, "*");
1815 			p = print_base(p, space, div, var);
1816 		}
1817 	} else {
1818 		p = isl_printer_print_str(p, "^");
1819 		p = isl_printer_print_int(p, exp);
1820 	}
1821 	return p;
1822 }
1823 
1824 /* Print the polynomial "poly" defined over the domain space "space" and
1825  * local variables defined by "div" to "p".
1826  */
poly_print(__isl_keep isl_poly * poly,__isl_keep isl_space * space,__isl_keep isl_mat * div,__isl_take isl_printer * p)1827 static __isl_give isl_printer *poly_print(__isl_keep isl_poly *poly,
1828 	__isl_keep isl_space *space, __isl_keep isl_mat *div,
1829 	__isl_take isl_printer *p)
1830 {
1831 	int i, first, print_parens;
1832 	isl_size n;
1833 	isl_bool is_cst;
1834 	isl_poly_rec *rec;
1835 
1836 	is_cst = isl_poly_is_cst(poly);
1837 	if (!p || is_cst < 0 || !space || !div)
1838 		goto error;
1839 
1840 	if (is_cst)
1841 		return poly_print_cst(poly, p, 1);
1842 
1843 	rec = isl_poly_as_rec(poly);
1844 	n = poly_rec_n_non_zero(rec);
1845 	if (n < 0)
1846 		return isl_printer_free(p);
1847 	print_parens = n > 1;
1848 	if (print_parens)
1849 		p = isl_printer_print_str(p, "(");
1850 	for (i = 0, first = 1; i < rec->n; ++i) {
1851 		isl_bool is_zero = isl_poly_is_zero(rec->p[i]);
1852 		isl_bool is_one = isl_poly_is_one(rec->p[i]);
1853 		isl_bool is_negone = isl_poly_is_negone(rec->p[i]);
1854 		isl_bool is_cst = isl_poly_is_cst(rec->p[i]);
1855 
1856 		if (is_zero < 0 || is_one < 0 || is_negone < 0)
1857 			return isl_printer_free(p);
1858 		if (is_zero)
1859 			continue;
1860 		if (is_negone) {
1861 			if (!i)
1862 				p = isl_printer_print_str(p, "-1");
1863 			else if (first)
1864 				p = isl_printer_print_str(p, "-");
1865 			else
1866 				p = isl_printer_print_str(p, " - ");
1867 		} else if (is_cst && !is_one)
1868 			p = poly_print_cst(rec->p[i], p, first);
1869 		else {
1870 			if (!first)
1871 				p = isl_printer_print_str(p, " + ");
1872 			if (i == 0 || !is_one)
1873 				p = poly_print(rec->p[i], space, div, p);
1874 		}
1875 		first = 0;
1876 		if (i == 0)
1877 			continue;
1878 		if (!is_one && !is_negone)
1879 			p = isl_printer_print_str(p, " * ");
1880 		p = print_pow(p, space, div, rec->poly.var, i);
1881 	}
1882 	if (print_parens)
1883 		p = isl_printer_print_str(p, ")");
1884 	return p;
1885 error:
1886 	isl_printer_free(p);
1887 	return NULL;
1888 }
1889 
print_qpolynomial(__isl_take isl_printer * p,__isl_keep isl_qpolynomial * qp)1890 static __isl_give isl_printer *print_qpolynomial(__isl_take isl_printer *p,
1891 	__isl_keep isl_qpolynomial *qp)
1892 {
1893 	if (!p || !qp)
1894 		goto error;
1895 	p = poly_print(qp->poly, qp->dim, qp->div, p);
1896 	return p;
1897 error:
1898 	isl_printer_free(p);
1899 	return NULL;
1900 }
1901 
print_qpolynomial_isl(__isl_take isl_printer * p,__isl_keep isl_qpolynomial * qp)1902 static __isl_give isl_printer *print_qpolynomial_isl(__isl_take isl_printer *p,
1903 	__isl_keep isl_qpolynomial *qp)
1904 {
1905 	struct isl_print_space_data data = { 0 };
1906 
1907 	if (!p || !qp)
1908 		goto error;
1909 
1910 	p = print_param_tuple(p, qp->dim, &data);
1911 	p = isl_printer_print_str(p, "{ ");
1912 	if (!isl_space_is_params(qp->dim)) {
1913 		p = isl_print_space(qp->dim, p, 0, &data);
1914 		p = isl_printer_print_str(p, " -> ");
1915 	}
1916 	p = print_qpolynomial(p, qp);
1917 	p = isl_printer_print_str(p, " }");
1918 	return p;
1919 error:
1920 	isl_printer_free(p);
1921 	return NULL;
1922 }
1923 
1924 /* Print the quasi-polynomial "qp" to "p" in C format, with the variable names
1925  * taken from the domain space "space".
1926  */
print_qpolynomial_c(__isl_take isl_printer * p,__isl_keep isl_space * space,__isl_keep isl_qpolynomial * qp)1927 static __isl_give isl_printer *print_qpolynomial_c(__isl_take isl_printer *p,
1928 	__isl_keep isl_space *space, __isl_keep isl_qpolynomial *qp)
1929 {
1930 	isl_bool is_one;
1931 	isl_val *den;
1932 
1933 	den = isl_qpolynomial_get_den(qp);
1934 	qp = isl_qpolynomial_copy(qp);
1935 	qp = isl_qpolynomial_scale_val(qp, isl_val_copy(den));
1936 	is_one = isl_val_is_one(den);
1937 	if (is_one < 0)
1938 		p = isl_printer_free(p);
1939 	if (!is_one)
1940 		p = isl_printer_print_str(p, "(");
1941 	if (qp)
1942 		p = poly_print(qp->poly, space, qp->div, p);
1943 	else
1944 		p = isl_printer_free(p);
1945 	if (!is_one) {
1946 		p = isl_printer_print_str(p, ")/");
1947 		p = isl_printer_print_val(p, den);
1948 	}
1949 	isl_qpolynomial_free(qp);
1950 	isl_val_free(den);
1951 	return p;
1952 }
1953 
isl_printer_print_qpolynomial(__isl_take isl_printer * p,__isl_keep isl_qpolynomial * qp)1954 __isl_give isl_printer *isl_printer_print_qpolynomial(
1955 	__isl_take isl_printer *p, __isl_keep isl_qpolynomial *qp)
1956 {
1957 	if (!p || !qp)
1958 		goto error;
1959 
1960 	if (p->output_format == ISL_FORMAT_ISL)
1961 		return print_qpolynomial_isl(p, qp);
1962 	else if (p->output_format == ISL_FORMAT_C)
1963 		return print_qpolynomial_c(p, qp->dim, qp);
1964 	else
1965 		isl_die(qp->dim->ctx, isl_error_unsupported,
1966 			"output format not supported for isl_qpolynomials",
1967 			goto error);
1968 error:
1969 	isl_printer_free(p);
1970 	return NULL;
1971 }
1972 
isl_qpolynomial_print(__isl_keep isl_qpolynomial * qp,FILE * out,unsigned output_format)1973 void isl_qpolynomial_print(__isl_keep isl_qpolynomial *qp, FILE *out,
1974 	unsigned output_format)
1975 {
1976 	isl_printer *p;
1977 
1978 	if  (!qp)
1979 		return;
1980 
1981 	isl_assert(qp->dim->ctx, output_format == ISL_FORMAT_ISL, return);
1982 	p = isl_printer_to_file(qp->dim->ctx, out);
1983 	p = isl_printer_print_qpolynomial(p, qp);
1984 	isl_printer_free(p);
1985 }
1986 
qpolynomial_fold_print(__isl_keep isl_qpolynomial_fold * fold,__isl_take isl_printer * p)1987 static __isl_give isl_printer *qpolynomial_fold_print(
1988 	__isl_keep isl_qpolynomial_fold *fold, __isl_take isl_printer *p)
1989 {
1990 	int i;
1991 	isl_qpolynomial_list *list;
1992 	isl_size n;
1993 
1994 	list = isl_qpolynomial_fold_peek_list(fold);
1995 	n = isl_qpolynomial_list_size(list);
1996 	if (n < 0)
1997 		return isl_printer_free(p);
1998 	if (fold->type == isl_fold_min)
1999 		p = isl_printer_print_str(p, "min");
2000 	else if (fold->type == isl_fold_max)
2001 		p = isl_printer_print_str(p, "max");
2002 	p = isl_printer_print_str(p, "(");
2003 	for (i = 0; i < n; ++i) {
2004 		isl_qpolynomial *qp;
2005 
2006 		if (i)
2007 			p = isl_printer_print_str(p, ", ");
2008 		qp = isl_qpolynomial_list_peek(list, i);
2009 		p = print_qpolynomial(p, qp);
2010 	}
2011 	p = isl_printer_print_str(p, ")");
2012 	return p;
2013 }
2014 
isl_qpolynomial_fold_print(__isl_keep isl_qpolynomial_fold * fold,FILE * out,unsigned output_format)2015 void isl_qpolynomial_fold_print(__isl_keep isl_qpolynomial_fold *fold,
2016 	FILE *out, unsigned output_format)
2017 {
2018 	isl_printer *p;
2019 
2020 	if (!fold)
2021 		return;
2022 
2023 	isl_assert(fold->dim->ctx, output_format == ISL_FORMAT_ISL, return);
2024 
2025 	p = isl_printer_to_file(fold->dim->ctx, out);
2026 	p = isl_printer_print_qpolynomial_fold(p, fold);
2027 
2028 	isl_printer_free(p);
2029 }
2030 
isl_pwqp_print_isl_body(__isl_take isl_printer * p,__isl_keep isl_pw_qpolynomial * pwqp)2031 static __isl_give isl_printer *isl_pwqp_print_isl_body(
2032 	__isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial *pwqp)
2033 {
2034 	struct isl_print_space_data data = { 0 };
2035 	int i = 0;
2036 
2037 	for (i = 0; i < pwqp->n; ++i) {
2038 		isl_space *space;
2039 
2040 		if (i)
2041 			p = isl_printer_print_str(p, "; ");
2042 		space = isl_qpolynomial_get_domain_space(pwqp->p[i].qp);
2043 		if (!isl_space_is_params(space)) {
2044 			p = isl_print_space(space, p, 0, &data);
2045 			p = isl_printer_print_str(p, " -> ");
2046 		}
2047 		p = print_qpolynomial(p, pwqp->p[i].qp);
2048 		p = print_disjuncts(set_to_map(pwqp->p[i].set), space, p, 0);
2049 		isl_space_free(space);
2050 	}
2051 
2052 	return p;
2053 }
2054 
print_pw_qpolynomial_isl(__isl_take isl_printer * p,__isl_keep isl_pw_qpolynomial * pwqp)2055 static __isl_give isl_printer *print_pw_qpolynomial_isl(
2056 	__isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial *pwqp)
2057 {
2058 	struct isl_print_space_data data = { 0 };
2059 
2060 	if (!p || !pwqp)
2061 		goto error;
2062 
2063 	p = print_param_tuple(p, pwqp->dim, &data);
2064 	p = isl_printer_print_str(p, "{ ");
2065 	if (pwqp->n == 0) {
2066 		if (!isl_space_is_set(pwqp->dim)) {
2067 			p = print_tuple(pwqp->dim, p, isl_dim_in, &data);
2068 			p = isl_printer_print_str(p, " -> ");
2069 		}
2070 		p = isl_printer_print_str(p, "0");
2071 	}
2072 	p = isl_pwqp_print_isl_body(p, pwqp);
2073 	p = isl_printer_print_str(p, " }");
2074 	return p;
2075 error:
2076 	isl_printer_free(p);
2077 	return NULL;
2078 }
2079 
isl_pw_qpolynomial_print(__isl_keep isl_pw_qpolynomial * pwqp,FILE * out,unsigned output_format)2080 void isl_pw_qpolynomial_print(__isl_keep isl_pw_qpolynomial *pwqp, FILE *out,
2081 	unsigned output_format)
2082 {
2083 	isl_printer *p;
2084 
2085 	if (!pwqp)
2086 		return;
2087 
2088 	p = isl_printer_to_file(pwqp->dim->ctx, out);
2089 	p = isl_printer_set_output_format(p, output_format);
2090 	p = isl_printer_print_pw_qpolynomial(p, pwqp);
2091 
2092 	isl_printer_free(p);
2093 }
2094 
isl_pwf_print_isl_body(__isl_take isl_printer * p,__isl_keep isl_pw_qpolynomial_fold * pwf)2095 static __isl_give isl_printer *isl_pwf_print_isl_body(
2096 	__isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial_fold *pwf)
2097 {
2098 	struct isl_print_space_data data = { 0 };
2099 	int i = 0;
2100 
2101 	for (i = 0; i < pwf->n; ++i) {
2102 		isl_space *space;
2103 
2104 		if (i)
2105 			p = isl_printer_print_str(p, "; ");
2106 		space = isl_qpolynomial_fold_get_domain_space(pwf->p[i].fold);
2107 		if (!isl_space_is_params(space)) {
2108 			p = isl_print_space(space, p, 0, &data);
2109 			p = isl_printer_print_str(p, " -> ");
2110 		}
2111 		p = qpolynomial_fold_print(pwf->p[i].fold, p);
2112 		p = print_disjuncts(set_to_map(pwf->p[i].set), space, p, 0);
2113 		isl_space_free(space);
2114 	}
2115 
2116 	return p;
2117 }
2118 
print_pw_qpolynomial_fold_isl(__isl_take isl_printer * p,__isl_keep isl_pw_qpolynomial_fold * pwf)2119 static __isl_give isl_printer *print_pw_qpolynomial_fold_isl(
2120 	__isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial_fold *pwf)
2121 {
2122 	struct isl_print_space_data data = { 0 };
2123 
2124 	p = print_param_tuple(p, pwf->dim, &data);
2125 	p = isl_printer_print_str(p, "{ ");
2126 	if (pwf->n == 0) {
2127 		if (!isl_space_is_set(pwf->dim)) {
2128 			p = print_tuple(pwf->dim, p, isl_dim_in, &data);
2129 			p = isl_printer_print_str(p, " -> ");
2130 		}
2131 		p = isl_printer_print_str(p, "0");
2132 	}
2133 	p = isl_pwf_print_isl_body(p, pwf);
2134 	p = isl_printer_print_str(p, " }");
2135 	return p;
2136 }
2137 
2138 static __isl_give isl_printer *print_ls_affine_c(__isl_take isl_printer *p,
2139 	__isl_keep isl_local_space *ls, isl_int *c);
2140 
2141 /* We skip the constraint if it is implied by the div expression.
2142  *
2143  * *first indicates whether this is the first constraint in the conjunction and
2144  * is updated if the constraint is actually printed.
2145  */
print_constraint_c(__isl_take isl_printer * p,__isl_keep isl_local_space * ls,isl_int * c,const char * op,int * first)2146 static __isl_give isl_printer *print_constraint_c(__isl_take isl_printer *p,
2147 	__isl_keep isl_local_space *ls, isl_int *c, const char *op, int *first)
2148 {
2149 	unsigned o_div;
2150 	isl_size n_div;
2151 	int div;
2152 
2153 	o_div = isl_local_space_offset(ls, isl_dim_div);
2154 	n_div = isl_local_space_dim(ls, isl_dim_div);
2155 	if (n_div < 0)
2156 		return isl_printer_free(p);
2157 	div = isl_seq_last_non_zero(c + o_div, n_div);
2158 	if (div >= 0) {
2159 		isl_bool is_div = isl_local_space_is_div_constraint(ls, c, div);
2160 		if (is_div < 0)
2161 			return isl_printer_free(p);
2162 		if (is_div)
2163 			return p;
2164 	}
2165 
2166 	if (!*first)
2167 		p = isl_printer_print_str(p, " && ");
2168 
2169 	p = print_ls_affine_c(p, ls, c);
2170 	p = isl_printer_print_str(p, " ");
2171 	p = isl_printer_print_str(p, op);
2172 	p = isl_printer_print_str(p, " 0");
2173 
2174 	*first = 0;
2175 
2176 	return p;
2177 }
2178 
2179 static __isl_give isl_printer *print_ls_partial_affine_c(
2180 	__isl_take isl_printer *p, __isl_keep isl_local_space *ls,
2181 	isl_int *c, unsigned len);
2182 
print_basic_set_c(__isl_take isl_printer * p,__isl_keep isl_space * space,__isl_keep isl_basic_set * bset)2183 static __isl_give isl_printer *print_basic_set_c(__isl_take isl_printer *p,
2184 	__isl_keep isl_space *space, __isl_keep isl_basic_set *bset)
2185 {
2186 	int i, j;
2187 	int first = 1;
2188 	isl_size n_div = isl_basic_set_dim(bset, isl_dim_div);
2189 	isl_size total = isl_basic_set_dim(bset, isl_dim_all);
2190 	isl_mat *div;
2191 	isl_local_space *ls;
2192 
2193 	if (n_div < 0 || total < 0)
2194 		return isl_printer_free(p);
2195 
2196 	total -= n_div;
2197 	div = isl_basic_set_get_divs(bset);
2198 	ls = isl_local_space_alloc_div(isl_space_copy(space), div);
2199 	for (i = 0; i < bset->n_eq; ++i) {
2200 		j = isl_seq_last_non_zero(bset->eq[i] + 1 + total, n_div);
2201 		if (j < 0)
2202 			p = print_constraint_c(p, ls,
2203 						bset->eq[i], "==", &first);
2204 		else {
2205 			if (i)
2206 				p = isl_printer_print_str(p, " && ");
2207 			p = isl_printer_print_str(p, "(");
2208 			p = print_ls_partial_affine_c(p, ls, bset->eq[i],
2209 						   1 + total + j);
2210 			p = isl_printer_print_str(p, ") % ");
2211 			p = isl_printer_print_isl_int(p,
2212 						bset->eq[i][1 + total + j]);
2213 			p = isl_printer_print_str(p, " == 0");
2214 			first = 0;
2215 		}
2216 	}
2217 	for (i = 0; i < bset->n_ineq; ++i)
2218 		p = print_constraint_c(p, ls, bset->ineq[i], ">=", &first);
2219 	isl_local_space_free(ls);
2220 	return p;
2221 }
2222 
print_set_c(__isl_take isl_printer * p,__isl_keep isl_space * space,__isl_keep isl_set * set)2223 static __isl_give isl_printer *print_set_c(__isl_take isl_printer *p,
2224 	__isl_keep isl_space *space, __isl_keep isl_set *set)
2225 {
2226 	int i;
2227 
2228 	if (!set)
2229 		return isl_printer_free(p);
2230 
2231 	if (set->n == 0)
2232 		p = isl_printer_print_str(p, "0");
2233 
2234 	for (i = 0; i < set->n; ++i) {
2235 		if (i)
2236 			p = isl_printer_print_str(p, " || ");
2237 		if (set->n > 1)
2238 			p = isl_printer_print_str(p, "(");
2239 		p = print_basic_set_c(p, space, set->p[i]);
2240 		if (set->n > 1)
2241 			p = isl_printer_print_str(p, ")");
2242 	}
2243 	return p;
2244 }
2245 
2246 /* Print the piecewise quasi-polynomial "pwqp" to "p" in C format.
2247  */
print_pw_qpolynomial_c(__isl_take isl_printer * p,__isl_keep isl_pw_qpolynomial * pwqp)2248 static __isl_give isl_printer *print_pw_qpolynomial_c(
2249 	__isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial *pwqp)
2250 {
2251 	int i;
2252 	isl_space *space;
2253 
2254 	space = isl_pw_qpolynomial_get_domain_space(pwqp);
2255 	if (pwqp->n == 1 && isl_set_plain_is_universe(pwqp->p[0].set)) {
2256 		p = print_qpolynomial_c(p, space, pwqp->p[0].qp);
2257 		isl_space_free(space);
2258 		return p;
2259 	}
2260 
2261 	for (i = 0; i < pwqp->n; ++i) {
2262 		p = isl_printer_print_str(p, "(");
2263 		p = print_set_c(p, space, pwqp->p[i].set);
2264 		p = isl_printer_print_str(p, ") ? (");
2265 		p = print_qpolynomial_c(p, space, pwqp->p[i].qp);
2266 		p = isl_printer_print_str(p, ") : ");
2267 	}
2268 
2269 	isl_space_free(space);
2270 	p = isl_printer_print_str(p, "0");
2271 	return p;
2272 }
2273 
isl_printer_print_pw_qpolynomial(__isl_take isl_printer * p,__isl_keep isl_pw_qpolynomial * pwqp)2274 __isl_give isl_printer *isl_printer_print_pw_qpolynomial(
2275 	__isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial *pwqp)
2276 {
2277 	if (!p || !pwqp)
2278 		goto error;
2279 
2280 	if (p->output_format == ISL_FORMAT_ISL)
2281 		return print_pw_qpolynomial_isl(p, pwqp);
2282 	else if (p->output_format == ISL_FORMAT_C)
2283 		return print_pw_qpolynomial_c(p, pwqp);
2284 	isl_assert(p->ctx, 0, goto error);
2285 error:
2286 	isl_printer_free(p);
2287 	return NULL;
2288 }
2289 
print_pwqp_body(__isl_take isl_pw_qpolynomial * pwqp,void * user)2290 static isl_stat print_pwqp_body(__isl_take isl_pw_qpolynomial *pwqp, void *user)
2291 {
2292 	struct isl_union_print_data *data;
2293 	data = (struct isl_union_print_data *)user;
2294 
2295 	if (!data->first)
2296 		data->p = isl_printer_print_str(data->p, "; ");
2297 	data->first = 0;
2298 
2299 	data->p = isl_pwqp_print_isl_body(data->p, pwqp);
2300 	isl_pw_qpolynomial_free(pwqp);
2301 
2302 	return isl_stat_ok;
2303 }
2304 
print_union_pw_qpolynomial_isl(__isl_take isl_printer * p,__isl_keep isl_union_pw_qpolynomial * upwqp)2305 static __isl_give isl_printer *print_union_pw_qpolynomial_isl(
2306 	__isl_take isl_printer *p, __isl_keep isl_union_pw_qpolynomial *upwqp)
2307 {
2308 	struct isl_union_print_data data;
2309 	struct isl_print_space_data space_data = { 0 };
2310 	isl_space *space;
2311 
2312 	space = isl_union_pw_qpolynomial_get_space(upwqp);
2313 	p = print_param_tuple(p, space, &space_data);
2314 	isl_space_free(space);
2315 	p = isl_printer_print_str(p, "{ ");
2316 	data.p = p;
2317 	data.first = 1;
2318 	isl_union_pw_qpolynomial_foreach_pw_qpolynomial(upwqp, &print_pwqp_body,
2319 							&data);
2320 	p = data.p;
2321 	p = isl_printer_print_str(p, " }");
2322 	return p;
2323 }
2324 
isl_printer_print_union_pw_qpolynomial(__isl_take isl_printer * p,__isl_keep isl_union_pw_qpolynomial * upwqp)2325 __isl_give isl_printer *isl_printer_print_union_pw_qpolynomial(
2326 	__isl_take isl_printer *p, __isl_keep isl_union_pw_qpolynomial *upwqp)
2327 {
2328 	if (!p || !upwqp)
2329 		goto error;
2330 
2331 	if (p->output_format == ISL_FORMAT_ISL)
2332 		return print_union_pw_qpolynomial_isl(p, upwqp);
2333 	isl_die(p->ctx, isl_error_invalid,
2334 		"invalid output format for isl_union_pw_qpolynomial",
2335 		goto error);
2336 error:
2337 	isl_printer_free(p);
2338 	return NULL;
2339 }
2340 
2341 /* Print the quasi-polynomial reduction "fold" to "p" in C format,
2342  * with the variable names taken from the domain space "space".
2343  */
print_qpolynomial_fold_c(__isl_take isl_printer * p,__isl_keep isl_space * space,__isl_keep isl_qpolynomial_fold * fold)2344 static __isl_give isl_printer *print_qpolynomial_fold_c(
2345 	__isl_take isl_printer *p, __isl_keep isl_space *space,
2346 	__isl_keep isl_qpolynomial_fold *fold)
2347 {
2348 	int i;
2349 	isl_qpolynomial_list *list;
2350 	isl_size n;
2351 
2352 	list = isl_qpolynomial_fold_peek_list(fold);
2353 	n = isl_qpolynomial_list_size(list);
2354 	if (n < 0)
2355 		return isl_printer_free(p);
2356 	for (i = 0; i < n - 1; ++i)
2357 		if (fold->type == isl_fold_min)
2358 			p = isl_printer_print_str(p, "min(");
2359 		else if (fold->type == isl_fold_max)
2360 			p = isl_printer_print_str(p, "max(");
2361 
2362 	for (i = 0; i < n; ++i) {
2363 		isl_qpolynomial *qp;
2364 
2365 		if (i)
2366 			p = isl_printer_print_str(p, ", ");
2367 		qp = isl_qpolynomial_list_peek(list, i);
2368 		p = print_qpolynomial_c(p, space, qp);
2369 		if (i)
2370 			p = isl_printer_print_str(p, ")");
2371 	}
2372 	return p;
2373 }
2374 
isl_printer_print_qpolynomial_fold(__isl_take isl_printer * p,__isl_keep isl_qpolynomial_fold * fold)2375 __isl_give isl_printer *isl_printer_print_qpolynomial_fold(
2376 	__isl_take isl_printer *p, __isl_keep isl_qpolynomial_fold *fold)
2377 {
2378 	if  (!p || !fold)
2379 		goto error;
2380 	if (p->output_format == ISL_FORMAT_ISL)
2381 		return qpolynomial_fold_print(fold, p);
2382 	else if (p->output_format == ISL_FORMAT_C)
2383 		return print_qpolynomial_fold_c(p, fold->dim, fold);
2384 	isl_die(p->ctx, isl_error_unsupported, "unsupported output format",
2385 		goto error);
2386 error:
2387 	isl_printer_free(p);
2388 	return NULL;
2389 }
2390 
2391 /* Print the piecewise quasi-polynomial reduction "pwf" to "p" in C format.
2392  */
print_pw_qpolynomial_fold_c(__isl_take isl_printer * p,__isl_keep isl_pw_qpolynomial_fold * pwf)2393 static __isl_give isl_printer *print_pw_qpolynomial_fold_c(
2394 	__isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial_fold *pwf)
2395 {
2396 	int i;
2397 	isl_space *space;
2398 
2399 	space = isl_pw_qpolynomial_fold_get_domain_space(pwf);
2400 	if (pwf->n == 1 && isl_set_plain_is_universe(pwf->p[0].set)) {
2401 		p = print_qpolynomial_fold_c(p, space, pwf->p[0].fold);
2402 		isl_space_free(space);
2403 		return p;
2404 	}
2405 
2406 	for (i = 0; i < pwf->n; ++i) {
2407 		p = isl_printer_print_str(p, "(");
2408 		p = print_set_c(p, space, pwf->p[i].set);
2409 		p = isl_printer_print_str(p, ") ? (");
2410 		p = print_qpolynomial_fold_c(p, space, pwf->p[i].fold);
2411 		p = isl_printer_print_str(p, ") : ");
2412 	}
2413 
2414 	isl_space_free(space);
2415 	p = isl_printer_print_str(p, "0");
2416 	return p;
2417 }
2418 
isl_printer_print_pw_qpolynomial_fold(__isl_take isl_printer * p,__isl_keep isl_pw_qpolynomial_fold * pwf)2419 __isl_give isl_printer *isl_printer_print_pw_qpolynomial_fold(
2420 	__isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial_fold *pwf)
2421 {
2422 	if (!p || !pwf)
2423 		goto error;
2424 
2425 	if (p->output_format == ISL_FORMAT_ISL)
2426 		return print_pw_qpolynomial_fold_isl(p, pwf);
2427 	else if (p->output_format == ISL_FORMAT_C)
2428 		return print_pw_qpolynomial_fold_c(p, pwf);
2429 	isl_assert(p->ctx, 0, goto error);
2430 error:
2431 	isl_printer_free(p);
2432 	return NULL;
2433 }
2434 
isl_pw_qpolynomial_fold_print(__isl_keep isl_pw_qpolynomial_fold * pwf,FILE * out,unsigned output_format)2435 void isl_pw_qpolynomial_fold_print(__isl_keep isl_pw_qpolynomial_fold *pwf,
2436 	FILE *out, unsigned output_format)
2437 {
2438 	isl_printer *p;
2439 
2440 	if (!pwf)
2441 		return;
2442 
2443 	p = isl_printer_to_file(pwf->dim->ctx, out);
2444 	p = isl_printer_set_output_format(p, output_format);
2445 	p = isl_printer_print_pw_qpolynomial_fold(p, pwf);
2446 
2447 	isl_printer_free(p);
2448 }
2449 
print_pwf_body(__isl_take isl_pw_qpolynomial_fold * pwf,void * user)2450 static isl_stat print_pwf_body(__isl_take isl_pw_qpolynomial_fold *pwf,
2451 	void *user)
2452 {
2453 	struct isl_union_print_data *data;
2454 	data = (struct isl_union_print_data *)user;
2455 
2456 	if (!data->first)
2457 		data->p = isl_printer_print_str(data->p, "; ");
2458 	data->first = 0;
2459 
2460 	data->p = isl_pwf_print_isl_body(data->p, pwf);
2461 	isl_pw_qpolynomial_fold_free(pwf);
2462 
2463 	return isl_stat_ok;
2464 }
2465 
print_union_pw_qpolynomial_fold_isl(__isl_take isl_printer * p,__isl_keep isl_union_pw_qpolynomial_fold * upwf)2466 static __isl_give isl_printer *print_union_pw_qpolynomial_fold_isl(
2467 	__isl_take isl_printer *p,
2468 	__isl_keep isl_union_pw_qpolynomial_fold *upwf)
2469 {
2470 	struct isl_union_print_data data;
2471 	struct isl_print_space_data space_data = { 0 };
2472 	isl_space *space;
2473 
2474 	space = isl_union_pw_qpolynomial_fold_get_space(upwf);
2475 	p = print_param_tuple(p, space, &space_data);
2476 	isl_space_free(space);
2477 	p = isl_printer_print_str(p, "{ ");
2478 	data.p = p;
2479 	data.first = 1;
2480 	isl_union_pw_qpolynomial_fold_foreach_pw_qpolynomial_fold(upwf,
2481 							&print_pwf_body, &data);
2482 	p = data.p;
2483 	p = isl_printer_print_str(p, " }");
2484 	return p;
2485 }
2486 
isl_printer_print_union_pw_qpolynomial_fold(__isl_take isl_printer * p,__isl_keep isl_union_pw_qpolynomial_fold * upwf)2487 __isl_give isl_printer *isl_printer_print_union_pw_qpolynomial_fold(
2488 	__isl_take isl_printer *p,
2489 	__isl_keep isl_union_pw_qpolynomial_fold *upwf)
2490 {
2491 	if (!p || !upwf)
2492 		goto error;
2493 
2494 	if (p->output_format == ISL_FORMAT_ISL)
2495 		return print_union_pw_qpolynomial_fold_isl(p, upwf);
2496 	isl_die(p->ctx, isl_error_invalid,
2497 		"invalid output format for isl_union_pw_qpolynomial_fold",
2498 		goto error);
2499 error:
2500 	isl_printer_free(p);
2501 	return NULL;
2502 }
2503 
2504 /* Print the isl_constraint "c" to "p".
2505  */
isl_printer_print_constraint(__isl_take isl_printer * p,__isl_keep isl_constraint * c)2506 __isl_give isl_printer *isl_printer_print_constraint(__isl_take isl_printer *p,
2507 	__isl_keep isl_constraint *c)
2508 {
2509 	struct isl_print_space_data data = { 0 };
2510 	isl_local_space *ls;
2511 	isl_space *space;
2512 	isl_bool exists;
2513 
2514 	if (!p || !c)
2515 		goto error;
2516 
2517 	ls = isl_constraint_get_local_space(c);
2518 	if (!ls)
2519 		return isl_printer_free(p);
2520 	space = isl_local_space_get_space(ls);
2521 	p = print_param_tuple(p, space, &data);
2522 	p = isl_printer_print_str(p, "{ ");
2523 	p = isl_print_space(space, p, 0, &data);
2524 	p = isl_printer_print_str(p, " : ");
2525 	exists = need_exists(p, ls->div);
2526 	if (exists < 0)
2527 		p = isl_printer_free(p);
2528 	if (exists >= 0 && exists)
2529 		p = open_exists(p, space, ls->div, 0);
2530 	p = print_affine_of_len(space, ls->div, p, c->v->el, c->v->size);
2531 	if (isl_constraint_is_equality(c))
2532 		p = isl_printer_print_str(p, " = 0");
2533 	else
2534 		p = isl_printer_print_str(p, " >= 0");
2535 	if (exists >= 0 && exists)
2536 		p = isl_printer_print_str(p, s_close_exists[0]);
2537 	p = isl_printer_print_str(p, " }");
2538 	isl_space_free(space);
2539 	isl_local_space_free(ls);
2540 
2541 	return p;
2542 error:
2543 	isl_printer_free(p);
2544 	return NULL;
2545 }
2546 
isl_printer_print_space_isl(__isl_take isl_printer * p,__isl_keep isl_space * space)2547 static __isl_give isl_printer *isl_printer_print_space_isl(
2548 	__isl_take isl_printer *p, __isl_keep isl_space *space)
2549 {
2550 	struct isl_print_space_data data = { 0 };
2551 
2552 	if (!space)
2553 		goto error;
2554 
2555 	p = print_param_tuple(p, space, &data);
2556 
2557 	p = isl_printer_print_str(p, "{ ");
2558 	if (isl_space_is_params(space))
2559 		p = isl_printer_print_str(p, s_such_that[0]);
2560 	else
2561 		p = isl_print_space(space, p, 0, &data);
2562 	p = isl_printer_print_str(p, " }");
2563 
2564 	return p;
2565 error:
2566 	isl_printer_free(p);
2567 	return NULL;
2568 }
2569 
isl_printer_print_space(__isl_take isl_printer * p,__isl_keep isl_space * space)2570 __isl_give isl_printer *isl_printer_print_space(__isl_take isl_printer *p,
2571 	__isl_keep isl_space *space)
2572 {
2573 	if (!p || !space)
2574 		return isl_printer_free(p);
2575 	if (p->output_format == ISL_FORMAT_ISL)
2576 		return isl_printer_print_space_isl(p, space);
2577 	else if (p->output_format == ISL_FORMAT_OMEGA)
2578 		return print_omega_parameters(space, p);
2579 
2580 	isl_die(isl_space_get_ctx(space), isl_error_unsupported,
2581 		"output format not supported for space",
2582 		return isl_printer_free(p));
2583 }
2584 
isl_printer_print_local_space(__isl_take isl_printer * p,__isl_keep isl_local_space * ls)2585 __isl_give isl_printer *isl_printer_print_local_space(__isl_take isl_printer *p,
2586 	__isl_keep isl_local_space *ls)
2587 {
2588 	struct isl_print_space_data data = { 0 };
2589 	isl_size n_div;
2590 
2591 	n_div = isl_local_space_dim(ls, isl_dim_div);
2592 	if (n_div < 0)
2593 		goto error;
2594 
2595 	p = print_param_tuple(p, ls->dim, &data);
2596 	p = isl_printer_print_str(p, "{ ");
2597 	p = isl_print_space(ls->dim, p, 0, &data);
2598 	if (n_div > 0) {
2599 		p = isl_printer_print_str(p, " : ");
2600 		p = isl_printer_print_str(p, s_open_exists[0]);
2601 		p = print_div_list(p, ls->dim, ls->div, 0, 1);
2602 		p = isl_printer_print_str(p, s_close_exists[0]);
2603 	} else if (isl_space_is_params(ls->dim))
2604 		p = isl_printer_print_str(p, s_such_that[0]);
2605 	p = isl_printer_print_str(p, " }");
2606 	return p;
2607 error:
2608 	isl_printer_free(p);
2609 	return NULL;
2610 }
2611 
2612 /* Look for the last of the "n" integer divisions that is used in "aff" and
2613  * that can be printed as a modulo and
2614  * return the position of this integer division.
2615  * Return "n" if no such integer division can be found.
2616  * Return isl_size_error on error.
2617  *
2618  * In particular, look for an integer division that appears in "aff"
2619  * with a coefficient that is a multiple of the denominator
2620  * of the integer division.
2621  * That is, check if the numerator of "aff" is of the form
2622  *
2623  *	f(...) + a m floor(g/m)
2624  *
2625  * and return the position of "floor(g/m)".
2626  *
2627  * Note that, unlike print_as_modulo_pos, no check needs to be made
2628  * for whether the integer division can be printed, since it will
2629  * need to be printed as an integer division anyway if it is not printed
2630  * as a modulo.
2631  */
last_modulo(__isl_keep isl_printer * p,__isl_keep isl_aff * aff,unsigned n)2632 static isl_size last_modulo(__isl_keep isl_printer *p, __isl_keep isl_aff *aff,
2633 	unsigned n)
2634 {
2635 	isl_size o_div;
2636 	int i;
2637 
2638 	if (n == 0)
2639 		return n;
2640 	o_div = isl_aff_domain_offset(aff, isl_dim_div);
2641 	if (o_div < 0)
2642 		return isl_size_error;
2643 	for (i = n - 1; i >= 0; --i) {
2644 		if (isl_int_is_zero(aff->v->el[1 + o_div + i]))
2645 			continue;
2646 		if (isl_int_is_divisible_by(aff->v->el[1 + o_div + i],
2647 					    aff->ls->div->row[i][0]))
2648 			return i;
2649 	}
2650 
2651 	return n;
2652 }
2653 
2654 /* Print the numerator of the affine expression "aff" to "p",
2655  * with the variable names taken from "space".
2656  */
print_aff_num_base(__isl_take isl_printer * p,__isl_keep isl_space * space,__isl_keep isl_aff * aff)2657 static __isl_give isl_printer *print_aff_num_base(__isl_take isl_printer *p,
2658 	__isl_keep isl_space *space, __isl_keep isl_aff *aff)
2659 {
2660 	isl_size total;
2661 
2662 	total = isl_aff_domain_dim(aff, isl_dim_all);
2663 	if (total < 0)
2664 		return isl_printer_free(p);
2665 	p = print_affine_of_len(space, aff->ls->div, p,
2666 				aff->v->el + 1, 1 + total);
2667 
2668 	return p;
2669 }
2670 
2671 static __isl_give isl_printer *print_aff_num(__isl_take isl_printer *p,
2672 	__isl_keep isl_space *space, __isl_keep isl_aff *aff);
2673 
2674 /* Print the modulo term "c" * ("aff" mod "mod") to "p",
2675  * with the variable names taken from "space".
2676  * If "first" is set, then this is the first term of an expression.
2677  */
print_mod_term(__isl_take isl_printer * p,__isl_keep isl_space * space,__isl_keep isl_aff * aff,int first,__isl_take isl_val * c,__isl_keep isl_val * mod)2678 static __isl_give isl_printer *print_mod_term(__isl_take isl_printer *p,
2679 	__isl_keep isl_space *space, __isl_keep isl_aff *aff, int first,
2680 	__isl_take isl_val *c, __isl_keep isl_val *mod)
2681 {
2682 	isl_bool is_one, is_neg;
2683 
2684 	is_neg = isl_val_is_neg(c);
2685 	if (is_neg < 0)
2686 		p = isl_printer_free(p);
2687 	if (!first) {
2688 		if (is_neg)
2689 			c = isl_val_neg(c);
2690 		p = isl_printer_print_str(p, is_neg ? " - " : " + ");
2691 	}
2692 	is_one = isl_val_is_one(c);
2693 	if (is_one < 0)
2694 		p = isl_printer_free(p);
2695 	if (!is_one) {
2696 		p = isl_printer_print_val(p, c);
2697 		p = isl_printer_print_str(p, "*(");
2698 	}
2699 	p = isl_printer_print_str(p, "(");
2700 	p = print_aff_num(p, space, aff);
2701 	p = isl_printer_print_str(p, ")");
2702 	p = isl_printer_print_str(p, " mod ");
2703 	p = isl_printer_print_val(p, mod);
2704 	if (!is_one)
2705 		p = isl_printer_print_str(p, ")");
2706 
2707 	isl_val_free(c);
2708 
2709 	return p;
2710 }
2711 
2712 /* Print the numerator of the affine expression "aff" to "p",
2713  * with the variable names taken from "space",
2714  * given that the numerator of "aff" is of the form
2715  *
2716  *	f(...) + a m floor(g/m)
2717  *
2718  * with "floor(g/m)" the integer division at position "last".
2719  *
2720  * First replace "aff" by its numerator and rewrite it as
2721  *
2722  *	f(...) + a g - a (g mod m)
2723  *
2724  * Recursively write out (the numerator of) "f(...) + a g"
2725  * (which may involve other modulo expressions) and
2726  * then write out "- a (g mod m)".
2727  */
print_aff_num_mod(__isl_take isl_printer * p,__isl_keep isl_space * space,__isl_keep isl_aff * aff,unsigned last)2728 static __isl_give isl_printer *print_aff_num_mod(__isl_take isl_printer *p,
2729 	__isl_keep isl_space *space, __isl_keep isl_aff *aff, unsigned last)
2730 {
2731 	isl_bool is_zero;
2732 	isl_val *a, *m;
2733 	isl_aff *div, *term;
2734 
2735 	aff = isl_aff_copy(aff);
2736 	aff = isl_aff_scale_val(aff, isl_aff_get_denominator_val(aff));
2737 	a = isl_aff_get_coefficient_val(aff, isl_dim_div, last);
2738 	aff = isl_aff_set_coefficient_si(aff, isl_dim_div, last, 0);
2739 	div = isl_aff_get_div(aff, last);
2740 	m = isl_aff_get_denominator_val(div);
2741 	a = isl_val_div(a, isl_val_copy(m));
2742 	div = isl_aff_scale_val(div, isl_val_copy(m));
2743 	term = isl_aff_scale_val(isl_aff_copy(div), isl_val_copy(a));
2744 	aff = isl_aff_add(aff, term);
2745 
2746 	is_zero = isl_aff_plain_is_zero(aff);
2747 	if (is_zero < 0) {
2748 		p = isl_printer_free(p);
2749 	} else {
2750 		if (!is_zero)
2751 			p = print_aff_num(p, space, aff);
2752 		a = isl_val_neg(a);
2753 		p = print_mod_term(p, space, div, is_zero, isl_val_copy(a), m);
2754 	}
2755 
2756 	isl_val_free(a);
2757 	isl_val_free(m);
2758 	isl_aff_free(aff);
2759 	isl_aff_free(div);
2760 
2761 	return p;
2762 }
2763 
2764 /* Print the numerator of the affine expression "aff" to "p",
2765  * with the variable names taken from "space",
2766  * separating out any (obvious) modulo expressions.
2767  *
2768  * In particular, look for modulo expressions in "aff",
2769  * separating them out if found and simply printing out "aff" otherwise.
2770  */
print_aff_num(__isl_take isl_printer * p,__isl_keep isl_space * space,__isl_keep isl_aff * aff)2771 static __isl_give isl_printer *print_aff_num(__isl_take isl_printer *p,
2772 	__isl_keep isl_space *space, __isl_keep isl_aff *aff)
2773 {
2774 	isl_size n_div, mod;
2775 
2776 	n_div = isl_aff_dim(aff, isl_dim_div);
2777 	if (n_div < 0)
2778 		return isl_printer_free(p);
2779 	mod = last_modulo(p, aff, n_div);
2780 	if (mod < 0)
2781 		return isl_printer_free(p);
2782 	if (mod < n_div)
2783 		return print_aff_num_mod(p, space, aff, mod);
2784 	else
2785 		return print_aff_num_base(p, space, aff);
2786 }
2787 
2788 /* Print the (potentially rational) affine expression "aff" to "p",
2789  * with the variable names taken from "space".
2790  */
print_aff_body(__isl_take isl_printer * p,__isl_keep isl_space * space,__isl_keep isl_aff * aff)2791 static __isl_give isl_printer *print_aff_body(__isl_take isl_printer *p,
2792 	__isl_keep isl_space *space, __isl_keep isl_aff *aff)
2793 {
2794 	if (isl_aff_is_nan(aff))
2795 		return isl_printer_print_str(p, "NaN");
2796 
2797 	p = isl_printer_print_str(p, "(");
2798 	p = print_aff_num(p, space, aff);
2799 	if (isl_int_is_one(aff->v->el[0]))
2800 		p = isl_printer_print_str(p, ")");
2801 	else {
2802 		p = isl_printer_print_str(p, ")/");
2803 		p = isl_printer_print_isl_int(p, aff->v->el[0]);
2804 	}
2805 
2806 	return p;
2807 }
2808 
print_aff(__isl_take isl_printer * p,__isl_keep isl_aff * aff)2809 static __isl_give isl_printer *print_aff(__isl_take isl_printer *p,
2810 	__isl_keep isl_aff *aff)
2811 {
2812 	struct isl_print_space_data data = { 0 };
2813 
2814 	if (isl_space_is_params(aff->ls->dim))
2815 		;
2816 	else {
2817 		p = print_tuple(aff->ls->dim, p, isl_dim_set, &data);
2818 		p = isl_printer_print_str(p, " -> ");
2819 	}
2820 	p = isl_printer_print_str(p, "[");
2821 	p = print_aff_body(p, aff->ls->dim, aff);
2822 	p = isl_printer_print_str(p, "]");
2823 
2824 	return p;
2825 }
2826 
print_aff_isl(__isl_take isl_printer * p,__isl_keep isl_aff * aff)2827 static __isl_give isl_printer *print_aff_isl(__isl_take isl_printer *p,
2828 	__isl_keep isl_aff *aff)
2829 {
2830 	struct isl_print_space_data data = { 0 };
2831 
2832 	if (!aff)
2833 		goto error;
2834 
2835 	p = print_param_tuple(p, aff->ls->dim, &data);
2836 	p = isl_printer_print_str(p, "{ ");
2837 	p = print_aff(p, aff);
2838 	p = isl_printer_print_str(p, " }");
2839 	return p;
2840 error:
2841 	isl_printer_free(p);
2842 	return NULL;
2843 }
2844 
2845 /* Print the body of an isl_pw_aff, i.e., a semicolon delimited
2846  * sequence of affine expressions, each followed by constraints.
2847  */
print_pw_aff_body(__isl_take isl_printer * p,__isl_keep isl_pw_aff * pa)2848 static __isl_give isl_printer *print_pw_aff_body(
2849 	__isl_take isl_printer *p, __isl_keep isl_pw_aff *pa)
2850 {
2851 	int i;
2852 
2853 	if (!pa)
2854 		return isl_printer_free(p);
2855 
2856 	for (i = 0; i < pa->n; ++i) {
2857 		isl_space *space;
2858 
2859 		if (i)
2860 			p = isl_printer_print_str(p, "; ");
2861 		p = print_aff(p, pa->p[i].aff);
2862 		space = isl_aff_get_domain_space(pa->p[i].aff);
2863 		p = print_disjuncts(set_to_map(pa->p[i].set), space, p, 0);
2864 		isl_space_free(space);
2865 	}
2866 
2867 	return p;
2868 }
2869 
print_pw_aff_isl(__isl_take isl_printer * p,__isl_keep isl_pw_aff * pwaff)2870 static __isl_give isl_printer *print_pw_aff_isl(__isl_take isl_printer *p,
2871 	__isl_keep isl_pw_aff *pwaff)
2872 {
2873 	struct isl_print_space_data data = { 0 };
2874 
2875 	if (!pwaff)
2876 		goto error;
2877 
2878 	p = print_param_tuple(p, pwaff->dim, &data);
2879 	p = isl_printer_print_str(p, "{ ");
2880 	p = print_pw_aff_body(p, pwaff);
2881 	p = isl_printer_print_str(p, " }");
2882 	return p;
2883 error:
2884 	isl_printer_free(p);
2885 	return NULL;
2886 }
2887 
print_ls_name_c(__isl_take isl_printer * p,__isl_keep isl_local_space * ls,enum isl_dim_type type,unsigned pos)2888 static __isl_give isl_printer *print_ls_name_c(__isl_take isl_printer *p,
2889 	__isl_keep isl_local_space *ls, enum isl_dim_type type, unsigned pos)
2890 {
2891 	if (type == isl_dim_div) {
2892 		p = isl_printer_print_str(p, "floord(");
2893 		p = print_ls_affine_c(p, ls, ls->div->row[pos] + 1);
2894 		p = isl_printer_print_str(p, ", ");
2895 		p = isl_printer_print_isl_int(p, ls->div->row[pos][0]);
2896 		p = isl_printer_print_str(p, ")");
2897 	} else {
2898 		const char *name;
2899 
2900 		name = isl_space_get_dim_name(ls->dim, type, pos);
2901 		if (!name)
2902 			name = "UNNAMED";
2903 		p = isl_printer_print_str(p, name);
2904 	}
2905 	return p;
2906 }
2907 
print_ls_term_c(__isl_take isl_printer * p,__isl_keep isl_local_space * ls,isl_int c,unsigned pos)2908 static __isl_give isl_printer *print_ls_term_c(__isl_take isl_printer *p,
2909 	__isl_keep isl_local_space *ls, isl_int c, unsigned pos)
2910 {
2911 	enum isl_dim_type type;
2912 
2913 	if (!p || !ls)
2914 		return isl_printer_free(p);
2915 
2916 	if (pos == 0)
2917 		return isl_printer_print_isl_int(p, c);
2918 
2919 	if (isl_int_is_one(c))
2920 		;
2921 	else if (isl_int_is_negone(c))
2922 		p = isl_printer_print_str(p, "-");
2923 	else {
2924 		p = isl_printer_print_isl_int(p, c);
2925 		p = isl_printer_print_str(p, "*");
2926 	}
2927 	if (pos2type(ls->dim, &type, &pos) < 0)
2928 		return isl_printer_free(p);
2929 	p = print_ls_name_c(p, ls, type, pos);
2930 	return p;
2931 }
2932 
print_ls_partial_affine_c(__isl_take isl_printer * p,__isl_keep isl_local_space * ls,isl_int * c,unsigned len)2933 static __isl_give isl_printer *print_ls_partial_affine_c(
2934 	__isl_take isl_printer *p, __isl_keep isl_local_space *ls,
2935 	isl_int *c, unsigned len)
2936 {
2937 	int i;
2938 	int first;
2939 
2940 	for (i = 0, first = 1; i < len; ++i) {
2941 		int flip = 0;
2942 		if (isl_int_is_zero(c[i]))
2943 			continue;
2944 		if (!first) {
2945 			if (isl_int_is_neg(c[i])) {
2946 				flip = 1;
2947 				isl_int_neg(c[i], c[i]);
2948 				p = isl_printer_print_str(p, " - ");
2949 			} else
2950 				p = isl_printer_print_str(p, " + ");
2951 		}
2952 		first = 0;
2953 		p = print_ls_term_c(p, ls, c[i], i);
2954 		if (flip)
2955 			isl_int_neg(c[i], c[i]);
2956 	}
2957 	if (first)
2958 		p = isl_printer_print_str(p, "0");
2959 	return p;
2960 }
2961 
print_ls_affine_c(__isl_take isl_printer * p,__isl_keep isl_local_space * ls,isl_int * c)2962 static __isl_give isl_printer *print_ls_affine_c(__isl_take isl_printer *p,
2963 	__isl_keep isl_local_space *ls, isl_int *c)
2964 {
2965 	isl_size total = isl_local_space_dim(ls, isl_dim_all);
2966 
2967 	if (total < 0)
2968 		return isl_printer_free(p);
2969 	return print_ls_partial_affine_c(p, ls, c, 1 + total);
2970 }
2971 
print_aff_c(__isl_take isl_printer * p,__isl_keep isl_aff * aff)2972 static __isl_give isl_printer *print_aff_c(__isl_take isl_printer *p,
2973 	__isl_keep isl_aff *aff)
2974 {
2975 	isl_size total;
2976 
2977 	total = isl_aff_domain_dim(aff, isl_dim_all);
2978 	if (total < 0)
2979 		return isl_printer_free(p);
2980 	if (!isl_int_is_one(aff->v->el[0]))
2981 		p = isl_printer_print_str(p, "(");
2982 	p = print_ls_partial_affine_c(p, aff->ls, aff->v->el + 1, 1 + total);
2983 	if (!isl_int_is_one(aff->v->el[0])) {
2984 		p = isl_printer_print_str(p, ")/");
2985 		p = isl_printer_print_isl_int(p, aff->v->el[0]);
2986 	}
2987 	return p;
2988 }
2989 
2990 /* In the C format, we cannot express that "pwaff" may be undefined
2991  * on parts of the domain space.  We therefore assume that the expression
2992  * will only be evaluated on its definition domain and compute the gist
2993  * of each cell with respect to this domain.
2994  */
print_pw_aff_c(__isl_take isl_printer * p,__isl_keep isl_pw_aff * pwaff)2995 static __isl_give isl_printer *print_pw_aff_c(__isl_take isl_printer *p,
2996 	__isl_keep isl_pw_aff *pwaff)
2997 {
2998 	isl_set *domain;
2999 	isl_ast_build *build;
3000 	isl_ast_expr *expr;
3001 
3002 	if (pwaff->n < 1)
3003 		isl_die(p->ctx, isl_error_unsupported,
3004 			"cannot print empty isl_pw_aff in C format",
3005 			return isl_printer_free(p));
3006 
3007 	domain = isl_pw_aff_domain(isl_pw_aff_copy(pwaff));
3008 	build = isl_ast_build_from_context(domain);
3009 	expr = isl_ast_build_expr_from_pw_aff(build, isl_pw_aff_copy(pwaff));
3010 	p = isl_printer_print_ast_expr(p, expr);
3011 	isl_ast_expr_free(expr);
3012 	isl_ast_build_free(build);
3013 
3014 	return p;
3015 }
3016 
isl_printer_print_aff(__isl_take isl_printer * p,__isl_keep isl_aff * aff)3017 __isl_give isl_printer *isl_printer_print_aff(__isl_take isl_printer *p,
3018 	__isl_keep isl_aff *aff)
3019 {
3020 	if (!p || !aff)
3021 		goto error;
3022 
3023 	if (p->output_format == ISL_FORMAT_ISL)
3024 		return print_aff_isl(p, aff);
3025 	else if (p->output_format == ISL_FORMAT_C)
3026 		return print_aff_c(p, aff);
3027 	isl_die(p->ctx, isl_error_unsupported, "unsupported output format",
3028 		goto error);
3029 error:
3030 	isl_printer_free(p);
3031 	return NULL;
3032 }
3033 
isl_printer_print_pw_aff(__isl_take isl_printer * p,__isl_keep isl_pw_aff * pwaff)3034 __isl_give isl_printer *isl_printer_print_pw_aff(__isl_take isl_printer *p,
3035 	__isl_keep isl_pw_aff *pwaff)
3036 {
3037 	if (!p || !pwaff)
3038 		goto error;
3039 
3040 	if (p->output_format == ISL_FORMAT_ISL)
3041 		return print_pw_aff_isl(p, pwaff);
3042 	else if (p->output_format == ISL_FORMAT_C)
3043 		return print_pw_aff_c(p, pwaff);
3044 	isl_die(p->ctx, isl_error_unsupported, "unsupported output format",
3045 		goto error);
3046 error:
3047 	isl_printer_free(p);
3048 	return NULL;
3049 }
3050 
3051 /* Print "pa" in a sequence of isl_pw_affs delimited by semicolons.
3052  * Each isl_pw_aff itself is also printed as semicolon delimited
3053  * sequence of pieces.
3054  * If data->first = 1, then this is the first in the sequence.
3055  * Update data->first to tell the next element that it is not the first.
3056  */
print_pw_aff_body_wrap(__isl_take isl_pw_aff * pa,void * user)3057 static isl_stat print_pw_aff_body_wrap(__isl_take isl_pw_aff *pa,
3058 	void *user)
3059 {
3060 	struct isl_union_print_data *data;
3061 	data = (struct isl_union_print_data *) user;
3062 
3063 	if (!data->first)
3064 		data->p = isl_printer_print_str(data->p, "; ");
3065 	data->first = 0;
3066 
3067 	data->p = print_pw_aff_body(data->p, pa);
3068 	isl_pw_aff_free(pa);
3069 
3070 	return data->p ? isl_stat_ok : isl_stat_error;
3071 }
3072 
3073 /* Print the body of an isl_union_pw_aff, i.e., a semicolon delimited
3074  * sequence of affine expressions, each followed by constraints,
3075  * with the sequence enclosed in braces.
3076  */
print_union_pw_aff_body(__isl_take isl_printer * p,__isl_keep isl_union_pw_aff * upa)3077 static __isl_give isl_printer *print_union_pw_aff_body(
3078 	__isl_take isl_printer *p, __isl_keep isl_union_pw_aff *upa)
3079 {
3080 	struct isl_union_print_data data = { p, 1 };
3081 
3082 	p = isl_printer_print_str(p, s_open_set[0]);
3083 	data.p = p;
3084 	if (isl_union_pw_aff_foreach_pw_aff(upa,
3085 					    &print_pw_aff_body_wrap, &data) < 0)
3086 		data.p = isl_printer_free(data.p);
3087 	p = data.p;
3088 	p = isl_printer_print_str(p, s_close_set[0]);
3089 
3090 	return p;
3091 }
3092 
3093 /* Print the isl_union_pw_aff "upa" to "p" in isl format.
3094  *
3095  * The individual isl_pw_affs are delimited by a semicolon.
3096  */
print_union_pw_aff_isl(__isl_take isl_printer * p,__isl_keep isl_union_pw_aff * upa)3097 static __isl_give isl_printer *print_union_pw_aff_isl(
3098 	__isl_take isl_printer *p, __isl_keep isl_union_pw_aff *upa)
3099 {
3100 	struct isl_print_space_data data = { 0 };
3101 	isl_space *space;
3102 
3103 	space = isl_union_pw_aff_get_space(upa);
3104 	p = print_param_tuple(p, space, &data);
3105 	isl_space_free(space);
3106 	p = print_union_pw_aff_body(p, upa);
3107 	return p;
3108 }
3109 
3110 /* Print the isl_union_pw_aff "upa" to "p".
3111  *
3112  * We currently only support an isl format.
3113  */
isl_printer_print_union_pw_aff(__isl_take isl_printer * p,__isl_keep isl_union_pw_aff * upa)3114 __isl_give isl_printer *isl_printer_print_union_pw_aff(
3115 	__isl_take isl_printer *p, __isl_keep isl_union_pw_aff *upa)
3116 {
3117 	if (!p || !upa)
3118 		return isl_printer_free(p);
3119 
3120 	if (p->output_format == ISL_FORMAT_ISL)
3121 		return print_union_pw_aff_isl(p, upa);
3122 	isl_die(isl_printer_get_ctx(p), isl_error_unsupported,
3123 		"unsupported output format", return isl_printer_free(p));
3124 }
3125 
3126 /* Print dimension "pos" of data->space to "p".
3127  *
3128  * data->user is assumed to be an isl_multi_aff.
3129  *
3130  * If the current dimension is an output dimension, then print
3131  * the corresponding expression.  Otherwise, print the name of the dimension.
3132  * Make sure to use the domain space for printing names as
3133  * that is the space that will be used for printing constraints (if any).
3134  */
print_dim_ma(__isl_take isl_printer * p,struct isl_print_space_data * data,unsigned pos)3135 static __isl_give isl_printer *print_dim_ma(__isl_take isl_printer *p,
3136 	struct isl_print_space_data *data, unsigned pos)
3137 {
3138 	isl_multi_aff *ma = data->user;
3139 	isl_space *space;
3140 
3141 	space = isl_multi_aff_get_domain_space(ma);
3142 	if (data->type == isl_dim_out) {
3143 		p = print_aff_body(p, space, ma->u.p[pos]);
3144 	} else {
3145 		enum isl_dim_type type = data->type;
3146 
3147 		if (type == isl_dim_in)
3148 			type = isl_dim_set;
3149 		p = print_name(space, p, type, pos, data->latex);
3150 	}
3151 	isl_space_free(space);
3152 
3153 	return p;
3154 }
3155 
print_multi_aff(__isl_take isl_printer * p,__isl_keep isl_multi_aff * maff)3156 static __isl_give isl_printer *print_multi_aff(__isl_take isl_printer *p,
3157 	__isl_keep isl_multi_aff *maff)
3158 {
3159 	struct isl_print_space_data data = { 0 };
3160 
3161 	data.print_dim = &print_dim_ma;
3162 	data.user = maff;
3163 	return isl_print_space(maff->space, p, 0, &data);
3164 }
3165 
print_multi_aff_isl(__isl_take isl_printer * p,__isl_keep isl_multi_aff * maff)3166 static __isl_give isl_printer *print_multi_aff_isl(__isl_take isl_printer *p,
3167 	__isl_keep isl_multi_aff *maff)
3168 {
3169 	struct isl_print_space_data data = { 0 };
3170 
3171 	if (!maff)
3172 		goto error;
3173 
3174 	p = print_param_tuple(p, maff->space, &data);
3175 	p = isl_printer_print_str(p, "{ ");
3176 	p = print_multi_aff(p, maff);
3177 	p = isl_printer_print_str(p, " }");
3178 	return p;
3179 error:
3180 	isl_printer_free(p);
3181 	return NULL;
3182 }
3183 
isl_printer_print_multi_aff(__isl_take isl_printer * p,__isl_keep isl_multi_aff * maff)3184 __isl_give isl_printer *isl_printer_print_multi_aff(__isl_take isl_printer *p,
3185 	__isl_keep isl_multi_aff *maff)
3186 {
3187 	if (!p || !maff)
3188 		goto error;
3189 
3190 	if (p->output_format == ISL_FORMAT_ISL)
3191 		return print_multi_aff_isl(p, maff);
3192 	isl_die(p->ctx, isl_error_unsupported, "unsupported output format",
3193 		goto error);
3194 error:
3195 	isl_printer_free(p);
3196 	return NULL;
3197 }
3198 
print_pw_multi_aff_body(__isl_take isl_printer * p,__isl_keep isl_pw_multi_aff * pma)3199 static __isl_give isl_printer *print_pw_multi_aff_body(
3200 	__isl_take isl_printer *p, __isl_keep isl_pw_multi_aff *pma)
3201 {
3202 	int i;
3203 
3204 	if (!pma)
3205 		goto error;
3206 
3207 	for (i = 0; i < pma->n; ++i) {
3208 		isl_space *space;
3209 
3210 		if (i)
3211 			p = isl_printer_print_str(p, "; ");
3212 		p = print_multi_aff(p, pma->p[i].maff);
3213 		space = isl_multi_aff_get_domain_space(pma->p[i].maff);
3214 		p = print_disjuncts(set_to_map(pma->p[i].set), space, p, 0);
3215 		isl_space_free(space);
3216 	}
3217 	return p;
3218 error:
3219 	isl_printer_free(p);
3220 	return NULL;
3221 }
3222 
print_pw_multi_aff_isl(__isl_take isl_printer * p,__isl_keep isl_pw_multi_aff * pma)3223 static __isl_give isl_printer *print_pw_multi_aff_isl(__isl_take isl_printer *p,
3224 	__isl_keep isl_pw_multi_aff *pma)
3225 {
3226 	struct isl_print_space_data data = { 0 };
3227 
3228 	if (!pma)
3229 		goto error;
3230 
3231 	p = print_param_tuple(p, pma->dim, &data);
3232 	p = isl_printer_print_str(p, "{ ");
3233 	p = print_pw_multi_aff_body(p, pma);
3234 	p = isl_printer_print_str(p, " }");
3235 	return p;
3236 error:
3237 	isl_printer_free(p);
3238 	return NULL;
3239 }
3240 
3241 /* Print the unnamed, single-dimensional piecewise multi affine expression "pma"
3242  * to "p".
3243  */
print_unnamed_pw_multi_aff_c(__isl_take isl_printer * p,__isl_keep isl_pw_multi_aff * pma)3244 static __isl_give isl_printer *print_unnamed_pw_multi_aff_c(
3245 	__isl_take isl_printer *p, __isl_keep isl_pw_multi_aff *pma)
3246 {
3247 	int i;
3248 	isl_space *space;
3249 
3250 	space = isl_pw_multi_aff_get_domain_space(pma);
3251 	for (i = 0; i < pma->n - 1; ++i) {
3252 		p = isl_printer_print_str(p, "(");
3253 		p = print_set_c(p, space, pma->p[i].set);
3254 		p = isl_printer_print_str(p, ") ? (");
3255 		p = print_aff_c(p, pma->p[i].maff->u.p[0]);
3256 		p = isl_printer_print_str(p, ") : ");
3257 	}
3258 	isl_space_free(space);
3259 
3260 	return print_aff_c(p, pma->p[pma->n - 1].maff->u.p[0]);
3261 }
3262 
print_pw_multi_aff_c(__isl_take isl_printer * p,__isl_keep isl_pw_multi_aff * pma)3263 static __isl_give isl_printer *print_pw_multi_aff_c(__isl_take isl_printer *p,
3264 	__isl_keep isl_pw_multi_aff *pma)
3265 {
3266 	isl_size n;
3267 	const char *name;
3268 
3269 	if (!pma)
3270 		goto error;
3271 	if (pma->n < 1)
3272 		isl_die(p->ctx, isl_error_unsupported,
3273 			"cannot print empty isl_pw_multi_aff in C format",
3274 			goto error);
3275 	n = isl_pw_multi_aff_dim(pma, isl_dim_out);
3276 	if (n < 0)
3277 		return isl_printer_free(p);
3278 	name = isl_pw_multi_aff_get_tuple_name(pma, isl_dim_out);
3279 	if (!name && n == 1)
3280 		return print_unnamed_pw_multi_aff_c(p, pma);
3281 	if (!name)
3282 		isl_die(p->ctx, isl_error_unsupported,
3283 			"cannot print unnamed isl_pw_multi_aff in C format",
3284 			goto error);
3285 
3286 	p = isl_printer_print_str(p, name);
3287 	if (n != 0)
3288 		isl_die(p->ctx, isl_error_unsupported,
3289 			"not supported yet", goto error);
3290 
3291 	return p;
3292 error:
3293 	isl_printer_free(p);
3294 	return NULL;
3295 }
3296 
isl_printer_print_pw_multi_aff(__isl_take isl_printer * p,__isl_keep isl_pw_multi_aff * pma)3297 __isl_give isl_printer *isl_printer_print_pw_multi_aff(
3298 	__isl_take isl_printer *p, __isl_keep isl_pw_multi_aff *pma)
3299 {
3300 	if (!p || !pma)
3301 		goto error;
3302 
3303 	if (p->output_format == ISL_FORMAT_ISL)
3304 		return print_pw_multi_aff_isl(p, pma);
3305 	if (p->output_format == ISL_FORMAT_C)
3306 		return print_pw_multi_aff_c(p, pma);
3307 	isl_die(p->ctx, isl_error_unsupported, "unsupported output format",
3308 		goto error);
3309 error:
3310 	isl_printer_free(p);
3311 	return NULL;
3312 }
3313 
print_pw_multi_aff_body_wrap(__isl_take isl_pw_multi_aff * pma,void * user)3314 static isl_stat print_pw_multi_aff_body_wrap(__isl_take isl_pw_multi_aff *pma,
3315 	void *user)
3316 {
3317 	struct isl_union_print_data *data;
3318 	data = (struct isl_union_print_data *) user;
3319 
3320 	if (!data->first)
3321 		data->p = isl_printer_print_str(data->p, "; ");
3322 	data->first = 0;
3323 
3324 	data->p = print_pw_multi_aff_body(data->p, pma);
3325 	isl_pw_multi_aff_free(pma);
3326 
3327 	return isl_stat_ok;
3328 }
3329 
print_union_pw_multi_aff_isl(__isl_take isl_printer * p,__isl_keep isl_union_pw_multi_aff * upma)3330 static __isl_give isl_printer *print_union_pw_multi_aff_isl(
3331 	__isl_take isl_printer *p, __isl_keep isl_union_pw_multi_aff *upma)
3332 {
3333 	struct isl_union_print_data data;
3334 	struct isl_print_space_data space_data = { 0 };
3335 	isl_space *space;
3336 
3337 	space = isl_union_pw_multi_aff_get_space(upma);
3338 	p = print_param_tuple(p, space, &space_data);
3339 	isl_space_free(space);
3340 	p = isl_printer_print_str(p, s_open_set[0]);
3341 	data.p = p;
3342 	data.first = 1;
3343 	isl_union_pw_multi_aff_foreach_pw_multi_aff(upma,
3344 					&print_pw_multi_aff_body_wrap, &data);
3345 	p = data.p;
3346 	p = isl_printer_print_str(p, s_close_set[0]);
3347 	return p;
3348 }
3349 
isl_printer_print_union_pw_multi_aff(__isl_take isl_printer * p,__isl_keep isl_union_pw_multi_aff * upma)3350 __isl_give isl_printer *isl_printer_print_union_pw_multi_aff(
3351 	__isl_take isl_printer *p, __isl_keep isl_union_pw_multi_aff *upma)
3352 {
3353 	if (!p || !upma)
3354 		goto error;
3355 
3356 	if (p->output_format == ISL_FORMAT_ISL)
3357 		return print_union_pw_multi_aff_isl(p, upma);
3358 	isl_die(p->ctx, isl_error_unsupported, "unsupported output format",
3359 		goto error);
3360 error:
3361 	isl_printer_free(p);
3362 	return NULL;
3363 }
3364 
3365 /* Print dimension "pos" of data->space to "p".
3366  *
3367  * data->user is assumed to be an isl_multi_pw_aff.
3368  *
3369  * If the current dimension is an output dimension, then print
3370  * the corresponding piecewise affine expression.
3371  * Otherwise, print the name of the dimension.
3372  * Make sure to use the same space in both cases.
3373  * In particular, use the domain space for printing names as
3374  * that is the space that is used for printing constraints.
3375  */
print_dim_mpa(__isl_take isl_printer * p,struct isl_print_space_data * data,unsigned pos)3376 static __isl_give isl_printer *print_dim_mpa(__isl_take isl_printer *p,
3377 	struct isl_print_space_data *data, unsigned pos)
3378 {
3379 	int i;
3380 	int need_parens;
3381 	isl_space *space;
3382 	isl_multi_pw_aff *mpa = data->user;
3383 	isl_pw_aff *pa;
3384 
3385 	if (data->type != isl_dim_out) {
3386 		enum isl_dim_type type = data->type;
3387 
3388 		if (type == isl_dim_in)
3389 			type = isl_dim_set;
3390 		space = isl_multi_pw_aff_get_domain_space(mpa);
3391 		p = print_name(space, p, type, pos, data->latex);
3392 		isl_space_free(space);
3393 		return p;
3394 	}
3395 
3396 	pa = mpa->u.p[pos];
3397 	if (pa->n == 0)
3398 		return isl_printer_print_str(p, "(0 : false)");
3399 
3400 	need_parens = pa->n != 1 || !isl_set_plain_is_universe(pa->p[0].set);
3401 	if (need_parens)
3402 		p = isl_printer_print_str(p, "(");
3403 	space = isl_multi_pw_aff_get_domain_space(mpa);
3404 	for (i = 0; i < pa->n; ++i) {
3405 
3406 		if (i)
3407 			p = isl_printer_print_str(p, "; ");
3408 		p = print_aff_body(p, space, pa->p[i].aff);
3409 		p = print_disjuncts(pa->p[i].set, space, p, 0);
3410 	}
3411 	isl_space_free(space);
3412 	if (need_parens)
3413 		p = isl_printer_print_str(p, ")");
3414 
3415 	return p;
3416 }
3417 
3418 /* Print "mpa" to "p" in isl format.
3419  *
3420  * If "mpa" is zero-dimensional and has a non-trivial explicit domain,
3421  * then it is printed after the tuple of affine expressions.
3422  */
print_multi_pw_aff_isl(__isl_take isl_printer * p,__isl_keep isl_multi_pw_aff * mpa)3423 static __isl_give isl_printer *print_multi_pw_aff_isl(__isl_take isl_printer *p,
3424 	__isl_keep isl_multi_pw_aff *mpa)
3425 {
3426 	struct isl_print_space_data data = { 0 };
3427 	isl_bool has_domain;
3428 
3429 	if (!mpa)
3430 		return isl_printer_free(p);
3431 
3432 	p = print_param_tuple(p, mpa->space, &data);
3433 	p = isl_printer_print_str(p, "{ ");
3434 	data.print_dim = &print_dim_mpa;
3435 	data.user = mpa;
3436 	p = isl_print_space(mpa->space, p, 0, &data);
3437 	has_domain = isl_multi_pw_aff_has_non_trivial_domain(mpa);
3438 	if (has_domain < 0)
3439 		return isl_printer_free(p);
3440 	if (has_domain) {
3441 		isl_space *space;
3442 
3443 		space = isl_space_domain(isl_space_copy(mpa->space));
3444 		p = print_disjuncts_set(mpa->u.dom, space, p, 0);
3445 		isl_space_free(space);
3446 	}
3447 	p = isl_printer_print_str(p, " }");
3448 	return p;
3449 }
3450 
isl_printer_print_multi_pw_aff(__isl_take isl_printer * p,__isl_keep isl_multi_pw_aff * mpa)3451 __isl_give isl_printer *isl_printer_print_multi_pw_aff(
3452 	__isl_take isl_printer *p, __isl_keep isl_multi_pw_aff *mpa)
3453 {
3454 	if (!p || !mpa)
3455 		return isl_printer_free(p);
3456 
3457 	if (p->output_format == ISL_FORMAT_ISL)
3458 		return print_multi_pw_aff_isl(p, mpa);
3459 	isl_die(p->ctx, isl_error_unsupported, "unsupported output format",
3460 		return isl_printer_free(p));
3461 }
3462 
3463 /* Print dimension "pos" of data->space to "p".
3464  *
3465  * data->user is assumed to be an isl_multi_val.
3466  *
3467  * If the current dimension is an output dimension, then print
3468  * the corresponding value.  Otherwise, print the name of the dimension.
3469  */
print_dim_mv(__isl_take isl_printer * p,struct isl_print_space_data * data,unsigned pos)3470 static __isl_give isl_printer *print_dim_mv(__isl_take isl_printer *p,
3471 	struct isl_print_space_data *data, unsigned pos)
3472 {
3473 	isl_multi_val *mv = data->user;
3474 
3475 	if (data->type == isl_dim_out)
3476 		return isl_printer_print_val(p, mv->u.p[pos]);
3477 	else
3478 		return print_name(data->space, p, data->type, pos, data->latex);
3479 }
3480 
3481 /* Print the isl_multi_val "mv" to "p" in isl format.
3482  */
print_multi_val_isl(__isl_take isl_printer * p,__isl_keep isl_multi_val * mv)3483 static __isl_give isl_printer *print_multi_val_isl(__isl_take isl_printer *p,
3484 	__isl_keep isl_multi_val *mv)
3485 {
3486 	struct isl_print_space_data data = { 0 };
3487 
3488 	if (!mv)
3489 		return isl_printer_free(p);
3490 
3491 	p = print_param_tuple(p, mv->space, &data);
3492 	p = isl_printer_print_str(p, "{ ");
3493 	data.print_dim = &print_dim_mv;
3494 	data.user = mv;
3495 	p = isl_print_space(mv->space, p, 0, &data);
3496 	p = isl_printer_print_str(p, " }");
3497 	return p;
3498 }
3499 
3500 /* Print the isl_multi_val "mv" to "p".
3501  *
3502  * Currently only supported in isl format.
3503  */
isl_printer_print_multi_val(__isl_take isl_printer * p,__isl_keep isl_multi_val * mv)3504 __isl_give isl_printer *isl_printer_print_multi_val(
3505 	__isl_take isl_printer *p, __isl_keep isl_multi_val *mv)
3506 {
3507 	if (!p || !mv)
3508 		return isl_printer_free(p);
3509 
3510 	if (p->output_format == ISL_FORMAT_ISL)
3511 		return print_multi_val_isl(p, mv);
3512 	isl_die(p->ctx, isl_error_unsupported, "unsupported output format",
3513 		return isl_printer_free(p));
3514 }
3515 
3516 /* Print dimension "pos" of data->space to "p".
3517  *
3518  * data->user is assumed to be an isl_multi_id.
3519  *
3520  * If the current dimension is an output dimension, then print
3521  * the corresponding identifier.  Otherwise, print the name of the dimension.
3522  */
print_dim_mi(__isl_take isl_printer * p,struct isl_print_space_data * data,unsigned pos)3523 static __isl_give isl_printer *print_dim_mi(__isl_take isl_printer *p,
3524 	struct isl_print_space_data *data, unsigned pos)
3525 {
3526 	isl_multi_id *mi = data->user;
3527 
3528 	if (data->type == isl_dim_out)
3529 		return isl_printer_print_id(p, mi->u.p[pos]);
3530 	else
3531 		return print_name(data->space, p, data->type, pos, data->latex);
3532 }
3533 
3534 /* Print the isl_multi_id "mi" to "p" in isl format.
3535  */
print_multi_id_isl(__isl_take isl_printer * p,__isl_keep isl_multi_id * mi)3536 static __isl_give isl_printer *print_multi_id_isl(__isl_take isl_printer *p,
3537 	__isl_keep isl_multi_id *mi)
3538 {
3539 	isl_space *space;
3540 	struct isl_print_space_data data = { 0 };
3541 
3542 	space = isl_multi_id_peek_space(mi);
3543 	p = print_param_tuple(p, space, &data);
3544 	p = isl_printer_print_str(p, "{ ");
3545 	data.print_dim = &print_dim_mi;
3546 	data.user = mi;
3547 	p = isl_print_space(space, p, 0, &data);
3548 	p = isl_printer_print_str(p, " }");
3549 	return p;
3550 }
3551 
3552 /* Print the isl_multi_id "mi" to "p".
3553  *
3554  * Currently only supported in isl format.
3555  */
isl_printer_print_multi_id(__isl_take isl_printer * p,__isl_keep isl_multi_id * mi)3556 __isl_give isl_printer *isl_printer_print_multi_id(
3557 	__isl_take isl_printer *p, __isl_keep isl_multi_id *mi)
3558 {
3559 	if (!p || !mi)
3560 		return isl_printer_free(p);
3561 
3562 	if (p->output_format == ISL_FORMAT_ISL)
3563 		return print_multi_id_isl(p, mi);
3564 	isl_die(isl_printer_get_ctx(p), isl_error_unsupported,
3565 		"unsupported output format", return isl_printer_free(p));
3566 }
3567 
3568 /* Print dimension "pos" of data->space to "p".
3569  *
3570  * data->user is assumed to be an isl_multi_union_pw_aff.
3571  *
3572  * The current dimension is necessarily a set dimension, so
3573  * we print the corresponding isl_union_pw_aff, including
3574  * the braces.
3575  */
print_union_pw_aff_dim(__isl_take isl_printer * p,struct isl_print_space_data * data,unsigned pos)3576 static __isl_give isl_printer *print_union_pw_aff_dim(__isl_take isl_printer *p,
3577 	struct isl_print_space_data *data, unsigned pos)
3578 {
3579 	isl_multi_union_pw_aff *mupa = data->user;
3580 	isl_union_pw_aff *upa;
3581 
3582 	upa = isl_multi_union_pw_aff_get_union_pw_aff(mupa, pos);
3583 	p = print_union_pw_aff_body(p, upa);
3584 	isl_union_pw_aff_free(upa);
3585 
3586 	return p;
3587 }
3588 
3589 /* Print the isl_multi_union_pw_aff "mupa" to "p" in isl format.
3590  *
3591  * If "mupa" is zero-dimensional and has a non-trivial explicit domain,
3592  * then it is printed after the tuple of affine expressions.
3593  * In order to clarify that this domain belongs to the expression,
3594  * the tuple along with the domain are placed inside parentheses.
3595  * If "mupa" has any parameters, then the opening parenthesis
3596  * appears after the parameter declarations.
3597  */
print_multi_union_pw_aff_isl(__isl_take isl_printer * p,__isl_keep isl_multi_union_pw_aff * mupa)3598 static __isl_give isl_printer *print_multi_union_pw_aff_isl(
3599 	__isl_take isl_printer *p, __isl_keep isl_multi_union_pw_aff *mupa)
3600 {
3601 	struct isl_print_space_data data = { 0 };
3602 	isl_bool has_domain;
3603 	isl_space *space;
3604 
3605 	if (!mupa)
3606 		return isl_printer_free(p);
3607 	has_domain = isl_multi_union_pw_aff_has_non_trivial_domain(mupa);
3608 	if (has_domain < 0)
3609 		return isl_printer_free(p);
3610 
3611 	space = isl_multi_union_pw_aff_get_space(mupa);
3612 	p = print_param_tuple(p, space, &data);
3613 
3614 	if (has_domain)
3615 		p = isl_printer_print_str(p, "(");
3616 
3617 	data.print_dim = &print_union_pw_aff_dim;
3618 	data.user = mupa;
3619 
3620 	p = isl_print_space(space, p, 0, &data);
3621 	isl_space_free(space);
3622 
3623 	if (has_domain) {
3624 		p = isl_printer_print_str(p, " : ");
3625 		p = isl_printer_print_union_set_isl_body(p, mupa->u.dom);
3626 		p = isl_printer_print_str(p, ")");
3627 	}
3628 
3629 	return p;
3630 }
3631 
3632 /* Print the isl_multi_union_pw_aff "mupa" to "p" in isl format.
3633  *
3634  * We currently only support an isl format.
3635  */
isl_printer_print_multi_union_pw_aff(__isl_take isl_printer * p,__isl_keep isl_multi_union_pw_aff * mupa)3636 __isl_give isl_printer *isl_printer_print_multi_union_pw_aff(
3637 	__isl_take isl_printer *p, __isl_keep isl_multi_union_pw_aff *mupa)
3638 {
3639 	if (!p || !mupa)
3640 		return isl_printer_free(p);
3641 
3642 	if (p->output_format == ISL_FORMAT_ISL)
3643 		return print_multi_union_pw_aff_isl(p, mupa);
3644 	isl_die(isl_printer_get_ctx(p), isl_error_unsupported,
3645 		"unsupported output format", return isl_printer_free(p));
3646 }
3647