hide isl_vec internals
[isl.git] / isl_output.c
blob82d22b42d0d2cb44d4135e6f9fc5425f465959d1
1 /*
2 * Copyright 2008-2009 Katholieke Universiteit Leuven
3 * Copyright 2010 INRIA Saclay
5 * Use of this software is governed by the MIT 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_space_private.h>
22 #include <isl_mat_private.h>
23 #include <isl_vec_private.h>
24 #include <isl/union_map.h>
25 #include <isl/constraint.h>
26 #include <isl_local_space_private.h>
27 #include <isl_aff_private.h>
28 #include <isl_ast_build_expr.h>
29 #include <isl_sort.h>
31 static const char *s_to[2] = { " -> ", " \\to " };
32 static const char *s_and[2] = { " and ", " \\wedge " };
33 static const char *s_or[2] = { " or ", " \\vee " };
34 static const char *s_le[2] = { "<=", "\\le" };
35 static const char *s_ge[2] = { ">=", "\\ge" };
36 static const char *s_open_set[2] = { "{ ", "\\{\\, " };
37 static const char *s_close_set[2] = { " }", " \\,\\}" };
38 static const char *s_open_list[2] = { "[", "(" };
39 static const char *s_close_list[2] = { "]", ")" };
40 static const char *s_such_that[2] = { " : ", " \\mid " };
41 static const char *s_open_exists[2] = { "exists (", "\\exists \\, " };
42 static const char *s_close_exists[2] = { ")", "" };
43 static const char *s_div_prefix[2] = { "e", "\\alpha_" };
44 static const char *s_param_prefix[2] = { "p", "p_" };
45 static const char *s_input_prefix[2] = { "i", "i_" };
46 static const char *s_output_prefix[2] = { "o", "o_" };
48 static __isl_give isl_printer *print_constraint_polylib(
49 struct isl_basic_map *bmap, int ineq, int n, __isl_take isl_printer *p)
51 int i;
52 unsigned n_in = isl_basic_map_dim(bmap, isl_dim_in);
53 unsigned n_out = isl_basic_map_dim(bmap, isl_dim_out);
54 unsigned nparam = isl_basic_map_dim(bmap, isl_dim_param);
55 isl_int *c = ineq ? bmap->ineq[n] : bmap->eq[n];
57 p = isl_printer_start_line(p);
58 p = isl_printer_print_int(p, ineq);
59 for (i = 0; i < n_out; ++i) {
60 p = isl_printer_print_str(p, " ");
61 p = isl_printer_print_isl_int(p, c[1+nparam+n_in+i]);
63 for (i = 0; i < n_in; ++i) {
64 p = isl_printer_print_str(p, " ");
65 p = isl_printer_print_isl_int(p, c[1+nparam+i]);
67 for (i = 0; i < bmap->n_div; ++i) {
68 p = isl_printer_print_str(p, " ");
69 p = isl_printer_print_isl_int(p, c[1+nparam+n_in+n_out+i]);
71 for (i = 0; i < nparam; ++i) {
72 p = isl_printer_print_str(p, " ");
73 p = isl_printer_print_isl_int(p, c[1+i]);
75 p = isl_printer_print_str(p, " ");
76 p = isl_printer_print_isl_int(p, c[0]);
77 p = isl_printer_end_line(p);
78 return p;
81 static __isl_give isl_printer *print_constraints_polylib(
82 struct isl_basic_map *bmap, __isl_take isl_printer *p)
84 int i;
86 p = isl_printer_set_isl_int_width(p, 5);
88 for (i = 0; i < bmap->n_eq; ++i)
89 p = print_constraint_polylib(bmap, 0, i, p);
90 for (i = 0; i < bmap->n_ineq; ++i)
91 p = print_constraint_polylib(bmap, 1, i, p);
93 return p;
96 static __isl_give isl_printer *bset_print_constraints_polylib(
97 struct isl_basic_set *bset, __isl_take isl_printer *p)
99 return print_constraints_polylib((struct isl_basic_map *)bset, p);
102 static __isl_give isl_printer *isl_basic_map_print_polylib(
103 __isl_keep isl_basic_map *bmap, __isl_take isl_printer *p, int ext)
105 unsigned total = isl_basic_map_total_dim(bmap);
106 p = isl_printer_start_line(p);
107 p = isl_printer_print_int(p, bmap->n_eq + bmap->n_ineq);
108 p = isl_printer_print_str(p, " ");
109 p = isl_printer_print_int(p, 1 + total + 1);
110 if (ext) {
111 p = isl_printer_print_str(p, " ");
112 p = isl_printer_print_int(p,
113 isl_basic_map_dim(bmap, isl_dim_out));
114 p = isl_printer_print_str(p, " ");
115 p = isl_printer_print_int(p,
116 isl_basic_map_dim(bmap, isl_dim_in));
117 p = isl_printer_print_str(p, " ");
118 p = isl_printer_print_int(p,
119 isl_basic_map_dim(bmap, isl_dim_div));
120 p = isl_printer_print_str(p, " ");
121 p = isl_printer_print_int(p,
122 isl_basic_map_dim(bmap, isl_dim_param));
124 p = isl_printer_end_line(p);
125 return print_constraints_polylib(bmap, p);
128 static __isl_give isl_printer *isl_basic_set_print_polylib(
129 __isl_keep isl_basic_set *bset, __isl_take isl_printer *p, int ext)
131 return isl_basic_map_print_polylib((struct isl_basic_map *)bset, p, ext);
134 static __isl_give isl_printer *isl_map_print_polylib(__isl_keep isl_map *map,
135 __isl_take isl_printer *p, int ext)
137 int i;
139 p = isl_printer_start_line(p);
140 p = isl_printer_print_int(p, map->n);
141 p = isl_printer_end_line(p);
142 for (i = 0; i < map->n; ++i) {
143 p = isl_printer_start_line(p);
144 p = isl_printer_end_line(p);
145 p = isl_basic_map_print_polylib(map->p[i], p, ext);
147 return p;
150 static __isl_give isl_printer *isl_set_print_polylib(__isl_keep isl_set *set,
151 __isl_take isl_printer *p, int ext)
153 return isl_map_print_polylib((struct isl_map *)set, p, ext);
156 static int count_same_name(__isl_keep isl_space *dim,
157 enum isl_dim_type type, unsigned pos, const char *name)
159 enum isl_dim_type t;
160 unsigned p, s;
161 int count = 0;
163 for (t = isl_dim_param; t <= type && t <= isl_dim_out; ++t) {
164 s = t == type ? pos : isl_space_dim(dim, t);
165 for (p = 0; p < s; ++p) {
166 const char *n = isl_space_get_dim_name(dim, t, p);
167 if (n && !strcmp(n, name))
168 count++;
171 return count;
174 static __isl_give isl_printer *print_name(__isl_keep isl_space *dim,
175 __isl_take isl_printer *p, enum isl_dim_type type, unsigned pos,
176 int latex)
178 const char *name;
179 char buffer[20];
180 int primes;
182 name = type == isl_dim_div ? NULL : isl_space_get_dim_name(dim, type, pos);
184 if (!name) {
185 const char *prefix;
186 if (type == isl_dim_param)
187 prefix = s_param_prefix[latex];
188 else if (type == isl_dim_div)
189 prefix = s_div_prefix[latex];
190 else if (isl_space_is_set(dim) || type == isl_dim_in)
191 prefix = s_input_prefix[latex];
192 else
193 prefix = s_output_prefix[latex];
194 snprintf(buffer, sizeof(buffer), "%s%d", prefix, pos);
195 name = buffer;
197 primes = count_same_name(dim, name == buffer ? isl_dim_div : type,
198 pos, name);
199 p = isl_printer_print_str(p, name);
200 while (primes-- > 0)
201 p = isl_printer_print_str(p, "'");
202 return p;
205 static enum isl_dim_type pos2type(__isl_keep isl_space *dim, unsigned *pos)
207 enum isl_dim_type type;
208 unsigned n_in = isl_space_dim(dim, isl_dim_in);
209 unsigned n_out = isl_space_dim(dim, isl_dim_out);
210 unsigned nparam = isl_space_dim(dim, isl_dim_param);
212 if (*pos < 1 + nparam) {
213 type = isl_dim_param;
214 *pos -= 1;
215 } else if (*pos < 1 + nparam + n_in) {
216 type = isl_dim_in;
217 *pos -= 1 + nparam;
218 } else if (*pos < 1 + nparam + n_in + n_out) {
219 type = isl_dim_out;
220 *pos -= 1 + nparam + n_in;
221 } else {
222 type = isl_dim_div;
223 *pos -= 1 + nparam + n_in + n_out;
226 return type;
229 static __isl_give isl_printer *print_div(__isl_keep isl_space *dim,
230 __isl_keep isl_mat *div, int pos, __isl_take isl_printer *p);
232 static __isl_give isl_printer *print_term(__isl_keep isl_space *dim,
233 __isl_keep isl_mat *div,
234 isl_int c, unsigned pos, __isl_take isl_printer *p, int latex)
236 enum isl_dim_type type;
237 int print_div_def;
239 if (pos == 0)
240 return isl_printer_print_isl_int(p, c);
242 type = pos2type(dim, &pos);
243 print_div_def = type == isl_dim_div && div &&
244 !isl_int_is_zero(div->row[pos][0]);
246 if (isl_int_is_one(c))
248 else if (isl_int_is_negone(c))
249 p = isl_printer_print_str(p, "-");
250 else {
251 p = isl_printer_print_isl_int(p, c);
252 if (p->output_format == ISL_FORMAT_C || print_div_def)
253 p = isl_printer_print_str(p, "*");
255 if (print_div_def)
256 p = print_div(dim, div, pos, p);
257 else
258 p = print_name(dim, p, type, pos, latex);
259 return p;
262 static __isl_give isl_printer *print_affine_of_len(__isl_keep isl_space *dim,
263 __isl_keep isl_mat *div,
264 __isl_take isl_printer *p, isl_int *c, int len)
266 int i;
267 int first;
269 for (i = 0, first = 1; i < len; ++i) {
270 int flip = 0;
271 if (isl_int_is_zero(c[i]))
272 continue;
273 if (!first) {
274 if (isl_int_is_neg(c[i])) {
275 flip = 1;
276 isl_int_neg(c[i], c[i]);
277 p = isl_printer_print_str(p, " - ");
278 } else
279 p = isl_printer_print_str(p, " + ");
281 first = 0;
282 p = print_term(dim, div, c[i], i, p, 0);
283 if (flip)
284 isl_int_neg(c[i], c[i]);
286 if (first)
287 p = isl_printer_print_str(p, "0");
288 return p;
291 static __isl_give isl_printer *print_affine(__isl_keep isl_basic_map *bmap,
292 __isl_keep isl_space *dim, __isl_take isl_printer *p, isl_int *c)
294 unsigned len = 1 + isl_basic_map_total_dim(bmap);
295 return print_affine_of_len(dim, NULL, p, c, len);
298 static int defining_equality(__isl_keep isl_basic_map *eq,
299 __isl_keep isl_space *dim, enum isl_dim_type type, int pos)
301 int i;
302 unsigned total;
304 if (!eq)
305 return -1;
307 pos += isl_space_offset(dim, type);
308 total = isl_basic_map_total_dim(eq);
310 for (i = 0; i < eq->n_eq; ++i) {
311 if (isl_seq_last_non_zero(eq->eq[i] + 1, total) != pos)
312 continue;
313 if (isl_int_is_one(eq->eq[i][1 + pos]))
314 isl_seq_neg(eq->eq[i], eq->eq[i], 1 + total);
315 return i;
318 return -1;
321 static __isl_give isl_printer *print_aff_body(__isl_take isl_printer *p,
322 __isl_keep isl_aff *aff);
324 /* offset is the offset of local_dim inside global_type of global_dim.
326 static __isl_give isl_printer *print_nested_var_list(__isl_take isl_printer *p,
327 __isl_keep isl_space *global_dim, enum isl_dim_type global_type,
328 __isl_keep isl_space *local_dim, enum isl_dim_type local_type,
329 int latex, __isl_keep isl_basic_map *eq,
330 __isl_keep isl_multi_aff *maff, int offset)
332 int i, j;
334 if (global_dim != local_dim && local_type == isl_dim_out)
335 offset += local_dim->n_in;
337 for (i = 0; i < isl_space_dim(local_dim, local_type); ++i) {
338 if (i)
339 p = isl_printer_print_str(p, ", ");
340 if (maff && global_type == isl_dim_out) {
341 p = print_aff_body(p, maff->p[offset + i]);
342 continue;
344 j = defining_equality(eq, global_dim, global_type, offset + i);
345 if (j >= 0) {
346 int pos = 1 + isl_space_offset(global_dim, global_type)
347 + offset + i;
348 p = print_affine_of_len(eq->dim, NULL,
349 p, eq->eq[j], pos);
350 } else {
351 p = print_name(global_dim, p, global_type, offset + i,
352 latex);
355 return p;
358 static __isl_give isl_printer *print_var_list(__isl_keep isl_space *dim,
359 __isl_take isl_printer *p, enum isl_dim_type type,
360 int latex, __isl_keep isl_basic_map *eq, __isl_keep isl_multi_aff *maff)
362 return print_nested_var_list(p, dim, type, dim, type, latex,
363 eq, maff, 0);
366 static __isl_give isl_printer *print_nested_map_dim(__isl_take isl_printer *p,
367 __isl_keep isl_space *global_dim, enum isl_dim_type global_type,
368 __isl_keep isl_space *local_dim,
369 int latex, __isl_keep isl_basic_map *eq,
370 __isl_keep isl_multi_aff *maff, int offset);
372 static __isl_give isl_printer *print_nested_tuple(__isl_take isl_printer *p,
373 __isl_keep isl_space *global_dim, enum isl_dim_type global_type,
374 __isl_keep isl_space *local_dim, enum isl_dim_type local_type,
375 int latex, __isl_keep isl_basic_map *eq,
376 __isl_keep isl_multi_aff *maff, int offset)
378 const char *name = NULL;
379 unsigned n = isl_space_dim(local_dim, local_type);
380 if ((local_type == isl_dim_in || local_type == isl_dim_out)) {
381 name = isl_space_get_tuple_name(local_dim, local_type);
382 if (name) {
383 if (latex)
384 p = isl_printer_print_str(p, "\\mathrm{");
385 p = isl_printer_print_str(p, name);
386 if (latex)
387 p = isl_printer_print_str(p, "}");
390 if (!latex || n != 1 || name)
391 p = isl_printer_print_str(p, s_open_list[latex]);
392 if ((local_type == isl_dim_in || local_type == isl_dim_out) &&
393 local_dim->nested[local_type - isl_dim_in]) {
394 if (global_dim != local_dim && local_type == isl_dim_out)
395 offset += local_dim->n_in;
396 p = print_nested_map_dim(p, global_dim, global_type,
397 local_dim->nested[local_type - isl_dim_in],
398 latex, eq, maff, offset);
399 } else
400 p = print_nested_var_list(p, global_dim, global_type,
401 local_dim, local_type, latex,
402 eq, maff, offset);
403 if (!latex || n != 1 || name)
404 p = isl_printer_print_str(p, s_close_list[latex]);
405 return p;
408 static __isl_give isl_printer *print_tuple(__isl_keep isl_space *dim,
409 __isl_take isl_printer *p, enum isl_dim_type type,
410 int latex, __isl_keep isl_basic_map *eq, __isl_keep isl_multi_aff *maff)
412 return print_nested_tuple(p, dim, type, dim, type, latex, eq, maff, 0);
415 static __isl_give isl_printer *print_nested_map_dim(__isl_take isl_printer *p,
416 __isl_keep isl_space *global_dim, enum isl_dim_type global_type,
417 __isl_keep isl_space *local_dim,
418 int latex, __isl_keep isl_basic_map *eq,
419 __isl_keep isl_multi_aff *maff, int offset)
421 p = print_nested_tuple(p, global_dim, global_type,
422 local_dim, isl_dim_in, latex, eq, maff, offset);
423 p = isl_printer_print_str(p, s_to[latex]);
424 p = print_nested_tuple(p, global_dim, global_type,
425 local_dim, isl_dim_out, latex, eq, maff, offset);
427 return p;
430 static __isl_give isl_printer *print_space(__isl_keep isl_space *dim,
431 __isl_take isl_printer *p, int latex, int rational,
432 __isl_keep isl_basic_map *eq, __isl_keep isl_multi_aff *maff)
434 if (rational && !latex)
435 p = isl_printer_print_str(p, "rat: ");
436 if (isl_space_is_params(dim))
438 else if (isl_space_is_set(dim))
439 p = print_tuple(dim, p, isl_dim_set, latex, eq, maff);
440 else {
441 p = print_tuple(dim, p, isl_dim_in, latex, eq, maff);
442 p = isl_printer_print_str(p, s_to[latex]);
443 p = print_tuple(dim, p, isl_dim_out, latex, eq, maff);
446 return p;
449 static __isl_give isl_printer *print_omega_parameters(__isl_keep isl_space *dim,
450 __isl_take isl_printer *p)
452 if (isl_space_dim(dim, isl_dim_param) == 0)
453 return p;
455 p = isl_printer_start_line(p);
456 p = isl_printer_print_str(p, "symbolic ");
457 p = print_var_list(dim, p, isl_dim_param, 0, NULL, NULL);
458 p = isl_printer_print_str(p, ";");
459 p = isl_printer_end_line(p);
460 return p;
463 static __isl_give isl_printer *print_constraint(struct isl_basic_map *bmap,
464 __isl_keep isl_space *dim, __isl_take isl_printer *p,
465 isl_int *c, int last, const char *op, int first_constraint, int latex)
467 if (!first_constraint)
468 p = isl_printer_print_str(p, s_and[latex]);
470 isl_int_abs(c[last], c[last]);
472 p = print_term(dim, NULL, c[last], last, p, latex);
474 p = isl_printer_print_str(p, " ");
475 p = isl_printer_print_str(p, op);
476 p = isl_printer_print_str(p, " ");
478 isl_int_set_si(c[last], 0);
479 p = print_affine(bmap, dim, p, c);
481 return p;
484 static __isl_give isl_printer *print_constraints(__isl_keep isl_basic_map *bmap,
485 __isl_keep isl_space *dim, __isl_take isl_printer *p, int latex)
487 int i;
488 struct isl_vec *c;
489 unsigned total = isl_basic_map_total_dim(bmap);
491 c = isl_vec_alloc(bmap->ctx, 1 + total);
492 if (!c)
493 goto error;
495 for (i = bmap->n_eq - 1; i >= 0; --i) {
496 int l = isl_seq_last_non_zero(bmap->eq[i], 1 + total);
497 if (l < 0) {
498 if (i != bmap->n_eq - 1)
499 p = isl_printer_print_str(p, s_and[latex]);
500 p = isl_printer_print_str(p, "0 = 0");
501 continue;
503 if (isl_int_is_neg(bmap->eq[i][l]))
504 isl_seq_cpy(c->el, bmap->eq[i], 1 + total);
505 else
506 isl_seq_neg(c->el, bmap->eq[i], 1 + total);
507 p = print_constraint(bmap, dim, p, c->el, l,
508 "=", i == bmap->n_eq - 1, latex);
510 for (i = 0; i < bmap->n_ineq; ++i) {
511 int l = isl_seq_last_non_zero(bmap->ineq[i], 1 + total);
512 int s;
513 const char *op;
514 if (l < 0)
515 continue;
516 s = isl_int_sgn(bmap->ineq[i][l]);
517 if (s < 0)
518 isl_seq_cpy(c->el, bmap->ineq[i], 1 + total);
519 else
520 isl_seq_neg(c->el, bmap->ineq[i], 1 + total);
521 op = s < 0 ? s_le[latex] : s_ge[latex];
522 p = print_constraint(bmap, dim, p, c->el, l,
523 op, !bmap->n_eq && !i, latex);
526 isl_vec_free(c);
528 return p;
529 error:
530 isl_vec_free(c);
531 isl_printer_free(p);
532 return NULL;
535 static __isl_give isl_printer *print_omega_constraints(
536 __isl_keep isl_basic_map *bmap, __isl_take isl_printer *p)
538 if (bmap->n_eq + bmap->n_ineq == 0)
539 return p;
541 p = isl_printer_print_str(p, ": ");
542 if (bmap->n_div > 0) {
543 int i;
544 p = isl_printer_print_str(p, "exists (");
545 for (i = 0; i < bmap->n_div; ++i) {
546 if (i)
547 p = isl_printer_print_str(p, ", ");
548 p = print_name(bmap->dim, p, isl_dim_div, i, 0);
550 p = isl_printer_print_str(p, ": ");
552 p = print_constraints(bmap, bmap->dim, p, 0);
553 if (bmap->n_div > 0)
554 p = isl_printer_print_str(p, ")");
555 return p;
558 static __isl_give isl_printer *basic_map_print_omega(
559 __isl_keep isl_basic_map *bmap, __isl_take isl_printer *p)
561 p = isl_printer_print_str(p, "{ [");
562 p = print_var_list(bmap->dim, p, isl_dim_in, 0, NULL, NULL);
563 p = isl_printer_print_str(p, "] -> [");
564 p = print_var_list(bmap->dim, p, isl_dim_out, 0, NULL, NULL);
565 p = isl_printer_print_str(p, "] ");
566 p = print_omega_constraints(bmap, p);
567 p = isl_printer_print_str(p, " }");
568 return p;
571 static __isl_give isl_printer *basic_set_print_omega(
572 __isl_keep isl_basic_set *bset, __isl_take isl_printer *p)
574 p = isl_printer_print_str(p, "{ [");
575 p = print_var_list(bset->dim, p, isl_dim_set, 0, NULL, NULL);
576 p = isl_printer_print_str(p, "] ");
577 p = print_omega_constraints((isl_basic_map *)bset, p);
578 p = isl_printer_print_str(p, " }");
579 return p;
582 static __isl_give isl_printer *isl_map_print_omega(__isl_keep isl_map *map,
583 __isl_take isl_printer *p)
585 int i;
587 for (i = 0; i < map->n; ++i) {
588 if (i)
589 p = isl_printer_print_str(p, " union ");
590 p = basic_map_print_omega(map->p[i], p);
592 return p;
595 static __isl_give isl_printer *isl_set_print_omega(__isl_keep isl_set *set,
596 __isl_take isl_printer *p)
598 int i;
600 for (i = 0; i < set->n; ++i) {
601 if (i)
602 p = isl_printer_print_str(p, " union ");
603 p = basic_set_print_omega(set->p[i], p);
605 return p;
608 static __isl_give isl_printer *print_disjunct(__isl_keep isl_basic_map *bmap,
609 __isl_keep isl_space *dim, __isl_take isl_printer *p, int latex)
611 if (bmap->n_div > 0) {
612 int i;
613 p = isl_printer_print_str(p, s_open_exists[latex]);
614 for (i = 0; i < bmap->n_div; ++i) {
615 if (i)
616 p = isl_printer_print_str(p, ", ");
617 p = print_name(dim, p, isl_dim_div, i, latex);
618 if (latex || isl_int_is_zero(bmap->div[i][0]))
619 continue;
620 p = isl_printer_print_str(p, " = [(");
621 p = print_affine(bmap, dim, p, bmap->div[i] + 1);
622 p = isl_printer_print_str(p, ")/");
623 p = isl_printer_print_isl_int(p, bmap->div[i][0]);
624 p = isl_printer_print_str(p, "]");
626 p = isl_printer_print_str(p, ": ");
629 p = print_constraints(bmap, dim, p, latex);
631 if (bmap->n_div > 0)
632 p = isl_printer_print_str(p, s_close_exists[latex]);
633 return p;
636 static __isl_give isl_printer *isl_basic_map_print_isl(
637 __isl_keep isl_basic_map *bmap, __isl_take isl_printer *p,
638 int latex)
640 int rational = ISL_F_ISSET(bmap, ISL_BASIC_MAP_RATIONAL);
641 if (isl_basic_map_dim(bmap, isl_dim_param) > 0) {
642 p = print_tuple(bmap->dim, p, isl_dim_param, latex, NULL, NULL);
643 p = isl_printer_print_str(p, " -> ");
645 p = isl_printer_print_str(p, "{ ");
646 p = print_space(bmap->dim, p, latex, rational, NULL, NULL);
647 p = isl_printer_print_str(p, " : ");
648 p = print_disjunct(bmap, bmap->dim, p, latex);
649 p = isl_printer_print_str(p, " }");
650 return p;
653 static __isl_give isl_printer *print_disjuncts(__isl_keep isl_map *map,
654 __isl_take isl_printer *p, int latex)
656 int i;
658 if (isl_map_plain_is_universe(map))
659 return p;
661 p = isl_printer_print_str(p, s_such_that[latex]);
662 if (map->n == 0)
663 p = isl_printer_print_str(p, "1 = 0");
664 for (i = 0; i < map->n; ++i) {
665 if (i)
666 p = isl_printer_print_str(p, s_or[latex]);
667 if (map->n > 1 && map->p[i]->n_eq + map->p[i]->n_ineq > 1)
668 p = isl_printer_print_str(p, "(");
669 p = print_disjunct(map->p[i], map->dim, p, latex);
670 if (map->n > 1 && map->p[i]->n_eq + map->p[i]->n_ineq > 1)
671 p = isl_printer_print_str(p, ")");
673 return p;
676 /* Print the disjuncts of a map (or set).
677 * If the map turns out to be a universal parameter domain, then
678 * we need to print the colon. Otherwise, the output looks identical
679 * to the empty set.
681 static __isl_give isl_printer *print_disjuncts_map(__isl_keep isl_map *map,
682 __isl_take isl_printer *p, int latex)
684 if (isl_map_plain_is_universe(map) && isl_space_is_params(map->dim))
685 return isl_printer_print_str(p, s_such_that[latex]);
686 else
687 return print_disjuncts(map, p, latex);
690 struct isl_aff_split {
691 isl_basic_map *aff;
692 isl_map *map;
695 static void free_split(__isl_take struct isl_aff_split *split, int n)
697 int i;
699 if (!split)
700 return;
702 for (i = 0; i < n; ++i) {
703 isl_basic_map_free(split[i].aff);
704 isl_map_free(split[i].map);
707 free(split);
710 static __isl_give isl_basic_map *get_aff(__isl_take isl_basic_map *bmap)
712 int i, j;
713 unsigned nparam, n_in, n_out, total;
715 bmap = isl_basic_map_cow(bmap);
716 if (!bmap)
717 return NULL;
718 if (isl_basic_map_free_inequality(bmap, bmap->n_ineq) < 0)
719 goto error;
721 nparam = isl_basic_map_dim(bmap, isl_dim_param);
722 n_in = isl_basic_map_dim(bmap, isl_dim_in);
723 n_out = isl_basic_map_dim(bmap, isl_dim_out);
724 total = isl_basic_map_dim(bmap, isl_dim_all);
725 for (i = bmap->n_eq - 1; i >= 0; --i) {
726 j = isl_seq_last_non_zero(bmap->eq[i] + 1, total);
727 if (j >= nparam && j < nparam + n_in + n_out &&
728 (isl_int_is_one(bmap->eq[i][1 + j]) ||
729 isl_int_is_negone(bmap->eq[i][1 + j])))
730 continue;
731 if (isl_basic_map_drop_equality(bmap, i) < 0)
732 goto error;
735 bmap = isl_basic_map_finalize(bmap);
737 return bmap;
738 error:
739 isl_basic_map_free(bmap);
740 return NULL;
743 static int aff_split_cmp(const void *p1, const void *p2, void *user)
745 const struct isl_aff_split *s1, *s2;
746 s1 = (const struct isl_aff_split *) p1;
747 s2 = (const struct isl_aff_split *) p2;
749 return isl_basic_map_plain_cmp(s1->aff, s2->aff);
752 static __isl_give isl_basic_map *drop_aff(__isl_take isl_basic_map *bmap,
753 __isl_keep isl_basic_map *aff)
755 int i, j;
756 unsigned total;
758 if (!bmap || !aff)
759 goto error;
761 total = isl_space_dim(bmap->dim, isl_dim_all);
763 for (i = bmap->n_eq - 1; i >= 0; --i) {
764 if (isl_seq_first_non_zero(bmap->eq[i] + 1 + total,
765 bmap->n_div) != -1)
766 continue;
767 for (j = 0; j < aff->n_eq; ++j) {
768 if (!isl_seq_eq(bmap->eq[i], aff->eq[j], 1 + total) &&
769 !isl_seq_is_neg(bmap->eq[i], aff->eq[j], 1 + total))
770 continue;
771 if (isl_basic_map_drop_equality(bmap, i) < 0)
772 goto error;
773 break;
777 return bmap;
778 error:
779 isl_basic_map_free(bmap);
780 return NULL;
783 static __isl_give struct isl_aff_split *split_aff(__isl_keep isl_map *map)
785 int i, n;
786 struct isl_aff_split *split;
787 isl_ctx *ctx;
789 ctx = isl_map_get_ctx(map);
790 split = isl_calloc_array(ctx, struct isl_aff_split, map->n);
791 if (!split)
792 return NULL;
794 for (i = 0; i < map->n; ++i) {
795 isl_basic_map *bmap;
796 split[i].aff = get_aff(isl_basic_map_copy(map->p[i]));
797 bmap = isl_basic_map_copy(map->p[i]);
798 bmap = isl_basic_map_cow(bmap);
799 bmap = drop_aff(bmap, split[i].aff);
800 split[i].map = isl_map_from_basic_map(bmap);
801 if (!split[i].aff || !split[i].map)
802 goto error;
805 if (isl_sort(split, map->n, sizeof(struct isl_aff_split),
806 &aff_split_cmp, NULL) < 0)
807 goto error;
809 n = map->n;
810 for (i = n - 1; i >= 1; --i) {
811 if (!isl_basic_map_plain_is_equal(split[i - 1].aff,
812 split[i].aff))
813 continue;
814 isl_basic_map_free(split[i].aff);
815 split[i - 1].map = isl_map_union(split[i - 1].map,
816 split[i].map);
817 if (i != n - 1)
818 split[i] = split[n - 1];
819 split[n - 1].aff = NULL;
820 split[n - 1].map = NULL;
821 --n;
824 return split;
825 error:
826 free_split(split, map->n);
827 return NULL;
830 static __isl_give isl_printer *print_split_map(__isl_take isl_printer *p,
831 struct isl_aff_split *split, int n)
833 int i;
834 int rational;
836 for (i = 0; i < n; ++i) {
837 isl_space *dim;
839 if (!split[i].map)
840 break;
841 dim = split[i].map->dim;
842 rational = split[i].map->n > 0 &&
843 ISL_F_ISSET(split[i].map->p[0], ISL_BASIC_MAP_RATIONAL);
844 if (i)
845 p = isl_printer_print_str(p, "; ");
846 p = print_space(dim, p, 0, rational, split[i].aff, NULL);
847 p = print_disjuncts_map(split[i].map, p, 0);
850 return p;
853 static __isl_give isl_printer *isl_map_print_isl_body(__isl_keep isl_map *map,
854 __isl_take isl_printer *p)
856 struct isl_aff_split *split = NULL;
857 int rational;
859 if (map->n > 0)
860 split = split_aff(map);
861 if (split) {
862 p = print_split_map(p, split, map->n);
863 } else {
864 rational = map->n > 0 &&
865 ISL_F_ISSET(map->p[0], ISL_BASIC_MAP_RATIONAL);
866 p = print_space(map->dim, p, 0, rational, NULL, NULL);
867 p = print_disjuncts_map(map, p, 0);
869 free_split(split, map->n);
870 return p;
873 static __isl_give isl_printer *isl_map_print_isl(__isl_keep isl_map *map,
874 __isl_take isl_printer *p)
876 if (isl_map_dim(map, isl_dim_param) > 0) {
877 p = print_tuple(map->dim, p, isl_dim_param, 0, NULL, NULL);
878 p = isl_printer_print_str(p, s_to[0]);
880 p = isl_printer_print_str(p, s_open_set[0]);
881 p = isl_map_print_isl_body(map, p);
882 p = isl_printer_print_str(p, s_close_set[0]);
883 return p;
886 static __isl_give isl_printer *print_latex_map(__isl_keep isl_map *map,
887 __isl_take isl_printer *p, __isl_keep isl_basic_map *aff)
889 if (isl_map_dim(map, isl_dim_param) > 0) {
890 p = print_tuple(map->dim, p, isl_dim_param, 1, NULL, NULL);
891 p = isl_printer_print_str(p, s_to[1]);
893 p = isl_printer_print_str(p, s_open_set[1]);
894 p = print_space(map->dim, p, 1, 0, aff, NULL);
895 p = print_disjuncts_map(map, p, 1);
896 p = isl_printer_print_str(p, s_close_set[1]);
898 return p;
901 static __isl_give isl_printer *isl_map_print_latex(__isl_keep isl_map *map,
902 __isl_take isl_printer *p)
904 int i;
905 struct isl_aff_split *split = NULL;
907 if (map->n > 0)
908 split = split_aff(map);
910 if (!split)
911 return print_latex_map(map, p, NULL);
913 for (i = 0; i < map->n; ++i) {
914 if (!split[i].map)
915 break;
916 if (i)
917 p = isl_printer_print_str(p, " \\cup ");
918 p = print_latex_map(split[i].map, p, split[i].aff);
921 free_split(split, map->n);
922 return p;
925 __isl_give isl_printer *isl_printer_print_basic_map(__isl_take isl_printer *p,
926 __isl_keep isl_basic_map *bmap)
928 if (!p || !bmap)
929 goto error;
930 if (p->output_format == ISL_FORMAT_ISL)
931 return isl_basic_map_print_isl(bmap, p, 0);
932 else if (p->output_format == ISL_FORMAT_OMEGA)
933 return basic_map_print_omega(bmap, p);
934 isl_assert(bmap->ctx, 0, goto error);
935 error:
936 isl_printer_free(p);
937 return NULL;
940 void isl_basic_map_print(__isl_keep isl_basic_map *bmap, FILE *out, int indent,
941 const char *prefix, const char *suffix, unsigned output_format)
943 isl_printer *printer;
945 if (!bmap)
946 return;
948 printer = isl_printer_to_file(bmap->ctx, out);
949 printer = isl_printer_set_indent(printer, indent);
950 printer = isl_printer_set_prefix(printer, prefix);
951 printer = isl_printer_set_suffix(printer, suffix);
952 printer = isl_printer_set_output_format(printer, output_format);
953 isl_printer_print_basic_map(printer, bmap);
955 isl_printer_free(printer);
958 __isl_give isl_printer *isl_printer_print_basic_set(__isl_take isl_printer *p,
959 __isl_keep isl_basic_set *bset)
961 if (!p || !bset)
962 goto error;
964 if (p->output_format == ISL_FORMAT_ISL)
965 return isl_basic_map_print_isl(bset, p, 0);
966 else if (p->output_format == ISL_FORMAT_POLYLIB)
967 return isl_basic_set_print_polylib(bset, p, 0);
968 else if (p->output_format == ISL_FORMAT_EXT_POLYLIB)
969 return isl_basic_set_print_polylib(bset, p, 1);
970 else if (p->output_format == ISL_FORMAT_POLYLIB_CONSTRAINTS)
971 return bset_print_constraints_polylib(bset, p);
972 else if (p->output_format == ISL_FORMAT_OMEGA)
973 return basic_set_print_omega(bset, p);
974 isl_assert(p->ctx, 0, goto error);
975 error:
976 isl_printer_free(p);
977 return NULL;
980 void isl_basic_set_print(struct isl_basic_set *bset, FILE *out, int indent,
981 const char *prefix, const char *suffix, unsigned output_format)
983 isl_printer *printer;
985 if (!bset)
986 return;
988 printer = isl_printer_to_file(bset->ctx, out);
989 printer = isl_printer_set_indent(printer, indent);
990 printer = isl_printer_set_prefix(printer, prefix);
991 printer = isl_printer_set_suffix(printer, suffix);
992 printer = isl_printer_set_output_format(printer, output_format);
993 isl_printer_print_basic_set(printer, bset);
995 isl_printer_free(printer);
998 __isl_give isl_printer *isl_printer_print_set(__isl_take isl_printer *p,
999 __isl_keep isl_set *set)
1001 if (!p || !set)
1002 goto error;
1003 if (p->output_format == ISL_FORMAT_ISL)
1004 return isl_map_print_isl((isl_map *)set, p);
1005 else if (p->output_format == ISL_FORMAT_POLYLIB)
1006 return isl_set_print_polylib(set, p, 0);
1007 else if (p->output_format == ISL_FORMAT_EXT_POLYLIB)
1008 return isl_set_print_polylib(set, p, 1);
1009 else if (p->output_format == ISL_FORMAT_OMEGA)
1010 return isl_set_print_omega(set, p);
1011 else if (p->output_format == ISL_FORMAT_LATEX)
1012 return isl_map_print_latex((isl_map *)set, p);
1013 isl_assert(set->ctx, 0, goto error);
1014 error:
1015 isl_printer_free(p);
1016 return NULL;
1019 void isl_set_print(struct isl_set *set, FILE *out, int indent,
1020 unsigned output_format)
1022 isl_printer *printer;
1024 if (!set)
1025 return;
1027 printer = isl_printer_to_file(set->ctx, out);
1028 printer = isl_printer_set_indent(printer, indent);
1029 printer = isl_printer_set_output_format(printer, output_format);
1030 printer = isl_printer_print_set(printer, set);
1032 isl_printer_free(printer);
1035 __isl_give isl_printer *isl_printer_print_map(__isl_take isl_printer *p,
1036 __isl_keep isl_map *map)
1038 if (!p || !map)
1039 goto error;
1041 if (p->output_format == ISL_FORMAT_ISL)
1042 return isl_map_print_isl(map, p);
1043 else if (p->output_format == ISL_FORMAT_POLYLIB)
1044 return isl_map_print_polylib(map, p, 0);
1045 else if (p->output_format == ISL_FORMAT_EXT_POLYLIB)
1046 return isl_map_print_polylib(map, p, 1);
1047 else if (p->output_format == ISL_FORMAT_OMEGA)
1048 return isl_map_print_omega(map, p);
1049 else if (p->output_format == ISL_FORMAT_LATEX)
1050 return isl_map_print_latex(map, p);
1051 isl_assert(map->ctx, 0, goto error);
1052 error:
1053 isl_printer_free(p);
1054 return NULL;
1057 struct isl_union_print_data {
1058 isl_printer *p;
1059 int first;
1062 static int print_map_body(__isl_take isl_map *map, void *user)
1064 struct isl_union_print_data *data;
1065 data = (struct isl_union_print_data *)user;
1067 if (!data->first)
1068 data->p = isl_printer_print_str(data->p, "; ");
1069 data->first = 0;
1071 data->p = isl_map_print_isl_body(map, data->p);
1072 isl_map_free(map);
1074 return 0;
1077 static __isl_give isl_printer *isl_union_map_print_isl(
1078 __isl_keep isl_union_map *umap, __isl_take isl_printer *p)
1080 struct isl_union_print_data data = { p, 1 };
1081 isl_space *dim;
1082 dim = isl_union_map_get_space(umap);
1083 if (isl_space_dim(dim, isl_dim_param) > 0) {
1084 p = print_tuple(dim, p, isl_dim_param, 0, NULL, NULL);
1085 p = isl_printer_print_str(p, s_to[0]);
1087 isl_space_free(dim);
1088 p = isl_printer_print_str(p, s_open_set[0]);
1089 isl_union_map_foreach_map(umap, &print_map_body, &data);
1090 p = data.p;
1091 p = isl_printer_print_str(p, s_close_set[0]);
1092 return p;
1095 static int print_latex_map_body(__isl_take isl_map *map, void *user)
1097 struct isl_union_print_data *data;
1098 data = (struct isl_union_print_data *)user;
1100 if (!data->first)
1101 data->p = isl_printer_print_str(data->p, " \\cup ");
1102 data->first = 0;
1104 data->p = isl_map_print_latex(map, data->p);
1105 isl_map_free(map);
1107 return 0;
1110 static __isl_give isl_printer *isl_union_map_print_latex(
1111 __isl_keep isl_union_map *umap, __isl_take isl_printer *p)
1113 struct isl_union_print_data data = { p, 1 };
1114 isl_union_map_foreach_map(umap, &print_latex_map_body, &data);
1115 p = data.p;
1116 return p;
1119 __isl_give isl_printer *isl_printer_print_union_map(__isl_take isl_printer *p,
1120 __isl_keep isl_union_map *umap)
1122 if (!p || !umap)
1123 goto error;
1125 if (p->output_format == ISL_FORMAT_ISL)
1126 return isl_union_map_print_isl(umap, p);
1127 if (p->output_format == ISL_FORMAT_LATEX)
1128 return isl_union_map_print_latex(umap, p);
1130 isl_die(p->ctx, isl_error_invalid,
1131 "invalid output format for isl_union_map", goto error);
1132 error:
1133 isl_printer_free(p);
1134 return NULL;
1137 __isl_give isl_printer *isl_printer_print_union_set(__isl_take isl_printer *p,
1138 __isl_keep isl_union_set *uset)
1140 if (!p || !uset)
1141 goto error;
1143 if (p->output_format == ISL_FORMAT_ISL)
1144 return isl_union_map_print_isl((isl_union_map *)uset, p);
1145 if (p->output_format == ISL_FORMAT_LATEX)
1146 return isl_union_map_print_latex((isl_union_map *)uset, p);
1148 isl_die(p->ctx, isl_error_invalid,
1149 "invalid output format for isl_union_set", goto error);
1150 error:
1151 isl_printer_free(p);
1152 return NULL;
1155 void isl_map_print(__isl_keep isl_map *map, FILE *out, int indent,
1156 unsigned output_format)
1158 isl_printer *printer;
1160 if (!map)
1161 return;
1163 printer = isl_printer_to_file(map->ctx, out);
1164 printer = isl_printer_set_indent(printer, indent);
1165 printer = isl_printer_set_output_format(printer, output_format);
1166 printer = isl_printer_print_map(printer, map);
1168 isl_printer_free(printer);
1171 static int upoly_rec_n_non_zero(__isl_keep struct isl_upoly_rec *rec)
1173 int i;
1174 int n;
1176 for (i = 0, n = 0; i < rec->n; ++i)
1177 if (!isl_upoly_is_zero(rec->p[i]))
1178 ++n;
1180 return n;
1183 static __isl_give isl_printer *print_div(__isl_keep isl_space *dim,
1184 __isl_keep isl_mat *div, int pos, __isl_take isl_printer *p)
1186 int c = p->output_format == ISL_FORMAT_C;
1187 p = isl_printer_print_str(p, c ? "floord(" : "[(");
1188 p = print_affine_of_len(dim, div, p,
1189 div->row[pos] + 1, div->n_col - 1);
1190 p = isl_printer_print_str(p, c ? ", " : ")/");
1191 p = isl_printer_print_isl_int(p, div->row[pos][0]);
1192 p = isl_printer_print_str(p, c ? ")" : "]");
1193 return p;
1196 static __isl_give isl_printer *upoly_print_cst(__isl_keep struct isl_upoly *up,
1197 __isl_take isl_printer *p, int first)
1199 struct isl_upoly_cst *cst;
1200 int neg;
1202 cst = isl_upoly_as_cst(up);
1203 if (!cst)
1204 goto error;
1205 neg = !first && isl_int_is_neg(cst->n);
1206 if (!first)
1207 p = isl_printer_print_str(p, neg ? " - " : " + ");
1208 if (neg)
1209 isl_int_neg(cst->n, cst->n);
1210 if (isl_int_is_zero(cst->d)) {
1211 int sgn = isl_int_sgn(cst->n);
1212 p = isl_printer_print_str(p, sgn < 0 ? "-infty" :
1213 sgn == 0 ? "NaN" : "infty");
1214 } else
1215 p = isl_printer_print_isl_int(p, cst->n);
1216 if (neg)
1217 isl_int_neg(cst->n, cst->n);
1218 if (!isl_int_is_zero(cst->d) && !isl_int_is_one(cst->d)) {
1219 p = isl_printer_print_str(p, "/");
1220 p = isl_printer_print_isl_int(p, cst->d);
1222 return p;
1223 error:
1224 isl_printer_free(p);
1225 return NULL;
1228 static __isl_give isl_printer *print_base(__isl_take isl_printer *p,
1229 __isl_keep isl_space *dim, __isl_keep isl_mat *div, int var)
1231 unsigned total;
1233 total = isl_space_dim(dim, isl_dim_all);
1234 if (var < total)
1235 p = print_term(dim, NULL, dim->ctx->one, 1 + var, p, 0);
1236 else
1237 p = print_div(dim, div, var - total, p);
1238 return p;
1241 static __isl_give isl_printer *print_pow(__isl_take isl_printer *p,
1242 __isl_keep isl_space *dim, __isl_keep isl_mat *div, int var, int exp)
1244 p = print_base(p, dim, div, var);
1245 if (exp == 1)
1246 return p;
1247 if (p->output_format == ISL_FORMAT_C) {
1248 int i;
1249 for (i = 1; i < exp; ++i) {
1250 p = isl_printer_print_str(p, "*");
1251 p = print_base(p, dim, div, var);
1253 } else {
1254 p = isl_printer_print_str(p, "^");
1255 p = isl_printer_print_int(p, exp);
1257 return p;
1260 static __isl_give isl_printer *upoly_print(__isl_keep struct isl_upoly *up,
1261 __isl_keep isl_space *dim, __isl_keep isl_mat *div,
1262 __isl_take isl_printer *p, int outer)
1264 int i, n, first, print_parens;
1265 struct isl_upoly_rec *rec;
1267 if (!p || !up || !dim || !div)
1268 goto error;
1270 if (isl_upoly_is_cst(up))
1271 return upoly_print_cst(up, p, 1);
1273 rec = isl_upoly_as_rec(up);
1274 if (!rec)
1275 goto error;
1276 n = upoly_rec_n_non_zero(rec);
1277 print_parens = n > 1 ||
1278 (outer && rec->up.var >= isl_space_dim(dim, isl_dim_all));
1279 if (print_parens)
1280 p = isl_printer_print_str(p, "(");
1281 for (i = 0, first = 1; i < rec->n; ++i) {
1282 if (isl_upoly_is_zero(rec->p[i]))
1283 continue;
1284 if (isl_upoly_is_negone(rec->p[i])) {
1285 if (!i)
1286 p = isl_printer_print_str(p, "-1");
1287 else if (first)
1288 p = isl_printer_print_str(p, "-");
1289 else
1290 p = isl_printer_print_str(p, " - ");
1291 } else if (isl_upoly_is_cst(rec->p[i]) &&
1292 !isl_upoly_is_one(rec->p[i]))
1293 p = upoly_print_cst(rec->p[i], p, first);
1294 else {
1295 if (!first)
1296 p = isl_printer_print_str(p, " + ");
1297 if (i == 0 || !isl_upoly_is_one(rec->p[i]))
1298 p = upoly_print(rec->p[i], dim, div, p, 0);
1300 first = 0;
1301 if (i == 0)
1302 continue;
1303 if (!isl_upoly_is_one(rec->p[i]) &&
1304 !isl_upoly_is_negone(rec->p[i]))
1305 p = isl_printer_print_str(p, " * ");
1306 p = print_pow(p, dim, div, rec->up.var, i);
1308 if (print_parens)
1309 p = isl_printer_print_str(p, ")");
1310 return p;
1311 error:
1312 isl_printer_free(p);
1313 return NULL;
1316 static __isl_give isl_printer *print_qpolynomial(__isl_take isl_printer *p,
1317 __isl_keep isl_qpolynomial *qp)
1319 if (!p || !qp)
1320 goto error;
1321 p = upoly_print(qp->upoly, qp->dim, qp->div, p, 1);
1322 return p;
1323 error:
1324 isl_printer_free(p);
1325 return NULL;
1328 static __isl_give isl_printer *print_qpolynomial_isl(__isl_take isl_printer *p,
1329 __isl_keep isl_qpolynomial *qp)
1331 if (!p || !qp)
1332 goto error;
1334 if (isl_space_dim(qp->dim, isl_dim_param) > 0) {
1335 p = print_tuple(qp->dim, p, isl_dim_param, 0, NULL, NULL);
1336 p = isl_printer_print_str(p, " -> ");
1338 p = isl_printer_print_str(p, "{ ");
1339 if (!isl_space_is_params(qp->dim)) {
1340 p = print_space(qp->dim, p, 0, 0, NULL, NULL);
1341 p = isl_printer_print_str(p, " -> ");
1343 p = print_qpolynomial(p, qp);
1344 p = isl_printer_print_str(p, " }");
1345 return p;
1346 error:
1347 isl_printer_free(p);
1348 return NULL;
1351 static __isl_give isl_printer *print_qpolynomial_c(__isl_take isl_printer *p,
1352 __isl_keep isl_space *dim, __isl_keep isl_qpolynomial *qp)
1354 isl_int den;
1356 isl_int_init(den);
1357 isl_qpolynomial_get_den(qp, &den);
1358 if (!isl_int_is_one(den)) {
1359 isl_qpolynomial *f;
1360 p = isl_printer_print_str(p, "(");
1361 qp = isl_qpolynomial_copy(qp);
1362 f = isl_qpolynomial_rat_cst_on_domain(isl_space_copy(qp->dim),
1363 den, qp->dim->ctx->one);
1364 qp = isl_qpolynomial_mul(qp, f);
1366 if (qp)
1367 p = upoly_print(qp->upoly, dim, qp->div, p, 0);
1368 else
1369 p = isl_printer_free(p);
1370 if (!isl_int_is_one(den)) {
1371 p = isl_printer_print_str(p, ")/");
1372 p = isl_printer_print_isl_int(p, den);
1373 isl_qpolynomial_free(qp);
1375 isl_int_clear(den);
1376 return p;
1379 __isl_give isl_printer *isl_printer_print_qpolynomial(
1380 __isl_take isl_printer *p, __isl_keep isl_qpolynomial *qp)
1382 if (!p || !qp)
1383 goto error;
1385 if (p->output_format == ISL_FORMAT_ISL)
1386 return print_qpolynomial_isl(p, qp);
1387 else if (p->output_format == ISL_FORMAT_C)
1388 return print_qpolynomial_c(p, qp->dim, qp);
1389 else
1390 isl_die(qp->dim->ctx, isl_error_unsupported,
1391 "output format not supported for isl_qpolynomials",
1392 goto error);
1393 error:
1394 isl_printer_free(p);
1395 return NULL;
1398 void isl_qpolynomial_print(__isl_keep isl_qpolynomial *qp, FILE *out,
1399 unsigned output_format)
1401 isl_printer *p;
1403 if (!qp)
1404 return;
1406 isl_assert(qp->dim->ctx, output_format == ISL_FORMAT_ISL, return);
1407 p = isl_printer_to_file(qp->dim->ctx, out);
1408 p = isl_printer_print_qpolynomial(p, qp);
1409 isl_printer_free(p);
1412 static __isl_give isl_printer *qpolynomial_fold_print(
1413 __isl_keep isl_qpolynomial_fold *fold, __isl_take isl_printer *p)
1415 int i;
1417 if (fold->type == isl_fold_min)
1418 p = isl_printer_print_str(p, "min");
1419 else if (fold->type == isl_fold_max)
1420 p = isl_printer_print_str(p, "max");
1421 p = isl_printer_print_str(p, "(");
1422 for (i = 0; i < fold->n; ++i) {
1423 if (i)
1424 p = isl_printer_print_str(p, ", ");
1425 p = print_qpolynomial(p, fold->qp[i]);
1427 p = isl_printer_print_str(p, ")");
1428 return p;
1431 void isl_qpolynomial_fold_print(__isl_keep isl_qpolynomial_fold *fold,
1432 FILE *out, unsigned output_format)
1434 isl_printer *p;
1436 if (!fold)
1437 return;
1439 isl_assert(fold->dim->ctx, output_format == ISL_FORMAT_ISL, return);
1441 p = isl_printer_to_file(fold->dim->ctx, out);
1442 p = isl_printer_print_qpolynomial_fold(p, fold);
1444 isl_printer_free(p);
1447 static __isl_give isl_printer *isl_pwqp_print_isl_body(
1448 __isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial *pwqp)
1450 int i = 0;
1452 for (i = 0; i < pwqp->n; ++i) {
1453 if (i)
1454 p = isl_printer_print_str(p, "; ");
1455 if (!isl_space_is_params(pwqp->p[i].set->dim)) {
1456 p = print_space(pwqp->p[i].set->dim, p, 0, 0, NULL, NULL);
1457 p = isl_printer_print_str(p, " -> ");
1459 p = print_qpolynomial(p, pwqp->p[i].qp);
1460 p = print_disjuncts((isl_map *)pwqp->p[i].set, p, 0);
1463 return p;
1466 static __isl_give isl_printer *print_pw_qpolynomial_isl(
1467 __isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial *pwqp)
1469 if (!p || !pwqp)
1470 goto error;
1472 if (isl_space_dim(pwqp->dim, isl_dim_param) > 0) {
1473 p = print_tuple(pwqp->dim, p, isl_dim_param, 0, NULL, NULL);
1474 p = isl_printer_print_str(p, " -> ");
1476 p = isl_printer_print_str(p, "{ ");
1477 if (pwqp->n == 0) {
1478 if (!isl_space_is_set(pwqp->dim)) {
1479 p = print_tuple(pwqp->dim, p, isl_dim_in, 0, NULL, NULL);
1480 p = isl_printer_print_str(p, " -> ");
1482 p = isl_printer_print_str(p, "0");
1484 p = isl_pwqp_print_isl_body(p, pwqp);
1485 p = isl_printer_print_str(p, " }");
1486 return p;
1487 error:
1488 isl_printer_free(p);
1489 return NULL;
1492 void isl_pw_qpolynomial_print(__isl_keep isl_pw_qpolynomial *pwqp, FILE *out,
1493 unsigned output_format)
1495 isl_printer *p;
1497 if (!pwqp)
1498 return;
1500 p = isl_printer_to_file(pwqp->dim->ctx, out);
1501 p = isl_printer_set_output_format(p, output_format);
1502 p = isl_printer_print_pw_qpolynomial(p, pwqp);
1504 isl_printer_free(p);
1507 static __isl_give isl_printer *isl_pwf_print_isl_body(
1508 __isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial_fold *pwf)
1510 int i = 0;
1512 for (i = 0; i < pwf->n; ++i) {
1513 if (i)
1514 p = isl_printer_print_str(p, "; ");
1515 if (!isl_space_is_params(pwf->p[i].set->dim)) {
1516 p = print_space(pwf->p[i].set->dim, p, 0, 0, NULL, NULL);
1517 p = isl_printer_print_str(p, " -> ");
1519 p = qpolynomial_fold_print(pwf->p[i].fold, p);
1520 p = print_disjuncts((isl_map *)pwf->p[i].set, p, 0);
1523 return p;
1526 static __isl_give isl_printer *print_pw_qpolynomial_fold_isl(
1527 __isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial_fold *pwf)
1529 if (isl_space_dim(pwf->dim, isl_dim_param) > 0) {
1530 p = print_tuple(pwf->dim, p, isl_dim_param, 0, NULL, NULL);
1531 p = isl_printer_print_str(p, " -> ");
1533 p = isl_printer_print_str(p, "{ ");
1534 if (pwf->n == 0) {
1535 if (!isl_space_is_set(pwf->dim)) {
1536 p = print_tuple(pwf->dim, p, isl_dim_in, 0, NULL, NULL);
1537 p = isl_printer_print_str(p, " -> ");
1539 p = isl_printer_print_str(p, "0");
1541 p = isl_pwf_print_isl_body(p, pwf);
1542 p = isl_printer_print_str(p, " }");
1543 return p;
1546 static __isl_give isl_printer *print_affine_c(__isl_take isl_printer *p,
1547 __isl_keep isl_space *dim, __isl_keep isl_basic_set *bset, isl_int *c);
1549 static __isl_give isl_printer *print_name_c(__isl_take isl_printer *p,
1550 __isl_keep isl_space *dim,
1551 __isl_keep isl_basic_set *bset, enum isl_dim_type type, unsigned pos)
1553 if (type == isl_dim_div) {
1554 p = isl_printer_print_str(p, "floord(");
1555 p = print_affine_c(p, dim, bset, bset->div[pos] + 1);
1556 p = isl_printer_print_str(p, ", ");
1557 p = isl_printer_print_isl_int(p, bset->div[pos][0]);
1558 p = isl_printer_print_str(p, ")");
1559 } else {
1560 const char *name;
1562 name = isl_space_get_dim_name(dim, type, pos);
1563 if (!name)
1564 name = "UNNAMED";
1565 p = isl_printer_print_str(p, name);
1567 return p;
1570 static __isl_give isl_printer *print_term_c(__isl_take isl_printer *p,
1571 __isl_keep isl_space *dim,
1572 __isl_keep isl_basic_set *bset, isl_int c, unsigned pos)
1574 enum isl_dim_type type;
1576 if (pos == 0)
1577 return isl_printer_print_isl_int(p, c);
1579 if (isl_int_is_one(c))
1581 else if (isl_int_is_negone(c))
1582 p = isl_printer_print_str(p, "-");
1583 else {
1584 p = isl_printer_print_isl_int(p, c);
1585 p = isl_printer_print_str(p, "*");
1587 type = pos2type(dim, &pos);
1588 p = print_name_c(p, dim, bset, type, pos);
1589 return p;
1592 static __isl_give isl_printer *print_partial_affine_c(__isl_take isl_printer *p,
1593 __isl_keep isl_space *dim,
1594 __isl_keep isl_basic_set *bset, isl_int *c, unsigned len)
1596 int i;
1597 int first;
1599 for (i = 0, first = 1; i < len; ++i) {
1600 int flip = 0;
1601 if (isl_int_is_zero(c[i]))
1602 continue;
1603 if (!first) {
1604 if (isl_int_is_neg(c[i])) {
1605 flip = 1;
1606 isl_int_neg(c[i], c[i]);
1607 p = isl_printer_print_str(p, " - ");
1608 } else
1609 p = isl_printer_print_str(p, " + ");
1611 first = 0;
1612 p = print_term_c(p, dim, bset, c[i], i);
1613 if (flip)
1614 isl_int_neg(c[i], c[i]);
1616 if (first)
1617 p = isl_printer_print_str(p, "0");
1618 return p;
1621 static __isl_give isl_printer *print_affine_c(__isl_take isl_printer *p,
1622 __isl_keep isl_space *dim, __isl_keep isl_basic_set *bset, isl_int *c)
1624 unsigned len = 1 + isl_basic_set_total_dim(bset);
1625 return print_partial_affine_c(p, dim, bset, c, len);
1628 /* We skip the constraint if it is implied by the div expression.
1630 static __isl_give isl_printer *print_constraint_c(__isl_take isl_printer *p,
1631 __isl_keep isl_space *dim,
1632 __isl_keep isl_basic_set *bset, isl_int *c, const char *op, int first)
1634 unsigned o_div;
1635 unsigned n_div;
1636 int div;
1638 o_div = isl_basic_set_offset(bset, isl_dim_div);
1639 n_div = isl_basic_set_dim(bset, isl_dim_div);
1640 div = isl_seq_last_non_zero(c + o_div, n_div);
1641 if (div >= 0 && isl_basic_set_is_div_constraint(bset, c, div))
1642 return p;
1644 if (!first)
1645 p = isl_printer_print_str(p, " && ");
1647 p = print_affine_c(p, dim, bset, c);
1648 p = isl_printer_print_str(p, " ");
1649 p = isl_printer_print_str(p, op);
1650 p = isl_printer_print_str(p, " 0");
1651 return p;
1654 static __isl_give isl_printer *print_basic_set_c(__isl_take isl_printer *p,
1655 __isl_keep isl_space *dim, __isl_keep isl_basic_set *bset)
1657 int i, j;
1658 unsigned n_div = isl_basic_set_dim(bset, isl_dim_div);
1659 unsigned total = isl_basic_set_total_dim(bset) - n_div;
1661 for (i = 0; i < bset->n_eq; ++i) {
1662 j = isl_seq_last_non_zero(bset->eq[i] + 1 + total, n_div);
1663 if (j < 0)
1664 p = print_constraint_c(p, dim, bset,
1665 bset->eq[i], "==", !i);
1666 else {
1667 if (i)
1668 p = isl_printer_print_str(p, " && ");
1669 p = isl_printer_print_str(p, "(");
1670 p = print_partial_affine_c(p, dim, bset, bset->eq[i],
1671 1 + total + j);
1672 p = isl_printer_print_str(p, ") % ");
1673 p = isl_printer_print_isl_int(p,
1674 bset->eq[i][1 + total + j]);
1675 p = isl_printer_print_str(p, " == 0");
1678 for (i = 0; i < bset->n_ineq; ++i)
1679 p = print_constraint_c(p, dim, bset, bset->ineq[i], ">=",
1680 !bset->n_eq && !i);
1681 return p;
1684 static __isl_give isl_printer *print_set_c(__isl_take isl_printer *p,
1685 __isl_keep isl_space *dim, __isl_keep isl_set *set)
1687 int i;
1689 if (!set)
1690 return isl_printer_free(p);
1692 if (set->n == 0)
1693 p = isl_printer_print_str(p, "0");
1695 for (i = 0; i < set->n; ++i) {
1696 if (i)
1697 p = isl_printer_print_str(p, " || ");
1698 if (set->n > 1)
1699 p = isl_printer_print_str(p, "(");
1700 p = print_basic_set_c(p, dim, set->p[i]);
1701 if (set->n > 1)
1702 p = isl_printer_print_str(p, ")");
1704 return p;
1707 static __isl_give isl_printer *print_pw_qpolynomial_c(
1708 __isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial *pwqp)
1710 int i;
1712 if (pwqp->n == 1 && isl_set_plain_is_universe(pwqp->p[0].set))
1713 return print_qpolynomial_c(p, pwqp->dim, pwqp->p[0].qp);
1715 for (i = 0; i < pwqp->n; ++i) {
1716 p = isl_printer_print_str(p, "(");
1717 p = print_set_c(p, pwqp->dim, pwqp->p[i].set);
1718 p = isl_printer_print_str(p, ") ? (");
1719 p = print_qpolynomial_c(p, pwqp->dim, pwqp->p[i].qp);
1720 p = isl_printer_print_str(p, ") : ");
1723 p = isl_printer_print_str(p, "0");
1724 return p;
1727 __isl_give isl_printer *isl_printer_print_pw_qpolynomial(
1728 __isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial *pwqp)
1730 if (!p || !pwqp)
1731 goto error;
1733 if (p->output_format == ISL_FORMAT_ISL)
1734 return print_pw_qpolynomial_isl(p, pwqp);
1735 else if (p->output_format == ISL_FORMAT_C)
1736 return print_pw_qpolynomial_c(p, pwqp);
1737 isl_assert(p->ctx, 0, goto error);
1738 error:
1739 isl_printer_free(p);
1740 return NULL;
1743 static int print_pwqp_body(__isl_take isl_pw_qpolynomial *pwqp, void *user)
1745 struct isl_union_print_data *data;
1746 data = (struct isl_union_print_data *)user;
1748 if (!data->first)
1749 data->p = isl_printer_print_str(data->p, "; ");
1750 data->first = 0;
1752 data->p = isl_pwqp_print_isl_body(data->p, pwqp);
1753 isl_pw_qpolynomial_free(pwqp);
1755 return 0;
1758 static __isl_give isl_printer *print_union_pw_qpolynomial_isl(
1759 __isl_take isl_printer *p, __isl_keep isl_union_pw_qpolynomial *upwqp)
1761 struct isl_union_print_data data = { p, 1 };
1762 isl_space *dim;
1763 dim = isl_union_pw_qpolynomial_get_space(upwqp);
1764 if (isl_space_dim(dim, isl_dim_param) > 0) {
1765 p = print_tuple(dim, p, isl_dim_param, 0, NULL, NULL);
1766 p = isl_printer_print_str(p, " -> ");
1768 isl_space_free(dim);
1769 p = isl_printer_print_str(p, "{ ");
1770 isl_union_pw_qpolynomial_foreach_pw_qpolynomial(upwqp, &print_pwqp_body,
1771 &data);
1772 p = data.p;
1773 p = isl_printer_print_str(p, " }");
1774 return p;
1777 __isl_give isl_printer *isl_printer_print_union_pw_qpolynomial(
1778 __isl_take isl_printer *p, __isl_keep isl_union_pw_qpolynomial *upwqp)
1780 if (!p || !upwqp)
1781 goto error;
1783 if (p->output_format == ISL_FORMAT_ISL)
1784 return print_union_pw_qpolynomial_isl(p, upwqp);
1785 isl_die(p->ctx, isl_error_invalid,
1786 "invalid output format for isl_union_pw_qpolynomial",
1787 goto error);
1788 error:
1789 isl_printer_free(p);
1790 return NULL;
1793 static __isl_give isl_printer *print_qpolynomial_fold_c(
1794 __isl_take isl_printer *p, __isl_keep isl_space *dim,
1795 __isl_keep isl_qpolynomial_fold *fold)
1797 int i;
1799 for (i = 0; i < fold->n - 1; ++i)
1800 if (fold->type == isl_fold_min)
1801 p = isl_printer_print_str(p, "min(");
1802 else if (fold->type == isl_fold_max)
1803 p = isl_printer_print_str(p, "max(");
1805 for (i = 0; i < fold->n; ++i) {
1806 if (i)
1807 p = isl_printer_print_str(p, ", ");
1808 p = print_qpolynomial_c(p, dim, fold->qp[i]);
1809 if (i)
1810 p = isl_printer_print_str(p, ")");
1812 return p;
1815 __isl_give isl_printer *isl_printer_print_qpolynomial_fold(
1816 __isl_take isl_printer *p, __isl_keep isl_qpolynomial_fold *fold)
1818 if (!p || !fold)
1819 goto error;
1820 if (p->output_format == ISL_FORMAT_ISL)
1821 return qpolynomial_fold_print(fold, p);
1822 else if (p->output_format == ISL_FORMAT_C)
1823 return print_qpolynomial_fold_c(p, fold->dim, fold);
1824 isl_die(p->ctx, isl_error_unsupported, "unsupported output format",
1825 goto error);
1826 error:
1827 isl_printer_free(p);
1828 return NULL;
1831 static __isl_give isl_printer *print_pw_qpolynomial_fold_c(
1832 __isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial_fold *pwf)
1834 int i;
1836 if (pwf->n == 1 && isl_set_plain_is_universe(pwf->p[0].set))
1837 return print_qpolynomial_fold_c(p, pwf->dim, pwf->p[0].fold);
1839 for (i = 0; i < pwf->n; ++i) {
1840 p = isl_printer_print_str(p, "(");
1841 p = print_set_c(p, pwf->dim, pwf->p[i].set);
1842 p = isl_printer_print_str(p, ") ? (");
1843 p = print_qpolynomial_fold_c(p, pwf->dim, pwf->p[i].fold);
1844 p = isl_printer_print_str(p, ") : ");
1847 p = isl_printer_print_str(p, "0");
1848 return p;
1851 __isl_give isl_printer *isl_printer_print_pw_qpolynomial_fold(
1852 __isl_take isl_printer *p, __isl_keep isl_pw_qpolynomial_fold *pwf)
1854 if (!p || !pwf)
1855 goto error;
1857 if (p->output_format == ISL_FORMAT_ISL)
1858 return print_pw_qpolynomial_fold_isl(p, pwf);
1859 else if (p->output_format == ISL_FORMAT_C)
1860 return print_pw_qpolynomial_fold_c(p, pwf);
1861 isl_assert(p->ctx, 0, goto error);
1862 error:
1863 isl_printer_free(p);
1864 return NULL;
1867 void isl_pw_qpolynomial_fold_print(__isl_keep isl_pw_qpolynomial_fold *pwf,
1868 FILE *out, unsigned output_format)
1870 isl_printer *p;
1872 if (!pwf)
1873 return;
1875 p = isl_printer_to_file(pwf->dim->ctx, out);
1876 p = isl_printer_set_output_format(p, output_format);
1877 p = isl_printer_print_pw_qpolynomial_fold(p, pwf);
1879 isl_printer_free(p);
1882 static int print_pwf_body(__isl_take isl_pw_qpolynomial_fold *pwf, void *user)
1884 struct isl_union_print_data *data;
1885 data = (struct isl_union_print_data *)user;
1887 if (!data->first)
1888 data->p = isl_printer_print_str(data->p, "; ");
1889 data->first = 0;
1891 data->p = isl_pwf_print_isl_body(data->p, pwf);
1892 isl_pw_qpolynomial_fold_free(pwf);
1894 return 0;
1897 static __isl_give isl_printer *print_union_pw_qpolynomial_fold_isl(
1898 __isl_take isl_printer *p,
1899 __isl_keep isl_union_pw_qpolynomial_fold *upwf)
1901 struct isl_union_print_data data = { p, 1 };
1902 isl_space *dim;
1903 dim = isl_union_pw_qpolynomial_fold_get_space(upwf);
1904 if (isl_space_dim(dim, isl_dim_param) > 0) {
1905 p = print_tuple(dim, p, isl_dim_param, 0, NULL, NULL);
1906 p = isl_printer_print_str(p, " -> ");
1908 isl_space_free(dim);
1909 p = isl_printer_print_str(p, "{ ");
1910 isl_union_pw_qpolynomial_fold_foreach_pw_qpolynomial_fold(upwf,
1911 &print_pwf_body, &data);
1912 p = data.p;
1913 p = isl_printer_print_str(p, " }");
1914 return p;
1917 __isl_give isl_printer *isl_printer_print_union_pw_qpolynomial_fold(
1918 __isl_take isl_printer *p,
1919 __isl_keep isl_union_pw_qpolynomial_fold *upwf)
1921 if (!p || !upwf)
1922 goto error;
1924 if (p->output_format == ISL_FORMAT_ISL)
1925 return print_union_pw_qpolynomial_fold_isl(p, upwf);
1926 isl_die(p->ctx, isl_error_invalid,
1927 "invalid output format for isl_union_pw_qpolynomial_fold",
1928 goto error);
1929 error:
1930 isl_printer_free(p);
1931 return NULL;
1934 __isl_give isl_printer *isl_printer_print_constraint(__isl_take isl_printer *p,
1935 __isl_keep isl_constraint *c)
1937 isl_basic_map *bmap;
1939 if (!p || !c)
1940 goto error;
1942 bmap = isl_basic_map_from_constraint(isl_constraint_copy(c));
1943 p = isl_printer_print_basic_map(p, bmap);
1944 isl_basic_map_free(bmap);
1945 return p;
1946 error:
1947 isl_printer_free(p);
1948 return NULL;
1951 static __isl_give isl_printer *isl_printer_print_space_isl(
1952 __isl_take isl_printer *p, __isl_keep isl_space *dim)
1954 if (!dim)
1955 goto error;
1957 if (isl_space_dim(dim, isl_dim_param) > 0) {
1958 p = print_tuple(dim, p, isl_dim_param, 0, NULL, NULL);
1959 p = isl_printer_print_str(p, " -> ");
1962 p = isl_printer_print_str(p, "{ ");
1963 if (isl_space_is_params(dim))
1964 p = isl_printer_print_str(p, s_such_that[0]);
1965 else
1966 p = print_space(dim, p, 0, 0, NULL, NULL);
1967 p = isl_printer_print_str(p, " }");
1969 return p;
1970 error:
1971 isl_printer_free(p);
1972 return NULL;
1975 __isl_give isl_printer *isl_printer_print_space(__isl_take isl_printer *p,
1976 __isl_keep isl_space *space)
1978 if (!p || !space)
1979 return isl_printer_free(p);
1980 if (p->output_format == ISL_FORMAT_ISL)
1981 return isl_printer_print_space_isl(p, space);
1982 else if (p->output_format == ISL_FORMAT_OMEGA)
1983 return print_omega_parameters(space, p);
1985 isl_die(isl_space_get_ctx(space), isl_error_unsupported,
1986 "output format not supported for space",
1987 return isl_printer_free(p));
1990 __isl_give isl_printer *isl_printer_print_local_space(__isl_take isl_printer *p,
1991 __isl_keep isl_local_space *ls)
1993 unsigned total;
1994 unsigned n_div;
1996 if (!ls)
1997 goto error;
1999 total = isl_local_space_dim(ls, isl_dim_all);
2000 if (isl_local_space_dim(ls, isl_dim_param) > 0) {
2001 p = print_tuple(ls->dim, p, isl_dim_param, 0, NULL, NULL);
2002 p = isl_printer_print_str(p, " -> ");
2004 p = isl_printer_print_str(p, "{ ");
2005 p = print_space(ls->dim, p, 0, 0, NULL, NULL);
2006 n_div = isl_local_space_dim(ls, isl_dim_div);
2007 if (n_div > 0) {
2008 int i;
2009 p = isl_printer_print_str(p, " : ");
2010 p = isl_printer_print_str(p, s_open_exists[0]);
2011 for (i = 0; i < n_div; ++i) {
2012 if (i)
2013 p = isl_printer_print_str(p, ", ");
2014 p = print_name(ls->dim, p, isl_dim_div, i, 0);
2015 if (isl_int_is_zero(ls->div->row[i][0]))
2016 continue;
2017 p = isl_printer_print_str(p, " = [(");
2018 p = print_affine_of_len(ls->dim, ls->div, p,
2019 ls->div->row[i] + 1, 1 + total);
2020 p = isl_printer_print_str(p, ")/");
2021 p = isl_printer_print_isl_int(p, ls->div->row[i][0]);
2022 p = isl_printer_print_str(p, "]");
2024 } else if (isl_space_is_params(ls->dim))
2025 p = isl_printer_print_str(p, s_such_that[0]);
2026 p = isl_printer_print_str(p, " }");
2027 return p;
2028 error:
2029 isl_printer_free(p);
2030 return NULL;
2033 static __isl_give isl_printer *print_aff_body(__isl_take isl_printer *p,
2034 __isl_keep isl_aff *aff)
2036 unsigned total;
2038 total = isl_local_space_dim(aff->ls, isl_dim_all);
2039 p = isl_printer_print_str(p, "(");
2040 p = print_affine_of_len(aff->ls->dim, aff->ls->div, p,
2041 aff->v->el + 1, 1 + total);
2042 if (isl_int_is_one(aff->v->el[0]))
2043 p = isl_printer_print_str(p, ")");
2044 else {
2045 p = isl_printer_print_str(p, ")/");
2046 p = isl_printer_print_isl_int(p, aff->v->el[0]);
2049 return p;
2052 static __isl_give isl_printer *print_aff(__isl_take isl_printer *p,
2053 __isl_keep isl_aff *aff)
2055 if (isl_space_is_params(aff->ls->dim))
2057 else {
2058 p = print_tuple(aff->ls->dim, p, isl_dim_set, 0, NULL, NULL);
2059 p = isl_printer_print_str(p, " -> ");
2061 p = isl_printer_print_str(p, "[");
2062 p = print_aff_body(p, aff);
2063 p = isl_printer_print_str(p, "]");
2065 return p;
2068 static __isl_give isl_printer *print_aff_isl(__isl_take isl_printer *p,
2069 __isl_keep isl_aff *aff)
2071 if (!aff)
2072 goto error;
2074 if (isl_local_space_dim(aff->ls, isl_dim_param) > 0) {
2075 p = print_tuple(aff->ls->dim, p, isl_dim_param, 0, NULL, NULL);
2076 p = isl_printer_print_str(p, " -> ");
2078 p = isl_printer_print_str(p, "{ ");
2079 p = print_aff(p, aff);
2080 p = isl_printer_print_str(p, " }");
2081 return p;
2082 error:
2083 isl_printer_free(p);
2084 return NULL;
2087 static __isl_give isl_printer *print_pw_aff_isl(__isl_take isl_printer *p,
2088 __isl_keep isl_pw_aff *pwaff)
2090 int i;
2092 if (!pwaff)
2093 goto error;
2095 if (isl_space_dim(pwaff->dim, isl_dim_param) > 0) {
2096 p = print_tuple(pwaff->dim, p, isl_dim_param, 0, NULL, NULL);
2097 p = isl_printer_print_str(p, " -> ");
2099 p = isl_printer_print_str(p, "{ ");
2100 for (i = 0; i < pwaff->n; ++i) {
2101 if (i)
2102 p = isl_printer_print_str(p, "; ");
2103 p = print_aff(p, pwaff->p[i].aff);
2104 p = print_disjuncts((isl_map *)pwaff->p[i].set, p, 0);
2106 p = isl_printer_print_str(p, " }");
2107 return p;
2108 error:
2109 isl_printer_free(p);
2110 return NULL;
2113 static __isl_give isl_printer *print_ls_affine_c(__isl_take isl_printer *p,
2114 __isl_keep isl_local_space *ls, isl_int *c);
2116 static __isl_give isl_printer *print_ls_name_c(__isl_take isl_printer *p,
2117 __isl_keep isl_local_space *ls, enum isl_dim_type type, unsigned pos)
2119 if (type == isl_dim_div) {
2120 p = isl_printer_print_str(p, "floord(");
2121 p = print_ls_affine_c(p, ls, ls->div->row[pos] + 1);
2122 p = isl_printer_print_str(p, ", ");
2123 p = isl_printer_print_isl_int(p, ls->div->row[pos][0]);
2124 p = isl_printer_print_str(p, ")");
2125 } else {
2126 const char *name;
2128 name = isl_space_get_dim_name(ls->dim, type, pos);
2129 if (!name)
2130 name = "UNNAMED";
2131 p = isl_printer_print_str(p, name);
2133 return p;
2136 static __isl_give isl_printer *print_ls_term_c(__isl_take isl_printer *p,
2137 __isl_keep isl_local_space *ls, isl_int c, unsigned pos)
2139 enum isl_dim_type type;
2141 if (pos == 0)
2142 return isl_printer_print_isl_int(p, c);
2144 if (isl_int_is_one(c))
2146 else if (isl_int_is_negone(c))
2147 p = isl_printer_print_str(p, "-");
2148 else {
2149 p = isl_printer_print_isl_int(p, c);
2150 p = isl_printer_print_str(p, "*");
2152 type = pos2type(ls->dim, &pos);
2153 p = print_ls_name_c(p, ls, type, pos);
2154 return p;
2157 static __isl_give isl_printer *print_ls_partial_affine_c(
2158 __isl_take isl_printer *p, __isl_keep isl_local_space *ls,
2159 isl_int *c, unsigned len)
2161 int i;
2162 int first;
2164 for (i = 0, first = 1; i < len; ++i) {
2165 int flip = 0;
2166 if (isl_int_is_zero(c[i]))
2167 continue;
2168 if (!first) {
2169 if (isl_int_is_neg(c[i])) {
2170 flip = 1;
2171 isl_int_neg(c[i], c[i]);
2172 p = isl_printer_print_str(p, " - ");
2173 } else
2174 p = isl_printer_print_str(p, " + ");
2176 first = 0;
2177 p = print_ls_term_c(p, ls, c[i], i);
2178 if (flip)
2179 isl_int_neg(c[i], c[i]);
2181 if (first)
2182 p = isl_printer_print_str(p, "0");
2183 return p;
2186 static __isl_give isl_printer *print_ls_affine_c(__isl_take isl_printer *p,
2187 __isl_keep isl_local_space *ls, isl_int *c)
2189 unsigned len = 1 + isl_local_space_dim(ls, isl_dim_all);
2190 return print_ls_partial_affine_c(p, ls, c, len);
2193 static __isl_give isl_printer *print_aff_c(__isl_take isl_printer *p,
2194 __isl_keep isl_aff *aff)
2196 unsigned total;
2198 total = isl_local_space_dim(aff->ls, isl_dim_all);
2199 if (!isl_int_is_one(aff->v->el[0]))
2200 p = isl_printer_print_str(p, "(");
2201 p = print_ls_partial_affine_c(p, aff->ls, aff->v->el + 1, 1 + total);
2202 if (!isl_int_is_one(aff->v->el[0])) {
2203 p = isl_printer_print_str(p, ")/");
2204 p = isl_printer_print_isl_int(p, aff->v->el[0]);
2206 return p;
2209 /* Print the affine expression "aff" in C format to "p".
2210 * The dimension names are taken from "space"
2211 * "set" represents the domain of the affine expression.
2213 static __isl_give isl_printer *print_aff_on_domain_c(__isl_take isl_printer *p,
2214 __isl_keep isl_space *space, __isl_keep isl_aff *aff,
2215 __isl_keep isl_set *set)
2217 isl_set *u;
2218 isl_ast_build *build;
2219 isl_ast_expr *expr;
2221 u = isl_set_universe(isl_space_copy(space));
2222 build = isl_ast_build_from_context(u);
2223 build = isl_ast_build_restrict(build, isl_set_copy(set));
2224 expr = isl_ast_expr_from_aff(isl_aff_copy(aff), build);
2225 p = isl_printer_print_ast_expr(p, expr);
2226 isl_ast_expr_free(expr);
2227 isl_ast_build_free(build);
2229 return p;
2232 /* In the C format, we cannot express that "pwaff" may be undefined
2233 * on parts of the domain space. We therefore assume that the expression
2234 * will only be evaluated on its definition domain and compute the gist
2235 * of each cell with respect to this domain.
2237 static __isl_give isl_printer *print_pw_aff_c(__isl_take isl_printer *p,
2238 __isl_keep isl_pw_aff *pwaff)
2240 int i;
2241 isl_set *domain;
2242 isl_space *space;
2244 if (pwaff->n < 1)
2245 isl_die(p->ctx, isl_error_unsupported,
2246 "cannot print empty isl_pw_aff in C format", goto error);
2247 space = isl_pw_aff_get_domain_space(pwaff);
2248 if (!space)
2249 goto error;
2251 domain = isl_pw_aff_domain(isl_pw_aff_copy(pwaff));
2253 for (i = 0; i < pwaff->n - 1; ++i) {
2254 isl_set *set_i;
2256 p = isl_printer_print_str(p, "(");
2258 set_i = isl_set_copy(pwaff->p[i].set);
2259 set_i = isl_set_gist(set_i, isl_set_copy(domain));
2260 p = print_set_c(p, space, set_i);
2261 isl_set_free(set_i);
2263 p = isl_printer_print_str(p, ") ? (");
2264 p = print_aff_on_domain_c(p, space,
2265 pwaff->p[i].aff, pwaff->p[i].set);
2266 p = isl_printer_print_str(p, ") : ");
2269 isl_set_free(domain);
2271 p = print_aff_on_domain_c(p, space, pwaff->p[pwaff->n - 1].aff,
2272 pwaff->p[pwaff->n - 1].set);
2273 isl_space_free(space);
2274 return p;
2275 error:
2276 isl_printer_free(p);
2277 return NULL;
2280 __isl_give isl_printer *isl_printer_print_aff(__isl_take isl_printer *p,
2281 __isl_keep isl_aff *aff)
2283 if (!p || !aff)
2284 goto error;
2286 if (p->output_format == ISL_FORMAT_ISL)
2287 return print_aff_isl(p, aff);
2288 else if (p->output_format == ISL_FORMAT_C)
2289 return print_aff_c(p, aff);
2290 isl_die(p->ctx, isl_error_unsupported, "unsupported output format",
2291 goto error);
2292 error:
2293 isl_printer_free(p);
2294 return NULL;
2297 __isl_give isl_printer *isl_printer_print_pw_aff(__isl_take isl_printer *p,
2298 __isl_keep isl_pw_aff *pwaff)
2300 if (!p || !pwaff)
2301 goto error;
2303 if (p->output_format == ISL_FORMAT_ISL)
2304 return print_pw_aff_isl(p, pwaff);
2305 else if (p->output_format == ISL_FORMAT_C)
2306 return print_pw_aff_c(p, pwaff);
2307 isl_die(p->ctx, isl_error_unsupported, "unsupported output format",
2308 goto error);
2309 error:
2310 isl_printer_free(p);
2311 return NULL;
2314 static __isl_give isl_printer *print_multi_aff(__isl_take isl_printer *p,
2315 __isl_keep isl_multi_aff *maff)
2317 return print_space(maff->space, p, 0, 0, NULL, maff);
2320 static __isl_give isl_printer *print_multi_aff_isl(__isl_take isl_printer *p,
2321 __isl_keep isl_multi_aff *maff)
2323 if (!maff)
2324 goto error;
2326 if (isl_space_dim(maff->space, isl_dim_param) > 0) {
2327 p = print_tuple(maff->space, p, isl_dim_param, 0, NULL, NULL);
2328 p = isl_printer_print_str(p, " -> ");
2330 p = isl_printer_print_str(p, "{ ");
2331 p = print_multi_aff(p, maff);
2332 p = isl_printer_print_str(p, " }");
2333 return p;
2334 error:
2335 isl_printer_free(p);
2336 return NULL;
2339 __isl_give isl_printer *isl_printer_print_multi_aff(__isl_take isl_printer *p,
2340 __isl_keep isl_multi_aff *maff)
2342 if (!p || !maff)
2343 goto error;
2345 if (p->output_format == ISL_FORMAT_ISL)
2346 return print_multi_aff_isl(p, maff);
2347 isl_die(p->ctx, isl_error_unsupported, "unsupported output format",
2348 goto error);
2349 error:
2350 isl_printer_free(p);
2351 return NULL;
2354 static __isl_give isl_printer *print_pw_multi_aff_body(
2355 __isl_take isl_printer *p, __isl_keep isl_pw_multi_aff *pma)
2357 int i;
2359 if (!pma)
2360 goto error;
2362 for (i = 0; i < pma->n; ++i) {
2363 if (i)
2364 p = isl_printer_print_str(p, "; ");
2365 p = print_multi_aff(p, pma->p[i].maff);
2366 p = print_disjuncts((isl_map *)pma->p[i].set, p, 0);
2368 return p;
2369 error:
2370 isl_printer_free(p);
2371 return NULL;
2374 static __isl_give isl_printer *print_pw_multi_aff_isl(__isl_take isl_printer *p,
2375 __isl_keep isl_pw_multi_aff *pma)
2377 if (!pma)
2378 goto error;
2380 if (isl_space_dim(pma->dim, isl_dim_param) > 0) {
2381 p = print_tuple(pma->dim, p, isl_dim_param, 0, NULL, NULL);
2382 p = isl_printer_print_str(p, " -> ");
2384 p = isl_printer_print_str(p, "{ ");
2385 p = print_pw_multi_aff_body(p, pma);
2386 p = isl_printer_print_str(p, " }");
2387 return p;
2388 error:
2389 isl_printer_free(p);
2390 return NULL;
2393 static __isl_give isl_printer *print_unnamed_pw_multi_aff_c(
2394 __isl_take isl_printer *p, __isl_keep isl_pw_multi_aff *pma)
2396 int i;
2398 for (i = 0; i < pma->n - 1; ++i) {
2399 p = isl_printer_print_str(p, "(");
2400 p = print_set_c(p, pma->dim, pma->p[i].set);
2401 p = isl_printer_print_str(p, ") ? (");
2402 p = print_aff_c(p, pma->p[i].maff->p[0]);
2403 p = isl_printer_print_str(p, ") : ");
2406 return print_aff_c(p, pma->p[pma->n - 1].maff->p[0]);
2409 static __isl_give isl_printer *print_pw_multi_aff_c(__isl_take isl_printer *p,
2410 __isl_keep isl_pw_multi_aff *pma)
2412 int n;
2413 const char *name;
2415 if (!pma)
2416 goto error;
2417 if (pma->n < 1)
2418 isl_die(p->ctx, isl_error_unsupported,
2419 "cannot print empty isl_pw_multi_aff in C format",
2420 goto error);
2421 name = isl_pw_multi_aff_get_tuple_name(pma, isl_dim_out);
2422 if (!name && isl_pw_multi_aff_dim(pma, isl_dim_out) == 1)
2423 return print_unnamed_pw_multi_aff_c(p, pma);
2424 if (!name)
2425 isl_die(p->ctx, isl_error_unsupported,
2426 "cannot print unnamed isl_pw_multi_aff in C format",
2427 goto error);
2429 p = isl_printer_print_str(p, name);
2430 n = isl_pw_multi_aff_dim(pma, isl_dim_out);
2431 if (n != 0)
2432 isl_die(p->ctx, isl_error_unsupported,
2433 "not supported yet", goto error);
2435 return p;
2436 error:
2437 isl_printer_free(p);
2438 return NULL;
2441 __isl_give isl_printer *isl_printer_print_pw_multi_aff(
2442 __isl_take isl_printer *p, __isl_keep isl_pw_multi_aff *pma)
2444 if (!p || !pma)
2445 goto error;
2447 if (p->output_format == ISL_FORMAT_ISL)
2448 return print_pw_multi_aff_isl(p, pma);
2449 if (p->output_format == ISL_FORMAT_C)
2450 return print_pw_multi_aff_c(p, pma);
2451 isl_die(p->ctx, isl_error_unsupported, "unsupported output format",
2452 goto error);
2453 error:
2454 isl_printer_free(p);
2455 return NULL;
2458 static int print_pw_multi_aff_body_wrap(__isl_take isl_pw_multi_aff *pma,
2459 void *user)
2461 struct isl_union_print_data *data;
2462 data = (struct isl_union_print_data *) user;
2464 if (!data->first)
2465 data->p = isl_printer_print_str(data->p, "; ");
2466 data->first = 0;
2468 data->p = print_pw_multi_aff_body(data->p, pma);
2469 isl_pw_multi_aff_free(pma);
2471 return 0;
2474 static __isl_give isl_printer *print_union_pw_multi_aff_isl(
2475 __isl_take isl_printer *p, __isl_keep isl_union_pw_multi_aff *upma)
2477 struct isl_union_print_data data = { p, 1 };
2478 isl_space *space;
2480 space = isl_union_pw_multi_aff_get_space(upma);
2481 if (isl_space_dim(space, isl_dim_param) > 0) {
2482 p = print_tuple(space, p, isl_dim_param, 0, NULL, NULL);
2483 p = isl_printer_print_str(p, s_to[0]);
2485 isl_space_free(space);
2486 p = isl_printer_print_str(p, s_open_set[0]);
2487 isl_union_pw_multi_aff_foreach_pw_multi_aff(upma,
2488 &print_pw_multi_aff_body_wrap, &data);
2489 p = data.p;
2490 p = isl_printer_print_str(p, s_close_set[0]);
2491 return p;
2494 __isl_give isl_printer *isl_printer_print_union_pw_multi_aff(
2495 __isl_take isl_printer *p, __isl_keep isl_union_pw_multi_aff *upma)
2497 if (!p || !upma)
2498 goto error;
2500 if (p->output_format == ISL_FORMAT_ISL)
2501 return print_union_pw_multi_aff_isl(p, upma);
2502 isl_die(p->ctx, isl_error_unsupported, "unsupported output format",
2503 goto error);
2504 error:
2505 isl_printer_free(p);
2506 return NULL;
2509 static __isl_give isl_printer *print_multi_pw_aff_isl(__isl_take isl_printer *p,
2510 __isl_keep isl_multi_pw_aff *mpa)
2512 int i;
2514 if (!mpa)
2515 return isl_printer_free(p);
2517 p = isl_printer_print_str(p, "(");
2518 for (i = 0; i < mpa->n; ++i) {
2519 if (i)
2520 p = isl_printer_print_str(p, ",");
2521 p = isl_printer_print_pw_aff(p, mpa->p[i]);
2523 p = isl_printer_print_str(p, ")");
2524 return p;
2527 __isl_give isl_printer *isl_printer_print_multi_pw_aff(
2528 __isl_take isl_printer *p, __isl_keep isl_multi_pw_aff *mpa)
2530 if (!p || !mpa)
2531 return isl_printer_free(p);
2533 if (p->output_format == ISL_FORMAT_ISL)
2534 return print_multi_pw_aff_isl(p, mpa);
2535 isl_die(p->ctx, isl_error_unsupported, "unsupported output format",
2536 return isl_printer_free(p));