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
17 #include <isl_map_private.h>
21 #include <isl/stream.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 /* Given an affine expression aff, return an affine expression
187 * for aff % d, with d the next token on the stream, which is
188 * assumed to be a constant.
190 * We introduce an integer division q = [aff/d] and the result
191 * is set to aff - d q.
193 static __isl_give isl_vec
*affine_mod(struct isl_stream
*s
,
194 struct vars
*v
, __isl_take isl_vec
*aff
)
196 struct isl_token
*tok
;
197 struct variable
*var
;
200 tok
= isl_stream_next_token(s
);
201 if (!tok
|| tok
->type
!= ISL_TOKEN_VALUE
) {
202 isl_stream_error(s
, tok
, "expecting constant value");
206 if (vars_add_anon(v
) < 0)
211 var
->def
= isl_vec_alloc(s
->ctx
, 2 + v
->n
);
214 isl_seq_cpy(var
->def
->el
+ 1, aff
->el
, aff
->size
);
215 isl_int_set_si(var
->def
->el
[1 + aff
->size
], 0);
216 isl_int_set(var
->def
->el
[0], tok
->u
.v
);
218 mod
= isl_vec_alloc(v
->ctx
, 1 + v
->n
);
222 isl_seq_cpy(mod
->el
, aff
->el
, aff
->size
);
223 isl_int_neg(mod
->el
[aff
->size
], tok
->u
.v
);
234 static struct isl_vec
*accept_affine(struct isl_stream
*s
, struct vars
*v
);
235 static int read_div_definition(struct isl_stream
*s
, struct vars
*v
);
237 static __isl_give isl_vec
*accept_affine_factor(struct isl_stream
*s
,
240 struct isl_token
*tok
= NULL
;
243 tok
= isl_stream_next_token(s
);
245 isl_stream_error(s
, NULL
, "unexpected EOF");
248 if (tok
->type
== ISL_TOKEN_IDENT
) {
250 int pos
= vars_pos(v
, tok
->u
.s
, -1);
254 isl_stream_error(s
, tok
, "unknown identifier");
258 aff
= isl_vec_alloc(v
->ctx
, 1 + v
->n
);
261 isl_seq_clr(aff
->el
, aff
->size
);
262 isl_int_set_si(aff
->el
[1 + pos
], 1);
264 } else if (tok
->type
== ISL_TOKEN_VALUE
) {
265 if (isl_stream_eat_if_available(s
, '*')) {
266 aff
= accept_affine_factor(s
, v
);
267 aff
= isl_vec_scale(aff
, tok
->u
.v
);
269 aff
= isl_vec_alloc(v
->ctx
, 1 + v
->n
);
272 isl_seq_clr(aff
->el
, aff
->size
);
273 isl_int_set(aff
->el
[0], tok
->u
.v
);
276 } else if (tok
->type
== '(') {
279 aff
= accept_affine(s
, v
);
282 if (isl_stream_eat(s
, ')'))
284 } else if (tok
->type
== '[') {
285 if (vars_add_anon(v
) < 0)
287 aff
= isl_vec_alloc(v
->ctx
, 1 + v
->n
);
290 isl_seq_clr(aff
->el
, aff
->size
);
291 isl_int_set_si(aff
->el
[1 + v
->n
- 1], 1);
292 isl_stream_push_token(s
, tok
);
294 if (read_div_definition(s
, v
) < 0)
297 isl_stream_error(s
, tok
, "expecting factor");
300 if (isl_stream_eat_if_available(s
, '%'))
301 return affine_mod(s
, v
, aff
);
302 if (isl_stream_eat_if_available(s
, '*')) {
305 isl_int_set_si(f
, 1);
306 if (accept_cst_factor(s
, &f
) < 0) {
310 aff
= isl_vec_scale(aff
, f
);
322 static struct isl_vec
*accept_affine(struct isl_stream
*s
, struct vars
*v
)
324 struct isl_token
*tok
= NULL
;
328 aff
= isl_vec_alloc(v
->ctx
, 1 + v
->n
);
331 isl_seq_clr(aff
->el
, aff
->size
);
334 tok
= isl_stream_next_token(s
);
336 isl_stream_error(s
, NULL
, "unexpected EOF");
339 if (tok
->type
== '-') {
344 if (tok
->type
== '(' || tok
->type
== '[' ||
345 tok
->type
== ISL_TOKEN_IDENT
) {
347 isl_stream_push_token(s
, tok
);
349 aff2
= accept_affine_factor(s
, v
);
351 aff2
= isl_vec_scale(aff2
, s
->ctx
->negone
);
352 aff
= isl_vec_zero_extend(aff
, 1 + v
->n
);
353 aff
= isl_vec_add(aff
, aff2
);
357 } else if (tok
->type
== ISL_TOKEN_VALUE
) {
359 isl_int_neg(tok
->u
.v
, tok
->u
.v
);
360 if (isl_stream_eat_if_available(s
, '*') ||
361 isl_stream_next_token_is(s
, ISL_TOKEN_IDENT
)) {
363 aff2
= accept_affine_factor(s
, v
);
364 aff2
= isl_vec_scale(aff2
, tok
->u
.v
);
365 aff
= isl_vec_zero_extend(aff
, 1 + v
->n
);
366 aff
= isl_vec_add(aff
, aff2
);
370 isl_int_add(aff
->el
[0], aff
->el
[0], tok
->u
.v
);
374 isl_stream_error(s
, tok
, "unexpected isl_token");
375 isl_stream_push_token(s
, tok
);
381 tok
= isl_stream_next_token(s
);
382 if (tok
&& tok
->type
== '-') {
385 } else if (tok
&& tok
->type
== '+') {
388 } else if (tok
&& tok
->type
== ISL_TOKEN_VALUE
&&
389 isl_int_is_neg(tok
->u
.v
)) {
390 isl_stream_push_token(s
, tok
);
393 isl_stream_push_token(s
, tok
);
405 /* Add any variables in the variable list "v" that are not already in "bmap"
406 * as existentially quantified variables in "bmap".
408 static __isl_give isl_basic_map
*add_divs(__isl_take isl_basic_map
*bmap
,
413 struct variable
*var
;
415 extra
= v
->n
- isl_basic_map_total_dim(bmap
);
420 bmap
= isl_basic_map_extend_dim(bmap
, isl_basic_map_get_dim(bmap
),
421 extra
, 0, 2 * extra
);
423 for (i
= 0; i
< extra
; ++i
)
424 if (isl_basic_map_alloc_div(bmap
) < 0)
427 for (i
= 0, var
= v
->v
; i
< extra
; ++i
, var
= var
->next
) {
428 int k
= bmap
->n_div
- 1 - i
;
430 isl_seq_cpy(bmap
->div
[k
], var
->def
->el
, var
->def
->size
);
431 isl_seq_clr(bmap
->div
[k
] + var
->def
->size
,
432 2 + v
->n
- var
->def
->size
);
434 if (isl_basic_map_add_div_constraints(bmap
, k
) < 0)
440 isl_basic_map_free(bmap
);
444 static __isl_give isl_basic_map
*read_var_def(struct isl_stream
*s
,
445 __isl_take isl_basic_map
*bmap
, enum isl_dim_type type
, struct vars
*v
)
448 isl_basic_map
*def
= NULL
;
453 if (vars_add_anon(v
) < 0)
457 vec
= accept_affine(s
, v
);
461 dim
= isl_basic_map_get_dim(bmap
);
462 def
= isl_basic_map_universe(dim
);
463 def
= add_divs(def
, v
);
464 def
= isl_basic_map_extend_constraints(def
, 1, 0);
465 k
= isl_basic_map_alloc_equality(def
);
467 isl_seq_cpy(def
->eq
[k
], vec
->el
, vec
->size
);
468 isl_int_set_si(def
->eq
[k
][1 + n
- 1], -1);
474 vars_drop(v
, v
->n
- n
);
476 def
= isl_basic_map_simplify(def
);
477 def
= isl_basic_map_finalize(def
);
478 bmap
= isl_basic_map_intersect(bmap
, def
);
481 isl_basic_map_free(bmap
);
482 isl_basic_map_free(def
);
486 static __isl_give isl_basic_map
*read_var_list(struct isl_stream
*s
,
487 __isl_take isl_basic_map
*bmap
, enum isl_dim_type type
, struct vars
*v
)
490 struct isl_token
*tok
;
492 while ((tok
= isl_stream_next_token(s
)) != NULL
) {
495 if (tok
->type
== ISL_TOKEN_IDENT
) {
497 int p
= vars_pos(v
, tok
->u
.s
, -1);
504 bmap
= isl_basic_map_add(bmap
, type
, 1);
505 bmap
= set_name(bmap
, type
, i
, v
->v
->name
);
507 } else if (tok
->type
== ISL_TOKEN_IDENT
||
508 tok
->type
== ISL_TOKEN_VALUE
||
511 if (type
== isl_dim_param
) {
512 isl_stream_error(s
, tok
,
513 "expecting unique identifier");
516 isl_stream_push_token(s
, tok
);
518 bmap
= isl_basic_map_add(bmap
, type
, 1);
519 bmap
= read_var_def(s
, bmap
, type
, v
);
523 tok
= isl_stream_next_token(s
);
524 if (!tok
|| tok
->type
!= ',')
531 isl_stream_push_token(s
, tok
);
536 isl_basic_map_free(bmap
);
540 static __isl_give isl_mat
*accept_affine_list(struct isl_stream
*s
,
545 struct isl_token
*tok
= NULL
;
547 vec
= accept_affine(s
, v
);
548 mat
= isl_mat_from_row_vec(vec
);
553 tok
= isl_stream_next_token(s
);
555 isl_stream_error(s
, NULL
, "unexpected EOF");
558 if (tok
->type
!= ',') {
559 isl_stream_push_token(s
, tok
);
564 vec
= accept_affine(s
, v
);
565 mat
= isl_mat_add_zero_cols(mat
, 1 + v
->n
- isl_mat_cols(mat
));
566 mat
= isl_mat_vec_concat(mat
, vec
);
577 static int read_div_definition(struct isl_stream
*s
, struct vars
*v
)
579 struct isl_token
*tok
;
582 struct variable
*var
;
584 if (isl_stream_eat(s
, '['))
587 tok
= isl_stream_next_token(s
);
590 if (tok
->type
== '(') {
594 isl_stream_push_token(s
, tok
);
598 aff
= accept_affine(s
, v
);
602 var
->def
= isl_vec_alloc(s
->ctx
, 2 + v
->n
);
608 isl_seq_cpy(var
->def
->el
+ 1, aff
->el
, aff
->size
);
612 if (seen_paren
&& isl_stream_eat(s
, ')'))
614 if (isl_stream_eat(s
, '/'))
617 tok
= isl_stream_next_token(s
);
620 if (tok
->type
!= ISL_TOKEN_VALUE
) {
621 isl_stream_error(s
, tok
, "expected denominator");
622 isl_stream_push_token(s
, tok
);
625 isl_int_set(var
->def
->el
[0], tok
->u
.v
);
628 if (isl_stream_eat(s
, ']'))
634 static struct isl_basic_map
*add_div_definition(struct isl_stream
*s
,
635 struct vars
*v
, struct isl_basic_map
*bmap
, int k
)
637 struct variable
*var
= v
->v
;
639 if (read_div_definition(s
, v
) < 0)
642 isl_seq_cpy(bmap
->div
[k
], var
->def
->el
, 2 + v
->n
);
644 if (isl_basic_map_add_div_constraints(bmap
, k
) < 0)
649 isl_basic_map_free(bmap
);
653 static struct isl_basic_map
*read_defined_var_list(struct isl_stream
*s
,
654 struct vars
*v
, struct isl_basic_map
*bmap
)
656 struct isl_token
*tok
;
658 while ((tok
= isl_stream_next_token(s
)) != NULL
) {
662 unsigned total
= isl_basic_map_total_dim(bmap
);
664 if (tok
->type
!= ISL_TOKEN_IDENT
)
667 p
= vars_pos(v
, tok
->u
.s
, -1);
671 isl_stream_error(s
, tok
, "expecting unique identifier");
675 bmap
= isl_basic_map_cow(bmap
);
676 bmap
= isl_basic_map_extend_dim(bmap
, isl_dim_copy(bmap
->dim
),
679 if ((k
= isl_basic_map_alloc_div(bmap
)) < 0)
681 isl_seq_clr(bmap
->div
[k
], 1 + 1 + total
);
684 tok
= isl_stream_next_token(s
);
685 if (tok
&& tok
->type
== '=') {
687 bmap
= add_div_definition(s
, v
, bmap
, k
);
688 tok
= isl_stream_next_token(s
);
691 if (!tok
|| tok
->type
!= ',')
697 isl_stream_push_token(s
, tok
);
702 isl_basic_map_free(bmap
);
706 static int next_is_tuple(struct isl_stream
*s
)
708 struct isl_token
*tok
;
711 tok
= isl_stream_next_token(s
);
714 if (tok
->type
== '[') {
715 isl_stream_push_token(s
, tok
);
718 if (tok
->type
!= ISL_TOKEN_IDENT
&& !tok
->is_keyword
) {
719 isl_stream_push_token(s
, tok
);
723 is_tuple
= isl_stream_next_token_is(s
, '[');
725 isl_stream_push_token(s
, tok
);
730 static __isl_give isl_basic_map
*read_tuple(struct isl_stream
*s
,
731 __isl_take isl_basic_map
*bmap
, enum isl_dim_type type
, struct vars
*v
);
733 static __isl_give isl_basic_map
*read_nested_tuple(struct isl_stream
*s
,
734 __isl_take isl_basic_map
*bmap
, struct vars
*v
)
736 bmap
= read_tuple(s
, bmap
, isl_dim_in
, v
);
737 if (isl_stream_eat(s
, ISL_TOKEN_TO
))
739 bmap
= read_tuple(s
, bmap
, isl_dim_out
, v
);
740 bmap
= isl_basic_map_from_range(isl_basic_map_wrap(bmap
));
743 isl_basic_map_free(bmap
);
747 static __isl_give isl_basic_map
*read_tuple(struct isl_stream
*s
,
748 __isl_take isl_basic_map
*bmap
, enum isl_dim_type type
, struct vars
*v
)
750 struct isl_token
*tok
;
753 tok
= isl_stream_next_token(s
);
754 if (tok
&& (tok
->type
== ISL_TOKEN_IDENT
|| tok
->is_keyword
)) {
755 name
= strdup(tok
->u
.s
);
759 tok
= isl_stream_next_token(s
);
761 if (!tok
|| tok
->type
!= '[') {
762 isl_stream_error(s
, tok
, "expecting '['");
766 if (type
!= isl_dim_param
&& next_is_tuple(s
)) {
767 isl_dim
*dim
= isl_basic_map_get_dim(bmap
);
768 int nparam
= isl_dim_size(dim
, isl_dim_param
);
769 int n_in
= isl_dim_size(dim
, isl_dim_in
);
770 isl_basic_map
*nested
;
771 if (type
== isl_dim_out
)
772 dim
= isl_dim_move(dim
, isl_dim_param
, nparam
,
773 isl_dim_in
, 0, n_in
);
774 nested
= isl_basic_map_alloc_dim(dim
, 0, 0, 0);
775 nested
= read_nested_tuple(s
, nested
, v
);
776 if (type
== isl_dim_in
) {
777 nested
= isl_basic_map_reverse(nested
);
778 bmap
= isl_basic_map_intersect(nested
, bmap
);
781 dim
= isl_dim_range(isl_basic_map_get_dim(nested
));
782 dim
= isl_dim_drop(dim
, isl_dim_param
, nparam
, n_in
);
783 dim
= isl_dim_join(isl_basic_map_get_dim(bmap
), dim
);
784 bset
= isl_basic_map_domain(bmap
);
785 nested
= isl_basic_map_reset_dim(nested
, dim
);
786 bmap
= isl_basic_map_intersect_domain(nested
, bset
);
789 bmap
= read_var_list(s
, bmap
, type
, v
);
790 tok
= isl_stream_next_token(s
);
791 if (!tok
|| tok
->type
!= ']') {
792 isl_stream_error(s
, tok
, "expecting ']'");
798 bmap
= isl_basic_map_set_tuple_name(bmap
, type
, name
);
806 isl_basic_map_free(bmap
);
810 static struct isl_basic_map
*add_constraints(struct isl_stream
*s
,
811 struct vars
*v
, struct isl_basic_map
*bmap
);
813 static struct isl_basic_map
*add_exists(struct isl_stream
*s
,
814 struct vars
*v
, struct isl_basic_map
*bmap
)
816 struct isl_token
*tok
;
823 tok
= isl_stream_next_token(s
);
826 if (tok
->type
== '(') {
830 isl_stream_push_token(s
, tok
);
832 bmap
= read_defined_var_list(s
, v
, bmap
);
836 if (isl_stream_eat(s
, ':'))
838 bmap
= add_constraints(s
, v
, bmap
);
839 if (seen_paren
&& isl_stream_eat(s
, ')'))
843 isl_basic_map_free(bmap
);
847 static __isl_give isl_basic_map
*construct_constraint(
848 __isl_take isl_basic_map
*bmap
, enum isl_token_type type
,
849 isl_int
*left
, isl_int
*right
)
857 len
= 1 + isl_basic_map_total_dim(bmap
);
860 k
= isl_basic_map_alloc_inequality(bmap
);
863 if (type
== ISL_TOKEN_LE
)
864 isl_seq_combine(bmap
->ineq
[k
], ctx
->negone
, left
,
867 else if (type
== ISL_TOKEN_GE
)
868 isl_seq_combine(bmap
->ineq
[k
], ctx
->one
, left
,
871 else if (type
== ISL_TOKEN_LT
) {
872 isl_seq_combine(bmap
->ineq
[k
], ctx
->negone
, left
,
875 isl_int_sub_ui(bmap
->ineq
[k
][0], bmap
->ineq
[k
][0], 1);
876 } else if (type
== ISL_TOKEN_GT
) {
877 isl_seq_combine(bmap
->ineq
[k
], ctx
->one
, left
,
880 isl_int_sub_ui(bmap
->ineq
[k
][0], bmap
->ineq
[k
][0], 1);
882 isl_seq_combine(bmap
->ineq
[k
], ctx
->one
, left
,
885 isl_basic_map_inequality_to_equality(bmap
, k
);
890 isl_basic_map_free(bmap
);
894 static int is_comparator(struct isl_token
*tok
)
911 static struct isl_basic_map
*add_constraint(struct isl_stream
*s
,
912 struct vars
*v
, struct isl_basic_map
*bmap
)
915 struct isl_token
*tok
= NULL
;
916 struct isl_mat
*aff1
= NULL
, *aff2
= NULL
;
918 tok
= isl_stream_next_token(s
);
921 if (tok
->type
== ISL_TOKEN_EXISTS
) {
923 return add_exists(s
, v
, bmap
);
925 isl_stream_push_token(s
, tok
);
928 bmap
= isl_basic_map_cow(bmap
);
930 aff1
= accept_affine_list(s
, v
);
933 tok
= isl_stream_next_token(s
);
934 if (!is_comparator(tok
)) {
935 isl_stream_error(s
, tok
, "missing operator");
937 isl_stream_push_token(s
, tok
);
942 aff2
= accept_affine_list(s
, v
);
946 aff1
= isl_mat_add_zero_cols(aff1
, aff2
->n_col
- aff1
->n_col
);
949 bmap
= add_divs(bmap
, v
);
950 bmap
= isl_basic_map_extend_constraints(bmap
, 0,
951 aff1
->n_row
* aff2
->n_row
);
952 for (i
= 0; i
< aff1
->n_row
; ++i
)
953 for (j
= 0; j
< aff2
->n_row
; ++j
)
954 bmap
= construct_constraint(bmap
, tok
->type
,
955 aff1
->row
[i
], aff2
->row
[j
]);
960 tok
= isl_stream_next_token(s
);
961 if (!is_comparator(tok
)) {
963 isl_stream_push_token(s
, tok
);
975 isl_basic_map_free(bmap
);
979 static struct isl_basic_map
*add_constraints(struct isl_stream
*s
,
980 struct vars
*v
, struct isl_basic_map
*bmap
)
982 struct isl_token
*tok
;
985 bmap
= add_constraint(s
, v
, bmap
);
988 tok
= isl_stream_next_token(s
);
990 isl_stream_error(s
, NULL
, "unexpected EOF");
993 if (tok
->type
!= ISL_TOKEN_AND
)
997 isl_stream_push_token(s
, tok
);
1002 isl_token_free(tok
);
1003 isl_basic_map_free(bmap
);
1007 static isl_map
*read_constraint(struct isl_stream
*s
,
1008 struct vars
*v
, __isl_take isl_dim
*dim
)
1010 struct isl_basic_map
*bmap
;
1013 bmap
= isl_basic_map_alloc_dim(dim
, 0, 0, 0);
1017 bmap
= add_constraint(s
, v
, bmap
);
1018 bmap
= isl_basic_map_simplify(bmap
);
1019 bmap
= isl_basic_map_finalize(bmap
);
1021 vars_drop(v
, v
->n
- n
);
1023 return isl_map_from_basic_map(bmap
);
1025 isl_basic_map_free(bmap
);
1029 static struct isl_map
*read_disjuncts(struct isl_stream
*s
,
1030 struct vars
*v
, __isl_take isl_dim
*dim
);
1032 static __isl_give isl_map
*read_conjunct(struct isl_stream
*s
,
1033 struct vars
*v
, __isl_take isl_dim
*dim
)
1037 if (isl_stream_eat_if_available(s
, '(')) {
1038 map
= read_disjuncts(s
, v
, dim
);
1039 if (isl_stream_eat(s
, ')'))
1044 return read_constraint(s
, v
, dim
);
1050 static __isl_give isl_map
*read_conjuncts(struct isl_stream
*s
,
1051 struct vars
*v
, __isl_take isl_dim
*dim
)
1055 map
= read_conjunct(s
, v
, isl_dim_copy(dim
));
1056 while (isl_stream_eat_if_available(s
, ISL_TOKEN_AND
)) {
1059 map_i
= read_conjunct(s
, v
, isl_dim_copy(dim
));
1060 map
= isl_map_intersect(map
, map_i
);
1067 static struct isl_map
*read_disjuncts(struct isl_stream
*s
,
1068 struct vars
*v
, __isl_take isl_dim
*dim
)
1070 struct isl_map
*map
;
1072 if (isl_stream_next_token_is(s
, '}'))
1073 return isl_map_universe(dim
);
1075 map
= read_conjuncts(s
, v
, isl_dim_copy(dim
));
1076 while (isl_stream_eat_if_available(s
, ISL_TOKEN_OR
)) {
1079 map_i
= read_conjuncts(s
, v
, isl_dim_copy(dim
));
1080 map
= isl_map_union(map
, map_i
);
1087 static int polylib_pos_to_isl_pos(__isl_keep isl_basic_map
*bmap
, int pos
)
1089 if (pos
< isl_basic_map_dim(bmap
, isl_dim_out
))
1090 return 1 + isl_basic_map_dim(bmap
, isl_dim_param
) +
1091 isl_basic_map_dim(bmap
, isl_dim_in
) + pos
;
1092 pos
-= isl_basic_map_dim(bmap
, isl_dim_out
);
1094 if (pos
< isl_basic_map_dim(bmap
, isl_dim_in
))
1095 return 1 + isl_basic_map_dim(bmap
, isl_dim_param
) + pos
;
1096 pos
-= isl_basic_map_dim(bmap
, isl_dim_in
);
1098 if (pos
< isl_basic_map_dim(bmap
, isl_dim_div
))
1099 return 1 + isl_basic_map_dim(bmap
, isl_dim_param
) +
1100 isl_basic_map_dim(bmap
, isl_dim_in
) +
1101 isl_basic_map_dim(bmap
, isl_dim_out
) + pos
;
1102 pos
-= isl_basic_map_dim(bmap
, isl_dim_div
);
1104 if (pos
< isl_basic_map_dim(bmap
, isl_dim_param
))
1110 static __isl_give isl_basic_map
*basic_map_read_polylib_constraint(
1111 struct isl_stream
*s
, __isl_take isl_basic_map
*bmap
)
1114 struct isl_token
*tok
;
1124 nparam
= isl_basic_map_dim(bmap
, isl_dim_param
);
1125 dim
= isl_basic_map_dim(bmap
, isl_dim_out
);
1127 tok
= isl_stream_next_token(s
);
1128 if (!tok
|| tok
->type
!= ISL_TOKEN_VALUE
) {
1129 isl_stream_error(s
, tok
, "expecting coefficient");
1131 isl_stream_push_token(s
, tok
);
1134 if (!tok
->on_new_line
) {
1135 isl_stream_error(s
, tok
, "coefficient should appear on new line");
1136 isl_stream_push_token(s
, tok
);
1140 type
= isl_int_get_si(tok
->u
.v
);
1141 isl_token_free(tok
);
1143 isl_assert(s
->ctx
, type
== 0 || type
== 1, goto error
);
1145 k
= isl_basic_map_alloc_equality(bmap
);
1148 k
= isl_basic_map_alloc_inequality(bmap
);
1154 for (j
= 0; j
< 1 + isl_basic_map_total_dim(bmap
); ++j
) {
1156 tok
= isl_stream_next_token(s
);
1157 if (!tok
|| tok
->type
!= ISL_TOKEN_VALUE
) {
1158 isl_stream_error(s
, tok
, "expecting coefficient");
1160 isl_stream_push_token(s
, tok
);
1163 if (tok
->on_new_line
) {
1164 isl_stream_error(s
, tok
,
1165 "coefficient should not appear on new line");
1166 isl_stream_push_token(s
, tok
);
1169 pos
= polylib_pos_to_isl_pos(bmap
, j
);
1170 isl_int_set(c
[pos
], tok
->u
.v
);
1171 isl_token_free(tok
);
1176 isl_basic_map_free(bmap
);
1180 static __isl_give isl_basic_map
*basic_map_read_polylib(struct isl_stream
*s
,
1184 struct isl_token
*tok
;
1185 struct isl_token
*tok2
;
1188 unsigned in
= 0, out
, local
= 0;
1189 struct isl_basic_map
*bmap
= NULL
;
1194 tok
= isl_stream_next_token(s
);
1196 isl_stream_error(s
, NULL
, "unexpected EOF");
1199 tok2
= isl_stream_next_token(s
);
1201 isl_token_free(tok
);
1202 isl_stream_error(s
, NULL
, "unexpected EOF");
1205 if (tok
->type
!= ISL_TOKEN_VALUE
|| tok2
->type
!= ISL_TOKEN_VALUE
) {
1206 isl_stream_push_token(s
, tok2
);
1207 isl_stream_push_token(s
, tok
);
1208 isl_stream_error(s
, NULL
,
1209 "expecting constraint matrix dimensions");
1212 n_row
= isl_int_get_si(tok
->u
.v
);
1213 n_col
= isl_int_get_si(tok2
->u
.v
);
1214 on_new_line
= tok2
->on_new_line
;
1215 isl_token_free(tok2
);
1216 isl_token_free(tok
);
1217 isl_assert(s
->ctx
, !on_new_line
, return NULL
);
1218 isl_assert(s
->ctx
, n_row
>= 0, return NULL
);
1219 isl_assert(s
->ctx
, n_col
>= 2 + nparam
, return NULL
);
1220 tok
= isl_stream_next_token_on_same_line(s
);
1222 if (tok
->type
!= ISL_TOKEN_VALUE
) {
1223 isl_stream_error(s
, tok
,
1224 "expecting number of output dimensions");
1225 isl_stream_push_token(s
, tok
);
1228 out
= isl_int_get_si(tok
->u
.v
);
1229 isl_token_free(tok
);
1231 tok
= isl_stream_next_token_on_same_line(s
);
1232 if (!tok
|| tok
->type
!= ISL_TOKEN_VALUE
) {
1233 isl_stream_error(s
, tok
,
1234 "expecting number of input dimensions");
1236 isl_stream_push_token(s
, tok
);
1239 in
= isl_int_get_si(tok
->u
.v
);
1240 isl_token_free(tok
);
1242 tok
= isl_stream_next_token_on_same_line(s
);
1243 if (!tok
|| tok
->type
!= ISL_TOKEN_VALUE
) {
1244 isl_stream_error(s
, tok
,
1245 "expecting number of existentials");
1247 isl_stream_push_token(s
, tok
);
1250 local
= isl_int_get_si(tok
->u
.v
);
1251 isl_token_free(tok
);
1253 tok
= isl_stream_next_token_on_same_line(s
);
1254 if (!tok
|| tok
->type
!= ISL_TOKEN_VALUE
) {
1255 isl_stream_error(s
, tok
,
1256 "expecting number of parameters");
1258 isl_stream_push_token(s
, tok
);
1261 nparam
= isl_int_get_si(tok
->u
.v
);
1262 isl_token_free(tok
);
1263 if (n_col
!= 1 + out
+ in
+ local
+ nparam
+ 1) {
1264 isl_stream_error(s
, NULL
,
1265 "dimensions don't match");
1269 out
= n_col
- 2 - nparam
;
1270 bmap
= isl_basic_map_alloc(s
->ctx
, nparam
, in
, out
, local
, n_row
, n_row
);
1274 for (i
= 0; i
< local
; ++i
) {
1275 int k
= isl_basic_map_alloc_div(bmap
);
1278 isl_seq_clr(bmap
->div
[k
], 1 + 1 + nparam
+ in
+ out
+ local
);
1281 for (i
= 0; i
< n_row
; ++i
)
1282 bmap
= basic_map_read_polylib_constraint(s
, bmap
);
1284 tok
= isl_stream_next_token_on_same_line(s
);
1286 isl_stream_error(s
, tok
, "unexpected extra token on line");
1287 isl_stream_push_token(s
, tok
);
1291 bmap
= isl_basic_map_simplify(bmap
);
1292 bmap
= isl_basic_map_finalize(bmap
);
1295 isl_basic_map_free(bmap
);
1299 static struct isl_map
*map_read_polylib(struct isl_stream
*s
, int nparam
)
1301 struct isl_token
*tok
;
1302 struct isl_token
*tok2
;
1304 struct isl_map
*map
;
1306 tok
= isl_stream_next_token(s
);
1308 isl_stream_error(s
, NULL
, "unexpected EOF");
1311 tok2
= isl_stream_next_token_on_same_line(s
);
1312 if (tok2
&& tok2
->type
== ISL_TOKEN_VALUE
) {
1313 isl_stream_push_token(s
, tok2
);
1314 isl_stream_push_token(s
, tok
);
1315 return isl_map_from_basic_map(basic_map_read_polylib(s
, nparam
));
1318 isl_stream_error(s
, tok2
, "unexpected token");
1319 isl_stream_push_token(s
, tok2
);
1320 isl_stream_push_token(s
, tok
);
1323 n
= isl_int_get_si(tok
->u
.v
);
1324 isl_token_free(tok
);
1326 isl_assert(s
->ctx
, n
>= 1, return NULL
);
1328 map
= isl_map_from_basic_map(basic_map_read_polylib(s
, nparam
));
1330 for (i
= 1; map
&& i
< n
; ++i
)
1331 map
= isl_map_union(map
,
1332 isl_map_from_basic_map(basic_map_read_polylib(s
, nparam
)));
1337 static int optional_power(struct isl_stream
*s
)
1340 struct isl_token
*tok
;
1342 tok
= isl_stream_next_token(s
);
1345 if (tok
->type
!= '^') {
1346 isl_stream_push_token(s
, tok
);
1349 isl_token_free(tok
);
1350 tok
= isl_stream_next_token(s
);
1351 if (!tok
|| tok
->type
!= ISL_TOKEN_VALUE
) {
1352 isl_stream_error(s
, tok
, "expecting exponent");
1354 isl_stream_push_token(s
, tok
);
1357 pow
= isl_int_get_si(tok
->u
.v
);
1358 isl_token_free(tok
);
1362 static __isl_give isl_div
*read_div(struct isl_stream
*s
,
1363 __isl_take isl_dim
*dim
, struct vars
*v
)
1366 isl_basic_map
*bmap
;
1369 bmap
= isl_basic_map_universe(dim
);
1371 if (vars_add_anon(v
) < 0)
1373 if (read_div_definition(s
, v
) < 0)
1375 bmap
= add_divs(bmap
, v
);
1376 bmap
= isl_basic_map_order_divs(bmap
);
1379 vars_drop(v
, v
->n
- n
);
1381 return isl_basic_map_div(bmap
, bmap
->n_div
- 1);
1383 isl_basic_map_free(bmap
);
1387 static __isl_give isl_qpolynomial
*read_term(struct isl_stream
*s
,
1388 __isl_keep isl_basic_map
*bmap
, struct vars
*v
);
1390 static __isl_give isl_qpolynomial
*read_factor(struct isl_stream
*s
,
1391 __isl_keep isl_basic_map
*bmap
, struct vars
*v
)
1393 struct isl_qpolynomial
*qp
;
1394 struct isl_token
*tok
;
1396 tok
= isl_stream_next_token(s
);
1398 isl_stream_error(s
, NULL
, "unexpected EOF");
1401 if (tok
->type
== '(') {
1404 isl_token_free(tok
);
1405 qp
= read_term(s
, bmap
, v
);
1408 if (isl_stream_eat(s
, ')'))
1410 pow
= optional_power(s
);
1411 qp
= isl_qpolynomial_pow(qp
, pow
);
1412 } else if (tok
->type
== ISL_TOKEN_VALUE
) {
1413 struct isl_token
*tok2
;
1414 tok2
= isl_stream_next_token(s
);
1415 if (tok2
&& tok2
->type
== '/') {
1416 isl_token_free(tok2
);
1417 tok2
= isl_stream_next_token(s
);
1418 if (!tok2
|| tok2
->type
!= ISL_TOKEN_VALUE
) {
1419 isl_stream_error(s
, tok2
, "expected denominator");
1420 isl_token_free(tok
);
1421 isl_token_free(tok2
);
1424 qp
= isl_qpolynomial_rat_cst(isl_basic_map_get_dim(bmap
),
1425 tok
->u
.v
, tok2
->u
.v
);
1426 isl_token_free(tok2
);
1428 isl_stream_push_token(s
, tok2
);
1429 qp
= isl_qpolynomial_cst(isl_basic_map_get_dim(bmap
),
1432 isl_token_free(tok
);
1433 } else if (tok
->type
== ISL_TOKEN_INFTY
) {
1434 isl_token_free(tok
);
1435 qp
= isl_qpolynomial_infty(isl_basic_map_get_dim(bmap
));
1436 } else if (tok
->type
== ISL_TOKEN_NAN
) {
1437 isl_token_free(tok
);
1438 qp
= isl_qpolynomial_nan(isl_basic_map_get_dim(bmap
));
1439 } else if (tok
->type
== ISL_TOKEN_IDENT
) {
1441 int pos
= vars_pos(v
, tok
->u
.s
, -1);
1444 isl_token_free(tok
);
1448 vars_drop(v
, v
->n
- n
);
1449 isl_stream_error(s
, tok
, "unknown identifier");
1450 isl_token_free(tok
);
1453 isl_token_free(tok
);
1454 pow
= optional_power(s
);
1455 qp
= isl_qpolynomial_var_pow(isl_basic_map_get_dim(bmap
), pos
, pow
);
1456 } else if (tok
->type
== '[') {
1460 isl_stream_push_token(s
, tok
);
1461 div
= read_div(s
, isl_basic_map_get_dim(bmap
), v
);
1462 pow
= optional_power(s
);
1463 qp
= isl_qpolynomial_div_pow(div
, pow
);
1464 } else if (tok
->type
== '-') {
1465 struct isl_qpolynomial
*qp2
;
1467 isl_token_free(tok
);
1468 qp
= isl_qpolynomial_cst(isl_basic_map_get_dim(bmap
),
1470 qp2
= read_factor(s
, bmap
, v
);
1471 qp
= isl_qpolynomial_mul(qp
, qp2
);
1473 isl_stream_error(s
, tok
, "unexpected isl_token");
1474 isl_stream_push_token(s
, tok
);
1478 if (isl_stream_eat_if_available(s
, '*') ||
1479 isl_stream_next_token_is(s
, ISL_TOKEN_IDENT
)) {
1480 struct isl_qpolynomial
*qp2
;
1482 qp2
= read_factor(s
, bmap
, v
);
1483 qp
= isl_qpolynomial_mul(qp
, qp2
);
1488 isl_qpolynomial_free(qp
);
1492 static __isl_give isl_qpolynomial
*read_term(struct isl_stream
*s
,
1493 __isl_keep isl_basic_map
*bmap
, struct vars
*v
)
1495 struct isl_token
*tok
;
1496 struct isl_qpolynomial
*qp
;
1498 qp
= read_factor(s
, bmap
, v
);
1501 tok
= isl_stream_next_token(s
);
1505 if (tok
->type
== '+') {
1506 struct isl_qpolynomial
*qp2
;
1508 isl_token_free(tok
);
1509 qp2
= read_factor(s
, bmap
, v
);
1510 qp
= isl_qpolynomial_add(qp
, qp2
);
1511 } else if (tok
->type
== '-') {
1512 struct isl_qpolynomial
*qp2
;
1514 isl_token_free(tok
);
1515 qp2
= read_factor(s
, bmap
, v
);
1516 qp
= isl_qpolynomial_sub(qp
, qp2
);
1517 } else if (tok
->type
== ISL_TOKEN_VALUE
&&
1518 isl_int_is_neg(tok
->u
.v
)) {
1519 struct isl_qpolynomial
*qp2
;
1521 isl_stream_push_token(s
, tok
);
1522 qp2
= read_factor(s
, bmap
, v
);
1523 qp
= isl_qpolynomial_add(qp
, qp2
);
1525 isl_stream_push_token(s
, tok
);
1533 static __isl_give isl_map
*read_optional_disjuncts(struct isl_stream
*s
,
1534 __isl_take isl_basic_map
*bmap
, struct vars
*v
)
1536 struct isl_token
*tok
;
1537 struct isl_map
*map
;
1539 tok
= isl_stream_next_token(s
);
1541 isl_stream_error(s
, NULL
, "unexpected EOF");
1544 map
= isl_map_from_basic_map(isl_basic_map_copy(bmap
));
1545 if (tok
->type
== ':' ||
1546 (tok
->type
== ISL_TOKEN_OR
&& !strcmp(tok
->u
.s
, "|"))) {
1547 isl_token_free(tok
);
1548 map
= isl_map_intersect(map
,
1549 read_disjuncts(s
, v
, isl_basic_map_get_dim(bmap
)));
1551 isl_stream_push_token(s
, tok
);
1553 isl_basic_map_free(bmap
);
1557 isl_basic_map_free(bmap
);
1561 static struct isl_obj
obj_read_poly(struct isl_stream
*s
,
1562 __isl_take isl_basic_map
*bmap
, struct vars
*v
, int n
)
1564 struct isl_obj obj
= { isl_obj_pw_qpolynomial
, NULL
};
1565 struct isl_pw_qpolynomial
*pwqp
;
1566 struct isl_qpolynomial
*qp
;
1567 struct isl_map
*map
;
1568 struct isl_set
*set
;
1570 qp
= read_term(s
, bmap
, v
);
1571 map
= read_optional_disjuncts(s
, bmap
, v
);
1572 set
= isl_map_range(map
);
1574 pwqp
= isl_pw_qpolynomial_alloc(set
, qp
);
1576 vars_drop(v
, v
->n
- n
);
1582 static struct isl_obj
obj_read_poly_or_fold(struct isl_stream
*s
,
1583 __isl_take isl_basic_map
*bmap
, struct vars
*v
, int n
)
1585 struct isl_obj obj
= { isl_obj_pw_qpolynomial_fold
, NULL
};
1586 struct isl_obj obj_p
;
1587 isl_qpolynomial
*qp
;
1588 isl_qpolynomial_fold
*fold
= NULL
;
1589 isl_pw_qpolynomial_fold
*pwf
;
1593 if (!isl_stream_eat_if_available(s
, ISL_TOKEN_MAX
))
1594 return obj_read_poly(s
, bmap
, v
, n
);
1596 if (isl_stream_eat(s
, '('))
1599 qp
= read_term(s
, bmap
, v
);
1600 fold
= isl_qpolynomial_fold_alloc(isl_fold_max
, qp
);
1602 while (isl_stream_eat_if_available(s
, ',')) {
1603 isl_qpolynomial_fold
*fold_i
;
1604 qp
= read_term(s
, bmap
, v
);
1605 fold_i
= isl_qpolynomial_fold_alloc(isl_fold_max
, qp
);
1606 fold
= isl_qpolynomial_fold_fold(fold
, fold_i
);
1609 if (isl_stream_eat(s
, ')'))
1612 map
= read_optional_disjuncts(s
, bmap
, v
);
1613 set
= isl_map_range(map
);
1614 pwf
= isl_pw_qpolynomial_fold_alloc(isl_fold_max
, set
, fold
);
1616 vars_drop(v
, v
->n
- n
);
1621 isl_basic_map_free(bmap
);
1622 isl_qpolynomial_fold_free(fold
);
1623 obj
.type
= isl_obj_none
;
1627 static struct isl_obj
obj_read_body(struct isl_stream
*s
,
1628 __isl_take isl_basic_map
*bmap
, struct vars
*v
)
1630 struct isl_map
*map
= NULL
;
1631 struct isl_token
*tok
;
1632 struct isl_obj obj
= { isl_obj_set
, NULL
};
1635 if (!next_is_tuple(s
))
1636 return obj_read_poly_or_fold(s
, bmap
, v
, n
);
1638 bmap
= read_tuple(s
, bmap
, isl_dim_in
, v
);
1641 tok
= isl_stream_next_token(s
);
1642 if (tok
&& tok
->type
== ISL_TOKEN_TO
) {
1643 obj
.type
= isl_obj_map
;
1644 isl_token_free(tok
);
1645 if (!next_is_tuple(s
)) {
1646 bmap
= isl_basic_map_reverse(bmap
);
1647 return obj_read_poly_or_fold(s
, bmap
, v
, n
);
1649 bmap
= read_tuple(s
, bmap
, isl_dim_out
, v
);
1653 bmap
= isl_basic_map_reverse(bmap
);
1655 isl_stream_push_token(s
, tok
);
1658 map
= read_optional_disjuncts(s
, bmap
, v
);
1660 vars_drop(v
, v
->n
- n
);
1665 isl_basic_map_free(bmap
);
1666 obj
.type
= isl_obj_none
;
1670 static struct isl_obj
to_union(isl_ctx
*ctx
, struct isl_obj obj
)
1672 if (obj
.type
== isl_obj_map
) {
1673 obj
.v
= isl_union_map_from_map(obj
.v
);
1674 obj
.type
= isl_obj_union_map
;
1675 } else if (obj
.type
== isl_obj_set
) {
1676 obj
.v
= isl_union_set_from_set(obj
.v
);
1677 obj
.type
= isl_obj_union_set
;
1678 } else if (obj
.type
== isl_obj_pw_qpolynomial
) {
1679 obj
.v
= isl_union_pw_qpolynomial_from_pw_qpolynomial(obj
.v
);
1680 obj
.type
= isl_obj_union_pw_qpolynomial
;
1681 } else if (obj
.type
== isl_obj_pw_qpolynomial_fold
) {
1682 obj
.v
= isl_union_pw_qpolynomial_fold_from_pw_qpolynomial_fold(obj
.v
);
1683 obj
.type
= isl_obj_union_pw_qpolynomial_fold
;
1685 isl_assert(ctx
, 0, goto error
);
1688 obj
.type
->free(obj
.v
);
1689 obj
.type
= isl_obj_none
;
1693 static struct isl_obj
obj_add(struct isl_ctx
*ctx
,
1694 struct isl_obj obj1
, struct isl_obj obj2
)
1696 if (obj1
.type
== isl_obj_set
&& obj2
.type
== isl_obj_union_set
)
1697 obj1
= to_union(ctx
, obj1
);
1698 if (obj1
.type
== isl_obj_union_set
&& obj2
.type
== isl_obj_set
)
1699 obj2
= to_union(ctx
, obj2
);
1700 if (obj1
.type
== isl_obj_map
&& obj2
.type
== isl_obj_union_map
)
1701 obj1
= to_union(ctx
, obj1
);
1702 if (obj1
.type
== isl_obj_union_map
&& obj2
.type
== isl_obj_map
)
1703 obj2
= to_union(ctx
, obj2
);
1704 if (obj1
.type
== isl_obj_pw_qpolynomial
&&
1705 obj2
.type
== isl_obj_union_pw_qpolynomial
)
1706 obj1
= to_union(ctx
, obj1
);
1707 if (obj1
.type
== isl_obj_union_pw_qpolynomial
&&
1708 obj2
.type
== isl_obj_pw_qpolynomial
)
1709 obj2
= to_union(ctx
, obj2
);
1710 if (obj1
.type
== isl_obj_pw_qpolynomial_fold
&&
1711 obj2
.type
== isl_obj_union_pw_qpolynomial_fold
)
1712 obj1
= to_union(ctx
, obj1
);
1713 if (obj1
.type
== isl_obj_union_pw_qpolynomial_fold
&&
1714 obj2
.type
== isl_obj_pw_qpolynomial_fold
)
1715 obj2
= to_union(ctx
, obj2
);
1716 isl_assert(ctx
, obj1
.type
== obj2
.type
, goto error
);
1717 if (obj1
.type
== isl_obj_map
&& !isl_map_has_equal_dim(obj1
.v
, obj2
.v
)) {
1718 obj1
= to_union(ctx
, obj1
);
1719 obj2
= to_union(ctx
, obj2
);
1721 if (obj1
.type
== isl_obj_set
&& !isl_set_has_equal_dim(obj1
.v
, obj2
.v
)) {
1722 obj1
= to_union(ctx
, obj1
);
1723 obj2
= to_union(ctx
, obj2
);
1725 if (obj1
.type
== isl_obj_pw_qpolynomial
&&
1726 !isl_pw_qpolynomial_has_equal_dim(obj1
.v
, obj2
.v
)) {
1727 obj1
= to_union(ctx
, obj1
);
1728 obj2
= to_union(ctx
, obj2
);
1730 if (obj1
.type
== isl_obj_pw_qpolynomial_fold
&&
1731 !isl_pw_qpolynomial_fold_has_equal_dim(obj1
.v
, obj2
.v
)) {
1732 obj1
= to_union(ctx
, obj1
);
1733 obj2
= to_union(ctx
, obj2
);
1735 obj1
.v
= obj1
.type
->add(obj1
.v
, obj2
.v
);
1738 obj1
.type
->free(obj1
.v
);
1739 obj2
.type
->free(obj2
.v
);
1740 obj1
.type
= isl_obj_none
;
1745 static struct isl_obj
obj_read(struct isl_stream
*s
, int nparam
)
1747 isl_basic_map
*bmap
= NULL
;
1748 struct isl_token
*tok
;
1749 struct vars
*v
= NULL
;
1750 struct isl_obj obj
= { isl_obj_set
, NULL
};
1752 tok
= isl_stream_next_token(s
);
1754 isl_stream_error(s
, NULL
, "unexpected EOF");
1757 if (tok
->type
== ISL_TOKEN_VALUE
) {
1758 struct isl_token
*tok2
;
1759 struct isl_map
*map
;
1761 tok2
= isl_stream_next_token(s
);
1762 if (!tok2
|| tok2
->type
!= ISL_TOKEN_VALUE
||
1763 isl_int_is_neg(tok2
->u
.v
)) {
1765 isl_stream_push_token(s
, tok2
);
1766 obj
.type
= isl_obj_int
;
1767 obj
.v
= isl_int_obj_alloc(s
->ctx
, tok
->u
.v
);
1768 isl_token_free(tok
);
1771 isl_stream_push_token(s
, tok2
);
1772 isl_stream_push_token(s
, tok
);
1773 map
= map_read_polylib(s
, nparam
);
1776 if (isl_map_dim(map
, isl_dim_in
) > 0)
1777 obj
.type
= isl_obj_map
;
1781 v
= vars_new(s
->ctx
);
1783 isl_stream_push_token(s
, tok
);
1786 bmap
= isl_basic_map_alloc(s
->ctx
, 0, 0, 0, 0, 0, 0);
1787 if (tok
->type
== '[') {
1788 isl_stream_push_token(s
, tok
);
1789 bmap
= read_tuple(s
, bmap
, isl_dim_param
, v
);
1793 isl_assert(s
->ctx
, nparam
== v
->n
, goto error
);
1794 tok
= isl_stream_next_token(s
);
1795 if (!tok
|| tok
->type
!= ISL_TOKEN_TO
) {
1796 isl_stream_error(s
, tok
, "expecting '->'");
1798 isl_stream_push_token(s
, tok
);
1801 isl_token_free(tok
);
1802 tok
= isl_stream_next_token(s
);
1803 } else if (nparam
> 0)
1804 bmap
= isl_basic_map_add(bmap
, isl_dim_param
, nparam
);
1805 if (!tok
|| tok
->type
!= '{') {
1806 isl_stream_error(s
, tok
, "expecting '{'");
1808 isl_stream_push_token(s
, tok
);
1811 isl_token_free(tok
);
1813 tok
= isl_stream_next_token(s
);
1816 else if (tok
->type
== ISL_TOKEN_IDENT
&& !strcmp(tok
->u
.s
, "Sym")) {
1817 isl_token_free(tok
);
1818 if (isl_stream_eat(s
, '='))
1820 bmap
= read_tuple(s
, bmap
, isl_dim_param
, v
);
1824 isl_assert(s
->ctx
, nparam
== v
->n
, goto error
);
1825 } else if (tok
->type
== '}') {
1826 obj
.type
= isl_obj_union_set
;
1827 obj
.v
= isl_union_set_empty(isl_basic_map_get_dim(bmap
));
1828 isl_token_free(tok
);
1831 isl_stream_push_token(s
, tok
);
1836 o
= obj_read_body(s
, isl_basic_map_copy(bmap
), v
);
1837 if (o
.type
== isl_obj_none
|| !o
.v
)
1842 obj
= obj_add(s
->ctx
, obj
, o
);
1843 if (obj
.type
== isl_obj_none
|| !obj
.v
)
1846 tok
= isl_stream_next_token(s
);
1847 if (!tok
|| tok
->type
!= ';')
1849 isl_token_free(tok
);
1852 if (tok
&& tok
->type
== '}') {
1853 isl_token_free(tok
);
1855 isl_stream_error(s
, tok
, "unexpected isl_token");
1857 isl_token_free(tok
);
1862 isl_basic_map_free(bmap
);
1866 isl_basic_map_free(bmap
);
1867 obj
.type
->free(obj
.v
);
1874 struct isl_obj
isl_stream_read_obj(struct isl_stream
*s
)
1876 return obj_read(s
, -1);
1879 __isl_give isl_map
*isl_stream_read_map(struct isl_stream
*s
, int nparam
)
1882 struct isl_map
*map
;
1884 obj
= obj_read(s
, nparam
);
1886 isl_assert(s
->ctx
, obj
.type
== isl_obj_map
||
1887 obj
.type
== isl_obj_set
, goto error
);
1891 obj
.type
->free(obj
.v
);
1895 __isl_give isl_set
*isl_stream_read_set(struct isl_stream
*s
, int nparam
)
1898 struct isl_set
*set
;
1900 obj
= obj_read(s
, nparam
);
1902 isl_assert(s
->ctx
, obj
.type
== isl_obj_set
, goto error
);
1906 obj
.type
->free(obj
.v
);
1910 __isl_give isl_union_map
*isl_stream_read_union_map(struct isl_stream
*s
)
1913 isl_union_map
*umap
;
1915 obj
= obj_read(s
, -1);
1916 if (obj
.type
== isl_obj_map
) {
1917 obj
.type
= isl_obj_union_map
;
1918 obj
.v
= isl_union_map_from_map(obj
.v
);
1920 if (obj
.type
== isl_obj_set
) {
1921 obj
.type
= isl_obj_union_set
;
1922 obj
.v
= isl_union_set_from_set(obj
.v
);
1925 isl_assert(s
->ctx
, obj
.type
== isl_obj_union_map
||
1926 obj
.type
== isl_obj_union_set
, goto error
);
1930 obj
.type
->free(obj
.v
);
1934 __isl_give isl_union_set
*isl_stream_read_union_set(struct isl_stream
*s
)
1937 isl_union_set
*uset
;
1939 obj
= obj_read(s
, -1);
1940 if (obj
.type
== isl_obj_set
) {
1941 obj
.type
= isl_obj_union_set
;
1942 obj
.v
= isl_union_set_from_set(obj
.v
);
1945 isl_assert(s
->ctx
, obj
.type
== isl_obj_union_set
, goto error
);
1949 obj
.type
->free(obj
.v
);
1953 static struct isl_basic_map
*basic_map_read(struct isl_stream
*s
, int nparam
)
1956 struct isl_map
*map
;
1957 struct isl_basic_map
*bmap
;
1959 obj
= obj_read(s
, nparam
);
1964 isl_assert(map
->ctx
, map
->n
<= 1, goto error
);
1967 bmap
= isl_basic_map_empty_like_map(map
);
1969 bmap
= isl_basic_map_copy(map
->p
[0]);
1979 __isl_give isl_basic_map
*isl_basic_map_read_from_file(isl_ctx
*ctx
,
1980 FILE *input
, int nparam
)
1982 struct isl_basic_map
*bmap
;
1983 struct isl_stream
*s
= isl_stream_new_file(ctx
, input
);
1986 bmap
= basic_map_read(s
, nparam
);
1991 __isl_give isl_basic_set
*isl_basic_set_read_from_file(isl_ctx
*ctx
,
1992 FILE *input
, int nparam
)
1994 struct isl_basic_map
*bmap
;
1995 bmap
= isl_basic_map_read_from_file(ctx
, input
, nparam
);
1998 isl_assert(ctx
, isl_basic_map_n_in(bmap
) == 0, goto error
);
1999 return (struct isl_basic_set
*)bmap
;
2001 isl_basic_map_free(bmap
);
2005 struct isl_basic_map
*isl_basic_map_read_from_str(struct isl_ctx
*ctx
,
2006 const char *str
, int nparam
)
2008 struct isl_basic_map
*bmap
;
2009 struct isl_stream
*s
= isl_stream_new_str(ctx
, str
);
2012 bmap
= basic_map_read(s
, nparam
);
2017 struct isl_basic_set
*isl_basic_set_read_from_str(struct isl_ctx
*ctx
,
2018 const char *str
, int nparam
)
2020 struct isl_basic_map
*bmap
;
2021 bmap
= isl_basic_map_read_from_str(ctx
, str
, nparam
);
2024 isl_assert(ctx
, isl_basic_map_n_in(bmap
) == 0, goto error
);
2025 return (struct isl_basic_set
*)bmap
;
2027 isl_basic_map_free(bmap
);
2031 __isl_give isl_map
*isl_map_read_from_file(struct isl_ctx
*ctx
,
2032 FILE *input
, int nparam
)
2034 struct isl_map
*map
;
2035 struct isl_stream
*s
= isl_stream_new_file(ctx
, input
);
2038 map
= isl_stream_read_map(s
, nparam
);
2043 __isl_give isl_map
*isl_map_read_from_str(struct isl_ctx
*ctx
,
2044 const char *str
, int nparam
)
2046 struct isl_map
*map
;
2047 struct isl_stream
*s
= isl_stream_new_str(ctx
, str
);
2050 map
= isl_stream_read_map(s
, nparam
);
2055 __isl_give isl_set
*isl_set_read_from_file(struct isl_ctx
*ctx
,
2056 FILE *input
, int nparam
)
2058 struct isl_map
*map
;
2059 map
= isl_map_read_from_file(ctx
, input
, nparam
);
2062 isl_assert(ctx
, isl_map_n_in(map
) == 0, goto error
);
2063 return (struct isl_set
*)map
;
2069 struct isl_set
*isl_set_read_from_str(struct isl_ctx
*ctx
,
2070 const char *str
, int nparam
)
2072 struct isl_map
*map
;
2073 map
= isl_map_read_from_str(ctx
, str
, nparam
);
2076 isl_assert(ctx
, isl_map_n_in(map
) == 0, goto error
);
2077 return (struct isl_set
*)map
;
2083 __isl_give isl_union_map
*isl_union_map_read_from_str(struct isl_ctx
*ctx
,
2086 isl_union_map
*umap
;
2087 struct isl_stream
*s
= isl_stream_new_str(ctx
, str
);
2090 umap
= isl_stream_read_union_map(s
);
2095 __isl_give isl_union_set
*isl_union_set_read_from_str(struct isl_ctx
*ctx
,
2098 isl_union_set
*uset
;
2099 struct isl_stream
*s
= isl_stream_new_str(ctx
, str
);
2102 uset
= isl_stream_read_union_set(s
);
2107 static __isl_give isl_vec
*isl_vec_read_polylib(struct isl_stream
*s
)
2109 struct isl_vec
*vec
= NULL
;
2110 struct isl_token
*tok
;
2114 tok
= isl_stream_next_token(s
);
2115 if (!tok
|| tok
->type
!= ISL_TOKEN_VALUE
) {
2116 isl_stream_error(s
, tok
, "expecting vector length");
2120 size
= isl_int_get_si(tok
->u
.v
);
2121 isl_token_free(tok
);
2123 vec
= isl_vec_alloc(s
->ctx
, size
);
2125 for (j
= 0; j
< size
; ++j
) {
2126 tok
= isl_stream_next_token(s
);
2127 if (!tok
|| tok
->type
!= ISL_TOKEN_VALUE
) {
2128 isl_stream_error(s
, tok
, "expecting constant value");
2131 isl_int_set(vec
->el
[j
], tok
->u
.v
);
2132 isl_token_free(tok
);
2137 isl_token_free(tok
);
2142 static __isl_give isl_vec
*vec_read(struct isl_stream
*s
,
2143 unsigned input_format
)
2145 if (input_format
== ISL_FORMAT_POLYLIB
)
2146 return isl_vec_read_polylib(s
);
2147 isl_assert(s
->ctx
, 0, return NULL
);
2150 struct isl_vec
*isl_vec_read_from_file(struct isl_ctx
*ctx
,
2151 FILE *input
, unsigned input_format
)
2154 struct isl_stream
*s
= isl_stream_new_file(ctx
, input
);
2157 v
= vec_read(s
, input_format
);
2162 __isl_give isl_pw_qpolynomial
*isl_stream_read_pw_qpolynomial(
2163 struct isl_stream
*s
)
2166 struct isl_pw_qpolynomial
*pwqp
;
2168 obj
= obj_read(s
, -1);
2170 isl_assert(s
->ctx
, obj
.type
== isl_obj_pw_qpolynomial
,
2175 obj
.type
->free(obj
.v
);
2179 __isl_give isl_pw_qpolynomial
*isl_pw_qpolynomial_read_from_str(isl_ctx
*ctx
,
2182 isl_pw_qpolynomial
*pwqp
;
2183 struct isl_stream
*s
= isl_stream_new_str(ctx
, str
);
2186 pwqp
= isl_stream_read_pw_qpolynomial(s
);