isl_dim_size: handle isl_dim_all
[isl.git] / isl_output.c
blob274582eaf2fe0c13775f7d82edb68562eb662dcc
1 /*
2 * Copyright 2008-2009 Katholieke Universiteit Leuven
3 * Copyright 2010 INRIA Saclay
5 * Use of this software is governed by the GNU LGPLv2.1 license
7 * Written by Sven Verdoolaege, K.U.Leuven, Departement
8 * Computerwetenschappen, Celestijnenlaan 200A, B-3001 Leuven, Belgium
9 * and INRIA Saclay - Ile-de-France, Parc Club Orsay Universite,
10 * ZAC des vignes, 4 rue Jacques Monod, 91893 Orsay, France
13 #include <stdlib.h>
14 #include <string.h>
15 #include <isl_ctx_private.h>
16 #include <isl_map_private.h>
17 #include <isl/set.h>
18 #include <isl/seq.h>
19 #include <isl_polynomial_private.h>
20 #include <isl_printer_private.h>
21 #include <isl_dim_private.h>
22 #include <isl_mat_private.h>
23 #include <isl/union_map.h>
24 #include <isl/constraint.h>
26 static const char *s_to[2] = { " -> ", " \\to " };
27 static const char *s_and[2] = { " and ", " \\wedge " };
28 static const char *s_or[2] = { " or ", " \\vee " };
29 static const char *s_le[2] = { "<=", "\\le" };
30 static const char *s_ge[2] = { ">=", "\\ge" };
31 static const char *s_open_set[2] = { "{ ", "\\{\\, " };
32 static const char *s_close_set[2] = { " }", " \\,\\}" };
33 static const char *s_open_list[2] = { "[", "(" };
34 static const char *s_close_list[2] = { "]", ")" };
35 static const char *s_such_that[2] = { " : ", " \\mid " };
36 static const char *s_open_exists[2] = { "exists (", "\\exists \\, " };
37 static const char *s_close_exists[2] = { ")", "" };
38 static const char *s_div_prefix[2] = { "e", "\\alpha_" };
39 static const char *s_param_prefix[2] = { "p", "p_" };
40 static const char *s_input_prefix[2] = { "i", "i_" };
41 static const char *s_output_prefix[2] = { "o", "o_" };
43 static __isl_give isl_printer *print_constraint_polylib(
44 struct isl_basic_map *bmap, int ineq, int n, __isl_take isl_printer *p)
46 int i;
47 unsigned n_in = isl_basic_map_dim(bmap, isl_dim_in);
48 unsigned n_out = isl_basic_map_dim(bmap, isl_dim_out);
49 unsigned nparam = isl_basic_map_dim(bmap, isl_dim_param);
50 isl_int *c = ineq ? bmap->ineq[n] : bmap->eq[n];
52 p = isl_printer_start_line(p);
53 p = isl_printer_print_int(p, ineq);
54 for (i = 0; i < n_out; ++i) {
55 p = isl_printer_print_str(p, " ");
56 p = isl_printer_print_isl_int(p, c[1+nparam+n_in+i]);
58 for (i = 0; i < n_in; ++i) {
59 p = isl_printer_print_str(p, " ");
60 p = isl_printer_print_isl_int(p, c[1+nparam+i]);
62 for (i = 0; i < bmap->n_div; ++i) {
63 p = isl_printer_print_str(p, " ");
64 p = isl_printer_print_isl_int(p, c[1+nparam+n_in+n_out+i]);
66 for (i = 0; i < nparam; ++i) {
67 p = isl_printer_print_str(p, " ");
68 p = isl_printer_print_isl_int(p, c[1+i]);
70 p = isl_printer_print_str(p, " ");
71 p = isl_printer_print_isl_int(p, c[0]);
72 p = isl_printer_end_line(p);
73 return p;
76 static __isl_give isl_printer *print_constraints_polylib(
77 struct isl_basic_map *bmap, __isl_take isl_printer *p)
79 int i;
81 p = isl_printer_set_isl_int_width(p, 5);
83 for (i = 0; i < bmap->n_eq; ++i)
84 p = print_constraint_polylib(bmap, 0, i, p);
85 for (i = 0; i < bmap->n_ineq; ++i)
86 p = print_constraint_polylib(bmap, 1, i, p);
88 return p;
91 static __isl_give isl_printer *bset_print_constraints_polylib(
92 struct isl_basic_set *bset, __isl_take isl_printer *p)
94 return print_constraints_polylib((struct isl_basic_map *)bset, p);
97 static __isl_give isl_printer *isl_basic_map_print_polylib(
98 __isl_keep isl_basic_map *bmap, __isl_take isl_printer *p, int ext)
100 unsigned total = isl_basic_map_total_dim(bmap);
101 p = isl_printer_start_line(p);
102 p = isl_printer_print_int(p, bmap->n_eq + bmap->n_ineq);
103 p = isl_printer_print_str(p, " ");
104 p = isl_printer_print_int(p, 1 + total + 1);
105 if (ext) {
106 p = isl_printer_print_str(p, " ");
107 p = isl_printer_print_int(p,
108 isl_basic_map_dim(bmap, isl_dim_out));
109 p = isl_printer_print_str(p, " ");
110 p = isl_printer_print_int(p,
111 isl_basic_map_dim(bmap, isl_dim_in));
112 p = isl_printer_print_str(p, " ");
113 p = isl_printer_print_int(p,
114 isl_basic_map_dim(bmap, isl_dim_div));
115 p = isl_printer_print_str(p, " ");
116 p = isl_printer_print_int(p,
117 isl_basic_map_dim(bmap, isl_dim_param));
119 p = isl_printer_end_line(p);
120 return print_constraints_polylib(bmap, p);
123 static __isl_give isl_printer *isl_basic_set_print_polylib(
124 __isl_keep isl_basic_set *bset, __isl_take isl_printer *p, int ext)
126 return isl_basic_map_print_polylib((struct isl_basic_map *)bset, p, ext);
129 static __isl_give isl_printer *isl_map_print_polylib(__isl_keep isl_map *map,
130 __isl_take isl_printer *p, int ext)
132 int i;
134 p = isl_printer_start_line(p);
135 p = isl_printer_print_int(p, map->n);
136 p = isl_printer_end_line(p);
137 for (i = 0; i < map->n; ++i) {
138 p = isl_printer_start_line(p);
139 p = isl_printer_end_line(p);
140 p = isl_basic_map_print_polylib(map->p[i], p, ext);
142 return p;
145 static __isl_give isl_printer *isl_set_print_polylib(__isl_keep isl_set *set,
146 __isl_take isl_printer *p, int ext)
148 return isl_map_print_polylib((struct isl_map *)set, p, ext);
151 static int count_same_name(__isl_keep isl_dim *dim,
152 enum isl_dim_type type, unsigned pos, const char *name)
154 enum isl_dim_type t;
155 unsigned p, s;
156 int count = 0;
158 for (t = isl_dim_param; t <= type && t <= isl_dim_out; ++t) {
159 s = t == type ? pos : isl_dim_size(dim, t);
160 for (p = 0; p < s; ++p) {
161 const char *n = isl_dim_get_name(dim, t, p);
162 if (n && !strcmp(n, name))
163 count++;
166 return count;
169 static __isl_give isl_printer *print_name(__isl_keep isl_dim *dim,
170 __isl_take isl_printer *p, enum isl_dim_type type, unsigned pos,
171 int set, int latex)
173 const char *name;
174 char buffer[20];
175 int primes;
177 name = type == isl_dim_div ? NULL : isl_dim_get_name(dim, type, pos);
179 if (!name) {
180 const char *prefix;
181 if (type == isl_dim_param)
182 prefix = s_param_prefix[latex];
183 else if (type == isl_dim_div)
184 prefix = s_div_prefix[latex];
185 else if (set || type == isl_dim_in)
186 prefix = s_input_prefix[latex];
187 else
188 prefix = s_output_prefix[latex];
189 snprintf(buffer, sizeof(buffer), "%s%d", prefix, pos);
190 name = buffer;
192 primes = count_same_name(dim, name == buffer ? isl_dim_div : type,
193 pos, name);
194 p = isl_printer_print_str(p, name);
195 while (primes-- > 0)
196 p = isl_printer_print_str(p, "'");
197 return p;
200 static enum isl_dim_type pos2type(__isl_keep isl_dim *dim, unsigned *pos)
202 enum isl_dim_type type;
203 unsigned n_in = isl_dim_size(dim, isl_dim_in);
204 unsigned n_out = isl_dim_size(dim, isl_dim_out);
205 unsigned nparam = isl_dim_size(dim, isl_dim_param);
207 if (*pos < 1 + nparam) {
208 type = isl_dim_param;
209 *pos -= 1;
210 } else if (*pos < 1 + nparam + n_in) {
211 type = isl_dim_in;
212 *pos -= 1 + nparam;
213 } else if (*pos < 1 + nparam + n_in + n_out) {
214 type = isl_dim_out;
215 *pos -= 1 + nparam + n_in;
216 } else {
217 type = isl_dim_div;
218 *pos -= 1 + nparam + n_in + n_out;
221 return type;
224 static __isl_give isl_printer *print_div(__isl_keep isl_dim *dim,
225 __isl_keep isl_mat *div, int pos, __isl_take isl_printer *p);
227 static __isl_give isl_printer *print_term(__isl_keep isl_dim *dim,
228 __isl_keep isl_mat *div,
229 isl_int c, unsigned pos, __isl_take isl_printer *p, int set, int latex)
231 enum isl_dim_type type;
232 int print_div_def;
234 if (pos == 0)
235 return isl_printer_print_isl_int(p, c);
237 type = pos2type(dim, &pos);
238 print_div_def = type == isl_dim_div && div &&
239 !isl_int_is_zero(div->row[pos][0]);
241 if (isl_int_is_one(c))
243 else if (isl_int_is_negone(c))
244 p = isl_printer_print_str(p, "-");
245 else {
246 p = isl_printer_print_isl_int(p, c);
247 if (p->output_format == ISL_FORMAT_C || print_div_def)
248 p = isl_printer_print_str(p, "*");
250 if (print_div_def)
251 p = print_div(dim, div, pos, p);
252 else
253 p = print_name(dim, p, type, pos, set, latex);
254 return p;
257 static __isl_give isl_printer *print_affine_of_len(__isl_keep isl_dim *dim,
258 __isl_keep isl_mat *div,
259 __isl_take isl_printer *p, isl_int *c, int len, int set)
261 int i;
262 int first;
264 for (i = 0, first = 1; i < len; ++i) {
265 int flip = 0;
266 if (isl_int_is_zero(c[i]))
267 continue;
268 if (!first) {
269 if (isl_int_is_neg(c[i])) {
270 flip = 1;
271 isl_int_neg(c[i], c[i]);
272 p = isl_printer_print_str(p, " - ");
273 } else
274 p = isl_printer_print_str(p, " + ");
276 first = 0;
277 p = print_term(dim, div, c[i], i, p, set, 0);
278 if (flip)
279 isl_int_neg(c[i], c[i]);
281 if (first)
282 p = isl_printer_print_str(p, "0");
283 return p;
286 static __isl_give isl_printer *print_affine(__isl_keep isl_basic_map *bmap,
287 __isl_keep isl_dim *dim, __isl_take isl_printer *p, isl_int *c, int set)
289 unsigned len = 1 + isl_basic_map_total_dim(bmap);
290 return print_affine_of_len(dim, NULL, p, c, len, set);
293 static int defining_equality(__isl_keep isl_basic_map *eq,
294 __isl_keep isl_dim *dim, enum isl_dim_type type, int pos)
296 int i;
297 unsigned total;
299 if (!eq)
300 return -1;
302 pos += isl_dim_offset(dim, type);
303 total = isl_basic_map_total_dim(eq);
305 for (i = 0; i < eq->n_eq; ++i) {
306 if (isl_seq_last_non_zero(eq->eq[i] + 1, total) != pos)
307 continue;
308 if (isl_int_is_one(eq->eq[i][1 + pos]))
309 isl_seq_neg(eq->eq[i], eq->eq[i], 1 + total);
310 return i;
313 return -1;
316 /* offset is the offset of local_dim inside global_type of global_dim.
318 static __isl_give isl_printer *print_nested_var_list(__isl_take isl_printer *p,
319 __isl_keep isl_dim *global_dim, enum isl_dim_type global_type,
320 __isl_keep isl_dim *local_dim, enum isl_dim_type local_type,
321 int set, int latex, __isl_keep isl_basic_map *eq, int offset)
323 int i, j;
325 if (global_dim != local_dim && local_type == isl_dim_out)
326 offset += local_dim->n_in;
328 for (i = 0; i < isl_dim_size(local_dim, local_type); ++i) {
329 if (i)
330 p = isl_printer_print_str(p, ", ");
331 j = defining_equality(eq, global_dim, global_type, offset + i);
332 if (j >= 0) {
333 int pos = 1 + isl_dim_offset(global_dim, global_type)
334 + offset + i;
335 p = print_affine_of_len(eq->dim, NULL,
336 p, eq->eq[j], pos, set);
337 } else {
338 p = print_name(global_dim, p, global_type, offset + i,
339 set, latex);
342 return p;
345 static __isl_give isl_printer *print_var_list(__isl_keep isl_dim *dim,
346 __isl_take isl_printer *p, enum isl_dim_type type,
347 int set, int latex, __isl_keep isl_basic_map *eq)
349 return print_nested_var_list(p, dim, type, dim, type, set, latex, eq, 0);
352 static __isl_give isl_printer *print_nested_map_dim(__isl_take isl_printer *p,
353 __isl_keep isl_dim *global_dim, enum isl_dim_type global_type,
354 __isl_keep isl_dim *local_dim,
355 int set, int latex, __isl_keep isl_basic_map *eq, int offset);
357 static __isl_give isl_printer *print_nested_tuple(__isl_take isl_printer *p,
358 __isl_keep isl_dim *global_dim, enum isl_dim_type global_type,
359 __isl_keep isl_dim *local_dim, enum isl_dim_type local_type,
360 int set, int latex, __isl_keep isl_basic_map *eq, int offset)
362 const char *name = NULL;
363 unsigned n = isl_dim_size(local_dim, local_type);
364 if ((local_type == isl_dim_in || local_type == isl_dim_out)) {
365 name = isl_dim_get_tuple_name(local_dim, local_type);
366 if (name) {
367 if (latex)
368 p = isl_printer_print_str(p, "\\mathrm{");
369 p = isl_printer_print_str(p, name);
370 if (latex)
371 p = isl_printer_print_str(p, "}");
374 if (!latex || n != 1 || name)
375 p = isl_printer_print_str(p, s_open_list[latex]);
376 if ((local_type == isl_dim_in || local_type == isl_dim_out) &&
377 local_dim->nested[local_type - isl_dim_in]) {
378 if (global_dim != local_dim && local_type == isl_dim_out)
379 offset += local_dim->n_in;
380 p = print_nested_map_dim(p, global_dim, global_type,
381 local_dim->nested[local_type - isl_dim_in],
382 set, latex, eq, offset);
383 } else
384 p = print_nested_var_list(p, global_dim, global_type,
385 local_dim, local_type, set, latex, eq, offset);
386 if (!latex || n != 1 || name)
387 p = isl_printer_print_str(p, s_close_list[latex]);
388 return p;
391 static __isl_give isl_printer *print_tuple(__isl_keep isl_dim *dim,
392 __isl_take isl_printer *p, enum isl_dim_type type,
393 int set, int latex, __isl_keep isl_basic_map *eq)
395 return print_nested_tuple(p, dim, type, dim, type, set, latex, eq, 0);
398 static __isl_give isl_printer *print_nested_map_dim(__isl_take isl_printer *p,
399 __isl_keep isl_dim *global_dim, enum isl_dim_type global_type,
400 __isl_keep isl_dim *local_dim,
401 int set, int latex, __isl_keep isl_basic_map *eq, int offset)
403 p = print_nested_tuple(p, global_dim, global_type,
404 local_dim, isl_dim_in, set, latex, eq, offset);
405 p = isl_printer_print_str(p, s_to[latex]);
406 p = print_nested_tuple(p, global_dim, global_type,
407 local_dim, isl_dim_out, set, latex, eq, offset);
409 return p;
412 static __isl_give isl_printer *print_dim(__isl_keep isl_dim *dim,
413 __isl_take isl_printer *p, int set, int latex, int rational,
414 __isl_keep isl_basic_map *eq)
416 if (rational && !latex)
417 p = isl_printer_print_str(p, "rat: ");
418 if (set)
419 p = print_tuple(dim, p, isl_dim_set, 1, latex, eq);
420 else {
421 p = print_tuple(dim, p, isl_dim_in, 0, latex, eq);
422 p = isl_printer_print_str(p, s_to[latex]);
423 p = print_tuple(dim, p, isl_dim_out, 0, latex, eq);
426 return p;
429 static __isl_give isl_printer *print_omega_parameters(__isl_keep isl_dim *dim,
430 __isl_take isl_printer *p)
432 if (isl_dim_size(dim, isl_dim_param) == 0)
433 return p;
435 p = isl_printer_start_line(p);
436 p = isl_printer_print_str(p, "symbolic ");
437 p = print_var_list(dim, p, isl_dim_param, 0, 0, NULL);
438 p = isl_printer_print_str(p, ";");
439 p = isl_printer_end_line(p);
440 return p;
443 static __isl_give isl_printer *print_constraint(struct isl_basic_map *bmap,
444 __isl_keep isl_dim *dim, __isl_take isl_printer *p,
445 isl_int *c, int last, const char *op, int first_constraint, int set,
446 int latex)
448 if (!first_constraint)
449 p = isl_printer_print_str(p, s_and[latex]);
451 isl_int_abs(c[last], c[last]);
453 p = print_term(dim, NULL, c[last], last, p, set, latex);
455 p = isl_printer_print_str(p, " ");
456 p = isl_printer_print_str(p, op);
457 p = isl_printer_print_str(p, " ");
459 isl_int_set_si(c[last], 0);
460 p = print_affine(bmap, dim, p, c, set);
462 return p;
465 static __isl_give isl_printer *print_constraints(__isl_keep isl_basic_map *bmap,
466 __isl_keep isl_dim *dim, __isl_take isl_printer *p, int set, int latex)
468 int i;
469 struct isl_vec *c;
470 unsigned total = isl_basic_map_total_dim(bmap);
472 c = isl_vec_alloc(bmap->ctx, 1 + total);
473 if (!c)
474 goto error;
476 for (i = bmap->n_eq - 1; i >= 0; --i) {
477 int l = isl_seq_last_non_zero(bmap->eq[i], 1 + total);
478 if (l < 0) {
479 if (i != bmap->n_eq - 1)
480 p = isl_printer_print_str(p, s_and[latex]);
481 p = isl_printer_print_str(p, "0 = 0");
482 continue;
484 if (isl_int_is_neg(bmap->eq[i][l]))
485 isl_seq_cpy(c->el, bmap->eq[i], 1 + total);
486 else
487 isl_seq_neg(c->el, bmap->eq[i], 1 + total);
488 p = print_constraint(bmap, dim, p, c->el, l,
489 "=", i == bmap->n_eq - 1, set, latex);
491 for (i = 0; i < bmap->n_ineq; ++i) {
492 int l = isl_seq_last_non_zero(bmap->ineq[i], 1 + total);
493 int s;
494 const char *op;
495 if (l < 0)
496 continue;
497 s = isl_int_sgn(bmap->ineq[i][l]);
498 if (s < 0)
499 isl_seq_cpy(c->el, bmap->ineq[i], 1 + total);
500 else
501 isl_seq_neg(c->el, bmap->ineq[i], 1 + total);
502 op = s < 0 ? s_le[latex] : s_ge[latex];
503 p = print_constraint(bmap, dim, p, c->el, l,
504 op, !bmap->n_eq && !i, set, latex);
507 isl_vec_free(c);
509 return p;
510 error:
511 isl_vec_free(c);
512 isl_printer_free(p);
513 return NULL;
516 static __isl_give isl_printer *print_omega_constraints(
517 __isl_keep isl_basic_map *bmap, __isl_take isl_printer *p, int set)
519 if (bmap->n_eq + bmap->n_ineq == 0)
520 return p;
522 p = isl_printer_print_str(p, ": ");
523 if (bmap->n_div > 0) {
524 int i;
525 p = isl_printer_print_str(p, "exists (");
526 for (i = 0; i < bmap->n_div; ++i) {
527 if (i)
528 p = isl_printer_print_str(p, ", ");
529 p = print_name(bmap->dim, p, isl_dim_div, i, 0, 0);
531 p = isl_printer_print_str(p, ": ");
533 p = print_constraints(bmap, bmap->dim, p, set, 0);
534 if (bmap->n_div > 0)
535 p = isl_printer_print_str(p, ")");
536 return p;
539 static __isl_give isl_printer *basic_map_print_omega(
540 __isl_keep isl_basic_map *bmap, __isl_take isl_printer *p)
542 p = isl_printer_print_str(p, "{ [");
543 p = print_var_list(bmap->dim, p, isl_dim_in, 0, 0, NULL);
544 p = isl_printer_print_str(p, "] -> [");
545 p = print_var_list(bmap->dim, p, isl_dim_out, 0, 0, NULL);
546 p = isl_printer_print_str(p, "] ");
547 p = print_omega_constraints(bmap, p, 0);
548 p = isl_printer_print_str(p, " }");
549 return p;
552 static __isl_give isl_printer *isl_basic_map_print_omega(
553 __isl_keep isl_basic_map *bmap, __isl_take isl_printer *p)
555 p = print_omega_parameters(bmap->dim, p);
557 p = isl_printer_start_line(p);
558 p = basic_map_print_omega(bmap, p);
559 p = isl_printer_end_line(p);
560 return p;
563 static __isl_give isl_printer *basic_set_print_omega(
564 __isl_keep isl_basic_set *bset, __isl_take isl_printer *p)
566 p = isl_printer_print_str(p, "{ [");
567 p = print_var_list(bset->dim, p, isl_dim_set, 1, 0, NULL);
568 p = isl_printer_print_str(p, "] ");
569 p = print_omega_constraints((isl_basic_map *)bset, p, 1);
570 p = isl_printer_print_str(p, " }");
571 return p;
574 static __isl_give isl_printer *isl_basic_set_print_omega(
575 __isl_keep isl_basic_set *bset, __isl_take isl_printer *p)
577 p = print_omega_parameters(bset->dim, p);
579 p = isl_printer_start_line(p);
580 p = basic_set_print_omega(bset, p);
581 p = isl_printer_end_line(p);
582 return p;
585 static __isl_give isl_printer *isl_map_print_omega(__isl_keep isl_map *map,
586 __isl_take isl_printer *p)
588 int i;
590 p = print_omega_parameters(map->dim, p);
592 p = isl_printer_start_line(p);
593 for (i = 0; i < map->n; ++i) {
594 if (i)
595 p = isl_printer_print_str(p, " union ");
596 p = basic_map_print_omega(map->p[i], p);
598 p = isl_printer_end_line(p);
599 return p;
602 static __isl_give isl_printer *isl_set_print_omega(__isl_keep isl_set *set,
603 __isl_take isl_printer *p)
605 int i;
607 p = print_omega_parameters(set->dim, p);
609 p = isl_printer_start_line(p);
610 for (i = 0; i < set->n; ++i) {
611 if (i)
612 p = isl_printer_print_str(p, " union ");
613 p = basic_set_print_omega(set->p[i], p);
615 p = isl_printer_end_line(p);
616 return p;
619 static __isl_give isl_printer *print_disjunct(__isl_keep isl_basic_map *bmap,
620 __isl_keep isl_dim *dim, __isl_take isl_printer *p, int set, int latex)
622 if (bmap->n_div > 0) {
623 int i;
624 p = isl_printer_print_str(p, s_open_exists[latex]);
625 for (i = 0; i < bmap->n_div; ++i) {
626 if (i)
627 p = isl_printer_print_str(p, ", ");
628 p = print_name(dim, p, isl_dim_div, i, 0, latex);
629 if (latex || isl_int_is_zero(bmap->div[i][0]))
630 continue;
631 p = isl_printer_print_str(p, " = [(");
632 p = print_affine(bmap, dim, p, bmap->div[i] + 1, set);
633 p = isl_printer_print_str(p, ")/");
634 p = isl_printer_print_isl_int(p, bmap->div[i][0]);
635 p = isl_printer_print_str(p, "]");
637 p = isl_printer_print_str(p, ": ");
640 p = print_constraints(bmap, dim, p, set, latex);
642 if (bmap->n_div > 0)
643 p = isl_printer_print_str(p, s_close_exists[latex]);
644 return p;
647 static __isl_give isl_printer *isl_basic_map_print_isl(
648 __isl_keep isl_basic_map *bmap, __isl_take isl_printer *p,
649 int set, int latex)
651 int rational = ISL_F_ISSET(bmap, ISL_BASIC_MAP_RATIONAL);
652 if (isl_basic_map_dim(bmap, isl_dim_param) > 0) {
653 p = print_tuple(bmap->dim, p, isl_dim_param, 0, latex, NULL);
654 p = isl_printer_print_str(p, " -> ");
656 p = isl_printer_print_str(p, "{ ");
657 p = print_dim(bmap->dim, p, set, latex, rational, NULL);
658 p = isl_printer_print_str(p, " : ");
659 p = print_disjunct(bmap, bmap->dim, p, set, latex);
660 p = isl_printer_print_str(p, " }");
661 return p;
664 static __isl_give isl_printer *print_disjuncts(__isl_keep isl_map *map,
665 __isl_take isl_printer *p, int set, int latex)
667 int i;
669 if (isl_map_plain_is_universe(map))
670 return p;
672 p = isl_printer_print_str(p, s_such_that[latex]);
673 if (map->n == 0)
674 p = isl_printer_print_str(p, "1 = 0");
675 for (i = 0; i < map->n; ++i) {
676 if (i)
677 p = isl_printer_print_str(p, s_or[latex]);
678 if (map->n > 1 && map->p[i]->n_eq + map->p[i]->n_ineq > 1)
679 p = isl_printer_print_str(p, "(");
680 p = print_disjunct(map->p[i], map->dim, p, set, latex);
681 if (map->n > 1 && map->p[i]->n_eq + map->p[i]->n_ineq > 1)
682 p = isl_printer_print_str(p, ")");
684 return p;
687 struct isl_aff_split {
688 isl_basic_map *aff;
689 isl_map *map;
692 static void free_split(__isl_take struct isl_aff_split *split, int n)
694 int i;
696 if (!split)
697 return;
699 for (i = 0; i < n; ++i) {
700 isl_basic_map_free(split[i].aff);
701 isl_map_free(split[i].map);
704 free(split);
707 static __isl_give isl_basic_map *get_aff(__isl_take isl_basic_map *bmap)
709 int i, j;
710 unsigned nparam, n_in, n_out, total;
712 bmap = isl_basic_map_cow(bmap);
713 if (!bmap)
714 return NULL;
715 if (isl_basic_map_free_inequality(bmap, bmap->n_ineq) < 0)
716 goto error;
718 nparam = isl_basic_map_dim(bmap, isl_dim_param);
719 n_in = isl_basic_map_dim(bmap, isl_dim_in);
720 n_out = isl_basic_map_dim(bmap, isl_dim_out);
721 total = isl_basic_map_dim(bmap, isl_dim_all);
722 for (i = bmap->n_eq - 1; i >= 0; --i) {
723 j = isl_seq_last_non_zero(bmap->eq[i] + 1, total);
724 if (j >= nparam && j < nparam + n_in + n_out &&
725 (isl_int_is_one(bmap->eq[i][1 + j]) ||
726 isl_int_is_negone(bmap->eq[i][1 + j])))
727 continue;
728 if (isl_basic_map_drop_equality(bmap, i) < 0)
729 goto error;
732 bmap = isl_basic_map_finalize(bmap);
734 return bmap;
735 error:
736 isl_basic_map_free(bmap);
737 return NULL;
740 static int aff_split_cmp(const void *p1, const void *p2)
742 const struct isl_aff_split *s1, *s2;
743 s1 = (const struct isl_aff_split *) p1;
744 s2 = (const struct isl_aff_split *) p2;
746 return isl_basic_map_plain_cmp(s1->aff, s2->aff);
749 static __isl_give isl_basic_map *drop_aff(__isl_take isl_basic_map *bmap,
750 __isl_keep isl_basic_map *aff)
752 int i, j;
753 unsigned total;
755 if (!bmap || !aff)
756 goto error;
758 total = isl_dim_total(bmap->dim);
760 for (i = bmap->n_eq - 1; i >= 0; --i) {
761 if (isl_seq_first_non_zero(bmap->eq[i] + 1 + total,
762 bmap->n_div) != -1)
763 continue;
764 for (j = 0; j < aff->n_eq; ++j) {
765 if (!isl_seq_eq(bmap->eq[i], aff->eq[j], 1 + total) &&
766 !isl_seq_is_neg(bmap->eq[i], aff->eq[j], 1 + total))
767 continue;
768 if (isl_basic_map_drop_equality(bmap, i) < 0)
769 goto error;
770 break;
774 return bmap;
775 error:
776 isl_basic_map_free(bmap);
777 return NULL;
780 static __isl_give struct isl_aff_split *split_aff(__isl_keep isl_map *map)
782 int i, n;
783 struct isl_aff_split *split;
785 split = isl_calloc_array(map->isl, struct isl_aff_split, map->n);
786 if (!split)
787 return NULL;
789 for (i = 0; i < map->n; ++i) {
790 isl_basic_map *bmap;
791 split[i].aff = get_aff(isl_basic_map_copy(map->p[i]));
792 bmap = isl_basic_map_copy(map->p[i]);
793 bmap = isl_basic_map_cow(bmap);
794 bmap = drop_aff(bmap, split[i].aff);
795 split[i].map = isl_map_from_basic_map(bmap);
796 if (!split[i].aff || !split[i].map)
797 goto error;
800 qsort(split, map->n, sizeof(struct isl_aff_split), &aff_split_cmp);
802 n = map->n;
803 for (i = n - 1; i >= 1; --i) {
804 if (!isl_basic_map_plain_is_equal(split[i - 1].aff,
805 split[i].aff))
806 continue;
807 isl_basic_map_free(split[i].aff);
808 split[i - 1].map = isl_map_union(split[i - 1].map,
809 split[i].map);
810 if (i != n - 1)
811 split[i] = split[n - 1];
812 split[n - 1].aff = NULL;
813 split[n - 1].map = NULL;
814 --n;
817 return split;
818 error:
819 free_split(split, map->n);
820 return NULL;
823 static __isl_give isl_printer *print_split_map(__isl_take isl_printer *p,
824 struct isl_aff_split *split, int n, int set)
826 int i;
827 int rational;
829 for (i = 0; i < n; ++i) {
830 isl_dim *dim;
832 if (!split[i].map)
833 break;
834 dim = split[i].map->dim;
835 rational = split[i].map->n > 0 &&
836 ISL_F_ISSET(split[i].map->p[0], ISL_BASIC_MAP_RATIONAL);
837 if (i)
838 p = isl_printer_print_str(p, "; ");
839 p = print_dim(dim, p, set, 0, rational, split[i].aff);
840 p = print_disjuncts(split[i].map, p, set, 0);
843 return p;
846 static __isl_give isl_printer *isl_map_print_isl_body(__isl_keep isl_map *map,
847 __isl_take isl_printer *p, int set)
849 struct isl_aff_split *split = NULL;
850 int rational;
852 if (map->n > 0)
853 split = split_aff(map);
854 if (split) {
855 p = print_split_map(p, split, map->n, set);
856 } else {
857 rational = map->n > 0 &&
858 ISL_F_ISSET(map->p[0], ISL_BASIC_MAP_RATIONAL);
859 p = print_dim(map->dim, p, set, 0, rational, NULL);
860 p = print_disjuncts(map, p, set, 0);
862 free_split(split, map->n);
863 return p;
866 static __isl_give isl_printer *isl_map_print_isl(__isl_keep isl_map *map,
867 __isl_take isl_printer *p, int set)
869 if (isl_map_dim(map, isl_dim_param) > 0) {
870 p = print_tuple(map->dim, p, isl_dim_param, set, 0, NULL);
871 p = isl_printer_print_str(p, s_to[0]);
873 p = isl_printer_print_str(p, s_open_set[0]);
874 p = isl_map_print_isl_body(map, p, set);
875 p = isl_printer_print_str(p, s_close_set[0]);
876 return p;
879 static __isl_give isl_printer *print_latex_map(__isl_keep isl_map *map,
880 __isl_take isl_printer *p, __isl_keep isl_basic_map *aff, int set)
882 if (isl_map_dim(map, isl_dim_param) > 0) {
883 p = print_tuple(map->dim, p, isl_dim_param, set, 1, NULL);
884 p = isl_printer_print_str(p, s_to[1]);
886 p = isl_printer_print_str(p, s_open_set[1]);
887 p = print_dim(map->dim, p, set, 1, 0, aff);
888 p = print_disjuncts(map, p, set, 1);
889 p = isl_printer_print_str(p, s_close_set[1]);
891 return p;
894 static __isl_give isl_printer *isl_map_print_latex(__isl_keep isl_map *map,
895 __isl_take isl_printer *p, int set)
897 int i;
898 struct isl_aff_split *split = NULL;
900 if (map->n > 0)
901 split = split_aff(map);
903 if (!split)
904 return print_latex_map(map, p, NULL, set);
906 for (i = 0; i < map->n; ++i) {
907 if (!split[i].map)
908 break;
909 if (i)
910 p = isl_printer_print_str(p, " \\cup ");
911 p = print_latex_map(split[i].map, p, split[i].aff, set);
914 free_split(split, map->n);
915 return p;
918 __isl_give isl_printer *isl_printer_print_basic_map(__isl_take isl_printer *p,
919 __isl_keep isl_basic_map *bmap)
921 if (!p || !bmap)
922 goto error;
923 if (p->output_format == ISL_FORMAT_ISL)
924 return isl_basic_map_print_isl(bmap, p, 0, 0);
925 else if (p->output_format == ISL_FORMAT_OMEGA)
926 return isl_basic_map_print_omega(bmap, p);
927 isl_assert(bmap->ctx, 0, goto error);
928 error:
929 isl_printer_free(p);
930 return NULL;
933 void isl_basic_map_print(__isl_keep isl_basic_map *bmap, FILE *out, int indent,
934 const char *prefix, const char *suffix, unsigned output_format)
936 isl_printer *printer;
938 if (!bmap)
939 return;
941 printer = isl_printer_to_file(bmap->ctx, out);
942 printer = isl_printer_set_indent(printer, indent);
943 printer = isl_printer_set_prefix(printer, prefix);
944 printer = isl_printer_set_suffix(printer, suffix);
945 printer = isl_printer_set_output_format(printer, output_format);
946 isl_printer_print_basic_map(printer, bmap);
948 isl_printer_free(printer);
951 __isl_give isl_printer *isl_printer_print_basic_set(__isl_take isl_printer *p,
952 __isl_keep isl_basic_set *bset)
954 if (!p || !bset)
955 goto error;
957 if (p->output_format == ISL_FORMAT_ISL)
958 return isl_basic_map_print_isl(bset, p, 1, 0);
959 else if (p->output_format == ISL_FORMAT_POLYLIB)
960 return isl_basic_set_print_polylib(bset, p, 0);
961 else if (p->output_format == ISL_FORMAT_EXT_POLYLIB)
962 return isl_basic_set_print_polylib(bset, p, 1);
963 else if (p->output_format == ISL_FORMAT_POLYLIB_CONSTRAINTS)
964 return bset_print_constraints_polylib(bset, p);
965 else if (p->output_format == ISL_FORMAT_OMEGA)
966 return isl_basic_set_print_omega(bset, p);
967 isl_assert(p->ctx, 0, goto error);
968 error:
969 isl_printer_free(p);
970 return NULL;
973 void isl_basic_set_print(struct isl_basic_set *bset, FILE *out, int indent,
974 const char *prefix, const char *suffix, unsigned output_format)
976 isl_printer *printer;
978 if (!bset)
979 return;
981 printer = isl_printer_to_file(bset->ctx, out);
982 printer = isl_printer_set_indent(printer, indent);
983 printer = isl_printer_set_prefix(printer, prefix);
984 printer = isl_printer_set_suffix(printer, suffix);
985 printer = isl_printer_set_output_format(printer, output_format);
986 isl_printer_print_basic_set(printer, bset);
988 isl_printer_free(printer);
991 __isl_give isl_printer *isl_printer_print_set(__isl_take isl_printer *p,
992 __isl_keep isl_set *set)
994 if (!p || !set)
995 goto error;
996 if (p->output_format == ISL_FORMAT_ISL)
997 return isl_map_print_isl((isl_map *)set, p, 1);
998 else if (p->output_format == ISL_FORMAT_POLYLIB)
999 return isl_set_print_polylib(set, p, 0);
1000 else if (p->output_format == ISL_FORMAT_EXT_POLYLIB)
1001 return isl_set_print_polylib(set, p, 1);
1002 else if (p->output_format == ISL_FORMAT_OMEGA)
1003 return isl_set_print_omega(set, p);
1004 else if (p->output_format == ISL_FORMAT_LATEX)
1005 return isl_map_print_latex((isl_map *)set, p, 1);
1006 isl_assert(set->ctx, 0, goto error);
1007 error:
1008 isl_printer_free(p);
1009 return NULL;
1012 void isl_set_print(struct isl_set *set, FILE *out, int indent,
1013 unsigned output_format)
1015 isl_printer *printer;
1017 if (!set)
1018 return;
1020 printer = isl_printer_to_file(set->ctx, out);
1021 printer = isl_printer_set_indent(printer, indent);
1022 printer = isl_printer_set_output_format(printer, output_format);
1023 printer = isl_printer_print_set(printer, set);
1025 isl_printer_free(printer);
1028 __isl_give isl_printer *isl_printer_print_map(__isl_take isl_printer *p,
1029 __isl_keep isl_map *map)
1031 if (!p || !map)
1032 goto error;
1034 if (p->output_format == ISL_FORMAT_ISL)
1035 return isl_map_print_isl(map, p, 0);
1036 else if (p->output_format == ISL_FORMAT_POLYLIB)
1037 return isl_map_print_polylib(map, p, 0);
1038 else if (p->output_format == ISL_FORMAT_EXT_POLYLIB)
1039 return isl_map_print_polylib(map, p, 1);
1040 else if (p->output_format == ISL_FORMAT_OMEGA)
1041 return isl_map_print_omega(map, p);
1042 else if (p->output_format == ISL_FORMAT_LATEX)
1043 return isl_map_print_latex(map, p, 0);
1044 isl_assert(map->ctx, 0, goto error);
1045 error:
1046 isl_printer_free(p);
1047 return NULL;
1050 struct isl_union_print_data {
1051 isl_printer *p;
1052 int set;
1053 int first;
1056 static int print_map_body(__isl_take isl_map *map, void *user)
1058 struct isl_union_print_data *data;
1059 data = (struct isl_union_print_data *)user;
1061 if (!data->first)
1062 data->p = isl_printer_print_str(data->p, "; ");
1063 data->first = 0;
1065 data->p = isl_map_print_isl_body(map, data->p, data->set);
1066 isl_map_free(map);
1068 return 0;
1071 static __isl_give isl_printer *isl_union_map_print_isl(
1072 __isl_keep isl_union_map *umap, __isl_take isl_printer *p, int set)
1074 struct isl_union_print_data data = { p, set, 1 };
1075 isl_dim *dim;
1076 dim = isl_union_map_get_dim(umap);
1077 if (isl_dim_size(dim, isl_dim_param) > 0) {
1078 p = print_tuple(dim, p, isl_dim_param, set, 0, NULL);
1079 p = isl_printer_print_str(p, s_to[0]);
1081 isl_dim_free(dim);
1082 p = isl_printer_print_str(p, s_open_set[0]);
1083 isl_union_map_foreach_map(umap, &print_map_body, &data);
1084 p = data.p;
1085 p = isl_printer_print_str(p, s_close_set[0]);
1086 return p;
1089 static int print_latex_map_body(__isl_take isl_map *map, void *user)
1091 struct isl_union_print_data *data;
1092 data = (struct isl_union_print_data *)user;
1094 if (!data->first)
1095 data->p = isl_printer_print_str(data->p, " \\cup ");
1096 data->first = 0;
1098 data->p = isl_map_print_latex(map, data->p, data->set);
1099 isl_map_free(map);
1101 return 0;
1104 static __isl_give isl_printer *isl_union_map_print_latex(
1105 __isl_keep isl_union_map *umap, __isl_take isl_printer *p, int set)
1107 struct isl_union_print_data data = { p, set, 1 };
1108 isl_union_map_foreach_map(umap, &print_latex_map_body, &data);
1109 p = data.p;
1110 return p;
1113 __isl_give isl_printer *isl_printer_print_union_map(__isl_take isl_printer *p,
1114 __isl_keep isl_union_map *umap)
1116 if (!p || !umap)
1117 goto error;
1119 if (p->output_format == ISL_FORMAT_ISL)
1120 return isl_union_map_print_isl(umap, p, 0);
1121 if (p->output_format == ISL_FORMAT_LATEX)
1122 return isl_union_map_print_latex(umap, p, 0);
1124 isl_die(p->ctx, isl_error_invalid,
1125 "invalid output format for isl_union_map", goto error);
1126 error:
1127 isl_printer_free(p);
1128 return NULL;
1131 __isl_give isl_printer *isl_printer_print_union_set(__isl_take isl_printer *p,
1132 __isl_keep isl_union_set *uset)
1134 if (!p || !uset)
1135 goto error;
1137 if (p->output_format == ISL_FORMAT_ISL)
1138 return isl_union_map_print_isl((isl_union_map *)uset, p, 1);
1139 if (p->output_format == ISL_FORMAT_LATEX)
1140 return isl_union_map_print_latex((isl_union_map *)uset, p, 1);
1142 isl_die(p->ctx, isl_error_invalid,
1143 "invalid output format for isl_union_set", goto error);
1144 error:
1145 isl_printer_free(p);
1146 return NULL;
1149 void isl_map_print(__isl_keep isl_map *map, FILE *out, int indent,
1150 unsigned output_format)
1152 isl_printer *printer;
1154 if (!map)
1155 return;
1157 printer = isl_printer_to_file(map->ctx, out);
1158 printer = isl_printer_set_indent(printer, indent);
1159 printer = isl_printer_set_output_format(printer, output_format);
1160 printer = isl_printer_print_map(printer, map);
1162 isl_printer_free(printer);
1165 void isl_basic_map_dump(__isl_keep isl_basic_map *bmap)
1167 isl_printer *printer;
1169 if (!bmap)
1170 return;
1172 printer = isl_printer_to_file(isl_basic_map_get_ctx(bmap), stderr);
1173 printer = isl_printer_print_basic_map(printer, bmap);
1174 printer = isl_printer_end_line(printer);
1176 isl_printer_free(printer);
1179 void isl_basic_set_dump(__isl_keep isl_basic_set *bset)
1181 isl_printer *printer;
1183 if (!bset)
1184 return;
1186 printer = isl_printer_to_file(isl_basic_set_get_ctx(bset), stderr);
1187 printer = isl_printer_print_basic_set(printer, bset);
1188 printer = isl_printer_end_line(printer);
1190 isl_printer_free(printer);
1193 void isl_map_dump(__isl_keep isl_map *map)
1195 isl_printer *printer;
1197 if (!map)
1198 return;
1200 printer = isl_printer_to_file(isl_map_get_ctx(map), stderr);
1201 printer = isl_printer_print_map(printer, map);
1202 printer = isl_printer_end_line(printer);
1204 isl_printer_free(printer);
1207 void isl_set_dump(__isl_keep isl_set *set)
1209 isl_printer *printer;
1211 if (!set)
1212 return;
1214 printer = isl_printer_to_file(isl_set_get_ctx(set), stderr);
1215 printer = isl_printer_print_set(printer, set);
1216 printer = isl_printer_end_line(printer);
1218 isl_printer_free(printer);
1221 void isl_union_map_dump(__isl_keep isl_union_map *umap)
1223 isl_printer *printer;
1225 if (!umap)
1226 return;
1228 printer = isl_printer_to_file(isl_union_map_get_ctx(umap), stderr);
1229 printer = isl_printer_print_union_map(printer, umap);
1230 printer = isl_printer_end_line(printer);
1232 isl_printer_free(printer);
1235 void isl_union_set_dump(__isl_keep isl_union_set *uset)
1237 isl_printer *printer;
1239 if (!uset)
1240 return;
1242 printer = isl_printer_to_file(isl_union_set_get_ctx(uset), stderr);
1243 printer = isl_printer_print_union_set(printer, uset);
1244 printer = isl_printer_end_line(printer);
1246 isl_printer_free(printer);
1249 void isl_union_pw_qpolynomial_dump(__isl_keep isl_union_pw_qpolynomial *upwqp)
1251 isl_printer *printer;
1253 if (!upwqp)
1254 return;
1256 printer = isl_printer_to_file(isl_union_pw_qpolynomial_get_ctx(upwqp),
1257 stderr);
1258 printer = isl_printer_print_union_pw_qpolynomial(printer, upwqp);
1259 printer = isl_printer_end_line(printer);
1261 isl_printer_free(printer);
1264 void isl_qpolynomial_dump(__isl_keep isl_qpolynomial *qp)
1266 isl_printer *printer;
1268 if (!qp)
1269 return;
1271 printer = isl_printer_to_file(isl_qpolynomial_get_ctx(qp), stderr);
1272 printer = isl_printer_print_qpolynomial(printer, qp);
1273 printer = isl_printer_end_line(printer);
1275 isl_printer_free(printer);
1278 void isl_qpolynomial_fold_dump(__isl_keep isl_qpolynomial_fold *fold)
1280 isl_printer *printer;
1282 if (!fold)
1283 return;
1285 printer = isl_printer_to_file(isl_qpolynomial_fold_get_ctx(fold),
1286 stderr);
1287 printer = isl_printer_print_qpolynomial_fold(printer, fold);
1288 printer = isl_printer_end_line(printer);
1290 isl_printer_free(printer);
1293 void isl_pw_qpolynomial_dump(__isl_keep isl_pw_qpolynomial *pwqp)
1295 isl_printer *printer;
1297 if (!pwqp)
1298 return;
1300 printer = isl_printer_to_file(isl_pw_qpolynomial_get_ctx(pwqp), stderr);
1301 printer = isl_printer_print_pw_qpolynomial(printer, pwqp);
1302 printer = isl_printer_end_line(printer);
1304 isl_printer_free(printer);
1307 void isl_pw_qpolynomial_fold_dump(__isl_keep isl_pw_qpolynomial_fold *pwf)
1309 isl_printer *printer;
1311 if (!pwf)
1312 return;
1314 printer = isl_printer_to_file(
1315 isl_pw_qpolynomial_fold_get_ctx(pwf), stderr);
1316 printer = isl_printer_print_pw_qpolynomial_fold(printer, pwf);
1317 printer = isl_printer_end_line(printer);
1319 isl_printer_free(printer);
1322 void isl_union_pw_qpolynomial_fold_dump(
1323 __isl_keep isl_union_pw_qpolynomial_fold *upwf)
1325 isl_printer *printer;
1327 if (!upwf)
1328 return;
1330 printer = isl_printer_to_file(
1331 isl_union_pw_qpolynomial_fold_get_ctx(upwf), stderr);
1332 printer = isl_printer_print_union_pw_qpolynomial_fold(printer, upwf);
1333 printer = isl_printer_end_line(printer);
1335 isl_printer_free(printer);
1338 static int upoly_rec_n_non_zero(__isl_keep struct isl_upoly_rec *rec)
1340 int i;
1341 int n;
1343 for (i = 0, n = 0; i < rec->n; ++i)
1344 if (!isl_upoly_is_zero(rec->p[i]))
1345 ++n;
1347 return n;
1350 static __isl_give isl_printer *print_div(__isl_keep isl_dim *dim,
1351 __isl_keep isl_mat *div, int pos, __isl_take isl_printer *p)
1353 int c = p->output_format == ISL_FORMAT_C;
1354 p = isl_printer_print_str(p, c ? "floord(" : "[(");
1355 p = print_affine_of_len(dim, div, p,
1356 div->row[pos] + 1, div->n_col - 1, 1);
1357 p = isl_printer_print_str(p, c ? ", " : ")/");
1358 p = isl_printer_print_isl_int(p, div->row[pos][0]);
1359 p = isl_printer_print_str(p, c ? ")" : "]");
1360 return p;
1363 static __isl_give isl_printer *upoly_print_cst(__isl_keep struct isl_upoly *up,
1364 __isl_take isl_printer *p, int first)
1366 struct isl_upoly_cst *cst;
1367 int neg;
1369 cst = isl_upoly_as_cst(up);
1370 if (!cst)
1371 goto error;
1372 neg = !first && isl_int_is_neg(cst->n);
1373 if (!first)
1374 p = isl_printer_print_str(p, neg ? " - " : " + ");
1375 if (neg)
1376 isl_int_neg(cst->n, cst->n);
1377 if (isl_int_is_zero(cst->d)) {
1378 int sgn = isl_int_sgn(cst->n);
1379 p = isl_printer_print_str(p, sgn < 0 ? "-infty" :
1380 sgn == 0 ? "NaN" : "infty");
1381 } else
1382 p = isl_printer_print_isl_int(p, cst->n);
1383 if (neg)
1384 isl_int_neg(cst->n, cst->n);
1385 if (!isl_int_is_zero(cst->d) && !isl_int_is_one(cst->d)) {
1386 p = isl_printer_print_str(p, "/");
1387 p = isl_printer_print_isl_int(p, cst->d);
1389 return p;
1390 error:
1391 isl_printer_free(p);
1392 return NULL;
1395 static __isl_give isl_printer *print_base(__isl_take isl_printer *p,
1396 __isl_keep isl_dim *dim, __isl_keep isl_mat *div, int var)
1398 unsigned total;
1400 total = isl_dim_total(dim);
1401 if (var < total)
1402 p = print_term(dim, NULL, dim->ctx->one, 1 + var, p, 1, 0);
1403 else
1404 p = print_div(dim, div, var - total, p);
1405 return p;
1408 static __isl_give isl_printer *print_pow(__isl_take isl_printer *p,
1409 __isl_keep isl_dim *dim, __isl_keep isl_mat *div, int var, int exp)
1411 p = print_base(p, dim, div, var);
1412 if (exp == 1)
1413 return p;
1414 if (p->output_format == ISL_FORMAT_C) {
1415 int i;
1416 for (i = 1; i < exp; ++i) {
1417 p = isl_printer_print_str(p, "*");
1418 p = print_base(p, dim, div, var);
1420 } else {
1421 p = isl_printer_print_str(p, "^");
1422 p = isl_printer_print_int(p, exp);
1424 return p;
1427 static __isl_give isl_printer *upoly_print(__isl_keep struct isl_upoly *up,
1428 __isl_keep isl_dim *dim, __isl_keep isl_mat *div,
1429 __isl_take isl_printer *p, int outer)
1431 int i, n, first, print_parens;
1432 struct isl_upoly_rec *rec;
1434 if (!p || !up || !dim || !div)
1435 goto error;
1437 if (isl_upoly_is_cst(up))
1438 return upoly_print_cst(up, p, 1);
1440 rec = isl_upoly_as_rec(up);
1441 if (!rec)
1442 goto error;
1443 n = upoly_rec_n_non_zero(rec);
1444 print_parens = n > 1 || (outer && rec->up.var >= isl_dim_total(dim));
1445 if (print_parens)
1446 p = isl_printer_print_str(p, "(");
1447 for (i = 0, first = 1; i < rec->n; ++i) {
1448 if (isl_upoly_is_zero(rec->p[i]))
1449 continue;
1450 if (isl_upoly_is_negone(rec->p[i])) {
1451 if (!i)
1452 p = isl_printer_print_str(p, "-1");
1453 else if (first)
1454 p = isl_printer_print_str(p, "-");
1455 else
1456 p = isl_printer_print_str(p, " - ");
1457 } else if (isl_upoly_is_cst(rec->p[i]) &&
1458 !isl_upoly_is_one(rec->p[i]))
1459 p = upoly_print_cst(rec->p[i], p, first);
1460 else {
1461 if (!first)
1462 p = isl_printer_print_str(p, " + ");
1463 if (i == 0 || !isl_upoly_is_one(rec->p[i]))
1464 p = upoly_print(rec->p[i], dim, div, p, 0);
1466 first = 0;
1467 if (i == 0)
1468 continue;
1469 if (!isl_upoly_is_one(rec->p[i]) &&
1470 !isl_upoly_is_negone(rec->p[i]))
1471 p = isl_printer_print_str(p, " * ");
1472 p = print_pow(p, dim, div, rec->up.var, i);
1474 if (print_parens)
1475 p = isl_printer_print_str(p, ")");
1476 return p;
1477 error:
1478 isl_printer_free(p);
1479 return NULL;
1482 static __isl_give isl_printer *print_qpolynomial(__isl_take isl_printer *p,
1483 __isl_keep isl_qpolynomial *qp)
1485 if (!p || !qp)
1486 goto error;
1487 p = upoly_print(qp->upoly, qp->dim, qp->div, p, 1);
1488 return p;
1489 error:
1490 isl_printer_free(p);
1491 return NULL;
1494 static __isl_give isl_printer *print_qpolynomial_isl(__isl_take isl_printer *p,
1495 __isl_keep isl_qpolynomial *qp)
1497 if (!p || !qp)
1498 goto error;
1500 if (isl_dim_size(qp->dim, isl_dim_param) > 0) {
1501 p = print_tuple(qp->dim, p, isl_dim_param, 0, 0, NULL);
1502 p = isl_printer_print_str(p, " -> ");
1504 p = isl_printer_print_str(p, "{ ");
1505 if (isl_dim_size(qp->dim, isl_dim_set) > 0 ||
1506 isl_dim_is_named_or_nested(qp->dim, isl_dim_set)) {
1507 p = print_dim(qp->dim, p, 1, 0, 0, NULL);
1508 p = isl_printer_print_str(p, " -> ");
1510 p = print_qpolynomial(p, qp);
1511 p = isl_printer_print_str(p, " }");
1512 return p;
1513 error:
1514 isl_printer_free(p);
1515 return NULL;
1518 static __isl_give isl_printer *print_qpolynomial_c(__isl_take isl_printer *p,
1519 __isl_keep isl_dim *dim, __isl_keep isl_qpolynomial *qp)
1521 isl_int den;
1523 isl_int_init(den);
1524 isl_qpolynomial_get_den(qp, &den);
1525 if (!isl_int_is_one(den)) {
1526 isl_qpolynomial *f;
1527 p = isl_printer_print_str(p, "(");
1528 qp = isl_qpolynomial_copy(qp);
1529 f = isl_qpolynomial_rat_cst(isl_dim_copy(qp->dim),
1530 den, qp->dim->ctx->one);
1531 qp = isl_qpolynomial_mul(qp, f);
1533 if (qp)
1534 p = upoly_print(qp->upoly, dim, qp->div, p, 0);
1535 if (!isl_int_is_one(den)) {
1536 p = isl_printer_print_str(p, ")/");
1537 p = isl_printer_print_isl_int(p, den);
1538 isl_qpolynomial_free(qp);
1540 isl_int_clear(den);
1541 return p;
1544 __isl_give isl_printer *isl_printer_print_qpolynomial(
1545 __isl_take isl_printer *p, __isl_keep isl_qpolynomial *qp)
1547 if (!p || !qp)
1548 goto error;
1550 if (p->output_format == ISL_FORMAT_ISL)
1551 return print_qpolynomial_isl(p, qp);
1552 else if (p->output_format == ISL_FORMAT_C)
1553 return print_qpolynomial_c(p, qp->dim, qp);
1554 else
1555 isl_die(qp->dim->ctx, isl_error_unsupported,
1556 "output format not supported for isl_qpolynomials",
1557 goto error);
1558 error:
1559 isl_printer_free(p);
1560 return NULL;
1563 void isl_qpolynomial_print(__isl_keep isl_qpolynomial *qp, FILE *out,
1564 unsigned output_format)
1566 isl_printer *p;
1568 if (!qp)
1569 return;
1571 isl_assert(qp->dim->ctx, output_format == ISL_FORMAT_ISL, return);
1572 p = isl_printer_to_file(qp->dim->ctx, out);
1573 p = isl_printer_print_qpolynomial(p, qp);
1574 isl_printer_free(p);
1577 static __isl_give isl_printer *qpolynomial_fold_print(
1578 __isl_keep isl_qpolynomial_fold *fold, __isl_take isl_printer *p)
1580 int i;
1582 if (fold->type == isl_fold_min)
1583 p = isl_printer_print_str(p, "min");
1584 else if (fold->type == isl_fold_max)
1585 p = isl_printer_print_str(p, "max");
1586 p = isl_printer_print_str(p, "(");
1587 for (i = 0; i < fold->n; ++i) {
1588 if (i)
1589 p = isl_printer_print_str(p, ", ");
1590 p = print_qpolynomial(p, fold->qp[i]);
1592 p = isl_printer_print_str(p, ")");
1593 return p;
1596 void isl_qpolynomial_fold_print(__isl_keep isl_qpolynomial_fold *fold,
1597 FILE *out, unsigned output_format)
1599 isl_printer *p;
1601 if (!fold)
1602 return;
1604 isl_assert(fold->dim->ctx, output_format == ISL_FORMAT_ISL, return);
1606 p = isl_printer_to_file(fold->dim->ctx, out);
1607 p = isl_printer_print_qpolynomial_fold(p, fold);
1609 isl_printer_free(p);
1612 static __isl_give isl_printer *isl_pwqp_print_isl_body(
1613 __isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial *pwqp)
1615 int i = 0;
1617 for (i = 0; i < pwqp->n; ++i) {
1618 if (i)
1619 p = isl_printer_print_str(p, "; ");
1620 if (isl_dim_size(pwqp->dim, isl_dim_set) > 0 ||
1621 isl_dim_is_named_or_nested(pwqp->dim, isl_dim_set)) {
1622 p = print_dim(pwqp->p[i].set->dim, p, 1, 0, 0, NULL);
1623 p = isl_printer_print_str(p, " -> ");
1625 p = print_qpolynomial(p, pwqp->p[i].qp);
1626 p = print_disjuncts((isl_map *)pwqp->p[i].set, p, 1, 0);
1629 return p;
1632 static __isl_give isl_printer *print_pw_qpolynomial_isl(
1633 __isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial *pwqp)
1635 if (!p || !pwqp)
1636 goto error;
1638 if (isl_dim_size(pwqp->dim, isl_dim_param) > 0) {
1639 p = print_tuple(pwqp->dim, p, isl_dim_param, 0, 0, NULL);
1640 p = isl_printer_print_str(p, " -> ");
1642 p = isl_printer_print_str(p, "{ ");
1643 if (pwqp->n == 0) {
1644 if (isl_dim_size(pwqp->dim, isl_dim_set) > 0 ||
1645 isl_dim_is_named_or_nested(pwqp->dim, isl_dim_set)) {
1646 p = print_dim(pwqp->dim, p, 1, 0, 0, NULL);
1647 p = isl_printer_print_str(p, " -> ");
1649 p = isl_printer_print_str(p, "0");
1651 p = isl_pwqp_print_isl_body(p, pwqp);
1652 p = isl_printer_print_str(p, " }");
1653 return p;
1654 error:
1655 isl_printer_free(p);
1656 return NULL;
1659 void isl_pw_qpolynomial_print(__isl_keep isl_pw_qpolynomial *pwqp, FILE *out,
1660 unsigned output_format)
1662 isl_printer *p;
1664 if (!pwqp)
1665 return;
1667 p = isl_printer_to_file(pwqp->dim->ctx, out);
1668 p = isl_printer_set_output_format(p, output_format);
1669 p = isl_printer_print_pw_qpolynomial(p, pwqp);
1671 isl_printer_free(p);
1674 static __isl_give isl_printer *isl_pwf_print_isl_body(
1675 __isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial_fold *pwf)
1677 int i = 0;
1679 for (i = 0; i < pwf->n; ++i) {
1680 if (i)
1681 p = isl_printer_print_str(p, "; ");
1682 if (isl_dim_size(pwf->dim, isl_dim_set) > 0 ||
1683 isl_dim_is_named_or_nested(pwf->dim, isl_dim_set)) {
1684 p = print_dim(pwf->p[i].set->dim, p, 1, 0, 0, NULL);
1685 p = isl_printer_print_str(p, " -> ");
1687 p = qpolynomial_fold_print(pwf->p[i].fold, p);
1688 p = print_disjuncts((isl_map *)pwf->p[i].set, p, 1, 0);
1691 return p;
1694 static __isl_give isl_printer *print_pw_qpolynomial_fold_isl(
1695 __isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial_fold *pwf)
1697 if (isl_dim_size(pwf->dim, isl_dim_param) > 0) {
1698 p = print_tuple(pwf->dim, p, isl_dim_param, 0, 0, NULL);
1699 p = isl_printer_print_str(p, " -> ");
1701 p = isl_printer_print_str(p, "{ ");
1702 if (pwf->n == 0) {
1703 if (isl_dim_size(pwf->dim, isl_dim_set) > 0 ||
1704 isl_dim_is_named_or_nested(pwf->dim, isl_dim_set)) {
1705 p = print_dim(pwf->dim, p, 1, 0, 0, NULL);
1706 p = isl_printer_print_str(p, " -> ");
1708 p = isl_printer_print_str(p, "0");
1710 p = isl_pwf_print_isl_body(p, pwf);
1711 p = isl_printer_print_str(p, " }");
1712 return p;
1715 static __isl_give isl_printer *print_affine_c(__isl_take isl_printer *p,
1716 __isl_keep isl_dim *dim, __isl_keep isl_basic_set *bset, isl_int *c);
1718 static __isl_give isl_printer *print_name_c(__isl_take isl_printer *p,
1719 __isl_keep isl_dim *dim,
1720 __isl_keep isl_basic_set *bset, enum isl_dim_type type, unsigned pos)
1722 if (type == isl_dim_div) {
1723 p = isl_printer_print_str(p, "floord(");
1724 p = print_affine_c(p, dim, bset, bset->div[pos] + 1);
1725 p = isl_printer_print_str(p, ", ");
1726 p = isl_printer_print_isl_int(p, bset->div[pos][0]);
1727 p = isl_printer_print_str(p, ")");
1728 } else {
1729 const char *name;
1731 name = isl_dim_get_name(dim, type, pos);
1732 if (!name)
1733 name = "UNNAMED";
1734 p = isl_printer_print_str(p, name);
1736 return p;
1739 static __isl_give isl_printer *print_term_c(__isl_take isl_printer *p,
1740 __isl_keep isl_dim *dim,
1741 __isl_keep isl_basic_set *bset, isl_int c, unsigned pos)
1743 enum isl_dim_type type;
1744 unsigned nparam = isl_basic_set_dim(bset, isl_dim_param);
1746 if (pos == 0)
1747 return isl_printer_print_isl_int(p, c);
1749 if (isl_int_is_one(c))
1751 else if (isl_int_is_negone(c))
1752 p = isl_printer_print_str(p, "-");
1753 else {
1754 p = isl_printer_print_isl_int(p, c);
1755 p = isl_printer_print_str(p, "*");
1757 type = pos2type(dim, &pos);
1758 p = print_name_c(p, dim, bset, type, pos);
1759 return p;
1762 static __isl_give isl_printer *print_partial_affine_c(__isl_take isl_printer *p,
1763 __isl_keep isl_dim *dim,
1764 __isl_keep isl_basic_set *bset, isl_int *c, unsigned len)
1766 int i;
1767 int first;
1769 for (i = 0, first = 1; i < len; ++i) {
1770 int flip = 0;
1771 if (isl_int_is_zero(c[i]))
1772 continue;
1773 if (!first) {
1774 if (isl_int_is_neg(c[i])) {
1775 flip = 1;
1776 isl_int_neg(c[i], c[i]);
1777 p = isl_printer_print_str(p, " - ");
1778 } else
1779 p = isl_printer_print_str(p, " + ");
1781 first = 0;
1782 p = print_term_c(p, dim, bset, c[i], i);
1783 if (flip)
1784 isl_int_neg(c[i], c[i]);
1786 if (first)
1787 p = isl_printer_print_str(p, "0");
1788 return p;
1791 static __isl_give isl_printer *print_affine_c(__isl_take isl_printer *p,
1792 __isl_keep isl_dim *dim, __isl_keep isl_basic_set *bset, isl_int *c)
1794 unsigned len = 1 + isl_basic_set_total_dim(bset);
1795 return print_partial_affine_c(p, dim, bset, c, len);
1798 static __isl_give isl_printer *print_constraint_c(__isl_take isl_printer *p,
1799 __isl_keep isl_dim *dim,
1800 __isl_keep isl_basic_set *bset, isl_int *c, const char *op, int first)
1802 if (!first)
1803 p = isl_printer_print_str(p, " && ");
1805 p = print_affine_c(p, dim, bset, c);
1806 p = isl_printer_print_str(p, " ");
1807 p = isl_printer_print_str(p, op);
1808 p = isl_printer_print_str(p, " 0");
1809 return p;
1812 static __isl_give isl_printer *print_basic_set_c(__isl_take isl_printer *p,
1813 __isl_keep isl_dim *dim, __isl_keep isl_basic_set *bset)
1815 int i, j;
1816 unsigned n_div = isl_basic_set_dim(bset, isl_dim_div);
1817 unsigned total = isl_basic_set_total_dim(bset) - n_div;
1819 for (i = 0; i < bset->n_eq; ++i) {
1820 j = isl_seq_last_non_zero(bset->eq[i] + 1 + total, n_div);
1821 if (j < 0)
1822 p = print_constraint_c(p, dim, bset,
1823 bset->eq[i], "==", !i);
1824 else {
1825 if (i)
1826 p = isl_printer_print_str(p, " && ");
1827 p = isl_printer_print_str(p, "(");
1828 p = print_partial_affine_c(p, dim, bset, bset->eq[i],
1829 1 + total + j);
1830 p = isl_printer_print_str(p, ") % ");
1831 p = isl_printer_print_isl_int(p,
1832 bset->eq[i][1 + total + j]);
1833 p = isl_printer_print_str(p, " == 0");
1836 for (i = 0; i < bset->n_ineq; ++i)
1837 p = print_constraint_c(p, dim, bset, bset->ineq[i], ">=",
1838 !bset->n_eq && !i);
1839 return p;
1842 static __isl_give isl_printer *print_set_c(__isl_take isl_printer *p,
1843 __isl_keep isl_dim *dim, __isl_keep isl_set *set)
1845 int i;
1847 if (set->n == 0)
1848 p = isl_printer_print_str(p, "0");
1850 for (i = 0; i < set->n; ++i) {
1851 if (i)
1852 p = isl_printer_print_str(p, " || ");
1853 if (set->n > 1)
1854 p = isl_printer_print_str(p, "(");
1855 p = print_basic_set_c(p, dim, set->p[i]);
1856 if (set->n > 1)
1857 p = isl_printer_print_str(p, ")");
1859 return p;
1862 static __isl_give isl_printer *print_pw_qpolynomial_c(
1863 __isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial *pwqp)
1865 int i;
1867 if (pwqp->n == 1 && isl_set_plain_is_universe(pwqp->p[0].set))
1868 return print_qpolynomial_c(p, pwqp->dim, pwqp->p[0].qp);
1870 for (i = 0; i < pwqp->n; ++i) {
1871 p = isl_printer_print_str(p, "(");
1872 p = print_set_c(p, pwqp->dim, pwqp->p[i].set);
1873 p = isl_printer_print_str(p, ") ? (");
1874 p = print_qpolynomial_c(p, pwqp->dim, pwqp->p[i].qp);
1875 p = isl_printer_print_str(p, ") : ");
1878 p = isl_printer_print_str(p, "0");
1879 return p;
1882 __isl_give isl_printer *isl_printer_print_pw_qpolynomial(
1883 __isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial *pwqp)
1885 if (!p || !pwqp)
1886 goto error;
1888 if (p->output_format == ISL_FORMAT_ISL)
1889 return print_pw_qpolynomial_isl(p, pwqp);
1890 else if (p->output_format == ISL_FORMAT_C)
1891 return print_pw_qpolynomial_c(p, pwqp);
1892 isl_assert(p->ctx, 0, goto error);
1893 error:
1894 isl_printer_free(p);
1895 return NULL;
1898 static int print_pwqp_body(__isl_take isl_pw_qpolynomial *pwqp, void *user)
1900 struct isl_union_print_data *data;
1901 data = (struct isl_union_print_data *)user;
1903 if (!data->first)
1904 data->p = isl_printer_print_str(data->p, "; ");
1905 data->first = 0;
1907 data->p = isl_pwqp_print_isl_body(data->p, pwqp);
1908 isl_pw_qpolynomial_free(pwqp);
1910 return 0;
1913 static __isl_give isl_printer *print_union_pw_qpolynomial_isl(
1914 __isl_take isl_printer *p, __isl_keep isl_union_pw_qpolynomial *upwqp)
1916 struct isl_union_print_data data = { p, 1, 1 };
1917 isl_dim *dim;
1918 dim = isl_union_pw_qpolynomial_get_dim(upwqp);
1919 if (isl_dim_size(dim, isl_dim_param) > 0) {
1920 p = print_tuple(dim, p, isl_dim_param, 1, 0, NULL);
1921 p = isl_printer_print_str(p, " -> ");
1923 isl_dim_free(dim);
1924 p = isl_printer_print_str(p, "{ ");
1925 isl_union_pw_qpolynomial_foreach_pw_qpolynomial(upwqp, &print_pwqp_body,
1926 &data);
1927 p = data.p;
1928 p = isl_printer_print_str(p, " }");
1929 return p;
1932 __isl_give isl_printer *isl_printer_print_union_pw_qpolynomial(
1933 __isl_take isl_printer *p, __isl_keep isl_union_pw_qpolynomial *upwqp)
1935 if (!p || !upwqp)
1936 goto error;
1938 if (p->output_format == ISL_FORMAT_ISL)
1939 return print_union_pw_qpolynomial_isl(p, upwqp);
1940 isl_die(p->ctx, isl_error_invalid,
1941 "invalid output format for isl_union_pw_qpolynomial",
1942 goto error);
1943 error:
1944 isl_printer_free(p);
1945 return NULL;
1948 static __isl_give isl_printer *print_qpolynomial_fold_c(
1949 __isl_take isl_printer *p, __isl_keep isl_dim *dim,
1950 __isl_keep isl_qpolynomial_fold *fold)
1952 int i;
1954 for (i = 0; i < fold->n - 1; ++i)
1955 if (fold->type == isl_fold_min)
1956 p = isl_printer_print_str(p, "min(");
1957 else if (fold->type == isl_fold_max)
1958 p = isl_printer_print_str(p, "max(");
1960 for (i = 0; i < fold->n; ++i) {
1961 if (i)
1962 p = isl_printer_print_str(p, ", ");
1963 p = print_qpolynomial_c(p, dim, fold->qp[i]);
1964 if (i)
1965 p = isl_printer_print_str(p, ")");
1967 return p;
1970 __isl_give isl_printer *isl_printer_print_qpolynomial_fold(
1971 __isl_take isl_printer *p, __isl_keep isl_qpolynomial_fold *fold)
1973 if (!p || !fold)
1974 goto error;
1975 if (p->output_format == ISL_FORMAT_ISL)
1976 return qpolynomial_fold_print(fold, p);
1977 else if (p->output_format == ISL_FORMAT_C)
1978 return print_qpolynomial_fold_c(p, fold->dim, fold);
1979 isl_die(p->ctx, isl_error_unsupported, "unsupported output format",
1980 goto error);
1981 error:
1982 isl_printer_free(p);
1983 return NULL;
1986 static __isl_give isl_printer *print_pw_qpolynomial_fold_c(
1987 __isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial_fold *pwf)
1989 int i;
1991 if (pwf->n == 1 && isl_set_plain_is_universe(pwf->p[0].set))
1992 return print_qpolynomial_fold_c(p, pwf->dim, pwf->p[0].fold);
1994 for (i = 0; i < pwf->n; ++i) {
1995 p = isl_printer_print_str(p, "(");
1996 p = print_set_c(p, pwf->dim, pwf->p[i].set);
1997 p = isl_printer_print_str(p, ") ? (");
1998 p = print_qpolynomial_fold_c(p, pwf->dim, pwf->p[i].fold);
1999 p = isl_printer_print_str(p, ") : ");
2002 p = isl_printer_print_str(p, "0");
2003 return p;
2006 __isl_give isl_printer *isl_printer_print_pw_qpolynomial_fold(
2007 __isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial_fold *pwf)
2009 if (!p || !pwf)
2010 goto error;
2012 if (p->output_format == ISL_FORMAT_ISL)
2013 return print_pw_qpolynomial_fold_isl(p, pwf);
2014 else if (p->output_format == ISL_FORMAT_C)
2015 return print_pw_qpolynomial_fold_c(p, pwf);
2016 isl_assert(p->ctx, 0, goto error);
2017 error:
2018 isl_printer_free(p);
2019 return NULL;
2022 void isl_pw_qpolynomial_fold_print(__isl_keep isl_pw_qpolynomial_fold *pwf,
2023 FILE *out, unsigned output_format)
2025 isl_printer *p;
2027 if (!pwf)
2028 return;
2030 p = isl_printer_to_file(pwf->dim->ctx, out);
2031 p = isl_printer_set_output_format(p, output_format);
2032 p = isl_printer_print_pw_qpolynomial_fold(p, pwf);
2034 isl_printer_free(p);
2037 static int print_pwf_body(__isl_take isl_pw_qpolynomial_fold *pwf, void *user)
2039 struct isl_union_print_data *data;
2040 data = (struct isl_union_print_data *)user;
2042 if (!data->first)
2043 data->p = isl_printer_print_str(data->p, "; ");
2044 data->first = 0;
2046 data->p = isl_pwf_print_isl_body(data->p, pwf);
2047 isl_pw_qpolynomial_fold_free(pwf);
2049 return 0;
2052 static __isl_give isl_printer *print_union_pw_qpolynomial_fold_isl(
2053 __isl_take isl_printer *p,
2054 __isl_keep isl_union_pw_qpolynomial_fold *upwf)
2056 struct isl_union_print_data data = { p, 1, 1 };
2057 isl_dim *dim;
2058 dim = isl_union_pw_qpolynomial_fold_get_dim(upwf);
2059 if (isl_dim_size(dim, isl_dim_param) > 0) {
2060 p = print_tuple(dim, p, isl_dim_param, 1, 0, NULL);
2061 p = isl_printer_print_str(p, " -> ");
2063 isl_dim_free(dim);
2064 p = isl_printer_print_str(p, "{ ");
2065 isl_union_pw_qpolynomial_fold_foreach_pw_qpolynomial_fold(upwf,
2066 &print_pwf_body, &data);
2067 p = data.p;
2068 p = isl_printer_print_str(p, " }");
2069 return p;
2072 __isl_give isl_printer *isl_printer_print_union_pw_qpolynomial_fold(
2073 __isl_take isl_printer *p,
2074 __isl_keep isl_union_pw_qpolynomial_fold *upwf)
2076 if (!p || !upwf)
2077 goto error;
2079 if (p->output_format == ISL_FORMAT_ISL)
2080 return print_union_pw_qpolynomial_fold_isl(p, upwf);
2081 isl_die(p->ctx, isl_error_invalid,
2082 "invalid output format for isl_union_pw_qpolynomial_fold",
2083 goto error);
2084 error:
2085 isl_printer_free(p);
2086 return NULL;
2089 __isl_give isl_printer *isl_printer_print_constraint(__isl_take isl_printer *p,
2090 __isl_keep isl_constraint *c)
2092 isl_basic_map *bmap;
2094 if (!p || !c)
2095 goto error;
2097 bmap = isl_basic_map_from_constraint(isl_constraint_copy(c));
2098 p = isl_printer_print_basic_map(p, bmap);
2099 isl_basic_map_free(bmap);
2100 return p;
2101 error:
2102 isl_printer_free(p);
2103 return NULL;
2106 void isl_constraint_dump(__isl_keep isl_constraint *c)
2108 isl_printer *printer;
2110 if (!c)
2111 return;
2113 printer = isl_printer_to_file(isl_constraint_get_ctx(c), stderr);
2114 printer = isl_printer_print_constraint(printer, c);
2115 printer = isl_printer_end_line(printer);
2117 isl_printer_free(printer);