?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
add isl_qpolynomial_one
[isl.git] / isl_input.c
blob0f825c21d303c98123354c11fbff0f847b8389d0
1 /*
2 * Copyright 2008-2009 Katholieke Universiteit Leuven
3 * Copyright 2010 INRIA Saclay
4 *
5 * Use of this software is governed by the GNU LGPLv2.1 license
6 *
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
11 */
12
13 #include <ctype.h>
14 #include <stdio.h>
15 #include <string.h>
16 #include <strings.h>
17 #include <isl_set.h>
18 #include <isl_seq.h>
19 #include <isl_div.h>
20 #include "isl_stream.h"
21 #include "isl_map_private.h"
22 #include "isl_obj.h"
23 #include "isl_polynomial_private.h"
24 #include <isl_union_map.h>
25 #include <isl_mat_private.h>
26
27 struct variable {
28 char *name;
29 int pos;
30 struct variable *next;
31 };
32
33 struct vars {
34 struct isl_ctx *ctx;
35 int n;
36 struct variable *v;
37 };
38
39 static struct vars *vars_new(struct isl_ctx *ctx)
40 {
41 struct vars *v;
42 v = isl_alloc_type(ctx, struct vars);
43 if (!v)
44 return NULL;
45 v->ctx = ctx;
46 v->n = 0;
47 v->v = NULL;
48 return v;
49 }
50
51 static void variable_free(struct variable *var)
52 {
53 while (var) {
54 struct variable *next = var->next;
55 free(var->name);
56 free(var);
57 var = next;
58 }
59 }
60
61 static void vars_free(struct vars *v)
62 {
63 if (!v)
64 return;
65 variable_free(v->v);
66 free(v);
67 }
68
69 static void vars_drop(struct vars *v, int n)
70 {
71 struct variable *var;
72
73 if (!v || !v->v)
74 return;
75
76 v->n -= n;
77
78 var = v->v;
79 while (--n >= 0) {
80 struct variable *next = var->next;
81 free(var->name);
82 free(var);
83 var = next;
84 }
85 v->v = var;
86 }
87
88 static struct variable *variable_new(struct vars *v, const char *name, int len,
89 int pos)
90 {
91 struct variable *var;
92 var = isl_alloc_type(v->ctx, struct variable);
93 if (!var)
94 goto error;
95 var->name = strdup(name);
96 var->name[len] = '\0';
97 var->pos = pos;
98 var->next = v->v;
99 return var;
100 error:
101 variable_free(v->v);
102 return NULL;
103 }
104
105 static int vars_pos(struct vars *v, const char *s, int len)
106 {
107 int pos;
108 struct variable *q;
109
110 if (len == -1)
111 len = strlen(s);
112 for (q = v->v; q; q = q->next) {
113 if (strncmp(q->name, s, len) == 0 && q->name[len] == '\0')
114 break;
115 }
116 if (q)
117 pos = q->pos;
118 else {
119 pos = v->n;
120 v->v = variable_new(v, s, len, v->n);
121 if (!v->v)
122 return -1;
123 v->n++;
124 }
125 return pos;
126 }
127
128 static __isl_give isl_dim *set_name(__isl_take isl_dim *dim,
129 enum isl_dim_type type, unsigned pos, char *name)
130 {
131 char *prime;
132
133 if (!dim)
134 return NULL;
135 if (!name)
136 return dim;
137
138 prime = strchr(name, '\'');
139 if (prime)
140 *prime = '\0';
141 dim = isl_dim_set_name(dim, type, pos, name);
142 if (prime)
143 *prime = '\'';
144
145 return dim;
146 }
147
148 static int accept_cst_factor(struct isl_stream *s, isl_int *f)
149 {
150 struct isl_token *tok;
151
152 tok = isl_stream_next_token(s);
153 if (!tok || tok->type != ISL_TOKEN_VALUE) {
154 isl_stream_error(s, tok, "expecting constant value");
155 goto error;
156 }
157
158 isl_int_mul(*f, *f, tok->u.v);
159
160 isl_token_free(tok);
161
162 if (isl_stream_eat_if_available(s, '*'))
163 return accept_cst_factor(s, f);
164
165 return 0;
166 error:
167 isl_token_free(tok);
168 return -1;
169 }
170
171 static struct isl_vec *accept_affine(struct isl_stream *s, struct vars *v);
172
173 static __isl_give isl_vec *accept_affine_factor(struct isl_stream *s,
174 struct vars *v)
175 {
176 struct isl_token *tok = NULL;
177 isl_vec *aff = NULL;
178
179 tok = isl_stream_next_token(s);
180 if (!tok) {
181 isl_stream_error(s, NULL, "unexpected EOF");
182 goto error;
183 }
184 if (tok->type == ISL_TOKEN_IDENT) {
185 int n = v->n;
186 int pos = vars_pos(v, tok->u.s, -1);
187 if (pos < 0)
188 goto error;
189 if (pos >= n) {
190 isl_stream_error(s, tok, "unknown identifier");
191 goto error;
192 }
193
194 aff = isl_vec_alloc(v->ctx, 1 + v->n);
195 if (!aff)
196 goto error;
197 isl_seq_clr(aff->el, aff->size);
198 isl_int_set_si(aff->el[1 + pos], 1);
199 isl_token_free(tok);
200 } else if (tok->type == ISL_TOKEN_VALUE) {
201 if (isl_stream_eat_if_available(s, '*')) {
202 aff = accept_affine_factor(s, v);
203 aff = isl_vec_scale(aff, tok->u.v);
204 } else {
205 aff = isl_vec_alloc(v->ctx, 1 + v->n);
206 if (!aff)
207 goto error;
208 isl_seq_clr(aff->el, aff->size);
209 isl_int_set(aff->el[0], tok->u.v);
210 }
211 isl_token_free(tok);
212 } else if (tok->type == '(') {
213 isl_token_free(tok);
214 tok = NULL;
215 aff = accept_affine(s, v);
216 if (!aff)
217 goto error;
218 if (isl_stream_eat(s, ')'))
219 goto error;
220 } else {
221 isl_stream_error(s, tok, "expecting factor");
222 goto error;
223 }
224 if (isl_stream_eat_if_available(s, '*')) {
225 isl_int f;
226 isl_int_init(f);
227 isl_int_set_si(f, 1);
228 if (accept_cst_factor(s, &f) < 0) {
229 isl_int_clear(f);
230 goto error;
231 }
232 aff = isl_vec_scale(aff, f);
233 isl_int_clear(f);
234 }
235
236 return aff;
237 error:
238 isl_token_free(tok);
239 isl_vec_free(aff);
240 return NULL;
241 }
242
243 static struct isl_vec *accept_affine(struct isl_stream *s, struct vars *v)
244 {
245 struct isl_token *tok = NULL;
246 struct isl_vec *aff;
247 int sign = 1;
248
249 aff = isl_vec_alloc(v->ctx, 1 + v->n);
250 if (!aff)
251 return NULL;
252 isl_seq_clr(aff->el, aff->size);
253
254 for (;;) {
255 tok = isl_stream_next_token(s);
256 if (!tok) {
257 isl_stream_error(s, NULL, "unexpected EOF");
258 goto error;
259 }
260 if (tok->type == '-') {
261 sign = -sign;
262 isl_token_free(tok);
263 continue;
264 }
265 if (tok->type == '(' || tok->type == ISL_TOKEN_IDENT) {
266 isl_vec *aff2;
267 isl_stream_push_token(s, tok);
268 tok = NULL;
269 aff2 = accept_affine_factor(s, v);
270 if (sign < 0)
271 aff2 = isl_vec_scale(aff2, s->ctx->negone);
272 aff = isl_vec_add(aff, aff2);
273 if (!aff)
274 goto error;
275 sign = 1;
276 } else if (tok->type == ISL_TOKEN_VALUE) {
277 if (sign < 0)
278 isl_int_neg(tok->u.v, tok->u.v);
279 if (isl_stream_eat_if_available(s, '*') ||
280 isl_stream_next_token_is(s, ISL_TOKEN_IDENT)) {
281 isl_vec *aff2;
282 aff2 = accept_affine_factor(s, v);
283 aff2 = isl_vec_scale(aff2, tok->u.v);
284 aff = isl_vec_add(aff, aff2);
285 if (!aff)
286 goto error;
287 } else {
288 isl_int_add(aff->el[0], aff->el[0], tok->u.v);
289 }
290 sign = 1;
291 }
292 isl_token_free(tok);
293
294 tok = isl_stream_next_token(s);
295 if (tok && tok->type == '-') {
296 sign = -sign;
297 isl_token_free(tok);
298 } else if (tok && tok->type == '+') {
299 /* nothing */
300 isl_token_free(tok);
301 } else if (tok && tok->type == ISL_TOKEN_VALUE &&
302 isl_int_is_neg(tok->u.v)) {
303 isl_stream_push_token(s, tok);
304 } else {
305 if (tok)
306 isl_stream_push_token(s, tok);
307 break;
308 }
309 }
310
311 return aff;
312 error:
313 isl_token_free(tok);
314 isl_vec_free(aff);
315 return NULL;
316 }
317
318 static __isl_give isl_mat *read_var_def(struct isl_stream *s,
319 __isl_take isl_mat *eq, enum isl_dim_type type, struct vars *v)
320 {
321 struct isl_vec *vec;
322
323 vec = accept_affine(s, v);
324 if (!vec)
325 goto error;
326 v->v = variable_new(v, "", 0, v->n);
327 if (!v->v)
328 goto error;
329 v->n++;
330 eq = isl_mat_add_rows(eq, 1);
331 if (eq) {
332 isl_seq_cpy(eq->row[eq->n_row - 1], vec->el, vec->size);
333 isl_int_set_si(eq->row[eq->n_row - 1][vec->size], -1);
334 }
335 isl_vec_free(vec);
336
337 return eq;
338 error:
339 isl_mat_free(eq);
340 return NULL;
341 }
342
343 static __isl_give isl_dim *read_var_list(struct isl_stream *s,
344 __isl_take isl_dim *dim, enum isl_dim_type type, struct vars *v,
345 __isl_keep isl_mat **eq)
346 {
347 int i = 0;
348 struct isl_token *tok;
349
350 while ((tok = isl_stream_next_token(s)) != NULL) {
351 int new_name = 0;
352
353 if (tok->type == ISL_TOKEN_IDENT) {
354 int n = v->n;
355 int p = vars_pos(v, tok->u.s, -1);
356 if (p < 0)
357 goto error;
358 new_name = p >= n;
359 }
360
361 if (new_name) {
362 dim = isl_dim_add(dim, type, 1);
363 if (eq)
364 *eq = isl_mat_add_zero_cols(*eq, 1);
365 dim = set_name(dim, type, i, v->v->name);
366 isl_token_free(tok);
367 } else if (tok->type == ISL_TOKEN_IDENT ||
368 tok->type == ISL_TOKEN_VALUE ||
369 tok->type == '-') {
370 if (type == isl_dim_param) {
371 isl_stream_error(s, tok,
372 "expecting unique identifier");
373 goto error;
374 }
375 isl_stream_push_token(s, tok);
376 tok = NULL;
377 dim = isl_dim_add(dim, type, 1);
378 *eq = isl_mat_add_zero_cols(*eq, 1);
379 *eq = read_var_def(s, *eq, type, v);
380 } else
381 break;
382
383 tok = isl_stream_next_token(s);
384 if (!tok || tok->type != ',')
385 break;
386
387 isl_token_free(tok);
388 i++;
389 }
390 if (tok)
391 isl_stream_push_token(s, tok);
392
393 return dim;
394 error:
395 isl_token_free(tok);
396 isl_dim_free(dim);
397 return NULL;
398 }
399
400 static __isl_give isl_mat *accept_affine_list(struct isl_stream *s,
401 struct vars *v)
402 {
403 struct isl_vec *vec;
404 struct isl_mat *mat;
405 struct isl_token *tok = NULL;
406
407 vec = accept_affine(s, v);
408 mat = isl_mat_from_row_vec(vec);
409 if (!mat)
410 return NULL;
411
412 for (;;) {
413 tok = isl_stream_next_token(s);
414 if (!tok) {
415 isl_stream_error(s, NULL, "unexpected EOF");
416 goto error;
417 }
418 if (tok->type != ',') {
419 isl_stream_push_token(s, tok);
420 break;
421 }
422 isl_token_free(tok);
423
424 vec = accept_affine(s, v);
425 mat = isl_mat_vec_concat(mat, vec);
426 if (!mat)
427 return NULL;
428 }
429
430 return mat;
431 error:
432 isl_mat_free(mat);
433 return NULL;
434 }
435
436 static struct isl_basic_map *add_div_definition(struct isl_stream *s,
437 struct vars *v, struct isl_basic_map *bmap, int k)
438 {
439 struct isl_token *tok;
440 int seen_paren = 0;
441 struct isl_vec *aff;
442
443 if (isl_stream_eat(s, '['))
444 goto error;
445
446 tok = isl_stream_next_token(s);
447 if (!tok)
448 goto error;
449 if (tok->type == '(') {
450 seen_paren = 1;
451 isl_token_free(tok);
452 } else
453 isl_stream_push_token(s, tok);
454
455 aff = accept_affine(s, v);
456 if (!aff)
457 goto error;
458
459 isl_seq_cpy(bmap->div[k] + 1, aff->el, aff->size);
460
461 isl_vec_free(aff);
462
463 if (seen_paren && isl_stream_eat(s, ')'))
464 goto error;
465 if (isl_stream_eat(s, '/'))
466 goto error;
467
468 tok = isl_stream_next_token(s);
469 if (!tok)
470 goto error;
471 if (tok->type != ISL_TOKEN_VALUE) {
472 isl_stream_error(s, tok, "expected denominator");
473 isl_stream_push_token(s, tok);
474 goto error;
475 }
476 isl_int_set(bmap->div[k][0], tok->u.v);
477 isl_token_free(tok);
478
479 if (isl_stream_eat(s, ']'))
480 goto error;
481
482 if (isl_basic_map_add_div_constraints(bmap, k) < 0)
483 goto error;
484
485 return bmap;
486 error:
487 isl_basic_map_free(bmap);
488 return NULL;
489 }
490
491 static struct isl_basic_map *read_defined_var_list(struct isl_stream *s,
492 struct vars *v, struct isl_basic_map *bmap)
493 {
494 struct isl_token *tok;
495
496 while ((tok = isl_stream_next_token(s)) != NULL) {
497 int k;
498 int p;
499 int n = v->n;
500 unsigned total = isl_basic_map_total_dim(bmap);
501
502 if (tok->type != ISL_TOKEN_IDENT)
503 break;
504
505 p = vars_pos(v, tok->u.s, -1);
506 if (p < 0)
507 goto error;
508 if (p < n) {
509 isl_stream_error(s, tok, "expecting unique identifier");
510 goto error;
511 }
512
513 bmap = isl_basic_map_cow(bmap);
514 bmap = isl_basic_map_extend_dim(bmap, isl_dim_copy(bmap->dim),
515 1, 0, 2);
516
517 if ((k = isl_basic_map_alloc_div(bmap)) < 0)
518 goto error;
519 isl_seq_clr(bmap->div[k], 1 + 1 + total);
520
521 isl_token_free(tok);
522 tok = isl_stream_next_token(s);
523 if (tok && tok->type == '=') {
524 isl_token_free(tok);
525 bmap = add_div_definition(s, v, bmap, k);
526 tok = isl_stream_next_token(s);
527 }
528
529 if (!tok || tok->type != ',')
530 break;
531
532 isl_token_free(tok);
533 }
534 if (tok)
535 isl_stream_push_token(s, tok);
536
537 return bmap;
538 error:
539 isl_token_free(tok);
540 isl_basic_map_free(bmap);
541 return NULL;
542 }
543
544 static int next_is_tuple(struct isl_stream *s)
545 {
546 struct isl_token *tok;
547 int is_tuple;
548
549 tok = isl_stream_next_token(s);
550 if (!tok)
551 return 0;
552 if (tok->type == '[') {
553 isl_stream_push_token(s, tok);
554 return 1;
555 }
556 if (tok->type != ISL_TOKEN_IDENT) {
557 isl_stream_push_token(s, tok);
558 return 0;
559 }
560
561 is_tuple = isl_stream_next_token_is(s, '[');
562
563 isl_stream_push_token(s, tok);
564
565 return is_tuple;
566 }
567
568 static __isl_give isl_dim *read_tuple(struct isl_stream *s,
569 __isl_take isl_dim *dim, enum isl_dim_type type, struct vars *v,
570 __isl_keep isl_mat **eq);
571
572 static __isl_give isl_dim *read_nested_tuple(struct isl_stream *s,
573 __isl_take isl_dim *dim, struct vars *v, __isl_keep isl_mat **eq)
574 {
575 dim = read_tuple(s, dim, isl_dim_in, v, eq);
576 if (isl_stream_eat(s, ISL_TOKEN_TO))
577 goto error;
578 dim = read_tuple(s, dim, isl_dim_out, v, eq);
579 dim = isl_dim_wrap(dim);
580 return dim;
581 error:
582 isl_dim_free(dim);
583 return NULL;
584 }
585
586 static __isl_give isl_dim *read_tuple(struct isl_stream *s,
587 __isl_take isl_dim *dim, enum isl_dim_type type, struct vars *v,
588 __isl_keep isl_mat **eq)
589 {
590 struct isl_token *tok;
591 char *name = NULL;
592
593 tok = isl_stream_next_token(s);
594 if (tok && tok->type == ISL_TOKEN_IDENT) {
595 name = strdup(tok->u.s);
596 if (!name)
597 goto error;
598 isl_token_free(tok);
599 tok = isl_stream_next_token(s);
600 }
601 if (!tok || tok->type != '[') {
602 isl_stream_error(s, tok, "expecting '['");
603 goto error;
604 }
605 isl_token_free(tok);
606 if (type != isl_dim_param && next_is_tuple(s)) {
607 isl_dim *nested = isl_dim_copy(dim);
608 nested = isl_dim_drop(nested, isl_dim_in, 0,
609 isl_dim_size(nested, isl_dim_in));
610 nested = isl_dim_drop(nested, isl_dim_out, 0,
611 isl_dim_size(nested, isl_dim_out));
612 nested = read_nested_tuple(s, nested, v, eq);
613 if (type == isl_dim_in) {
614 isl_dim_free(dim);
615 dim = isl_dim_reverse(nested);
616 } else {
617 dim = isl_dim_join(dim, nested);
618 }
619 } else
620 dim = read_var_list(s, dim, type, v, eq);
621 tok = isl_stream_next_token(s);
622 if (!tok || tok->type != ']') {
623 isl_stream_error(s, tok, "expecting ']'");
624 goto error;
625 }
626 isl_token_free(tok);
627
628 if (name) {
629 dim = isl_dim_set_tuple_name(dim, type, name);
630 free(name);
631 }
632
633 return dim;
634 error:
635 if (tok)
636 isl_token_free(tok);
637 isl_dim_free(dim);
638 return NULL;
639 }
640
641 static struct isl_basic_map *add_constraints(struct isl_stream *s,
642 struct vars *v, struct isl_basic_map *bmap);
643
644 static struct isl_basic_map *add_exists(struct isl_stream *s,
645 struct vars *v, struct isl_basic_map *bmap)
646 {
647 struct isl_token *tok;
648 int n = v->n;
649 int extra;
650 int seen_paren = 0;
651 int i;
652 unsigned total;
653
654 tok = isl_stream_next_token(s);
655 if (!tok)
656 goto error;
657 if (tok->type == '(') {
658 seen_paren = 1;
659 isl_token_free(tok);
660 } else
661 isl_stream_push_token(s, tok);
662
663 bmap = read_defined_var_list(s, v, bmap);
664 if (!bmap)
665 goto error;
666
667 if (isl_stream_eat(s, ':'))
668 goto error;
669 bmap = add_constraints(s, v, bmap);
670 if (seen_paren && isl_stream_eat(s, ')'))
671 goto error;
672 return bmap;
673 error:
674 isl_basic_map_free(bmap);
675 return NULL;
676 }
677
678 static __isl_give isl_basic_map *construct_constraint(
679 __isl_take isl_basic_map *bmap, enum isl_token_type type,
680 isl_int *left, isl_int *right)
681 {
682 int k;
683 unsigned len;
684 struct isl_ctx *ctx;
685
686 if (!bmap)
687 return NULL;
688 len = 1 + isl_basic_map_total_dim(bmap);
689 ctx = bmap->ctx;
690
691 k = isl_basic_map_alloc_inequality(bmap);
692 if (k < 0)
693 goto error;
694 if (type == ISL_TOKEN_LE)
695 isl_seq_combine(bmap->ineq[k], ctx->negone, left,
696 ctx->one, right,
697 len);
698 else if (type == ISL_TOKEN_GE)
699 isl_seq_combine(bmap->ineq[k], ctx->one, left,
700 ctx->negone, right,
701 len);
702 else if (type == ISL_TOKEN_LT) {
703 isl_seq_combine(bmap->ineq[k], ctx->negone, left,
704 ctx->one, right,
705 len);
706 isl_int_sub_ui(bmap->ineq[k][0], bmap->ineq[k][0], 1);
707 } else if (type == ISL_TOKEN_GT) {
708 isl_seq_combine(bmap->ineq[k], ctx->one, left,
709 ctx->negone, right,
710 len);
711 isl_int_sub_ui(bmap->ineq[k][0], bmap->ineq[k][0], 1);
712 } else {
713 isl_seq_combine(bmap->ineq[k], ctx->one, left,
714 ctx->negone, right,
715 len);
716 isl_basic_map_inequality_to_equality(bmap, k);
717 }
718
719 return bmap;
720 error:
721 isl_basic_map_free(bmap);
722 return NULL;
723 }
724
725 static int is_comparator(struct isl_token *tok)
726 {
727 if (!tok)
728 return 0;
729
730 switch (tok->type) {
731 case ISL_TOKEN_LT:
732 case ISL_TOKEN_GT:
733 case ISL_TOKEN_LE:
734 case ISL_TOKEN_GE:
735 case '=':
736 return 1;
737 default:
738 return 0;
739 }
740 }
741
742 static struct isl_basic_map *add_constraint(struct isl_stream *s,
743 struct vars *v, struct isl_basic_map *bmap)
744 {
745 int i, j;
746 unsigned total = isl_basic_map_total_dim(bmap);
747 struct isl_token *tok = NULL;
748 struct isl_mat *aff1 = NULL, *aff2 = NULL;
749
750 tok = isl_stream_next_token(s);
751 if (!tok)
752 goto error;
753 if (tok->type == ISL_TOKEN_EXISTS) {
754 isl_token_free(tok);
755 return add_exists(s, v, bmap);
756 }
757 isl_stream_push_token(s, tok);
758 tok = NULL;
759
760 bmap = isl_basic_map_cow(bmap);
761
762 aff1 = accept_affine_list(s, v);
763 if (!aff1)
764 goto error;
765 tok = isl_stream_next_token(s);
766 if (!is_comparator(tok)) {
767 isl_stream_error(s, tok, "missing operator");
768 if (tok)
769 isl_stream_push_token(s, tok);
770 tok = NULL;
771 goto error;
772 }
773 isl_assert(aff1->ctx, aff1->n_col == 1 + total, goto error);
774 for (;;) {
775 aff2 = accept_affine_list(s, v);
776 if (!aff2)
777 goto error;
778 isl_assert(aff2->ctx, aff2->n_col == 1 + total, goto error);
779
780 bmap = isl_basic_map_extend_constraints(bmap, 0,
781 aff1->n_row * aff2->n_row);
782 for (i = 0; i < aff1->n_row; ++i)
783 for (j = 0; j < aff2->n_row; ++j)
784 bmap = construct_constraint(bmap, tok->type,
785 aff1->row[i], aff2->row[j]);
786 isl_token_free(tok);
787 isl_mat_free(aff1);
788 aff1 = aff2;
789
790 tok = isl_stream_next_token(s);
791 if (!is_comparator(tok)) {
792 if (tok)
793 isl_stream_push_token(s, tok);
794 break;
795 }
796 }
797 isl_mat_free(aff1);
798
799 return bmap;
800 error:
801 if (tok)
802 isl_token_free(tok);
803 isl_mat_free(aff1);
804 isl_mat_free(aff2);
805 isl_basic_map_free(bmap);
806 return NULL;
807 }
808
809 static struct isl_basic_map *add_constraints(struct isl_stream *s,
810 struct vars *v, struct isl_basic_map *bmap)
811 {
812 struct isl_token *tok;
813
814 for (;;) {
815 bmap = add_constraint(s, v, bmap);
816 if (!bmap)
817 return NULL;
818 tok = isl_stream_next_token(s);
819 if (!tok) {
820 isl_stream_error(s, NULL, "unexpected EOF");
821 goto error;
822 }
823 if (tok->type != ISL_TOKEN_AND)
824 break;
825 isl_token_free(tok);
826 }
827 isl_stream_push_token(s, tok);
828
829 return bmap;
830 error:
831 if (tok)
832 isl_token_free(tok);
833 isl_basic_map_free(bmap);
834 return NULL;
835 }
836
837 static struct isl_basic_map *read_disjunct(struct isl_stream *s,
838 struct vars *v, __isl_take isl_dim *dim)
839 {
840 int seen_paren = 0;
841 struct isl_token *tok;
842 struct isl_basic_map *bmap;
843
844 bmap = isl_basic_map_alloc_dim(dim, 0, 0, 0);
845 if (!bmap)
846 return NULL;
847
848 tok = isl_stream_next_token(s);
849 if (!tok)
850 goto error;
851 if (tok->type == '(') {
852 seen_paren = 1;
853 isl_token_free(tok);
854 } else
855 isl_stream_push_token(s, tok);
856
857 bmap = add_constraints(s, v, bmap);
858 bmap = isl_basic_map_simplify(bmap);
859 bmap = isl_basic_map_finalize(bmap);
860
861 if (seen_paren && isl_stream_eat(s, ')'))
862 goto error;
863
864 return bmap;
865 error:
866 isl_basic_map_free(bmap);
867 return NULL;
868 }
869
870 static struct isl_map *read_disjuncts(struct isl_stream *s,
871 struct vars *v, __isl_take isl_dim *dim)
872 {
873 struct isl_token *tok;
874 struct isl_map *map;
875
876 tok = isl_stream_next_token(s);
877 if (!tok) {
878 isl_stream_error(s, NULL, "unexpected EOF");
879 goto error;
880 }
881 if (tok->type == '}') {
882 isl_stream_push_token(s, tok);
883 return isl_map_universe(dim);
884 }
885 isl_stream_push_token(s, tok);
886
887 map = isl_map_empty(isl_dim_copy(dim));
888 for (;;) {
889 struct isl_basic_map *bmap;
890 int n = v->n;
891
892 bmap = read_disjunct(s, v, isl_dim_copy(dim));
893 map = isl_map_union(map, isl_map_from_basic_map(bmap));
894
895 vars_drop(v, v->n - n);
896
897 tok = isl_stream_next_token(s);
898 if (!tok || tok->type != ISL_TOKEN_OR)
899 break;
900 isl_token_free(tok);
901 }
902 if (tok)
903 isl_stream_push_token(s, tok);
904
905 isl_dim_free(dim);
906 return map;
907 error:
908 isl_dim_free(dim);
909 return NULL;
910 }
911
912 static int polylib_pos_to_isl_pos(__isl_keep isl_basic_map *bmap, int pos)
913 {
914 if (pos < isl_basic_map_dim(bmap, isl_dim_out))
915 return 1 + isl_basic_map_dim(bmap, isl_dim_param) +
916 isl_basic_map_dim(bmap, isl_dim_in) + pos;
917 pos -= isl_basic_map_dim(bmap, isl_dim_out);
918
919 if (pos < isl_basic_map_dim(bmap, isl_dim_in))
920 return 1 + isl_basic_map_dim(bmap, isl_dim_param) + pos;
921 pos -= isl_basic_map_dim(bmap, isl_dim_in);
922
923 if (pos < isl_basic_map_dim(bmap, isl_dim_div))
924 return 1 + isl_basic_map_dim(bmap, isl_dim_param) +
925 isl_basic_map_dim(bmap, isl_dim_in) +
926 isl_basic_map_dim(bmap, isl_dim_out) + pos;
927 pos -= isl_basic_map_dim(bmap, isl_dim_div);
928
929 if (pos < isl_basic_map_dim(bmap, isl_dim_param))
930 return 1 + pos;
931
932 return 0;
933 }
934
935 static __isl_give isl_basic_map *basic_map_read_polylib_constraint(
936 struct isl_stream *s, __isl_take isl_basic_map *bmap)
937 {
938 int j;
939 struct isl_token *tok;
940 int type;
941 int k;
942 isl_int *c;
943 unsigned nparam;
944 unsigned dim;
945
946 if (!bmap)
947 return NULL;
948
949 nparam = isl_basic_map_dim(bmap, isl_dim_param);
950 dim = isl_basic_map_dim(bmap, isl_dim_out);
951
952 tok = isl_stream_next_token(s);
953 if (!tok || tok->type != ISL_TOKEN_VALUE) {
954 isl_stream_error(s, tok, "expecting coefficient");
955 if (tok)
956 isl_stream_push_token(s, tok);
957 goto error;
958 }
959 if (!tok->on_new_line) {
960 isl_stream_error(s, tok, "coefficient should appear on new line");
961 isl_stream_push_token(s, tok);
962 goto error;
963 }
964
965 type = isl_int_get_si(tok->u.v);
966 isl_token_free(tok);
967
968 isl_assert(s->ctx, type == 0 || type == 1, goto error);
969 if (type == 0) {
970 k = isl_basic_map_alloc_equality(bmap);
971 c = bmap->eq[k];
972 } else {
973 k = isl_basic_map_alloc_inequality(bmap);
974 c = bmap->ineq[k];
975 }
976 if (k < 0)
977 goto error;
978
979 for (j = 0; j < 1 + isl_basic_map_total_dim(bmap); ++j) {
980 int pos;
981 tok = isl_stream_next_token(s);
982 if (!tok || tok->type != ISL_TOKEN_VALUE) {
983 isl_stream_error(s, tok, "expecting coefficient");
984 if (tok)
985 isl_stream_push_token(s, tok);
986 goto error;
987 }
988 if (tok->on_new_line) {
989 isl_stream_error(s, tok,
990 "coefficient should not appear on new line");
991 isl_stream_push_token(s, tok);
992 goto error;
993 }
994 pos = polylib_pos_to_isl_pos(bmap, j);
995 isl_int_set(c[pos], tok->u.v);
996 isl_token_free(tok);
997 }
998
999 return bmap;
1000 error:
1001 isl_basic_map_free(bmap);
1002 return NULL;
1003 }
1004
1005 static __isl_give isl_basic_map *basic_map_read_polylib(struct isl_stream *s,
1006 int nparam)
1007 {
1008 int i;
1009 struct isl_token *tok;
1010 struct isl_token *tok2;
1011 int n_row, n_col;
1012 int on_new_line;
1013 unsigned in = 0, out, local = 0;
1014 struct isl_basic_map *bmap = NULL;
1015
1016 if (nparam < 0)
1017 nparam = 0;
1018
1019 tok = isl_stream_next_token(s);
1020 if (!tok) {
1021 isl_stream_error(s, NULL, "unexpected EOF");
1022 return NULL;
1023 }
1024 tok2 = isl_stream_next_token(s);
1025 if (!tok2) {
1026 isl_token_free(tok);
1027 isl_stream_error(s, NULL, "unexpected EOF");
1028 return NULL;
1029 }
1030 n_row = isl_int_get_si(tok->u.v);
1031 n_col = isl_int_get_si(tok2->u.v);
1032 on_new_line = tok2->on_new_line;
1033 isl_token_free(tok2);
1034 isl_token_free(tok);
1035 isl_assert(s->ctx, !on_new_line, return NULL);
1036 isl_assert(s->ctx, n_row >= 0, return NULL);
1037 isl_assert(s->ctx, n_col >= 2 + nparam, return NULL);
1038 tok = isl_stream_next_token_on_same_line(s);
1039 if (tok) {
1040 if (tok->type != ISL_TOKEN_VALUE) {
1041 isl_stream_error(s, tok,
1042 "expecting number of output dimensions");
1043 isl_stream_push_token(s, tok);
1044 goto error;
1045 }
1046 out = isl_int_get_si(tok->u.v);
1047 isl_token_free(tok);
1048
1049 tok = isl_stream_next_token_on_same_line(s);
1050 if (!tok || tok->type != ISL_TOKEN_VALUE) {
1051 isl_stream_error(s, tok,
1052 "expecting number of input dimensions");
1053 if (tok)
1054 isl_stream_push_token(s, tok);
1055 goto error;
1056 }
1057 in = isl_int_get_si(tok->u.v);
1058 isl_token_free(tok);
1059
1060 tok = isl_stream_next_token_on_same_line(s);
1061 if (!tok || tok->type != ISL_TOKEN_VALUE) {
1062 isl_stream_error(s, tok,
1063 "expecting number of existentials");
1064 if (tok)
1065 isl_stream_push_token(s, tok);
1066 goto error;
1067 }
1068 local = isl_int_get_si(tok->u.v);
1069 isl_token_free(tok);
1070
1071 tok = isl_stream_next_token_on_same_line(s);
1072 if (!tok || tok->type != ISL_TOKEN_VALUE) {
1073 isl_stream_error(s, tok,
1074 "expecting number of parameters");
1075 if (tok)
1076 isl_stream_push_token(s, tok);
1077 goto error;
1078 }
1079 nparam = isl_int_get_si(tok->u.v);
1080 isl_token_free(tok);
1081 if (n_col != 1 + out + in + local + nparam + 1) {
1082 isl_stream_error(s, NULL,
1083 "dimensions don't match");
1084 goto error;
1085 }
1086 } else
1087 out = n_col - 2 - nparam;
1088 bmap = isl_basic_map_alloc(s->ctx, nparam, in, out, local, n_row, n_row);
1089 if (!bmap)
1090 return NULL;
1091
1092 for (i = 0; i < local; ++i) {
1093 int k = isl_basic_map_alloc_div(bmap);
1094 if (k < 0)
1095 goto error;
1096 }
1097
1098 for (i = 0; i < n_row; ++i)
1099 bmap = basic_map_read_polylib_constraint(s, bmap);
1100
1101 tok = isl_stream_next_token_on_same_line(s);
1102 if (tok) {
1103 isl_stream_error(s, tok, "unexpected extra token on line");
1104 isl_stream_push_token(s, tok);
1105 goto error;
1106 }
1107
1108 bmap = isl_basic_map_simplify(bmap);
1109 bmap = isl_basic_map_finalize(bmap);
1110 return bmap;
1111 error:
1112 isl_basic_map_free(bmap);
1113 return NULL;
1114 }
1115
1116 static struct isl_map *map_read_polylib(struct isl_stream *s, int nparam)
1117 {
1118 struct isl_token *tok;
1119 struct isl_token *tok2;
1120 int i, n;
1121 struct isl_map *map;
1122
1123 tok = isl_stream_next_token(s);
1124 if (!tok) {
1125 isl_stream_error(s, NULL, "unexpected EOF");
1126 return NULL;
1127 }
1128 tok2 = isl_stream_next_token_on_same_line(s);
1129 if (tok2) {
1130 isl_stream_push_token(s, tok2);
1131 isl_stream_push_token(s, tok);
1132 return isl_map_from_basic_map(basic_map_read_polylib(s, nparam));
1133 }
1134 n = isl_int_get_si(tok->u.v);
1135 isl_token_free(tok);
1136
1137 isl_assert(s->ctx, n >= 1, return NULL);
1138
1139 map = isl_map_from_basic_map(basic_map_read_polylib(s, nparam));
1140
1141 for (i = 1; i < n; ++i)
1142 map = isl_map_union(map,
1143 isl_map_from_basic_map(basic_map_read_polylib(s, nparam)));
1144
1145 return map;
1146 }
1147
1148 static int optional_power(struct isl_stream *s)
1149 {
1150 int pow;
1151 struct isl_token *tok;
1152
1153 tok = isl_stream_next_token(s);
1154 if (!tok)
1155 return 1;
1156 if (tok->type != '^') {
1157 isl_stream_push_token(s, tok);
1158 return 1;
1159 }
1160 isl_token_free(tok);
1161 tok = isl_stream_next_token(s);
1162 if (!tok || tok->type != ISL_TOKEN_VALUE) {
1163 isl_stream_error(s, tok, "expecting exponent");
1164 if (tok)
1165 isl_stream_push_token(s, tok);
1166 return 1;
1167 }
1168 pow = isl_int_get_si(tok->u.v);
1169 isl_token_free(tok);
1170 return pow;
1171 }
1172
1173 static __isl_give isl_div *read_div(struct isl_stream *s,
1174 __isl_keep isl_basic_map *bmap, struct vars *v)
1175 {
1176 unsigned total = isl_basic_map_total_dim(bmap);
1177 int k;
1178
1179 bmap = isl_basic_map_copy(bmap);
1180 bmap = isl_basic_map_cow(bmap);
1181 bmap = isl_basic_map_extend_dim(bmap, isl_dim_copy(bmap->dim),
1182 1, 0, 2);
1183
1184 if ((k = isl_basic_map_alloc_div(bmap)) < 0)
1185 goto error;
1186 isl_seq_clr(bmap->div[k], 1 + 1 + total);
1187 bmap = add_div_definition(s, v, bmap, k);
1188
1189 return isl_basic_map_div(bmap, k);
1190 error:
1191 isl_basic_map_free(bmap);
1192 return NULL;
1193 }
1194
1195 static __isl_give isl_qpolynomial *read_term(struct isl_stream *s,
1196 __isl_keep isl_basic_map *bmap, struct vars *v);
1197
1198 static __isl_give isl_qpolynomial *read_factor(struct isl_stream *s,
1199 __isl_keep isl_basic_map *bmap, struct vars *v)
1200 {
1201 struct isl_qpolynomial *qp;
1202 struct isl_token *tok;
1203
1204 tok = isl_stream_next_token(s);
1205 if (!tok) {
1206 isl_stream_error(s, NULL, "unexpected EOF");
1207 return NULL;
1208 }
1209 if (tok->type == '(') {
1210 isl_token_free(tok);
1211 qp = read_term(s, bmap, v);
1212 if (!qp)
1213 return NULL;
1214 if (isl_stream_eat(s, ')'))
1215 goto error;
1216 } else if (tok->type == ISL_TOKEN_VALUE) {
1217 struct isl_token *tok2;
1218 tok2 = isl_stream_next_token(s);
1219 if (tok2 && tok2->type == '/') {
1220 isl_token_free(tok2);
1221 tok2 = isl_stream_next_token(s);
1222 if (!tok2 || tok2->type != ISL_TOKEN_VALUE) {
1223 isl_stream_error(s, tok2, "expected denominator");
1224 isl_token_free(tok);
1225 isl_token_free(tok2);
1226 return NULL;
1227 }
1228 qp = isl_qpolynomial_rat_cst(isl_basic_map_get_dim(bmap),
1229 tok->u.v, tok2->u.v);
1230 isl_token_free(tok2);
1231 } else {
1232 isl_stream_push_token(s, tok2);
1233 qp = isl_qpolynomial_cst(isl_basic_map_get_dim(bmap),
1234 tok->u.v);
1235 }
1236 isl_token_free(tok);
1237 } else if (tok->type == ISL_TOKEN_INFTY) {
1238 isl_token_free(tok);
1239 qp = isl_qpolynomial_infty(isl_basic_map_get_dim(bmap));
1240 } else if (tok->type == ISL_TOKEN_NAN) {
1241 isl_token_free(tok);
1242 qp = isl_qpolynomial_nan(isl_basic_map_get_dim(bmap));
1243 } else if (tok->type == ISL_TOKEN_IDENT) {
1244 int n = v->n;
1245 int pos = vars_pos(v, tok->u.s, -1);
1246 int pow;
1247 if (pos < 0) {
1248 isl_token_free(tok);
1249 return NULL;
1250 }
1251 if (pos >= n) {
1252 isl_stream_error(s, tok, "unknown identifier");
1253 isl_token_free(tok);
1254 return NULL;
1255 }
1256 isl_token_free(tok);
1257 pow = optional_power(s);
1258 qp = isl_qpolynomial_pow(isl_basic_map_get_dim(bmap), pos, pow);
1259 } else if (tok->type == '[') {
1260 isl_div *div;
1261 int pow;
1262
1263 isl_stream_push_token(s, tok);
1264 div = read_div(s, bmap, v);
1265 pow = optional_power(s);
1266 qp = isl_qpolynomial_div_pow(div, pow);
1267 } else if (tok->type == '-') {
1268 struct isl_qpolynomial *qp2;
1269
1270 isl_token_free(tok);
1271 qp = isl_qpolynomial_cst(isl_basic_map_get_dim(bmap),
1272 s->ctx->negone);
1273 qp2 = read_factor(s, bmap, v);
1274 qp = isl_qpolynomial_mul(qp, qp2);
1275 } else {
1276 isl_stream_error(s, tok, "unexpected isl_token");
1277 isl_stream_push_token(s, tok);
1278 return NULL;
1279 }
1280
1281 if (isl_stream_eat_if_available(s, '*') ||
1282 isl_stream_next_token_is(s, ISL_TOKEN_IDENT)) {
1283 struct isl_qpolynomial *qp2;
1284
1285 qp2 = read_factor(s, bmap, v);
1286 qp = isl_qpolynomial_mul(qp, qp2);
1287 }
1288
1289 return qp;
1290 error:
1291 isl_qpolynomial_free(qp);
1292 return NULL;
1293 }
1294
1295 static __isl_give isl_qpolynomial *read_term(struct isl_stream *s,
1296 __isl_keep isl_basic_map *bmap, struct vars *v)
1297 {
1298 struct isl_token *tok;
1299 struct isl_qpolynomial *qp;
1300
1301 qp = read_factor(s, bmap, v);
1302
1303 for (;;) {
1304 tok = isl_stream_next_token(s);
1305 if (!tok)
1306 return qp;
1307
1308 if (tok->type == '+') {
1309 struct isl_qpolynomial *qp2;
1310
1311 isl_token_free(tok);
1312 qp2 = read_factor(s, bmap, v);
1313 qp = isl_qpolynomial_add(qp, qp2);
1314 } else if (tok->type == '-') {
1315 struct isl_qpolynomial *qp2;
1316
1317 isl_token_free(tok);
1318 qp2 = read_factor(s, bmap, v);
1319 qp = isl_qpolynomial_sub(qp, qp2);
1320 } else if (tok->type == ISL_TOKEN_VALUE &&
1321 isl_int_is_neg(tok->u.v)) {
1322 struct isl_qpolynomial *qp2;
1323
1324 isl_stream_push_token(s, tok);
1325 qp2 = read_factor(s, bmap, v);
1326 qp = isl_qpolynomial_add(qp, qp2);
1327 } else {
1328 isl_stream_push_token(s, tok);
1329 break;
1330 }
1331 }
1332
1333 return qp;
1334 }
1335
1336 static __isl_give isl_map *read_optional_disjuncts(struct isl_stream *s,
1337 __isl_take isl_basic_map *bmap, struct vars *v)
1338 {
1339 struct isl_token *tok;
1340 struct isl_map *map;
1341
1342 tok = isl_stream_next_token(s);
1343 if (!tok) {
1344 isl_stream_error(s, NULL, "unexpected EOF");
1345 goto error;
1346 }
1347 map = isl_map_from_basic_map(isl_basic_map_copy(bmap));
1348 if (tok->type == ':') {
1349 isl_token_free(tok);
1350 map = isl_map_intersect(map,
1351 read_disjuncts(s, v, isl_basic_map_get_dim(bmap)));
1352 } else
1353 isl_stream_push_token(s, tok);
1354
1355 isl_basic_map_free(bmap);
1356
1357 return map;
1358 error:
1359 isl_basic_map_free(bmap);
1360 return NULL;
1361 }
1362
1363 static struct isl_obj obj_read_poly(struct isl_stream *s,
1364 __isl_take isl_basic_map *bmap, struct vars *v, int n)
1365 {
1366 struct isl_obj obj = { isl_obj_pw_qpolynomial, NULL };
1367 struct isl_pw_qpolynomial *pwqp;
1368 struct isl_qpolynomial *qp;
1369 struct isl_map *map;
1370 struct isl_set *set;
1371
1372 qp = read_term(s, bmap, v);
1373 map = read_optional_disjuncts(s, bmap, v);
1374 set = isl_map_range(map);
1375
1376 pwqp = isl_pw_qpolynomial_alloc(set, qp);
1377
1378 vars_drop(v, v->n - n);
1379
1380 obj.v = pwqp;
1381 return obj;
1382 }
1383
1384 static struct isl_obj obj_read_poly_or_fold(struct isl_stream *s,
1385 __isl_take isl_basic_map *bmap, struct vars *v, int n)
1386 {
1387 struct isl_obj obj = { isl_obj_pw_qpolynomial_fold, NULL };
1388 struct isl_obj obj_p;
1389 isl_qpolynomial *qp;
1390 isl_qpolynomial_fold *fold = NULL;
1391 isl_pw_qpolynomial_fold *pwf;
1392 isl_map *map;
1393 isl_set *set;
1394
1395 if (!isl_stream_eat_if_available(s, ISL_TOKEN_MAX))
1396 return obj_read_poly(s, bmap, v, n);
1397
1398 if (isl_stream_eat(s, '('))
1399 goto error;
1400
1401 qp = read_term(s, bmap, v);
1402 fold = isl_qpolynomial_fold_alloc(isl_fold_max, qp);
1403
1404 while (isl_stream_eat_if_available(s, ',')) {
1405 isl_qpolynomial_fold *fold_i;
1406 qp = read_term(s, bmap, v);
1407 fold_i = isl_qpolynomial_fold_alloc(isl_fold_max, qp);
1408 fold = isl_qpolynomial_fold_fold(fold, fold_i);
1409 }
1410
1411 if (isl_stream_eat(s, ')'))
1412 goto error;
1413
1414 map = read_optional_disjuncts(s, bmap, v);
1415 set = isl_map_range(map);
1416 pwf = isl_pw_qpolynomial_fold_alloc(isl_fold_max, set, fold);
1417
1418 vars_drop(v, v->n - n);
1419
1420 obj.v = pwf;
1421 return obj;
1422 error:
1423 isl_basic_map_free(bmap);
1424 isl_qpolynomial_fold_free(fold);
1425 obj.type = isl_obj_none;
1426 return obj;
1427 }
1428
1429 static __isl_give isl_basic_map *add_equalities(__isl_take isl_basic_map *bmap,
1430 isl_mat *eq)
1431 {
1432 int i, k;
1433 unsigned total = 1 + isl_basic_map_total_dim(bmap);
1434
1435 if (!bmap || !eq)
1436 goto error;
1437
1438 bmap = isl_basic_map_extend_constraints(bmap, eq->n_row, 0);
1439
1440 for (i = 0; i < eq->n_row; ++i) {
1441 k = isl_basic_map_alloc_equality(bmap);
1442 if (k < 0)
1443 goto error;
1444 isl_seq_cpy(bmap->eq[k], eq->row[i], eq->n_col);
1445 isl_seq_clr(bmap->eq[k] + eq->n_col, total - eq->n_col);
1446 }
1447
1448 isl_mat_free(eq);
1449 bmap = isl_basic_map_gauss(bmap, NULL);
1450 bmap = isl_basic_map_finalize(bmap);
1451 return bmap;
1452 error:
1453 isl_mat_free(eq);
1454 isl_basic_map_free(bmap);
1455 return NULL;
1456 }
1457
1458 static struct isl_obj obj_read_body(struct isl_stream *s,
1459 __isl_take isl_dim *dim, struct vars *v)
1460 {
1461 struct isl_map *map = NULL;
1462 struct isl_token *tok;
1463 struct isl_obj obj = { isl_obj_set, NULL };
1464 int n = v->n;
1465 isl_mat *eq = NULL;
1466 isl_basic_map *bmap;
1467
1468 if (!next_is_tuple(s)) {
1469 bmap = isl_basic_map_alloc_dim(dim, 0, 0, 0);
1470 return obj_read_poly_or_fold(s, bmap, v, n);
1471 }
1472
1473 eq = isl_mat_alloc(s->ctx, 0, 1 + n);
1474
1475 dim = read_tuple(s, dim, isl_dim_in, v, &eq);
1476 if (!dim)
1477 goto error;
1478 tok = isl_stream_next_token(s);
1479 if (tok && tok->type == ISL_TOKEN_TO) {
1480 obj.type = isl_obj_map;
1481 isl_token_free(tok);
1482 if (!next_is_tuple(s)) {
1483 bmap = isl_basic_map_alloc_dim(dim, 0, 0, 0);
1484 bmap = add_equalities(bmap, eq);
1485 bmap = isl_basic_map_reverse(bmap);
1486 return obj_read_poly_or_fold(s, bmap, v, n);
1487 }
1488 dim = read_tuple(s, dim, isl_dim_out, v, &eq);
1489 if (!dim)
1490 goto error;
1491 } else {
1492 dim = isl_dim_reverse(dim);
1493 if (tok)
1494 isl_stream_push_token(s, tok);
1495 }
1496
1497 bmap = isl_basic_map_alloc_dim(dim, 0, 0, 0);
1498 bmap = add_equalities(bmap, eq);
1499
1500 map = read_optional_disjuncts(s, bmap, v);
1501
1502 vars_drop(v, v->n - n);
1503
1504 obj.v = map;
1505 return obj;
1506 error:
1507 isl_mat_free(eq);
1508 isl_dim_free(dim);
1509 obj.type = isl_obj_none;
1510 return obj;
1511 }
1512
1513 static struct isl_obj to_union(isl_ctx *ctx, struct isl_obj obj)
1514 {
1515 if (obj.type == isl_obj_map) {
1516 obj.v = isl_union_map_from_map(obj.v);
1517 obj.type = isl_obj_union_map;
1518 } else if (obj.type == isl_obj_set) {
1519 obj.v = isl_union_set_from_set(obj.v);
1520 obj.type = isl_obj_union_set;
1521 } else if (obj.type == isl_obj_pw_qpolynomial) {
1522 obj.v = isl_union_pw_qpolynomial_from_pw_qpolynomial(obj.v);
1523 obj.type = isl_obj_union_pw_qpolynomial;
1524 } else if (obj.type == isl_obj_pw_qpolynomial_fold) {
1525 obj.v = isl_union_pw_qpolynomial_fold_from_pw_qpolynomial_fold(obj.v);
1526 obj.type = isl_obj_union_pw_qpolynomial_fold;
1527 } else
1528 isl_assert(ctx, 0, goto error);
1529 return obj;
1530 error:
1531 obj.type->free(obj.v);
1532 obj.type = isl_obj_none;
1533 return obj;
1534 }
1535
1536 static struct isl_obj obj_add(struct isl_ctx *ctx,
1537 struct isl_obj obj1, struct isl_obj obj2)
1538 {
1539 if (obj1.type == isl_obj_set && obj2.type == isl_obj_union_set)
1540 obj1 = to_union(ctx, obj1);
1541 if (obj1.type == isl_obj_union_set && obj2.type == isl_obj_set)
1542 obj2 = to_union(ctx, obj2);
1543 if (obj1.type == isl_obj_map && obj2.type == isl_obj_union_map)
1544 obj1 = to_union(ctx, obj1);
1545 if (obj1.type == isl_obj_union_map && obj2.type == isl_obj_map)
1546 obj2 = to_union(ctx, obj2);
1547 if (obj1.type == isl_obj_pw_qpolynomial &&
1548 obj2.type == isl_obj_union_pw_qpolynomial)
1549 obj1 = to_union(ctx, obj1);
1550 if (obj1.type == isl_obj_union_pw_qpolynomial &&
1551 obj2.type == isl_obj_pw_qpolynomial)
1552 obj2 = to_union(ctx, obj2);
1553 if (obj1.type == isl_obj_pw_qpolynomial_fold &&
1554 obj2.type == isl_obj_union_pw_qpolynomial_fold)
1555 obj1 = to_union(ctx, obj1);
1556 if (obj1.type == isl_obj_union_pw_qpolynomial_fold &&
1557 obj2.type == isl_obj_pw_qpolynomial_fold)
1558 obj2 = to_union(ctx, obj2);
1559 isl_assert(ctx, obj1.type == obj2.type, goto error);
1560 if (obj1.type == isl_obj_map && !isl_map_has_equal_dim(obj1.v, obj2.v)) {
1561 obj1 = to_union(ctx, obj1);
1562 obj2 = to_union(ctx, obj2);
1563 }
1564 if (obj1.type == isl_obj_set && !isl_set_has_equal_dim(obj1.v, obj2.v)) {
1565 obj1 = to_union(ctx, obj1);
1566 obj2 = to_union(ctx, obj2);
1567 }
1568 if (obj1.type == isl_obj_pw_qpolynomial &&
1569 !isl_pw_qpolynomial_has_equal_dim(obj1.v, obj2.v)) {
1570 obj1 = to_union(ctx, obj1);
1571 obj2 = to_union(ctx, obj2);
1572 }
1573 if (obj1.type == isl_obj_pw_qpolynomial_fold &&
1574 !isl_pw_qpolynomial_fold_has_equal_dim(obj1.v, obj2.v)) {
1575 obj1 = to_union(ctx, obj1);
1576 obj2 = to_union(ctx, obj2);
1577 }
1578 obj1.v = obj1.type->add(obj1.v, obj2.v);
1579 return obj1;
1580 error:
1581 obj1.type->free(obj1.v);
1582 obj2.type->free(obj2.v);
1583 obj1.type = isl_obj_none;
1584 obj1.v = NULL;
1585 return obj1;
1586 }
1587
1588 static struct isl_obj obj_read(struct isl_stream *s, int nparam)
1589 {
1590 isl_dim *dim = NULL;
1591 struct isl_token *tok;
1592 struct vars *v = NULL;
1593 struct isl_obj obj = { isl_obj_set, NULL };
1594
1595 tok = isl_stream_next_token(s);
1596 if (!tok) {
1597 isl_stream_error(s, NULL, "unexpected EOF");
1598 goto error;
1599 }
1600 if (tok->type == ISL_TOKEN_VALUE) {
1601 struct isl_map *map;
1602 isl_stream_push_token(s, tok);
1603 map = map_read_polylib(s, nparam);
1604 if (!map)
1605 goto error;
1606 if (isl_map_dim(map, isl_dim_in) > 0)
1607 obj.type = isl_obj_map;
1608 obj.v = map;
1609 return obj;
1610 }
1611 v = vars_new(s->ctx);
1612 if (!v) {
1613 isl_stream_push_token(s, tok);
1614 goto error;
1615 }
1616 dim = isl_dim_alloc(s->ctx, 0, 0, 0);
1617 if (tok->type == '[') {
1618 isl_stream_push_token(s, tok);
1619 dim = read_tuple(s, dim, isl_dim_param, v, NULL);
1620 if (!dim)
1621 goto error;
1622 if (nparam >= 0)
1623 isl_assert(s->ctx, nparam == v->n, goto error);
1624 tok = isl_stream_next_token(s);
1625 if (!tok || tok->type != ISL_TOKEN_TO) {
1626 isl_stream_error(s, tok, "expecting '->'");
1627 if (tok)
1628 isl_stream_push_token(s, tok);
1629 goto error;
1630 }
1631 isl_token_free(tok);
1632 tok = isl_stream_next_token(s);
1633 } else if (nparam > 0)
1634 dim = isl_dim_add(dim, isl_dim_param, nparam);
1635 if (!tok || tok->type != '{') {
1636 isl_stream_error(s, tok, "expecting '{'");
1637 if (tok)
1638 isl_stream_push_token(s, tok);
1639 goto error;
1640 }
1641 isl_token_free(tok);
1642
1643 tok = isl_stream_next_token(s);
1644 if (!tok)
1645 ;
1646 else if (tok->type == ISL_TOKEN_IDENT && !strcmp(tok->u.s, "Sym")) {
1647 isl_token_free(tok);
1648 if (isl_stream_eat(s, '='))
1649 goto error;
1650 dim = read_tuple(s, dim, isl_dim_param, v, NULL);
1651 if (!dim)
1652 goto error;
1653 if (nparam >= 0)
1654 isl_assert(s->ctx, nparam == v->n, goto error);
1655 } else if (tok->type == '}') {
1656 obj.type = isl_obj_union_set;
1657 obj.v = isl_union_set_empty(isl_dim_copy(dim));
1658 isl_token_free(tok);
1659 goto done;
1660 } else
1661 isl_stream_push_token(s, tok);
1662
1663 for (;;) {
1664 struct isl_obj o;
1665 tok = NULL;
1666 o = obj_read_body(s, isl_dim_copy(dim), v);
1667 if (o.type == isl_obj_none)
1668 break;
1669 if (!obj.v)
1670 obj = o;
1671 else {
1672 obj = obj_add(s->ctx, obj, o);
1673 if (obj.type == isl_obj_none)
1674 break;
1675 }
1676 tok = isl_stream_next_token(s);
1677 if (!tok || tok->type != ';')
1678 break;
1679 isl_token_free(tok);
1680 }
1681
1682 if (tok && tok->type == '}') {
1683 isl_token_free(tok);
1684 } else {
1685 isl_stream_error(s, tok, "unexpected isl_token");
1686 if (tok)
1687 isl_token_free(tok);
1688 goto error;
1689 }
1690 done:
1691 vars_free(v);
1692 isl_dim_free(dim);
1693
1694 return obj;
1695 error:
1696 isl_dim_free(dim);
1697 obj.type->free(obj.v);
1698 if (v)
1699 vars_free(v);
1700 obj.v = NULL;
1701 return obj;
1702 }
1703
1704 struct isl_obj isl_stream_read_obj(struct isl_stream *s)
1705 {
1706 return obj_read(s, -1);
1707 }
1708
1709 __isl_give isl_map *isl_stream_read_map(struct isl_stream *s, int nparam)
1710 {
1711 struct isl_obj obj;
1712 struct isl_map *map;
1713
1714 obj = obj_read(s, nparam);
1715 if (obj.v)
1716 isl_assert(s->ctx, obj.type == isl_obj_map ||
1717 obj.type == isl_obj_set, goto error);
1718
1719 return obj.v;
1720 error:
1721 obj.type->free(obj.v);
1722 return NULL;
1723 }
1724
1725 __isl_give isl_set *isl_stream_read_set(struct isl_stream *s, int nparam)
1726 {
1727 struct isl_obj obj;
1728 struct isl_set *set;
1729
1730 obj = obj_read(s, nparam);
1731 if (obj.v)
1732 isl_assert(s->ctx, obj.type == isl_obj_set, goto error);
1733
1734 return obj.v;
1735 error:
1736 obj.type->free(obj.v);
1737 return NULL;
1738 }
1739
1740 static struct isl_basic_map *basic_map_read(struct isl_stream *s, int nparam)
1741 {
1742 struct isl_obj obj;
1743 struct isl_map *map;
1744 struct isl_basic_map *bmap;
1745
1746 obj = obj_read(s, nparam);
1747 map = obj.v;
1748 if (!map)
1749 return NULL;
1750
1751 isl_assert(map->ctx, map->n <= 1, goto error);
1752
1753 if (map->n == 0)
1754 bmap = isl_basic_map_empty_like_map(map);
1755 else
1756 bmap = isl_basic_map_copy(map->p[0]);
1757
1758 isl_map_free(map);
1759
1760 return bmap;
1761 error:
1762 isl_map_free(map);
1763 return NULL;
1764 }
1765
1766 __isl_give isl_basic_map *isl_basic_map_read_from_file(isl_ctx *ctx,
1767 FILE *input, int nparam)
1768 {
1769 struct isl_basic_map *bmap;
1770 struct isl_stream *s = isl_stream_new_file(ctx, input);
1771 if (!s)
1772 return NULL;
1773 bmap = basic_map_read(s, nparam);
1774 isl_stream_free(s);
1775 return bmap;
1776 }
1777
1778 __isl_give isl_basic_set *isl_basic_set_read_from_file(isl_ctx *ctx,
1779 FILE *input, int nparam)
1780 {
1781 struct isl_basic_map *bmap;
1782 bmap = isl_basic_map_read_from_file(ctx, input, nparam);
1783 if (!bmap)
1784 return NULL;
1785 isl_assert(ctx, isl_basic_map_n_in(bmap) == 0, goto error);
1786 return (struct isl_basic_set *)bmap;
1787 error:
1788 isl_basic_map_free(bmap);
1789 return NULL;
1790 }
1791
1792 struct isl_basic_map *isl_basic_map_read_from_str(struct isl_ctx *ctx,
1793 const char *str, int nparam)
1794 {
1795 struct isl_basic_map *bmap;
1796 struct isl_stream *s = isl_stream_new_str(ctx, str);
1797 if (!s)
1798 return NULL;
1799 bmap = basic_map_read(s, nparam);
1800 isl_stream_free(s);
1801 return bmap;
1802 }
1803
1804 struct isl_basic_set *isl_basic_set_read_from_str(struct isl_ctx *ctx,
1805 const char *str, int nparam)
1806 {
1807 struct isl_basic_map *bmap;
1808 bmap = isl_basic_map_read_from_str(ctx, str, nparam);
1809 if (!bmap)
1810 return NULL;
1811 isl_assert(ctx, isl_basic_map_n_in(bmap) == 0, goto error);
1812 return (struct isl_basic_set *)bmap;
1813 error:
1814 isl_basic_map_free(bmap);
1815 return NULL;
1816 }
1817
1818 __isl_give isl_map *isl_map_read_from_file(struct isl_ctx *ctx,
1819 FILE *input, int nparam)
1820 {
1821 struct isl_map *map;
1822 struct isl_stream *s = isl_stream_new_file(ctx, input);
1823 if (!s)
1824 return NULL;
1825 map = isl_stream_read_map(s, nparam);
1826 isl_stream_free(s);
1827 return map;
1828 }
1829
1830 __isl_give isl_map *isl_map_read_from_str(struct isl_ctx *ctx,
1831 const char *str, int nparam)
1832 {
1833 struct isl_map *map;
1834 struct isl_stream *s = isl_stream_new_str(ctx, str);
1835 if (!s)
1836 return NULL;
1837 map = isl_stream_read_map(s, nparam);
1838 isl_stream_free(s);
1839 return map;
1840 }
1841
1842 __isl_give isl_set *isl_set_read_from_file(struct isl_ctx *ctx,
1843 FILE *input, int nparam)
1844 {
1845 struct isl_map *map;
1846 map = isl_map_read_from_file(ctx, input, nparam);
1847 if (!map)
1848 return NULL;
1849 isl_assert(ctx, isl_map_n_in(map) == 0, goto error);
1850 return (struct isl_set *)map;
1851 error:
1852 isl_map_free(map);
1853 return NULL;
1854 }
1855
1856 struct isl_set *isl_set_read_from_str(struct isl_ctx *ctx,
1857 const char *str, int nparam)
1858 {
1859 struct isl_map *map;
1860 map = isl_map_read_from_str(ctx, str, nparam);
1861 if (!map)
1862 return NULL;
1863 isl_assert(ctx, isl_map_n_in(map) == 0, goto error);
1864 return (struct isl_set *)map;
1865 error:
1866 isl_map_free(map);
1867 return NULL;
1868 }
1869
1870 static char *next_line(FILE *input, char *line, unsigned len)
1871 {
1872 char *p;
1873
1874 do {
1875 if (!(p = fgets(line, len, input)))
1876 return NULL;
1877 while (isspace(*p) && *p != '\n')
1878 ++p;
1879 } while (*p == '#' || *p == '\n');
1880
1881 return p;
1882 }
1883
1884 static struct isl_vec *isl_vec_read_from_file_polylib(struct isl_ctx *ctx,
1885 FILE *input)
1886 {
1887 struct isl_vec *vec = NULL;
1888 char line[1024];
1889 char val[1024];
1890 char *p;
1891 unsigned size;
1892 int j;
1893 int n;
1894 int offset;
1895
1896 isl_assert(ctx, next_line(input, line, sizeof(line)), return NULL);
1897 isl_assert(ctx, sscanf(line, "%u", &size) == 1, return NULL);
1898
1899 vec = isl_vec_alloc(ctx, size);
1900
1901 p = next_line(input, line, sizeof(line));
1902 isl_assert(ctx, p, goto error);
1903
1904 for (j = 0; j < size; ++j) {
1905 n = sscanf(p, "%s%n", val, &offset);
1906 isl_assert(ctx, n != 0, goto error);
1907 isl_int_read(vec->el[j], val);
1908 p += offset;
1909 }
1910
1911 return vec;
1912 error:
1913 isl_vec_free(vec);
1914 return NULL;
1915 }
1916
1917 struct isl_vec *isl_vec_read_from_file(struct isl_ctx *ctx,
1918 FILE *input, unsigned input_format)
1919 {
1920 if (input_format == ISL_FORMAT_POLYLIB)
1921 return isl_vec_read_from_file_polylib(ctx, input);
1922 else
1923 isl_assert(ctx, 0, return NULL);
1924 }
1925
1926 __isl_give isl_pw_qpolynomial *isl_stream_read_pw_qpolynomial(
1927 struct isl_stream *s)
1928 {
1929 struct isl_obj obj;
1930 struct isl_pw_qpolynomial *pwqp;
1931
1932 obj = obj_read(s, -1);
1933 if (obj.v)
1934 isl_assert(s->ctx, obj.type == isl_obj_pw_qpolynomial,
1935 goto error);
1936
1937 return obj.v;
1938 error:
1939 obj.type->free(obj.v);
1940 return NULL;
1941 }
1942
1943 __isl_give isl_pw_qpolynomial *isl_pw_qpolynomial_read_from_str(isl_ctx *ctx,
1944 const char *str)
1945 {
1946 isl_pw_qpolynomial *pwqp;
1947 struct isl_stream *s = isl_stream_new_str(ctx, str);
1948 if (!s)
1949 return NULL;
1950 pwqp = isl_stream_read_pw_qpolynomial(s);
1951 isl_stream_free(s);
1952 return pwqp;
1953 }
Integer set library