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
);