add isl_local_space_is_params
[isl.git] / isl_output.c
blobb5d6c9233843b79485b4eb65fd987311948756af
1 /*
2 * Copyright 2008-2009 Katholieke Universiteit Leuven
3 * Copyright 2010 INRIA Saclay
4 * Copyright 2012-2013 Ecole Normale Superieure
6 * Use of this software is governed by the MIT license
8 * Written by Sven Verdoolaege, K.U.Leuven, Departement
9 * Computerwetenschappen, Celestijnenlaan 200A, B-3001 Leuven, Belgium
10 * and INRIA Saclay - Ile-de-France, Parc Club Orsay Universite,
11 * ZAC des vignes, 4 rue Jacques Monod, 91893 Orsay, France
12 * and Ecole Normale Superieure, 45 rue d’Ulm, 75230 Paris, France
15 #include <stdlib.h>
16 #include <string.h>
17 #include <isl_ctx_private.h>
18 #include <isl_map_private.h>
19 #include <isl/set.h>
20 #include <isl_seq.h>
21 #include <isl_polynomial_private.h>
22 #include <isl_printer_private.h>
23 #include <isl_space_private.h>
24 #include <isl_mat_private.h>
25 #include <isl_vec_private.h>
26 #include <isl/union_map.h>
27 #include <isl/constraint.h>
28 #include <isl_local_space_private.h>
29 #include <isl_aff_private.h>
30 #include <isl_val_private.h>
31 #include <isl/ast_build.h>
32 #include <isl_sort.h>
34 static const char *s_to[2] = { " -> ", " \\to " };
35 static const char *s_and[2] = { " and ", " \\wedge " };
36 static const char *s_or[2] = { " or ", " \\vee " };
37 static const char *s_le[2] = { "<=", "\\le" };
38 static const char *s_ge[2] = { ">=", "\\ge" };
39 static const char *s_open_set[2] = { "{ ", "\\{\\, " };
40 static const char *s_close_set[2] = { " }", " \\,\\}" };
41 static const char *s_open_list[2] = { "[", "(" };
42 static const char *s_close_list[2] = { "]", ")" };
43 static const char *s_such_that[2] = { " : ", " \\mid " };
44 static const char *s_open_exists[2] = { "exists (", "\\exists \\, " };
45 static const char *s_close_exists[2] = { ")", "" };
46 static const char *s_div_prefix[2] = { "e", "\\alpha_" };
47 static const char *s_param_prefix[2] = { "p", "p_" };
48 static const char *s_input_prefix[2] = { "i", "i_" };
49 static const char *s_output_prefix[2] = { "o", "o_" };
51 static __isl_give isl_printer *print_constraint_polylib(
52 struct isl_basic_map *bmap, int ineq, int n, __isl_take isl_printer *p)
54 int i;
55 unsigned n_in = isl_basic_map_dim(bmap, isl_dim_in);
56 unsigned n_out = isl_basic_map_dim(bmap, isl_dim_out);
57 unsigned nparam = isl_basic_map_dim(bmap, isl_dim_param);
58 isl_int *c = ineq ? bmap->ineq[n] : bmap->eq[n];
60 p = isl_printer_start_line(p);
61 p = isl_printer_print_int(p, ineq);
62 for (i = 0; i < n_out; ++i) {
63 p = isl_printer_print_str(p, " ");
64 p = isl_printer_print_isl_int(p, c[1+nparam+n_in+i]);
66 for (i = 0; i < n_in; ++i) {
67 p = isl_printer_print_str(p, " ");
68 p = isl_printer_print_isl_int(p, c[1+nparam+i]);
70 for (i = 0; i < bmap->n_div; ++i) {
71 p = isl_printer_print_str(p, " ");
72 p = isl_printer_print_isl_int(p, c[1+nparam+n_in+n_out+i]);
74 for (i = 0; i < nparam; ++i) {
75 p = isl_printer_print_str(p, " ");
76 p = isl_printer_print_isl_int(p, c[1+i]);
78 p = isl_printer_print_str(p, " ");
79 p = isl_printer_print_isl_int(p, c[0]);
80 p = isl_printer_end_line(p);
81 return p;
84 static __isl_give isl_printer *print_constraints_polylib(
85 struct isl_basic_map *bmap, __isl_take isl_printer *p)
87 int i;
89 p = isl_printer_set_isl_int_width(p, 5);
91 for (i = 0; i < bmap->n_eq; ++i)
92 p = print_constraint_polylib(bmap, 0, i, p);
93 for (i = 0; i < bmap->n_ineq; ++i)
94 p = print_constraint_polylib(bmap, 1, i, p);
96 return p;
99 static __isl_give isl_printer *bset_print_constraints_polylib(
100 struct isl_basic_set *bset, __isl_take isl_printer *p)
102 return print_constraints_polylib((struct isl_basic_map *)bset, p);
105 static __isl_give isl_printer *isl_basic_map_print_polylib(
106 __isl_keep isl_basic_map *bmap, __isl_take isl_printer *p, int ext)
108 unsigned total = isl_basic_map_total_dim(bmap);
109 p = isl_printer_start_line(p);
110 p = isl_printer_print_int(p, bmap->n_eq + bmap->n_ineq);
111 p = isl_printer_print_str(p, " ");
112 p = isl_printer_print_int(p, 1 + total + 1);
113 if (ext) {
114 p = isl_printer_print_str(p, " ");
115 p = isl_printer_print_int(p,
116 isl_basic_map_dim(bmap, isl_dim_out));
117 p = isl_printer_print_str(p, " ");
118 p = isl_printer_print_int(p,
119 isl_basic_map_dim(bmap, isl_dim_in));
120 p = isl_printer_print_str(p, " ");
121 p = isl_printer_print_int(p,
122 isl_basic_map_dim(bmap, isl_dim_div));
123 p = isl_printer_print_str(p, " ");
124 p = isl_printer_print_int(p,
125 isl_basic_map_dim(bmap, isl_dim_param));
127 p = isl_printer_end_line(p);
128 return print_constraints_polylib(bmap, p);
131 static __isl_give isl_printer *isl_basic_set_print_polylib(
132 __isl_keep isl_basic_set *bset, __isl_take isl_printer *p, int ext)
134 return isl_basic_map_print_polylib((struct isl_basic_map *)bset, p, ext);
137 static __isl_give isl_printer *isl_map_print_polylib(__isl_keep isl_map *map,
138 __isl_take isl_printer *p, int ext)
140 int i;
142 p = isl_printer_start_line(p);
143 p = isl_printer_print_int(p, map->n);
144 p = isl_printer_end_line(p);
145 for (i = 0; i < map->n; ++i) {
146 p = isl_printer_start_line(p);
147 p = isl_printer_end_line(p);
148 p = isl_basic_map_print_polylib(map->p[i], p, ext);
150 return p;
153 static __isl_give isl_printer *isl_set_print_polylib(__isl_keep isl_set *set,
154 __isl_take isl_printer *p, int ext)
156 return isl_map_print_polylib((struct isl_map *)set, p, ext);
159 static int count_same_name(__isl_keep isl_space *dim,
160 enum isl_dim_type type, unsigned pos, const char *name)
162 enum isl_dim_type t;
163 unsigned p, s;
164 int count = 0;
166 for (t = isl_dim_param; t <= type && t <= isl_dim_out; ++t) {
167 s = t == type ? pos : isl_space_dim(dim, t);
168 for (p = 0; p < s; ++p) {
169 const char *n = isl_space_get_dim_name(dim, t, p);
170 if (n && !strcmp(n, name))
171 count++;
174 return count;
177 static __isl_give isl_printer *print_name(__isl_keep isl_space *dim,
178 __isl_take isl_printer *p, enum isl_dim_type type, unsigned pos,
179 int latex)
181 const char *name;
182 char buffer[20];
183 int primes;
185 name = type == isl_dim_div ? NULL : isl_space_get_dim_name(dim, type, pos);
187 if (!name) {
188 const char *prefix;
189 if (type == isl_dim_param)
190 prefix = s_param_prefix[latex];
191 else if (type == isl_dim_div)
192 prefix = s_div_prefix[latex];
193 else if (isl_space_is_set(dim) || type == isl_dim_in)
194 prefix = s_input_prefix[latex];
195 else
196 prefix = s_output_prefix[latex];
197 snprintf(buffer, sizeof(buffer), "%s%d", prefix, pos);
198 name = buffer;
200 primes = count_same_name(dim, name == buffer ? isl_dim_div : type,
201 pos, name);
202 p = isl_printer_print_str(p, name);
203 while (primes-- > 0)
204 p = isl_printer_print_str(p, "'");
205 return p;
208 static enum isl_dim_type pos2type(__isl_keep isl_space *dim, unsigned *pos)
210 enum isl_dim_type type;
211 unsigned n_in = isl_space_dim(dim, isl_dim_in);
212 unsigned n_out = isl_space_dim(dim, isl_dim_out);
213 unsigned nparam = isl_space_dim(dim, isl_dim_param);
215 if (*pos < 1 + nparam) {
216 type = isl_dim_param;
217 *pos -= 1;
218 } else if (*pos < 1 + nparam + n_in) {
219 type = isl_dim_in;
220 *pos -= 1 + nparam;
221 } else if (*pos < 1 + nparam + n_in + n_out) {
222 type = isl_dim_out;
223 *pos -= 1 + nparam + n_in;
224 } else {
225 type = isl_dim_div;
226 *pos -= 1 + nparam + n_in + n_out;
229 return type;
232 static __isl_give isl_printer *print_div(__isl_keep isl_space *dim,
233 __isl_keep isl_mat *div, int pos, __isl_take isl_printer *p);
235 static __isl_give isl_printer *print_term(__isl_keep isl_space *dim,
236 __isl_keep isl_mat *div,
237 isl_int c, unsigned pos, __isl_take isl_printer *p, int latex)
239 enum isl_dim_type type;
240 int print_div_def;
242 if (pos == 0)
243 return isl_printer_print_isl_int(p, c);
245 type = pos2type(dim, &pos);
246 print_div_def = type == isl_dim_div && div &&
247 !isl_int_is_zero(div->row[pos][0]);
249 if (isl_int_is_one(c))
251 else if (isl_int_is_negone(c))
252 p = isl_printer_print_str(p, "-");
253 else {
254 p = isl_printer_print_isl_int(p, c);
255 if (p->output_format == ISL_FORMAT_C || print_div_def)
256 p = isl_printer_print_str(p, "*");
258 if (print_div_def)
259 p = print_div(dim, div, pos, p);
260 else
261 p = print_name(dim, p, type, pos, latex);
262 return p;
265 static __isl_give isl_printer *print_affine_of_len(__isl_keep isl_space *dim,
266 __isl_keep isl_mat *div,
267 __isl_take isl_printer *p, isl_int *c, int len)
269 int i;
270 int first;
272 for (i = 0, first = 1; i < len; ++i) {
273 int flip = 0;
274 if (isl_int_is_zero(c[i]))
275 continue;
276 if (!first) {
277 if (isl_int_is_neg(c[i])) {
278 flip = 1;
279 isl_int_neg(c[i], c[i]);
280 p = isl_printer_print_str(p, " - ");
281 } else
282 p = isl_printer_print_str(p, " + ");
284 first = 0;
285 p = print_term(dim, div, c[i], i, p, 0);
286 if (flip)
287 isl_int_neg(c[i], c[i]);
289 if (first)
290 p = isl_printer_print_str(p, "0");
291 return p;
294 static __isl_give isl_printer *print_affine(__isl_keep isl_basic_map *bmap,
295 __isl_keep isl_space *dim, __isl_take isl_printer *p, isl_int *c)
297 unsigned len = 1 + isl_basic_map_total_dim(bmap);
298 return print_affine_of_len(dim, NULL, p, c, len);
301 /* Internal data structure for print_space.
303 * latex is set if that is the output format.
304 * print_dim (if not NULL) is called on each dimension.
305 * user is set by the caller of print_space and may be used inside print_dim.
307 * space is the global space that is being printed. This field is set by
308 * print_space.
309 * type is the tuple of the global space that is currently being printed.
310 * This field is set by print_space.
312 struct isl_print_space_data {
313 int latex;
314 __isl_give isl_printer *(*print_dim)(__isl_take isl_printer *p,
315 struct isl_print_space_data *data, unsigned pos);
316 void *user;
318 isl_space *space;
319 enum isl_dim_type type;
322 /* offset is the offset of local_dim inside data->type of data->space.
324 static __isl_give isl_printer *print_nested_var_list(__isl_take isl_printer *p,
325 __isl_keep isl_space *local_dim, enum isl_dim_type local_type,
326 struct isl_print_space_data *data, int offset)
328 int i;
330 if (data->space != local_dim && local_type == isl_dim_out)
331 offset += local_dim->n_in;
333 for (i = 0; i < isl_space_dim(local_dim, local_type); ++i) {
334 if (i)
335 p = isl_printer_print_str(p, ", ");
336 if (data->print_dim)
337 p = data->print_dim(p, data, offset + i);
338 else
339 p = print_name(data->space, p, data->type, offset + i,
340 data->latex);
342 return p;
345 static __isl_give isl_printer *print_var_list(__isl_take isl_printer *p,
346 __isl_keep isl_space *space, enum isl_dim_type type)
348 struct isl_print_space_data data = { .space = space, .type = type };
350 return print_nested_var_list(p, space, type, &data, 0);
353 static __isl_give isl_printer *print_nested_map_dim(__isl_take isl_printer *p,
354 __isl_keep isl_space *local_dim,
355 struct isl_print_space_data *data, int offset);
357 static __isl_give isl_printer *print_nested_tuple(__isl_take isl_printer *p,
358 __isl_keep isl_space *local_dim, enum isl_dim_type local_type,
359 struct isl_print_space_data *data, int offset)
361 const char *name = NULL;
362 unsigned n = isl_space_dim(local_dim, local_type);
363 if ((local_type == isl_dim_in || local_type == isl_dim_out)) {
364 name = isl_space_get_tuple_name(local_dim, local_type);
365 if (name) {
366 if (data->latex)
367 p = isl_printer_print_str(p, "\\mathrm{");
368 p = isl_printer_print_str(p, name);
369 if (data->latex)
370 p = isl_printer_print_str(p, "}");
373 if (!data->latex || n != 1 || name)
374 p = isl_printer_print_str(p, s_open_list[data->latex]);
375 if ((local_type == isl_dim_in || local_type == isl_dim_out) &&
376 local_dim->nested[local_type - isl_dim_in]) {
377 if (data->space != local_dim && local_type == isl_dim_out)
378 offset += local_dim->n_in;
379 p = print_nested_map_dim(p,
380 local_dim->nested[local_type - isl_dim_in],
381 data, offset);
382 } else
383 p = print_nested_var_list(p, local_dim, local_type, data,
384 offset);
385 if (!data->latex || n != 1 || name)
386 p = isl_printer_print_str(p, s_close_list[data->latex]);
387 return p;
390 static __isl_give isl_printer *print_tuple(__isl_keep isl_space *dim,
391 __isl_take isl_printer *p, enum isl_dim_type type,
392 struct isl_print_space_data *data)
394 data->space = dim;
395 data->type = type;
396 return print_nested_tuple(p, dim, type, data, 0);
399 static __isl_give isl_printer *print_nested_map_dim(__isl_take isl_printer *p,
400 __isl_keep isl_space *local_dim,
401 struct isl_print_space_data *data, int offset)
403 p = print_nested_tuple(p, local_dim, isl_dim_in, data, offset);
404 p = isl_printer_print_str(p, s_to[data->latex]);
405 p = print_nested_tuple(p, local_dim, isl_dim_out, data, offset);
407 return p;
410 static __isl_give isl_printer *print_space(__isl_keep isl_space *dim,
411 __isl_take isl_printer *p, int rational,
412 struct isl_print_space_data *data)
414 if (rational && !data->latex)
415 p = isl_printer_print_str(p, "rat: ");
416 if (isl_space_is_params(dim))
418 else if (isl_space_is_set(dim))
419 p = print_tuple(dim, p, isl_dim_set, data);
420 else {
421 p = print_tuple(dim, p, isl_dim_in, data);
422 p = isl_printer_print_str(p, s_to[data->latex]);
423 p = print_tuple(dim, p, isl_dim_out, data);
426 return p;
429 static __isl_give isl_printer *print_omega_parameters(__isl_keep isl_space *dim,
430 __isl_take isl_printer *p)
432 if (isl_space_dim(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(p, dim, isl_dim_param);
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_space *dim, __isl_take isl_printer *p,
445 isl_int *c, int last, const char *op, int first_constraint, int latex)
447 if (!first_constraint)
448 p = isl_printer_print_str(p, s_and[latex]);
450 isl_int_abs(c[last], c[last]);
452 p = print_term(dim, NULL, c[last], last, p, latex);
454 p = isl_printer_print_str(p, " ");
455 p = isl_printer_print_str(p, op);
456 p = isl_printer_print_str(p, " ");
458 isl_int_set_si(c[last], 0);
459 p = print_affine(bmap, dim, p, c);
461 return p;
464 static __isl_give isl_printer *print_constraints(__isl_keep isl_basic_map *bmap,
465 __isl_keep isl_space *dim, __isl_take isl_printer *p, int latex)
467 int i;
468 struct isl_vec *c;
469 unsigned total = isl_basic_map_total_dim(bmap);
471 c = isl_vec_alloc(bmap->ctx, 1 + total);
472 if (!c)
473 goto error;
475 for (i = bmap->n_eq - 1; i >= 0; --i) {
476 int l = isl_seq_last_non_zero(bmap->eq[i], 1 + total);
477 if (l < 0) {
478 if (i != bmap->n_eq - 1)
479 p = isl_printer_print_str(p, s_and[latex]);
480 p = isl_printer_print_str(p, "0 = 0");
481 continue;
483 if (isl_int_is_neg(bmap->eq[i][l]))
484 isl_seq_cpy(c->el, bmap->eq[i], 1 + total);
485 else
486 isl_seq_neg(c->el, bmap->eq[i], 1 + total);
487 p = print_constraint(bmap, dim, p, c->el, l,
488 "=", i == bmap->n_eq - 1, latex);
490 for (i = 0; i < bmap->n_ineq; ++i) {
491 int l = isl_seq_last_non_zero(bmap->ineq[i], 1 + total);
492 int s;
493 const char *op;
494 if (l < 0)
495 continue;
496 s = isl_int_sgn(bmap->ineq[i][l]);
497 if (s < 0)
498 isl_seq_cpy(c->el, bmap->ineq[i], 1 + total);
499 else
500 isl_seq_neg(c->el, bmap->ineq[i], 1 + total);
501 op = s < 0 ? s_le[latex] : s_ge[latex];
502 p = print_constraint(bmap, dim, p, c->el, l,
503 op, !bmap->n_eq && !i, latex);
506 isl_vec_free(c);
508 return p;
509 error:
510 isl_vec_free(c);
511 isl_printer_free(p);
512 return NULL;
515 static __isl_give isl_printer *print_div(__isl_keep isl_space *dim,
516 __isl_keep isl_mat *div, int pos, __isl_take isl_printer *p)
518 int c = p->output_format == ISL_FORMAT_C;
519 p = isl_printer_print_str(p, c ? "floord(" : "floor((");
520 p = print_affine_of_len(dim, div, p,
521 div->row[pos] + 1, div->n_col - 1);
522 p = isl_printer_print_str(p, c ? ", " : ")/");
523 p = isl_printer_print_isl_int(p, div->row[pos][0]);
524 p = isl_printer_print_str(p, ")");
525 return p;
528 /* Print a comma separated list of div names, with their definitions
529 * (provided that they have a definition and we are printing in isl format).
531 static __isl_give isl_printer *print_div_list(__isl_take isl_printer *p,
532 __isl_keep isl_space *space, __isl_keep isl_mat *div, int latex)
534 int i;
535 unsigned n_div;
537 if (!p || !space || !div)
538 return isl_printer_free(p);
540 n_div = isl_mat_rows(div);
542 for (i = 0; i < n_div; ++i) {
543 if (i)
544 p = isl_printer_print_str(p, ", ");
545 p = print_name(space, p, isl_dim_div, i, latex);
546 if (p->output_format != ISL_FORMAT_ISL ||
547 isl_int_is_zero(div->row[i][0]))
548 continue;
549 p = isl_printer_print_str(p, " = ");
550 p = print_div(space, div, i, p);
553 return p;
556 static __isl_give isl_printer *print_disjunct(__isl_keep isl_basic_map *bmap,
557 __isl_keep isl_space *dim, __isl_take isl_printer *p, int latex)
559 if (bmap->n_div > 0) {
560 isl_space *space;
561 isl_mat *div;
563 space = isl_basic_map_get_space(bmap);
564 div = isl_basic_map_get_divs(bmap);
565 p = isl_printer_print_str(p, s_open_exists[latex]);
566 p = print_div_list(p, space, div, latex);
567 isl_space_free(space);
568 isl_mat_free(div);
569 p = isl_printer_print_str(p, ": ");
572 p = print_constraints(bmap, dim, p, latex);
574 if (bmap->n_div > 0)
575 p = isl_printer_print_str(p, s_close_exists[latex]);
576 return p;
579 /* Print a colon followed by the constraints of "bmap"
580 * to "p", provided there are any constraints.
581 * The names of the variables are taken from "space".
582 * "latex" is set if the constraints should be printed in LaTeX format.
584 static __isl_give isl_printer *print_optional_disjunct(
585 __isl_keep isl_basic_map *bmap, __isl_keep isl_space *space,
586 __isl_take isl_printer *p, int latex)
588 if (isl_basic_map_is_universe(bmap))
589 return p;
591 p = isl_printer_print_str(p, ": ");
592 p = print_disjunct(bmap, space, p, latex);
594 return p;
597 static __isl_give isl_printer *basic_map_print_omega(
598 __isl_keep isl_basic_map *bmap, __isl_take isl_printer *p)
600 p = isl_printer_print_str(p, "{ [");
601 p = print_var_list(p, bmap->dim, isl_dim_in);
602 p = isl_printer_print_str(p, "] -> [");
603 p = print_var_list(p, bmap->dim, isl_dim_out);
604 p = isl_printer_print_str(p, "] ");
605 p = print_optional_disjunct(bmap, bmap->dim, p, 0);
606 p = isl_printer_print_str(p, " }");
607 return p;
610 static __isl_give isl_printer *basic_set_print_omega(
611 __isl_keep isl_basic_set *bset, __isl_take isl_printer *p)
613 p = isl_printer_print_str(p, "{ [");
614 p = print_var_list(p, bset->dim, isl_dim_set);
615 p = isl_printer_print_str(p, "] ");
616 p = print_optional_disjunct(bset, bset->dim, p, 0);
617 p = isl_printer_print_str(p, " }");
618 return p;
621 static __isl_give isl_printer *isl_map_print_omega(__isl_keep isl_map *map,
622 __isl_take isl_printer *p)
624 int i;
626 for (i = 0; i < map->n; ++i) {
627 if (i)
628 p = isl_printer_print_str(p, " union ");
629 p = basic_map_print_omega(map->p[i], p);
631 return p;
634 static __isl_give isl_printer *isl_set_print_omega(__isl_keep isl_set *set,
635 __isl_take isl_printer *p)
637 int i;
639 for (i = 0; i < set->n; ++i) {
640 if (i)
641 p = isl_printer_print_str(p, " union ");
642 p = basic_set_print_omega(set->p[i], p);
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 latex)
651 struct isl_print_space_data data = { .latex = latex };
652 int rational = ISL_F_ISSET(bmap, ISL_BASIC_MAP_RATIONAL);
654 if (isl_basic_map_dim(bmap, isl_dim_param) > 0) {
655 p = print_tuple(bmap->dim, p, isl_dim_param, &data);
656 p = isl_printer_print_str(p, " -> ");
658 p = isl_printer_print_str(p, "{ ");
659 p = print_space(bmap->dim, p, rational, &data);
660 p = isl_printer_print_str(p, " : ");
661 p = print_disjunct(bmap, bmap->dim, p, latex);
662 p = isl_printer_print_str(p, " }");
663 return p;
666 static __isl_give isl_printer *print_disjuncts(__isl_keep isl_map *map,
667 __isl_take isl_printer *p, int latex)
669 int i;
671 if (isl_map_plain_is_universe(map))
672 return p;
674 p = isl_printer_print_str(p, s_such_that[latex]);
675 if (map->n == 0)
676 p = isl_printer_print_str(p, "1 = 0");
677 for (i = 0; i < map->n; ++i) {
678 if (i)
679 p = isl_printer_print_str(p, s_or[latex]);
680 if (map->n > 1 && map->p[i]->n_eq + map->p[i]->n_ineq > 1)
681 p = isl_printer_print_str(p, "(");
682 p = print_disjunct(map->p[i], map->dim, p, latex);
683 if (map->n > 1 && map->p[i]->n_eq + map->p[i]->n_ineq > 1)
684 p = isl_printer_print_str(p, ")");
686 return p;
689 /* Print the disjuncts of a map (or set).
690 * If the map turns out to be a universal parameter domain, then
691 * we need to print the colon. Otherwise, the output looks identical
692 * to the empty set.
694 static __isl_give isl_printer *print_disjuncts_map(__isl_keep isl_map *map,
695 __isl_take isl_printer *p, int latex)
697 if (isl_map_plain_is_universe(map) && isl_space_is_params(map->dim))
698 return isl_printer_print_str(p, s_such_that[latex]);
699 else
700 return print_disjuncts(map, p, latex);
703 struct isl_aff_split {
704 isl_basic_map *aff;
705 isl_map *map;
708 static void free_split(__isl_take struct isl_aff_split *split, int n)
710 int i;
712 if (!split)
713 return;
715 for (i = 0; i < n; ++i) {
716 isl_basic_map_free(split[i].aff);
717 isl_map_free(split[i].map);
720 free(split);
723 static __isl_give isl_basic_map *get_aff(__isl_take isl_basic_map *bmap)
725 int i, j;
726 unsigned nparam, n_in, n_out, total;
728 bmap = isl_basic_map_cow(bmap);
729 if (!bmap)
730 return NULL;
731 if (isl_basic_map_free_inequality(bmap, bmap->n_ineq) < 0)
732 goto error;
734 nparam = isl_basic_map_dim(bmap, isl_dim_param);
735 n_in = isl_basic_map_dim(bmap, isl_dim_in);
736 n_out = isl_basic_map_dim(bmap, isl_dim_out);
737 total = isl_basic_map_dim(bmap, isl_dim_all);
738 for (i = bmap->n_eq - 1; i >= 0; --i) {
739 j = isl_seq_last_non_zero(bmap->eq[i] + 1, total);
740 if (j >= nparam && j < nparam + n_in + n_out &&
741 (isl_int_is_one(bmap->eq[i][1 + j]) ||
742 isl_int_is_negone(bmap->eq[i][1 + j])))
743 continue;
744 if (isl_basic_map_drop_equality(bmap, i) < 0)
745 goto error;
748 bmap = isl_basic_map_finalize(bmap);
750 return bmap;
751 error:
752 isl_basic_map_free(bmap);
753 return NULL;
756 static int aff_split_cmp(const void *p1, const void *p2, void *user)
758 const struct isl_aff_split *s1, *s2;
759 s1 = (const struct isl_aff_split *) p1;
760 s2 = (const struct isl_aff_split *) p2;
762 return isl_basic_map_plain_cmp(s1->aff, s2->aff);
765 static __isl_give isl_basic_map *drop_aff(__isl_take isl_basic_map *bmap,
766 __isl_keep isl_basic_map *aff)
768 int i, j;
769 unsigned total;
771 if (!bmap || !aff)
772 goto error;
774 total = isl_space_dim(bmap->dim, isl_dim_all);
776 for (i = bmap->n_eq - 1; i >= 0; --i) {
777 if (isl_seq_first_non_zero(bmap->eq[i] + 1 + total,
778 bmap->n_div) != -1)
779 continue;
780 for (j = 0; j < aff->n_eq; ++j) {
781 if (!isl_seq_eq(bmap->eq[i], aff->eq[j], 1 + total) &&
782 !isl_seq_is_neg(bmap->eq[i], aff->eq[j], 1 + total))
783 continue;
784 if (isl_basic_map_drop_equality(bmap, i) < 0)
785 goto error;
786 break;
790 return bmap;
791 error:
792 isl_basic_map_free(bmap);
793 return NULL;
796 static __isl_give struct isl_aff_split *split_aff(__isl_keep isl_map *map)
798 int i, n;
799 struct isl_aff_split *split;
800 isl_ctx *ctx;
802 ctx = isl_map_get_ctx(map);
803 split = isl_calloc_array(ctx, struct isl_aff_split, map->n);
804 if (!split)
805 return NULL;
807 for (i = 0; i < map->n; ++i) {
808 isl_basic_map *bmap;
809 split[i].aff = get_aff(isl_basic_map_copy(map->p[i]));
810 bmap = isl_basic_map_copy(map->p[i]);
811 bmap = isl_basic_map_cow(bmap);
812 bmap = drop_aff(bmap, split[i].aff);
813 split[i].map = isl_map_from_basic_map(bmap);
814 if (!split[i].aff || !split[i].map)
815 goto error;
818 if (isl_sort(split, map->n, sizeof(struct isl_aff_split),
819 &aff_split_cmp, NULL) < 0)
820 goto error;
822 n = map->n;
823 for (i = n - 1; i >= 1; --i) {
824 if (!isl_basic_map_plain_is_equal(split[i - 1].aff,
825 split[i].aff))
826 continue;
827 isl_basic_map_free(split[i].aff);
828 split[i - 1].map = isl_map_union(split[i - 1].map,
829 split[i].map);
830 if (i != n - 1)
831 split[i] = split[n - 1];
832 split[n - 1].aff = NULL;
833 split[n - 1].map = NULL;
834 --n;
837 return split;
838 error:
839 free_split(split, map->n);
840 return NULL;
843 static int defining_equality(__isl_keep isl_basic_map *eq,
844 __isl_keep isl_space *dim, enum isl_dim_type type, int pos)
846 int i;
847 unsigned total;
849 if (!eq)
850 return -1;
852 pos += isl_space_offset(dim, type);
853 total = isl_basic_map_total_dim(eq);
855 for (i = 0; i < eq->n_eq; ++i) {
856 if (isl_seq_last_non_zero(eq->eq[i] + 1, total) != pos)
857 continue;
858 if (isl_int_is_one(eq->eq[i][1 + pos]))
859 isl_seq_neg(eq->eq[i], eq->eq[i], 1 + total);
860 return i;
863 return -1;
866 /* Print dimension "pos" of data->space to "p".
868 * data->user is assumed to be an isl_basic_map keeping track of equalities.
870 * If the current dimension is defined by these equalities, then print
871 * the corresponding expression. Otherwise, print the name of the dimension.
873 static __isl_give isl_printer *print_dim_eq(__isl_take isl_printer *p,
874 struct isl_print_space_data *data, unsigned pos)
876 isl_basic_map *eq = data->user;
877 int j;
879 j = defining_equality(eq, data->space, data->type, pos);
880 if (j >= 0) {
881 pos += 1 + isl_space_offset(data->space, data->type);
882 p = print_affine_of_len(eq->dim, NULL, p, eq->eq[j], pos);
883 } else {
884 p = print_name(data->space, p, data->type, pos, data->latex);
887 return p;
890 static __isl_give isl_printer *print_split_map(__isl_take isl_printer *p,
891 struct isl_aff_split *split, int n)
893 struct isl_print_space_data data = { 0 };
894 int i;
895 int rational;
897 data.print_dim = &print_dim_eq;
898 for (i = 0; i < n; ++i) {
899 isl_space *dim;
901 if (!split[i].map)
902 break;
903 dim = split[i].map->dim;
904 rational = split[i].map->n > 0 &&
905 ISL_F_ISSET(split[i].map->p[0], ISL_BASIC_MAP_RATIONAL);
906 if (i)
907 p = isl_printer_print_str(p, "; ");
908 data.user = split[i].aff;
909 p = print_space(dim, p, rational, &data);
910 p = print_disjuncts_map(split[i].map, p, 0);
913 return p;
916 static __isl_give isl_printer *isl_map_print_isl_body(__isl_keep isl_map *map,
917 __isl_take isl_printer *p)
919 struct isl_print_space_data data = { 0 };
920 struct isl_aff_split *split = NULL;
921 int rational;
923 if (map->n > 0)
924 split = split_aff(map);
925 if (split) {
926 p = print_split_map(p, split, map->n);
927 } else {
928 rational = map->n > 0 &&
929 ISL_F_ISSET(map->p[0], ISL_BASIC_MAP_RATIONAL);
930 p = print_space(map->dim, p, rational, &data);
931 p = print_disjuncts_map(map, p, 0);
933 free_split(split, map->n);
934 return p;
937 static __isl_give isl_printer *isl_map_print_isl(__isl_keep isl_map *map,
938 __isl_take isl_printer *p)
940 struct isl_print_space_data data = { 0 };
942 if (isl_map_dim(map, isl_dim_param) > 0) {
943 p = print_tuple(map->dim, p, isl_dim_param, &data);
944 p = isl_printer_print_str(p, s_to[0]);
946 p = isl_printer_print_str(p, s_open_set[0]);
947 p = isl_map_print_isl_body(map, p);
948 p = isl_printer_print_str(p, s_close_set[0]);
949 return p;
952 static __isl_give isl_printer *print_latex_map(__isl_keep isl_map *map,
953 __isl_take isl_printer *p, __isl_keep isl_basic_map *aff)
955 struct isl_print_space_data data = { 0 };
957 data.latex = 1;
958 if (isl_map_dim(map, isl_dim_param) > 0) {
959 p = print_tuple(map->dim, p, isl_dim_param, &data);
960 p = isl_printer_print_str(p, s_to[1]);
962 p = isl_printer_print_str(p, s_open_set[1]);
963 data.print_dim = &print_dim_eq;
964 data.user = aff;
965 p = print_space(map->dim, p, 0, &data);
966 p = print_disjuncts_map(map, p, 1);
967 p = isl_printer_print_str(p, s_close_set[1]);
969 return p;
972 static __isl_give isl_printer *isl_map_print_latex(__isl_keep isl_map *map,
973 __isl_take isl_printer *p)
975 int i;
976 struct isl_aff_split *split = NULL;
978 if (map->n > 0)
979 split = split_aff(map);
981 if (!split)
982 return print_latex_map(map, p, NULL);
984 for (i = 0; i < map->n; ++i) {
985 if (!split[i].map)
986 break;
987 if (i)
988 p = isl_printer_print_str(p, " \\cup ");
989 p = print_latex_map(split[i].map, p, split[i].aff);
992 free_split(split, map->n);
993 return p;
996 __isl_give isl_printer *isl_printer_print_basic_map(__isl_take isl_printer *p,
997 __isl_keep isl_basic_map *bmap)
999 if (!p || !bmap)
1000 goto error;
1001 if (p->output_format == ISL_FORMAT_ISL)
1002 return isl_basic_map_print_isl(bmap, p, 0);
1003 else if (p->output_format == ISL_FORMAT_OMEGA)
1004 return basic_map_print_omega(bmap, p);
1005 isl_assert(bmap->ctx, 0, goto error);
1006 error:
1007 isl_printer_free(p);
1008 return NULL;
1011 void isl_basic_map_print(__isl_keep isl_basic_map *bmap, FILE *out, int indent,
1012 const char *prefix, const char *suffix, unsigned output_format)
1014 isl_printer *printer;
1016 if (!bmap)
1017 return;
1019 printer = isl_printer_to_file(bmap->ctx, out);
1020 printer = isl_printer_set_indent(printer, indent);
1021 printer = isl_printer_set_prefix(printer, prefix);
1022 printer = isl_printer_set_suffix(printer, suffix);
1023 printer = isl_printer_set_output_format(printer, output_format);
1024 isl_printer_print_basic_map(printer, bmap);
1026 isl_printer_free(printer);
1029 __isl_give isl_printer *isl_printer_print_basic_set(__isl_take isl_printer *p,
1030 __isl_keep isl_basic_set *bset)
1032 if (!p || !bset)
1033 goto error;
1035 if (p->output_format == ISL_FORMAT_ISL)
1036 return isl_basic_map_print_isl(bset, p, 0);
1037 else if (p->output_format == ISL_FORMAT_POLYLIB)
1038 return isl_basic_set_print_polylib(bset, p, 0);
1039 else if (p->output_format == ISL_FORMAT_EXT_POLYLIB)
1040 return isl_basic_set_print_polylib(bset, p, 1);
1041 else if (p->output_format == ISL_FORMAT_POLYLIB_CONSTRAINTS)
1042 return bset_print_constraints_polylib(bset, p);
1043 else if (p->output_format == ISL_FORMAT_OMEGA)
1044 return basic_set_print_omega(bset, p);
1045 isl_assert(p->ctx, 0, goto error);
1046 error:
1047 isl_printer_free(p);
1048 return NULL;
1051 void isl_basic_set_print(struct isl_basic_set *bset, FILE *out, int indent,
1052 const char *prefix, const char *suffix, unsigned output_format)
1054 isl_printer *printer;
1056 if (!bset)
1057 return;
1059 printer = isl_printer_to_file(bset->ctx, out);
1060 printer = isl_printer_set_indent(printer, indent);
1061 printer = isl_printer_set_prefix(printer, prefix);
1062 printer = isl_printer_set_suffix(printer, suffix);
1063 printer = isl_printer_set_output_format(printer, output_format);
1064 isl_printer_print_basic_set(printer, bset);
1066 isl_printer_free(printer);
1069 __isl_give isl_printer *isl_printer_print_set(__isl_take isl_printer *p,
1070 __isl_keep isl_set *set)
1072 if (!p || !set)
1073 goto error;
1074 if (p->output_format == ISL_FORMAT_ISL)
1075 return isl_map_print_isl((isl_map *)set, p);
1076 else if (p->output_format == ISL_FORMAT_POLYLIB)
1077 return isl_set_print_polylib(set, p, 0);
1078 else if (p->output_format == ISL_FORMAT_EXT_POLYLIB)
1079 return isl_set_print_polylib(set, p, 1);
1080 else if (p->output_format == ISL_FORMAT_OMEGA)
1081 return isl_set_print_omega(set, p);
1082 else if (p->output_format == ISL_FORMAT_LATEX)
1083 return isl_map_print_latex((isl_map *)set, p);
1084 isl_assert(set->ctx, 0, goto error);
1085 error:
1086 isl_printer_free(p);
1087 return NULL;
1090 void isl_set_print(struct isl_set *set, FILE *out, int indent,
1091 unsigned output_format)
1093 isl_printer *printer;
1095 if (!set)
1096 return;
1098 printer = isl_printer_to_file(set->ctx, out);
1099 printer = isl_printer_set_indent(printer, indent);
1100 printer = isl_printer_set_output_format(printer, output_format);
1101 printer = isl_printer_print_set(printer, set);
1103 isl_printer_free(printer);
1106 __isl_give isl_printer *isl_printer_print_map(__isl_take isl_printer *p,
1107 __isl_keep isl_map *map)
1109 if (!p || !map)
1110 goto error;
1112 if (p->output_format == ISL_FORMAT_ISL)
1113 return isl_map_print_isl(map, p);
1114 else if (p->output_format == ISL_FORMAT_POLYLIB)
1115 return isl_map_print_polylib(map, p, 0);
1116 else if (p->output_format == ISL_FORMAT_EXT_POLYLIB)
1117 return isl_map_print_polylib(map, p, 1);
1118 else if (p->output_format == ISL_FORMAT_OMEGA)
1119 return isl_map_print_omega(map, p);
1120 else if (p->output_format == ISL_FORMAT_LATEX)
1121 return isl_map_print_latex(map, p);
1122 isl_assert(map->ctx, 0, goto error);
1123 error:
1124 isl_printer_free(p);
1125 return NULL;
1128 struct isl_union_print_data {
1129 isl_printer *p;
1130 int first;
1133 static int print_map_body(__isl_take isl_map *map, void *user)
1135 struct isl_union_print_data *data;
1136 data = (struct isl_union_print_data *)user;
1138 if (!data->first)
1139 data->p = isl_printer_print_str(data->p, "; ");
1140 data->first = 0;
1142 data->p = isl_map_print_isl_body(map, data->p);
1143 isl_map_free(map);
1145 return 0;
1148 static __isl_give isl_printer *isl_union_map_print_isl(
1149 __isl_keep isl_union_map *umap, __isl_take isl_printer *p)
1151 struct isl_union_print_data data = { p, 1 };
1152 struct isl_print_space_data space_data = { 0 };
1153 isl_space *dim;
1155 dim = isl_union_map_get_space(umap);
1156 if (isl_space_dim(dim, isl_dim_param) > 0) {
1157 p = print_tuple(dim, p, isl_dim_param, &space_data);
1158 p = isl_printer_print_str(p, s_to[0]);
1160 isl_space_free(dim);
1161 p = isl_printer_print_str(p, s_open_set[0]);
1162 isl_union_map_foreach_map(umap, &print_map_body, &data);
1163 p = data.p;
1164 p = isl_printer_print_str(p, s_close_set[0]);
1165 return p;
1168 static int print_latex_map_body(__isl_take isl_map *map, void *user)
1170 struct isl_union_print_data *data;
1171 data = (struct isl_union_print_data *)user;
1173 if (!data->first)
1174 data->p = isl_printer_print_str(data->p, " \\cup ");
1175 data->first = 0;
1177 data->p = isl_map_print_latex(map, data->p);
1178 isl_map_free(map);
1180 return 0;
1183 static __isl_give isl_printer *isl_union_map_print_latex(
1184 __isl_keep isl_union_map *umap, __isl_take isl_printer *p)
1186 struct isl_union_print_data data = { p, 1 };
1187 isl_union_map_foreach_map(umap, &print_latex_map_body, &data);
1188 p = data.p;
1189 return p;
1192 __isl_give isl_printer *isl_printer_print_union_map(__isl_take isl_printer *p,
1193 __isl_keep isl_union_map *umap)
1195 if (!p || !umap)
1196 goto error;
1198 if (p->output_format == ISL_FORMAT_ISL)
1199 return isl_union_map_print_isl(umap, p);
1200 if (p->output_format == ISL_FORMAT_LATEX)
1201 return isl_union_map_print_latex(umap, p);
1203 isl_die(p->ctx, isl_error_invalid,
1204 "invalid output format for isl_union_map", goto error);
1205 error:
1206 isl_printer_free(p);
1207 return NULL;
1210 __isl_give isl_printer *isl_printer_print_union_set(__isl_take isl_printer *p,
1211 __isl_keep isl_union_set *uset)
1213 if (!p || !uset)
1214 goto error;
1216 if (p->output_format == ISL_FORMAT_ISL)
1217 return isl_union_map_print_isl((isl_union_map *)uset, p);
1218 if (p->output_format == ISL_FORMAT_LATEX)
1219 return isl_union_map_print_latex((isl_union_map *)uset, p);
1221 isl_die(p->ctx, isl_error_invalid,
1222 "invalid output format for isl_union_set", goto error);
1223 error:
1224 isl_printer_free(p);
1225 return NULL;
1228 void isl_map_print(__isl_keep isl_map *map, FILE *out, int indent,
1229 unsigned output_format)
1231 isl_printer *printer;
1233 if (!map)
1234 return;
1236 printer = isl_printer_to_file(map->ctx, out);
1237 printer = isl_printer_set_indent(printer, indent);
1238 printer = isl_printer_set_output_format(printer, output_format);
1239 printer = isl_printer_print_map(printer, map);
1241 isl_printer_free(printer);
1244 static int upoly_rec_n_non_zero(__isl_keep struct isl_upoly_rec *rec)
1246 int i;
1247 int n;
1249 for (i = 0, n = 0; i < rec->n; ++i)
1250 if (!isl_upoly_is_zero(rec->p[i]))
1251 ++n;
1253 return n;
1256 static __isl_give isl_printer *upoly_print_cst(__isl_keep struct isl_upoly *up,
1257 __isl_take isl_printer *p, int first)
1259 struct isl_upoly_cst *cst;
1260 int neg;
1262 cst = isl_upoly_as_cst(up);
1263 if (!cst)
1264 goto error;
1265 neg = !first && isl_int_is_neg(cst->n);
1266 if (!first)
1267 p = isl_printer_print_str(p, neg ? " - " : " + ");
1268 if (neg)
1269 isl_int_neg(cst->n, cst->n);
1270 if (isl_int_is_zero(cst->d)) {
1271 int sgn = isl_int_sgn(cst->n);
1272 p = isl_printer_print_str(p, sgn < 0 ? "-infty" :
1273 sgn == 0 ? "NaN" : "infty");
1274 } else
1275 p = isl_printer_print_isl_int(p, cst->n);
1276 if (neg)
1277 isl_int_neg(cst->n, cst->n);
1278 if (!isl_int_is_zero(cst->d) && !isl_int_is_one(cst->d)) {
1279 p = isl_printer_print_str(p, "/");
1280 p = isl_printer_print_isl_int(p, cst->d);
1282 return p;
1283 error:
1284 isl_printer_free(p);
1285 return NULL;
1288 static __isl_give isl_printer *print_base(__isl_take isl_printer *p,
1289 __isl_keep isl_space *dim, __isl_keep isl_mat *div, int var)
1291 unsigned total;
1293 total = isl_space_dim(dim, isl_dim_all);
1294 if (var < total)
1295 p = print_term(dim, NULL, dim->ctx->one, 1 + var, p, 0);
1296 else
1297 p = print_div(dim, div, var - total, p);
1298 return p;
1301 static __isl_give isl_printer *print_pow(__isl_take isl_printer *p,
1302 __isl_keep isl_space *dim, __isl_keep isl_mat *div, int var, int exp)
1304 p = print_base(p, dim, div, var);
1305 if (exp == 1)
1306 return p;
1307 if (p->output_format == ISL_FORMAT_C) {
1308 int i;
1309 for (i = 1; i < exp; ++i) {
1310 p = isl_printer_print_str(p, "*");
1311 p = print_base(p, dim, div, var);
1313 } else {
1314 p = isl_printer_print_str(p, "^");
1315 p = isl_printer_print_int(p, exp);
1317 return p;
1320 static __isl_give isl_printer *upoly_print(__isl_keep struct isl_upoly *up,
1321 __isl_keep isl_space *dim, __isl_keep isl_mat *div,
1322 __isl_take isl_printer *p, int outer)
1324 int i, n, first, print_parens;
1325 struct isl_upoly_rec *rec;
1327 if (!p || !up || !dim || !div)
1328 goto error;
1330 if (isl_upoly_is_cst(up))
1331 return upoly_print_cst(up, p, 1);
1333 rec = isl_upoly_as_rec(up);
1334 if (!rec)
1335 goto error;
1336 n = upoly_rec_n_non_zero(rec);
1337 print_parens = n > 1 ||
1338 (outer && rec->up.var >= isl_space_dim(dim, isl_dim_all));
1339 if (print_parens)
1340 p = isl_printer_print_str(p, "(");
1341 for (i = 0, first = 1; i < rec->n; ++i) {
1342 if (isl_upoly_is_zero(rec->p[i]))
1343 continue;
1344 if (isl_upoly_is_negone(rec->p[i])) {
1345 if (!i)
1346 p = isl_printer_print_str(p, "-1");
1347 else if (first)
1348 p = isl_printer_print_str(p, "-");
1349 else
1350 p = isl_printer_print_str(p, " - ");
1351 } else if (isl_upoly_is_cst(rec->p[i]) &&
1352 !isl_upoly_is_one(rec->p[i]))
1353 p = upoly_print_cst(rec->p[i], p, first);
1354 else {
1355 if (!first)
1356 p = isl_printer_print_str(p, " + ");
1357 if (i == 0 || !isl_upoly_is_one(rec->p[i]))
1358 p = upoly_print(rec->p[i], dim, div, p, 0);
1360 first = 0;
1361 if (i == 0)
1362 continue;
1363 if (!isl_upoly_is_one(rec->p[i]) &&
1364 !isl_upoly_is_negone(rec->p[i]))
1365 p = isl_printer_print_str(p, " * ");
1366 p = print_pow(p, dim, div, rec->up.var, i);
1368 if (print_parens)
1369 p = isl_printer_print_str(p, ")");
1370 return p;
1371 error:
1372 isl_printer_free(p);
1373 return NULL;
1376 static __isl_give isl_printer *print_qpolynomial(__isl_take isl_printer *p,
1377 __isl_keep isl_qpolynomial *qp)
1379 if (!p || !qp)
1380 goto error;
1381 p = upoly_print(qp->upoly, qp->dim, qp->div, p, 1);
1382 return p;
1383 error:
1384 isl_printer_free(p);
1385 return NULL;
1388 static __isl_give isl_printer *print_qpolynomial_isl(__isl_take isl_printer *p,
1389 __isl_keep isl_qpolynomial *qp)
1391 struct isl_print_space_data data = { 0 };
1393 if (!p || !qp)
1394 goto error;
1396 if (isl_space_dim(qp->dim, isl_dim_param) > 0) {
1397 p = print_tuple(qp->dim, p, isl_dim_param, &data);
1398 p = isl_printer_print_str(p, " -> ");
1400 p = isl_printer_print_str(p, "{ ");
1401 if (!isl_space_is_params(qp->dim)) {
1402 p = print_space(qp->dim, p, 0, &data);
1403 p = isl_printer_print_str(p, " -> ");
1405 p = print_qpolynomial(p, qp);
1406 p = isl_printer_print_str(p, " }");
1407 return p;
1408 error:
1409 isl_printer_free(p);
1410 return NULL;
1413 static __isl_give isl_printer *print_qpolynomial_c(__isl_take isl_printer *p,
1414 __isl_keep isl_space *dim, __isl_keep isl_qpolynomial *qp)
1416 isl_int den;
1418 isl_int_init(den);
1419 isl_qpolynomial_get_den(qp, &den);
1420 if (!isl_int_is_one(den)) {
1421 isl_qpolynomial *f;
1422 p = isl_printer_print_str(p, "(");
1423 qp = isl_qpolynomial_copy(qp);
1424 f = isl_qpolynomial_rat_cst_on_domain(isl_space_copy(qp->dim),
1425 den, qp->dim->ctx->one);
1426 qp = isl_qpolynomial_mul(qp, f);
1428 if (qp)
1429 p = upoly_print(qp->upoly, dim, qp->div, p, 0);
1430 else
1431 p = isl_printer_free(p);
1432 if (!isl_int_is_one(den)) {
1433 p = isl_printer_print_str(p, ")/");
1434 p = isl_printer_print_isl_int(p, den);
1435 isl_qpolynomial_free(qp);
1437 isl_int_clear(den);
1438 return p;
1441 __isl_give isl_printer *isl_printer_print_qpolynomial(
1442 __isl_take isl_printer *p, __isl_keep isl_qpolynomial *qp)
1444 if (!p || !qp)
1445 goto error;
1447 if (p->output_format == ISL_FORMAT_ISL)
1448 return print_qpolynomial_isl(p, qp);
1449 else if (p->output_format == ISL_FORMAT_C)
1450 return print_qpolynomial_c(p, qp->dim, qp);
1451 else
1452 isl_die(qp->dim->ctx, isl_error_unsupported,
1453 "output format not supported for isl_qpolynomials",
1454 goto error);
1455 error:
1456 isl_printer_free(p);
1457 return NULL;
1460 void isl_qpolynomial_print(__isl_keep isl_qpolynomial *qp, FILE *out,
1461 unsigned output_format)
1463 isl_printer *p;
1465 if (!qp)
1466 return;
1468 isl_assert(qp->dim->ctx, output_format == ISL_FORMAT_ISL, return);
1469 p = isl_printer_to_file(qp->dim->ctx, out);
1470 p = isl_printer_print_qpolynomial(p, qp);
1471 isl_printer_free(p);
1474 static __isl_give isl_printer *qpolynomial_fold_print(
1475 __isl_keep isl_qpolynomial_fold *fold, __isl_take isl_printer *p)
1477 int i;
1479 if (fold->type == isl_fold_min)
1480 p = isl_printer_print_str(p, "min");
1481 else if (fold->type == isl_fold_max)
1482 p = isl_printer_print_str(p, "max");
1483 p = isl_printer_print_str(p, "(");
1484 for (i = 0; i < fold->n; ++i) {
1485 if (i)
1486 p = isl_printer_print_str(p, ", ");
1487 p = print_qpolynomial(p, fold->qp[i]);
1489 p = isl_printer_print_str(p, ")");
1490 return p;
1493 void isl_qpolynomial_fold_print(__isl_keep isl_qpolynomial_fold *fold,
1494 FILE *out, unsigned output_format)
1496 isl_printer *p;
1498 if (!fold)
1499 return;
1501 isl_assert(fold->dim->ctx, output_format == ISL_FORMAT_ISL, return);
1503 p = isl_printer_to_file(fold->dim->ctx, out);
1504 p = isl_printer_print_qpolynomial_fold(p, fold);
1506 isl_printer_free(p);
1509 static __isl_give isl_printer *isl_pwqp_print_isl_body(
1510 __isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial *pwqp)
1512 struct isl_print_space_data data = { 0 };
1513 int i = 0;
1515 for (i = 0; i < pwqp->n; ++i) {
1516 if (i)
1517 p = isl_printer_print_str(p, "; ");
1518 if (!isl_space_is_params(pwqp->p[i].set->dim)) {
1519 p = print_space(pwqp->p[i].set->dim, p, 0, &data);
1520 p = isl_printer_print_str(p, " -> ");
1522 p = print_qpolynomial(p, pwqp->p[i].qp);
1523 p = print_disjuncts((isl_map *)pwqp->p[i].set, p, 0);
1526 return p;
1529 static __isl_give isl_printer *print_pw_qpolynomial_isl(
1530 __isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial *pwqp)
1532 struct isl_print_space_data data = { 0 };
1534 if (!p || !pwqp)
1535 goto error;
1537 if (isl_space_dim(pwqp->dim, isl_dim_param) > 0) {
1538 p = print_tuple(pwqp->dim, p, isl_dim_param, &data);
1539 p = isl_printer_print_str(p, " -> ");
1541 p = isl_printer_print_str(p, "{ ");
1542 if (pwqp->n == 0) {
1543 if (!isl_space_is_set(pwqp->dim)) {
1544 p = print_tuple(pwqp->dim, p, isl_dim_in, &data);
1545 p = isl_printer_print_str(p, " -> ");
1547 p = isl_printer_print_str(p, "0");
1549 p = isl_pwqp_print_isl_body(p, pwqp);
1550 p = isl_printer_print_str(p, " }");
1551 return p;
1552 error:
1553 isl_printer_free(p);
1554 return NULL;
1557 void isl_pw_qpolynomial_print(__isl_keep isl_pw_qpolynomial *pwqp, FILE *out,
1558 unsigned output_format)
1560 isl_printer *p;
1562 if (!pwqp)
1563 return;
1565 p = isl_printer_to_file(pwqp->dim->ctx, out);
1566 p = isl_printer_set_output_format(p, output_format);
1567 p = isl_printer_print_pw_qpolynomial(p, pwqp);
1569 isl_printer_free(p);
1572 static __isl_give isl_printer *isl_pwf_print_isl_body(
1573 __isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial_fold *pwf)
1575 struct isl_print_space_data data = { 0 };
1576 int i = 0;
1578 for (i = 0; i < pwf->n; ++i) {
1579 if (i)
1580 p = isl_printer_print_str(p, "; ");
1581 if (!isl_space_is_params(pwf->p[i].set->dim)) {
1582 p = print_space(pwf->p[i].set->dim, p, 0, &data);
1583 p = isl_printer_print_str(p, " -> ");
1585 p = qpolynomial_fold_print(pwf->p[i].fold, p);
1586 p = print_disjuncts((isl_map *)pwf->p[i].set, p, 0);
1589 return p;
1592 static __isl_give isl_printer *print_pw_qpolynomial_fold_isl(
1593 __isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial_fold *pwf)
1595 struct isl_print_space_data data = { 0 };
1597 if (isl_space_dim(pwf->dim, isl_dim_param) > 0) {
1598 p = print_tuple(pwf->dim, p, isl_dim_param, &data);
1599 p = isl_printer_print_str(p, " -> ");
1601 p = isl_printer_print_str(p, "{ ");
1602 if (pwf->n == 0) {
1603 if (!isl_space_is_set(pwf->dim)) {
1604 p = print_tuple(pwf->dim, p, isl_dim_in, &data);
1605 p = isl_printer_print_str(p, " -> ");
1607 p = isl_printer_print_str(p, "0");
1609 p = isl_pwf_print_isl_body(p, pwf);
1610 p = isl_printer_print_str(p, " }");
1611 return p;
1614 static __isl_give isl_printer *print_affine_c(__isl_take isl_printer *p,
1615 __isl_keep isl_space *dim, __isl_keep isl_basic_set *bset, isl_int *c);
1617 static __isl_give isl_printer *print_name_c(__isl_take isl_printer *p,
1618 __isl_keep isl_space *dim,
1619 __isl_keep isl_basic_set *bset, enum isl_dim_type type, unsigned pos)
1621 if (type == isl_dim_div) {
1622 p = isl_printer_print_str(p, "floord(");
1623 p = print_affine_c(p, dim, bset, bset->div[pos] + 1);
1624 p = isl_printer_print_str(p, ", ");
1625 p = isl_printer_print_isl_int(p, bset->div[pos][0]);
1626 p = isl_printer_print_str(p, ")");
1627 } else {
1628 const char *name;
1630 name = isl_space_get_dim_name(dim, type, pos);
1631 if (!name)
1632 name = "UNNAMED";
1633 p = isl_printer_print_str(p, name);
1635 return p;
1638 static __isl_give isl_printer *print_term_c(__isl_take isl_printer *p,
1639 __isl_keep isl_space *dim,
1640 __isl_keep isl_basic_set *bset, isl_int c, unsigned pos)
1642 enum isl_dim_type type;
1644 if (pos == 0)
1645 return isl_printer_print_isl_int(p, c);
1647 if (isl_int_is_one(c))
1649 else if (isl_int_is_negone(c))
1650 p = isl_printer_print_str(p, "-");
1651 else {
1652 p = isl_printer_print_isl_int(p, c);
1653 p = isl_printer_print_str(p, "*");
1655 type = pos2type(dim, &pos);
1656 p = print_name_c(p, dim, bset, type, pos);
1657 return p;
1660 static __isl_give isl_printer *print_partial_affine_c(__isl_take isl_printer *p,
1661 __isl_keep isl_space *dim,
1662 __isl_keep isl_basic_set *bset, isl_int *c, unsigned len)
1664 int i;
1665 int first;
1667 for (i = 0, first = 1; i < len; ++i) {
1668 int flip = 0;
1669 if (isl_int_is_zero(c[i]))
1670 continue;
1671 if (!first) {
1672 if (isl_int_is_neg(c[i])) {
1673 flip = 1;
1674 isl_int_neg(c[i], c[i]);
1675 p = isl_printer_print_str(p, " - ");
1676 } else
1677 p = isl_printer_print_str(p, " + ");
1679 first = 0;
1680 p = print_term_c(p, dim, bset, c[i], i);
1681 if (flip)
1682 isl_int_neg(c[i], c[i]);
1684 if (first)
1685 p = isl_printer_print_str(p, "0");
1686 return p;
1689 static __isl_give isl_printer *print_affine_c(__isl_take isl_printer *p,
1690 __isl_keep isl_space *dim, __isl_keep isl_basic_set *bset, isl_int *c)
1692 unsigned len = 1 + isl_basic_set_total_dim(bset);
1693 return print_partial_affine_c(p, dim, bset, c, len);
1696 /* We skip the constraint if it is implied by the div expression.
1698 * *first indicates whether this is the first constraint in the conjunction and
1699 * is updated if the constraint is actually printed.
1701 static __isl_give isl_printer *print_constraint_c(__isl_take isl_printer *p,
1702 __isl_keep isl_space *dim,
1703 __isl_keep isl_basic_set *bset, isl_int *c, const char *op, int *first)
1705 unsigned o_div;
1706 unsigned n_div;
1707 int div;
1709 o_div = isl_basic_set_offset(bset, isl_dim_div);
1710 n_div = isl_basic_set_dim(bset, isl_dim_div);
1711 div = isl_seq_last_non_zero(c + o_div, n_div);
1712 if (div >= 0 && isl_basic_set_is_div_constraint(bset, c, div))
1713 return p;
1715 if (!*first)
1716 p = isl_printer_print_str(p, " && ");
1718 p = print_affine_c(p, dim, bset, c);
1719 p = isl_printer_print_str(p, " ");
1720 p = isl_printer_print_str(p, op);
1721 p = isl_printer_print_str(p, " 0");
1723 *first = 0;
1725 return p;
1728 static __isl_give isl_printer *print_basic_set_c(__isl_take isl_printer *p,
1729 __isl_keep isl_space *dim, __isl_keep isl_basic_set *bset)
1731 int i, j;
1732 int first = 1;
1733 unsigned n_div = isl_basic_set_dim(bset, isl_dim_div);
1734 unsigned total = isl_basic_set_total_dim(bset) - n_div;
1736 for (i = 0; i < bset->n_eq; ++i) {
1737 j = isl_seq_last_non_zero(bset->eq[i] + 1 + total, n_div);
1738 if (j < 0)
1739 p = print_constraint_c(p, dim, bset,
1740 bset->eq[i], "==", &first);
1741 else {
1742 if (i)
1743 p = isl_printer_print_str(p, " && ");
1744 p = isl_printer_print_str(p, "(");
1745 p = print_partial_affine_c(p, dim, bset, bset->eq[i],
1746 1 + total + j);
1747 p = isl_printer_print_str(p, ") % ");
1748 p = isl_printer_print_isl_int(p,
1749 bset->eq[i][1 + total + j]);
1750 p = isl_printer_print_str(p, " == 0");
1751 first = 0;
1754 for (i = 0; i < bset->n_ineq; ++i)
1755 p = print_constraint_c(p, dim, bset, bset->ineq[i], ">=",
1756 &first);
1757 return p;
1760 static __isl_give isl_printer *print_set_c(__isl_take isl_printer *p,
1761 __isl_keep isl_space *dim, __isl_keep isl_set *set)
1763 int i;
1765 if (!set)
1766 return isl_printer_free(p);
1768 if (set->n == 0)
1769 p = isl_printer_print_str(p, "0");
1771 for (i = 0; i < set->n; ++i) {
1772 if (i)
1773 p = isl_printer_print_str(p, " || ");
1774 if (set->n > 1)
1775 p = isl_printer_print_str(p, "(");
1776 p = print_basic_set_c(p, dim, set->p[i]);
1777 if (set->n > 1)
1778 p = isl_printer_print_str(p, ")");
1780 return p;
1783 static __isl_give isl_printer *print_pw_qpolynomial_c(
1784 __isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial *pwqp)
1786 int i;
1788 if (pwqp->n == 1 && isl_set_plain_is_universe(pwqp->p[0].set))
1789 return print_qpolynomial_c(p, pwqp->dim, pwqp->p[0].qp);
1791 for (i = 0; i < pwqp->n; ++i) {
1792 p = isl_printer_print_str(p, "(");
1793 p = print_set_c(p, pwqp->dim, pwqp->p[i].set);
1794 p = isl_printer_print_str(p, ") ? (");
1795 p = print_qpolynomial_c(p, pwqp->dim, pwqp->p[i].qp);
1796 p = isl_printer_print_str(p, ") : ");
1799 p = isl_printer_print_str(p, "0");
1800 return p;
1803 __isl_give isl_printer *isl_printer_print_pw_qpolynomial(
1804 __isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial *pwqp)
1806 if (!p || !pwqp)
1807 goto error;
1809 if (p->output_format == ISL_FORMAT_ISL)
1810 return print_pw_qpolynomial_isl(p, pwqp);
1811 else if (p->output_format == ISL_FORMAT_C)
1812 return print_pw_qpolynomial_c(p, pwqp);
1813 isl_assert(p->ctx, 0, goto error);
1814 error:
1815 isl_printer_free(p);
1816 return NULL;
1819 static int print_pwqp_body(__isl_take isl_pw_qpolynomial *pwqp, void *user)
1821 struct isl_union_print_data *data;
1822 data = (struct isl_union_print_data *)user;
1824 if (!data->first)
1825 data->p = isl_printer_print_str(data->p, "; ");
1826 data->first = 0;
1828 data->p = isl_pwqp_print_isl_body(data->p, pwqp);
1829 isl_pw_qpolynomial_free(pwqp);
1831 return 0;
1834 static __isl_give isl_printer *print_union_pw_qpolynomial_isl(
1835 __isl_take isl_printer *p, __isl_keep isl_union_pw_qpolynomial *upwqp)
1837 struct isl_union_print_data data = { p, 1 };
1838 struct isl_print_space_data space_data = { 0 };
1839 isl_space *dim;
1841 dim = isl_union_pw_qpolynomial_get_space(upwqp);
1842 if (isl_space_dim(dim, isl_dim_param) > 0) {
1843 p = print_tuple(dim, p, isl_dim_param, &space_data);
1844 p = isl_printer_print_str(p, " -> ");
1846 isl_space_free(dim);
1847 p = isl_printer_print_str(p, "{ ");
1848 isl_union_pw_qpolynomial_foreach_pw_qpolynomial(upwqp, &print_pwqp_body,
1849 &data);
1850 p = data.p;
1851 p = isl_printer_print_str(p, " }");
1852 return p;
1855 __isl_give isl_printer *isl_printer_print_union_pw_qpolynomial(
1856 __isl_take isl_printer *p, __isl_keep isl_union_pw_qpolynomial *upwqp)
1858 if (!p || !upwqp)
1859 goto error;
1861 if (p->output_format == ISL_FORMAT_ISL)
1862 return print_union_pw_qpolynomial_isl(p, upwqp);
1863 isl_die(p->ctx, isl_error_invalid,
1864 "invalid output format for isl_union_pw_qpolynomial",
1865 goto error);
1866 error:
1867 isl_printer_free(p);
1868 return NULL;
1871 static __isl_give isl_printer *print_qpolynomial_fold_c(
1872 __isl_take isl_printer *p, __isl_keep isl_space *dim,
1873 __isl_keep isl_qpolynomial_fold *fold)
1875 int i;
1877 for (i = 0; i < fold->n - 1; ++i)
1878 if (fold->type == isl_fold_min)
1879 p = isl_printer_print_str(p, "min(");
1880 else if (fold->type == isl_fold_max)
1881 p = isl_printer_print_str(p, "max(");
1883 for (i = 0; i < fold->n; ++i) {
1884 if (i)
1885 p = isl_printer_print_str(p, ", ");
1886 p = print_qpolynomial_c(p, dim, fold->qp[i]);
1887 if (i)
1888 p = isl_printer_print_str(p, ")");
1890 return p;
1893 __isl_give isl_printer *isl_printer_print_qpolynomial_fold(
1894 __isl_take isl_printer *p, __isl_keep isl_qpolynomial_fold *fold)
1896 if (!p || !fold)
1897 goto error;
1898 if (p->output_format == ISL_FORMAT_ISL)
1899 return qpolynomial_fold_print(fold, p);
1900 else if (p->output_format == ISL_FORMAT_C)
1901 return print_qpolynomial_fold_c(p, fold->dim, fold);
1902 isl_die(p->ctx, isl_error_unsupported, "unsupported output format",
1903 goto error);
1904 error:
1905 isl_printer_free(p);
1906 return NULL;
1909 static __isl_give isl_printer *print_pw_qpolynomial_fold_c(
1910 __isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial_fold *pwf)
1912 int i;
1914 if (pwf->n == 1 && isl_set_plain_is_universe(pwf->p[0].set))
1915 return print_qpolynomial_fold_c(p, pwf->dim, pwf->p[0].fold);
1917 for (i = 0; i < pwf->n; ++i) {
1918 p = isl_printer_print_str(p, "(");
1919 p = print_set_c(p, pwf->dim, pwf->p[i].set);
1920 p = isl_printer_print_str(p, ") ? (");
1921 p = print_qpolynomial_fold_c(p, pwf->dim, pwf->p[i].fold);
1922 p = isl_printer_print_str(p, ") : ");
1925 p = isl_printer_print_str(p, "0");
1926 return p;
1929 __isl_give isl_printer *isl_printer_print_pw_qpolynomial_fold(
1930 __isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial_fold *pwf)
1932 if (!p || !pwf)
1933 goto error;
1935 if (p->output_format == ISL_FORMAT_ISL)
1936 return print_pw_qpolynomial_fold_isl(p, pwf);
1937 else if (p->output_format == ISL_FORMAT_C)
1938 return print_pw_qpolynomial_fold_c(p, pwf);
1939 isl_assert(p->ctx, 0, goto error);
1940 error:
1941 isl_printer_free(p);
1942 return NULL;
1945 void isl_pw_qpolynomial_fold_print(__isl_keep isl_pw_qpolynomial_fold *pwf,
1946 FILE *out, unsigned output_format)
1948 isl_printer *p;
1950 if (!pwf)
1951 return;
1953 p = isl_printer_to_file(pwf->dim->ctx, out);
1954 p = isl_printer_set_output_format(p, output_format);
1955 p = isl_printer_print_pw_qpolynomial_fold(p, pwf);
1957 isl_printer_free(p);
1960 static int print_pwf_body(__isl_take isl_pw_qpolynomial_fold *pwf, void *user)
1962 struct isl_union_print_data *data;
1963 data = (struct isl_union_print_data *)user;
1965 if (!data->first)
1966 data->p = isl_printer_print_str(data->p, "; ");
1967 data->first = 0;
1969 data->p = isl_pwf_print_isl_body(data->p, pwf);
1970 isl_pw_qpolynomial_fold_free(pwf);
1972 return 0;
1975 static __isl_give isl_printer *print_union_pw_qpolynomial_fold_isl(
1976 __isl_take isl_printer *p,
1977 __isl_keep isl_union_pw_qpolynomial_fold *upwf)
1979 struct isl_union_print_data data = { p, 1 };
1980 struct isl_print_space_data space_data = { 0 };
1981 isl_space *dim;
1983 dim = isl_union_pw_qpolynomial_fold_get_space(upwf);
1984 if (isl_space_dim(dim, isl_dim_param) > 0) {
1985 p = print_tuple(dim, p, isl_dim_param, &space_data);
1986 p = isl_printer_print_str(p, " -> ");
1988 isl_space_free(dim);
1989 p = isl_printer_print_str(p, "{ ");
1990 isl_union_pw_qpolynomial_fold_foreach_pw_qpolynomial_fold(upwf,
1991 &print_pwf_body, &data);
1992 p = data.p;
1993 p = isl_printer_print_str(p, " }");
1994 return p;
1997 __isl_give isl_printer *isl_printer_print_union_pw_qpolynomial_fold(
1998 __isl_take isl_printer *p,
1999 __isl_keep isl_union_pw_qpolynomial_fold *upwf)
2001 if (!p || !upwf)
2002 goto error;
2004 if (p->output_format == ISL_FORMAT_ISL)
2005 return print_union_pw_qpolynomial_fold_isl(p, upwf);
2006 isl_die(p->ctx, isl_error_invalid,
2007 "invalid output format for isl_union_pw_qpolynomial_fold",
2008 goto error);
2009 error:
2010 isl_printer_free(p);
2011 return NULL;
2014 __isl_give isl_printer *isl_printer_print_constraint(__isl_take isl_printer *p,
2015 __isl_keep isl_constraint *c)
2017 isl_basic_map *bmap;
2019 if (!p || !c)
2020 goto error;
2022 bmap = isl_basic_map_from_constraint(isl_constraint_copy(c));
2023 p = isl_printer_print_basic_map(p, bmap);
2024 isl_basic_map_free(bmap);
2025 return p;
2026 error:
2027 isl_printer_free(p);
2028 return NULL;
2031 static __isl_give isl_printer *isl_printer_print_space_isl(
2032 __isl_take isl_printer *p, __isl_keep isl_space *dim)
2034 struct isl_print_space_data data = { 0 };
2036 if (!dim)
2037 goto error;
2039 if (isl_space_dim(dim, isl_dim_param) > 0) {
2040 p = print_tuple(dim, p, isl_dim_param, &data);
2041 p = isl_printer_print_str(p, " -> ");
2044 p = isl_printer_print_str(p, "{ ");
2045 if (isl_space_is_params(dim))
2046 p = isl_printer_print_str(p, s_such_that[0]);
2047 else
2048 p = print_space(dim, p, 0, &data);
2049 p = isl_printer_print_str(p, " }");
2051 return p;
2052 error:
2053 isl_printer_free(p);
2054 return NULL;
2057 __isl_give isl_printer *isl_printer_print_space(__isl_take isl_printer *p,
2058 __isl_keep isl_space *space)
2060 if (!p || !space)
2061 return isl_printer_free(p);
2062 if (p->output_format == ISL_FORMAT_ISL)
2063 return isl_printer_print_space_isl(p, space);
2064 else if (p->output_format == ISL_FORMAT_OMEGA)
2065 return print_omega_parameters(space, p);
2067 isl_die(isl_space_get_ctx(space), isl_error_unsupported,
2068 "output format not supported for space",
2069 return isl_printer_free(p));
2072 __isl_give isl_printer *isl_printer_print_local_space(__isl_take isl_printer *p,
2073 __isl_keep isl_local_space *ls)
2075 struct isl_print_space_data data = { 0 };
2076 unsigned total;
2077 unsigned n_div;
2079 if (!ls)
2080 goto error;
2082 total = isl_local_space_dim(ls, isl_dim_all);
2083 if (isl_local_space_dim(ls, isl_dim_param) > 0) {
2084 p = print_tuple(ls->dim, p, isl_dim_param, &data);
2085 p = isl_printer_print_str(p, " -> ");
2087 p = isl_printer_print_str(p, "{ ");
2088 p = print_space(ls->dim, p, 0, &data);
2089 n_div = isl_local_space_dim(ls, isl_dim_div);
2090 if (n_div > 0) {
2091 p = isl_printer_print_str(p, " : ");
2092 p = isl_printer_print_str(p, s_open_exists[0]);
2093 p = print_div_list(p, ls->dim, ls->div, 0);
2094 } else if (isl_space_is_params(ls->dim))
2095 p = isl_printer_print_str(p, s_such_that[0]);
2096 p = isl_printer_print_str(p, " }");
2097 return p;
2098 error:
2099 isl_printer_free(p);
2100 return NULL;
2103 static __isl_give isl_printer *print_aff_body(__isl_take isl_printer *p,
2104 __isl_keep isl_aff *aff)
2106 unsigned total;
2108 if (isl_aff_is_nan(aff))
2109 return isl_printer_print_str(p, "NaN");
2111 total = isl_local_space_dim(aff->ls, isl_dim_all);
2112 p = isl_printer_print_str(p, "(");
2113 p = print_affine_of_len(aff->ls->dim, aff->ls->div, p,
2114 aff->v->el + 1, 1 + total);
2115 if (isl_int_is_one(aff->v->el[0]))
2116 p = isl_printer_print_str(p, ")");
2117 else {
2118 p = isl_printer_print_str(p, ")/");
2119 p = isl_printer_print_isl_int(p, aff->v->el[0]);
2122 return p;
2125 static __isl_give isl_printer *print_aff(__isl_take isl_printer *p,
2126 __isl_keep isl_aff *aff)
2128 struct isl_print_space_data data = { 0 };
2130 if (isl_space_is_params(aff->ls->dim))
2132 else {
2133 p = print_tuple(aff->ls->dim, p, isl_dim_set, &data);
2134 p = isl_printer_print_str(p, " -> ");
2136 p = isl_printer_print_str(p, "[");
2137 p = print_aff_body(p, aff);
2138 p = isl_printer_print_str(p, "]");
2140 return p;
2143 static __isl_give isl_printer *print_aff_isl(__isl_take isl_printer *p,
2144 __isl_keep isl_aff *aff)
2146 struct isl_print_space_data data = { 0 };
2148 if (!aff)
2149 goto error;
2151 if (isl_local_space_dim(aff->ls, isl_dim_param) > 0) {
2152 p = print_tuple(aff->ls->dim, p, isl_dim_param, &data);
2153 p = isl_printer_print_str(p, " -> ");
2155 p = isl_printer_print_str(p, "{ ");
2156 p = print_aff(p, aff);
2157 p = isl_printer_print_str(p, " }");
2158 return p;
2159 error:
2160 isl_printer_free(p);
2161 return NULL;
2164 static __isl_give isl_printer *print_pw_aff_isl(__isl_take isl_printer *p,
2165 __isl_keep isl_pw_aff *pwaff)
2167 struct isl_print_space_data data = { 0 };
2168 int i;
2170 if (!pwaff)
2171 goto error;
2173 if (isl_space_dim(pwaff->dim, isl_dim_param) > 0) {
2174 p = print_tuple(pwaff->dim, p, isl_dim_param, &data);
2175 p = isl_printer_print_str(p, " -> ");
2177 p = isl_printer_print_str(p, "{ ");
2178 for (i = 0; i < pwaff->n; ++i) {
2179 if (i)
2180 p = isl_printer_print_str(p, "; ");
2181 p = print_aff(p, pwaff->p[i].aff);
2182 p = print_disjuncts((isl_map *)pwaff->p[i].set, p, 0);
2184 p = isl_printer_print_str(p, " }");
2185 return p;
2186 error:
2187 isl_printer_free(p);
2188 return NULL;
2191 static __isl_give isl_printer *print_ls_affine_c(__isl_take isl_printer *p,
2192 __isl_keep isl_local_space *ls, isl_int *c);
2194 static __isl_give isl_printer *print_ls_name_c(__isl_take isl_printer *p,
2195 __isl_keep isl_local_space *ls, enum isl_dim_type type, unsigned pos)
2197 if (type == isl_dim_div) {
2198 p = isl_printer_print_str(p, "floord(");
2199 p = print_ls_affine_c(p, ls, ls->div->row[pos] + 1);
2200 p = isl_printer_print_str(p, ", ");
2201 p = isl_printer_print_isl_int(p, ls->div->row[pos][0]);
2202 p = isl_printer_print_str(p, ")");
2203 } else {
2204 const char *name;
2206 name = isl_space_get_dim_name(ls->dim, type, pos);
2207 if (!name)
2208 name = "UNNAMED";
2209 p = isl_printer_print_str(p, name);
2211 return p;
2214 static __isl_give isl_printer *print_ls_term_c(__isl_take isl_printer *p,
2215 __isl_keep isl_local_space *ls, isl_int c, unsigned pos)
2217 enum isl_dim_type type;
2219 if (pos == 0)
2220 return isl_printer_print_isl_int(p, c);
2222 if (isl_int_is_one(c))
2224 else if (isl_int_is_negone(c))
2225 p = isl_printer_print_str(p, "-");
2226 else {
2227 p = isl_printer_print_isl_int(p, c);
2228 p = isl_printer_print_str(p, "*");
2230 type = pos2type(ls->dim, &pos);
2231 p = print_ls_name_c(p, ls, type, pos);
2232 return p;
2235 static __isl_give isl_printer *print_ls_partial_affine_c(
2236 __isl_take isl_printer *p, __isl_keep isl_local_space *ls,
2237 isl_int *c, unsigned len)
2239 int i;
2240 int first;
2242 for (i = 0, first = 1; i < len; ++i) {
2243 int flip = 0;
2244 if (isl_int_is_zero(c[i]))
2245 continue;
2246 if (!first) {
2247 if (isl_int_is_neg(c[i])) {
2248 flip = 1;
2249 isl_int_neg(c[i], c[i]);
2250 p = isl_printer_print_str(p, " - ");
2251 } else
2252 p = isl_printer_print_str(p, " + ");
2254 first = 0;
2255 p = print_ls_term_c(p, ls, c[i], i);
2256 if (flip)
2257 isl_int_neg(c[i], c[i]);
2259 if (first)
2260 p = isl_printer_print_str(p, "0");
2261 return p;
2264 static __isl_give isl_printer *print_ls_affine_c(__isl_take isl_printer *p,
2265 __isl_keep isl_local_space *ls, isl_int *c)
2267 unsigned len = 1 + isl_local_space_dim(ls, isl_dim_all);
2268 return print_ls_partial_affine_c(p, ls, c, len);
2271 static __isl_give isl_printer *print_aff_c(__isl_take isl_printer *p,
2272 __isl_keep isl_aff *aff)
2274 unsigned total;
2276 total = isl_local_space_dim(aff->ls, isl_dim_all);
2277 if (!isl_int_is_one(aff->v->el[0]))
2278 p = isl_printer_print_str(p, "(");
2279 p = print_ls_partial_affine_c(p, aff->ls, aff->v->el + 1, 1 + total);
2280 if (!isl_int_is_one(aff->v->el[0])) {
2281 p = isl_printer_print_str(p, ")/");
2282 p = isl_printer_print_isl_int(p, aff->v->el[0]);
2284 return p;
2287 /* In the C format, we cannot express that "pwaff" may be undefined
2288 * on parts of the domain space. We therefore assume that the expression
2289 * will only be evaluated on its definition domain and compute the gist
2290 * of each cell with respect to this domain.
2292 static __isl_give isl_printer *print_pw_aff_c(__isl_take isl_printer *p,
2293 __isl_keep isl_pw_aff *pwaff)
2295 isl_set *domain;
2296 isl_ast_build *build;
2297 isl_ast_expr *expr;
2299 if (pwaff->n < 1)
2300 isl_die(p->ctx, isl_error_unsupported,
2301 "cannot print empty isl_pw_aff in C format",
2302 return isl_printer_free(p));
2304 domain = isl_pw_aff_domain(isl_pw_aff_copy(pwaff));
2305 build = isl_ast_build_from_context(domain);
2306 expr = isl_ast_build_expr_from_pw_aff(build, isl_pw_aff_copy(pwaff));
2307 p = isl_printer_print_ast_expr(p, expr);
2308 isl_ast_expr_free(expr);
2309 isl_ast_build_free(build);
2311 return p;
2314 __isl_give isl_printer *isl_printer_print_aff(__isl_take isl_printer *p,
2315 __isl_keep isl_aff *aff)
2317 if (!p || !aff)
2318 goto error;
2320 if (p->output_format == ISL_FORMAT_ISL)
2321 return print_aff_isl(p, aff);
2322 else if (p->output_format == ISL_FORMAT_C)
2323 return print_aff_c(p, aff);
2324 isl_die(p->ctx, isl_error_unsupported, "unsupported output format",
2325 goto error);
2326 error:
2327 isl_printer_free(p);
2328 return NULL;
2331 __isl_give isl_printer *isl_printer_print_pw_aff(__isl_take isl_printer *p,
2332 __isl_keep isl_pw_aff *pwaff)
2334 if (!p || !pwaff)
2335 goto error;
2337 if (p->output_format == ISL_FORMAT_ISL)
2338 return print_pw_aff_isl(p, pwaff);
2339 else if (p->output_format == ISL_FORMAT_C)
2340 return print_pw_aff_c(p, pwaff);
2341 isl_die(p->ctx, isl_error_unsupported, "unsupported output format",
2342 goto error);
2343 error:
2344 isl_printer_free(p);
2345 return NULL;
2348 /* Print dimension "pos" of data->space to "p".
2350 * data->user is assumed to be an isl_multi_aff.
2352 * If the current dimension is an output dimension, then print
2353 * the corresponding expression. Otherwise, print the name of the dimension.
2355 static __isl_give isl_printer *print_dim_ma(__isl_take isl_printer *p,
2356 struct isl_print_space_data *data, unsigned pos)
2358 isl_multi_aff *ma = data->user;
2360 if (data->type == isl_dim_out)
2361 p = print_aff_body(p, ma->p[pos]);
2362 else
2363 p = print_name(data->space, p, data->type, pos, data->latex);
2365 return p;
2368 static __isl_give isl_printer *print_multi_aff(__isl_take isl_printer *p,
2369 __isl_keep isl_multi_aff *maff)
2371 struct isl_print_space_data data = { 0 };
2373 data.print_dim = &print_dim_ma;
2374 data.user = maff;
2375 return print_space(maff->space, p, 0, &data);
2378 static __isl_give isl_printer *print_multi_aff_isl(__isl_take isl_printer *p,
2379 __isl_keep isl_multi_aff *maff)
2381 struct isl_print_space_data data = { 0 };
2383 if (!maff)
2384 goto error;
2386 if (isl_space_dim(maff->space, isl_dim_param) > 0) {
2387 p = print_tuple(maff->space, p, isl_dim_param, &data);
2388 p = isl_printer_print_str(p, " -> ");
2390 p = isl_printer_print_str(p, "{ ");
2391 p = print_multi_aff(p, maff);
2392 p = isl_printer_print_str(p, " }");
2393 return p;
2394 error:
2395 isl_printer_free(p);
2396 return NULL;
2399 __isl_give isl_printer *isl_printer_print_multi_aff(__isl_take isl_printer *p,
2400 __isl_keep isl_multi_aff *maff)
2402 if (!p || !maff)
2403 goto error;
2405 if (p->output_format == ISL_FORMAT_ISL)
2406 return print_multi_aff_isl(p, maff);
2407 isl_die(p->ctx, isl_error_unsupported, "unsupported output format",
2408 goto error);
2409 error:
2410 isl_printer_free(p);
2411 return NULL;
2414 static __isl_give isl_printer *print_pw_multi_aff_body(
2415 __isl_take isl_printer *p, __isl_keep isl_pw_multi_aff *pma)
2417 int i;
2419 if (!pma)
2420 goto error;
2422 for (i = 0; i < pma->n; ++i) {
2423 if (i)
2424 p = isl_printer_print_str(p, "; ");
2425 p = print_multi_aff(p, pma->p[i].maff);
2426 p = print_disjuncts((isl_map *)pma->p[i].set, p, 0);
2428 return p;
2429 error:
2430 isl_printer_free(p);
2431 return NULL;
2434 static __isl_give isl_printer *print_pw_multi_aff_isl(__isl_take isl_printer *p,
2435 __isl_keep isl_pw_multi_aff *pma)
2437 struct isl_print_space_data data = { 0 };
2439 if (!pma)
2440 goto error;
2442 if (isl_space_dim(pma->dim, isl_dim_param) > 0) {
2443 p = print_tuple(pma->dim, p, isl_dim_param, &data);
2444 p = isl_printer_print_str(p, " -> ");
2446 p = isl_printer_print_str(p, "{ ");
2447 p = print_pw_multi_aff_body(p, pma);
2448 p = isl_printer_print_str(p, " }");
2449 return p;
2450 error:
2451 isl_printer_free(p);
2452 return NULL;
2455 static __isl_give isl_printer *print_unnamed_pw_multi_aff_c(
2456 __isl_take isl_printer *p, __isl_keep isl_pw_multi_aff *pma)
2458 int i;
2460 for (i = 0; i < pma->n - 1; ++i) {
2461 p = isl_printer_print_str(p, "(");
2462 p = print_set_c(p, pma->dim, pma->p[i].set);
2463 p = isl_printer_print_str(p, ") ? (");
2464 p = print_aff_c(p, pma->p[i].maff->p[0]);
2465 p = isl_printer_print_str(p, ") : ");
2468 return print_aff_c(p, pma->p[pma->n - 1].maff->p[0]);
2471 static __isl_give isl_printer *print_pw_multi_aff_c(__isl_take isl_printer *p,
2472 __isl_keep isl_pw_multi_aff *pma)
2474 int n;
2475 const char *name;
2477 if (!pma)
2478 goto error;
2479 if (pma->n < 1)
2480 isl_die(p->ctx, isl_error_unsupported,
2481 "cannot print empty isl_pw_multi_aff in C format",
2482 goto error);
2483 name = isl_pw_multi_aff_get_tuple_name(pma, isl_dim_out);
2484 if (!name && isl_pw_multi_aff_dim(pma, isl_dim_out) == 1)
2485 return print_unnamed_pw_multi_aff_c(p, pma);
2486 if (!name)
2487 isl_die(p->ctx, isl_error_unsupported,
2488 "cannot print unnamed isl_pw_multi_aff in C format",
2489 goto error);
2491 p = isl_printer_print_str(p, name);
2492 n = isl_pw_multi_aff_dim(pma, isl_dim_out);
2493 if (n != 0)
2494 isl_die(p->ctx, isl_error_unsupported,
2495 "not supported yet", goto error);
2497 return p;
2498 error:
2499 isl_printer_free(p);
2500 return NULL;
2503 __isl_give isl_printer *isl_printer_print_pw_multi_aff(
2504 __isl_take isl_printer *p, __isl_keep isl_pw_multi_aff *pma)
2506 if (!p || !pma)
2507 goto error;
2509 if (p->output_format == ISL_FORMAT_ISL)
2510 return print_pw_multi_aff_isl(p, pma);
2511 if (p->output_format == ISL_FORMAT_C)
2512 return print_pw_multi_aff_c(p, pma);
2513 isl_die(p->ctx, isl_error_unsupported, "unsupported output format",
2514 goto error);
2515 error:
2516 isl_printer_free(p);
2517 return NULL;
2520 static int print_pw_multi_aff_body_wrap(__isl_take isl_pw_multi_aff *pma,
2521 void *user)
2523 struct isl_union_print_data *data;
2524 data = (struct isl_union_print_data *) user;
2526 if (!data->first)
2527 data->p = isl_printer_print_str(data->p, "; ");
2528 data->first = 0;
2530 data->p = print_pw_multi_aff_body(data->p, pma);
2531 isl_pw_multi_aff_free(pma);
2533 return 0;
2536 static __isl_give isl_printer *print_union_pw_multi_aff_isl(
2537 __isl_take isl_printer *p, __isl_keep isl_union_pw_multi_aff *upma)
2539 struct isl_union_print_data data = { p, 1 };
2540 struct isl_print_space_data space_data = { 0 };
2541 isl_space *space;
2543 space = isl_union_pw_multi_aff_get_space(upma);
2544 if (isl_space_dim(space, isl_dim_param) > 0) {
2545 p = print_tuple(space, p, isl_dim_param, &space_data);
2546 p = isl_printer_print_str(p, s_to[0]);
2548 isl_space_free(space);
2549 p = isl_printer_print_str(p, s_open_set[0]);
2550 isl_union_pw_multi_aff_foreach_pw_multi_aff(upma,
2551 &print_pw_multi_aff_body_wrap, &data);
2552 p = data.p;
2553 p = isl_printer_print_str(p, s_close_set[0]);
2554 return p;
2557 __isl_give isl_printer *isl_printer_print_union_pw_multi_aff(
2558 __isl_take isl_printer *p, __isl_keep isl_union_pw_multi_aff *upma)
2560 if (!p || !upma)
2561 goto error;
2563 if (p->output_format == ISL_FORMAT_ISL)
2564 return print_union_pw_multi_aff_isl(p, upma);
2565 isl_die(p->ctx, isl_error_unsupported, "unsupported output format",
2566 goto error);
2567 error:
2568 isl_printer_free(p);
2569 return NULL;
2572 /* Print dimension "pos" of data->space to "p".
2574 * data->user is assumed to be an isl_multi_pw_aff.
2576 * If the current dimension is an output dimension, then print
2577 * the corresponding piecewise affine expression.
2578 * Otherwise, print the name of the dimension.
2580 static __isl_give isl_printer *print_dim_mpa(__isl_take isl_printer *p,
2581 struct isl_print_space_data *data, unsigned pos)
2583 int i;
2584 int need_parens;
2585 isl_multi_pw_aff *mpa = data->user;
2586 isl_pw_aff *pa;
2588 if (data->type != isl_dim_out)
2589 return print_name(data->space, p, data->type, pos, data->latex);
2591 pa = mpa->p[pos];
2592 if (pa->n == 0)
2593 return isl_printer_print_str(p, "(0 : 1 = 0)");
2595 need_parens = pa->n != 1 || !isl_set_plain_is_universe(pa->p[0].set);
2596 if (need_parens)
2597 p = isl_printer_print_str(p, "(");
2598 for (i = 0; i < pa->n; ++i) {
2599 if (i)
2600 p = isl_printer_print_str(p, "; ");
2601 p = print_aff_body(p, pa->p[i].aff);
2602 p = print_disjuncts(pa->p[i].set, p, 0);
2604 if (need_parens)
2605 p = isl_printer_print_str(p, ")");
2607 return p;
2610 /* Print "mpa" to "p" in isl format.
2612 static __isl_give isl_printer *print_multi_pw_aff_isl(__isl_take isl_printer *p,
2613 __isl_keep isl_multi_pw_aff *mpa)
2615 struct isl_print_space_data data = { 0 };
2617 if (!mpa)
2618 return isl_printer_free(p);
2620 if (isl_space_dim(mpa->space, isl_dim_param) > 0) {
2621 p = print_tuple(mpa->space, p, isl_dim_param, &data);
2622 p = isl_printer_print_str(p, " -> ");
2624 p = isl_printer_print_str(p, "{ ");
2625 data.print_dim = &print_dim_mpa;
2626 data.user = mpa;
2627 p = print_space(mpa->space, p, 0, &data);
2628 p = isl_printer_print_str(p, " }");
2629 return p;
2632 __isl_give isl_printer *isl_printer_print_multi_pw_aff(
2633 __isl_take isl_printer *p, __isl_keep isl_multi_pw_aff *mpa)
2635 if (!p || !mpa)
2636 return isl_printer_free(p);
2638 if (p->output_format == ISL_FORMAT_ISL)
2639 return print_multi_pw_aff_isl(p, mpa);
2640 isl_die(p->ctx, isl_error_unsupported, "unsupported output format",
2641 return isl_printer_free(p));
2644 /* Print dimension "pos" of data->space to "p".
2646 * data->user is assumed to be an isl_multi_val.
2648 * If the current dimension is an output dimension, then print
2649 * the corresponding value. Otherwise, print the name of the dimension.
2651 static __isl_give isl_printer *print_dim_mv(__isl_take isl_printer *p,
2652 struct isl_print_space_data *data, unsigned pos)
2654 isl_multi_val *mv = data->user;
2656 if (data->type == isl_dim_out)
2657 return isl_printer_print_val(p, mv->p[pos]);
2658 else
2659 return print_name(data->space, p, data->type, pos, data->latex);
2662 /* Print the isl_multi_val "mv" to "p" in isl format.
2664 static __isl_give isl_printer *print_multi_val_isl(__isl_take isl_printer *p,
2665 __isl_keep isl_multi_val *mv)
2667 struct isl_print_space_data data = { 0 };
2669 if (!mv)
2670 return isl_printer_free(p);
2672 if (isl_space_dim(mv->space, isl_dim_param) > 0) {
2673 p = print_tuple(mv->space, p, isl_dim_param, &data);
2674 p = isl_printer_print_str(p, " -> ");
2676 p = isl_printer_print_str(p, "{ ");
2677 data.print_dim = &print_dim_mv;
2678 data.user = mv;
2679 p = print_space(mv->space, p, 0, &data);
2680 p = isl_printer_print_str(p, " }");
2681 return p;
2684 /* Print the isl_multi_val "mv" to "p".
2686 * Currently only supported in isl format.
2688 __isl_give isl_printer *isl_printer_print_multi_val(
2689 __isl_take isl_printer *p, __isl_keep isl_multi_val *mv)
2691 if (!p || !mv)
2692 return isl_printer_free(p);
2694 if (p->output_format == ISL_FORMAT_ISL)
2695 return print_multi_val_isl(p, mv);
2696 isl_die(p->ctx, isl_error_unsupported, "unsupported output format",
2697 return isl_printer_free(p));