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
)
396 isl_basic_map
*def
= NULL
;
401 if (vars_add_anon(v
) < 0)
405 vec
= accept_affine(s
, v
);
409 dim
= isl_basic_map_get_dim(bmap
);
410 def
= isl_basic_map_universe(dim
);
411 def
= add_divs(def
, v
);
412 def
= isl_basic_map_extend_constraints(def
, 1, 0);
413 k
= isl_basic_map_alloc_equality(def
);
415 isl_seq_cpy(def
->eq
[k
], vec
->el
, vec
->size
);
416 isl_int_set_si(def
->eq
[k
][1 + n
- 1], -1);
422 vars_drop(v
, v
->n
- n
);
424 def
= isl_basic_map_simplify(def
);
425 def
= isl_basic_map_finalize(def
);
426 bmap
= isl_basic_map_intersect(bmap
, def
);
429 isl_basic_map_free(bmap
);
430 isl_basic_map_free(def
);
434 static __isl_give isl_basic_map
*read_var_list(struct isl_stream
*s
,
435 __isl_take isl_basic_map
*bmap
, enum isl_dim_type type
, struct vars
*v
)
438 struct isl_token
*tok
;
440 while ((tok
= isl_stream_next_token(s
)) != NULL
) {
443 if (tok
->type
== ISL_TOKEN_IDENT
) {
445 int p
= vars_pos(v
, tok
->u
.s
, -1);
452 bmap
= isl_basic_map_add(bmap
, type
, 1);
453 bmap
= set_name(bmap
, type
, i
, v
->v
->name
);
455 } else if (tok
->type
== ISL_TOKEN_IDENT
||
456 tok
->type
== ISL_TOKEN_VALUE
||
459 if (type
== isl_dim_param
) {
460 isl_stream_error(s
, tok
,
461 "expecting unique identifier");
464 isl_stream_push_token(s
, tok
);
466 bmap
= isl_basic_map_add(bmap
, type
, 1);
467 bmap
= read_var_def(s
, bmap
, type
, v
);
471 tok
= isl_stream_next_token(s
);
472 if (!tok
|| tok
->type
!= ',')
479 isl_stream_push_token(s
, tok
);
484 isl_basic_map_free(bmap
);
488 static __isl_give isl_mat
*accept_affine_list(struct isl_stream
*s
,
493 struct isl_token
*tok
= NULL
;
495 vec
= accept_affine(s
, v
);
496 mat
= isl_mat_from_row_vec(vec
);
501 tok
= isl_stream_next_token(s
);
503 isl_stream_error(s
, NULL
, "unexpected EOF");
506 if (tok
->type
!= ',') {
507 isl_stream_push_token(s
, tok
);
512 vec
= accept_affine(s
, v
);
513 mat
= isl_mat_add_zero_cols(mat
, 1 + v
->n
- isl_mat_cols(mat
));
514 mat
= isl_mat_vec_concat(mat
, vec
);
525 static int read_div_definition(struct isl_stream
*s
, struct vars
*v
)
527 struct isl_token
*tok
;
530 struct variable
*var
;
532 if (isl_stream_eat(s
, '['))
535 tok
= isl_stream_next_token(s
);
538 if (tok
->type
== '(') {
542 isl_stream_push_token(s
, tok
);
546 aff
= accept_affine(s
, v
);
550 var
->def
= isl_vec_alloc(s
->ctx
, 2 + v
->n
);
556 isl_seq_cpy(var
->def
->el
+ 1, aff
->el
, aff
->size
);
560 if (seen_paren
&& isl_stream_eat(s
, ')'))
562 if (isl_stream_eat(s
, '/'))
565 tok
= isl_stream_next_token(s
);
568 if (tok
->type
!= ISL_TOKEN_VALUE
) {
569 isl_stream_error(s
, tok
, "expected denominator");
570 isl_stream_push_token(s
, tok
);
573 isl_int_set(var
->def
->el
[0], tok
->u
.v
);
576 if (isl_stream_eat(s
, ']'))
582 static struct isl_basic_map
*add_div_definition(struct isl_stream
*s
,
583 struct vars
*v
, struct isl_basic_map
*bmap
, int k
)
585 struct variable
*var
= v
->v
;
587 if (read_div_definition(s
, v
) < 0)
590 isl_seq_cpy(bmap
->div
[k
], var
->def
->el
, 2 + v
->n
);
592 if (isl_basic_map_add_div_constraints(bmap
, k
) < 0)
597 isl_basic_map_free(bmap
);
601 static struct isl_basic_map
*read_defined_var_list(struct isl_stream
*s
,
602 struct vars
*v
, struct isl_basic_map
*bmap
)
604 struct isl_token
*tok
;
606 while ((tok
= isl_stream_next_token(s
)) != NULL
) {
610 unsigned total
= isl_basic_map_total_dim(bmap
);
612 if (tok
->type
!= ISL_TOKEN_IDENT
)
615 p
= vars_pos(v
, tok
->u
.s
, -1);
619 isl_stream_error(s
, tok
, "expecting unique identifier");
623 bmap
= isl_basic_map_cow(bmap
);
624 bmap
= isl_basic_map_extend_dim(bmap
, isl_dim_copy(bmap
->dim
),
627 if ((k
= isl_basic_map_alloc_div(bmap
)) < 0)
629 isl_seq_clr(bmap
->div
[k
], 1 + 1 + total
);
632 tok
= isl_stream_next_token(s
);
633 if (tok
&& tok
->type
== '=') {
635 bmap
= add_div_definition(s
, v
, bmap
, k
);
636 tok
= isl_stream_next_token(s
);
639 if (!tok
|| tok
->type
!= ',')
645 isl_stream_push_token(s
, tok
);
650 isl_basic_map_free(bmap
);
654 static int next_is_tuple(struct isl_stream
*s
)
656 struct isl_token
*tok
;
659 tok
= isl_stream_next_token(s
);
662 if (tok
->type
== '[') {
663 isl_stream_push_token(s
, tok
);
666 if (tok
->type
!= ISL_TOKEN_IDENT
) {
667 isl_stream_push_token(s
, tok
);
671 is_tuple
= isl_stream_next_token_is(s
, '[');
673 isl_stream_push_token(s
, tok
);
678 static __isl_give isl_basic_map
*read_tuple(struct isl_stream
*s
,
679 __isl_take isl_basic_map
*bmap
, enum isl_dim_type type
, struct vars
*v
);
681 static __isl_give isl_basic_map
*read_nested_tuple(struct isl_stream
*s
,
682 __isl_take isl_basic_map
*bmap
, struct vars
*v
)
684 bmap
= read_tuple(s
, bmap
, isl_dim_in
, v
);
685 if (isl_stream_eat(s
, ISL_TOKEN_TO
))
687 bmap
= read_tuple(s
, bmap
, isl_dim_out
, v
);
688 bmap
= isl_basic_map_from_range(isl_basic_map_wrap(bmap
));
691 isl_basic_map_free(bmap
);
695 static __isl_give isl_basic_map
*read_tuple(struct isl_stream
*s
,
696 __isl_take isl_basic_map
*bmap
, enum isl_dim_type type
, struct vars
*v
)
698 struct isl_token
*tok
;
701 tok
= isl_stream_next_token(s
);
702 if (tok
&& tok
->type
== ISL_TOKEN_IDENT
) {
703 name
= strdup(tok
->u
.s
);
707 tok
= isl_stream_next_token(s
);
709 if (!tok
|| tok
->type
!= '[') {
710 isl_stream_error(s
, tok
, "expecting '['");
714 if (type
!= isl_dim_param
&& next_is_tuple(s
)) {
715 isl_dim
*dim
= isl_basic_map_get_dim(bmap
);
716 int nparam
= isl_dim_size(dim
, isl_dim_param
);
717 int n_in
= isl_dim_size(dim
, isl_dim_in
);
718 isl_basic_map
*nested
;
719 if (type
== isl_dim_out
)
720 dim
= isl_dim_move(dim
, isl_dim_param
, nparam
,
721 isl_dim_in
, 0, n_in
);
722 nested
= isl_basic_map_alloc_dim(dim
, 0, 0, 0);
723 nested
= read_nested_tuple(s
, nested
, v
);
724 if (type
== isl_dim_in
) {
725 nested
= isl_basic_map_reverse(nested
);
726 bmap
= isl_basic_map_intersect(nested
, bmap
);
729 dim
= isl_dim_range(isl_basic_map_get_dim(nested
));
730 dim
= isl_dim_drop(dim
, isl_dim_param
, nparam
, n_in
);
731 dim
= isl_dim_join(isl_basic_map_get_dim(bmap
), dim
);
732 bset
= isl_basic_map_domain(bmap
);
733 nested
= isl_basic_map_reset_dim(nested
, dim
);
734 bmap
= isl_basic_map_intersect_domain(nested
, bset
);
737 bmap
= read_var_list(s
, bmap
, type
, v
);
738 tok
= isl_stream_next_token(s
);
739 if (!tok
|| tok
->type
!= ']') {
740 isl_stream_error(s
, tok
, "expecting ']'");
746 bmap
= isl_basic_map_set_tuple_name(bmap
, type
, name
);
754 isl_basic_map_free(bmap
);
758 static struct isl_basic_map
*add_constraints(struct isl_stream
*s
,
759 struct vars
*v
, struct isl_basic_map
*bmap
);
761 static struct isl_basic_map
*add_exists(struct isl_stream
*s
,
762 struct vars
*v
, struct isl_basic_map
*bmap
)
764 struct isl_token
*tok
;
771 tok
= isl_stream_next_token(s
);
774 if (tok
->type
== '(') {
778 isl_stream_push_token(s
, tok
);
780 bmap
= read_defined_var_list(s
, v
, bmap
);
784 if (isl_stream_eat(s
, ':'))
786 bmap
= add_constraints(s
, v
, bmap
);
787 if (seen_paren
&& isl_stream_eat(s
, ')'))
791 isl_basic_map_free(bmap
);
795 static __isl_give isl_basic_map
*construct_constraint(
796 __isl_take isl_basic_map
*bmap
, enum isl_token_type type
,
797 isl_int
*left
, isl_int
*right
)
805 len
= 1 + isl_basic_map_total_dim(bmap
);
808 k
= isl_basic_map_alloc_inequality(bmap
);
811 if (type
== ISL_TOKEN_LE
)
812 isl_seq_combine(bmap
->ineq
[k
], ctx
->negone
, left
,
815 else if (type
== ISL_TOKEN_GE
)
816 isl_seq_combine(bmap
->ineq
[k
], ctx
->one
, left
,
819 else if (type
== ISL_TOKEN_LT
) {
820 isl_seq_combine(bmap
->ineq
[k
], ctx
->negone
, left
,
823 isl_int_sub_ui(bmap
->ineq
[k
][0], bmap
->ineq
[k
][0], 1);
824 } else if (type
== ISL_TOKEN_GT
) {
825 isl_seq_combine(bmap
->ineq
[k
], ctx
->one
, left
,
828 isl_int_sub_ui(bmap
->ineq
[k
][0], bmap
->ineq
[k
][0], 1);
830 isl_seq_combine(bmap
->ineq
[k
], ctx
->one
, left
,
833 isl_basic_map_inequality_to_equality(bmap
, k
);
838 isl_basic_map_free(bmap
);
842 static int is_comparator(struct isl_token
*tok
)
859 static struct isl_basic_map
*add_constraint(struct isl_stream
*s
,
860 struct vars
*v
, struct isl_basic_map
*bmap
)
863 struct isl_token
*tok
= NULL
;
864 struct isl_mat
*aff1
= NULL
, *aff2
= NULL
;
866 tok
= isl_stream_next_token(s
);
869 if (tok
->type
== ISL_TOKEN_EXISTS
) {
871 return add_exists(s
, v
, bmap
);
873 isl_stream_push_token(s
, tok
);
876 bmap
= isl_basic_map_cow(bmap
);
878 aff1
= accept_affine_list(s
, v
);
881 tok
= isl_stream_next_token(s
);
882 if (!is_comparator(tok
)) {
883 isl_stream_error(s
, tok
, "missing operator");
885 isl_stream_push_token(s
, tok
);
890 aff2
= accept_affine_list(s
, v
);
894 aff1
= isl_mat_add_zero_cols(aff1
, aff2
->n_col
- aff1
->n_col
);
897 bmap
= add_divs(bmap
, v
);
898 bmap
= isl_basic_map_extend_constraints(bmap
, 0,
899 aff1
->n_row
* aff2
->n_row
);
900 for (i
= 0; i
< aff1
->n_row
; ++i
)
901 for (j
= 0; j
< aff2
->n_row
; ++j
)
902 bmap
= construct_constraint(bmap
, tok
->type
,
903 aff1
->row
[i
], aff2
->row
[j
]);
908 tok
= isl_stream_next_token(s
);
909 if (!is_comparator(tok
)) {
911 isl_stream_push_token(s
, tok
);
923 isl_basic_map_free(bmap
);
927 static struct isl_basic_map
*add_constraints(struct isl_stream
*s
,
928 struct vars
*v
, struct isl_basic_map
*bmap
)
930 struct isl_token
*tok
;
933 bmap
= add_constraint(s
, v
, bmap
);
936 tok
= isl_stream_next_token(s
);
938 isl_stream_error(s
, NULL
, "unexpected EOF");
941 if (tok
->type
!= ISL_TOKEN_AND
)
945 isl_stream_push_token(s
, tok
);
951 isl_basic_map_free(bmap
);
955 static struct isl_basic_map
*read_disjunct(struct isl_stream
*s
,
956 struct vars
*v
, __isl_take isl_dim
*dim
)
959 struct isl_token
*tok
;
960 struct isl_basic_map
*bmap
;
962 bmap
= isl_basic_map_alloc_dim(dim
, 0, 0, 0);
966 tok
= isl_stream_next_token(s
);
969 if (tok
->type
== '(') {
973 isl_stream_push_token(s
, tok
);
975 bmap
= add_constraints(s
, v
, bmap
);
976 bmap
= isl_basic_map_simplify(bmap
);
977 bmap
= isl_basic_map_finalize(bmap
);
979 if (seen_paren
&& isl_stream_eat(s
, ')'))
984 isl_basic_map_free(bmap
);
988 static struct isl_map
*read_disjuncts(struct isl_stream
*s
,
989 struct vars
*v
, __isl_take isl_dim
*dim
)
991 struct isl_token
*tok
;
994 tok
= isl_stream_next_token(s
);
996 isl_stream_error(s
, NULL
, "unexpected EOF");
999 if (tok
->type
== '}') {
1000 isl_stream_push_token(s
, tok
);
1001 return isl_map_universe(dim
);
1003 isl_stream_push_token(s
, tok
);
1005 map
= isl_map_empty(isl_dim_copy(dim
));
1007 struct isl_basic_map
*bmap
;
1010 bmap
= read_disjunct(s
, v
, isl_dim_copy(dim
));
1011 map
= isl_map_union(map
, isl_map_from_basic_map(bmap
));
1013 vars_drop(v
, v
->n
- n
);
1015 tok
= isl_stream_next_token(s
);
1016 if (!tok
|| tok
->type
!= ISL_TOKEN_OR
)
1018 isl_token_free(tok
);
1021 isl_stream_push_token(s
, tok
);
1030 static int polylib_pos_to_isl_pos(__isl_keep isl_basic_map
*bmap
, int pos
)
1032 if (pos
< isl_basic_map_dim(bmap
, isl_dim_out
))
1033 return 1 + isl_basic_map_dim(bmap
, isl_dim_param
) +
1034 isl_basic_map_dim(bmap
, isl_dim_in
) + pos
;
1035 pos
-= isl_basic_map_dim(bmap
, isl_dim_out
);
1037 if (pos
< isl_basic_map_dim(bmap
, isl_dim_in
))
1038 return 1 + isl_basic_map_dim(bmap
, isl_dim_param
) + pos
;
1039 pos
-= isl_basic_map_dim(bmap
, isl_dim_in
);
1041 if (pos
< isl_basic_map_dim(bmap
, isl_dim_div
))
1042 return 1 + isl_basic_map_dim(bmap
, isl_dim_param
) +
1043 isl_basic_map_dim(bmap
, isl_dim_in
) +
1044 isl_basic_map_dim(bmap
, isl_dim_out
) + pos
;
1045 pos
-= isl_basic_map_dim(bmap
, isl_dim_div
);
1047 if (pos
< isl_basic_map_dim(bmap
, isl_dim_param
))
1053 static __isl_give isl_basic_map
*basic_map_read_polylib_constraint(
1054 struct isl_stream
*s
, __isl_take isl_basic_map
*bmap
)
1057 struct isl_token
*tok
;
1067 nparam
= isl_basic_map_dim(bmap
, isl_dim_param
);
1068 dim
= isl_basic_map_dim(bmap
, isl_dim_out
);
1070 tok
= isl_stream_next_token(s
);
1071 if (!tok
|| tok
->type
!= ISL_TOKEN_VALUE
) {
1072 isl_stream_error(s
, tok
, "expecting coefficient");
1074 isl_stream_push_token(s
, tok
);
1077 if (!tok
->on_new_line
) {
1078 isl_stream_error(s
, tok
, "coefficient should appear on new line");
1079 isl_stream_push_token(s
, tok
);
1083 type
= isl_int_get_si(tok
->u
.v
);
1084 isl_token_free(tok
);
1086 isl_assert(s
->ctx
, type
== 0 || type
== 1, goto error
);
1088 k
= isl_basic_map_alloc_equality(bmap
);
1091 k
= isl_basic_map_alloc_inequality(bmap
);
1097 for (j
= 0; j
< 1 + isl_basic_map_total_dim(bmap
); ++j
) {
1099 tok
= isl_stream_next_token(s
);
1100 if (!tok
|| tok
->type
!= ISL_TOKEN_VALUE
) {
1101 isl_stream_error(s
, tok
, "expecting coefficient");
1103 isl_stream_push_token(s
, tok
);
1106 if (tok
->on_new_line
) {
1107 isl_stream_error(s
, tok
,
1108 "coefficient should not appear on new line");
1109 isl_stream_push_token(s
, tok
);
1112 pos
= polylib_pos_to_isl_pos(bmap
, j
);
1113 isl_int_set(c
[pos
], tok
->u
.v
);
1114 isl_token_free(tok
);
1119 isl_basic_map_free(bmap
);
1123 static __isl_give isl_basic_map
*basic_map_read_polylib(struct isl_stream
*s
,
1127 struct isl_token
*tok
;
1128 struct isl_token
*tok2
;
1131 unsigned in
= 0, out
, local
= 0;
1132 struct isl_basic_map
*bmap
= NULL
;
1137 tok
= isl_stream_next_token(s
);
1139 isl_stream_error(s
, NULL
, "unexpected EOF");
1142 tok2
= isl_stream_next_token(s
);
1144 isl_token_free(tok
);
1145 isl_stream_error(s
, NULL
, "unexpected EOF");
1148 n_row
= isl_int_get_si(tok
->u
.v
);
1149 n_col
= isl_int_get_si(tok2
->u
.v
);
1150 on_new_line
= tok2
->on_new_line
;
1151 isl_token_free(tok2
);
1152 isl_token_free(tok
);
1153 isl_assert(s
->ctx
, !on_new_line
, return NULL
);
1154 isl_assert(s
->ctx
, n_row
>= 0, return NULL
);
1155 isl_assert(s
->ctx
, n_col
>= 2 + nparam
, return NULL
);
1156 tok
= isl_stream_next_token_on_same_line(s
);
1158 if (tok
->type
!= ISL_TOKEN_VALUE
) {
1159 isl_stream_error(s
, tok
,
1160 "expecting number of output dimensions");
1161 isl_stream_push_token(s
, tok
);
1164 out
= isl_int_get_si(tok
->u
.v
);
1165 isl_token_free(tok
);
1167 tok
= isl_stream_next_token_on_same_line(s
);
1168 if (!tok
|| tok
->type
!= ISL_TOKEN_VALUE
) {
1169 isl_stream_error(s
, tok
,
1170 "expecting number of input dimensions");
1172 isl_stream_push_token(s
, tok
);
1175 in
= isl_int_get_si(tok
->u
.v
);
1176 isl_token_free(tok
);
1178 tok
= isl_stream_next_token_on_same_line(s
);
1179 if (!tok
|| tok
->type
!= ISL_TOKEN_VALUE
) {
1180 isl_stream_error(s
, tok
,
1181 "expecting number of existentials");
1183 isl_stream_push_token(s
, tok
);
1186 local
= isl_int_get_si(tok
->u
.v
);
1187 isl_token_free(tok
);
1189 tok
= isl_stream_next_token_on_same_line(s
);
1190 if (!tok
|| tok
->type
!= ISL_TOKEN_VALUE
) {
1191 isl_stream_error(s
, tok
,
1192 "expecting number of parameters");
1194 isl_stream_push_token(s
, tok
);
1197 nparam
= isl_int_get_si(tok
->u
.v
);
1198 isl_token_free(tok
);
1199 if (n_col
!= 1 + out
+ in
+ local
+ nparam
+ 1) {
1200 isl_stream_error(s
, NULL
,
1201 "dimensions don't match");
1205 out
= n_col
- 2 - nparam
;
1206 bmap
= isl_basic_map_alloc(s
->ctx
, nparam
, in
, out
, local
, n_row
, n_row
);
1210 for (i
= 0; i
< local
; ++i
) {
1211 int k
= isl_basic_map_alloc_div(bmap
);
1214 isl_seq_clr(bmap
->div
[k
], 1 + 1 + nparam
+ in
+ out
+ local
);
1217 for (i
= 0; i
< n_row
; ++i
)
1218 bmap
= basic_map_read_polylib_constraint(s
, bmap
);
1220 tok
= isl_stream_next_token_on_same_line(s
);
1222 isl_stream_error(s
, tok
, "unexpected extra token on line");
1223 isl_stream_push_token(s
, tok
);
1227 bmap
= isl_basic_map_simplify(bmap
);
1228 bmap
= isl_basic_map_finalize(bmap
);
1231 isl_basic_map_free(bmap
);
1235 static struct isl_map
*map_read_polylib(struct isl_stream
*s
, int nparam
)
1237 struct isl_token
*tok
;
1238 struct isl_token
*tok2
;
1240 struct isl_map
*map
;
1242 tok
= isl_stream_next_token(s
);
1244 isl_stream_error(s
, NULL
, "unexpected EOF");
1247 tok2
= isl_stream_next_token_on_same_line(s
);
1249 isl_stream_push_token(s
, tok2
);
1250 isl_stream_push_token(s
, tok
);
1251 return isl_map_from_basic_map(basic_map_read_polylib(s
, nparam
));
1253 n
= isl_int_get_si(tok
->u
.v
);
1254 isl_token_free(tok
);
1256 isl_assert(s
->ctx
, n
>= 1, return NULL
);
1258 map
= isl_map_from_basic_map(basic_map_read_polylib(s
, nparam
));
1260 for (i
= 1; i
< n
; ++i
)
1261 map
= isl_map_union(map
,
1262 isl_map_from_basic_map(basic_map_read_polylib(s
, nparam
)));
1267 static int optional_power(struct isl_stream
*s
)
1270 struct isl_token
*tok
;
1272 tok
= isl_stream_next_token(s
);
1275 if (tok
->type
!= '^') {
1276 isl_stream_push_token(s
, tok
);
1279 isl_token_free(tok
);
1280 tok
= isl_stream_next_token(s
);
1281 if (!tok
|| tok
->type
!= ISL_TOKEN_VALUE
) {
1282 isl_stream_error(s
, tok
, "expecting exponent");
1284 isl_stream_push_token(s
, tok
);
1287 pow
= isl_int_get_si(tok
->u
.v
);
1288 isl_token_free(tok
);
1292 static __isl_give isl_div
*read_div(struct isl_stream
*s
,
1293 __isl_keep isl_basic_map
*bmap
, struct vars
*v
)
1295 unsigned total
= isl_basic_map_total_dim(bmap
);
1298 bmap
= isl_basic_map_copy(bmap
);
1299 bmap
= isl_basic_map_cow(bmap
);
1300 bmap
= isl_basic_map_extend_dim(bmap
, isl_dim_copy(bmap
->dim
),
1303 if ((k
= isl_basic_map_alloc_div(bmap
)) < 0)
1305 isl_seq_clr(bmap
->div
[k
], 1 + 1 + total
);
1307 if (vars_add_anon(v
) < 0)
1309 bmap
= add_div_definition(s
, v
, bmap
, k
);
1312 return isl_basic_map_div(bmap
, k
);
1314 isl_basic_map_free(bmap
);
1318 static __isl_give isl_qpolynomial
*read_term(struct isl_stream
*s
,
1319 __isl_keep isl_basic_map
*bmap
, struct vars
*v
);
1321 static __isl_give isl_qpolynomial
*read_factor(struct isl_stream
*s
,
1322 __isl_keep isl_basic_map
*bmap
, struct vars
*v
)
1324 struct isl_qpolynomial
*qp
;
1325 struct isl_token
*tok
;
1327 tok
= isl_stream_next_token(s
);
1329 isl_stream_error(s
, NULL
, "unexpected EOF");
1332 if (tok
->type
== '(') {
1333 isl_token_free(tok
);
1334 qp
= read_term(s
, bmap
, v
);
1337 if (isl_stream_eat(s
, ')'))
1339 } else if (tok
->type
== ISL_TOKEN_VALUE
) {
1340 struct isl_token
*tok2
;
1341 tok2
= isl_stream_next_token(s
);
1342 if (tok2
&& tok2
->type
== '/') {
1343 isl_token_free(tok2
);
1344 tok2
= isl_stream_next_token(s
);
1345 if (!tok2
|| tok2
->type
!= ISL_TOKEN_VALUE
) {
1346 isl_stream_error(s
, tok2
, "expected denominator");
1347 isl_token_free(tok
);
1348 isl_token_free(tok2
);
1351 qp
= isl_qpolynomial_rat_cst(isl_basic_map_get_dim(bmap
),
1352 tok
->u
.v
, tok2
->u
.v
);
1353 isl_token_free(tok2
);
1355 isl_stream_push_token(s
, tok2
);
1356 qp
= isl_qpolynomial_cst(isl_basic_map_get_dim(bmap
),
1359 isl_token_free(tok
);
1360 } else if (tok
->type
== ISL_TOKEN_INFTY
) {
1361 isl_token_free(tok
);
1362 qp
= isl_qpolynomial_infty(isl_basic_map_get_dim(bmap
));
1363 } else if (tok
->type
== ISL_TOKEN_NAN
) {
1364 isl_token_free(tok
);
1365 qp
= isl_qpolynomial_nan(isl_basic_map_get_dim(bmap
));
1366 } else if (tok
->type
== ISL_TOKEN_IDENT
) {
1368 int pos
= vars_pos(v
, tok
->u
.s
, -1);
1371 isl_token_free(tok
);
1375 isl_stream_error(s
, tok
, "unknown identifier");
1376 isl_token_free(tok
);
1379 isl_token_free(tok
);
1380 pow
= optional_power(s
);
1381 qp
= isl_qpolynomial_pow(isl_basic_map_get_dim(bmap
), pos
, pow
);
1382 } else if (tok
->type
== '[') {
1386 isl_stream_push_token(s
, tok
);
1387 div
= read_div(s
, bmap
, v
);
1388 pow
= optional_power(s
);
1389 qp
= isl_qpolynomial_div_pow(div
, pow
);
1390 } else if (tok
->type
== '-') {
1391 struct isl_qpolynomial
*qp2
;
1393 isl_token_free(tok
);
1394 qp
= isl_qpolynomial_cst(isl_basic_map_get_dim(bmap
),
1396 qp2
= read_factor(s
, bmap
, v
);
1397 qp
= isl_qpolynomial_mul(qp
, qp2
);
1399 isl_stream_error(s
, tok
, "unexpected isl_token");
1400 isl_stream_push_token(s
, tok
);
1404 if (isl_stream_eat_if_available(s
, '*') ||
1405 isl_stream_next_token_is(s
, ISL_TOKEN_IDENT
)) {
1406 struct isl_qpolynomial
*qp2
;
1408 qp2
= read_factor(s
, bmap
, v
);
1409 qp
= isl_qpolynomial_mul(qp
, qp2
);
1414 isl_qpolynomial_free(qp
);
1418 static __isl_give isl_qpolynomial
*read_term(struct isl_stream
*s
,
1419 __isl_keep isl_basic_map
*bmap
, struct vars
*v
)
1421 struct isl_token
*tok
;
1422 struct isl_qpolynomial
*qp
;
1424 qp
= read_factor(s
, bmap
, v
);
1427 tok
= isl_stream_next_token(s
);
1431 if (tok
->type
== '+') {
1432 struct isl_qpolynomial
*qp2
;
1434 isl_token_free(tok
);
1435 qp2
= read_factor(s
, bmap
, v
);
1436 qp
= isl_qpolynomial_add(qp
, qp2
);
1437 } else if (tok
->type
== '-') {
1438 struct isl_qpolynomial
*qp2
;
1440 isl_token_free(tok
);
1441 qp2
= read_factor(s
, bmap
, v
);
1442 qp
= isl_qpolynomial_sub(qp
, qp2
);
1443 } else if (tok
->type
== ISL_TOKEN_VALUE
&&
1444 isl_int_is_neg(tok
->u
.v
)) {
1445 struct isl_qpolynomial
*qp2
;
1447 isl_stream_push_token(s
, tok
);
1448 qp2
= read_factor(s
, bmap
, v
);
1449 qp
= isl_qpolynomial_add(qp
, qp2
);
1451 isl_stream_push_token(s
, tok
);
1459 static __isl_give isl_map
*read_optional_disjuncts(struct isl_stream
*s
,
1460 __isl_take isl_basic_map
*bmap
, struct vars
*v
)
1462 struct isl_token
*tok
;
1463 struct isl_map
*map
;
1465 tok
= isl_stream_next_token(s
);
1467 isl_stream_error(s
, NULL
, "unexpected EOF");
1470 map
= isl_map_from_basic_map(isl_basic_map_copy(bmap
));
1471 if (tok
->type
== ':') {
1472 isl_token_free(tok
);
1473 map
= isl_map_intersect(map
,
1474 read_disjuncts(s
, v
, isl_basic_map_get_dim(bmap
)));
1476 isl_stream_push_token(s
, tok
);
1478 isl_basic_map_free(bmap
);
1482 isl_basic_map_free(bmap
);
1486 static struct isl_obj
obj_read_poly(struct isl_stream
*s
,
1487 __isl_take isl_basic_map
*bmap
, struct vars
*v
, int n
)
1489 struct isl_obj obj
= { isl_obj_pw_qpolynomial
, NULL
};
1490 struct isl_pw_qpolynomial
*pwqp
;
1491 struct isl_qpolynomial
*qp
;
1492 struct isl_map
*map
;
1493 struct isl_set
*set
;
1495 qp
= read_term(s
, bmap
, v
);
1496 map
= read_optional_disjuncts(s
, bmap
, v
);
1497 set
= isl_map_range(map
);
1499 pwqp
= isl_pw_qpolynomial_alloc(set
, qp
);
1501 vars_drop(v
, v
->n
- n
);
1507 static struct isl_obj
obj_read_poly_or_fold(struct isl_stream
*s
,
1508 __isl_take isl_basic_map
*bmap
, struct vars
*v
, int n
)
1510 struct isl_obj obj
= { isl_obj_pw_qpolynomial_fold
, NULL
};
1511 struct isl_obj obj_p
;
1512 isl_qpolynomial
*qp
;
1513 isl_qpolynomial_fold
*fold
= NULL
;
1514 isl_pw_qpolynomial_fold
*pwf
;
1518 if (!isl_stream_eat_if_available(s
, ISL_TOKEN_MAX
))
1519 return obj_read_poly(s
, bmap
, v
, n
);
1521 if (isl_stream_eat(s
, '('))
1524 qp
= read_term(s
, bmap
, v
);
1525 fold
= isl_qpolynomial_fold_alloc(isl_fold_max
, qp
);
1527 while (isl_stream_eat_if_available(s
, ',')) {
1528 isl_qpolynomial_fold
*fold_i
;
1529 qp
= read_term(s
, bmap
, v
);
1530 fold_i
= isl_qpolynomial_fold_alloc(isl_fold_max
, qp
);
1531 fold
= isl_qpolynomial_fold_fold(fold
, fold_i
);
1534 if (isl_stream_eat(s
, ')'))
1537 map
= read_optional_disjuncts(s
, bmap
, v
);
1538 set
= isl_map_range(map
);
1539 pwf
= isl_pw_qpolynomial_fold_alloc(isl_fold_max
, set
, fold
);
1541 vars_drop(v
, v
->n
- n
);
1546 isl_basic_map_free(bmap
);
1547 isl_qpolynomial_fold_free(fold
);
1548 obj
.type
= isl_obj_none
;
1552 static struct isl_obj
obj_read_body(struct isl_stream
*s
,
1553 __isl_take isl_basic_map
*bmap
, struct vars
*v
)
1555 struct isl_map
*map
= NULL
;
1556 struct isl_token
*tok
;
1557 struct isl_obj obj
= { isl_obj_set
, NULL
};
1560 if (!next_is_tuple(s
))
1561 return obj_read_poly_or_fold(s
, bmap
, v
, n
);
1563 bmap
= read_tuple(s
, bmap
, isl_dim_in
, v
);
1566 tok
= isl_stream_next_token(s
);
1567 if (tok
&& tok
->type
== ISL_TOKEN_TO
) {
1568 obj
.type
= isl_obj_map
;
1569 isl_token_free(tok
);
1570 if (!next_is_tuple(s
)) {
1571 bmap
= isl_basic_map_reverse(bmap
);
1572 return obj_read_poly_or_fold(s
, bmap
, v
, n
);
1574 bmap
= read_tuple(s
, bmap
, isl_dim_out
, v
);
1578 bmap
= isl_basic_map_reverse(bmap
);
1580 isl_stream_push_token(s
, tok
);
1583 map
= read_optional_disjuncts(s
, bmap
, v
);
1585 vars_drop(v
, v
->n
- n
);
1590 isl_basic_map_free(bmap
);
1591 obj
.type
= isl_obj_none
;
1595 static struct isl_obj
to_union(isl_ctx
*ctx
, struct isl_obj obj
)
1597 if (obj
.type
== isl_obj_map
) {
1598 obj
.v
= isl_union_map_from_map(obj
.v
);
1599 obj
.type
= isl_obj_union_map
;
1600 } else if (obj
.type
== isl_obj_set
) {
1601 obj
.v
= isl_union_set_from_set(obj
.v
);
1602 obj
.type
= isl_obj_union_set
;
1603 } else if (obj
.type
== isl_obj_pw_qpolynomial
) {
1604 obj
.v
= isl_union_pw_qpolynomial_from_pw_qpolynomial(obj
.v
);
1605 obj
.type
= isl_obj_union_pw_qpolynomial
;
1606 } else if (obj
.type
== isl_obj_pw_qpolynomial_fold
) {
1607 obj
.v
= isl_union_pw_qpolynomial_fold_from_pw_qpolynomial_fold(obj
.v
);
1608 obj
.type
= isl_obj_union_pw_qpolynomial_fold
;
1610 isl_assert(ctx
, 0, goto error
);
1613 obj
.type
->free(obj
.v
);
1614 obj
.type
= isl_obj_none
;
1618 static struct isl_obj
obj_add(struct isl_ctx
*ctx
,
1619 struct isl_obj obj1
, struct isl_obj obj2
)
1621 if (obj1
.type
== isl_obj_set
&& obj2
.type
== isl_obj_union_set
)
1622 obj1
= to_union(ctx
, obj1
);
1623 if (obj1
.type
== isl_obj_union_set
&& obj2
.type
== isl_obj_set
)
1624 obj2
= to_union(ctx
, obj2
);
1625 if (obj1
.type
== isl_obj_map
&& obj2
.type
== isl_obj_union_map
)
1626 obj1
= to_union(ctx
, obj1
);
1627 if (obj1
.type
== isl_obj_union_map
&& obj2
.type
== isl_obj_map
)
1628 obj2
= to_union(ctx
, obj2
);
1629 if (obj1
.type
== isl_obj_pw_qpolynomial
&&
1630 obj2
.type
== isl_obj_union_pw_qpolynomial
)
1631 obj1
= to_union(ctx
, obj1
);
1632 if (obj1
.type
== isl_obj_union_pw_qpolynomial
&&
1633 obj2
.type
== isl_obj_pw_qpolynomial
)
1634 obj2
= to_union(ctx
, obj2
);
1635 if (obj1
.type
== isl_obj_pw_qpolynomial_fold
&&
1636 obj2
.type
== isl_obj_union_pw_qpolynomial_fold
)
1637 obj1
= to_union(ctx
, obj1
);
1638 if (obj1
.type
== isl_obj_union_pw_qpolynomial_fold
&&
1639 obj2
.type
== isl_obj_pw_qpolynomial_fold
)
1640 obj2
= to_union(ctx
, obj2
);
1641 isl_assert(ctx
, obj1
.type
== obj2
.type
, goto error
);
1642 if (obj1
.type
== isl_obj_map
&& !isl_map_has_equal_dim(obj1
.v
, obj2
.v
)) {
1643 obj1
= to_union(ctx
, obj1
);
1644 obj2
= to_union(ctx
, obj2
);
1646 if (obj1
.type
== isl_obj_set
&& !isl_set_has_equal_dim(obj1
.v
, obj2
.v
)) {
1647 obj1
= to_union(ctx
, obj1
);
1648 obj2
= to_union(ctx
, obj2
);
1650 if (obj1
.type
== isl_obj_pw_qpolynomial
&&
1651 !isl_pw_qpolynomial_has_equal_dim(obj1
.v
, obj2
.v
)) {
1652 obj1
= to_union(ctx
, obj1
);
1653 obj2
= to_union(ctx
, obj2
);
1655 if (obj1
.type
== isl_obj_pw_qpolynomial_fold
&&
1656 !isl_pw_qpolynomial_fold_has_equal_dim(obj1
.v
, obj2
.v
)) {
1657 obj1
= to_union(ctx
, obj1
);
1658 obj2
= to_union(ctx
, obj2
);
1660 obj1
.v
= obj1
.type
->add(obj1
.v
, obj2
.v
);
1663 obj1
.type
->free(obj1
.v
);
1664 obj2
.type
->free(obj2
.v
);
1665 obj1
.type
= isl_obj_none
;
1670 static struct isl_obj
obj_read(struct isl_stream
*s
, int nparam
)
1672 isl_basic_map
*bmap
= NULL
;
1673 struct isl_token
*tok
;
1674 struct vars
*v
= NULL
;
1675 struct isl_obj obj
= { isl_obj_set
, NULL
};
1677 tok
= isl_stream_next_token(s
);
1679 isl_stream_error(s
, NULL
, "unexpected EOF");
1682 if (tok
->type
== ISL_TOKEN_VALUE
) {
1683 struct isl_map
*map
;
1684 isl_stream_push_token(s
, tok
);
1685 map
= map_read_polylib(s
, nparam
);
1688 if (isl_map_dim(map
, isl_dim_in
) > 0)
1689 obj
.type
= isl_obj_map
;
1693 v
= vars_new(s
->ctx
);
1695 isl_stream_push_token(s
, tok
);
1698 bmap
= isl_basic_map_alloc(s
->ctx
, 0, 0, 0, 0, 0, 0);
1699 if (tok
->type
== '[') {
1700 isl_stream_push_token(s
, tok
);
1701 bmap
= read_tuple(s
, bmap
, isl_dim_param
, v
);
1705 isl_assert(s
->ctx
, nparam
== v
->n
, goto error
);
1706 tok
= isl_stream_next_token(s
);
1707 if (!tok
|| tok
->type
!= ISL_TOKEN_TO
) {
1708 isl_stream_error(s
, tok
, "expecting '->'");
1710 isl_stream_push_token(s
, tok
);
1713 isl_token_free(tok
);
1714 tok
= isl_stream_next_token(s
);
1715 } else if (nparam
> 0)
1716 bmap
= isl_basic_map_add(bmap
, isl_dim_param
, nparam
);
1717 if (!tok
|| tok
->type
!= '{') {
1718 isl_stream_error(s
, tok
, "expecting '{'");
1720 isl_stream_push_token(s
, tok
);
1723 isl_token_free(tok
);
1725 tok
= isl_stream_next_token(s
);
1728 else if (tok
->type
== ISL_TOKEN_IDENT
&& !strcmp(tok
->u
.s
, "Sym")) {
1729 isl_token_free(tok
);
1730 if (isl_stream_eat(s
, '='))
1732 bmap
= read_tuple(s
, bmap
, isl_dim_param
, v
);
1736 isl_assert(s
->ctx
, nparam
== v
->n
, goto error
);
1737 } else if (tok
->type
== '}') {
1738 obj
.type
= isl_obj_union_set
;
1739 obj
.v
= isl_union_set_empty(isl_basic_map_get_dim(bmap
));
1740 isl_token_free(tok
);
1743 isl_stream_push_token(s
, tok
);
1748 o
= obj_read_body(s
, isl_basic_map_copy(bmap
), v
);
1749 if (o
.type
== isl_obj_none
|| !o
.v
)
1754 obj
= obj_add(s
->ctx
, obj
, o
);
1755 if (obj
.type
== isl_obj_none
|| !obj
.v
)
1758 tok
= isl_stream_next_token(s
);
1759 if (!tok
|| tok
->type
!= ';')
1761 isl_token_free(tok
);
1764 if (tok
&& tok
->type
== '}') {
1765 isl_token_free(tok
);
1767 isl_stream_error(s
, tok
, "unexpected isl_token");
1769 isl_token_free(tok
);
1774 isl_basic_map_free(bmap
);
1778 isl_basic_map_free(bmap
);
1779 obj
.type
->free(obj
.v
);
1786 struct isl_obj
isl_stream_read_obj(struct isl_stream
*s
)
1788 return obj_read(s
, -1);
1791 __isl_give isl_map
*isl_stream_read_map(struct isl_stream
*s
, int nparam
)
1794 struct isl_map
*map
;
1796 obj
= obj_read(s
, nparam
);
1798 isl_assert(s
->ctx
, obj
.type
== isl_obj_map
||
1799 obj
.type
== isl_obj_set
, goto error
);
1803 obj
.type
->free(obj
.v
);
1807 __isl_give isl_set
*isl_stream_read_set(struct isl_stream
*s
, int nparam
)
1810 struct isl_set
*set
;
1812 obj
= obj_read(s
, nparam
);
1814 isl_assert(s
->ctx
, obj
.type
== isl_obj_set
, goto error
);
1818 obj
.type
->free(obj
.v
);
1822 static struct isl_basic_map
*basic_map_read(struct isl_stream
*s
, int nparam
)
1825 struct isl_map
*map
;
1826 struct isl_basic_map
*bmap
;
1828 obj
= obj_read(s
, nparam
);
1833 isl_assert(map
->ctx
, map
->n
<= 1, goto error
);
1836 bmap
= isl_basic_map_empty_like_map(map
);
1838 bmap
= isl_basic_map_copy(map
->p
[0]);
1848 __isl_give isl_basic_map
*isl_basic_map_read_from_file(isl_ctx
*ctx
,
1849 FILE *input
, int nparam
)
1851 struct isl_basic_map
*bmap
;
1852 struct isl_stream
*s
= isl_stream_new_file(ctx
, input
);
1855 bmap
= basic_map_read(s
, nparam
);
1860 __isl_give isl_basic_set
*isl_basic_set_read_from_file(isl_ctx
*ctx
,
1861 FILE *input
, int nparam
)
1863 struct isl_basic_map
*bmap
;
1864 bmap
= isl_basic_map_read_from_file(ctx
, input
, nparam
);
1867 isl_assert(ctx
, isl_basic_map_n_in(bmap
) == 0, goto error
);
1868 return (struct isl_basic_set
*)bmap
;
1870 isl_basic_map_free(bmap
);
1874 struct isl_basic_map
*isl_basic_map_read_from_str(struct isl_ctx
*ctx
,
1875 const char *str
, int nparam
)
1877 struct isl_basic_map
*bmap
;
1878 struct isl_stream
*s
= isl_stream_new_str(ctx
, str
);
1881 bmap
= basic_map_read(s
, nparam
);
1886 struct isl_basic_set
*isl_basic_set_read_from_str(struct isl_ctx
*ctx
,
1887 const char *str
, int nparam
)
1889 struct isl_basic_map
*bmap
;
1890 bmap
= isl_basic_map_read_from_str(ctx
, str
, nparam
);
1893 isl_assert(ctx
, isl_basic_map_n_in(bmap
) == 0, goto error
);
1894 return (struct isl_basic_set
*)bmap
;
1896 isl_basic_map_free(bmap
);
1900 __isl_give isl_map
*isl_map_read_from_file(struct isl_ctx
*ctx
,
1901 FILE *input
, int nparam
)
1903 struct isl_map
*map
;
1904 struct isl_stream
*s
= isl_stream_new_file(ctx
, input
);
1907 map
= isl_stream_read_map(s
, nparam
);
1912 __isl_give isl_map
*isl_map_read_from_str(struct isl_ctx
*ctx
,
1913 const char *str
, int nparam
)
1915 struct isl_map
*map
;
1916 struct isl_stream
*s
= isl_stream_new_str(ctx
, str
);
1919 map
= isl_stream_read_map(s
, nparam
);
1924 __isl_give isl_set
*isl_set_read_from_file(struct isl_ctx
*ctx
,
1925 FILE *input
, int nparam
)
1927 struct isl_map
*map
;
1928 map
= isl_map_read_from_file(ctx
, input
, nparam
);
1931 isl_assert(ctx
, isl_map_n_in(map
) == 0, goto error
);
1932 return (struct isl_set
*)map
;
1938 struct isl_set
*isl_set_read_from_str(struct isl_ctx
*ctx
,
1939 const char *str
, int nparam
)
1941 struct isl_map
*map
;
1942 map
= isl_map_read_from_str(ctx
, str
, nparam
);
1945 isl_assert(ctx
, isl_map_n_in(map
) == 0, goto error
);
1946 return (struct isl_set
*)map
;
1952 static char *next_line(FILE *input
, char *line
, unsigned len
)
1957 if (!(p
= fgets(line
, len
, input
)))
1959 while (isspace(*p
) && *p
!= '\n')
1961 } while (*p
== '#' || *p
== '\n');
1966 static struct isl_vec
*isl_vec_read_from_file_polylib(struct isl_ctx
*ctx
,
1969 struct isl_vec
*vec
= NULL
;
1978 isl_assert(ctx
, next_line(input
, line
, sizeof(line
)), return NULL
);
1979 isl_assert(ctx
, sscanf(line
, "%u", &size
) == 1, return NULL
);
1981 vec
= isl_vec_alloc(ctx
, size
);
1983 p
= next_line(input
, line
, sizeof(line
));
1984 isl_assert(ctx
, p
, goto error
);
1986 for (j
= 0; j
< size
; ++j
) {
1987 n
= sscanf(p
, "%s%n", val
, &offset
);
1988 isl_assert(ctx
, n
!= 0, goto error
);
1989 isl_int_read(vec
->el
[j
], val
);
1999 struct isl_vec
*isl_vec_read_from_file(struct isl_ctx
*ctx
,
2000 FILE *input
, unsigned input_format
)
2002 if (input_format
== ISL_FORMAT_POLYLIB
)
2003 return isl_vec_read_from_file_polylib(ctx
, input
);
2005 isl_assert(ctx
, 0, return NULL
);
2008 __isl_give isl_pw_qpolynomial
*isl_stream_read_pw_qpolynomial(
2009 struct isl_stream
*s
)
2012 struct isl_pw_qpolynomial
*pwqp
;
2014 obj
= obj_read(s
, -1);
2016 isl_assert(s
->ctx
, obj
.type
== isl_obj_pw_qpolynomial
,
2021 obj
.type
->free(obj
.v
);
2025 __isl_give isl_pw_qpolynomial
*isl_pw_qpolynomial_read_from_str(isl_ctx
*ctx
,
2028 isl_pw_qpolynomial
*pwqp
;
2029 struct isl_stream
*s
= isl_stream_new_str(ctx
, str
);
2032 pwqp
= isl_stream_read_pw_qpolynomial(s
);