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>
29 struct variable *next;
38 static struct vars *vars_new(struct isl_ctx *ctx)
41 v = isl_alloc_type(ctx, struct vars);
50 static void variable_free(struct variable *var)
53 struct variable *next = var->next;
60 static void vars_free(struct vars *v)
68 static void vars_drop(struct vars *v, int n)
79 struct variable *next = var->next;
87 static struct variable *variable_new(struct vars *v, const char *name, int len,
91 var = isl_alloc_type(v->ctx, struct variable);
94 var->name = strdup(name);
95 var->name[len] = '\0';
104 static int vars_pos(struct vars *v, const char *s, int len)
111 for (q = v->v; q; q = q->next) {
112 if (strncmp(q->name, s, len) == 0 && q->name[len] == '\0')
119 v->v = variable_new(v, s, len, v->n);
127 static __isl_give isl_dim *set_name(__isl_take isl_dim *dim,
128 enum isl_dim_type type, unsigned pos, char *name)
137 prime = strchr(name, '\'');
140 dim = isl_dim_set_name(dim, type, pos, name);
147 static int accept_cst_factor(struct isl_stream *s, isl_int *f)
149 struct isl_token *tok;
151 tok = isl_stream_next_token(s);
152 if (!tok || tok->type != ISL_TOKEN_VALUE) {
153 isl_stream_error(s, tok, "expecting constant value");
157 isl_int_mul(*f, *f, tok->u.v);
161 if (isl_stream_eat_if_available(s, '*'))
162 return accept_cst_factor(s, f);
170 static struct isl_vec *accept_affine(struct isl_stream *s, struct vars *v);
172 static __isl_give isl_vec *accept_affine_factor(struct isl_stream *s,
175 struct isl_token *tok = NULL;
178 tok = isl_stream_next_token(s);
180 isl_stream_error(s, NULL, "unexpected EOF");
183 if (tok->type == ISL_TOKEN_IDENT) {
185 int pos = vars_pos(v, tok->u.s, -1);
189 isl_stream_error(s, tok, "unknown identifier");
193 aff = isl_vec_alloc(v->ctx, 1 + v->n);
196 isl_seq_clr(aff->el, aff->size);
197 isl_int_set_si(aff->el[1 + pos], 1);
199 } else if (tok->type == ISL_TOKEN_VALUE) {
200 if (isl_stream_eat_if_available(s, '*')) {
201 aff = accept_affine_factor(s, v);
202 aff = isl_vec_scale(aff, tok->u.v);
204 aff = isl_vec_alloc(v->ctx, 1 + v->n);
207 isl_seq_clr(aff->el, aff->size);
208 isl_int_set(aff->el[0], tok->u.v);
211 } else if (tok->type == '(') {
214 aff = accept_affine(s, v);
217 if (isl_stream_eat(s, ')'))
220 isl_stream_error(s, tok, "expecting factor");
223 if (isl_stream_eat_if_available(s, '*')) {
226 isl_int_set_si(f, 1);
227 if (accept_cst_factor(s, &f) < 0) {
231 aff = isl_vec_scale(aff, f);
242 static struct isl_vec *accept_affine(struct isl_stream *s, struct vars *v)
244 struct isl_token *tok = NULL;
248 aff = isl_vec_alloc(v->ctx, 1 + v->n);
251 isl_seq_clr(aff->el, aff->size);
254 tok = isl_stream_next_token(s);
256 isl_stream_error(s, NULL, "unexpected EOF");
259 if (tok->type == '-') {
264 if (tok->type == '(' || tok->type == ISL_TOKEN_IDENT) {
266 isl_stream_push_token(s, tok);
268 aff2 = accept_affine_factor(s, v);
270 aff2 = isl_vec_scale(aff2, s->ctx->negone);
271 aff = isl_vec_add(aff, aff2);
275 } else if (tok->type == ISL_TOKEN_VALUE) {
277 isl_int_neg(tok->u.v, tok->u.v);
278 if (isl_stream_eat_if_available(s, '*') ||
279 isl_stream_next_token_is(s, ISL_TOKEN_IDENT)) {
281 aff2 = accept_affine_factor(s, v);
282 aff2 = isl_vec_scale(aff2, tok->u.v);
283 aff = isl_vec_add(aff, aff2);
287 isl_int_add(aff->el[0], aff->el[0], tok->u.v);
293 tok = isl_stream_next_token(s);
294 if (tok && tok->type == '-') {
297 } else if (tok && tok->type == '+') {
300 } else if (tok && tok->type == ISL_TOKEN_VALUE &&
301 isl_int_is_neg(tok->u.v)) {
302 isl_stream_push_token(s, tok);
305 isl_stream_push_token(s, tok);
317 static __isl_give isl_mat *read_var_def(struct isl_stream *s,
318 __isl_take isl_mat *eq, enum isl_dim_type type, struct vars *v)
322 vec = accept_affine(s, v);
325 v->v = variable_new(v, "", 0, v->n);
329 eq = isl_mat_add_rows(eq, 1);
331 isl_seq_cpy(eq->row[eq->n_row - 1], vec->el, vec->size);
332 isl_int_set_si(eq->row[eq->n_row - 1][vec->size], -1);
342 static __isl_give isl_dim *read_var_list(struct isl_stream *s,
343 __isl_take isl_dim *dim, enum isl_dim_type type, struct vars *v,
344 __isl_keep isl_mat **eq)
347 struct isl_token *tok;
349 while ((tok = isl_stream_next_token(s)) != NULL) {
352 if (tok->type == ISL_TOKEN_IDENT) {
354 int p = vars_pos(v, tok->u.s, -1);
361 dim = isl_dim_add(dim, type, 1);
363 *eq = isl_mat_add_zero_cols(*eq, 1);
364 dim = set_name(dim, type, i, v->v->name);
366 } else if (tok->type == ISL_TOKEN_IDENT ||
367 tok->type == ISL_TOKEN_VALUE) {
368 if (type == isl_dim_param) {
369 isl_stream_error(s, tok,
370 "expecting unique identifier");
373 isl_stream_push_token(s, tok);
375 dim = isl_dim_add(dim, type, 1);
376 *eq = isl_mat_add_zero_cols(*eq, 1);
377 *eq = read_var_def(s, *eq, type, v);
381 tok = isl_stream_next_token(s);
382 if (!tok || tok->type != ',')
389 isl_stream_push_token(s, tok);
398 static __isl_give isl_mat *accept_affine_list(struct isl_stream *s,
403 struct isl_token *tok = NULL;
405 vec = accept_affine(s, v);
406 mat = isl_mat_from_row_vec(vec);
411 tok = isl_stream_next_token(s);
413 isl_stream_error(s, NULL, "unexpected EOF");
416 if (tok->type != ',') {
417 isl_stream_push_token(s, tok);
422 vec = accept_affine(s, v);
423 mat = isl_mat_vec_concat(mat, vec);
434 static struct isl_basic_map *add_div_definition(struct isl_stream *s,
435 struct vars *v, struct isl_basic_map *bmap, int k)
437 struct isl_token *tok;
441 if (isl_stream_eat(s, '['))
444 tok = isl_stream_next_token(s);
447 if (tok->type == '(') {
451 isl_stream_push_token(s, tok);
453 aff = accept_affine(s, v);
457 isl_seq_cpy(bmap->div[k] + 1, aff->el, aff->size);
461 if (seen_paren && isl_stream_eat(s, ')'))
463 if (isl_stream_eat(s, '/'))
466 tok = isl_stream_next_token(s);
469 if (tok->type != ISL_TOKEN_VALUE) {
470 isl_stream_error(s, tok, "expected denominator");
471 isl_stream_push_token(s, tok);
474 isl_int_set(bmap->div[k][0], tok->u.v);
477 if (isl_stream_eat(s, ']'))
480 if (isl_basic_map_add_div_constraints(bmap, k) < 0)
485 isl_basic_map_free(bmap);
489 static struct isl_basic_map *read_defined_var_list(struct isl_stream *s,
490 struct vars *v, struct isl_basic_map *bmap)
492 struct isl_token *tok;
494 while ((tok = isl_stream_next_token(s)) != NULL) {
498 unsigned total = isl_basic_map_total_dim(bmap);
500 if (tok->type != ISL_TOKEN_IDENT)
503 p = vars_pos(v, tok->u.s, -1);
507 isl_stream_error(s, tok, "expecting unique identifier");
511 bmap = isl_basic_map_cow(bmap);
512 bmap = isl_basic_map_extend_dim(bmap, isl_dim_copy(bmap->dim),
515 if ((k = isl_basic_map_alloc_div(bmap)) < 0)
517 isl_seq_clr(bmap->div[k], 1 + 1 + total);
520 tok = isl_stream_next_token(s);
521 if (tok && tok->type == '=') {
523 bmap = add_div_definition(s, v, bmap, k);
524 tok = isl_stream_next_token(s);
527 if (!tok || tok->type != ',')
533 isl_stream_push_token(s, tok);
538 isl_basic_map_free(bmap);
542 static int next_is_tuple(struct isl_stream *s)
544 struct isl_token *tok;
547 tok = isl_stream_next_token(s);
550 if (tok->type == '[') {
551 isl_stream_push_token(s, tok);
554 if (tok->type != ISL_TOKEN_IDENT) {
555 isl_stream_push_token(s, tok);
559 is_tuple = isl_stream_next_token_is(s, '[');
561 isl_stream_push_token(s, tok);
566 static __isl_give isl_dim *read_tuple(struct isl_stream *s,
567 __isl_take isl_dim *dim, enum isl_dim_type type, struct vars *v,
568 __isl_keep isl_mat **eq);
570 static __isl_give isl_dim *read_nested_tuple(struct isl_stream *s,
571 __isl_take isl_dim *dim, struct vars *v, __isl_keep isl_mat **eq)
573 dim = read_tuple(s, dim, isl_dim_in, v, eq);
574 if (isl_stream_eat(s, ISL_TOKEN_TO))
576 dim = read_tuple(s, dim, isl_dim_out, v, eq);
577 dim = isl_dim_wrap(dim);
584 static __isl_give isl_dim *read_tuple(struct isl_stream *s,
585 __isl_take isl_dim *dim, enum isl_dim_type type, struct vars *v,
586 __isl_keep isl_mat **eq)
588 struct isl_token *tok;
591 tok = isl_stream_next_token(s);
592 if (tok && tok->type == ISL_TOKEN_IDENT) {
593 name = strdup(tok->u.s);
597 tok = isl_stream_next_token(s);
599 if (!tok || tok->type != '[') {
600 isl_stream_error(s, tok, "expecting '['");
604 if (type != isl_dim_param && next_is_tuple(s)) {
605 isl_dim *nested = isl_dim_copy(dim);
606 nested = isl_dim_drop(nested, isl_dim_in, 0,
607 isl_dim_size(nested, isl_dim_in));
608 nested = isl_dim_drop(nested, isl_dim_out, 0,
609 isl_dim_size(nested, isl_dim_out));
610 nested = read_nested_tuple(s, nested, v, eq);
611 if (type == isl_dim_in) {
613 dim = isl_dim_reverse(nested);
615 dim = isl_dim_join(dim, nested);
618 dim = read_var_list(s, dim, type, v, eq);
619 tok = isl_stream_next_token(s);
620 if (!tok || tok->type != ']') {
621 isl_stream_error(s, tok, "expecting ']'");
627 dim = isl_dim_set_tuple_name(dim, type, name);
639 static struct isl_basic_map *add_constraints(struct isl_stream *s,
640 struct vars *v, struct isl_basic_map *bmap);
642 static struct isl_basic_map *add_exists(struct isl_stream *s,
643 struct vars *v, struct isl_basic_map *bmap)
645 struct isl_token *tok;
652 tok = isl_stream_next_token(s);
655 if (tok->type == '(') {
659 isl_stream_push_token(s, tok);
661 bmap = read_defined_var_list(s, v, bmap);
665 if (isl_stream_eat(s, ':'))
667 bmap = add_constraints(s, v, bmap);
668 if (seen_paren && isl_stream_eat(s, ')'))
672 isl_basic_map_free(bmap);
676 static __isl_give isl_basic_map *construct_constraint(
677 __isl_take isl_basic_map *bmap, enum isl_token_type type,
678 isl_int *left, isl_int *right)
686 len = 1 + isl_basic_map_total_dim(bmap);
689 k = isl_basic_map_alloc_inequality(bmap);
692 if (type == ISL_TOKEN_LE)
693 isl_seq_combine(bmap->ineq[k], ctx->negone, left,
696 else if (type == ISL_TOKEN_GE)
697 isl_seq_combine(bmap->ineq[k], ctx->one, left,
700 else if (type == ISL_TOKEN_LT) {
701 isl_seq_combine(bmap->ineq[k], ctx->negone, left,
704 isl_int_sub_ui(bmap->ineq[k][0], bmap->ineq[k][0], 1);
705 } else if (type == ISL_TOKEN_GT) {
706 isl_seq_combine(bmap->ineq[k], ctx->one, left,
709 isl_int_sub_ui(bmap->ineq[k][0], bmap->ineq[k][0], 1);
711 isl_seq_combine(bmap->ineq[k], ctx->one, left,
714 isl_basic_map_inequality_to_equality(bmap, k);
719 isl_basic_map_free(bmap);
723 static int is_comparator(struct isl_token *tok)
740 static struct isl_basic_map *add_constraint(struct isl_stream *s,
741 struct vars *v, struct isl_basic_map *bmap)
744 unsigned total = isl_basic_map_total_dim(bmap);
745 struct isl_token *tok = NULL;
746 struct isl_mat *aff1 = NULL, *aff2 = NULL;
748 tok = isl_stream_next_token(s);
751 if (tok->type == ISL_TOKEN_EXISTS) {
753 return add_exists(s, v, bmap);
755 isl_stream_push_token(s, tok);
758 bmap = isl_basic_map_cow(bmap);
760 aff1 = accept_affine_list(s, v);
763 tok = isl_stream_next_token(s);
764 if (!is_comparator(tok)) {
765 isl_stream_error(s, tok, "missing operator");
767 isl_stream_push_token(s, tok);
771 isl_assert(aff1->ctx, aff1->n_col == 1 + total, goto error);
773 aff2 = accept_affine_list(s, v);
776 isl_assert(aff2->ctx, aff2->n_col == 1 + total, goto error);
778 bmap = isl_basic_map_extend_constraints(bmap, 0,
779 aff1->n_row * aff2->n_row);
780 for (i = 0; i < aff1->n_row; ++i)
781 for (j = 0; j < aff2->n_row; ++j)
782 bmap = construct_constraint(bmap, tok->type,
783 aff1->row[i], aff2->row[j]);
788 tok = isl_stream_next_token(s);
789 if (!is_comparator(tok)) {
791 isl_stream_push_token(s, tok);
803 isl_basic_map_free(bmap);
807 static struct isl_basic_map *add_constraints(struct isl_stream *s,
808 struct vars *v, struct isl_basic_map *bmap)
810 struct isl_token *tok;
813 bmap = add_constraint(s, v, bmap);
816 tok = isl_stream_next_token(s);
818 isl_stream_error(s, NULL, "unexpected EOF");
821 if (tok->type != ISL_TOKEN_AND)
825 isl_stream_push_token(s, tok);
831 isl_basic_map_free(bmap);
835 static struct isl_basic_map *read_disjunct(struct isl_stream *s,
836 struct vars *v, __isl_take isl_dim *dim)
839 struct isl_token *tok;
840 struct isl_basic_map *bmap;
842 bmap = isl_basic_map_alloc_dim(dim, 0, 0, 0);
846 tok = isl_stream_next_token(s);
849 if (tok->type == '(') {
853 isl_stream_push_token(s, tok);
855 bmap = add_constraints(s, v, bmap);
856 bmap = isl_basic_map_simplify(bmap);
857 bmap = isl_basic_map_finalize(bmap);
859 if (seen_paren && isl_stream_eat(s, ')'))
864 isl_basic_map_free(bmap);
868 static struct isl_map *read_disjuncts(struct isl_stream *s,
869 struct vars *v, __isl_take isl_dim *dim)
871 struct isl_token *tok;
874 tok = isl_stream_next_token(s);
876 isl_stream_error(s, NULL, "unexpected EOF");
879 if (tok->type == '}') {
880 isl_stream_push_token(s, tok);
881 return isl_map_universe(dim);
883 isl_stream_push_token(s, tok);
885 map = isl_map_empty(isl_dim_copy(dim));
887 struct isl_basic_map *bmap;
890 bmap = read_disjunct(s, v, isl_dim_copy(dim));
891 map = isl_map_union(map, isl_map_from_basic_map(bmap));
893 vars_drop(v, v->n - n);
895 tok = isl_stream_next_token(s);
896 if (!tok || tok->type != ISL_TOKEN_OR)
901 isl_stream_push_token(s, tok);
910 static int polylib_pos_to_isl_pos(__isl_keep isl_basic_map *bmap, int pos)
912 if (pos < isl_basic_map_dim(bmap, isl_dim_out))
913 return 1 + isl_basic_map_dim(bmap, isl_dim_param) +
914 isl_basic_map_dim(bmap, isl_dim_in) + pos;
915 pos -= isl_basic_map_dim(bmap, isl_dim_out);
917 if (pos < isl_basic_map_dim(bmap, isl_dim_in))
918 return 1 + isl_basic_map_dim(bmap, isl_dim_param) + pos;
919 pos -= isl_basic_map_dim(bmap, isl_dim_in);
921 if (pos < isl_basic_map_dim(bmap, isl_dim_div))
922 return 1 + isl_basic_map_dim(bmap, isl_dim_param) +
923 isl_basic_map_dim(bmap, isl_dim_in) +
924 isl_basic_map_dim(bmap, isl_dim_out) + pos;
925 pos -= isl_basic_map_dim(bmap, isl_dim_div);
927 if (pos < isl_basic_map_dim(bmap, isl_dim_param))
933 static __isl_give isl_basic_map *basic_map_read_polylib_constraint(
934 struct isl_stream *s, __isl_take isl_basic_map *bmap)
937 struct isl_token *tok;
947 nparam = isl_basic_map_dim(bmap, isl_dim_param);
948 dim = isl_basic_map_dim(bmap, isl_dim_out);
950 tok = isl_stream_next_token(s);
951 if (!tok || tok->type != ISL_TOKEN_VALUE) {
952 isl_stream_error(s, tok, "expecting coefficient");
954 isl_stream_push_token(s, tok);
957 if (!tok->on_new_line) {
958 isl_stream_error(s, tok, "coefficient should appear on new line");
959 isl_stream_push_token(s, tok);
963 type = isl_int_get_si(tok->u.v);
966 isl_assert(s->ctx, type == 0 || type == 1, goto error);
968 k = isl_basic_map_alloc_equality(bmap);
971 k = isl_basic_map_alloc_inequality(bmap);
977 for (j = 0; j < 1 + isl_basic_map_total_dim(bmap); ++j) {
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,
988 "coefficient should not appear on new line");
989 isl_stream_push_token(s, tok);
992 pos = polylib_pos_to_isl_pos(bmap, j);
993 isl_int_set(c[pos], tok->u.v);
999 isl_basic_map_free(bmap);
1003 static __isl_give isl_basic_map *basic_map_read_polylib(struct isl_stream *s,
1007 struct isl_token *tok;
1008 struct isl_token *tok2;
1011 unsigned in = 0, out, local = 0;
1012 struct isl_basic_map *bmap = NULL;
1017 tok = isl_stream_next_token(s);
1019 isl_stream_error(s, NULL, "unexpected EOF");
1022 tok2 = isl_stream_next_token(s);
1024 isl_token_free(tok);
1025 isl_stream_error(s, NULL, "unexpected EOF");
1028 n_row = isl_int_get_si(tok->u.v);
1029 n_col = isl_int_get_si(tok2->u.v);
1030 on_new_line = tok2->on_new_line;
1031 isl_token_free(tok2);
1032 isl_token_free(tok);
1033 isl_assert(s->ctx, !on_new_line, return NULL);
1034 isl_assert(s->ctx, n_row >= 0, return NULL);
1035 isl_assert(s->ctx, n_col >= 2 + nparam, return NULL);
1036 tok = isl_stream_next_token_on_same_line(s);
1038 if (tok->type != ISL_TOKEN_VALUE) {
1039 isl_stream_error(s, tok,
1040 "expecting number of output dimensions");
1041 isl_stream_push_token(s, tok);
1044 out = isl_int_get_si(tok->u.v);
1045 isl_token_free(tok);
1047 tok = isl_stream_next_token_on_same_line(s);
1048 if (!tok || tok->type != ISL_TOKEN_VALUE) {
1049 isl_stream_error(s, tok,
1050 "expecting number of input dimensions");
1052 isl_stream_push_token(s, tok);
1055 in = isl_int_get_si(tok->u.v);
1056 isl_token_free(tok);
1058 tok = isl_stream_next_token_on_same_line(s);
1059 if (!tok || tok->type != ISL_TOKEN_VALUE) {
1060 isl_stream_error(s, tok,
1061 "expecting number of existentials");
1063 isl_stream_push_token(s, tok);
1066 local = isl_int_get_si(tok->u.v);
1067 isl_token_free(tok);
1069 tok = isl_stream_next_token_on_same_line(s);
1070 if (!tok || tok->type != ISL_TOKEN_VALUE) {
1071 isl_stream_error(s, tok,
1072 "expecting number of parameters");
1074 isl_stream_push_token(s, tok);
1077 nparam = isl_int_get_si(tok->u.v);
1078 isl_token_free(tok);
1079 if (n_col != 1 + out + in + local + nparam + 1) {
1080 isl_stream_error(s, NULL,
1081 "dimensions don't match");
1085 out = n_col - 2 - nparam;
1086 bmap = isl_basic_map_alloc(s->ctx, nparam, in, out, local, n_row, n_row);
1090 for (i = 0; i < local; ++i) {
1091 int k = isl_basic_map_alloc_div(bmap);
1096 for (i = 0; i < n_row; ++i)
1097 bmap = basic_map_read_polylib_constraint(s, bmap);
1099 tok = isl_stream_next_token_on_same_line(s);
1101 isl_stream_error(s, tok, "unexpected extra token on line");
1102 isl_stream_push_token(s, tok);
1106 bmap = isl_basic_map_simplify(bmap);
1107 bmap = isl_basic_map_finalize(bmap);
1110 isl_basic_map_free(bmap);
1114 static struct isl_map *map_read_polylib(struct isl_stream *s, int nparam)
1116 struct isl_token *tok;
1117 struct isl_token *tok2;
1119 struct isl_map *map;
1121 tok = isl_stream_next_token(s);
1123 isl_stream_error(s, NULL, "unexpected EOF");
1126 tok2 = isl_stream_next_token_on_same_line(s);
1128 isl_stream_push_token(s, tok2);
1129 isl_stream_push_token(s, tok);
1130 return isl_map_from_basic_map(basic_map_read_polylib(s, nparam));
1132 n = isl_int_get_si(tok->u.v);
1133 isl_token_free(tok);
1135 isl_assert(s->ctx, n >= 1, return NULL);
1137 map = isl_map_from_basic_map(basic_map_read_polylib(s, nparam));
1139 for (i = 1; i < n; ++i)
1140 map = isl_map_union(map,
1141 isl_map_from_basic_map(basic_map_read_polylib(s, nparam)));
1146 static int optional_power(struct isl_stream *s)
1149 struct isl_token *tok;
1151 tok = isl_stream_next_token(s);
1154 if (tok->type != '^') {
1155 isl_stream_push_token(s, tok);
1158 isl_token_free(tok);
1159 tok = isl_stream_next_token(s);
1160 if (!tok || tok->type != ISL_TOKEN_VALUE) {
1161 isl_stream_error(s, tok, "expecting exponent");
1163 isl_stream_push_token(s, tok);
1166 pow = isl_int_get_si(tok->u.v);
1167 isl_token_free(tok);
1171 static __isl_give isl_div *read_div(struct isl_stream *s,
1172 __isl_keep isl_basic_map *bmap, struct vars *v)
1174 unsigned total = isl_basic_map_total_dim(bmap);
1177 bmap = isl_basic_map_copy(bmap);
1178 bmap = isl_basic_map_cow(bmap);
1179 bmap = isl_basic_map_extend_dim(bmap, isl_dim_copy(bmap->dim),
1182 if ((k = isl_basic_map_alloc_div(bmap)) < 0)
1184 isl_seq_clr(bmap->div[k], 1 + 1 + total);
1185 bmap = add_div_definition(s, v, bmap, k);
1187 return isl_basic_map_div(bmap, k);
1189 isl_basic_map_free(bmap);
1193 static __isl_give isl_qpolynomial *read_term(struct isl_stream *s,
1194 __isl_keep isl_basic_map *bmap, struct vars *v);
1196 static __isl_give isl_qpolynomial *read_factor(struct isl_stream *s,
1197 __isl_keep isl_basic_map *bmap, struct vars *v)
1199 struct isl_qpolynomial *qp;
1200 struct isl_token *tok;
1202 tok = isl_stream_next_token(s);
1204 isl_stream_error(s, NULL, "unexpected EOF");
1207 if (tok->type == '(') {
1208 isl_token_free(tok);
1209 qp = read_term(s, bmap, v);
1212 if (isl_stream_eat(s, ')'))
1214 } else if (tok->type == ISL_TOKEN_VALUE) {
1215 struct isl_token *tok2;
1216 tok2 = isl_stream_next_token(s);
1217 if (tok2 && tok2->type == '/') {
1218 isl_token_free(tok2);
1219 tok2 = isl_stream_next_token(s);
1220 if (!tok2 || tok2->type != ISL_TOKEN_VALUE) {
1221 isl_stream_error(s, tok2, "expected denominator");
1222 isl_token_free(tok);
1223 isl_token_free(tok2);
1226 qp = isl_qpolynomial_rat_cst(isl_basic_map_get_dim(bmap),
1227 tok->u.v, tok2->u.v);
1228 isl_token_free(tok2);
1230 isl_stream_push_token(s, tok2);
1231 qp = isl_qpolynomial_cst(isl_basic_map_get_dim(bmap),
1234 isl_token_free(tok);
1235 } else if (tok->type == ISL_TOKEN_INFTY) {
1236 isl_token_free(tok);
1237 qp = isl_qpolynomial_infty(isl_basic_map_get_dim(bmap));
1238 } else if (tok->type == ISL_TOKEN_NAN) {
1239 isl_token_free(tok);
1240 qp = isl_qpolynomial_nan(isl_basic_map_get_dim(bmap));
1241 } else if (tok->type == ISL_TOKEN_IDENT) {
1243 int pos = vars_pos(v, tok->u.s, -1);
1245 isl_token_free(tok);
1249 isl_stream_error(s, tok, "unknown identifier");
1252 pow = optional_power(s);
1253 qp = isl_qpolynomial_pow(isl_basic_map_get_dim(bmap), pos, pow);
1254 } else if (tok->type == '[') {
1258 isl_stream_push_token(s, tok);
1259 div = read_div(s, bmap, v);
1260 pow = optional_power(s);
1261 qp = isl_qpolynomial_div_pow(div, pow);
1262 } else if (tok->type == '-') {
1263 struct isl_qpolynomial *qp2;
1265 isl_token_free(tok);
1266 qp = isl_qpolynomial_cst(isl_basic_map_get_dim(bmap),
1268 qp2 = read_factor(s, bmap, v);
1269 qp = isl_qpolynomial_mul(qp, qp2);
1271 isl_stream_error(s, tok, "unexpected isl_token");
1272 isl_stream_push_token(s, tok);
1276 if (isl_stream_eat_if_available(s, '*') ||
1277 isl_stream_next_token_is(s, ISL_TOKEN_IDENT)) {
1278 struct isl_qpolynomial *qp2;
1280 qp2 = read_factor(s, bmap, v);
1281 qp = isl_qpolynomial_mul(qp, qp2);
1286 isl_qpolynomial_free(qp);
1290 static __isl_give isl_qpolynomial *read_term(struct isl_stream *s,
1291 __isl_keep isl_basic_map *bmap, struct vars *v)
1293 struct isl_token *tok;
1294 struct isl_qpolynomial *qp;
1296 qp = read_factor(s, bmap, v);
1299 tok = isl_stream_next_token(s);
1303 if (tok->type == '+') {
1304 struct isl_qpolynomial *qp2;
1306 isl_token_free(tok);
1307 qp2 = read_factor(s, bmap, v);
1308 qp = isl_qpolynomial_add(qp, qp2);
1309 } else if (tok->type == '-') {
1310 struct isl_qpolynomial *qp2;
1312 isl_token_free(tok);
1313 qp2 = read_factor(s, bmap, v);
1314 qp = isl_qpolynomial_sub(qp, qp2);
1315 } else if (tok->type == ISL_TOKEN_VALUE &&
1316 isl_int_is_neg(tok->u.v)) {
1317 struct isl_qpolynomial *qp2;
1319 qp2 = isl_qpolynomial_cst(isl_basic_map_get_dim(bmap),
1321 isl_token_free(tok);
1322 qp = isl_qpolynomial_add(qp, qp2);
1324 isl_stream_push_token(s, tok);
1332 static __isl_give isl_map *read_optional_disjuncts(struct isl_stream *s,
1333 __isl_take isl_basic_map *bmap, struct vars *v)
1335 struct isl_token *tok;
1336 struct isl_map *map;
1338 tok = isl_stream_next_token(s);
1340 isl_stream_error(s, NULL, "unexpected EOF");
1343 map = isl_map_from_basic_map(isl_basic_map_copy(bmap));
1344 if (tok->type == ':') {
1345 isl_token_free(tok);
1346 map = isl_map_intersect(map,
1347 read_disjuncts(s, v, isl_basic_map_get_dim(bmap)));
1349 isl_stream_push_token(s, tok);
1351 isl_basic_map_free(bmap);
1355 isl_basic_map_free(bmap);
1359 static struct isl_obj obj_read_poly(struct isl_stream *s,
1360 __isl_take isl_basic_map *bmap, struct vars *v, int n)
1362 struct isl_obj obj = { isl_obj_pw_qpolynomial, NULL };
1363 struct isl_pw_qpolynomial *pwqp;
1364 struct isl_qpolynomial *qp;
1365 struct isl_map *map;
1366 struct isl_set *set;
1368 bmap = isl_basic_map_reverse(bmap);
1370 qp = read_term(s, bmap, v);
1371 map = read_optional_disjuncts(s, bmap, v);
1372 set = isl_map_range(map);
1374 pwqp = isl_pw_qpolynomial_alloc(set, qp);
1376 vars_drop(v, v->n - n);
1382 static __isl_give isl_basic_map *add_equalities(__isl_take isl_basic_map *bmap,
1386 unsigned total = 1 + isl_basic_map_total_dim(bmap);
1391 bmap = isl_basic_map_extend_constraints(bmap, eq->n_row, 0);
1393 for (i = 0; i < eq->n_row; ++i) {
1394 k = isl_basic_map_alloc_equality(bmap);
1397 isl_seq_cpy(bmap->eq[k], eq->row[i], eq->n_col);
1398 isl_seq_clr(bmap->eq[k] + eq->n_col, total - eq->n_col);
1405 isl_basic_map_free(bmap);
1409 static struct isl_obj obj_read_body(struct isl_stream *s,
1410 __isl_take isl_dim *dim, struct vars *v)
1412 struct isl_map *map = NULL;
1413 struct isl_token *tok;
1414 struct isl_obj obj = { isl_obj_set, NULL };
1417 isl_basic_map *bmap;
1419 if (!next_is_tuple(s)) {
1420 bmap = isl_basic_map_alloc_dim(dim, 0, 0, 0);
1421 return obj_read_poly(s, bmap, v, n);
1424 eq = isl_mat_alloc(s->ctx, 0, 1 + n);
1426 dim = read_tuple(s, dim, isl_dim_in, v, &eq);
1429 tok = isl_stream_next_token(s);
1430 if (tok && tok->type == ISL_TOKEN_TO) {
1431 obj.type = isl_obj_map;
1432 isl_token_free(tok);
1433 if (!next_is_tuple(s)) {
1434 bmap = isl_basic_map_alloc_dim(dim, 0, 0, 0);
1435 bmap = add_equalities(bmap, eq);
1436 return obj_read_poly(s, bmap, v, n);
1438 dim = read_tuple(s, dim, isl_dim_out, v, &eq);
1442 dim = isl_dim_reverse(dim);
1444 isl_stream_push_token(s, tok);
1447 bmap = isl_basic_map_alloc_dim(dim, 0, 0, 0);
1448 bmap = add_equalities(bmap, eq);
1450 map = read_optional_disjuncts(s, bmap, v);
1452 vars_drop(v, v->n - n);
1459 obj.type = isl_obj_none;
1463 static struct isl_obj to_union(isl_ctx *ctx, struct isl_obj obj)
1465 if (obj.type == isl_obj_map) {
1466 obj.v = isl_union_map_from_map(obj.v);
1467 obj.type = isl_obj_union_map;
1468 } else if (obj.type == isl_obj_set) {
1469 obj.v = isl_union_set_from_set(obj.v);
1470 obj.type = isl_obj_union_set;
1471 } else if (obj.type == isl_obj_pw_qpolynomial) {
1472 obj.v = isl_union_pw_qpolynomial_from_pw_qpolynomial(obj.v);
1473 obj.type = isl_obj_union_pw_qpolynomial;
1474 } else if (obj.type == isl_obj_pw_qpolynomial_fold) {
1475 obj.v = isl_union_pw_qpolynomial_fold_from_pw_qpolynomial_fold(obj.v);
1476 obj.type = isl_obj_union_pw_qpolynomial_fold;
1478 isl_assert(ctx, 0, goto error);
1481 obj.type->free(obj.v);
1482 obj.type = isl_obj_none;
1486 static struct isl_obj obj_add(struct isl_ctx *ctx,
1487 struct isl_obj obj1, struct isl_obj obj2)
1489 if (obj1.type == isl_obj_set && obj2.type == isl_obj_union_set)
1490 obj1 = to_union(ctx, obj1);
1491 if (obj1.type == isl_obj_union_set && obj2.type == isl_obj_set)
1492 obj2 = to_union(ctx, obj2);
1493 if (obj1.type == isl_obj_map && obj2.type == isl_obj_union_map)
1494 obj1 = to_union(ctx, obj1);
1495 if (obj1.type == isl_obj_union_map && obj2.type == isl_obj_map)
1496 obj2 = to_union(ctx, obj2);
1497 if (obj1.type == isl_obj_pw_qpolynomial &&
1498 obj2.type == isl_obj_union_pw_qpolynomial)
1499 obj1 = to_union(ctx, obj1);
1500 if (obj1.type == isl_obj_union_pw_qpolynomial &&
1501 obj2.type == isl_obj_pw_qpolynomial)
1502 obj2 = to_union(ctx, obj2);
1503 if (obj1.type == isl_obj_pw_qpolynomial_fold &&
1504 obj2.type == isl_obj_union_pw_qpolynomial_fold)
1505 obj1 = to_union(ctx, obj1);
1506 if (obj1.type == isl_obj_union_pw_qpolynomial_fold &&
1507 obj2.type == isl_obj_pw_qpolynomial_fold)
1508 obj2 = to_union(ctx, obj2);
1509 isl_assert(ctx, obj1.type == obj2.type, goto error);
1510 if (obj1.type == isl_obj_map && !isl_map_has_equal_dim(obj1.v, obj2.v)) {
1511 obj1 = to_union(ctx, obj1);
1512 obj2 = to_union(ctx, obj2);
1514 if (obj1.type == isl_obj_set && !isl_set_has_equal_dim(obj1.v, obj2.v)) {
1515 obj1 = to_union(ctx, obj1);
1516 obj2 = to_union(ctx, obj2);
1518 if (obj1.type == isl_obj_pw_qpolynomial &&
1519 !isl_pw_qpolynomial_has_equal_dim(obj1.v, obj2.v)) {
1520 obj1 = to_union(ctx, obj1);
1521 obj2 = to_union(ctx, obj2);
1523 if (obj1.type == isl_obj_pw_qpolynomial_fold &&
1524 !isl_pw_qpolynomial_fold_has_equal_dim(obj1.v, obj2.v)) {
1525 obj1 = to_union(ctx, obj1);
1526 obj2 = to_union(ctx, obj2);
1528 obj1.v = obj1.type->add(obj1.v, obj2.v);
1531 obj1.type->free(obj1.v);
1532 obj2.type->free(obj2.v);
1533 obj1.type = isl_obj_none;
1538 static struct isl_obj obj_read(struct isl_stream *s, int nparam)
1540 isl_dim *dim = NULL;
1541 struct isl_token *tok;
1542 struct vars *v = NULL;
1543 struct isl_obj obj = { isl_obj_set, NULL };
1545 tok = isl_stream_next_token(s);
1547 isl_stream_error(s, NULL, "unexpected EOF");
1550 if (tok->type == ISL_TOKEN_VALUE) {
1551 struct isl_map *map;
1552 isl_stream_push_token(s, tok);
1553 map = map_read_polylib(s, nparam);
1556 if (isl_map_dim(map, isl_dim_in) > 0)
1557 obj.type = isl_obj_map;
1561 v = vars_new(s->ctx);
1563 isl_stream_push_token(s, tok);
1566 dim = isl_dim_alloc(s->ctx, 0, 0, 0);
1567 if (tok->type == '[') {
1568 isl_stream_push_token(s, tok);
1569 dim = read_tuple(s, dim, isl_dim_param, v, NULL);
1573 isl_assert(s->ctx, nparam == v->n, goto error);
1574 tok = isl_stream_next_token(s);
1575 if (!tok || tok->type != ISL_TOKEN_TO) {
1576 isl_stream_error(s, tok, "expecting '->'");
1578 isl_stream_push_token(s, tok);
1581 isl_token_free(tok);
1582 tok = isl_stream_next_token(s);
1583 } else if (nparam > 0)
1584 dim = isl_dim_add(dim, isl_dim_param, nparam);
1585 if (!tok || tok->type != '{') {
1586 isl_stream_error(s, tok, "expecting '{'");
1588 isl_stream_push_token(s, tok);
1591 isl_token_free(tok);
1593 tok = isl_stream_next_token(s);
1596 else if (tok->type == ISL_TOKEN_IDENT && !strcmp(tok->u.s, "Sym")) {
1597 isl_token_free(tok);
1598 if (isl_stream_eat(s, '='))
1600 dim = read_tuple(s, dim, isl_dim_param, v, NULL);
1604 isl_assert(s->ctx, nparam == v->n, goto error);
1605 } else if (tok->type == '}') {
1606 obj.type = isl_obj_union_set;
1607 obj.v = isl_union_set_empty(isl_dim_copy(dim));
1608 isl_token_free(tok);
1611 isl_stream_push_token(s, tok);
1616 o = obj_read_body(s, isl_dim_copy(dim), v);
1617 if (o.type == isl_obj_none)
1622 obj = obj_add(s->ctx, obj, o);
1623 if (obj.type == isl_obj_none)
1626 tok = isl_stream_next_token(s);
1627 if (!tok || tok->type != ';')
1629 isl_token_free(tok);
1632 if (tok && tok->type == '}') {
1633 isl_token_free(tok);
1635 isl_stream_error(s, tok, "unexpected isl_token");
1637 isl_token_free(tok);
1647 obj.type->free(obj.v);
1654 struct isl_obj isl_stream_read_obj(struct isl_stream *s)
1656 return obj_read(s, -1);
1659 __isl_give isl_map *isl_stream_read_map(struct isl_stream *s, int nparam)
1662 struct isl_map *map;
1664 obj = obj_read(s, nparam);
1666 isl_assert(s->ctx, obj.type == isl_obj_map ||
1667 obj.type == isl_obj_set, goto error);
1671 obj.type->free(obj.v);
1675 __isl_give isl_set *isl_stream_read_set(struct isl_stream *s, int nparam)
1678 struct isl_set *set;
1680 obj = obj_read(s, nparam);
1682 isl_assert(s->ctx, obj.type == isl_obj_set, goto error);
1686 obj.type->free(obj.v);
1690 static struct isl_basic_map *basic_map_read(struct isl_stream *s, int nparam)
1693 struct isl_map *map;
1694 struct isl_basic_map *bmap;
1696 obj = obj_read(s, nparam);
1701 isl_assert(map->ctx, map->n <= 1, goto error);
1704 bmap = isl_basic_map_empty_like_map(map);
1706 bmap = isl_basic_map_copy(map->p[0]);
1716 __isl_give isl_basic_map *isl_basic_map_read_from_file(isl_ctx *ctx,
1717 FILE *input, int nparam)
1719 struct isl_basic_map *bmap;
1720 struct isl_stream *s = isl_stream_new_file(ctx, input);
1723 bmap = basic_map_read(s, nparam);
1728 __isl_give isl_basic_set *isl_basic_set_read_from_file(isl_ctx *ctx,
1729 FILE *input, int nparam)
1731 struct isl_basic_map *bmap;
1732 bmap = isl_basic_map_read_from_file(ctx, input, nparam);
1735 isl_assert(ctx, isl_basic_map_n_in(bmap) == 0, goto error);
1736 return (struct isl_basic_set *)bmap;
1738 isl_basic_map_free(bmap);
1742 struct isl_basic_map *isl_basic_map_read_from_str(struct isl_ctx *ctx,
1743 const char *str, int nparam)
1745 struct isl_basic_map *bmap;
1746 struct isl_stream *s = isl_stream_new_str(ctx, str);
1749 bmap = basic_map_read(s, nparam);
1754 struct isl_basic_set *isl_basic_set_read_from_str(struct isl_ctx *ctx,
1755 const char *str, int nparam)
1757 struct isl_basic_map *bmap;
1758 bmap = isl_basic_map_read_from_str(ctx, str, nparam);
1761 isl_assert(ctx, isl_basic_map_n_in(bmap) == 0, goto error);
1762 return (struct isl_basic_set *)bmap;
1764 isl_basic_map_free(bmap);
1768 __isl_give isl_map *isl_map_read_from_file(struct isl_ctx *ctx,
1769 FILE *input, int nparam)
1771 struct isl_map *map;
1772 struct isl_stream *s = isl_stream_new_file(ctx, input);
1775 map = isl_stream_read_map(s, nparam);
1780 __isl_give isl_map *isl_map_read_from_str(struct isl_ctx *ctx,
1781 const char *str, int nparam)
1783 struct isl_map *map;
1784 struct isl_stream *s = isl_stream_new_str(ctx, str);
1787 map = isl_stream_read_map(s, nparam);
1792 __isl_give isl_set *isl_set_read_from_file(struct isl_ctx *ctx,
1793 FILE *input, int nparam)
1795 struct isl_map *map;
1796 map = isl_map_read_from_file(ctx, input, nparam);
1799 isl_assert(ctx, isl_map_n_in(map) == 0, goto error);
1800 return (struct isl_set *)map;
1806 struct isl_set *isl_set_read_from_str(struct isl_ctx *ctx,
1807 const char *str, int nparam)
1809 struct isl_map *map;
1810 map = isl_map_read_from_str(ctx, str, nparam);
1813 isl_assert(ctx, isl_map_n_in(map) == 0, goto error);
1814 return (struct isl_set *)map;
1820 static char *next_line(FILE *input, char *line, unsigned len)
1825 if (!(p = fgets(line, len, input)))
1827 while (isspace(*p) && *p != '\n')
1829 } while (*p == '#' || *p == '\n');
1834 static struct isl_vec *isl_vec_read_from_file_polylib(struct isl_ctx *ctx,
1837 struct isl_vec *vec = NULL;
1846 isl_assert(ctx, next_line(input, line, sizeof(line)), return NULL);
1847 isl_assert(ctx, sscanf(line, "%u", &size) == 1, return NULL);
1849 vec = isl_vec_alloc(ctx, size);
1851 p = next_line(input, line, sizeof(line));
1852 isl_assert(ctx, p, goto error);
1854 for (j = 0; j < size; ++j) {
1855 n = sscanf(p, "%s%n", val, &offset);
1856 isl_assert(ctx, n != 0, goto error);
1857 isl_int_read(vec->el[j], val);
1867 struct isl_vec *isl_vec_read_from_file(struct isl_ctx *ctx,
1868 FILE *input, unsigned input_format)
1870 if (input_format == ISL_FORMAT_POLYLIB)
1871 return isl_vec_read_from_file_polylib(ctx, input);
1873 isl_assert(ctx, 0, return NULL);
1876 __isl_give isl_pw_qpolynomial *isl_stream_read_pw_qpolynomial(
1877 struct isl_stream *s)
1880 struct isl_pw_qpolynomial *pwqp;
1882 obj = obj_read(s, -1);
1884 isl_assert(s->ctx, obj.type == isl_obj_pw_qpolynomial,
1889 obj.type->free(obj.v);
1893 __isl_give isl_pw_qpolynomial *isl_pw_qpolynomial_read_from_str(isl_ctx *ctx,
1896 isl_pw_qpolynomial *pwqp;
1897 struct isl_stream *s = isl_stream_new_str(ctx, str);
1900 pwqp = isl_stream_read_pw_qpolynomial(s);