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_basic_map *set_name(__isl_take isl_basic_map *bmap,
144 enum isl_dim_type type, unsigned pos, char *name)
153 prime = strchr(name, '\'');
156 bmap = isl_basic_map_set_dim_name(bmap, 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);
187 static int read_div_definition(struct isl_stream *s, struct vars *v);
189 static __isl_give isl_vec *accept_affine_factor(struct isl_stream *s,
192 struct isl_token *tok = NULL;
195 tok = isl_stream_next_token(s);
197 isl_stream_error(s, NULL, "unexpected EOF");
200 if (tok->type == ISL_TOKEN_IDENT) {
202 int pos = vars_pos(v, tok->u.s, -1);
206 isl_stream_error(s, tok, "unknown identifier");
210 aff = isl_vec_alloc(v->ctx, 1 + v->n);
213 isl_seq_clr(aff->el, aff->size);
214 isl_int_set_si(aff->el[1 + pos], 1);
216 } else if (tok->type == ISL_TOKEN_VALUE) {
217 if (isl_stream_eat_if_available(s, '*')) {
218 aff = accept_affine_factor(s, v);
219 aff = isl_vec_scale(aff, tok->u.v);
221 aff = isl_vec_alloc(v->ctx, 1 + v->n);
224 isl_seq_clr(aff->el, aff->size);
225 isl_int_set(aff->el[0], tok->u.v);
228 } else if (tok->type == '(') {
231 aff = accept_affine(s, v);
234 if (isl_stream_eat(s, ')'))
236 } else if (tok->type == '[') {
237 if (vars_add_anon(v) < 0)
239 aff = isl_vec_alloc(v->ctx, 1 + v->n);
242 isl_seq_clr(aff->el, aff->size);
243 isl_int_set_si(aff->el[1 + v->n - 1], 1);
244 isl_stream_push_token(s, tok);
246 if (read_div_definition(s, v) < 0)
249 isl_stream_error(s, tok, "expecting factor");
252 if (isl_stream_eat_if_available(s, '*')) {
255 isl_int_set_si(f, 1);
256 if (accept_cst_factor(s, &f) < 0) {
260 aff = isl_vec_scale(aff, f);
271 static struct isl_vec *accept_affine(struct isl_stream *s, struct vars *v)
273 struct isl_token *tok = NULL;
277 aff = isl_vec_alloc(v->ctx, 1 + v->n);
280 isl_seq_clr(aff->el, aff->size);
283 tok = isl_stream_next_token(s);
285 isl_stream_error(s, NULL, "unexpected EOF");
288 if (tok->type == '-') {
293 if (tok->type == '(' || tok->type == '[' ||
294 tok->type == ISL_TOKEN_IDENT) {
296 isl_stream_push_token(s, tok);
298 aff2 = accept_affine_factor(s, v);
300 aff2 = isl_vec_scale(aff2, s->ctx->negone);
301 aff = isl_vec_zero_extend(aff, 1 + v->n);
302 aff = isl_vec_add(aff, aff2);
306 } else if (tok->type == ISL_TOKEN_VALUE) {
308 isl_int_neg(tok->u.v, tok->u.v);
309 if (isl_stream_eat_if_available(s, '*') ||
310 isl_stream_next_token_is(s, ISL_TOKEN_IDENT)) {
312 aff2 = accept_affine_factor(s, v);
313 aff2 = isl_vec_scale(aff2, tok->u.v);
314 aff = isl_vec_zero_extend(aff, 1 + v->n);
315 aff = isl_vec_add(aff, aff2);
319 isl_int_add(aff->el[0], aff->el[0], tok->u.v);
323 isl_stream_error(s, tok, "unexpected isl_token");
324 isl_stream_push_token(s, tok);
330 tok = isl_stream_next_token(s);
331 if (tok && tok->type == '-') {
334 } else if (tok && tok->type == '+') {
337 } else if (tok && tok->type == ISL_TOKEN_VALUE &&
338 isl_int_is_neg(tok->u.v)) {
339 isl_stream_push_token(s, tok);
342 isl_stream_push_token(s, tok);
354 /* Add any variables in the variable list "v" that are not already in "bmap"
355 * as existentially quantified variables in "bmap".
357 static __isl_give isl_basic_map *add_divs(__isl_take isl_basic_map *bmap,
362 struct variable *var;
364 extra = v->n - isl_basic_map_total_dim(bmap);
369 bmap = isl_basic_map_extend_dim(bmap, isl_basic_map_get_dim(bmap),
370 extra, 0, 2 * extra);
372 for (i = 0; i < extra; ++i)
373 if (isl_basic_map_alloc_div(bmap) < 0)
376 for (i = 0, var = v->v; i < extra; ++i, var = var->next) {
377 int k = bmap->n_div - 1 - i;
379 isl_seq_cpy(bmap->div[k], var->def->el, 2 + var->pos);
380 isl_seq_clr(bmap->div[k] + 2 + var->pos, v->n - var->pos);
382 if (isl_basic_map_add_div_constraints(bmap, k) < 0)
388 isl_basic_map_free(bmap);
392 static __isl_give isl_basic_map *read_var_def(struct isl_stream *s,
393 __isl_take isl_basic_map *bmap, enum isl_dim_type type, struct vars *v)
399 if (vars_add_anon(v) < 0)
403 vec = accept_affine(s, v);
407 bmap = add_divs(bmap, v);
408 bmap = isl_basic_map_extend_constraints(bmap, 1, 0);
409 k = isl_basic_map_alloc_equality(bmap);
411 isl_seq_cpy(bmap->eq[k], vec->el, vec->size);
412 isl_int_set_si(bmap->eq[k][1 + n - 1], -1);
418 vars_drop(v, v->n - n);
422 isl_basic_map_free(bmap);
426 static __isl_give isl_basic_map *read_var_list(struct isl_stream *s,
427 __isl_take isl_basic_map *bmap, enum isl_dim_type type, struct vars *v)
430 struct isl_token *tok;
432 while ((tok = isl_stream_next_token(s)) != NULL) {
435 if (tok->type == ISL_TOKEN_IDENT) {
437 int p = vars_pos(v, tok->u.s, -1);
444 bmap = isl_basic_map_add(bmap, type, 1);
445 bmap = set_name(bmap, type, i, v->v->name);
447 } else if (tok->type == ISL_TOKEN_IDENT ||
448 tok->type == ISL_TOKEN_VALUE ||
451 if (type == isl_dim_param) {
452 isl_stream_error(s, tok,
453 "expecting unique identifier");
456 isl_stream_push_token(s, tok);
458 bmap = isl_basic_map_add(bmap, type, 1);
459 bmap = read_var_def(s, bmap, type, v);
463 tok = isl_stream_next_token(s);
464 if (!tok || tok->type != ',')
471 isl_stream_push_token(s, tok);
476 isl_basic_map_free(bmap);
480 static __isl_give isl_mat *accept_affine_list(struct isl_stream *s,
485 struct isl_token *tok = NULL;
487 vec = accept_affine(s, v);
488 mat = isl_mat_from_row_vec(vec);
493 tok = isl_stream_next_token(s);
495 isl_stream_error(s, NULL, "unexpected EOF");
498 if (tok->type != ',') {
499 isl_stream_push_token(s, tok);
504 vec = accept_affine(s, v);
505 mat = isl_mat_add_zero_cols(mat, 1 + v->n - isl_mat_cols(mat));
506 mat = isl_mat_vec_concat(mat, vec);
517 static int read_div_definition(struct isl_stream *s, struct vars *v)
519 struct isl_token *tok;
522 struct variable *var;
524 if (isl_stream_eat(s, '['))
527 tok = isl_stream_next_token(s);
530 if (tok->type == '(') {
534 isl_stream_push_token(s, tok);
538 aff = accept_affine(s, v);
542 var->def = isl_vec_alloc(s->ctx, 2 + v->n);
548 isl_seq_cpy(var->def->el + 1, aff->el, aff->size);
552 if (seen_paren && isl_stream_eat(s, ')'))
554 if (isl_stream_eat(s, '/'))
557 tok = isl_stream_next_token(s);
560 if (tok->type != ISL_TOKEN_VALUE) {
561 isl_stream_error(s, tok, "expected denominator");
562 isl_stream_push_token(s, tok);
565 isl_int_set(var->def->el[0], tok->u.v);
568 if (isl_stream_eat(s, ']'))
574 static struct isl_basic_map *add_div_definition(struct isl_stream *s,
575 struct vars *v, struct isl_basic_map *bmap, int k)
577 struct variable *var = v->v;
579 if (read_div_definition(s, v) < 0)
582 isl_seq_cpy(bmap->div[k], var->def->el, 2 + v->n);
584 if (isl_basic_map_add_div_constraints(bmap, k) < 0)
589 isl_basic_map_free(bmap);
593 static struct isl_basic_map *read_defined_var_list(struct isl_stream *s,
594 struct vars *v, struct isl_basic_map *bmap)
596 struct isl_token *tok;
598 while ((tok = isl_stream_next_token(s)) != NULL) {
602 unsigned total = isl_basic_map_total_dim(bmap);
604 if (tok->type != ISL_TOKEN_IDENT)
607 p = vars_pos(v, tok->u.s, -1);
611 isl_stream_error(s, tok, "expecting unique identifier");
615 bmap = isl_basic_map_cow(bmap);
616 bmap = isl_basic_map_extend_dim(bmap, isl_dim_copy(bmap->dim),
619 if ((k = isl_basic_map_alloc_div(bmap)) < 0)
621 isl_seq_clr(bmap->div[k], 1 + 1 + total);
624 tok = isl_stream_next_token(s);
625 if (tok && tok->type == '=') {
627 bmap = add_div_definition(s, v, bmap, k);
628 tok = isl_stream_next_token(s);
631 if (!tok || tok->type != ',')
637 isl_stream_push_token(s, tok);
642 isl_basic_map_free(bmap);
646 static int next_is_tuple(struct isl_stream *s)
648 struct isl_token *tok;
651 tok = isl_stream_next_token(s);
654 if (tok->type == '[') {
655 isl_stream_push_token(s, tok);
658 if (tok->type != ISL_TOKEN_IDENT) {
659 isl_stream_push_token(s, tok);
663 is_tuple = isl_stream_next_token_is(s, '[');
665 isl_stream_push_token(s, tok);
670 static __isl_give isl_basic_map *read_tuple(struct isl_stream *s,
671 __isl_take isl_basic_map *bmap, enum isl_dim_type type, struct vars *v);
673 static __isl_give isl_basic_map *read_nested_tuple(struct isl_stream *s,
674 __isl_take isl_basic_map *bmap, struct vars *v)
676 bmap = read_tuple(s, bmap, isl_dim_in, v);
677 if (isl_stream_eat(s, ISL_TOKEN_TO))
679 bmap = read_tuple(s, bmap, isl_dim_out, v);
680 bmap = isl_basic_map_from_range(isl_basic_map_wrap(bmap));
683 isl_basic_map_free(bmap);
687 static __isl_give isl_basic_map *read_tuple(struct isl_stream *s,
688 __isl_take isl_basic_map *bmap, enum isl_dim_type type, struct vars *v)
690 struct isl_token *tok;
693 tok = isl_stream_next_token(s);
694 if (tok && tok->type == ISL_TOKEN_IDENT) {
695 name = strdup(tok->u.s);
699 tok = isl_stream_next_token(s);
701 if (!tok || tok->type != '[') {
702 isl_stream_error(s, tok, "expecting '['");
706 if (type != isl_dim_param && next_is_tuple(s)) {
707 isl_dim *dim = isl_basic_map_get_dim(bmap);
708 isl_basic_map *nested;
709 dim = isl_dim_drop(dim, isl_dim_in, 0,
710 isl_dim_size(dim, isl_dim_in));
711 dim = isl_dim_drop(dim, isl_dim_out, 0,
712 isl_dim_size(dim, isl_dim_out));
713 nested = isl_basic_map_alloc_dim(dim, 0, 0, 0);
714 nested = read_nested_tuple(s, nested, v);
715 if (type == isl_dim_in) {
716 isl_basic_map_free(bmap);
717 bmap = isl_basic_map_reverse(nested);
719 bmap = isl_basic_map_apply_range(bmap, nested);
722 bmap = read_var_list(s, bmap, type, v);
723 tok = isl_stream_next_token(s);
724 if (!tok || tok->type != ']') {
725 isl_stream_error(s, tok, "expecting ']'");
731 bmap = isl_basic_map_set_tuple_name(bmap, type, name);
739 isl_basic_map_free(bmap);
743 static struct isl_basic_map *add_constraints(struct isl_stream *s,
744 struct vars *v, struct isl_basic_map *bmap);
746 static struct isl_basic_map *add_exists(struct isl_stream *s,
747 struct vars *v, struct isl_basic_map *bmap)
749 struct isl_token *tok;
756 tok = isl_stream_next_token(s);
759 if (tok->type == '(') {
763 isl_stream_push_token(s, tok);
765 bmap = read_defined_var_list(s, v, bmap);
769 if (isl_stream_eat(s, ':'))
771 bmap = add_constraints(s, v, bmap);
772 if (seen_paren && isl_stream_eat(s, ')'))
776 isl_basic_map_free(bmap);
780 static __isl_give isl_basic_map *construct_constraint(
781 __isl_take isl_basic_map *bmap, enum isl_token_type type,
782 isl_int *left, isl_int *right)
790 len = 1 + isl_basic_map_total_dim(bmap);
793 k = isl_basic_map_alloc_inequality(bmap);
796 if (type == ISL_TOKEN_LE)
797 isl_seq_combine(bmap->ineq[k], ctx->negone, left,
800 else if (type == ISL_TOKEN_GE)
801 isl_seq_combine(bmap->ineq[k], ctx->one, left,
804 else if (type == ISL_TOKEN_LT) {
805 isl_seq_combine(bmap->ineq[k], ctx->negone, left,
808 isl_int_sub_ui(bmap->ineq[k][0], bmap->ineq[k][0], 1);
809 } else if (type == ISL_TOKEN_GT) {
810 isl_seq_combine(bmap->ineq[k], ctx->one, left,
813 isl_int_sub_ui(bmap->ineq[k][0], bmap->ineq[k][0], 1);
815 isl_seq_combine(bmap->ineq[k], ctx->one, left,
818 isl_basic_map_inequality_to_equality(bmap, k);
823 isl_basic_map_free(bmap);
827 static int is_comparator(struct isl_token *tok)
844 static struct isl_basic_map *add_constraint(struct isl_stream *s,
845 struct vars *v, struct isl_basic_map *bmap)
848 struct isl_token *tok = NULL;
849 struct isl_mat *aff1 = NULL, *aff2 = NULL;
851 tok = isl_stream_next_token(s);
854 if (tok->type == ISL_TOKEN_EXISTS) {
856 return add_exists(s, v, bmap);
858 isl_stream_push_token(s, tok);
861 bmap = isl_basic_map_cow(bmap);
863 aff1 = accept_affine_list(s, v);
866 tok = isl_stream_next_token(s);
867 if (!is_comparator(tok)) {
868 isl_stream_error(s, tok, "missing operator");
870 isl_stream_push_token(s, tok);
875 aff2 = accept_affine_list(s, v);
879 aff1 = isl_mat_add_zero_cols(aff1, aff2->n_col - aff1->n_col);
882 bmap = add_divs(bmap, v);
883 bmap = isl_basic_map_extend_constraints(bmap, 0,
884 aff1->n_row * aff2->n_row);
885 for (i = 0; i < aff1->n_row; ++i)
886 for (j = 0; j < aff2->n_row; ++j)
887 bmap = construct_constraint(bmap, tok->type,
888 aff1->row[i], aff2->row[j]);
893 tok = isl_stream_next_token(s);
894 if (!is_comparator(tok)) {
896 isl_stream_push_token(s, tok);
908 isl_basic_map_free(bmap);
912 static struct isl_basic_map *add_constraints(struct isl_stream *s,
913 struct vars *v, struct isl_basic_map *bmap)
915 struct isl_token *tok;
918 bmap = add_constraint(s, v, bmap);
921 tok = isl_stream_next_token(s);
923 isl_stream_error(s, NULL, "unexpected EOF");
926 if (tok->type != ISL_TOKEN_AND)
930 isl_stream_push_token(s, tok);
936 isl_basic_map_free(bmap);
940 static struct isl_basic_map *read_disjunct(struct isl_stream *s,
941 struct vars *v, __isl_take isl_dim *dim)
944 struct isl_token *tok;
945 struct isl_basic_map *bmap;
947 bmap = isl_basic_map_alloc_dim(dim, 0, 0, 0);
951 tok = isl_stream_next_token(s);
954 if (tok->type == '(') {
958 isl_stream_push_token(s, tok);
960 bmap = add_constraints(s, v, bmap);
961 bmap = isl_basic_map_simplify(bmap);
962 bmap = isl_basic_map_finalize(bmap);
964 if (seen_paren && isl_stream_eat(s, ')'))
969 isl_basic_map_free(bmap);
973 static struct isl_map *read_disjuncts(struct isl_stream *s,
974 struct vars *v, __isl_take isl_dim *dim)
976 struct isl_token *tok;
979 tok = isl_stream_next_token(s);
981 isl_stream_error(s, NULL, "unexpected EOF");
984 if (tok->type == '}') {
985 isl_stream_push_token(s, tok);
986 return isl_map_universe(dim);
988 isl_stream_push_token(s, tok);
990 map = isl_map_empty(isl_dim_copy(dim));
992 struct isl_basic_map *bmap;
995 bmap = read_disjunct(s, v, isl_dim_copy(dim));
996 map = isl_map_union(map, isl_map_from_basic_map(bmap));
998 vars_drop(v, v->n - n);
1000 tok = isl_stream_next_token(s);
1001 if (!tok || tok->type != ISL_TOKEN_OR)
1003 isl_token_free(tok);
1006 isl_stream_push_token(s, tok);
1015 static int polylib_pos_to_isl_pos(__isl_keep isl_basic_map *bmap, int pos)
1017 if (pos < isl_basic_map_dim(bmap, isl_dim_out))
1018 return 1 + isl_basic_map_dim(bmap, isl_dim_param) +
1019 isl_basic_map_dim(bmap, isl_dim_in) + pos;
1020 pos -= isl_basic_map_dim(bmap, isl_dim_out);
1022 if (pos < isl_basic_map_dim(bmap, isl_dim_in))
1023 return 1 + isl_basic_map_dim(bmap, isl_dim_param) + pos;
1024 pos -= isl_basic_map_dim(bmap, isl_dim_in);
1026 if (pos < isl_basic_map_dim(bmap, isl_dim_div))
1027 return 1 + isl_basic_map_dim(bmap, isl_dim_param) +
1028 isl_basic_map_dim(bmap, isl_dim_in) +
1029 isl_basic_map_dim(bmap, isl_dim_out) + pos;
1030 pos -= isl_basic_map_dim(bmap, isl_dim_div);
1032 if (pos < isl_basic_map_dim(bmap, isl_dim_param))
1038 static __isl_give isl_basic_map *basic_map_read_polylib_constraint(
1039 struct isl_stream *s, __isl_take isl_basic_map *bmap)
1042 struct isl_token *tok;
1052 nparam = isl_basic_map_dim(bmap, isl_dim_param);
1053 dim = isl_basic_map_dim(bmap, isl_dim_out);
1055 tok = isl_stream_next_token(s);
1056 if (!tok || tok->type != ISL_TOKEN_VALUE) {
1057 isl_stream_error(s, tok, "expecting coefficient");
1059 isl_stream_push_token(s, tok);
1062 if (!tok->on_new_line) {
1063 isl_stream_error(s, tok, "coefficient should appear on new line");
1064 isl_stream_push_token(s, tok);
1068 type = isl_int_get_si(tok->u.v);
1069 isl_token_free(tok);
1071 isl_assert(s->ctx, type == 0 || type == 1, goto error);
1073 k = isl_basic_map_alloc_equality(bmap);
1076 k = isl_basic_map_alloc_inequality(bmap);
1082 for (j = 0; j < 1 + isl_basic_map_total_dim(bmap); ++j) {
1084 tok = isl_stream_next_token(s);
1085 if (!tok || tok->type != ISL_TOKEN_VALUE) {
1086 isl_stream_error(s, tok, "expecting coefficient");
1088 isl_stream_push_token(s, tok);
1091 if (tok->on_new_line) {
1092 isl_stream_error(s, tok,
1093 "coefficient should not appear on new line");
1094 isl_stream_push_token(s, tok);
1097 pos = polylib_pos_to_isl_pos(bmap, j);
1098 isl_int_set(c[pos], tok->u.v);
1099 isl_token_free(tok);
1104 isl_basic_map_free(bmap);
1108 static __isl_give isl_basic_map *basic_map_read_polylib(struct isl_stream *s,
1112 struct isl_token *tok;
1113 struct isl_token *tok2;
1116 unsigned in = 0, out, local = 0;
1117 struct isl_basic_map *bmap = NULL;
1122 tok = isl_stream_next_token(s);
1124 isl_stream_error(s, NULL, "unexpected EOF");
1127 tok2 = isl_stream_next_token(s);
1129 isl_token_free(tok);
1130 isl_stream_error(s, NULL, "unexpected EOF");
1133 n_row = isl_int_get_si(tok->u.v);
1134 n_col = isl_int_get_si(tok2->u.v);
1135 on_new_line = tok2->on_new_line;
1136 isl_token_free(tok2);
1137 isl_token_free(tok);
1138 isl_assert(s->ctx, !on_new_line, return NULL);
1139 isl_assert(s->ctx, n_row >= 0, return NULL);
1140 isl_assert(s->ctx, n_col >= 2 + nparam, return NULL);
1141 tok = isl_stream_next_token_on_same_line(s);
1143 if (tok->type != ISL_TOKEN_VALUE) {
1144 isl_stream_error(s, tok,
1145 "expecting number of output dimensions");
1146 isl_stream_push_token(s, tok);
1149 out = isl_int_get_si(tok->u.v);
1150 isl_token_free(tok);
1152 tok = isl_stream_next_token_on_same_line(s);
1153 if (!tok || tok->type != ISL_TOKEN_VALUE) {
1154 isl_stream_error(s, tok,
1155 "expecting number of input dimensions");
1157 isl_stream_push_token(s, tok);
1160 in = isl_int_get_si(tok->u.v);
1161 isl_token_free(tok);
1163 tok = isl_stream_next_token_on_same_line(s);
1164 if (!tok || tok->type != ISL_TOKEN_VALUE) {
1165 isl_stream_error(s, tok,
1166 "expecting number of existentials");
1168 isl_stream_push_token(s, tok);
1171 local = isl_int_get_si(tok->u.v);
1172 isl_token_free(tok);
1174 tok = isl_stream_next_token_on_same_line(s);
1175 if (!tok || tok->type != ISL_TOKEN_VALUE) {
1176 isl_stream_error(s, tok,
1177 "expecting number of parameters");
1179 isl_stream_push_token(s, tok);
1182 nparam = isl_int_get_si(tok->u.v);
1183 isl_token_free(tok);
1184 if (n_col != 1 + out + in + local + nparam + 1) {
1185 isl_stream_error(s, NULL,
1186 "dimensions don't match");
1190 out = n_col - 2 - nparam;
1191 bmap = isl_basic_map_alloc(s->ctx, nparam, in, out, local, n_row, n_row);
1195 for (i = 0; i < local; ++i) {
1196 int k = isl_basic_map_alloc_div(bmap);
1201 for (i = 0; i < n_row; ++i)
1202 bmap = basic_map_read_polylib_constraint(s, bmap);
1204 tok = isl_stream_next_token_on_same_line(s);
1206 isl_stream_error(s, tok, "unexpected extra token on line");
1207 isl_stream_push_token(s, tok);
1211 bmap = isl_basic_map_simplify(bmap);
1212 bmap = isl_basic_map_finalize(bmap);
1215 isl_basic_map_free(bmap);
1219 static struct isl_map *map_read_polylib(struct isl_stream *s, int nparam)
1221 struct isl_token *tok;
1222 struct isl_token *tok2;
1224 struct isl_map *map;
1226 tok = isl_stream_next_token(s);
1228 isl_stream_error(s, NULL, "unexpected EOF");
1231 tok2 = isl_stream_next_token_on_same_line(s);
1233 isl_stream_push_token(s, tok2);
1234 isl_stream_push_token(s, tok);
1235 return isl_map_from_basic_map(basic_map_read_polylib(s, nparam));
1237 n = isl_int_get_si(tok->u.v);
1238 isl_token_free(tok);
1240 isl_assert(s->ctx, n >= 1, return NULL);
1242 map = isl_map_from_basic_map(basic_map_read_polylib(s, nparam));
1244 for (i = 1; i < n; ++i)
1245 map = isl_map_union(map,
1246 isl_map_from_basic_map(basic_map_read_polylib(s, nparam)));
1251 static int optional_power(struct isl_stream *s)
1254 struct isl_token *tok;
1256 tok = isl_stream_next_token(s);
1259 if (tok->type != '^') {
1260 isl_stream_push_token(s, tok);
1263 isl_token_free(tok);
1264 tok = isl_stream_next_token(s);
1265 if (!tok || tok->type != ISL_TOKEN_VALUE) {
1266 isl_stream_error(s, tok, "expecting exponent");
1268 isl_stream_push_token(s, tok);
1271 pow = isl_int_get_si(tok->u.v);
1272 isl_token_free(tok);
1276 static __isl_give isl_div *read_div(struct isl_stream *s,
1277 __isl_keep isl_basic_map *bmap, struct vars *v)
1279 unsigned total = isl_basic_map_total_dim(bmap);
1282 bmap = isl_basic_map_copy(bmap);
1283 bmap = isl_basic_map_cow(bmap);
1284 bmap = isl_basic_map_extend_dim(bmap, isl_dim_copy(bmap->dim),
1287 if ((k = isl_basic_map_alloc_div(bmap)) < 0)
1289 isl_seq_clr(bmap->div[k], 1 + 1 + total);
1291 if (vars_add_anon(v) < 0)
1293 bmap = add_div_definition(s, v, bmap, k);
1296 return isl_basic_map_div(bmap, k);
1298 isl_basic_map_free(bmap);
1302 static __isl_give isl_qpolynomial *read_term(struct isl_stream *s,
1303 __isl_keep isl_basic_map *bmap, struct vars *v);
1305 static __isl_give isl_qpolynomial *read_factor(struct isl_stream *s,
1306 __isl_keep isl_basic_map *bmap, struct vars *v)
1308 struct isl_qpolynomial *qp;
1309 struct isl_token *tok;
1311 tok = isl_stream_next_token(s);
1313 isl_stream_error(s, NULL, "unexpected EOF");
1316 if (tok->type == '(') {
1317 isl_token_free(tok);
1318 qp = read_term(s, bmap, v);
1321 if (isl_stream_eat(s, ')'))
1323 } else if (tok->type == ISL_TOKEN_VALUE) {
1324 struct isl_token *tok2;
1325 tok2 = isl_stream_next_token(s);
1326 if (tok2 && tok2->type == '/') {
1327 isl_token_free(tok2);
1328 tok2 = isl_stream_next_token(s);
1329 if (!tok2 || tok2->type != ISL_TOKEN_VALUE) {
1330 isl_stream_error(s, tok2, "expected denominator");
1331 isl_token_free(tok);
1332 isl_token_free(tok2);
1335 qp = isl_qpolynomial_rat_cst(isl_basic_map_get_dim(bmap),
1336 tok->u.v, tok2->u.v);
1337 isl_token_free(tok2);
1339 isl_stream_push_token(s, tok2);
1340 qp = isl_qpolynomial_cst(isl_basic_map_get_dim(bmap),
1343 isl_token_free(tok);
1344 } else if (tok->type == ISL_TOKEN_INFTY) {
1345 isl_token_free(tok);
1346 qp = isl_qpolynomial_infty(isl_basic_map_get_dim(bmap));
1347 } else if (tok->type == ISL_TOKEN_NAN) {
1348 isl_token_free(tok);
1349 qp = isl_qpolynomial_nan(isl_basic_map_get_dim(bmap));
1350 } else if (tok->type == ISL_TOKEN_IDENT) {
1352 int pos = vars_pos(v, tok->u.s, -1);
1355 isl_token_free(tok);
1359 isl_stream_error(s, tok, "unknown identifier");
1360 isl_token_free(tok);
1363 isl_token_free(tok);
1364 pow = optional_power(s);
1365 qp = isl_qpolynomial_pow(isl_basic_map_get_dim(bmap), pos, pow);
1366 } else if (tok->type == '[') {
1370 isl_stream_push_token(s, tok);
1371 div = read_div(s, bmap, v);
1372 pow = optional_power(s);
1373 qp = isl_qpolynomial_div_pow(div, pow);
1374 } else if (tok->type == '-') {
1375 struct isl_qpolynomial *qp2;
1377 isl_token_free(tok);
1378 qp = isl_qpolynomial_cst(isl_basic_map_get_dim(bmap),
1380 qp2 = read_factor(s, bmap, v);
1381 qp = isl_qpolynomial_mul(qp, qp2);
1383 isl_stream_error(s, tok, "unexpected isl_token");
1384 isl_stream_push_token(s, tok);
1388 if (isl_stream_eat_if_available(s, '*') ||
1389 isl_stream_next_token_is(s, ISL_TOKEN_IDENT)) {
1390 struct isl_qpolynomial *qp2;
1392 qp2 = read_factor(s, bmap, v);
1393 qp = isl_qpolynomial_mul(qp, qp2);
1398 isl_qpolynomial_free(qp);
1402 static __isl_give isl_qpolynomial *read_term(struct isl_stream *s,
1403 __isl_keep isl_basic_map *bmap, struct vars *v)
1405 struct isl_token *tok;
1406 struct isl_qpolynomial *qp;
1408 qp = read_factor(s, bmap, v);
1411 tok = isl_stream_next_token(s);
1415 if (tok->type == '+') {
1416 struct isl_qpolynomial *qp2;
1418 isl_token_free(tok);
1419 qp2 = read_factor(s, bmap, v);
1420 qp = isl_qpolynomial_add(qp, qp2);
1421 } else if (tok->type == '-') {
1422 struct isl_qpolynomial *qp2;
1424 isl_token_free(tok);
1425 qp2 = read_factor(s, bmap, v);
1426 qp = isl_qpolynomial_sub(qp, qp2);
1427 } else if (tok->type == ISL_TOKEN_VALUE &&
1428 isl_int_is_neg(tok->u.v)) {
1429 struct isl_qpolynomial *qp2;
1431 isl_stream_push_token(s, tok);
1432 qp2 = read_factor(s, bmap, v);
1433 qp = isl_qpolynomial_add(qp, qp2);
1435 isl_stream_push_token(s, tok);
1443 static __isl_give isl_map *read_optional_disjuncts(struct isl_stream *s,
1444 __isl_take isl_basic_map *bmap, struct vars *v)
1446 struct isl_token *tok;
1447 struct isl_map *map;
1449 tok = isl_stream_next_token(s);
1451 isl_stream_error(s, NULL, "unexpected EOF");
1454 map = isl_map_from_basic_map(isl_basic_map_copy(bmap));
1455 if (tok->type == ':') {
1456 isl_token_free(tok);
1457 map = isl_map_intersect(map,
1458 read_disjuncts(s, v, isl_basic_map_get_dim(bmap)));
1460 isl_stream_push_token(s, tok);
1462 isl_basic_map_free(bmap);
1466 isl_basic_map_free(bmap);
1470 static struct isl_obj obj_read_poly(struct isl_stream *s,
1471 __isl_take isl_basic_map *bmap, struct vars *v, int n)
1473 struct isl_obj obj = { isl_obj_pw_qpolynomial, NULL };
1474 struct isl_pw_qpolynomial *pwqp;
1475 struct isl_qpolynomial *qp;
1476 struct isl_map *map;
1477 struct isl_set *set;
1479 qp = read_term(s, bmap, v);
1480 map = read_optional_disjuncts(s, bmap, v);
1481 set = isl_map_range(map);
1483 pwqp = isl_pw_qpolynomial_alloc(set, qp);
1485 vars_drop(v, v->n - n);
1491 static struct isl_obj obj_read_poly_or_fold(struct isl_stream *s,
1492 __isl_take isl_basic_map *bmap, struct vars *v, int n)
1494 struct isl_obj obj = { isl_obj_pw_qpolynomial_fold, NULL };
1495 struct isl_obj obj_p;
1496 isl_qpolynomial *qp;
1497 isl_qpolynomial_fold *fold = NULL;
1498 isl_pw_qpolynomial_fold *pwf;
1502 if (!isl_stream_eat_if_available(s, ISL_TOKEN_MAX))
1503 return obj_read_poly(s, bmap, v, n);
1505 if (isl_stream_eat(s, '('))
1508 qp = read_term(s, bmap, v);
1509 fold = isl_qpolynomial_fold_alloc(isl_fold_max, qp);
1511 while (isl_stream_eat_if_available(s, ',')) {
1512 isl_qpolynomial_fold *fold_i;
1513 qp = read_term(s, bmap, v);
1514 fold_i = isl_qpolynomial_fold_alloc(isl_fold_max, qp);
1515 fold = isl_qpolynomial_fold_fold(fold, fold_i);
1518 if (isl_stream_eat(s, ')'))
1521 map = read_optional_disjuncts(s, bmap, v);
1522 set = isl_map_range(map);
1523 pwf = isl_pw_qpolynomial_fold_alloc(isl_fold_max, set, fold);
1525 vars_drop(v, v->n - n);
1530 isl_basic_map_free(bmap);
1531 isl_qpolynomial_fold_free(fold);
1532 obj.type = isl_obj_none;
1536 static struct isl_obj obj_read_body(struct isl_stream *s,
1537 __isl_take isl_basic_map *bmap, struct vars *v)
1539 struct isl_map *map = NULL;
1540 struct isl_token *tok;
1541 struct isl_obj obj = { isl_obj_set, NULL };
1544 if (!next_is_tuple(s))
1545 return obj_read_poly_or_fold(s, bmap, v, n);
1547 bmap = read_tuple(s, bmap, isl_dim_in, v);
1550 tok = isl_stream_next_token(s);
1551 if (tok && tok->type == ISL_TOKEN_TO) {
1552 obj.type = isl_obj_map;
1553 isl_token_free(tok);
1554 if (!next_is_tuple(s)) {
1555 bmap = isl_basic_map_reverse(bmap);
1556 return obj_read_poly_or_fold(s, bmap, v, n);
1558 bmap = read_tuple(s, bmap, isl_dim_out, v);
1562 bmap = isl_basic_map_reverse(bmap);
1564 isl_stream_push_token(s, tok);
1567 map = read_optional_disjuncts(s, bmap, v);
1569 vars_drop(v, v->n - n);
1574 isl_basic_map_free(bmap);
1575 obj.type = isl_obj_none;
1579 static struct isl_obj to_union(isl_ctx *ctx, struct isl_obj obj)
1581 if (obj.type == isl_obj_map) {
1582 obj.v = isl_union_map_from_map(obj.v);
1583 obj.type = isl_obj_union_map;
1584 } else if (obj.type == isl_obj_set) {
1585 obj.v = isl_union_set_from_set(obj.v);
1586 obj.type = isl_obj_union_set;
1587 } else if (obj.type == isl_obj_pw_qpolynomial) {
1588 obj.v = isl_union_pw_qpolynomial_from_pw_qpolynomial(obj.v);
1589 obj.type = isl_obj_union_pw_qpolynomial;
1590 } else if (obj.type == isl_obj_pw_qpolynomial_fold) {
1591 obj.v = isl_union_pw_qpolynomial_fold_from_pw_qpolynomial_fold(obj.v);
1592 obj.type = isl_obj_union_pw_qpolynomial_fold;
1594 isl_assert(ctx, 0, goto error);
1597 obj.type->free(obj.v);
1598 obj.type = isl_obj_none;
1602 static struct isl_obj obj_add(struct isl_ctx *ctx,
1603 struct isl_obj obj1, struct isl_obj obj2)
1605 if (obj1.type == isl_obj_set && obj2.type == isl_obj_union_set)
1606 obj1 = to_union(ctx, obj1);
1607 if (obj1.type == isl_obj_union_set && obj2.type == isl_obj_set)
1608 obj2 = to_union(ctx, obj2);
1609 if (obj1.type == isl_obj_map && obj2.type == isl_obj_union_map)
1610 obj1 = to_union(ctx, obj1);
1611 if (obj1.type == isl_obj_union_map && obj2.type == isl_obj_map)
1612 obj2 = to_union(ctx, obj2);
1613 if (obj1.type == isl_obj_pw_qpolynomial &&
1614 obj2.type == isl_obj_union_pw_qpolynomial)
1615 obj1 = to_union(ctx, obj1);
1616 if (obj1.type == isl_obj_union_pw_qpolynomial &&
1617 obj2.type == isl_obj_pw_qpolynomial)
1618 obj2 = to_union(ctx, obj2);
1619 if (obj1.type == isl_obj_pw_qpolynomial_fold &&
1620 obj2.type == isl_obj_union_pw_qpolynomial_fold)
1621 obj1 = to_union(ctx, obj1);
1622 if (obj1.type == isl_obj_union_pw_qpolynomial_fold &&
1623 obj2.type == isl_obj_pw_qpolynomial_fold)
1624 obj2 = to_union(ctx, obj2);
1625 isl_assert(ctx, obj1.type == obj2.type, goto error);
1626 if (obj1.type == isl_obj_map && !isl_map_has_equal_dim(obj1.v, obj2.v)) {
1627 obj1 = to_union(ctx, obj1);
1628 obj2 = to_union(ctx, obj2);
1630 if (obj1.type == isl_obj_set && !isl_set_has_equal_dim(obj1.v, obj2.v)) {
1631 obj1 = to_union(ctx, obj1);
1632 obj2 = to_union(ctx, obj2);
1634 if (obj1.type == isl_obj_pw_qpolynomial &&
1635 !isl_pw_qpolynomial_has_equal_dim(obj1.v, obj2.v)) {
1636 obj1 = to_union(ctx, obj1);
1637 obj2 = to_union(ctx, obj2);
1639 if (obj1.type == isl_obj_pw_qpolynomial_fold &&
1640 !isl_pw_qpolynomial_fold_has_equal_dim(obj1.v, obj2.v)) {
1641 obj1 = to_union(ctx, obj1);
1642 obj2 = to_union(ctx, obj2);
1644 obj1.v = obj1.type->add(obj1.v, obj2.v);
1647 obj1.type->free(obj1.v);
1648 obj2.type->free(obj2.v);
1649 obj1.type = isl_obj_none;
1654 static struct isl_obj obj_read(struct isl_stream *s, int nparam)
1656 isl_basic_map *bmap = NULL;
1657 struct isl_token *tok;
1658 struct vars *v = NULL;
1659 struct isl_obj obj = { isl_obj_set, NULL };
1661 tok = isl_stream_next_token(s);
1663 isl_stream_error(s, NULL, "unexpected EOF");
1666 if (tok->type == ISL_TOKEN_VALUE) {
1667 struct isl_map *map;
1668 isl_stream_push_token(s, tok);
1669 map = map_read_polylib(s, nparam);
1672 if (isl_map_dim(map, isl_dim_in) > 0)
1673 obj.type = isl_obj_map;
1677 v = vars_new(s->ctx);
1679 isl_stream_push_token(s, tok);
1682 bmap = isl_basic_map_alloc(s->ctx, 0, 0, 0, 0, 0, 0);
1683 if (tok->type == '[') {
1684 isl_stream_push_token(s, tok);
1685 bmap = read_tuple(s, bmap, isl_dim_param, v);
1689 isl_assert(s->ctx, nparam == v->n, goto error);
1690 tok = isl_stream_next_token(s);
1691 if (!tok || tok->type != ISL_TOKEN_TO) {
1692 isl_stream_error(s, tok, "expecting '->'");
1694 isl_stream_push_token(s, tok);
1697 isl_token_free(tok);
1698 tok = isl_stream_next_token(s);
1699 } else if (nparam > 0)
1700 bmap = isl_basic_map_add(bmap, isl_dim_param, nparam);
1701 if (!tok || tok->type != '{') {
1702 isl_stream_error(s, tok, "expecting '{'");
1704 isl_stream_push_token(s, tok);
1707 isl_token_free(tok);
1709 tok = isl_stream_next_token(s);
1712 else if (tok->type == ISL_TOKEN_IDENT && !strcmp(tok->u.s, "Sym")) {
1713 isl_token_free(tok);
1714 if (isl_stream_eat(s, '='))
1716 bmap = read_tuple(s, bmap, isl_dim_param, v);
1720 isl_assert(s->ctx, nparam == v->n, goto error);
1721 } else if (tok->type == '}') {
1722 obj.type = isl_obj_union_set;
1723 obj.v = isl_union_set_empty(isl_basic_map_get_dim(bmap));
1724 isl_token_free(tok);
1727 isl_stream_push_token(s, tok);
1732 o = obj_read_body(s, isl_basic_map_copy(bmap), v);
1733 if (o.type == isl_obj_none || !o.v)
1738 obj = obj_add(s->ctx, obj, o);
1739 if (obj.type == isl_obj_none || !obj.v)
1742 tok = isl_stream_next_token(s);
1743 if (!tok || tok->type != ';')
1745 isl_token_free(tok);
1748 if (tok && tok->type == '}') {
1749 isl_token_free(tok);
1751 isl_stream_error(s, tok, "unexpected isl_token");
1753 isl_token_free(tok);
1758 isl_basic_map_free(bmap);
1762 isl_basic_map_free(bmap);
1763 obj.type->free(obj.v);
1770 struct isl_obj isl_stream_read_obj(struct isl_stream *s)
1772 return obj_read(s, -1);
1775 __isl_give isl_map *isl_stream_read_map(struct isl_stream *s, int nparam)
1778 struct isl_map *map;
1780 obj = obj_read(s, nparam);
1782 isl_assert(s->ctx, obj.type == isl_obj_map ||
1783 obj.type == isl_obj_set, goto error);
1787 obj.type->free(obj.v);
1791 __isl_give isl_set *isl_stream_read_set(struct isl_stream *s, int nparam)
1794 struct isl_set *set;
1796 obj = obj_read(s, nparam);
1798 isl_assert(s->ctx, obj.type == isl_obj_set, goto error);
1802 obj.type->free(obj.v);
1806 static struct isl_basic_map *basic_map_read(struct isl_stream *s, int nparam)
1809 struct isl_map *map;
1810 struct isl_basic_map *bmap;
1812 obj = obj_read(s, nparam);
1817 isl_assert(map->ctx, map->n <= 1, goto error);
1820 bmap = isl_basic_map_empty_like_map(map);
1822 bmap = isl_basic_map_copy(map->p[0]);
1832 __isl_give isl_basic_map *isl_basic_map_read_from_file(isl_ctx *ctx,
1833 FILE *input, int nparam)
1835 struct isl_basic_map *bmap;
1836 struct isl_stream *s = isl_stream_new_file(ctx, input);
1839 bmap = basic_map_read(s, nparam);
1844 __isl_give isl_basic_set *isl_basic_set_read_from_file(isl_ctx *ctx,
1845 FILE *input, int nparam)
1847 struct isl_basic_map *bmap;
1848 bmap = isl_basic_map_read_from_file(ctx, input, nparam);
1851 isl_assert(ctx, isl_basic_map_n_in(bmap) == 0, goto error);
1852 return (struct isl_basic_set *)bmap;
1854 isl_basic_map_free(bmap);
1858 struct isl_basic_map *isl_basic_map_read_from_str(struct isl_ctx *ctx,
1859 const char *str, int nparam)
1861 struct isl_basic_map *bmap;
1862 struct isl_stream *s = isl_stream_new_str(ctx, str);
1865 bmap = basic_map_read(s, nparam);
1870 struct isl_basic_set *isl_basic_set_read_from_str(struct isl_ctx *ctx,
1871 const char *str, int nparam)
1873 struct isl_basic_map *bmap;
1874 bmap = isl_basic_map_read_from_str(ctx, str, nparam);
1877 isl_assert(ctx, isl_basic_map_n_in(bmap) == 0, goto error);
1878 return (struct isl_basic_set *)bmap;
1880 isl_basic_map_free(bmap);
1884 __isl_give isl_map *isl_map_read_from_file(struct isl_ctx *ctx,
1885 FILE *input, int nparam)
1887 struct isl_map *map;
1888 struct isl_stream *s = isl_stream_new_file(ctx, input);
1891 map = isl_stream_read_map(s, nparam);
1896 __isl_give isl_map *isl_map_read_from_str(struct isl_ctx *ctx,
1897 const char *str, int nparam)
1899 struct isl_map *map;
1900 struct isl_stream *s = isl_stream_new_str(ctx, str);
1903 map = isl_stream_read_map(s, nparam);
1908 __isl_give isl_set *isl_set_read_from_file(struct isl_ctx *ctx,
1909 FILE *input, int nparam)
1911 struct isl_map *map;
1912 map = isl_map_read_from_file(ctx, input, nparam);
1915 isl_assert(ctx, isl_map_n_in(map) == 0, goto error);
1916 return (struct isl_set *)map;
1922 struct isl_set *isl_set_read_from_str(struct isl_ctx *ctx,
1923 const char *str, int nparam)
1925 struct isl_map *map;
1926 map = isl_map_read_from_str(ctx, str, nparam);
1929 isl_assert(ctx, isl_map_n_in(map) == 0, goto error);
1930 return (struct isl_set *)map;
1936 static char *next_line(FILE *input, char *line, unsigned len)
1941 if (!(p = fgets(line, len, input)))
1943 while (isspace(*p) && *p != '\n')
1945 } while (*p == '#' || *p == '\n');
1950 static struct isl_vec *isl_vec_read_from_file_polylib(struct isl_ctx *ctx,
1953 struct isl_vec *vec = NULL;
1962 isl_assert(ctx, next_line(input, line, sizeof(line)), return NULL);
1963 isl_assert(ctx, sscanf(line, "%u", &size) == 1, return NULL);
1965 vec = isl_vec_alloc(ctx, size);
1967 p = next_line(input, line, sizeof(line));
1968 isl_assert(ctx, p, goto error);
1970 for (j = 0; j < size; ++j) {
1971 n = sscanf(p, "%s%n", val, &offset);
1972 isl_assert(ctx, n != 0, goto error);
1973 isl_int_read(vec->el[j], val);
1983 struct isl_vec *isl_vec_read_from_file(struct isl_ctx *ctx,
1984 FILE *input, unsigned input_format)
1986 if (input_format == ISL_FORMAT_POLYLIB)
1987 return isl_vec_read_from_file_polylib(ctx, input);
1989 isl_assert(ctx, 0, return NULL);
1992 __isl_give isl_pw_qpolynomial *isl_stream_read_pw_qpolynomial(
1993 struct isl_stream *s)
1996 struct isl_pw_qpolynomial *pwqp;
1998 obj = obj_read(s, -1);
2000 isl_assert(s->ctx, obj.type == isl_obj_pw_qpolynomial,
2005 obj.type->free(obj.v);
2009 __isl_give isl_pw_qpolynomial *isl_pw_qpolynomial_read_from_str(isl_ctx *ctx,
2012 isl_pw_qpolynomial *pwqp;
2013 struct isl_stream *s = isl_stream_new_str(ctx, str);
2016 pwqp = isl_stream_read_pw_qpolynomial(s);