2 * Copyright 2008-2009 Katholieke Universiteit Leuven
3 * Copyright 2010 INRIA Saclay
5 * Use of this software is governed by the GNU LGPLv2.1 license
7 * Written by Sven Verdoolaege, K.U.Leuven, Departement
8 * Computerwetenschappen, Celestijnenlaan 200A, B-3001 Leuven, Belgium
9 * and INRIA Saclay - Ile-de-France, Parc Club Orsay Universite,
10 * ZAC des vignes, 4 rue Jacques Monod, 91893 Orsay, France
20 #include "isl_stream.h"
21 #include "isl_map_private.h"
23 #include "isl_polynomial_private.h"
24 #include <isl_union_map.h>
25 #include <isl_mat_private.h>
31 struct variable *next;
40 static struct vars *vars_new(struct isl_ctx *ctx)
43 v = isl_alloc_type(ctx, struct vars);
52 static void variable_free(struct variable *var)
55 struct variable *next = var->next;
56 isl_vec_free(var->def);
63 static void vars_free(struct vars *v)
71 static void vars_drop(struct vars *v, int n)
82 struct variable *next = var->next;
83 isl_vec_free(var->def);
91 static struct variable *variable_new(struct vars *v, const char *name, int len,
95 var = isl_alloc_type(v->ctx, struct variable);
98 var->name = strdup(name);
99 var->name[len] = '\0';
109 static int vars_pos(struct vars *v, const char *s, int len)
116 for (q = v->v; q; q = q->next) {
117 if (strncmp(q->name, s, len) == 0 && q->name[len] == '\0')
124 v->v = variable_new(v, s, len, v->n);
132 static int vars_add_anon(struct vars *v)
134 v->v = variable_new(v, "", 0, v->n);
143 static __isl_give isl_dim *set_name(__isl_take isl_dim *dim,
144 enum isl_dim_type type, unsigned pos, char *name)
153 prime = strchr(name, '\'');
156 dim = isl_dim_set_name(dim, type, pos, name);
163 static int accept_cst_factor(struct isl_stream *s, isl_int *f)
165 struct isl_token *tok;
167 tok = isl_stream_next_token(s);
168 if (!tok || tok->type != ISL_TOKEN_VALUE) {
169 isl_stream_error(s, tok, "expecting constant value");
173 isl_int_mul(*f, *f, tok->u.v);
177 if (isl_stream_eat_if_available(s, '*'))
178 return accept_cst_factor(s, f);
186 static struct isl_vec *accept_affine(struct isl_stream *s, struct vars *v);
188 static __isl_give isl_vec *accept_affine_factor(struct isl_stream *s,
191 struct isl_token *tok = NULL;
194 tok = isl_stream_next_token(s);
196 isl_stream_error(s, NULL, "unexpected EOF");
199 if (tok->type == ISL_TOKEN_IDENT) {
201 int pos = vars_pos(v, tok->u.s, -1);
205 isl_stream_error(s, tok, "unknown identifier");
209 aff = isl_vec_alloc(v->ctx, 1 + v->n);
212 isl_seq_clr(aff->el, aff->size);
213 isl_int_set_si(aff->el[1 + pos], 1);
215 } else if (tok->type == ISL_TOKEN_VALUE) {
216 if (isl_stream_eat_if_available(s, '*')) {
217 aff = accept_affine_factor(s, v);
218 aff = isl_vec_scale(aff, tok->u.v);
220 aff = isl_vec_alloc(v->ctx, 1 + v->n);
223 isl_seq_clr(aff->el, aff->size);
224 isl_int_set(aff->el[0], tok->u.v);
227 } else if (tok->type == '(') {
230 aff = accept_affine(s, v);
233 if (isl_stream_eat(s, ')'))
236 isl_stream_error(s, tok, "expecting factor");
239 if (isl_stream_eat_if_available(s, '*')) {
242 isl_int_set_si(f, 1);
243 if (accept_cst_factor(s, &f) < 0) {
247 aff = isl_vec_scale(aff, f);
258 static struct isl_vec *accept_affine(struct isl_stream *s, struct vars *v)
260 struct isl_token *tok = NULL;
264 aff = isl_vec_alloc(v->ctx, 1 + v->n);
267 isl_seq_clr(aff->el, aff->size);
270 tok = isl_stream_next_token(s);
272 isl_stream_error(s, NULL, "unexpected EOF");
275 if (tok->type == '-') {
280 if (tok->type == '(' || tok->type == ISL_TOKEN_IDENT) {
282 isl_stream_push_token(s, tok);
284 aff2 = accept_affine_factor(s, v);
286 aff2 = isl_vec_scale(aff2, s->ctx->negone);
287 aff = isl_vec_add(aff, aff2);
291 } else if (tok->type == ISL_TOKEN_VALUE) {
293 isl_int_neg(tok->u.v, tok->u.v);
294 if (isl_stream_eat_if_available(s, '*') ||
295 isl_stream_next_token_is(s, ISL_TOKEN_IDENT)) {
297 aff2 = accept_affine_factor(s, v);
298 aff2 = isl_vec_scale(aff2, tok->u.v);
299 aff = isl_vec_add(aff, aff2);
303 isl_int_add(aff->el[0], aff->el[0], tok->u.v);
309 tok = isl_stream_next_token(s);
310 if (tok && tok->type == '-') {
313 } else if (tok && tok->type == '+') {
316 } else if (tok && tok->type == ISL_TOKEN_VALUE &&
317 isl_int_is_neg(tok->u.v)) {
318 isl_stream_push_token(s, tok);
321 isl_stream_push_token(s, tok);
333 static __isl_give isl_mat *read_var_def(struct isl_stream *s,
334 __isl_take isl_mat *eq, enum isl_dim_type type, struct vars *v)
338 vec = accept_affine(s, v);
341 if (vars_add_anon(v) < 0)
343 eq = isl_mat_add_rows(eq, 1);
345 isl_seq_cpy(eq->row[eq->n_row - 1], vec->el, vec->size);
346 isl_int_set_si(eq->row[eq->n_row - 1][vec->size], -1);
356 static __isl_give isl_dim *read_var_list(struct isl_stream *s,
357 __isl_take isl_dim *dim, enum isl_dim_type type, struct vars *v,
358 __isl_keep isl_mat **eq)
361 struct isl_token *tok;
363 while ((tok = isl_stream_next_token(s)) != NULL) {
366 if (tok->type == ISL_TOKEN_IDENT) {
368 int p = vars_pos(v, tok->u.s, -1);
375 dim = isl_dim_add(dim, type, 1);
377 *eq = isl_mat_add_zero_cols(*eq, 1);
378 dim = set_name(dim, type, i, v->v->name);
380 } else if (tok->type == ISL_TOKEN_IDENT ||
381 tok->type == ISL_TOKEN_VALUE ||
383 if (type == isl_dim_param) {
384 isl_stream_error(s, tok,
385 "expecting unique identifier");
388 isl_stream_push_token(s, tok);
390 dim = isl_dim_add(dim, type, 1);
391 *eq = isl_mat_add_zero_cols(*eq, 1);
392 *eq = read_var_def(s, *eq, type, v);
396 tok = isl_stream_next_token(s);
397 if (!tok || tok->type != ',')
404 isl_stream_push_token(s, tok);
413 static __isl_give isl_mat *accept_affine_list(struct isl_stream *s,
418 struct isl_token *tok = NULL;
420 vec = accept_affine(s, v);
421 mat = isl_mat_from_row_vec(vec);
426 tok = isl_stream_next_token(s);
428 isl_stream_error(s, NULL, "unexpected EOF");
431 if (tok->type != ',') {
432 isl_stream_push_token(s, tok);
437 vec = accept_affine(s, v);
438 mat = isl_mat_vec_concat(mat, vec);
449 static int read_div_definition(struct isl_stream *s, struct vars *v)
451 struct isl_token *tok;
455 v->v->def = isl_vec_alloc(s->ctx, 2 + v->n);
459 if (isl_stream_eat(s, '['))
462 tok = isl_stream_next_token(s);
465 if (tok->type == '(') {
469 isl_stream_push_token(s, tok);
471 aff = accept_affine(s, v);
475 isl_seq_cpy(v->v->def->el + 1, aff->el, aff->size);
479 if (seen_paren && isl_stream_eat(s, ')'))
481 if (isl_stream_eat(s, '/'))
484 tok = isl_stream_next_token(s);
487 if (tok->type != ISL_TOKEN_VALUE) {
488 isl_stream_error(s, tok, "expected denominator");
489 isl_stream_push_token(s, tok);
492 isl_int_set(v->v->def->el[0], tok->u.v);
495 if (isl_stream_eat(s, ']'))
501 static struct isl_basic_map *add_div_definition(struct isl_stream *s,
502 struct vars *v, struct isl_basic_map *bmap, int k)
504 if (read_div_definition(s, v) < 0)
507 isl_seq_cpy(bmap->div[k], v->v->def->el, 2 + v->n);
509 if (isl_basic_map_add_div_constraints(bmap, k) < 0)
514 isl_basic_map_free(bmap);
518 static struct isl_basic_map *read_defined_var_list(struct isl_stream *s,
519 struct vars *v, struct isl_basic_map *bmap)
521 struct isl_token *tok;
523 while ((tok = isl_stream_next_token(s)) != NULL) {
527 unsigned total = isl_basic_map_total_dim(bmap);
529 if (tok->type != ISL_TOKEN_IDENT)
532 p = vars_pos(v, tok->u.s, -1);
536 isl_stream_error(s, tok, "expecting unique identifier");
540 bmap = isl_basic_map_cow(bmap);
541 bmap = isl_basic_map_extend_dim(bmap, isl_dim_copy(bmap->dim),
544 if ((k = isl_basic_map_alloc_div(bmap)) < 0)
546 isl_seq_clr(bmap->div[k], 1 + 1 + total);
549 tok = isl_stream_next_token(s);
550 if (tok && tok->type == '=') {
552 bmap = add_div_definition(s, v, bmap, k);
553 tok = isl_stream_next_token(s);
556 if (!tok || tok->type != ',')
562 isl_stream_push_token(s, tok);
567 isl_basic_map_free(bmap);
571 static int next_is_tuple(struct isl_stream *s)
573 struct isl_token *tok;
576 tok = isl_stream_next_token(s);
579 if (tok->type == '[') {
580 isl_stream_push_token(s, tok);
583 if (tok->type != ISL_TOKEN_IDENT) {
584 isl_stream_push_token(s, tok);
588 is_tuple = isl_stream_next_token_is(s, '[');
590 isl_stream_push_token(s, tok);
595 static __isl_give isl_dim *read_tuple(struct isl_stream *s,
596 __isl_take isl_dim *dim, enum isl_dim_type type, struct vars *v,
597 __isl_keep isl_mat **eq);
599 static __isl_give isl_dim *read_nested_tuple(struct isl_stream *s,
600 __isl_take isl_dim *dim, struct vars *v, __isl_keep isl_mat **eq)
602 dim = read_tuple(s, dim, isl_dim_in, v, eq);
603 if (isl_stream_eat(s, ISL_TOKEN_TO))
605 dim = read_tuple(s, dim, isl_dim_out, v, eq);
606 dim = isl_dim_wrap(dim);
613 static __isl_give isl_dim *read_tuple(struct isl_stream *s,
614 __isl_take isl_dim *dim, enum isl_dim_type type, struct vars *v,
615 __isl_keep isl_mat **eq)
617 struct isl_token *tok;
620 tok = isl_stream_next_token(s);
621 if (tok && tok->type == ISL_TOKEN_IDENT) {
622 name = strdup(tok->u.s);
626 tok = isl_stream_next_token(s);
628 if (!tok || tok->type != '[') {
629 isl_stream_error(s, tok, "expecting '['");
633 if (type != isl_dim_param && next_is_tuple(s)) {
634 isl_dim *nested = isl_dim_copy(dim);
635 nested = isl_dim_drop(nested, isl_dim_in, 0,
636 isl_dim_size(nested, isl_dim_in));
637 nested = isl_dim_drop(nested, isl_dim_out, 0,
638 isl_dim_size(nested, isl_dim_out));
639 nested = read_nested_tuple(s, nested, v, eq);
640 if (type == isl_dim_in) {
642 dim = isl_dim_reverse(nested);
644 dim = isl_dim_join(dim, nested);
647 dim = read_var_list(s, dim, type, v, eq);
648 tok = isl_stream_next_token(s);
649 if (!tok || tok->type != ']') {
650 isl_stream_error(s, tok, "expecting ']'");
656 dim = isl_dim_set_tuple_name(dim, type, name);
668 static struct isl_basic_map *add_constraints(struct isl_stream *s,
669 struct vars *v, struct isl_basic_map *bmap);
671 static struct isl_basic_map *add_exists(struct isl_stream *s,
672 struct vars *v, struct isl_basic_map *bmap)
674 struct isl_token *tok;
681 tok = isl_stream_next_token(s);
684 if (tok->type == '(') {
688 isl_stream_push_token(s, tok);
690 bmap = read_defined_var_list(s, v, bmap);
694 if (isl_stream_eat(s, ':'))
696 bmap = add_constraints(s, v, bmap);
697 if (seen_paren && isl_stream_eat(s, ')'))
701 isl_basic_map_free(bmap);
705 static __isl_give isl_basic_map *construct_constraint(
706 __isl_take isl_basic_map *bmap, enum isl_token_type type,
707 isl_int *left, isl_int *right)
715 len = 1 + isl_basic_map_total_dim(bmap);
718 k = isl_basic_map_alloc_inequality(bmap);
721 if (type == ISL_TOKEN_LE)
722 isl_seq_combine(bmap->ineq[k], ctx->negone, left,
725 else if (type == ISL_TOKEN_GE)
726 isl_seq_combine(bmap->ineq[k], ctx->one, left,
729 else if (type == ISL_TOKEN_LT) {
730 isl_seq_combine(bmap->ineq[k], ctx->negone, left,
733 isl_int_sub_ui(bmap->ineq[k][0], bmap->ineq[k][0], 1);
734 } else if (type == ISL_TOKEN_GT) {
735 isl_seq_combine(bmap->ineq[k], ctx->one, left,
738 isl_int_sub_ui(bmap->ineq[k][0], bmap->ineq[k][0], 1);
740 isl_seq_combine(bmap->ineq[k], ctx->one, left,
743 isl_basic_map_inequality_to_equality(bmap, k);
748 isl_basic_map_free(bmap);
752 static int is_comparator(struct isl_token *tok)
769 static struct isl_basic_map *add_constraint(struct isl_stream *s,
770 struct vars *v, struct isl_basic_map *bmap)
773 unsigned total = isl_basic_map_total_dim(bmap);
774 struct isl_token *tok = NULL;
775 struct isl_mat *aff1 = NULL, *aff2 = NULL;
777 tok = isl_stream_next_token(s);
780 if (tok->type == ISL_TOKEN_EXISTS) {
782 return add_exists(s, v, bmap);
784 isl_stream_push_token(s, tok);
787 bmap = isl_basic_map_cow(bmap);
789 aff1 = accept_affine_list(s, v);
792 tok = isl_stream_next_token(s);
793 if (!is_comparator(tok)) {
794 isl_stream_error(s, tok, "missing operator");
796 isl_stream_push_token(s, tok);
800 isl_assert(aff1->ctx, aff1->n_col == 1 + total, goto error);
802 aff2 = accept_affine_list(s, v);
805 isl_assert(aff2->ctx, aff2->n_col == 1 + total, goto error);
807 bmap = isl_basic_map_extend_constraints(bmap, 0,
808 aff1->n_row * aff2->n_row);
809 for (i = 0; i < aff1->n_row; ++i)
810 for (j = 0; j < aff2->n_row; ++j)
811 bmap = construct_constraint(bmap, tok->type,
812 aff1->row[i], aff2->row[j]);
817 tok = isl_stream_next_token(s);
818 if (!is_comparator(tok)) {
820 isl_stream_push_token(s, tok);
832 isl_basic_map_free(bmap);
836 static struct isl_basic_map *add_constraints(struct isl_stream *s,
837 struct vars *v, struct isl_basic_map *bmap)
839 struct isl_token *tok;
842 bmap = add_constraint(s, v, bmap);
845 tok = isl_stream_next_token(s);
847 isl_stream_error(s, NULL, "unexpected EOF");
850 if (tok->type != ISL_TOKEN_AND)
854 isl_stream_push_token(s, tok);
860 isl_basic_map_free(bmap);
864 static struct isl_basic_map *read_disjunct(struct isl_stream *s,
865 struct vars *v, __isl_take isl_dim *dim)
868 struct isl_token *tok;
869 struct isl_basic_map *bmap;
871 bmap = isl_basic_map_alloc_dim(dim, 0, 0, 0);
875 tok = isl_stream_next_token(s);
878 if (tok->type == '(') {
882 isl_stream_push_token(s, tok);
884 bmap = add_constraints(s, v, bmap);
885 bmap = isl_basic_map_simplify(bmap);
886 bmap = isl_basic_map_finalize(bmap);
888 if (seen_paren && isl_stream_eat(s, ')'))
893 isl_basic_map_free(bmap);
897 static struct isl_map *read_disjuncts(struct isl_stream *s,
898 struct vars *v, __isl_take isl_dim *dim)
900 struct isl_token *tok;
903 tok = isl_stream_next_token(s);
905 isl_stream_error(s, NULL, "unexpected EOF");
908 if (tok->type == '}') {
909 isl_stream_push_token(s, tok);
910 return isl_map_universe(dim);
912 isl_stream_push_token(s, tok);
914 map = isl_map_empty(isl_dim_copy(dim));
916 struct isl_basic_map *bmap;
919 bmap = read_disjunct(s, v, isl_dim_copy(dim));
920 map = isl_map_union(map, isl_map_from_basic_map(bmap));
922 vars_drop(v, v->n - n);
924 tok = isl_stream_next_token(s);
925 if (!tok || tok->type != ISL_TOKEN_OR)
930 isl_stream_push_token(s, tok);
939 static int polylib_pos_to_isl_pos(__isl_keep isl_basic_map *bmap, int pos)
941 if (pos < isl_basic_map_dim(bmap, isl_dim_out))
942 return 1 + isl_basic_map_dim(bmap, isl_dim_param) +
943 isl_basic_map_dim(bmap, isl_dim_in) + pos;
944 pos -= isl_basic_map_dim(bmap, isl_dim_out);
946 if (pos < isl_basic_map_dim(bmap, isl_dim_in))
947 return 1 + isl_basic_map_dim(bmap, isl_dim_param) + pos;
948 pos -= isl_basic_map_dim(bmap, isl_dim_in);
950 if (pos < isl_basic_map_dim(bmap, isl_dim_div))
951 return 1 + isl_basic_map_dim(bmap, isl_dim_param) +
952 isl_basic_map_dim(bmap, isl_dim_in) +
953 isl_basic_map_dim(bmap, isl_dim_out) + pos;
954 pos -= isl_basic_map_dim(bmap, isl_dim_div);
956 if (pos < isl_basic_map_dim(bmap, isl_dim_param))
962 static __isl_give isl_basic_map *basic_map_read_polylib_constraint(
963 struct isl_stream *s, __isl_take isl_basic_map *bmap)
966 struct isl_token *tok;
976 nparam = isl_basic_map_dim(bmap, isl_dim_param);
977 dim = isl_basic_map_dim(bmap, isl_dim_out);
979 tok = isl_stream_next_token(s);
980 if (!tok || tok->type != ISL_TOKEN_VALUE) {
981 isl_stream_error(s, tok, "expecting coefficient");
983 isl_stream_push_token(s, tok);
986 if (!tok->on_new_line) {
987 isl_stream_error(s, tok, "coefficient should appear on new line");
988 isl_stream_push_token(s, tok);
992 type = isl_int_get_si(tok->u.v);
995 isl_assert(s->ctx, type == 0 || type == 1, goto error);
997 k = isl_basic_map_alloc_equality(bmap);
1000 k = isl_basic_map_alloc_inequality(bmap);
1006 for (j = 0; j < 1 + isl_basic_map_total_dim(bmap); ++j) {
1008 tok = isl_stream_next_token(s);
1009 if (!tok || tok->type != ISL_TOKEN_VALUE) {
1010 isl_stream_error(s, tok, "expecting coefficient");
1012 isl_stream_push_token(s, tok);
1015 if (tok->on_new_line) {
1016 isl_stream_error(s, tok,
1017 "coefficient should not appear on new line");
1018 isl_stream_push_token(s, tok);
1021 pos = polylib_pos_to_isl_pos(bmap, j);
1022 isl_int_set(c[pos], tok->u.v);
1023 isl_token_free(tok);
1028 isl_basic_map_free(bmap);
1032 static __isl_give isl_basic_map *basic_map_read_polylib(struct isl_stream *s,
1036 struct isl_token *tok;
1037 struct isl_token *tok2;
1040 unsigned in = 0, out, local = 0;
1041 struct isl_basic_map *bmap = NULL;
1046 tok = isl_stream_next_token(s);
1048 isl_stream_error(s, NULL, "unexpected EOF");
1051 tok2 = isl_stream_next_token(s);
1053 isl_token_free(tok);
1054 isl_stream_error(s, NULL, "unexpected EOF");
1057 n_row = isl_int_get_si(tok->u.v);
1058 n_col = isl_int_get_si(tok2->u.v);
1059 on_new_line = tok2->on_new_line;
1060 isl_token_free(tok2);
1061 isl_token_free(tok);
1062 isl_assert(s->ctx, !on_new_line, return NULL);
1063 isl_assert(s->ctx, n_row >= 0, return NULL);
1064 isl_assert(s->ctx, n_col >= 2 + nparam, return NULL);
1065 tok = isl_stream_next_token_on_same_line(s);
1067 if (tok->type != ISL_TOKEN_VALUE) {
1068 isl_stream_error(s, tok,
1069 "expecting number of output dimensions");
1070 isl_stream_push_token(s, tok);
1073 out = isl_int_get_si(tok->u.v);
1074 isl_token_free(tok);
1076 tok = isl_stream_next_token_on_same_line(s);
1077 if (!tok || tok->type != ISL_TOKEN_VALUE) {
1078 isl_stream_error(s, tok,
1079 "expecting number of input dimensions");
1081 isl_stream_push_token(s, tok);
1084 in = isl_int_get_si(tok->u.v);
1085 isl_token_free(tok);
1087 tok = isl_stream_next_token_on_same_line(s);
1088 if (!tok || tok->type != ISL_TOKEN_VALUE) {
1089 isl_stream_error(s, tok,
1090 "expecting number of existentials");
1092 isl_stream_push_token(s, tok);
1095 local = isl_int_get_si(tok->u.v);
1096 isl_token_free(tok);
1098 tok = isl_stream_next_token_on_same_line(s);
1099 if (!tok || tok->type != ISL_TOKEN_VALUE) {
1100 isl_stream_error(s, tok,
1101 "expecting number of parameters");
1103 isl_stream_push_token(s, tok);
1106 nparam = isl_int_get_si(tok->u.v);
1107 isl_token_free(tok);
1108 if (n_col != 1 + out + in + local + nparam + 1) {
1109 isl_stream_error(s, NULL,
1110 "dimensions don't match");
1114 out = n_col - 2 - nparam;
1115 bmap = isl_basic_map_alloc(s->ctx, nparam, in, out, local, n_row, n_row);
1119 for (i = 0; i < local; ++i) {
1120 int k = isl_basic_map_alloc_div(bmap);
1125 for (i = 0; i < n_row; ++i)
1126 bmap = basic_map_read_polylib_constraint(s, bmap);
1128 tok = isl_stream_next_token_on_same_line(s);
1130 isl_stream_error(s, tok, "unexpected extra token on line");
1131 isl_stream_push_token(s, tok);
1135 bmap = isl_basic_map_simplify(bmap);
1136 bmap = isl_basic_map_finalize(bmap);
1139 isl_basic_map_free(bmap);
1143 static struct isl_map *map_read_polylib(struct isl_stream *s, int nparam)
1145 struct isl_token *tok;
1146 struct isl_token *tok2;
1148 struct isl_map *map;
1150 tok = isl_stream_next_token(s);
1152 isl_stream_error(s, NULL, "unexpected EOF");
1155 tok2 = isl_stream_next_token_on_same_line(s);
1157 isl_stream_push_token(s, tok2);
1158 isl_stream_push_token(s, tok);
1159 return isl_map_from_basic_map(basic_map_read_polylib(s, nparam));
1161 n = isl_int_get_si(tok->u.v);
1162 isl_token_free(tok);
1164 isl_assert(s->ctx, n >= 1, return NULL);
1166 map = isl_map_from_basic_map(basic_map_read_polylib(s, nparam));
1168 for (i = 1; i < n; ++i)
1169 map = isl_map_union(map,
1170 isl_map_from_basic_map(basic_map_read_polylib(s, nparam)));
1175 static int optional_power(struct isl_stream *s)
1178 struct isl_token *tok;
1180 tok = isl_stream_next_token(s);
1183 if (tok->type != '^') {
1184 isl_stream_push_token(s, tok);
1187 isl_token_free(tok);
1188 tok = isl_stream_next_token(s);
1189 if (!tok || tok->type != ISL_TOKEN_VALUE) {
1190 isl_stream_error(s, tok, "expecting exponent");
1192 isl_stream_push_token(s, tok);
1195 pow = isl_int_get_si(tok->u.v);
1196 isl_token_free(tok);
1200 static __isl_give isl_div *read_div(struct isl_stream *s,
1201 __isl_keep isl_basic_map *bmap, struct vars *v)
1203 unsigned total = isl_basic_map_total_dim(bmap);
1206 bmap = isl_basic_map_copy(bmap);
1207 bmap = isl_basic_map_cow(bmap);
1208 bmap = isl_basic_map_extend_dim(bmap, isl_dim_copy(bmap->dim),
1211 if ((k = isl_basic_map_alloc_div(bmap)) < 0)
1213 isl_seq_clr(bmap->div[k], 1 + 1 + total);
1215 if (vars_add_anon(v) < 0)
1217 bmap = add_div_definition(s, v, bmap, k);
1220 return isl_basic_map_div(bmap, k);
1222 isl_basic_map_free(bmap);
1226 static __isl_give isl_qpolynomial *read_term(struct isl_stream *s,
1227 __isl_keep isl_basic_map *bmap, struct vars *v);
1229 static __isl_give isl_qpolynomial *read_factor(struct isl_stream *s,
1230 __isl_keep isl_basic_map *bmap, struct vars *v)
1232 struct isl_qpolynomial *qp;
1233 struct isl_token *tok;
1235 tok = isl_stream_next_token(s);
1237 isl_stream_error(s, NULL, "unexpected EOF");
1240 if (tok->type == '(') {
1241 isl_token_free(tok);
1242 qp = read_term(s, bmap, v);
1245 if (isl_stream_eat(s, ')'))
1247 } else if (tok->type == ISL_TOKEN_VALUE) {
1248 struct isl_token *tok2;
1249 tok2 = isl_stream_next_token(s);
1250 if (tok2 && tok2->type == '/') {
1251 isl_token_free(tok2);
1252 tok2 = isl_stream_next_token(s);
1253 if (!tok2 || tok2->type != ISL_TOKEN_VALUE) {
1254 isl_stream_error(s, tok2, "expected denominator");
1255 isl_token_free(tok);
1256 isl_token_free(tok2);
1259 qp = isl_qpolynomial_rat_cst(isl_basic_map_get_dim(bmap),
1260 tok->u.v, tok2->u.v);
1261 isl_token_free(tok2);
1263 isl_stream_push_token(s, tok2);
1264 qp = isl_qpolynomial_cst(isl_basic_map_get_dim(bmap),
1267 isl_token_free(tok);
1268 } else if (tok->type == ISL_TOKEN_INFTY) {
1269 isl_token_free(tok);
1270 qp = isl_qpolynomial_infty(isl_basic_map_get_dim(bmap));
1271 } else if (tok->type == ISL_TOKEN_NAN) {
1272 isl_token_free(tok);
1273 qp = isl_qpolynomial_nan(isl_basic_map_get_dim(bmap));
1274 } else if (tok->type == ISL_TOKEN_IDENT) {
1276 int pos = vars_pos(v, tok->u.s, -1);
1279 isl_token_free(tok);
1283 isl_stream_error(s, tok, "unknown identifier");
1284 isl_token_free(tok);
1287 isl_token_free(tok);
1288 pow = optional_power(s);
1289 qp = isl_qpolynomial_pow(isl_basic_map_get_dim(bmap), pos, pow);
1290 } else if (tok->type == '[') {
1294 isl_stream_push_token(s, tok);
1295 div = read_div(s, bmap, v);
1296 pow = optional_power(s);
1297 qp = isl_qpolynomial_div_pow(div, pow);
1298 } else if (tok->type == '-') {
1299 struct isl_qpolynomial *qp2;
1301 isl_token_free(tok);
1302 qp = isl_qpolynomial_cst(isl_basic_map_get_dim(bmap),
1304 qp2 = read_factor(s, bmap, v);
1305 qp = isl_qpolynomial_mul(qp, qp2);
1307 isl_stream_error(s, tok, "unexpected isl_token");
1308 isl_stream_push_token(s, tok);
1312 if (isl_stream_eat_if_available(s, '*') ||
1313 isl_stream_next_token_is(s, ISL_TOKEN_IDENT)) {
1314 struct isl_qpolynomial *qp2;
1316 qp2 = read_factor(s, bmap, v);
1317 qp = isl_qpolynomial_mul(qp, qp2);
1322 isl_qpolynomial_free(qp);
1326 static __isl_give isl_qpolynomial *read_term(struct isl_stream *s,
1327 __isl_keep isl_basic_map *bmap, struct vars *v)
1329 struct isl_token *tok;
1330 struct isl_qpolynomial *qp;
1332 qp = read_factor(s, bmap, v);
1335 tok = isl_stream_next_token(s);
1339 if (tok->type == '+') {
1340 struct isl_qpolynomial *qp2;
1342 isl_token_free(tok);
1343 qp2 = read_factor(s, bmap, v);
1344 qp = isl_qpolynomial_add(qp, qp2);
1345 } else if (tok->type == '-') {
1346 struct isl_qpolynomial *qp2;
1348 isl_token_free(tok);
1349 qp2 = read_factor(s, bmap, v);
1350 qp = isl_qpolynomial_sub(qp, qp2);
1351 } else if (tok->type == ISL_TOKEN_VALUE &&
1352 isl_int_is_neg(tok->u.v)) {
1353 struct isl_qpolynomial *qp2;
1355 isl_stream_push_token(s, tok);
1356 qp2 = read_factor(s, bmap, v);
1357 qp = isl_qpolynomial_add(qp, qp2);
1359 isl_stream_push_token(s, tok);
1367 static __isl_give isl_map *read_optional_disjuncts(struct isl_stream *s,
1368 __isl_take isl_basic_map *bmap, struct vars *v)
1370 struct isl_token *tok;
1371 struct isl_map *map;
1373 tok = isl_stream_next_token(s);
1375 isl_stream_error(s, NULL, "unexpected EOF");
1378 map = isl_map_from_basic_map(isl_basic_map_copy(bmap));
1379 if (tok->type == ':') {
1380 isl_token_free(tok);
1381 map = isl_map_intersect(map,
1382 read_disjuncts(s, v, isl_basic_map_get_dim(bmap)));
1384 isl_stream_push_token(s, tok);
1386 isl_basic_map_free(bmap);
1390 isl_basic_map_free(bmap);
1394 static struct isl_obj obj_read_poly(struct isl_stream *s,
1395 __isl_take isl_basic_map *bmap, struct vars *v, int n)
1397 struct isl_obj obj = { isl_obj_pw_qpolynomial, NULL };
1398 struct isl_pw_qpolynomial *pwqp;
1399 struct isl_qpolynomial *qp;
1400 struct isl_map *map;
1401 struct isl_set *set;
1403 qp = read_term(s, bmap, v);
1404 map = read_optional_disjuncts(s, bmap, v);
1405 set = isl_map_range(map);
1407 pwqp = isl_pw_qpolynomial_alloc(set, qp);
1409 vars_drop(v, v->n - n);
1415 static struct isl_obj obj_read_poly_or_fold(struct isl_stream *s,
1416 __isl_take isl_basic_map *bmap, struct vars *v, int n)
1418 struct isl_obj obj = { isl_obj_pw_qpolynomial_fold, NULL };
1419 struct isl_obj obj_p;
1420 isl_qpolynomial *qp;
1421 isl_qpolynomial_fold *fold = NULL;
1422 isl_pw_qpolynomial_fold *pwf;
1426 if (!isl_stream_eat_if_available(s, ISL_TOKEN_MAX))
1427 return obj_read_poly(s, bmap, v, n);
1429 if (isl_stream_eat(s, '('))
1432 qp = read_term(s, bmap, v);
1433 fold = isl_qpolynomial_fold_alloc(isl_fold_max, qp);
1435 while (isl_stream_eat_if_available(s, ',')) {
1436 isl_qpolynomial_fold *fold_i;
1437 qp = read_term(s, bmap, v);
1438 fold_i = isl_qpolynomial_fold_alloc(isl_fold_max, qp);
1439 fold = isl_qpolynomial_fold_fold(fold, fold_i);
1442 if (isl_stream_eat(s, ')'))
1445 map = read_optional_disjuncts(s, bmap, v);
1446 set = isl_map_range(map);
1447 pwf = isl_pw_qpolynomial_fold_alloc(isl_fold_max, set, fold);
1449 vars_drop(v, v->n - n);
1454 isl_basic_map_free(bmap);
1455 isl_qpolynomial_fold_free(fold);
1456 obj.type = isl_obj_none;
1460 static __isl_give isl_basic_map *add_equalities(__isl_take isl_basic_map *bmap,
1464 unsigned total = 1 + isl_basic_map_total_dim(bmap);
1469 bmap = isl_basic_map_extend_constraints(bmap, eq->n_row, 0);
1471 for (i = 0; i < eq->n_row; ++i) {
1472 k = isl_basic_map_alloc_equality(bmap);
1475 isl_seq_cpy(bmap->eq[k], eq->row[i], eq->n_col);
1476 isl_seq_clr(bmap->eq[k] + eq->n_col, total - eq->n_col);
1480 bmap = isl_basic_map_gauss(bmap, NULL);
1481 bmap = isl_basic_map_finalize(bmap);
1485 isl_basic_map_free(bmap);
1489 static struct isl_obj obj_read_body(struct isl_stream *s,
1490 __isl_take isl_dim *dim, struct vars *v)
1492 struct isl_map *map = NULL;
1493 struct isl_token *tok;
1494 struct isl_obj obj = { isl_obj_set, NULL };
1497 isl_basic_map *bmap;
1499 if (!next_is_tuple(s)) {
1500 bmap = isl_basic_map_alloc_dim(dim, 0, 0, 0);
1501 return obj_read_poly_or_fold(s, bmap, v, n);
1504 eq = isl_mat_alloc(s->ctx, 0, 1 + n);
1506 dim = read_tuple(s, dim, isl_dim_in, v, &eq);
1509 tok = isl_stream_next_token(s);
1510 if (tok && tok->type == ISL_TOKEN_TO) {
1511 obj.type = isl_obj_map;
1512 isl_token_free(tok);
1513 if (!next_is_tuple(s)) {
1514 bmap = isl_basic_map_alloc_dim(dim, 0, 0, 0);
1515 bmap = add_equalities(bmap, eq);
1516 bmap = isl_basic_map_reverse(bmap);
1517 return obj_read_poly_or_fold(s, bmap, v, n);
1519 dim = read_tuple(s, dim, isl_dim_out, v, &eq);
1523 dim = isl_dim_reverse(dim);
1525 isl_stream_push_token(s, tok);
1528 bmap = isl_basic_map_alloc_dim(dim, 0, 0, 0);
1529 bmap = add_equalities(bmap, eq);
1531 map = read_optional_disjuncts(s, bmap, v);
1533 vars_drop(v, v->n - n);
1540 obj.type = isl_obj_none;
1544 static struct isl_obj to_union(isl_ctx *ctx, struct isl_obj obj)
1546 if (obj.type == isl_obj_map) {
1547 obj.v = isl_union_map_from_map(obj.v);
1548 obj.type = isl_obj_union_map;
1549 } else if (obj.type == isl_obj_set) {
1550 obj.v = isl_union_set_from_set(obj.v);
1551 obj.type = isl_obj_union_set;
1552 } else if (obj.type == isl_obj_pw_qpolynomial) {
1553 obj.v = isl_union_pw_qpolynomial_from_pw_qpolynomial(obj.v);
1554 obj.type = isl_obj_union_pw_qpolynomial;
1555 } else if (obj.type == isl_obj_pw_qpolynomial_fold) {
1556 obj.v = isl_union_pw_qpolynomial_fold_from_pw_qpolynomial_fold(obj.v);
1557 obj.type = isl_obj_union_pw_qpolynomial_fold;
1559 isl_assert(ctx, 0, goto error);
1562 obj.type->free(obj.v);
1563 obj.type = isl_obj_none;
1567 static struct isl_obj obj_add(struct isl_ctx *ctx,
1568 struct isl_obj obj1, struct isl_obj obj2)
1570 if (obj1.type == isl_obj_set && obj2.type == isl_obj_union_set)
1571 obj1 = to_union(ctx, obj1);
1572 if (obj1.type == isl_obj_union_set && obj2.type == isl_obj_set)
1573 obj2 = to_union(ctx, obj2);
1574 if (obj1.type == isl_obj_map && obj2.type == isl_obj_union_map)
1575 obj1 = to_union(ctx, obj1);
1576 if (obj1.type == isl_obj_union_map && obj2.type == isl_obj_map)
1577 obj2 = to_union(ctx, obj2);
1578 if (obj1.type == isl_obj_pw_qpolynomial &&
1579 obj2.type == isl_obj_union_pw_qpolynomial)
1580 obj1 = to_union(ctx, obj1);
1581 if (obj1.type == isl_obj_union_pw_qpolynomial &&
1582 obj2.type == isl_obj_pw_qpolynomial)
1583 obj2 = to_union(ctx, obj2);
1584 if (obj1.type == isl_obj_pw_qpolynomial_fold &&
1585 obj2.type == isl_obj_union_pw_qpolynomial_fold)
1586 obj1 = to_union(ctx, obj1);
1587 if (obj1.type == isl_obj_union_pw_qpolynomial_fold &&
1588 obj2.type == isl_obj_pw_qpolynomial_fold)
1589 obj2 = to_union(ctx, obj2);
1590 isl_assert(ctx, obj1.type == obj2.type, goto error);
1591 if (obj1.type == isl_obj_map && !isl_map_has_equal_dim(obj1.v, obj2.v)) {
1592 obj1 = to_union(ctx, obj1);
1593 obj2 = to_union(ctx, obj2);
1595 if (obj1.type == isl_obj_set && !isl_set_has_equal_dim(obj1.v, obj2.v)) {
1596 obj1 = to_union(ctx, obj1);
1597 obj2 = to_union(ctx, obj2);
1599 if (obj1.type == isl_obj_pw_qpolynomial &&
1600 !isl_pw_qpolynomial_has_equal_dim(obj1.v, obj2.v)) {
1601 obj1 = to_union(ctx, obj1);
1602 obj2 = to_union(ctx, obj2);
1604 if (obj1.type == isl_obj_pw_qpolynomial_fold &&
1605 !isl_pw_qpolynomial_fold_has_equal_dim(obj1.v, obj2.v)) {
1606 obj1 = to_union(ctx, obj1);
1607 obj2 = to_union(ctx, obj2);
1609 obj1.v = obj1.type->add(obj1.v, obj2.v);
1612 obj1.type->free(obj1.v);
1613 obj2.type->free(obj2.v);
1614 obj1.type = isl_obj_none;
1619 static struct isl_obj obj_read(struct isl_stream *s, int nparam)
1621 isl_dim *dim = NULL;
1622 struct isl_token *tok;
1623 struct vars *v = NULL;
1624 struct isl_obj obj = { isl_obj_set, NULL };
1626 tok = isl_stream_next_token(s);
1628 isl_stream_error(s, NULL, "unexpected EOF");
1631 if (tok->type == ISL_TOKEN_VALUE) {
1632 struct isl_map *map;
1633 isl_stream_push_token(s, tok);
1634 map = map_read_polylib(s, nparam);
1637 if (isl_map_dim(map, isl_dim_in) > 0)
1638 obj.type = isl_obj_map;
1642 v = vars_new(s->ctx);
1644 isl_stream_push_token(s, tok);
1647 dim = isl_dim_alloc(s->ctx, 0, 0, 0);
1648 if (tok->type == '[') {
1649 isl_stream_push_token(s, tok);
1650 dim = read_tuple(s, dim, isl_dim_param, v, NULL);
1654 isl_assert(s->ctx, nparam == v->n, goto error);
1655 tok = isl_stream_next_token(s);
1656 if (!tok || tok->type != ISL_TOKEN_TO) {
1657 isl_stream_error(s, tok, "expecting '->'");
1659 isl_stream_push_token(s, tok);
1662 isl_token_free(tok);
1663 tok = isl_stream_next_token(s);
1664 } else if (nparam > 0)
1665 dim = isl_dim_add(dim, isl_dim_param, nparam);
1666 if (!tok || tok->type != '{') {
1667 isl_stream_error(s, tok, "expecting '{'");
1669 isl_stream_push_token(s, tok);
1672 isl_token_free(tok);
1674 tok = isl_stream_next_token(s);
1677 else if (tok->type == ISL_TOKEN_IDENT && !strcmp(tok->u.s, "Sym")) {
1678 isl_token_free(tok);
1679 if (isl_stream_eat(s, '='))
1681 dim = read_tuple(s, dim, isl_dim_param, v, NULL);
1685 isl_assert(s->ctx, nparam == v->n, goto error);
1686 } else if (tok->type == '}') {
1687 obj.type = isl_obj_union_set;
1688 obj.v = isl_union_set_empty(isl_dim_copy(dim));
1689 isl_token_free(tok);
1692 isl_stream_push_token(s, tok);
1697 o = obj_read_body(s, isl_dim_copy(dim), v);
1698 if (o.type == isl_obj_none || !o.v)
1703 obj = obj_add(s->ctx, obj, o);
1704 if (obj.type == isl_obj_none || !obj.v)
1707 tok = isl_stream_next_token(s);
1708 if (!tok || tok->type != ';')
1710 isl_token_free(tok);
1713 if (tok && tok->type == '}') {
1714 isl_token_free(tok);
1716 isl_stream_error(s, tok, "unexpected isl_token");
1718 isl_token_free(tok);
1728 obj.type->free(obj.v);
1735 struct isl_obj isl_stream_read_obj(struct isl_stream *s)
1737 return obj_read(s, -1);
1740 __isl_give isl_map *isl_stream_read_map(struct isl_stream *s, int nparam)
1743 struct isl_map *map;
1745 obj = obj_read(s, nparam);
1747 isl_assert(s->ctx, obj.type == isl_obj_map ||
1748 obj.type == isl_obj_set, goto error);
1752 obj.type->free(obj.v);
1756 __isl_give isl_set *isl_stream_read_set(struct isl_stream *s, int nparam)
1759 struct isl_set *set;
1761 obj = obj_read(s, nparam);
1763 isl_assert(s->ctx, obj.type == isl_obj_set, goto error);
1767 obj.type->free(obj.v);
1771 static struct isl_basic_map *basic_map_read(struct isl_stream *s, int nparam)
1774 struct isl_map *map;
1775 struct isl_basic_map *bmap;
1777 obj = obj_read(s, nparam);
1782 isl_assert(map->ctx, map->n <= 1, goto error);
1785 bmap = isl_basic_map_empty_like_map(map);
1787 bmap = isl_basic_map_copy(map->p[0]);
1797 __isl_give isl_basic_map *isl_basic_map_read_from_file(isl_ctx *ctx,
1798 FILE *input, int nparam)
1800 struct isl_basic_map *bmap;
1801 struct isl_stream *s = isl_stream_new_file(ctx, input);
1804 bmap = basic_map_read(s, nparam);
1809 __isl_give isl_basic_set *isl_basic_set_read_from_file(isl_ctx *ctx,
1810 FILE *input, int nparam)
1812 struct isl_basic_map *bmap;
1813 bmap = isl_basic_map_read_from_file(ctx, input, nparam);
1816 isl_assert(ctx, isl_basic_map_n_in(bmap) == 0, goto error);
1817 return (struct isl_basic_set *)bmap;
1819 isl_basic_map_free(bmap);
1823 struct isl_basic_map *isl_basic_map_read_from_str(struct isl_ctx *ctx,
1824 const char *str, int nparam)
1826 struct isl_basic_map *bmap;
1827 struct isl_stream *s = isl_stream_new_str(ctx, str);
1830 bmap = basic_map_read(s, nparam);
1835 struct isl_basic_set *isl_basic_set_read_from_str(struct isl_ctx *ctx,
1836 const char *str, int nparam)
1838 struct isl_basic_map *bmap;
1839 bmap = isl_basic_map_read_from_str(ctx, str, nparam);
1842 isl_assert(ctx, isl_basic_map_n_in(bmap) == 0, goto error);
1843 return (struct isl_basic_set *)bmap;
1845 isl_basic_map_free(bmap);
1849 __isl_give isl_map *isl_map_read_from_file(struct isl_ctx *ctx,
1850 FILE *input, int nparam)
1852 struct isl_map *map;
1853 struct isl_stream *s = isl_stream_new_file(ctx, input);
1856 map = isl_stream_read_map(s, nparam);
1861 __isl_give isl_map *isl_map_read_from_str(struct isl_ctx *ctx,
1862 const char *str, int nparam)
1864 struct isl_map *map;
1865 struct isl_stream *s = isl_stream_new_str(ctx, str);
1868 map = isl_stream_read_map(s, nparam);
1873 __isl_give isl_set *isl_set_read_from_file(struct isl_ctx *ctx,
1874 FILE *input, int nparam)
1876 struct isl_map *map;
1877 map = isl_map_read_from_file(ctx, input, nparam);
1880 isl_assert(ctx, isl_map_n_in(map) == 0, goto error);
1881 return (struct isl_set *)map;
1887 struct isl_set *isl_set_read_from_str(struct isl_ctx *ctx,
1888 const char *str, int nparam)
1890 struct isl_map *map;
1891 map = isl_map_read_from_str(ctx, str, nparam);
1894 isl_assert(ctx, isl_map_n_in(map) == 0, goto error);
1895 return (struct isl_set *)map;
1901 static char *next_line(FILE *input, char *line, unsigned len)
1906 if (!(p = fgets(line, len, input)))
1908 while (isspace(*p) && *p != '\n')
1910 } while (*p == '#' || *p == '\n');
1915 static struct isl_vec *isl_vec_read_from_file_polylib(struct isl_ctx *ctx,
1918 struct isl_vec *vec = NULL;
1927 isl_assert(ctx, next_line(input, line, sizeof(line)), return NULL);
1928 isl_assert(ctx, sscanf(line, "%u", &size) == 1, return NULL);
1930 vec = isl_vec_alloc(ctx, size);
1932 p = next_line(input, line, sizeof(line));
1933 isl_assert(ctx, p, goto error);
1935 for (j = 0; j < size; ++j) {
1936 n = sscanf(p, "%s%n", val, &offset);
1937 isl_assert(ctx, n != 0, goto error);
1938 isl_int_read(vec->el[j], val);
1948 struct isl_vec *isl_vec_read_from_file(struct isl_ctx *ctx,
1949 FILE *input, unsigned input_format)
1951 if (input_format == ISL_FORMAT_POLYLIB)
1952 return isl_vec_read_from_file_polylib(ctx, input);
1954 isl_assert(ctx, 0, return NULL);
1957 __isl_give isl_pw_qpolynomial *isl_stream_read_pw_qpolynomial(
1958 struct isl_stream *s)
1961 struct isl_pw_qpolynomial *pwqp;
1963 obj = obj_read(s, -1);
1965 isl_assert(s->ctx, obj.type == isl_obj_pw_qpolynomial,
1970 obj.type->free(obj.v);
1974 __isl_give isl_pw_qpolynomial *isl_pw_qpolynomial_read_from_str(isl_ctx *ctx,
1977 isl_pw_qpolynomial *pwqp;
1978 struct isl_stream *s = isl_stream_new_str(ctx, str);
1981 pwqp = isl_stream_read_pw_qpolynomial(s);