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
17 #include <isl_ctx_private.h>
18 #include <isl_map_private.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>
33 #include <isl_output_private.h>
35 static const char *s_to
[2] = { " -> ", " \\to " };
36 static const char *s_and
[2] = { " and ", " \\wedge " };
37 static const char *s_or
[2] = { " or ", " \\vee " };
38 static const char *s_le
[2] = { "<=", "\\le" };
39 static const char *s_ge
[2] = { ">=", "\\ge" };
40 static const char *s_open_set
[2] = { "{ ", "\\{\\, " };
41 static const char *s_close_set
[2] = { " }", " \\,\\}" };
42 static const char *s_open_list
[2] = { "[", "(" };
43 static const char *s_close_list
[2] = { "]", ")" };
44 static const char *s_such_that
[2] = { " : ", " \\mid " };
45 static const char *s_open_exists
[2] = { "exists (", "\\exists \\, " };
46 static const char *s_close_exists
[2] = { ")", "" };
47 static const char *s_div_prefix
[2] = { "e", "\\alpha_" };
48 static const char *s_param_prefix
[2] = { "p", "p_" };
49 static const char *s_input_prefix
[2] = { "i", "i_" };
50 static const char *s_output_prefix
[2] = { "o", "o_" };
52 static __isl_give isl_printer
*print_constraint_polylib(
53 struct isl_basic_map
*bmap
, int ineq
, int n
, __isl_take isl_printer
*p
)
56 unsigned n_in
= isl_basic_map_dim(bmap
, isl_dim_in
);
57 unsigned n_out
= isl_basic_map_dim(bmap
, isl_dim_out
);
58 unsigned nparam
= isl_basic_map_dim(bmap
, isl_dim_param
);
59 isl_int
*c
= ineq
? bmap
->ineq
[n
] : bmap
->eq
[n
];
61 p
= isl_printer_start_line(p
);
62 p
= isl_printer_print_int(p
, ineq
);
63 for (i
= 0; i
< n_out
; ++i
) {
64 p
= isl_printer_print_str(p
, " ");
65 p
= isl_printer_print_isl_int(p
, c
[1+nparam
+n_in
+i
]);
67 for (i
= 0; i
< n_in
; ++i
) {
68 p
= isl_printer_print_str(p
, " ");
69 p
= isl_printer_print_isl_int(p
, c
[1+nparam
+i
]);
71 for (i
= 0; i
< bmap
->n_div
; ++i
) {
72 p
= isl_printer_print_str(p
, " ");
73 p
= isl_printer_print_isl_int(p
, c
[1+nparam
+n_in
+n_out
+i
]);
75 for (i
= 0; i
< nparam
; ++i
) {
76 p
= isl_printer_print_str(p
, " ");
77 p
= isl_printer_print_isl_int(p
, c
[1+i
]);
79 p
= isl_printer_print_str(p
, " ");
80 p
= isl_printer_print_isl_int(p
, c
[0]);
81 p
= isl_printer_end_line(p
);
85 static __isl_give isl_printer
*print_constraints_polylib(
86 struct isl_basic_map
*bmap
, __isl_take isl_printer
*p
)
90 p
= isl_printer_set_isl_int_width(p
, 5);
92 for (i
= 0; i
< bmap
->n_eq
; ++i
)
93 p
= print_constraint_polylib(bmap
, 0, i
, p
);
94 for (i
= 0; i
< bmap
->n_ineq
; ++i
)
95 p
= print_constraint_polylib(bmap
, 1, i
, p
);
100 static __isl_give isl_printer
*bset_print_constraints_polylib(
101 struct isl_basic_set
*bset
, __isl_take isl_printer
*p
)
103 return print_constraints_polylib((struct isl_basic_map
*)bset
, p
);
106 static __isl_give isl_printer
*isl_basic_map_print_polylib(
107 __isl_keep isl_basic_map
*bmap
, __isl_take isl_printer
*p
, int ext
)
109 unsigned total
= isl_basic_map_total_dim(bmap
);
110 p
= isl_printer_start_line(p
);
111 p
= isl_printer_print_int(p
, bmap
->n_eq
+ bmap
->n_ineq
);
112 p
= isl_printer_print_str(p
, " ");
113 p
= isl_printer_print_int(p
, 1 + total
+ 1);
115 p
= isl_printer_print_str(p
, " ");
116 p
= isl_printer_print_int(p
,
117 isl_basic_map_dim(bmap
, isl_dim_out
));
118 p
= isl_printer_print_str(p
, " ");
119 p
= isl_printer_print_int(p
,
120 isl_basic_map_dim(bmap
, isl_dim_in
));
121 p
= isl_printer_print_str(p
, " ");
122 p
= isl_printer_print_int(p
,
123 isl_basic_map_dim(bmap
, isl_dim_div
));
124 p
= isl_printer_print_str(p
, " ");
125 p
= isl_printer_print_int(p
,
126 isl_basic_map_dim(bmap
, isl_dim_param
));
128 p
= isl_printer_end_line(p
);
129 return print_constraints_polylib(bmap
, p
);
132 static __isl_give isl_printer
*isl_basic_set_print_polylib(
133 __isl_keep isl_basic_set
*bset
, __isl_take isl_printer
*p
, int ext
)
135 return isl_basic_map_print_polylib((struct isl_basic_map
*)bset
, p
, ext
);
138 static __isl_give isl_printer
*isl_map_print_polylib(__isl_keep isl_map
*map
,
139 __isl_take isl_printer
*p
, int ext
)
143 p
= isl_printer_start_line(p
);
144 p
= isl_printer_print_int(p
, map
->n
);
145 p
= isl_printer_end_line(p
);
146 for (i
= 0; i
< map
->n
; ++i
) {
147 p
= isl_printer_start_line(p
);
148 p
= isl_printer_end_line(p
);
149 p
= isl_basic_map_print_polylib(map
->p
[i
], p
, ext
);
154 static __isl_give isl_printer
*isl_set_print_polylib(__isl_keep isl_set
*set
,
155 __isl_take isl_printer
*p
, int ext
)
157 return isl_map_print_polylib((struct isl_map
*)set
, p
, ext
);
160 static int count_same_name(__isl_keep isl_space
*dim
,
161 enum isl_dim_type type
, unsigned pos
, const char *name
)
167 for (t
= isl_dim_param
; t
<= type
&& t
<= isl_dim_out
; ++t
) {
168 s
= t
== type
? pos
: isl_space_dim(dim
, t
);
169 for (p
= 0; p
< s
; ++p
) {
170 const char *n
= isl_space_get_dim_name(dim
, t
, p
);
171 if (n
&& !strcmp(n
, name
))
178 static __isl_give isl_printer
*print_name(__isl_keep isl_space
*dim
,
179 __isl_take isl_printer
*p
, enum isl_dim_type type
, unsigned pos
,
186 name
= type
== isl_dim_div
? NULL
: isl_space_get_dim_name(dim
, type
, pos
);
190 if (type
== isl_dim_param
)
191 prefix
= s_param_prefix
[latex
];
192 else if (type
== isl_dim_div
)
193 prefix
= s_div_prefix
[latex
];
194 else if (isl_space_is_set(dim
) || type
== isl_dim_in
)
195 prefix
= s_input_prefix
[latex
];
197 prefix
= s_output_prefix
[latex
];
198 snprintf(buffer
, sizeof(buffer
), "%s%d", prefix
, pos
);
201 primes
= count_same_name(dim
, name
== buffer
? isl_dim_div
: type
,
203 p
= isl_printer_print_str(p
, name
);
205 p
= isl_printer_print_str(p
, "'");
209 static enum isl_dim_type
pos2type(__isl_keep isl_space
*dim
, unsigned *pos
)
211 enum isl_dim_type type
;
212 unsigned n_in
= isl_space_dim(dim
, isl_dim_in
);
213 unsigned n_out
= isl_space_dim(dim
, isl_dim_out
);
214 unsigned nparam
= isl_space_dim(dim
, isl_dim_param
);
216 if (*pos
< 1 + nparam
) {
217 type
= isl_dim_param
;
219 } else if (*pos
< 1 + nparam
+ n_in
) {
222 } else if (*pos
< 1 + nparam
+ n_in
+ n_out
) {
224 *pos
-= 1 + nparam
+ n_in
;
227 *pos
-= 1 + nparam
+ n_in
+ n_out
;
233 static __isl_give isl_printer
*print_div(__isl_keep isl_space
*dim
,
234 __isl_keep isl_mat
*div
, int pos
, __isl_take isl_printer
*p
);
236 static __isl_give isl_printer
*print_term(__isl_keep isl_space
*dim
,
237 __isl_keep isl_mat
*div
,
238 isl_int c
, unsigned pos
, __isl_take isl_printer
*p
, int latex
)
240 enum isl_dim_type type
;
244 return isl_printer_print_isl_int(p
, c
);
246 type
= pos2type(dim
, &pos
);
247 print_div_def
= type
== isl_dim_div
&& div
&&
248 !isl_int_is_zero(div
->row
[pos
][0]);
250 if (isl_int_is_one(c
))
252 else if (isl_int_is_negone(c
))
253 p
= isl_printer_print_str(p
, "-");
255 p
= isl_printer_print_isl_int(p
, c
);
256 if (p
->output_format
== ISL_FORMAT_C
|| print_div_def
)
257 p
= isl_printer_print_str(p
, "*");
260 p
= print_div(dim
, div
, pos
, p
);
262 p
= print_name(dim
, p
, type
, pos
, latex
);
266 static __isl_give isl_printer
*print_affine_of_len(__isl_keep isl_space
*dim
,
267 __isl_keep isl_mat
*div
,
268 __isl_take isl_printer
*p
, isl_int
*c
, int len
)
273 for (i
= 0, first
= 1; i
< len
; ++i
) {
275 if (isl_int_is_zero(c
[i
]))
278 if (isl_int_is_neg(c
[i
])) {
280 isl_int_neg(c
[i
], c
[i
]);
281 p
= isl_printer_print_str(p
, " - ");
283 p
= isl_printer_print_str(p
, " + ");
286 p
= print_term(dim
, div
, c
[i
], i
, p
, 0);
288 isl_int_neg(c
[i
], c
[i
]);
291 p
= isl_printer_print_str(p
, "0");
295 static __isl_give isl_printer
*print_affine(__isl_keep isl_basic_map
*bmap
,
296 __isl_keep isl_space
*dim
, __isl_take isl_printer
*p
, isl_int
*c
)
298 unsigned len
= 1 + isl_basic_map_total_dim(bmap
);
299 return print_affine_of_len(dim
, NULL
, p
, c
, len
);
302 /* offset is the offset of local_dim inside data->type of data->space.
304 static __isl_give isl_printer
*print_nested_var_list(__isl_take isl_printer
*p
,
305 __isl_keep isl_space
*local_dim
, enum isl_dim_type local_type
,
306 struct isl_print_space_data
*data
, int offset
)
310 if (data
->space
!= local_dim
&& local_type
== isl_dim_out
)
311 offset
+= local_dim
->n_in
;
313 for (i
= 0; i
< isl_space_dim(local_dim
, local_type
); ++i
) {
315 p
= isl_printer_print_str(p
, ", ");
317 p
= data
->print_dim(p
, data
, offset
+ i
);
319 p
= print_name(data
->space
, p
, data
->type
, offset
+ i
,
325 static __isl_give isl_printer
*print_var_list(__isl_take isl_printer
*p
,
326 __isl_keep isl_space
*space
, enum isl_dim_type type
)
328 struct isl_print_space_data data
= { .space
= space
, .type
= type
};
330 return print_nested_var_list(p
, space
, type
, &data
, 0);
333 static __isl_give isl_printer
*print_nested_map_dim(__isl_take isl_printer
*p
,
334 __isl_keep isl_space
*local_dim
,
335 struct isl_print_space_data
*data
, int offset
);
337 static __isl_give isl_printer
*print_nested_tuple(__isl_take isl_printer
*p
,
338 __isl_keep isl_space
*local_dim
, enum isl_dim_type local_type
,
339 struct isl_print_space_data
*data
, int offset
)
341 const char *name
= NULL
;
342 unsigned n
= isl_space_dim(local_dim
, local_type
);
343 if ((local_type
== isl_dim_in
|| local_type
== isl_dim_out
)) {
344 name
= isl_space_get_tuple_name(local_dim
, local_type
);
347 p
= isl_printer_print_str(p
, "\\mathrm{");
348 p
= isl_printer_print_str(p
, name
);
350 p
= isl_printer_print_str(p
, "}");
353 if (!data
->latex
|| n
!= 1 || name
)
354 p
= isl_printer_print_str(p
, s_open_list
[data
->latex
]);
355 if ((local_type
== isl_dim_in
|| local_type
== isl_dim_out
) &&
356 local_dim
->nested
[local_type
- isl_dim_in
]) {
357 if (data
->space
!= local_dim
&& local_type
== isl_dim_out
)
358 offset
+= local_dim
->n_in
;
359 p
= print_nested_map_dim(p
,
360 local_dim
->nested
[local_type
- isl_dim_in
],
363 p
= print_nested_var_list(p
, local_dim
, local_type
, data
,
365 if (!data
->latex
|| n
!= 1 || name
)
366 p
= isl_printer_print_str(p
, s_close_list
[data
->latex
]);
370 static __isl_give isl_printer
*print_tuple(__isl_keep isl_space
*dim
,
371 __isl_take isl_printer
*p
, enum isl_dim_type type
,
372 struct isl_print_space_data
*data
)
376 return print_nested_tuple(p
, dim
, type
, data
, 0);
379 static __isl_give isl_printer
*print_nested_map_dim(__isl_take isl_printer
*p
,
380 __isl_keep isl_space
*local_dim
,
381 struct isl_print_space_data
*data
, int offset
)
383 p
= print_nested_tuple(p
, local_dim
, isl_dim_in
, data
, offset
);
384 p
= isl_printer_print_str(p
, s_to
[data
->latex
]);
385 p
= print_nested_tuple(p
, local_dim
, isl_dim_out
, data
, offset
);
390 __isl_give isl_printer
*isl_print_space(__isl_keep isl_space
*space
,
391 __isl_take isl_printer
*p
, int rational
,
392 struct isl_print_space_data
*data
)
394 if (rational
&& !data
->latex
)
395 p
= isl_printer_print_str(p
, "rat: ");
396 if (isl_space_is_params(space
))
398 else if (isl_space_is_set(space
))
399 p
= print_tuple(space
, p
, isl_dim_set
, data
);
401 p
= print_tuple(space
, p
, isl_dim_in
, data
);
402 p
= isl_printer_print_str(p
, s_to
[data
->latex
]);
403 p
= print_tuple(space
, p
, isl_dim_out
, data
);
409 static __isl_give isl_printer
*print_omega_parameters(__isl_keep isl_space
*dim
,
410 __isl_take isl_printer
*p
)
412 if (isl_space_dim(dim
, isl_dim_param
) == 0)
415 p
= isl_printer_start_line(p
);
416 p
= isl_printer_print_str(p
, "symbolic ");
417 p
= print_var_list(p
, dim
, isl_dim_param
);
418 p
= isl_printer_print_str(p
, ";");
419 p
= isl_printer_end_line(p
);
423 static __isl_give isl_printer
*print_constraint(struct isl_basic_map
*bmap
,
424 __isl_keep isl_space
*dim
, __isl_take isl_printer
*p
,
425 isl_int
*c
, int last
, const char *op
, int first_constraint
, int latex
)
427 if (!first_constraint
)
428 p
= isl_printer_print_str(p
, s_and
[latex
]);
430 isl_int_abs(c
[last
], c
[last
]);
432 p
= print_term(dim
, NULL
, c
[last
], last
, p
, latex
);
434 p
= isl_printer_print_str(p
, " ");
435 p
= isl_printer_print_str(p
, op
);
436 p
= isl_printer_print_str(p
, " ");
438 isl_int_set_si(c
[last
], 0);
439 p
= print_affine(bmap
, dim
, p
, c
);
444 static __isl_give isl_printer
*print_constraints(__isl_keep isl_basic_map
*bmap
,
445 __isl_keep isl_space
*dim
, __isl_take isl_printer
*p
, int latex
)
449 unsigned total
= isl_basic_map_total_dim(bmap
);
451 c
= isl_vec_alloc(bmap
->ctx
, 1 + total
);
455 for (i
= bmap
->n_eq
- 1; i
>= 0; --i
) {
456 int l
= isl_seq_last_non_zero(bmap
->eq
[i
], 1 + total
);
458 if (i
!= bmap
->n_eq
- 1)
459 p
= isl_printer_print_str(p
, s_and
[latex
]);
460 p
= isl_printer_print_str(p
, "0 = 0");
463 if (isl_int_is_neg(bmap
->eq
[i
][l
]))
464 isl_seq_cpy(c
->el
, bmap
->eq
[i
], 1 + total
);
466 isl_seq_neg(c
->el
, bmap
->eq
[i
], 1 + total
);
467 p
= print_constraint(bmap
, dim
, p
, c
->el
, l
,
468 "=", i
== bmap
->n_eq
- 1, latex
);
470 for (i
= 0; i
< bmap
->n_ineq
; ++i
) {
471 int l
= isl_seq_last_non_zero(bmap
->ineq
[i
], 1 + total
);
476 s
= isl_int_sgn(bmap
->ineq
[i
][l
]);
478 isl_seq_cpy(c
->el
, bmap
->ineq
[i
], 1 + total
);
480 isl_seq_neg(c
->el
, bmap
->ineq
[i
], 1 + total
);
481 op
= s
< 0 ? s_le
[latex
] : s_ge
[latex
];
482 p
= print_constraint(bmap
, dim
, p
, c
->el
, l
,
483 op
, !bmap
->n_eq
&& !i
, latex
);
495 static __isl_give isl_printer
*print_div(__isl_keep isl_space
*dim
,
496 __isl_keep isl_mat
*div
, int pos
, __isl_take isl_printer
*p
)
501 return isl_printer_free(p
);
503 c
= p
->output_format
== ISL_FORMAT_C
;
504 p
= isl_printer_print_str(p
, c
? "floord(" : "floor((");
505 p
= print_affine_of_len(dim
, div
, p
,
506 div
->row
[pos
] + 1, div
->n_col
- 1);
507 p
= isl_printer_print_str(p
, c
? ", " : ")/");
508 p
= isl_printer_print_isl_int(p
, div
->row
[pos
][0]);
509 p
= isl_printer_print_str(p
, ")");
513 /* Print a comma separated list of div names, with their definitions
514 * (provided that they have a definition and we are printing in isl format).
516 static __isl_give isl_printer
*print_div_list(__isl_take isl_printer
*p
,
517 __isl_keep isl_space
*space
, __isl_keep isl_mat
*div
, int latex
)
522 if (!p
|| !space
|| !div
)
523 return isl_printer_free(p
);
525 n_div
= isl_mat_rows(div
);
527 for (i
= 0; i
< n_div
; ++i
) {
529 p
= isl_printer_print_str(p
, ", ");
530 p
= print_name(space
, p
, isl_dim_div
, i
, latex
);
531 if (p
->output_format
!= ISL_FORMAT_ISL
||
532 isl_int_is_zero(div
->row
[i
][0]))
534 p
= isl_printer_print_str(p
, " = ");
535 p
= print_div(space
, div
, i
, p
);
541 static __isl_give isl_printer
*print_disjunct(__isl_keep isl_basic_map
*bmap
,
542 __isl_keep isl_space
*space
, __isl_take isl_printer
*p
, int latex
)
544 if (bmap
->n_div
> 0) {
547 div
= isl_basic_map_get_divs(bmap
);
548 p
= isl_printer_print_str(p
, s_open_exists
[latex
]);
549 p
= print_div_list(p
, space
, div
, latex
);
551 p
= isl_printer_print_str(p
, ": ");
554 p
= print_constraints(bmap
, space
, p
, latex
);
557 p
= isl_printer_print_str(p
, s_close_exists
[latex
]);
561 /* Print a colon followed by the constraints of "bmap"
562 * to "p", provided there are any constraints.
563 * The names of the variables are taken from "space".
564 * "latex" is set if the constraints should be printed in LaTeX format.
566 static __isl_give isl_printer
*print_optional_disjunct(
567 __isl_keep isl_basic_map
*bmap
, __isl_keep isl_space
*space
,
568 __isl_take isl_printer
*p
, int latex
)
570 if (isl_basic_map_is_universe(bmap
))
573 p
= isl_printer_print_str(p
, ": ");
574 p
= print_disjunct(bmap
, space
, p
, latex
);
579 static __isl_give isl_printer
*basic_map_print_omega(
580 __isl_keep isl_basic_map
*bmap
, __isl_take isl_printer
*p
)
582 p
= isl_printer_print_str(p
, "{ [");
583 p
= print_var_list(p
, bmap
->dim
, isl_dim_in
);
584 p
= isl_printer_print_str(p
, "] -> [");
585 p
= print_var_list(p
, bmap
->dim
, isl_dim_out
);
586 p
= isl_printer_print_str(p
, "] ");
587 p
= print_optional_disjunct(bmap
, bmap
->dim
, p
, 0);
588 p
= isl_printer_print_str(p
, " }");
592 static __isl_give isl_printer
*basic_set_print_omega(
593 __isl_keep isl_basic_set
*bset
, __isl_take isl_printer
*p
)
595 p
= isl_printer_print_str(p
, "{ [");
596 p
= print_var_list(p
, bset
->dim
, isl_dim_set
);
597 p
= isl_printer_print_str(p
, "] ");
598 p
= print_optional_disjunct(bset
, bset
->dim
, p
, 0);
599 p
= isl_printer_print_str(p
, " }");
603 static __isl_give isl_printer
*isl_map_print_omega(__isl_keep isl_map
*map
,
604 __isl_take isl_printer
*p
)
608 for (i
= 0; i
< map
->n
; ++i
) {
610 p
= isl_printer_print_str(p
, " union ");
611 p
= basic_map_print_omega(map
->p
[i
], p
);
616 static __isl_give isl_printer
*isl_set_print_omega(__isl_keep isl_set
*set
,
617 __isl_take isl_printer
*p
)
621 for (i
= 0; i
< set
->n
; ++i
) {
623 p
= isl_printer_print_str(p
, " union ");
624 p
= basic_set_print_omega(set
->p
[i
], p
);
629 static __isl_give isl_printer
*isl_basic_map_print_isl(
630 __isl_keep isl_basic_map
*bmap
, __isl_take isl_printer
*p
,
633 struct isl_print_space_data data
= { .latex
= latex
};
634 int rational
= ISL_F_ISSET(bmap
, ISL_BASIC_MAP_RATIONAL
);
636 if (isl_basic_map_dim(bmap
, isl_dim_param
) > 0) {
637 p
= print_tuple(bmap
->dim
, p
, isl_dim_param
, &data
);
638 p
= isl_printer_print_str(p
, " -> ");
640 p
= isl_printer_print_str(p
, "{ ");
641 p
= isl_print_space(bmap
->dim
, p
, rational
, &data
);
642 p
= isl_printer_print_str(p
, " : ");
643 p
= print_disjunct(bmap
, bmap
->dim
, p
, latex
);
644 p
= isl_printer_print_str(p
, " }");
648 static __isl_give isl_printer
*print_disjuncts(__isl_keep isl_map
*map
,
649 __isl_take isl_printer
*p
, int latex
)
653 if (isl_map_plain_is_universe(map
))
656 p
= isl_printer_print_str(p
, s_such_that
[latex
]);
658 p
= isl_printer_print_str(p
, "1 = 0");
659 for (i
= 0; i
< map
->n
; ++i
) {
661 p
= isl_printer_print_str(p
, s_or
[latex
]);
662 if (map
->n
> 1 && map
->p
[i
]->n_eq
+ map
->p
[i
]->n_ineq
> 1)
663 p
= isl_printer_print_str(p
, "(");
664 p
= print_disjunct(map
->p
[i
], map
->dim
, p
, latex
);
665 if (map
->n
> 1 && map
->p
[i
]->n_eq
+ map
->p
[i
]->n_ineq
> 1)
666 p
= isl_printer_print_str(p
, ")");
671 /* Print the disjuncts of a map (or set).
672 * If the map turns out to be a universal parameter domain, then
673 * we need to print the colon. Otherwise, the output looks identical
676 static __isl_give isl_printer
*print_disjuncts_map(__isl_keep isl_map
*map
,
677 __isl_take isl_printer
*p
, int latex
)
679 if (isl_map_plain_is_universe(map
) && isl_space_is_params(map
->dim
))
680 return isl_printer_print_str(p
, s_such_that
[latex
]);
682 return print_disjuncts(map
, p
, latex
);
685 struct isl_aff_split
{
690 static void free_split(__isl_take
struct isl_aff_split
*split
, int n
)
697 for (i
= 0; i
< n
; ++i
) {
698 isl_basic_map_free(split
[i
].aff
);
699 isl_map_free(split
[i
].map
);
705 static __isl_give isl_basic_map
*get_aff(__isl_take isl_basic_map
*bmap
)
708 unsigned nparam
, n_in
, n_out
, total
;
710 bmap
= isl_basic_map_cow(bmap
);
713 if (isl_basic_map_free_inequality(bmap
, bmap
->n_ineq
) < 0)
716 nparam
= isl_basic_map_dim(bmap
, isl_dim_param
);
717 n_in
= isl_basic_map_dim(bmap
, isl_dim_in
);
718 n_out
= isl_basic_map_dim(bmap
, isl_dim_out
);
719 total
= isl_basic_map_dim(bmap
, isl_dim_all
);
720 for (i
= bmap
->n_eq
- 1; i
>= 0; --i
) {
721 j
= isl_seq_last_non_zero(bmap
->eq
[i
] + 1, total
);
722 if (j
>= nparam
&& j
< nparam
+ n_in
+ n_out
&&
723 (isl_int_is_one(bmap
->eq
[i
][1 + j
]) ||
724 isl_int_is_negone(bmap
->eq
[i
][1 + j
])))
726 if (isl_basic_map_drop_equality(bmap
, i
) < 0)
730 bmap
= isl_basic_map_finalize(bmap
);
734 isl_basic_map_free(bmap
);
738 static int aff_split_cmp(const void *p1
, const void *p2
, void *user
)
740 const struct isl_aff_split
*s1
, *s2
;
741 s1
= (const struct isl_aff_split
*) p1
;
742 s2
= (const struct isl_aff_split
*) p2
;
744 return isl_basic_map_plain_cmp(s1
->aff
, s2
->aff
);
747 static __isl_give isl_basic_map
*drop_aff(__isl_take isl_basic_map
*bmap
,
748 __isl_keep isl_basic_map
*aff
)
756 total
= isl_space_dim(bmap
->dim
, isl_dim_all
);
758 for (i
= bmap
->n_eq
- 1; i
>= 0; --i
) {
759 if (isl_seq_first_non_zero(bmap
->eq
[i
] + 1 + total
,
762 for (j
= 0; j
< aff
->n_eq
; ++j
) {
763 if (!isl_seq_eq(bmap
->eq
[i
], aff
->eq
[j
], 1 + total
) &&
764 !isl_seq_is_neg(bmap
->eq
[i
], aff
->eq
[j
], 1 + total
))
766 if (isl_basic_map_drop_equality(bmap
, i
) < 0)
774 isl_basic_map_free(bmap
);
778 static __isl_give
struct isl_aff_split
*split_aff(__isl_keep isl_map
*map
)
781 struct isl_aff_split
*split
;
784 ctx
= isl_map_get_ctx(map
);
785 split
= isl_calloc_array(ctx
, struct isl_aff_split
, map
->n
);
789 for (i
= 0; i
< map
->n
; ++i
) {
791 split
[i
].aff
= get_aff(isl_basic_map_copy(map
->p
[i
]));
792 bmap
= isl_basic_map_copy(map
->p
[i
]);
793 bmap
= isl_basic_map_cow(bmap
);
794 bmap
= drop_aff(bmap
, split
[i
].aff
);
795 split
[i
].map
= isl_map_from_basic_map(bmap
);
796 if (!split
[i
].aff
|| !split
[i
].map
)
800 if (isl_sort(split
, map
->n
, sizeof(struct isl_aff_split
),
801 &aff_split_cmp
, NULL
) < 0)
805 for (i
= n
- 1; i
>= 1; --i
) {
806 if (!isl_basic_map_plain_is_equal(split
[i
- 1].aff
,
809 isl_basic_map_free(split
[i
].aff
);
810 split
[i
- 1].map
= isl_map_union(split
[i
- 1].map
,
813 split
[i
] = split
[n
- 1];
814 split
[n
- 1].aff
= NULL
;
815 split
[n
- 1].map
= NULL
;
821 free_split(split
, map
->n
);
825 static int defining_equality(__isl_keep isl_basic_map
*eq
,
826 __isl_keep isl_space
*dim
, enum isl_dim_type type
, int pos
)
834 pos
+= isl_space_offset(dim
, type
);
835 total
= isl_basic_map_total_dim(eq
);
837 for (i
= 0; i
< eq
->n_eq
; ++i
) {
838 if (isl_seq_last_non_zero(eq
->eq
[i
] + 1, total
) != pos
)
840 if (isl_int_is_one(eq
->eq
[i
][1 + pos
]))
841 isl_seq_neg(eq
->eq
[i
], eq
->eq
[i
], 1 + total
);
848 /* Print dimension "pos" of data->space to "p".
850 * data->user is assumed to be an isl_basic_map keeping track of equalities.
852 * If the current dimension is defined by these equalities, then print
853 * the corresponding expression. Otherwise, print the name of the dimension.
855 static __isl_give isl_printer
*print_dim_eq(__isl_take isl_printer
*p
,
856 struct isl_print_space_data
*data
, unsigned pos
)
858 isl_basic_map
*eq
= data
->user
;
861 j
= defining_equality(eq
, data
->space
, data
->type
, pos
);
863 pos
+= 1 + isl_space_offset(data
->space
, data
->type
);
864 p
= print_affine_of_len(eq
->dim
, NULL
, p
, eq
->eq
[j
], pos
);
866 p
= print_name(data
->space
, p
, data
->type
, pos
, data
->latex
);
872 static __isl_give isl_printer
*print_split_map(__isl_take isl_printer
*p
,
873 struct isl_aff_split
*split
, int n
)
875 struct isl_print_space_data data
= { 0 };
879 data
.print_dim
= &print_dim_eq
;
880 for (i
= 0; i
< n
; ++i
) {
885 space
= split
[i
].map
->dim
;
886 rational
= split
[i
].map
->n
> 0 &&
887 ISL_F_ISSET(split
[i
].map
->p
[0], ISL_BASIC_MAP_RATIONAL
);
889 p
= isl_printer_print_str(p
, "; ");
890 data
.user
= split
[i
].aff
;
891 p
= isl_print_space(space
, p
, rational
, &data
);
892 p
= print_disjuncts_map(split
[i
].map
, p
, 0);
898 static __isl_give isl_printer
*isl_map_print_isl_body(__isl_keep isl_map
*map
,
899 __isl_take isl_printer
*p
)
901 struct isl_print_space_data data
= { 0 };
902 struct isl_aff_split
*split
= NULL
;
906 split
= split_aff(map
);
908 p
= print_split_map(p
, split
, map
->n
);
910 rational
= map
->n
> 0 &&
911 ISL_F_ISSET(map
->p
[0], ISL_BASIC_MAP_RATIONAL
);
912 p
= isl_print_space(map
->dim
, p
, rational
, &data
);
913 p
= print_disjuncts_map(map
, p
, 0);
915 free_split(split
, map
->n
);
919 static __isl_give isl_printer
*isl_map_print_isl(__isl_keep isl_map
*map
,
920 __isl_take isl_printer
*p
)
922 struct isl_print_space_data data
= { 0 };
924 if (isl_map_dim(map
, isl_dim_param
) > 0) {
925 p
= print_tuple(map
->dim
, p
, isl_dim_param
, &data
);
926 p
= isl_printer_print_str(p
, s_to
[0]);
928 p
= isl_printer_print_str(p
, s_open_set
[0]);
929 p
= isl_map_print_isl_body(map
, p
);
930 p
= isl_printer_print_str(p
, s_close_set
[0]);
934 static __isl_give isl_printer
*print_latex_map(__isl_keep isl_map
*map
,
935 __isl_take isl_printer
*p
, __isl_keep isl_basic_map
*aff
)
937 struct isl_print_space_data data
= { 0 };
940 if (isl_map_dim(map
, isl_dim_param
) > 0) {
941 p
= print_tuple(map
->dim
, p
, isl_dim_param
, &data
);
942 p
= isl_printer_print_str(p
, s_to
[1]);
944 p
= isl_printer_print_str(p
, s_open_set
[1]);
945 data
.print_dim
= &print_dim_eq
;
947 p
= isl_print_space(map
->dim
, p
, 0, &data
);
948 p
= print_disjuncts_map(map
, p
, 1);
949 p
= isl_printer_print_str(p
, s_close_set
[1]);
954 static __isl_give isl_printer
*isl_map_print_latex(__isl_keep isl_map
*map
,
955 __isl_take isl_printer
*p
)
958 struct isl_aff_split
*split
= NULL
;
961 split
= split_aff(map
);
964 return print_latex_map(map
, p
, NULL
);
966 for (i
= 0; i
< map
->n
; ++i
) {
970 p
= isl_printer_print_str(p
, " \\cup ");
971 p
= print_latex_map(split
[i
].map
, p
, split
[i
].aff
);
974 free_split(split
, map
->n
);
978 __isl_give isl_printer
*isl_printer_print_basic_map(__isl_take isl_printer
*p
,
979 __isl_keep isl_basic_map
*bmap
)
983 if (p
->output_format
== ISL_FORMAT_ISL
)
984 return isl_basic_map_print_isl(bmap
, p
, 0);
985 else if (p
->output_format
== ISL_FORMAT_OMEGA
)
986 return basic_map_print_omega(bmap
, p
);
987 isl_assert(bmap
->ctx
, 0, goto error
);
993 __isl_give isl_printer
*isl_printer_print_basic_set(__isl_take isl_printer
*p
,
994 __isl_keep isl_basic_set
*bset
)
999 if (p
->output_format
== ISL_FORMAT_ISL
)
1000 return isl_basic_map_print_isl(bset
, p
, 0);
1001 else if (p
->output_format
== ISL_FORMAT_POLYLIB
)
1002 return isl_basic_set_print_polylib(bset
, p
, 0);
1003 else if (p
->output_format
== ISL_FORMAT_EXT_POLYLIB
)
1004 return isl_basic_set_print_polylib(bset
, p
, 1);
1005 else if (p
->output_format
== ISL_FORMAT_POLYLIB_CONSTRAINTS
)
1006 return bset_print_constraints_polylib(bset
, p
);
1007 else if (p
->output_format
== ISL_FORMAT_OMEGA
)
1008 return basic_set_print_omega(bset
, p
);
1009 isl_assert(p
->ctx
, 0, goto error
);
1011 isl_printer_free(p
);
1015 __isl_give isl_printer
*isl_printer_print_set(__isl_take isl_printer
*p
,
1016 __isl_keep isl_set
*set
)
1020 if (p
->output_format
== ISL_FORMAT_ISL
)
1021 return isl_map_print_isl((isl_map
*)set
, p
);
1022 else if (p
->output_format
== ISL_FORMAT_POLYLIB
)
1023 return isl_set_print_polylib(set
, p
, 0);
1024 else if (p
->output_format
== ISL_FORMAT_EXT_POLYLIB
)
1025 return isl_set_print_polylib(set
, p
, 1);
1026 else if (p
->output_format
== ISL_FORMAT_OMEGA
)
1027 return isl_set_print_omega(set
, p
);
1028 else if (p
->output_format
== ISL_FORMAT_LATEX
)
1029 return isl_map_print_latex((isl_map
*)set
, p
);
1030 isl_assert(set
->ctx
, 0, goto error
);
1032 isl_printer_free(p
);
1036 __isl_give isl_printer
*isl_printer_print_map(__isl_take isl_printer
*p
,
1037 __isl_keep isl_map
*map
)
1042 if (p
->output_format
== ISL_FORMAT_ISL
)
1043 return isl_map_print_isl(map
, p
);
1044 else if (p
->output_format
== ISL_FORMAT_POLYLIB
)
1045 return isl_map_print_polylib(map
, p
, 0);
1046 else if (p
->output_format
== ISL_FORMAT_EXT_POLYLIB
)
1047 return isl_map_print_polylib(map
, p
, 1);
1048 else if (p
->output_format
== ISL_FORMAT_OMEGA
)
1049 return isl_map_print_omega(map
, p
);
1050 else if (p
->output_format
== ISL_FORMAT_LATEX
)
1051 return isl_map_print_latex(map
, p
);
1052 isl_assert(map
->ctx
, 0, goto error
);
1054 isl_printer_free(p
);
1058 struct isl_union_print_data
{
1063 static isl_stat
print_map_body(__isl_take isl_map
*map
, void *user
)
1065 struct isl_union_print_data
*data
;
1066 data
= (struct isl_union_print_data
*)user
;
1069 data
->p
= isl_printer_print_str(data
->p
, "; ");
1072 data
->p
= isl_map_print_isl_body(map
, data
->p
);
1078 static __isl_give isl_printer
*isl_union_map_print_isl(
1079 __isl_keep isl_union_map
*umap
, __isl_take isl_printer
*p
)
1081 struct isl_union_print_data data
;
1082 struct isl_print_space_data space_data
= { 0 };
1085 dim
= isl_union_map_get_space(umap
);
1086 if (isl_space_dim(dim
, isl_dim_param
) > 0) {
1087 p
= print_tuple(dim
, p
, isl_dim_param
, &space_data
);
1088 p
= isl_printer_print_str(p
, s_to
[0]);
1090 isl_space_free(dim
);
1091 p
= isl_printer_print_str(p
, s_open_set
[0]);
1094 isl_union_map_foreach_map(umap
, &print_map_body
, &data
);
1096 p
= isl_printer_print_str(p
, s_close_set
[0]);
1100 static isl_stat
print_latex_map_body(__isl_take isl_map
*map
, void *user
)
1102 struct isl_union_print_data
*data
;
1103 data
= (struct isl_union_print_data
*)user
;
1106 data
->p
= isl_printer_print_str(data
->p
, " \\cup ");
1109 data
->p
= isl_map_print_latex(map
, data
->p
);
1115 static __isl_give isl_printer
*isl_union_map_print_latex(
1116 __isl_keep isl_union_map
*umap
, __isl_take isl_printer
*p
)
1118 struct isl_union_print_data data
= { p
, 1 };
1119 isl_union_map_foreach_map(umap
, &print_latex_map_body
, &data
);
1124 __isl_give isl_printer
*isl_printer_print_union_map(__isl_take isl_printer
*p
,
1125 __isl_keep isl_union_map
*umap
)
1130 if (p
->output_format
== ISL_FORMAT_ISL
)
1131 return isl_union_map_print_isl(umap
, p
);
1132 if (p
->output_format
== ISL_FORMAT_LATEX
)
1133 return isl_union_map_print_latex(umap
, p
);
1135 isl_die(p
->ctx
, isl_error_invalid
,
1136 "invalid output format for isl_union_map", goto error
);
1138 isl_printer_free(p
);
1142 __isl_give isl_printer
*isl_printer_print_union_set(__isl_take isl_printer
*p
,
1143 __isl_keep isl_union_set
*uset
)
1148 if (p
->output_format
== ISL_FORMAT_ISL
)
1149 return isl_union_map_print_isl((isl_union_map
*)uset
, p
);
1150 if (p
->output_format
== ISL_FORMAT_LATEX
)
1151 return isl_union_map_print_latex((isl_union_map
*)uset
, p
);
1153 isl_die(p
->ctx
, isl_error_invalid
,
1154 "invalid output format for isl_union_set", goto error
);
1156 isl_printer_free(p
);
1160 static int upoly_rec_n_non_zero(__isl_keep
struct isl_upoly_rec
*rec
)
1165 for (i
= 0, n
= 0; i
< rec
->n
; ++i
)
1166 if (!isl_upoly_is_zero(rec
->p
[i
]))
1172 static __isl_give isl_printer
*upoly_print_cst(__isl_keep
struct isl_upoly
*up
,
1173 __isl_take isl_printer
*p
, int first
)
1175 struct isl_upoly_cst
*cst
;
1178 cst
= isl_upoly_as_cst(up
);
1181 neg
= !first
&& isl_int_is_neg(cst
->n
);
1183 p
= isl_printer_print_str(p
, neg
? " - " : " + ");
1185 isl_int_neg(cst
->n
, cst
->n
);
1186 if (isl_int_is_zero(cst
->d
)) {
1187 int sgn
= isl_int_sgn(cst
->n
);
1188 p
= isl_printer_print_str(p
, sgn
< 0 ? "-infty" :
1189 sgn
== 0 ? "NaN" : "infty");
1191 p
= isl_printer_print_isl_int(p
, cst
->n
);
1193 isl_int_neg(cst
->n
, cst
->n
);
1194 if (!isl_int_is_zero(cst
->d
) && !isl_int_is_one(cst
->d
)) {
1195 p
= isl_printer_print_str(p
, "/");
1196 p
= isl_printer_print_isl_int(p
, cst
->d
);
1200 isl_printer_free(p
);
1204 static __isl_give isl_printer
*print_base(__isl_take isl_printer
*p
,
1205 __isl_keep isl_space
*dim
, __isl_keep isl_mat
*div
, int var
)
1209 total
= isl_space_dim(dim
, isl_dim_all
);
1211 p
= print_term(dim
, NULL
, dim
->ctx
->one
, 1 + var
, p
, 0);
1213 p
= print_div(dim
, div
, var
- total
, p
);
1217 static __isl_give isl_printer
*print_pow(__isl_take isl_printer
*p
,
1218 __isl_keep isl_space
*dim
, __isl_keep isl_mat
*div
, int var
, int exp
)
1220 p
= print_base(p
, dim
, div
, var
);
1223 if (p
->output_format
== ISL_FORMAT_C
) {
1225 for (i
= 1; i
< exp
; ++i
) {
1226 p
= isl_printer_print_str(p
, "*");
1227 p
= print_base(p
, dim
, div
, var
);
1230 p
= isl_printer_print_str(p
, "^");
1231 p
= isl_printer_print_int(p
, exp
);
1236 static __isl_give isl_printer
*upoly_print(__isl_keep
struct isl_upoly
*up
,
1237 __isl_keep isl_space
*dim
, __isl_keep isl_mat
*div
,
1238 __isl_take isl_printer
*p
, int outer
)
1240 int i
, n
, first
, print_parens
;
1241 struct isl_upoly_rec
*rec
;
1243 if (!p
|| !up
|| !dim
|| !div
)
1246 if (isl_upoly_is_cst(up
))
1247 return upoly_print_cst(up
, p
, 1);
1249 rec
= isl_upoly_as_rec(up
);
1252 n
= upoly_rec_n_non_zero(rec
);
1253 print_parens
= n
> 1 ||
1254 (outer
&& rec
->up
.var
>= isl_space_dim(dim
, isl_dim_all
));
1256 p
= isl_printer_print_str(p
, "(");
1257 for (i
= 0, first
= 1; i
< rec
->n
; ++i
) {
1258 if (isl_upoly_is_zero(rec
->p
[i
]))
1260 if (isl_upoly_is_negone(rec
->p
[i
])) {
1262 p
= isl_printer_print_str(p
, "-1");
1264 p
= isl_printer_print_str(p
, "-");
1266 p
= isl_printer_print_str(p
, " - ");
1267 } else if (isl_upoly_is_cst(rec
->p
[i
]) &&
1268 !isl_upoly_is_one(rec
->p
[i
]))
1269 p
= upoly_print_cst(rec
->p
[i
], p
, first
);
1272 p
= isl_printer_print_str(p
, " + ");
1273 if (i
== 0 || !isl_upoly_is_one(rec
->p
[i
]))
1274 p
= upoly_print(rec
->p
[i
], dim
, div
, p
, 0);
1279 if (!isl_upoly_is_one(rec
->p
[i
]) &&
1280 !isl_upoly_is_negone(rec
->p
[i
]))
1281 p
= isl_printer_print_str(p
, " * ");
1282 p
= print_pow(p
, dim
, div
, rec
->up
.var
, i
);
1285 p
= isl_printer_print_str(p
, ")");
1288 isl_printer_free(p
);
1292 static __isl_give isl_printer
*print_qpolynomial(__isl_take isl_printer
*p
,
1293 __isl_keep isl_qpolynomial
*qp
)
1297 p
= upoly_print(qp
->upoly
, qp
->dim
, qp
->div
, p
, 1);
1300 isl_printer_free(p
);
1304 static __isl_give isl_printer
*print_qpolynomial_isl(__isl_take isl_printer
*p
,
1305 __isl_keep isl_qpolynomial
*qp
)
1307 struct isl_print_space_data data
= { 0 };
1312 if (isl_space_dim(qp
->dim
, isl_dim_param
) > 0) {
1313 p
= print_tuple(qp
->dim
, p
, isl_dim_param
, &data
);
1314 p
= isl_printer_print_str(p
, " -> ");
1316 p
= isl_printer_print_str(p
, "{ ");
1317 if (!isl_space_is_params(qp
->dim
)) {
1318 p
= isl_print_space(qp
->dim
, p
, 0, &data
);
1319 p
= isl_printer_print_str(p
, " -> ");
1321 p
= print_qpolynomial(p
, qp
);
1322 p
= isl_printer_print_str(p
, " }");
1325 isl_printer_free(p
);
1329 static __isl_give isl_printer
*print_qpolynomial_c(__isl_take isl_printer
*p
,
1330 __isl_keep isl_space
*dim
, __isl_keep isl_qpolynomial
*qp
)
1335 isl_qpolynomial_get_den(qp
, &den
);
1336 if (!isl_int_is_one(den
)) {
1338 p
= isl_printer_print_str(p
, "(");
1339 qp
= isl_qpolynomial_copy(qp
);
1340 f
= isl_qpolynomial_rat_cst_on_domain(isl_space_copy(qp
->dim
),
1341 den
, qp
->dim
->ctx
->one
);
1342 qp
= isl_qpolynomial_mul(qp
, f
);
1345 p
= upoly_print(qp
->upoly
, dim
, qp
->div
, p
, 0);
1347 p
= isl_printer_free(p
);
1348 if (!isl_int_is_one(den
)) {
1349 p
= isl_printer_print_str(p
, ")/");
1350 p
= isl_printer_print_isl_int(p
, den
);
1351 isl_qpolynomial_free(qp
);
1357 __isl_give isl_printer
*isl_printer_print_qpolynomial(
1358 __isl_take isl_printer
*p
, __isl_keep isl_qpolynomial
*qp
)
1363 if (p
->output_format
== ISL_FORMAT_ISL
)
1364 return print_qpolynomial_isl(p
, qp
);
1365 else if (p
->output_format
== ISL_FORMAT_C
)
1366 return print_qpolynomial_c(p
, qp
->dim
, qp
);
1368 isl_die(qp
->dim
->ctx
, isl_error_unsupported
,
1369 "output format not supported for isl_qpolynomials",
1372 isl_printer_free(p
);
1376 void isl_qpolynomial_print(__isl_keep isl_qpolynomial
*qp
, FILE *out
,
1377 unsigned output_format
)
1384 isl_assert(qp
->dim
->ctx
, output_format
== ISL_FORMAT_ISL
, return);
1385 p
= isl_printer_to_file(qp
->dim
->ctx
, out
);
1386 p
= isl_printer_print_qpolynomial(p
, qp
);
1387 isl_printer_free(p
);
1390 static __isl_give isl_printer
*qpolynomial_fold_print(
1391 __isl_keep isl_qpolynomial_fold
*fold
, __isl_take isl_printer
*p
)
1395 if (fold
->type
== isl_fold_min
)
1396 p
= isl_printer_print_str(p
, "min");
1397 else if (fold
->type
== isl_fold_max
)
1398 p
= isl_printer_print_str(p
, "max");
1399 p
= isl_printer_print_str(p
, "(");
1400 for (i
= 0; i
< fold
->n
; ++i
) {
1402 p
= isl_printer_print_str(p
, ", ");
1403 p
= print_qpolynomial(p
, fold
->qp
[i
]);
1405 p
= isl_printer_print_str(p
, ")");
1409 void isl_qpolynomial_fold_print(__isl_keep isl_qpolynomial_fold
*fold
,
1410 FILE *out
, unsigned output_format
)
1417 isl_assert(fold
->dim
->ctx
, output_format
== ISL_FORMAT_ISL
, return);
1419 p
= isl_printer_to_file(fold
->dim
->ctx
, out
);
1420 p
= isl_printer_print_qpolynomial_fold(p
, fold
);
1422 isl_printer_free(p
);
1425 static __isl_give isl_printer
*isl_pwqp_print_isl_body(
1426 __isl_take isl_printer
*p
, __isl_keep isl_pw_qpolynomial
*pwqp
)
1428 struct isl_print_space_data data
= { 0 };
1431 for (i
= 0; i
< pwqp
->n
; ++i
) {
1433 p
= isl_printer_print_str(p
, "; ");
1434 if (!isl_space_is_params(pwqp
->p
[i
].set
->dim
)) {
1435 p
= isl_print_space(pwqp
->p
[i
].set
->dim
, p
, 0, &data
);
1436 p
= isl_printer_print_str(p
, " -> ");
1438 p
= print_qpolynomial(p
, pwqp
->p
[i
].qp
);
1439 p
= print_disjuncts((isl_map
*)pwqp
->p
[i
].set
, p
, 0);
1445 static __isl_give isl_printer
*print_pw_qpolynomial_isl(
1446 __isl_take isl_printer
*p
, __isl_keep isl_pw_qpolynomial
*pwqp
)
1448 struct isl_print_space_data data
= { 0 };
1453 if (isl_space_dim(pwqp
->dim
, isl_dim_param
) > 0) {
1454 p
= print_tuple(pwqp
->dim
, p
, isl_dim_param
, &data
);
1455 p
= isl_printer_print_str(p
, " -> ");
1457 p
= isl_printer_print_str(p
, "{ ");
1459 if (!isl_space_is_set(pwqp
->dim
)) {
1460 p
= print_tuple(pwqp
->dim
, p
, isl_dim_in
, &data
);
1461 p
= isl_printer_print_str(p
, " -> ");
1463 p
= isl_printer_print_str(p
, "0");
1465 p
= isl_pwqp_print_isl_body(p
, pwqp
);
1466 p
= isl_printer_print_str(p
, " }");
1469 isl_printer_free(p
);
1473 void isl_pw_qpolynomial_print(__isl_keep isl_pw_qpolynomial
*pwqp
, FILE *out
,
1474 unsigned output_format
)
1481 p
= isl_printer_to_file(pwqp
->dim
->ctx
, out
);
1482 p
= isl_printer_set_output_format(p
, output_format
);
1483 p
= isl_printer_print_pw_qpolynomial(p
, pwqp
);
1485 isl_printer_free(p
);
1488 static __isl_give isl_printer
*isl_pwf_print_isl_body(
1489 __isl_take isl_printer
*p
, __isl_keep isl_pw_qpolynomial_fold
*pwf
)
1491 struct isl_print_space_data data
= { 0 };
1494 for (i
= 0; i
< pwf
->n
; ++i
) {
1496 p
= isl_printer_print_str(p
, "; ");
1497 if (!isl_space_is_params(pwf
->p
[i
].set
->dim
)) {
1498 p
= isl_print_space(pwf
->p
[i
].set
->dim
, p
, 0, &data
);
1499 p
= isl_printer_print_str(p
, " -> ");
1501 p
= qpolynomial_fold_print(pwf
->p
[i
].fold
, p
);
1502 p
= print_disjuncts((isl_map
*)pwf
->p
[i
].set
, p
, 0);
1508 static __isl_give isl_printer
*print_pw_qpolynomial_fold_isl(
1509 __isl_take isl_printer
*p
, __isl_keep isl_pw_qpolynomial_fold
*pwf
)
1511 struct isl_print_space_data data
= { 0 };
1513 if (isl_space_dim(pwf
->dim
, isl_dim_param
) > 0) {
1514 p
= print_tuple(pwf
->dim
, p
, isl_dim_param
, &data
);
1515 p
= isl_printer_print_str(p
, " -> ");
1517 p
= isl_printer_print_str(p
, "{ ");
1519 if (!isl_space_is_set(pwf
->dim
)) {
1520 p
= print_tuple(pwf
->dim
, p
, isl_dim_in
, &data
);
1521 p
= isl_printer_print_str(p
, " -> ");
1523 p
= isl_printer_print_str(p
, "0");
1525 p
= isl_pwf_print_isl_body(p
, pwf
);
1526 p
= isl_printer_print_str(p
, " }");
1530 static __isl_give isl_printer
*print_affine_c(__isl_take isl_printer
*p
,
1531 __isl_keep isl_space
*dim
, __isl_keep isl_basic_set
*bset
, isl_int
*c
);
1533 static __isl_give isl_printer
*print_name_c(__isl_take isl_printer
*p
,
1534 __isl_keep isl_space
*dim
,
1535 __isl_keep isl_basic_set
*bset
, enum isl_dim_type type
, unsigned pos
)
1537 if (type
== isl_dim_div
) {
1538 p
= isl_printer_print_str(p
, "floord(");
1539 p
= print_affine_c(p
, dim
, bset
, bset
->div
[pos
] + 1);
1540 p
= isl_printer_print_str(p
, ", ");
1541 p
= isl_printer_print_isl_int(p
, bset
->div
[pos
][0]);
1542 p
= isl_printer_print_str(p
, ")");
1546 name
= isl_space_get_dim_name(dim
, type
, pos
);
1549 p
= isl_printer_print_str(p
, name
);
1554 static __isl_give isl_printer
*print_term_c(__isl_take isl_printer
*p
,
1555 __isl_keep isl_space
*dim
,
1556 __isl_keep isl_basic_set
*bset
, isl_int c
, unsigned pos
)
1558 enum isl_dim_type type
;
1561 return isl_printer_print_isl_int(p
, c
);
1563 if (isl_int_is_one(c
))
1565 else if (isl_int_is_negone(c
))
1566 p
= isl_printer_print_str(p
, "-");
1568 p
= isl_printer_print_isl_int(p
, c
);
1569 p
= isl_printer_print_str(p
, "*");
1571 type
= pos2type(dim
, &pos
);
1572 p
= print_name_c(p
, dim
, bset
, type
, pos
);
1576 static __isl_give isl_printer
*print_partial_affine_c(__isl_take isl_printer
*p
,
1577 __isl_keep isl_space
*dim
,
1578 __isl_keep isl_basic_set
*bset
, isl_int
*c
, unsigned len
)
1583 for (i
= 0, first
= 1; i
< len
; ++i
) {
1585 if (isl_int_is_zero(c
[i
]))
1588 if (isl_int_is_neg(c
[i
])) {
1590 isl_int_neg(c
[i
], c
[i
]);
1591 p
= isl_printer_print_str(p
, " - ");
1593 p
= isl_printer_print_str(p
, " + ");
1596 p
= print_term_c(p
, dim
, bset
, c
[i
], i
);
1598 isl_int_neg(c
[i
], c
[i
]);
1601 p
= isl_printer_print_str(p
, "0");
1605 static __isl_give isl_printer
*print_affine_c(__isl_take isl_printer
*p
,
1606 __isl_keep isl_space
*dim
, __isl_keep isl_basic_set
*bset
, isl_int
*c
)
1608 unsigned len
= 1 + isl_basic_set_total_dim(bset
);
1609 return print_partial_affine_c(p
, dim
, bset
, c
, len
);
1612 /* We skip the constraint if it is implied by the div expression.
1614 * *first indicates whether this is the first constraint in the conjunction and
1615 * is updated if the constraint is actually printed.
1617 static __isl_give isl_printer
*print_constraint_c(__isl_take isl_printer
*p
,
1618 __isl_keep isl_space
*dim
,
1619 __isl_keep isl_basic_set
*bset
, isl_int
*c
, const char *op
, int *first
)
1625 o_div
= isl_basic_set_offset(bset
, isl_dim_div
);
1626 n_div
= isl_basic_set_dim(bset
, isl_dim_div
);
1627 div
= isl_seq_last_non_zero(c
+ o_div
, n_div
);
1628 if (div
>= 0 && isl_basic_set_is_div_constraint(bset
, c
, div
))
1632 p
= isl_printer_print_str(p
, " && ");
1634 p
= print_affine_c(p
, dim
, bset
, c
);
1635 p
= isl_printer_print_str(p
, " ");
1636 p
= isl_printer_print_str(p
, op
);
1637 p
= isl_printer_print_str(p
, " 0");
1644 static __isl_give isl_printer
*print_basic_set_c(__isl_take isl_printer
*p
,
1645 __isl_keep isl_space
*dim
, __isl_keep isl_basic_set
*bset
)
1649 unsigned n_div
= isl_basic_set_dim(bset
, isl_dim_div
);
1650 unsigned total
= isl_basic_set_total_dim(bset
) - n_div
;
1652 for (i
= 0; i
< bset
->n_eq
; ++i
) {
1653 j
= isl_seq_last_non_zero(bset
->eq
[i
] + 1 + total
, n_div
);
1655 p
= print_constraint_c(p
, dim
, bset
,
1656 bset
->eq
[i
], "==", &first
);
1659 p
= isl_printer_print_str(p
, " && ");
1660 p
= isl_printer_print_str(p
, "(");
1661 p
= print_partial_affine_c(p
, dim
, bset
, bset
->eq
[i
],
1663 p
= isl_printer_print_str(p
, ") % ");
1664 p
= isl_printer_print_isl_int(p
,
1665 bset
->eq
[i
][1 + total
+ j
]);
1666 p
= isl_printer_print_str(p
, " == 0");
1670 for (i
= 0; i
< bset
->n_ineq
; ++i
)
1671 p
= print_constraint_c(p
, dim
, bset
, bset
->ineq
[i
], ">=",
1676 static __isl_give isl_printer
*print_set_c(__isl_take isl_printer
*p
,
1677 __isl_keep isl_space
*dim
, __isl_keep isl_set
*set
)
1682 return isl_printer_free(p
);
1685 p
= isl_printer_print_str(p
, "0");
1687 for (i
= 0; i
< set
->n
; ++i
) {
1689 p
= isl_printer_print_str(p
, " || ");
1691 p
= isl_printer_print_str(p
, "(");
1692 p
= print_basic_set_c(p
, dim
, set
->p
[i
]);
1694 p
= isl_printer_print_str(p
, ")");
1699 static __isl_give isl_printer
*print_pw_qpolynomial_c(
1700 __isl_take isl_printer
*p
, __isl_keep isl_pw_qpolynomial
*pwqp
)
1704 if (pwqp
->n
== 1 && isl_set_plain_is_universe(pwqp
->p
[0].set
))
1705 return print_qpolynomial_c(p
, pwqp
->dim
, pwqp
->p
[0].qp
);
1707 for (i
= 0; i
< pwqp
->n
; ++i
) {
1708 p
= isl_printer_print_str(p
, "(");
1709 p
= print_set_c(p
, pwqp
->dim
, pwqp
->p
[i
].set
);
1710 p
= isl_printer_print_str(p
, ") ? (");
1711 p
= print_qpolynomial_c(p
, pwqp
->dim
, pwqp
->p
[i
].qp
);
1712 p
= isl_printer_print_str(p
, ") : ");
1715 p
= isl_printer_print_str(p
, "0");
1719 __isl_give isl_printer
*isl_printer_print_pw_qpolynomial(
1720 __isl_take isl_printer
*p
, __isl_keep isl_pw_qpolynomial
*pwqp
)
1725 if (p
->output_format
== ISL_FORMAT_ISL
)
1726 return print_pw_qpolynomial_isl(p
, pwqp
);
1727 else if (p
->output_format
== ISL_FORMAT_C
)
1728 return print_pw_qpolynomial_c(p
, pwqp
);
1729 isl_assert(p
->ctx
, 0, goto error
);
1731 isl_printer_free(p
);
1735 static isl_stat
print_pwqp_body(__isl_take isl_pw_qpolynomial
*pwqp
, void *user
)
1737 struct isl_union_print_data
*data
;
1738 data
= (struct isl_union_print_data
*)user
;
1741 data
->p
= isl_printer_print_str(data
->p
, "; ");
1744 data
->p
= isl_pwqp_print_isl_body(data
->p
, pwqp
);
1745 isl_pw_qpolynomial_free(pwqp
);
1750 static __isl_give isl_printer
*print_union_pw_qpolynomial_isl(
1751 __isl_take isl_printer
*p
, __isl_keep isl_union_pw_qpolynomial
*upwqp
)
1753 struct isl_union_print_data data
;
1754 struct isl_print_space_data space_data
= { 0 };
1757 dim
= isl_union_pw_qpolynomial_get_space(upwqp
);
1758 if (isl_space_dim(dim
, isl_dim_param
) > 0) {
1759 p
= print_tuple(dim
, p
, isl_dim_param
, &space_data
);
1760 p
= isl_printer_print_str(p
, " -> ");
1762 isl_space_free(dim
);
1763 p
= isl_printer_print_str(p
, "{ ");
1766 isl_union_pw_qpolynomial_foreach_pw_qpolynomial(upwqp
, &print_pwqp_body
,
1769 p
= isl_printer_print_str(p
, " }");
1773 __isl_give isl_printer
*isl_printer_print_union_pw_qpolynomial(
1774 __isl_take isl_printer
*p
, __isl_keep isl_union_pw_qpolynomial
*upwqp
)
1779 if (p
->output_format
== ISL_FORMAT_ISL
)
1780 return print_union_pw_qpolynomial_isl(p
, upwqp
);
1781 isl_die(p
->ctx
, isl_error_invalid
,
1782 "invalid output format for isl_union_pw_qpolynomial",
1785 isl_printer_free(p
);
1789 static __isl_give isl_printer
*print_qpolynomial_fold_c(
1790 __isl_take isl_printer
*p
, __isl_keep isl_space
*dim
,
1791 __isl_keep isl_qpolynomial_fold
*fold
)
1795 for (i
= 0; i
< fold
->n
- 1; ++i
)
1796 if (fold
->type
== isl_fold_min
)
1797 p
= isl_printer_print_str(p
, "min(");
1798 else if (fold
->type
== isl_fold_max
)
1799 p
= isl_printer_print_str(p
, "max(");
1801 for (i
= 0; i
< fold
->n
; ++i
) {
1803 p
= isl_printer_print_str(p
, ", ");
1804 p
= print_qpolynomial_c(p
, dim
, fold
->qp
[i
]);
1806 p
= isl_printer_print_str(p
, ")");
1811 __isl_give isl_printer
*isl_printer_print_qpolynomial_fold(
1812 __isl_take isl_printer
*p
, __isl_keep isl_qpolynomial_fold
*fold
)
1816 if (p
->output_format
== ISL_FORMAT_ISL
)
1817 return qpolynomial_fold_print(fold
, p
);
1818 else if (p
->output_format
== ISL_FORMAT_C
)
1819 return print_qpolynomial_fold_c(p
, fold
->dim
, fold
);
1820 isl_die(p
->ctx
, isl_error_unsupported
, "unsupported output format",
1823 isl_printer_free(p
);
1827 static __isl_give isl_printer
*print_pw_qpolynomial_fold_c(
1828 __isl_take isl_printer
*p
, __isl_keep isl_pw_qpolynomial_fold
*pwf
)
1832 if (pwf
->n
== 1 && isl_set_plain_is_universe(pwf
->p
[0].set
))
1833 return print_qpolynomial_fold_c(p
, pwf
->dim
, pwf
->p
[0].fold
);
1835 for (i
= 0; i
< pwf
->n
; ++i
) {
1836 p
= isl_printer_print_str(p
, "(");
1837 p
= print_set_c(p
, pwf
->dim
, pwf
->p
[i
].set
);
1838 p
= isl_printer_print_str(p
, ") ? (");
1839 p
= print_qpolynomial_fold_c(p
, pwf
->dim
, pwf
->p
[i
].fold
);
1840 p
= isl_printer_print_str(p
, ") : ");
1843 p
= isl_printer_print_str(p
, "0");
1847 __isl_give isl_printer
*isl_printer_print_pw_qpolynomial_fold(
1848 __isl_take isl_printer
*p
, __isl_keep isl_pw_qpolynomial_fold
*pwf
)
1853 if (p
->output_format
== ISL_FORMAT_ISL
)
1854 return print_pw_qpolynomial_fold_isl(p
, pwf
);
1855 else if (p
->output_format
== ISL_FORMAT_C
)
1856 return print_pw_qpolynomial_fold_c(p
, pwf
);
1857 isl_assert(p
->ctx
, 0, goto error
);
1859 isl_printer_free(p
);
1863 void isl_pw_qpolynomial_fold_print(__isl_keep isl_pw_qpolynomial_fold
*pwf
,
1864 FILE *out
, unsigned output_format
)
1871 p
= isl_printer_to_file(pwf
->dim
->ctx
, out
);
1872 p
= isl_printer_set_output_format(p
, output_format
);
1873 p
= isl_printer_print_pw_qpolynomial_fold(p
, pwf
);
1875 isl_printer_free(p
);
1878 static isl_stat
print_pwf_body(__isl_take isl_pw_qpolynomial_fold
*pwf
,
1881 struct isl_union_print_data
*data
;
1882 data
= (struct isl_union_print_data
*)user
;
1885 data
->p
= isl_printer_print_str(data
->p
, "; ");
1888 data
->p
= isl_pwf_print_isl_body(data
->p
, pwf
);
1889 isl_pw_qpolynomial_fold_free(pwf
);
1894 static __isl_give isl_printer
*print_union_pw_qpolynomial_fold_isl(
1895 __isl_take isl_printer
*p
,
1896 __isl_keep isl_union_pw_qpolynomial_fold
*upwf
)
1898 struct isl_union_print_data data
;
1899 struct isl_print_space_data space_data
= { 0 };
1902 dim
= isl_union_pw_qpolynomial_fold_get_space(upwf
);
1903 if (isl_space_dim(dim
, isl_dim_param
) > 0) {
1904 p
= print_tuple(dim
, p
, isl_dim_param
, &space_data
);
1905 p
= isl_printer_print_str(p
, " -> ");
1907 isl_space_free(dim
);
1908 p
= isl_printer_print_str(p
, "{ ");
1911 isl_union_pw_qpolynomial_fold_foreach_pw_qpolynomial_fold(upwf
,
1912 &print_pwf_body
, &data
);
1914 p
= isl_printer_print_str(p
, " }");
1918 __isl_give isl_printer
*isl_printer_print_union_pw_qpolynomial_fold(
1919 __isl_take isl_printer
*p
,
1920 __isl_keep isl_union_pw_qpolynomial_fold
*upwf
)
1925 if (p
->output_format
== ISL_FORMAT_ISL
)
1926 return print_union_pw_qpolynomial_fold_isl(p
, upwf
);
1927 isl_die(p
->ctx
, isl_error_invalid
,
1928 "invalid output format for isl_union_pw_qpolynomial_fold",
1931 isl_printer_free(p
);
1935 __isl_give isl_printer
*isl_printer_print_constraint(__isl_take isl_printer
*p
,
1936 __isl_keep isl_constraint
*c
)
1938 isl_basic_map
*bmap
;
1943 bmap
= isl_basic_map_from_constraint(isl_constraint_copy(c
));
1944 p
= isl_printer_print_basic_map(p
, bmap
);
1945 isl_basic_map_free(bmap
);
1948 isl_printer_free(p
);
1952 static __isl_give isl_printer
*isl_printer_print_space_isl(
1953 __isl_take isl_printer
*p
, __isl_keep isl_space
*space
)
1955 struct isl_print_space_data data
= { 0 };
1960 if (isl_space_dim(space
, isl_dim_param
) > 0) {
1961 p
= print_tuple(space
, p
, isl_dim_param
, &data
);
1962 p
= isl_printer_print_str(p
, " -> ");
1965 p
= isl_printer_print_str(p
, "{ ");
1966 if (isl_space_is_params(space
))
1967 p
= isl_printer_print_str(p
, s_such_that
[0]);
1969 p
= isl_print_space(space
, p
, 0, &data
);
1970 p
= isl_printer_print_str(p
, " }");
1974 isl_printer_free(p
);
1978 __isl_give isl_printer
*isl_printer_print_space(__isl_take isl_printer
*p
,
1979 __isl_keep isl_space
*space
)
1982 return isl_printer_free(p
);
1983 if (p
->output_format
== ISL_FORMAT_ISL
)
1984 return isl_printer_print_space_isl(p
, space
);
1985 else if (p
->output_format
== ISL_FORMAT_OMEGA
)
1986 return print_omega_parameters(space
, p
);
1988 isl_die(isl_space_get_ctx(space
), isl_error_unsupported
,
1989 "output format not supported for space",
1990 return isl_printer_free(p
));
1993 __isl_give isl_printer
*isl_printer_print_local_space(__isl_take isl_printer
*p
,
1994 __isl_keep isl_local_space
*ls
)
1996 struct isl_print_space_data data
= { 0 };
2002 if (isl_local_space_dim(ls
, isl_dim_param
) > 0) {
2003 p
= print_tuple(ls
->dim
, p
, isl_dim_param
, &data
);
2004 p
= isl_printer_print_str(p
, " -> ");
2006 p
= isl_printer_print_str(p
, "{ ");
2007 p
= isl_print_space(ls
->dim
, p
, 0, &data
);
2008 n_div
= isl_local_space_dim(ls
, isl_dim_div
);
2010 p
= isl_printer_print_str(p
, " : ");
2011 p
= isl_printer_print_str(p
, s_open_exists
[0]);
2012 p
= print_div_list(p
, ls
->dim
, ls
->div
, 0);
2013 p
= isl_printer_print_str(p
, s_close_exists
[0]);
2014 } else if (isl_space_is_params(ls
->dim
))
2015 p
= isl_printer_print_str(p
, s_such_that
[0]);
2016 p
= isl_printer_print_str(p
, " }");
2019 isl_printer_free(p
);
2023 static __isl_give isl_printer
*print_aff_body(__isl_take isl_printer
*p
,
2024 __isl_keep isl_aff
*aff
)
2028 if (isl_aff_is_nan(aff
))
2029 return isl_printer_print_str(p
, "NaN");
2031 total
= isl_local_space_dim(aff
->ls
, isl_dim_all
);
2032 p
= isl_printer_print_str(p
, "(");
2033 p
= print_affine_of_len(aff
->ls
->dim
, aff
->ls
->div
, p
,
2034 aff
->v
->el
+ 1, 1 + total
);
2035 if (isl_int_is_one(aff
->v
->el
[0]))
2036 p
= isl_printer_print_str(p
, ")");
2038 p
= isl_printer_print_str(p
, ")/");
2039 p
= isl_printer_print_isl_int(p
, aff
->v
->el
[0]);
2045 static __isl_give isl_printer
*print_aff(__isl_take isl_printer
*p
,
2046 __isl_keep isl_aff
*aff
)
2048 struct isl_print_space_data data
= { 0 };
2050 if (isl_space_is_params(aff
->ls
->dim
))
2053 p
= print_tuple(aff
->ls
->dim
, p
, isl_dim_set
, &data
);
2054 p
= isl_printer_print_str(p
, " -> ");
2056 p
= isl_printer_print_str(p
, "[");
2057 p
= print_aff_body(p
, aff
);
2058 p
= isl_printer_print_str(p
, "]");
2063 static __isl_give isl_printer
*print_aff_isl(__isl_take isl_printer
*p
,
2064 __isl_keep isl_aff
*aff
)
2066 struct isl_print_space_data data
= { 0 };
2071 if (isl_local_space_dim(aff
->ls
, isl_dim_param
) > 0) {
2072 p
= print_tuple(aff
->ls
->dim
, p
, isl_dim_param
, &data
);
2073 p
= isl_printer_print_str(p
, " -> ");
2075 p
= isl_printer_print_str(p
, "{ ");
2076 p
= print_aff(p
, aff
);
2077 p
= isl_printer_print_str(p
, " }");
2080 isl_printer_free(p
);
2084 /* Print the body of an isl_pw_aff, i.e., a semicolon delimited
2085 * sequence of affine expressions, each followed by constraints.
2087 static __isl_give isl_printer
*print_pw_aff_body(
2088 __isl_take isl_printer
*p
, __isl_keep isl_pw_aff
*pa
)
2093 return isl_printer_free(p
);
2095 for (i
= 0; i
< pa
->n
; ++i
) {
2097 p
= isl_printer_print_str(p
, "; ");
2098 p
= print_aff(p
, pa
->p
[i
].aff
);
2099 p
= print_disjuncts((isl_map
*)pa
->p
[i
].set
, p
, 0);
2105 static __isl_give isl_printer
*print_pw_aff_isl(__isl_take isl_printer
*p
,
2106 __isl_keep isl_pw_aff
*pwaff
)
2108 struct isl_print_space_data data
= { 0 };
2113 if (isl_space_dim(pwaff
->dim
, isl_dim_param
) > 0) {
2114 p
= print_tuple(pwaff
->dim
, p
, isl_dim_param
, &data
);
2115 p
= isl_printer_print_str(p
, " -> ");
2117 p
= isl_printer_print_str(p
, "{ ");
2118 p
= print_pw_aff_body(p
, pwaff
);
2119 p
= isl_printer_print_str(p
, " }");
2122 isl_printer_free(p
);
2126 static __isl_give isl_printer
*print_ls_affine_c(__isl_take isl_printer
*p
,
2127 __isl_keep isl_local_space
*ls
, isl_int
*c
);
2129 static __isl_give isl_printer
*print_ls_name_c(__isl_take isl_printer
*p
,
2130 __isl_keep isl_local_space
*ls
, enum isl_dim_type type
, unsigned pos
)
2132 if (type
== isl_dim_div
) {
2133 p
= isl_printer_print_str(p
, "floord(");
2134 p
= print_ls_affine_c(p
, ls
, ls
->div
->row
[pos
] + 1);
2135 p
= isl_printer_print_str(p
, ", ");
2136 p
= isl_printer_print_isl_int(p
, ls
->div
->row
[pos
][0]);
2137 p
= isl_printer_print_str(p
, ")");
2141 name
= isl_space_get_dim_name(ls
->dim
, type
, pos
);
2144 p
= isl_printer_print_str(p
, name
);
2149 static __isl_give isl_printer
*print_ls_term_c(__isl_take isl_printer
*p
,
2150 __isl_keep isl_local_space
*ls
, isl_int c
, unsigned pos
)
2152 enum isl_dim_type type
;
2155 return isl_printer_print_isl_int(p
, c
);
2157 if (isl_int_is_one(c
))
2159 else if (isl_int_is_negone(c
))
2160 p
= isl_printer_print_str(p
, "-");
2162 p
= isl_printer_print_isl_int(p
, c
);
2163 p
= isl_printer_print_str(p
, "*");
2165 type
= pos2type(ls
->dim
, &pos
);
2166 p
= print_ls_name_c(p
, ls
, type
, pos
);
2170 static __isl_give isl_printer
*print_ls_partial_affine_c(
2171 __isl_take isl_printer
*p
, __isl_keep isl_local_space
*ls
,
2172 isl_int
*c
, unsigned len
)
2177 for (i
= 0, first
= 1; i
< len
; ++i
) {
2179 if (isl_int_is_zero(c
[i
]))
2182 if (isl_int_is_neg(c
[i
])) {
2184 isl_int_neg(c
[i
], c
[i
]);
2185 p
= isl_printer_print_str(p
, " - ");
2187 p
= isl_printer_print_str(p
, " + ");
2190 p
= print_ls_term_c(p
, ls
, c
[i
], i
);
2192 isl_int_neg(c
[i
], c
[i
]);
2195 p
= isl_printer_print_str(p
, "0");
2199 static __isl_give isl_printer
*print_ls_affine_c(__isl_take isl_printer
*p
,
2200 __isl_keep isl_local_space
*ls
, isl_int
*c
)
2202 unsigned len
= 1 + isl_local_space_dim(ls
, isl_dim_all
);
2203 return print_ls_partial_affine_c(p
, ls
, c
, len
);
2206 static __isl_give isl_printer
*print_aff_c(__isl_take isl_printer
*p
,
2207 __isl_keep isl_aff
*aff
)
2211 total
= isl_local_space_dim(aff
->ls
, isl_dim_all
);
2212 if (!isl_int_is_one(aff
->v
->el
[0]))
2213 p
= isl_printer_print_str(p
, "(");
2214 p
= print_ls_partial_affine_c(p
, aff
->ls
, aff
->v
->el
+ 1, 1 + total
);
2215 if (!isl_int_is_one(aff
->v
->el
[0])) {
2216 p
= isl_printer_print_str(p
, ")/");
2217 p
= isl_printer_print_isl_int(p
, aff
->v
->el
[0]);
2222 /* In the C format, we cannot express that "pwaff" may be undefined
2223 * on parts of the domain space. We therefore assume that the expression
2224 * will only be evaluated on its definition domain and compute the gist
2225 * of each cell with respect to this domain.
2227 static __isl_give isl_printer
*print_pw_aff_c(__isl_take isl_printer
*p
,
2228 __isl_keep isl_pw_aff
*pwaff
)
2231 isl_ast_build
*build
;
2235 isl_die(p
->ctx
, isl_error_unsupported
,
2236 "cannot print empty isl_pw_aff in C format",
2237 return isl_printer_free(p
));
2239 domain
= isl_pw_aff_domain(isl_pw_aff_copy(pwaff
));
2240 build
= isl_ast_build_from_context(domain
);
2241 expr
= isl_ast_build_expr_from_pw_aff(build
, isl_pw_aff_copy(pwaff
));
2242 p
= isl_printer_print_ast_expr(p
, expr
);
2243 isl_ast_expr_free(expr
);
2244 isl_ast_build_free(build
);
2249 __isl_give isl_printer
*isl_printer_print_aff(__isl_take isl_printer
*p
,
2250 __isl_keep isl_aff
*aff
)
2255 if (p
->output_format
== ISL_FORMAT_ISL
)
2256 return print_aff_isl(p
, aff
);
2257 else if (p
->output_format
== ISL_FORMAT_C
)
2258 return print_aff_c(p
, aff
);
2259 isl_die(p
->ctx
, isl_error_unsupported
, "unsupported output format",
2262 isl_printer_free(p
);
2266 __isl_give isl_printer
*isl_printer_print_pw_aff(__isl_take isl_printer
*p
,
2267 __isl_keep isl_pw_aff
*pwaff
)
2272 if (p
->output_format
== ISL_FORMAT_ISL
)
2273 return print_pw_aff_isl(p
, pwaff
);
2274 else if (p
->output_format
== ISL_FORMAT_C
)
2275 return print_pw_aff_c(p
, pwaff
);
2276 isl_die(p
->ctx
, isl_error_unsupported
, "unsupported output format",
2279 isl_printer_free(p
);
2283 /* Print "pa" in a sequence of isl_pw_affs delimited by semicolons.
2284 * Each isl_pw_aff itself is also printed as semicolon delimited
2285 * sequence of pieces.
2286 * If data->first = 1, then this is the first in the sequence.
2287 * Update data->first to tell the next element that it is not the first.
2289 static isl_stat
print_pw_aff_body_wrap(__isl_take isl_pw_aff
*pa
,
2292 struct isl_union_print_data
*data
;
2293 data
= (struct isl_union_print_data
*) user
;
2296 data
->p
= isl_printer_print_str(data
->p
, "; ");
2299 data
->p
= print_pw_aff_body(data
->p
, pa
);
2300 isl_pw_aff_free(pa
);
2302 return data
->p
? isl_stat_ok
: isl_stat_error
;
2305 /* Print the body of an isl_union_pw_aff, i.e., a semicolon delimited
2306 * sequence of affine expressions, each followed by constraints,
2307 * with the sequence enclosed in braces.
2309 static __isl_give isl_printer
*print_union_pw_aff_body(
2310 __isl_take isl_printer
*p
, __isl_keep isl_union_pw_aff
*upa
)
2312 struct isl_union_print_data data
= { p
, 1 };
2314 p
= isl_printer_print_str(p
, s_open_set
[0]);
2316 if (isl_union_pw_aff_foreach_pw_aff(upa
,
2317 &print_pw_aff_body_wrap
, &data
) < 0)
2318 data
.p
= isl_printer_free(p
);
2320 p
= isl_printer_print_str(p
, s_close_set
[0]);
2325 /* Print the isl_union_pw_aff "upa" to "p" in isl format.
2327 * The individual isl_pw_affs are delimited by a semicolon.
2329 static __isl_give isl_printer
*print_union_pw_aff_isl(
2330 __isl_take isl_printer
*p
, __isl_keep isl_union_pw_aff
*upa
)
2332 struct isl_print_space_data data
= { 0 };
2335 space
= isl_union_pw_aff_get_space(upa
);
2336 if (isl_space_dim(space
, isl_dim_param
) > 0) {
2337 p
= print_tuple(space
, p
, isl_dim_param
, &data
);
2338 p
= isl_printer_print_str(p
, s_to
[0]);
2340 isl_space_free(space
);
2341 p
= print_union_pw_aff_body(p
, upa
);
2345 /* Print the isl_union_pw_aff "upa" to "p".
2347 * We currently only support an isl format.
2349 __isl_give isl_printer
*isl_printer_print_union_pw_aff(
2350 __isl_take isl_printer
*p
, __isl_keep isl_union_pw_aff
*upa
)
2353 return isl_printer_free(p
);
2355 if (p
->output_format
== ISL_FORMAT_ISL
)
2356 return print_union_pw_aff_isl(p
, upa
);
2357 isl_die(isl_printer_get_ctx(p
), isl_error_unsupported
,
2358 "unsupported output format", return isl_printer_free(p
));
2361 /* Print dimension "pos" of data->space to "p".
2363 * data->user is assumed to be an isl_multi_aff.
2365 * If the current dimension is an output dimension, then print
2366 * the corresponding expression. Otherwise, print the name of the dimension.
2368 static __isl_give isl_printer
*print_dim_ma(__isl_take isl_printer
*p
,
2369 struct isl_print_space_data
*data
, unsigned pos
)
2371 isl_multi_aff
*ma
= data
->user
;
2373 if (data
->type
== isl_dim_out
)
2374 p
= print_aff_body(p
, ma
->p
[pos
]);
2376 p
= print_name(data
->space
, p
, data
->type
, pos
, data
->latex
);
2381 static __isl_give isl_printer
*print_multi_aff(__isl_take isl_printer
*p
,
2382 __isl_keep isl_multi_aff
*maff
)
2384 struct isl_print_space_data data
= { 0 };
2386 data
.print_dim
= &print_dim_ma
;
2388 return isl_print_space(maff
->space
, p
, 0, &data
);
2391 static __isl_give isl_printer
*print_multi_aff_isl(__isl_take isl_printer
*p
,
2392 __isl_keep isl_multi_aff
*maff
)
2394 struct isl_print_space_data data
= { 0 };
2399 if (isl_space_dim(maff
->space
, isl_dim_param
) > 0) {
2400 p
= print_tuple(maff
->space
, p
, isl_dim_param
, &data
);
2401 p
= isl_printer_print_str(p
, " -> ");
2403 p
= isl_printer_print_str(p
, "{ ");
2404 p
= print_multi_aff(p
, maff
);
2405 p
= isl_printer_print_str(p
, " }");
2408 isl_printer_free(p
);
2412 __isl_give isl_printer
*isl_printer_print_multi_aff(__isl_take isl_printer
*p
,
2413 __isl_keep isl_multi_aff
*maff
)
2418 if (p
->output_format
== ISL_FORMAT_ISL
)
2419 return print_multi_aff_isl(p
, maff
);
2420 isl_die(p
->ctx
, isl_error_unsupported
, "unsupported output format",
2423 isl_printer_free(p
);
2427 static __isl_give isl_printer
*print_pw_multi_aff_body(
2428 __isl_take isl_printer
*p
, __isl_keep isl_pw_multi_aff
*pma
)
2435 for (i
= 0; i
< pma
->n
; ++i
) {
2437 p
= isl_printer_print_str(p
, "; ");
2438 p
= print_multi_aff(p
, pma
->p
[i
].maff
);
2439 p
= print_disjuncts((isl_map
*)pma
->p
[i
].set
, p
, 0);
2443 isl_printer_free(p
);
2447 static __isl_give isl_printer
*print_pw_multi_aff_isl(__isl_take isl_printer
*p
,
2448 __isl_keep isl_pw_multi_aff
*pma
)
2450 struct isl_print_space_data data
= { 0 };
2455 if (isl_space_dim(pma
->dim
, isl_dim_param
) > 0) {
2456 p
= print_tuple(pma
->dim
, p
, isl_dim_param
, &data
);
2457 p
= isl_printer_print_str(p
, " -> ");
2459 p
= isl_printer_print_str(p
, "{ ");
2460 p
= print_pw_multi_aff_body(p
, pma
);
2461 p
= isl_printer_print_str(p
, " }");
2464 isl_printer_free(p
);
2468 static __isl_give isl_printer
*print_unnamed_pw_multi_aff_c(
2469 __isl_take isl_printer
*p
, __isl_keep isl_pw_multi_aff
*pma
)
2473 for (i
= 0; i
< pma
->n
- 1; ++i
) {
2474 p
= isl_printer_print_str(p
, "(");
2475 p
= print_set_c(p
, pma
->dim
, pma
->p
[i
].set
);
2476 p
= isl_printer_print_str(p
, ") ? (");
2477 p
= print_aff_c(p
, pma
->p
[i
].maff
->p
[0]);
2478 p
= isl_printer_print_str(p
, ") : ");
2481 return print_aff_c(p
, pma
->p
[pma
->n
- 1].maff
->p
[0]);
2484 static __isl_give isl_printer
*print_pw_multi_aff_c(__isl_take isl_printer
*p
,
2485 __isl_keep isl_pw_multi_aff
*pma
)
2493 isl_die(p
->ctx
, isl_error_unsupported
,
2494 "cannot print empty isl_pw_multi_aff in C format",
2496 name
= isl_pw_multi_aff_get_tuple_name(pma
, isl_dim_out
);
2497 if (!name
&& isl_pw_multi_aff_dim(pma
, isl_dim_out
) == 1)
2498 return print_unnamed_pw_multi_aff_c(p
, pma
);
2500 isl_die(p
->ctx
, isl_error_unsupported
,
2501 "cannot print unnamed isl_pw_multi_aff in C format",
2504 p
= isl_printer_print_str(p
, name
);
2505 n
= isl_pw_multi_aff_dim(pma
, isl_dim_out
);
2507 isl_die(p
->ctx
, isl_error_unsupported
,
2508 "not supported yet", goto error
);
2512 isl_printer_free(p
);
2516 __isl_give isl_printer
*isl_printer_print_pw_multi_aff(
2517 __isl_take isl_printer
*p
, __isl_keep isl_pw_multi_aff
*pma
)
2522 if (p
->output_format
== ISL_FORMAT_ISL
)
2523 return print_pw_multi_aff_isl(p
, pma
);
2524 if (p
->output_format
== ISL_FORMAT_C
)
2525 return print_pw_multi_aff_c(p
, pma
);
2526 isl_die(p
->ctx
, isl_error_unsupported
, "unsupported output format",
2529 isl_printer_free(p
);
2533 static isl_stat
print_pw_multi_aff_body_wrap(__isl_take isl_pw_multi_aff
*pma
,
2536 struct isl_union_print_data
*data
;
2537 data
= (struct isl_union_print_data
*) user
;
2540 data
->p
= isl_printer_print_str(data
->p
, "; ");
2543 data
->p
= print_pw_multi_aff_body(data
->p
, pma
);
2544 isl_pw_multi_aff_free(pma
);
2549 static __isl_give isl_printer
*print_union_pw_multi_aff_isl(
2550 __isl_take isl_printer
*p
, __isl_keep isl_union_pw_multi_aff
*upma
)
2552 struct isl_union_print_data data
;
2553 struct isl_print_space_data space_data
= { 0 };
2556 space
= isl_union_pw_multi_aff_get_space(upma
);
2557 if (isl_space_dim(space
, isl_dim_param
) > 0) {
2558 p
= print_tuple(space
, p
, isl_dim_param
, &space_data
);
2559 p
= isl_printer_print_str(p
, s_to
[0]);
2561 isl_space_free(space
);
2562 p
= isl_printer_print_str(p
, s_open_set
[0]);
2565 isl_union_pw_multi_aff_foreach_pw_multi_aff(upma
,
2566 &print_pw_multi_aff_body_wrap
, &data
);
2568 p
= isl_printer_print_str(p
, s_close_set
[0]);
2572 __isl_give isl_printer
*isl_printer_print_union_pw_multi_aff(
2573 __isl_take isl_printer
*p
, __isl_keep isl_union_pw_multi_aff
*upma
)
2578 if (p
->output_format
== ISL_FORMAT_ISL
)
2579 return print_union_pw_multi_aff_isl(p
, upma
);
2580 isl_die(p
->ctx
, isl_error_unsupported
, "unsupported output format",
2583 isl_printer_free(p
);
2587 /* Print dimension "pos" of data->space to "p".
2589 * data->user is assumed to be an isl_multi_pw_aff.
2591 * If the current dimension is an output dimension, then print
2592 * the corresponding piecewise affine expression.
2593 * Otherwise, print the name of the dimension.
2595 static __isl_give isl_printer
*print_dim_mpa(__isl_take isl_printer
*p
,
2596 struct isl_print_space_data
*data
, unsigned pos
)
2600 isl_multi_pw_aff
*mpa
= data
->user
;
2603 if (data
->type
!= isl_dim_out
)
2604 return print_name(data
->space
, p
, data
->type
, pos
, data
->latex
);
2608 return isl_printer_print_str(p
, "(0 : 1 = 0)");
2610 need_parens
= pa
->n
!= 1 || !isl_set_plain_is_universe(pa
->p
[0].set
);
2612 p
= isl_printer_print_str(p
, "(");
2613 for (i
= 0; i
< pa
->n
; ++i
) {
2615 p
= isl_printer_print_str(p
, "; ");
2616 p
= print_aff_body(p
, pa
->p
[i
].aff
);
2617 p
= print_disjuncts(pa
->p
[i
].set
, p
, 0);
2620 p
= isl_printer_print_str(p
, ")");
2625 /* Print "mpa" to "p" in isl format.
2627 static __isl_give isl_printer
*print_multi_pw_aff_isl(__isl_take isl_printer
*p
,
2628 __isl_keep isl_multi_pw_aff
*mpa
)
2630 struct isl_print_space_data data
= { 0 };
2633 return isl_printer_free(p
);
2635 if (isl_space_dim(mpa
->space
, isl_dim_param
) > 0) {
2636 p
= print_tuple(mpa
->space
, p
, isl_dim_param
, &data
);
2637 p
= isl_printer_print_str(p
, " -> ");
2639 p
= isl_printer_print_str(p
, "{ ");
2640 data
.print_dim
= &print_dim_mpa
;
2642 p
= isl_print_space(mpa
->space
, p
, 0, &data
);
2643 p
= isl_printer_print_str(p
, " }");
2647 __isl_give isl_printer
*isl_printer_print_multi_pw_aff(
2648 __isl_take isl_printer
*p
, __isl_keep isl_multi_pw_aff
*mpa
)
2651 return isl_printer_free(p
);
2653 if (p
->output_format
== ISL_FORMAT_ISL
)
2654 return print_multi_pw_aff_isl(p
, mpa
);
2655 isl_die(p
->ctx
, isl_error_unsupported
, "unsupported output format",
2656 return isl_printer_free(p
));
2659 /* Print dimension "pos" of data->space to "p".
2661 * data->user is assumed to be an isl_multi_val.
2663 * If the current dimension is an output dimension, then print
2664 * the corresponding value. Otherwise, print the name of the dimension.
2666 static __isl_give isl_printer
*print_dim_mv(__isl_take isl_printer
*p
,
2667 struct isl_print_space_data
*data
, unsigned pos
)
2669 isl_multi_val
*mv
= data
->user
;
2671 if (data
->type
== isl_dim_out
)
2672 return isl_printer_print_val(p
, mv
->p
[pos
]);
2674 return print_name(data
->space
, p
, data
->type
, pos
, data
->latex
);
2677 /* Print the isl_multi_val "mv" to "p" in isl format.
2679 static __isl_give isl_printer
*print_multi_val_isl(__isl_take isl_printer
*p
,
2680 __isl_keep isl_multi_val
*mv
)
2682 struct isl_print_space_data data
= { 0 };
2685 return isl_printer_free(p
);
2687 if (isl_space_dim(mv
->space
, isl_dim_param
) > 0) {
2688 p
= print_tuple(mv
->space
, p
, isl_dim_param
, &data
);
2689 p
= isl_printer_print_str(p
, " -> ");
2691 p
= isl_printer_print_str(p
, "{ ");
2692 data
.print_dim
= &print_dim_mv
;
2694 p
= isl_print_space(mv
->space
, p
, 0, &data
);
2695 p
= isl_printer_print_str(p
, " }");
2699 /* Print the isl_multi_val "mv" to "p".
2701 * Currently only supported in isl format.
2703 __isl_give isl_printer
*isl_printer_print_multi_val(
2704 __isl_take isl_printer
*p
, __isl_keep isl_multi_val
*mv
)
2707 return isl_printer_free(p
);
2709 if (p
->output_format
== ISL_FORMAT_ISL
)
2710 return print_multi_val_isl(p
, mv
);
2711 isl_die(p
->ctx
, isl_error_unsupported
, "unsupported output format",
2712 return isl_printer_free(p
));
2715 /* Print dimension "pos" of data->space to "p".
2717 * data->user is assumed to be an isl_multi_union_pw_aff.
2719 * The current dimension is necessarily a set dimension, so
2720 * we print the corresponding isl_union_pw_aff, including
2723 static __isl_give isl_printer
*print_union_pw_aff_dim(__isl_take isl_printer
*p
,
2724 struct isl_print_space_data
*data
, unsigned pos
)
2726 isl_multi_union_pw_aff
*mupa
= data
->user
;
2727 isl_union_pw_aff
*upa
;
2729 upa
= isl_multi_union_pw_aff_get_union_pw_aff(mupa
, pos
);
2730 p
= print_union_pw_aff_body(p
, upa
);
2731 isl_union_pw_aff_free(upa
);
2736 /* Print the isl_multi_union_pw_aff "mupa" to "p" in isl format.
2738 static __isl_give isl_printer
*print_multi_union_pw_aff_isl(
2739 __isl_take isl_printer
*p
, __isl_keep isl_multi_union_pw_aff
*mupa
)
2741 struct isl_print_space_data data
= { 0 };
2744 space
= isl_multi_union_pw_aff_get_space(mupa
);
2745 if (isl_space_dim(space
, isl_dim_param
) > 0) {
2746 struct isl_print_space_data space_data
= { 0 };
2747 p
= print_tuple(space
, p
, isl_dim_param
, &space_data
);
2748 p
= isl_printer_print_str(p
, s_to
[0]);
2751 data
.print_dim
= &print_union_pw_aff_dim
;
2754 p
= isl_print_space(space
, p
, 0, &data
);
2755 isl_space_free(space
);
2760 /* Print the isl_multi_union_pw_aff "mupa" to "p" in isl format.
2762 * We currently only support an isl format.
2764 __isl_give isl_printer
*isl_printer_print_multi_union_pw_aff(
2765 __isl_take isl_printer
*p
, __isl_keep isl_multi_union_pw_aff
*mupa
)
2768 return isl_printer_free(p
);
2770 if (p
->output_format
== ISL_FORMAT_ISL
)
2771 return print_multi_union_pw_aff_isl(p
, mupa
);
2772 isl_die(isl_printer_get_ctx(p
), isl_error_unsupported
,
2773 "unsupported output format", return isl_printer_free(p
));