X-Git-Url: http://review.tizen.org/git/?a=blobdiff_plain;f=isl_input.c;h=164a269f3b1d7283e8612a0f5c804e4745be5a06;hb=63fb8a7f484648c3caa25351c8c94ac2395ec563;hp=3465e1fdfbd3cfe8c6896d68121481bda3e7310d;hpb=775f713b1f34049373157f8f4db6f1ea844d18cd;p=platform%2Fupstream%2Fisl.git diff --git a/isl_input.c b/isl_input.c index 3465e1f..164a269 100644 --- a/isl_input.c +++ b/isl_input.c @@ -1,13 +1,15 @@ /* * Copyright 2008-2009 Katholieke Universiteit Leuven * Copyright 2010 INRIA Saclay + * Copyright 2012 Ecole Normale Superieure * - * Use of this software is governed by the GNU LGPLv2.1 license + * Use of this software is governed by the MIT license * * Written by Sven Verdoolaege, K.U.Leuven, Departement * Computerwetenschappen, Celestijnenlaan 200A, B-3001 Leuven, Belgium * and INRIA Saclay - Ile-de-France, Parc Club Orsay Universite, * ZAC des vignes, 4 rue Jacques Monod, 91893 Orsay, France + * and Ecole Normale Superieure, 45 rue d’Ulm, 75230 Paris, France */ #include @@ -17,7 +19,6 @@ #include #include #include -#include #include #include #include "isl_polynomial_private.h" @@ -25,6 +26,7 @@ #include #include #include +#include struct variable { char *name; @@ -138,26 +140,6 @@ static int vars_add_anon(struct vars *v) return 0; } -static __isl_give isl_map *set_name(__isl_take isl_map *map, - enum isl_dim_type type, unsigned pos, char *name) -{ - char *prime; - - if (!map) - return NULL; - if (!name) - return map; - - prime = strchr(name, '\''); - if (prime) - *prime = '\0'; - map = isl_map_set_dim_name(map, type, pos, name); - if (prime) - *prime = '\''; - - return map; -} - /* Obtain next token, with some preprocessing. * In particular, evaluate expressions of the form x^y, * with x and y values. @@ -187,6 +169,86 @@ error: return NULL; } +/* Read an isl_val from "s". + * + * The following token sequences are recognized + * + * "infty" -> infty + * "-" "infty" -> -infty + * "NaN" -> NaN + * n "/" d -> n/d + * v -> v + * + * where n, d and v are integer constants. + */ +__isl_give isl_val *isl_stream_read_val(struct isl_stream *s) +{ + struct isl_token *tok = NULL; + struct isl_token *tok2 = NULL; + isl_val *val; + + tok = next_token(s); + if (!tok) { + isl_stream_error(s, NULL, "unexpected EOF"); + goto error; + } + if (tok->type == ISL_TOKEN_INFTY) { + isl_token_free(tok); + return isl_val_infty(s->ctx); + } + if (tok->type == '-' && + isl_stream_eat_if_available(s, ISL_TOKEN_INFTY)) { + isl_token_free(tok); + return isl_val_neginfty(s->ctx); + } + if (tok->type == ISL_TOKEN_NAN) { + isl_token_free(tok); + return isl_val_nan(s->ctx); + } + if (tok->type != ISL_TOKEN_VALUE) { + isl_stream_error(s, tok, "expecting value"); + goto error; + } + + if (isl_stream_eat_if_available(s, '/')) { + tok2 = next_token(s); + if (!tok2) { + isl_stream_error(s, NULL, "unexpected EOF"); + goto error; + } + if (tok2->type != ISL_TOKEN_VALUE) { + isl_stream_error(s, tok2, "expecting value"); + goto error; + } + val = isl_val_rat_from_isl_int(s->ctx, tok->u.v, tok2->u.v); + val = isl_val_normalize(val); + } else { + val = isl_val_int_from_isl_int(s->ctx, tok->u.v); + } + + isl_token_free(tok); + isl_token_free(tok2); + return val; +error: + isl_token_free(tok); + isl_token_free(tok2); + return NULL; +} + +/* Read an isl_val from "str". + */ +struct isl_val *isl_val_read_from_str(struct isl_ctx *ctx, + const char *str) +{ + isl_val *val; + struct isl_stream *s = isl_stream_new_str(ctx, str); + if (!s) + return NULL; + val = isl_stream_read_val(s); + isl_stream_free(s); + return val; +} + static int accept_cst_factor(struct isl_stream *s, isl_int *f) { struct isl_token *tok; @@ -245,12 +307,12 @@ error: } static __isl_give isl_pw_aff *accept_affine(struct isl_stream *s, - __isl_take isl_dim *dim, struct vars *v); + __isl_take isl_space *dim, struct vars *v); static __isl_give isl_pw_aff_list *accept_affine_list(struct isl_stream *s, - __isl_take isl_dim *dim, struct vars *v); + __isl_take isl_space *dim, struct vars *v); static __isl_give isl_pw_aff *accept_minmax(struct isl_stream *s, - __isl_take isl_dim *dim, struct vars *v) + __isl_take isl_space *dim, struct vars *v) { struct isl_token *tok; isl_pw_aff_list *list = NULL; @@ -265,26 +327,25 @@ static __isl_give isl_pw_aff *accept_minmax(struct isl_stream *s, if (isl_stream_eat(s, '(')) goto error; - list = accept_affine_list(s, isl_dim_copy(dim), v); + list = accept_affine_list(s, isl_space_copy(dim), v); if (!list) goto error; if (isl_stream_eat(s, ')')) goto error; - isl_dim_free(dim); + isl_space_free(dim); return min ? isl_pw_aff_list_min(list) : isl_pw_aff_list_max(list); error: - isl_dim_free(dim); + isl_space_free(dim); isl_pw_aff_list_free(list); return NULL; } static __isl_give isl_pw_aff *accept_div(struct isl_stream *s, - __isl_take isl_dim *dim, struct vars *v) + __isl_take isl_space *dim, struct vars *v) { struct isl_token *tok; - int seen_paren = 0; int f = 0; int c = 0; isl_pw_aff *pwaff = NULL; @@ -299,33 +360,26 @@ static __isl_give isl_pw_aff *accept_div(struct isl_stream *s, } else { if (isl_stream_eat(s, '[')) goto error; - if (isl_stream_eat_if_available(s, '(')) - seen_paren = 1; } - pwaff = accept_affine(s, isl_dim_copy(dim), v); + pwaff = accept_affine(s, isl_space_copy(dim), v); if (f || c) { if (isl_stream_eat(s, ',')) goto error; - } else { - if (seen_paren && isl_stream_eat(s, ')')) + + tok = next_token(s); + if (!tok) goto error; - if (isl_stream_eat(s, '/')) + if (tok->type != ISL_TOKEN_VALUE) { + isl_stream_error(s, tok, "expected denominator"); + isl_stream_push_token(s, tok); goto error; + } + isl_pw_aff_scale_down(pwaff, tok->u.v); + isl_token_free(tok); } - tok = next_token(s); - if (!tok) - goto error; - if (tok->type != ISL_TOKEN_VALUE) { - isl_stream_error(s, tok, "expected denominator"); - isl_stream_push_token(s, tok); - goto error; - } - isl_pw_aff_scale_down(pwaff, tok->u.v); - isl_token_free(tok); - if (c) pwaff = isl_pw_aff_ceil(pwaff); else @@ -339,16 +393,16 @@ static __isl_give isl_pw_aff *accept_div(struct isl_stream *s, goto error; } - isl_dim_free(dim); + isl_space_free(dim); return pwaff; error: - isl_dim_free(dim); + isl_space_free(dim); isl_pw_aff_free(pwaff); return NULL; } static __isl_give isl_pw_aff *accept_affine_factor(struct isl_stream *s, - __isl_take isl_dim *dim, struct vars *v) + __isl_take isl_space *dim, struct vars *v) { struct isl_token *tok = NULL; isl_pw_aff *res = NULL; @@ -370,11 +424,12 @@ static __isl_give isl_pw_aff *accept_affine_factor(struct isl_stream *s, if (pos < 0) goto error; if (pos >= n) { + vars_drop(v, v->n - n); isl_stream_error(s, tok, "unknown identifier"); goto error; } - aff = isl_aff_zero(isl_local_space_from_dim(isl_dim_copy(dim))); + aff = isl_aff_zero_on_domain(isl_local_space_from_space(isl_space_copy(dim))); if (!aff) goto error; isl_int_set_si(aff->v->el[2 + pos], 1); @@ -382,13 +437,13 @@ static __isl_give isl_pw_aff *accept_affine_factor(struct isl_stream *s, isl_token_free(tok); } else if (tok->type == ISL_TOKEN_VALUE) { if (isl_stream_eat_if_available(s, '*')) { - res = accept_affine_factor(s, isl_dim_copy(dim), v); + res = accept_affine_factor(s, isl_space_copy(dim), v); res = isl_pw_aff_scale(res, tok->u.v); } else { isl_local_space *ls; isl_aff *aff; - ls = isl_local_space_from_dim(isl_dim_copy(dim)); - aff = isl_aff_zero(ls); + ls = isl_local_space_from_space(isl_space_copy(dim)); + aff = isl_aff_zero_on_domain(ls); aff = isl_aff_add_constant(aff, tok->u.v); res = isl_pw_aff_from_aff(aff); } @@ -396,7 +451,7 @@ static __isl_give isl_pw_aff *accept_affine_factor(struct isl_stream *s, } else if (tok->type == '(') { isl_token_free(tok); tok = NULL; - res = accept_affine(s, isl_dim_copy(dim), v); + res = accept_affine(s, isl_space_copy(dim), v); if (!res) goto error; if (isl_stream_eat(s, ')')) @@ -406,18 +461,18 @@ static __isl_give isl_pw_aff *accept_affine_factor(struct isl_stream *s, tok->type == ISL_TOKEN_CEILD) { isl_stream_push_token(s, tok); tok = NULL; - res = accept_div(s, isl_dim_copy(dim), v); + res = accept_div(s, isl_space_copy(dim), v); } else if (tok->type == ISL_TOKEN_MIN || tok->type == ISL_TOKEN_MAX) { isl_stream_push_token(s, tok); tok = NULL; - res = accept_minmax(s, isl_dim_copy(dim), v); + res = accept_minmax(s, isl_space_copy(dim), v); } else { isl_stream_error(s, tok, "expecting factor"); goto error; } if (isl_stream_eat_if_available(s, '%') || isl_stream_eat_if_available(s, ISL_TOKEN_MOD)) { - isl_dim_free(dim); + isl_space_free(dim); return affine_mod(s, v, res); } if (isl_stream_eat_if_available(s, '*')) { @@ -431,37 +486,50 @@ static __isl_give isl_pw_aff *accept_affine_factor(struct isl_stream *s, res = isl_pw_aff_scale(res, f); isl_int_clear(f); } + if (isl_stream_eat_if_available(s, '/')) { + isl_int f; + isl_int_init(f); + isl_int_set_si(f, 1); + if (accept_cst_factor(s, &f) < 0) { + isl_int_clear(f); + goto error2; + } + res = isl_pw_aff_scale_down(res, f); + isl_int_clear(f); + } - isl_dim_free(dim); + isl_space_free(dim); return res; error: isl_token_free(tok); error2: isl_pw_aff_free(res); - isl_dim_free(dim); + isl_space_free(dim); return NULL; } static __isl_give isl_pw_aff *add_cst(__isl_take isl_pw_aff *pwaff, isl_int v) { isl_aff *aff; + isl_space *space; - aff = isl_aff_zero(isl_local_space_from_dim(isl_pw_aff_get_dim(pwaff))); + space = isl_pw_aff_get_domain_space(pwaff); + aff = isl_aff_zero_on_domain(isl_local_space_from_space(space)); aff = isl_aff_add_constant(aff, v); return isl_pw_aff_add(pwaff, isl_pw_aff_from_aff(aff)); } static __isl_give isl_pw_aff *accept_affine(struct isl_stream *s, - __isl_take isl_dim *dim, struct vars *v) + __isl_take isl_space *dim, struct vars *v) { struct isl_token *tok = NULL; isl_local_space *ls; isl_pw_aff *res; int sign = 1; - ls = isl_local_space_from_dim(isl_dim_copy(dim)); - res = isl_pw_aff_from_aff(isl_aff_zero(ls)); + ls = isl_local_space_from_space(isl_space_copy(dim)); + res = isl_pw_aff_from_aff(isl_aff_zero_on_domain(ls)); if (!res) goto error; @@ -485,7 +553,7 @@ static __isl_give isl_pw_aff *accept_affine(struct isl_stream *s, isl_pw_aff *term; isl_stream_push_token(s, tok); tok = NULL; - term = accept_affine_factor(s, isl_dim_copy(dim), v); + term = accept_affine_factor(s, isl_space_copy(dim), v); if (sign < 0) res = isl_pw_aff_sub(res, term); else @@ -500,7 +568,7 @@ static __isl_give isl_pw_aff *accept_affine(struct isl_stream *s, isl_stream_next_token_is(s, ISL_TOKEN_IDENT)) { isl_pw_aff *term; term = accept_affine_factor(s, - isl_dim_copy(dim), v); + isl_space_copy(dim), v); term = isl_pw_aff_scale(term, tok->u.v); res = isl_pw_aff_add(res, term); if (!res) @@ -513,7 +581,7 @@ static __isl_give isl_pw_aff *accept_affine(struct isl_stream *s, isl_stream_error(s, tok, "unexpected isl_token"); isl_stream_push_token(s, tok); isl_pw_aff_free(res); - isl_dim_free(dim); + isl_space_free(dim); return NULL; } isl_token_free(tok); @@ -535,10 +603,10 @@ static __isl_give isl_pw_aff *accept_affine(struct isl_stream *s, } } - isl_dim_free(dim); + isl_space_free(dim); return res; error: - isl_dim_free(dim); + isl_space_free(dim); isl_token_free(tok); isl_pw_aff_free(res); return NULL; @@ -563,17 +631,17 @@ static int is_comparator(struct isl_token *tok) } static struct isl_map *read_disjuncts(struct isl_stream *s, - struct vars *v, __isl_take isl_map *map); + struct vars *v, __isl_take isl_map *map, int rational); static __isl_give isl_pw_aff *accept_extended_affine(struct isl_stream *s, - __isl_take isl_dim *dim, struct vars *v); + __isl_take isl_space *dim, struct vars *v, int rational); /* Accept a ternary operator, given the first argument. */ static __isl_give isl_pw_aff *accept_ternary(struct isl_stream *s, - __isl_take isl_map *cond, struct vars *v) + __isl_take isl_map *cond, struct vars *v, int rational) { - isl_dim *dim; - isl_pw_aff *pwaff1 = NULL, *pwaff2 = NULL; + isl_space *dim; + isl_pw_aff *pwaff1 = NULL, *pwaff2 = NULL, *pa_cond; if (!cond) return NULL; @@ -581,19 +649,21 @@ static __isl_give isl_pw_aff *accept_ternary(struct isl_stream *s, if (isl_stream_eat(s, '?')) goto error; - dim = isl_dim_wrap(isl_map_get_dim(cond)); - pwaff1 = accept_extended_affine(s, dim, v); + dim = isl_space_wrap(isl_map_get_space(cond)); + pwaff1 = accept_extended_affine(s, dim, v, rational); if (!pwaff1) goto error; if (isl_stream_eat(s, ':')) goto error; - pwaff2 = accept_extended_affine(s, isl_pw_aff_get_dim(pwaff1), v); + dim = isl_pw_aff_get_domain_space(pwaff1); + pwaff2 = accept_extended_affine(s, dim, v, rational); if (!pwaff1) goto error; - return isl_pw_aff_cond(isl_map_wrap(cond), pwaff1, pwaff2); + pa_cond = isl_set_indicator_function(isl_map_wrap(cond)); + return isl_pw_aff_cond(pa_cond, pwaff1, pwaff2); error: isl_map_free(cond); isl_pw_aff_free(pwaff1); @@ -608,8 +678,9 @@ error: * argument of a ternary operator and try to parse that. */ static __isl_give isl_pw_aff *accept_extended_affine(struct isl_stream *s, - __isl_take isl_dim *dim, struct vars *v) + __isl_take isl_space *dim, struct vars *v, int rational) { + isl_space *space; isl_map *cond; isl_pw_aff *pwaff; struct isl_token *tok; @@ -624,6 +695,8 @@ static __isl_give isl_pw_aff *accept_extended_affine(struct isl_stream *s, } pwaff = accept_affine(s, dim, v); + if (rational) + pwaff = isl_pw_aff_set_rational(pwaff); if (!pwaff) return NULL; @@ -642,30 +715,38 @@ static __isl_give isl_pw_aff *accept_extended_affine(struct isl_stream *s, tok->type = ISL_TOKEN_AFF; tok->u.pwaff = pwaff; - cond = isl_map_universe(isl_dim_unwrap(isl_pw_aff_get_dim(pwaff))); + space = isl_pw_aff_get_domain_space(pwaff); + cond = isl_map_universe(isl_space_unwrap(space)); isl_stream_push_token(s, tok); - cond = read_disjuncts(s, v, cond); + cond = read_disjuncts(s, v, cond, rational); - return accept_ternary(s, cond, v); + return accept_ternary(s, cond, v, rational); } static __isl_give isl_map *read_var_def(struct isl_stream *s, - __isl_take isl_map *map, enum isl_dim_type type, struct vars *v) + __isl_take isl_map *map, enum isl_dim_type type, struct vars *v, + int rational) { isl_pw_aff *def; int pos; isl_map *def_map; - pos = isl_map_dim(map, isl_dim_in); - if (type == isl_dim_out) - pos += isl_map_dim(map, isl_dim_out); + if (type == isl_dim_param) + pos = isl_map_dim(map, isl_dim_param); + else { + pos = isl_map_dim(map, isl_dim_in); + if (type == isl_dim_out) + pos += isl_map_dim(map, isl_dim_out); + type = isl_dim_in; + } --pos; - def = accept_extended_affine(s, isl_dim_wrap(isl_map_get_dim(map)), v); + def = accept_extended_affine(s, isl_space_wrap(isl_map_get_space(map)), + v, rational); def_map = isl_map_from_pw_aff(def); - def_map = isl_map_equate(def_map, isl_dim_in, pos, isl_dim_out, 0); + def_map = isl_map_equate(def_map, type, pos, isl_dim_out, 0); def_map = isl_set_unwrap(isl_map_domain(def_map)); map = isl_map_intersect(map, def_map); @@ -673,73 +754,14 @@ static __isl_give isl_map *read_var_def(struct isl_stream *s, return map; } -static __isl_give isl_map *read_var_list(struct isl_stream *s, - __isl_take isl_map *map, enum isl_dim_type type, struct vars *v) -{ - int i = 0; - struct isl_token *tok; - - if (isl_stream_next_token_is(s, ']')) - return map; - - while ((tok = next_token(s)) != NULL) { - int new_name = 0; - - if (tok->type == ISL_TOKEN_IDENT) { - int n = v->n; - int p = vars_pos(v, tok->u.s, -1); - if (p < 0) - goto error; - new_name = p >= n; - } - - if (new_name) { - map = isl_map_add_dims(map, type, 1); - map = set_name(map, type, i, v->v->name); - isl_token_free(tok); - } else { - if (type == isl_dim_param) { - isl_stream_error(s, tok, - "expecting unique identifier"); - goto error; - } - isl_stream_push_token(s, tok); - tok = NULL; - if (vars_add_anon(v) < 0) - goto error; - map = isl_map_add_dims(map, type, 1); - map = read_var_def(s, map, type, v); - } - - tok = isl_stream_next_token(s); - if (tok && tok->type == ']' && - isl_stream_next_token_is(s, '[')) { - isl_token_free(tok); - tok = isl_stream_next_token(s); - } else if (!tok || tok->type != ',') - break; - - isl_token_free(tok); - i++; - } - if (tok) - isl_stream_push_token(s, tok); - - return map; -error: - isl_token_free(tok); - isl_map_free(map); - return NULL; -} - static __isl_give isl_pw_aff_list *accept_affine_list(struct isl_stream *s, - __isl_take isl_dim *dim, struct vars *v) + __isl_take isl_space *dim, struct vars *v) { isl_pw_aff *pwaff; isl_pw_aff_list *list; struct isl_token *tok = NULL; - pwaff = accept_affine(s, isl_dim_copy(dim), v); + pwaff = accept_affine(s, isl_space_copy(dim), v); list = isl_pw_aff_list_from_pw_aff(pwaff); if (!list) goto error; @@ -756,23 +778,23 @@ static __isl_give isl_pw_aff_list *accept_affine_list(struct isl_stream *s, } isl_token_free(tok); - pwaff = accept_affine(s, isl_dim_copy(dim), v); + pwaff = accept_affine(s, isl_space_copy(dim), v); list = isl_pw_aff_list_concat(list, isl_pw_aff_list_from_pw_aff(pwaff)); if (!list) - return NULL; + goto error; } - isl_dim_free(dim); + isl_space_free(dim); return list; error: - isl_dim_free(dim); + isl_space_free(dim); isl_pw_aff_list_free(list); return NULL; } static __isl_give isl_map *read_defined_var_list(struct isl_stream *s, - struct vars *v, __isl_take isl_map *map) + struct vars *v, __isl_take isl_map *map, int rational) { struct isl_token *tok; @@ -797,7 +819,7 @@ static __isl_give isl_map *read_defined_var_list(struct isl_stream *s, tok = isl_stream_next_token(s); if (tok && tok->type == '=') { isl_token_free(tok); - map = read_var_def(s, map, isl_dim_out, v); + map = read_var_def(s, map, isl_dim_out, v, rational); tok = isl_stream_next_token(s); } @@ -840,175 +862,432 @@ static int next_is_tuple(struct isl_stream *s) return is_tuple; } -static __isl_give isl_map *read_tuple(struct isl_stream *s, - __isl_take isl_map *map, enum isl_dim_type type, struct vars *v); - -static __isl_give isl_map *read_nested_tuple(struct isl_stream *s, - __isl_take isl_map *map, struct vars *v) +/* Allocate an initial tuple with zero dimensions and an anonymous, + * unstructured space. + * A tuple is represented as an isl_multi_pw_aff. + * The range space is the space of the tuple. + * The domain space is an anonymous space + * with a dimension for each variable in the set of variables in "v". + * If a given dimension is not defined in terms of earlier dimensions in + * the input, then the corresponding isl_pw_aff is set equal to one time + * the variable corresponding to the dimension being defined. + */ +static __isl_give isl_multi_pw_aff *tuple_alloc(struct vars *v) { - map = read_tuple(s, map, isl_dim_in, v); - if (isl_stream_eat(s, ISL_TOKEN_TO)) - goto error; - map = read_tuple(s, map, isl_dim_out, v); - map = isl_map_from_range(isl_map_wrap(map)); - return map; -error: - isl_map_free(map); - return NULL; + return isl_multi_pw_aff_alloc(isl_space_alloc(v->ctx, 0, v->n, 0)); } -static __isl_give isl_map *read_tuple(struct isl_stream *s, - __isl_take isl_map *map, enum isl_dim_type type, struct vars *v) +/* Is "pa" an expression in term of earlier dimensions? + * The alternative is that the dimension is defined to be equal to itself, + * meaning that it has a universe domain and an expression that depends + * on itself. "i" is the position of the expression in a sequence + * of "n" expressions. The final dimensions of "pa" correspond to + * these "n" expressions. + */ +static int pw_aff_is_expr(__isl_keep isl_pw_aff *pa, int i, int n) { - struct isl_token *tok; - char *name = NULL; - - tok = isl_stream_next_token(s); - if (tok && (tok->type == ISL_TOKEN_IDENT || tok->is_keyword)) { - name = strdup(tok->u.s); - if (!name) - goto error; - isl_token_free(tok); - tok = isl_stream_next_token(s); - } - if (!tok || tok->type != '[') { - isl_stream_error(s, tok, "expecting '['"); - goto error; - } - isl_token_free(tok); - if (type != isl_dim_param && next_is_tuple(s)) { - isl_dim *dim = isl_map_get_dim(map); - int nparam = isl_dim_size(dim, isl_dim_param); - int n_in = isl_dim_size(dim, isl_dim_in); - isl_map *nested; - if (type == isl_dim_out) - dim = isl_dim_move(dim, isl_dim_param, nparam, - isl_dim_in, 0, n_in); - nested = isl_map_universe(dim); - nested = read_nested_tuple(s, nested, v); - if (type == isl_dim_in) { - nested = isl_map_reverse(nested); - map = isl_map_intersect(nested, map); - } else { - isl_set *set; - dim = isl_dim_range(isl_map_get_dim(nested)); - dim = isl_dim_drop(dim, isl_dim_param, nparam, n_in); - dim = isl_dim_join(isl_map_get_dim(map), dim); - set = isl_map_domain(map); - nested = isl_map_reset_dim(nested, dim); - map = isl_map_intersect_domain(nested, set); - } - } else - map = read_var_list(s, map, type, v); - tok = isl_stream_next_token(s); - if (!tok || tok->type != ']') { - isl_stream_error(s, tok, "expecting ']'"); - goto error; - } - isl_token_free(tok); + isl_aff *aff; - if (name) { - map = isl_map_set_tuple_name(map, type, name); - free(name); - } + if (!pa) + return -1; + if (pa->n != 1) + return 1; + if (!isl_set_plain_is_universe(pa->p[0].set)) + return 1; - return map; -error: - if (tok) - isl_token_free(tok); - isl_map_free(map); - return NULL; + aff = pa->p[0].aff; + if (isl_int_is_zero(aff->v->el[aff->v->size - n + i])) + return 1; + return 0; } -static __isl_give isl_set *construct_constraints( - __isl_take isl_set *set, enum isl_token_type type, - __isl_keep isl_pw_aff_list *left, __isl_keep isl_pw_aff_list *right) +/* Does the tuple contain any dimensions that are defined + * in terms of earlier dimensions? + */ +static int tuple_has_expr(__isl_keep isl_multi_pw_aff *tuple) { - isl_set *cond; + int i, n; + int has_expr = 0; + isl_pw_aff *pa; - if (type == ISL_TOKEN_LE) - cond = isl_pw_aff_list_le_set(isl_pw_aff_list_copy(left), - isl_pw_aff_list_copy(right)); - else if (type == ISL_TOKEN_GE) - cond = isl_pw_aff_list_ge_set(isl_pw_aff_list_copy(left), - isl_pw_aff_list_copy(right)); - else if (type == ISL_TOKEN_LT) - cond = isl_pw_aff_list_lt_set(isl_pw_aff_list_copy(left), - isl_pw_aff_list_copy(right)); - else if (type == ISL_TOKEN_GT) - cond = isl_pw_aff_list_gt_set(isl_pw_aff_list_copy(left), - isl_pw_aff_list_copy(right)); - else if (type == ISL_TOKEN_NE) - cond = isl_pw_aff_list_ne_set(isl_pw_aff_list_copy(left), - isl_pw_aff_list_copy(right)); - else - cond = isl_pw_aff_list_eq_set(isl_pw_aff_list_copy(left), - isl_pw_aff_list_copy(right)); + if (!tuple) + return -1; + n = isl_multi_pw_aff_dim(tuple, isl_dim_out); + for (i = 0; i < n; ++i) { + pa = isl_multi_pw_aff_get_pw_aff(tuple, i); + has_expr = pw_aff_is_expr(pa, i, n); + isl_pw_aff_free(pa); + if (has_expr < 0 || has_expr) + break; + } - return isl_set_intersect(set, cond); + return has_expr; } -static __isl_give isl_map *add_constraint(struct isl_stream *s, - struct vars *v, __isl_take isl_map *map) +/* Add a dimension to the given tuple. + * The dimension is initially undefined, so it is encoded + * as one times itself. + */ +static __isl_give isl_multi_pw_aff *tuple_add_dim( + __isl_take isl_multi_pw_aff *tuple, struct vars *v) { - struct isl_token *tok = NULL; - isl_pw_aff_list *list1 = NULL, *list2 = NULL; - isl_set *set; - - set = isl_map_wrap(map); - list1 = accept_affine_list(s, isl_set_get_dim(set), v); - if (!list1) - goto error; - tok = isl_stream_next_token(s); - if (!is_comparator(tok)) { - isl_stream_error(s, tok, "missing operator"); - if (tok) - isl_stream_push_token(s, tok); - tok = NULL; - goto error; - } - for (;;) { - list2 = accept_affine_list(s, isl_set_get_dim(set), v); - if (!list2) - goto error; - - set = construct_constraints(set, tok->type, list1, list2); - isl_token_free(tok); - isl_pw_aff_list_free(list1); - list1 = list2; + isl_space *space; + isl_aff *aff; + isl_pw_aff *pa; - tok = isl_stream_next_token(s); - if (!is_comparator(tok)) { - if (tok) - isl_stream_push_token(s, tok); - break; - } - } - isl_pw_aff_list_free(list1); + tuple = isl_multi_pw_aff_add_dims(tuple, isl_dim_in, 1); + space = isl_multi_pw_aff_get_domain_space(tuple); + aff = isl_aff_zero_on_domain(isl_local_space_from_space(space)); + aff = isl_aff_add_coefficient_si(aff, isl_dim_in, v->n, 1); + pa = isl_pw_aff_from_aff(aff); + tuple = isl_multi_pw_aff_flat_range_product(tuple, + isl_multi_pw_aff_from_pw_aff(pa)); - return isl_set_unwrap(set); -error: - if (tok) - isl_token_free(tok); - isl_pw_aff_list_free(list1); - isl_pw_aff_list_free(list2); - isl_set_free(set); - return NULL; + return tuple; } -static __isl_give isl_map *read_exists(struct isl_stream *s, - struct vars *v, __isl_take isl_map *map) +/* Set the name of dimension "pos" in "tuple" to "name". + * During printing, we add primes if the same name appears more than once + * to distinguish the occurrences. Here, we remove those primes from "name" + * before setting the name of the dimension. + */ +static __isl_give isl_multi_pw_aff *tuple_set_dim_name( + __isl_take isl_multi_pw_aff *tuple, int pos, char *name) { - int n = v->n; - int seen_paren = isl_stream_eat_if_available(s, '('); - - map = isl_map_from_domain(isl_map_wrap(map)); - map = read_defined_var_list(s, v, map); + char *prime; - if (isl_stream_eat(s, ':')) - goto error; + if (!name) + return tuple; - map = read_disjuncts(s, v, map); + prime = strchr(name, '\''); + if (prime) + *prime = '\0'; + tuple = isl_multi_pw_aff_set_dim_name(tuple, isl_dim_set, pos, name); + if (prime) + *prime = '\''; + + return tuple; +} + +/* Read an affine expression from "s" and replace the definition + * of dimension "pos" in "tuple" by this expression. + * + * accept_extended_affine requires a wrapped space as input. + * The domain space of "tuple", on the other hand is an anonymous space, + * so we have to adjust the space of the isl_pw_aff before adding it + * to "tuple". + */ +static __isl_give isl_multi_pw_aff *read_tuple_var_def(struct isl_stream *s, + __isl_take isl_multi_pw_aff *tuple, int pos, struct vars *v, + int rational) +{ + isl_space *space; + isl_pw_aff *def; + + space = isl_space_wrap(isl_space_alloc(s->ctx, 0, v->n, 0)); + def = accept_extended_affine(s, space, v, rational); + space = isl_space_set_alloc(s->ctx, 0, v->n); + def = isl_pw_aff_reset_domain_space(def, space); + tuple = isl_multi_pw_aff_set_pw_aff(tuple, pos, def); + + return tuple; +} + +/* Read a list of variables and/or affine expressions and return the list + * as an isl_multi_pw_aff. + * The elements in the list are separated by either "," or "][". + * If "comma" is set then only "," is allowed. + */ +static __isl_give isl_multi_pw_aff *read_tuple_var_list(struct isl_stream *s, + struct vars *v, int rational, int comma) +{ + int i = 0; + struct isl_token *tok; + isl_multi_pw_aff *res; + + res = tuple_alloc(v); + + if (isl_stream_next_token_is(s, ']')) + return res; + + while ((tok = next_token(s)) != NULL) { + int new_name = 0; + + res = tuple_add_dim(res, v); + + if (tok->type == ISL_TOKEN_IDENT) { + int n = v->n; + int p = vars_pos(v, tok->u.s, -1); + if (p < 0) + goto error; + new_name = p >= n; + } + + if (tok->type == '*') { + if (vars_add_anon(v) < 0) + goto error; + isl_token_free(tok); + } else if (new_name) { + res = tuple_set_dim_name(res, i, v->v->name); + isl_token_free(tok); + if (isl_stream_eat_if_available(s, '=')) + res = read_tuple_var_def(s, res, i, v, + rational); + } else { + isl_stream_push_token(s, tok); + tok = NULL; + if (vars_add_anon(v) < 0) + goto error; + res = read_tuple_var_def(s, res, i, v, rational); + } + + tok = isl_stream_next_token(s); + if (!comma && tok && tok->type == ']' && + isl_stream_next_token_is(s, '[')) { + isl_token_free(tok); + tok = isl_stream_next_token(s); + } else if (!tok || tok->type != ',') + break; + + isl_token_free(tok); + i++; + } + if (tok) + isl_stream_push_token(s, tok); + + return res; +error: + isl_token_free(tok); + return isl_multi_pw_aff_free(res); +} + +/* Read a tuple and represent it as an isl_multi_pw_aff. See tuple_alloc. + */ +static __isl_give isl_multi_pw_aff *read_tuple(struct isl_stream *s, + struct vars *v, int rational, int comma) +{ + struct isl_token *tok; + char *name = NULL; + isl_multi_pw_aff *res = NULL; + + tok = isl_stream_next_token(s); + if (!tok) + goto error; + if (tok->type == ISL_TOKEN_IDENT || tok->is_keyword) { + name = strdup(tok->u.s); + isl_token_free(tok); + if (!name) + goto error; + } else + isl_stream_push_token(s, tok); + if (isl_stream_eat(s, '[')) + goto error; + if (next_is_tuple(s)) { + isl_multi_pw_aff *out; + int n; + res = read_tuple(s, v, rational, comma); + if (isl_stream_eat(s, ISL_TOKEN_TO)) + goto error; + out = read_tuple(s, v, rational, comma); + n = isl_multi_pw_aff_dim(out, isl_dim_out); + res = isl_multi_pw_aff_add_dims(res, isl_dim_in, n); + res = isl_multi_pw_aff_range_product(res, out); + } else + res = read_tuple_var_list(s, v, rational, comma); + if (isl_stream_eat(s, ']')) + goto error; + + if (name) { + res = isl_multi_pw_aff_set_tuple_name(res, isl_dim_out, name); + free(name); + } + + return res; +error: + free(name); + return isl_multi_pw_aff_free(res); +} + +/* Read a tuple from "s" and add it to "map". + * The tuple is initially represented as an isl_multi_pw_aff. + * We first create the appropriate space in "map" based on the range + * space of this isl_multi_pw_aff. Then, we add equalities based + * on the affine expressions. These live in an anonymous space, + * however, so we first need to reset the space to that of "map". + */ +static __isl_give isl_map *read_map_tuple(struct isl_stream *s, + __isl_take isl_map *map, enum isl_dim_type type, struct vars *v, + int rational, int comma) +{ + int i, n; + isl_multi_pw_aff *tuple; + isl_space *space = NULL; + + tuple = read_tuple(s, v, rational, comma); + if (!tuple) + goto error; + + n = isl_multi_pw_aff_dim(tuple, isl_dim_out); + space = isl_space_range(isl_multi_pw_aff_get_space(tuple)); + if (!space) + goto error; + + if (type == isl_dim_param) { + if (isl_space_has_tuple_name(space, isl_dim_set) || + isl_space_is_wrapping(space)) { + isl_die(s->ctx, isl_error_invalid, + "parameter tuples cannot be named or nested", + goto error); + } + map = isl_map_add_dims(map, type, n); + for (i = 0; i < n; ++i) { + isl_id *id; + if (!isl_space_has_dim_name(space, isl_dim_set, i)) + isl_die(s->ctx, isl_error_invalid, + "parameters must be named", + goto error); + id = isl_space_get_dim_id(space, isl_dim_set, i); + map = isl_map_set_dim_id(map, isl_dim_param, i, id); + } + } else if (type == isl_dim_in) { + isl_set *set; + + set = isl_set_universe(isl_space_copy(space)); + if (rational) + set = isl_set_set_rational(set); + set = isl_set_intersect_params(set, isl_map_params(map)); + map = isl_map_from_domain(set); + } else { + isl_set *set; + + set = isl_set_universe(isl_space_copy(space)); + if (rational) + set = isl_set_set_rational(set); + map = isl_map_from_domain_and_range(isl_map_domain(map), set); + } + + for (i = 0; i < n; ++i) { + isl_pw_aff *pa; + isl_space *space; + isl_aff *aff; + isl_set *set; + isl_map *map_i; + + pa = isl_multi_pw_aff_get_pw_aff(tuple, i); + space = isl_pw_aff_get_domain_space(pa); + aff = isl_aff_zero_on_domain(isl_local_space_from_space(space)); + aff = isl_aff_add_coefficient_si(aff, + isl_dim_in, v->n - n + i, -1); + pa = isl_pw_aff_add(pa, isl_pw_aff_from_aff(aff)); + if (rational) + pa = isl_pw_aff_set_rational(pa); + set = isl_pw_aff_zero_set(pa); + map_i = isl_map_from_range(set); + map_i = isl_map_reset_space(map_i, isl_map_get_space(map)); + map = isl_map_intersect(map, map_i); + } + + isl_space_free(space); + isl_multi_pw_aff_free(tuple); + return map; +error: + isl_space_free(space); + isl_multi_pw_aff_free(tuple); + isl_map_free(map); + return NULL; +} + +static __isl_give isl_set *construct_constraints( + __isl_take isl_set *set, int type, + __isl_keep isl_pw_aff_list *left, __isl_keep isl_pw_aff_list *right, + int rational) +{ + isl_set *cond; + + left = isl_pw_aff_list_copy(left); + right = isl_pw_aff_list_copy(right); + if (rational) { + left = isl_pw_aff_list_set_rational(left); + right = isl_pw_aff_list_set_rational(right); + } + if (type == ISL_TOKEN_LE) + cond = isl_pw_aff_list_le_set(left, right); + else if (type == ISL_TOKEN_GE) + cond = isl_pw_aff_list_ge_set(left, right); + else if (type == ISL_TOKEN_LT) + cond = isl_pw_aff_list_lt_set(left, right); + else if (type == ISL_TOKEN_GT) + cond = isl_pw_aff_list_gt_set(left, right); + else if (type == ISL_TOKEN_NE) + cond = isl_pw_aff_list_ne_set(left, right); + else + cond = isl_pw_aff_list_eq_set(left, right); + + return isl_set_intersect(set, cond); +} + +static __isl_give isl_map *add_constraint(struct isl_stream *s, + struct vars *v, __isl_take isl_map *map, int rational) +{ + struct isl_token *tok = NULL; + isl_pw_aff_list *list1 = NULL, *list2 = NULL; + isl_set *set; + + set = isl_map_wrap(map); + list1 = accept_affine_list(s, isl_set_get_space(set), v); + if (!list1) + goto error; + tok = isl_stream_next_token(s); + if (!is_comparator(tok)) { + isl_stream_error(s, tok, "missing operator"); + if (tok) + isl_stream_push_token(s, tok); + tok = NULL; + goto error; + } + for (;;) { + list2 = accept_affine_list(s, isl_set_get_space(set), v); + if (!list2) + goto error; + + set = construct_constraints(set, tok->type, list1, list2, + rational); + isl_token_free(tok); + isl_pw_aff_list_free(list1); + list1 = list2; + + tok = isl_stream_next_token(s); + if (!is_comparator(tok)) { + if (tok) + isl_stream_push_token(s, tok); + break; + } + } + isl_pw_aff_list_free(list1); + + return isl_set_unwrap(set); +error: + if (tok) + isl_token_free(tok); + isl_pw_aff_list_free(list1); + isl_pw_aff_list_free(list2); + isl_set_free(set); + return NULL; +} + +static __isl_give isl_map *read_exists(struct isl_stream *s, + struct vars *v, __isl_take isl_map *map, int rational) +{ + int n = v->n; + int seen_paren = isl_stream_eat_if_available(s, '('); + + map = isl_map_from_domain(isl_map_wrap(map)); + map = read_defined_var_list(s, v, map, rational); + + if (isl_stream_eat(s, ':')) + goto error; + + map = read_disjuncts(s, v, map, rational); map = isl_set_unwrap(isl_map_domain(map)); vars_drop(v, v->n - n); @@ -1037,7 +1316,7 @@ error: * start of a condition and continue parsing. */ static int resolve_paren_expr(struct isl_stream *s, - struct vars *v, __isl_take isl_map *map) + struct vars *v, __isl_take isl_map *map, int rational) { struct isl_token *tok, *tok2; int line, col; @@ -1047,10 +1326,16 @@ static int resolve_paren_expr(struct isl_stream *s, if (!tok || tok->type != '(') goto error; + if (isl_stream_next_token_is(s, '(')) + if (resolve_paren_expr(s, v, isl_map_copy(map), rational)) + goto error; + if (isl_stream_next_token_is(s, ISL_TOKEN_EXISTS) || + isl_stream_next_token_is(s, ISL_TOKEN_NOT) || isl_stream_next_token_is(s, ISL_TOKEN_TRUE) || - isl_stream_next_token_is(s, ISL_TOKEN_FALSE)) { - map = read_disjuncts(s, v, map); + isl_stream_next_token_is(s, ISL_TOKEN_FALSE) || + isl_stream_next_token_is(s, ISL_TOKEN_MAP)) { + map = read_disjuncts(s, v, map, rational); if (isl_stream_eat(s, ')')) goto error; tok->type = ISL_TOKEN_MAP; @@ -1066,7 +1351,7 @@ static int resolve_paren_expr(struct isl_stream *s, col = tok2->col; isl_stream_push_token(s, tok2); - pwaff = accept_affine(s, isl_dim_wrap(isl_map_get_dim(map)), v); + pwaff = accept_affine(s, isl_space_wrap(isl_map_get_space(map)), v); if (!pwaff) goto error; @@ -1085,7 +1370,7 @@ static int resolve_paren_expr(struct isl_stream *s, isl_stream_push_token(s, tok2); - map = read_disjuncts(s, v, map); + map = read_disjuncts(s, v, map, rational); if (isl_stream_eat(s, ')')) goto error; @@ -1103,10 +1388,10 @@ error: } static __isl_give isl_map *read_conjunct(struct isl_stream *s, - struct vars *v, __isl_take isl_map *map) + struct vars *v, __isl_take isl_map *map, int rational) { if (isl_stream_next_token_is(s, '(')) - if (resolve_paren_expr(s, v, isl_map_copy(map))) + if (resolve_paren_expr(s, v, isl_map_copy(map), rational)) goto error; if (isl_stream_next_token_is(s, ISL_TOKEN_MAP)) { @@ -1121,39 +1406,39 @@ static __isl_give isl_map *read_conjunct(struct isl_stream *s, } if (isl_stream_eat_if_available(s, ISL_TOKEN_EXISTS)) - return read_exists(s, v, map); + return read_exists(s, v, map, rational); if (isl_stream_eat_if_available(s, ISL_TOKEN_TRUE)) return map; if (isl_stream_eat_if_available(s, ISL_TOKEN_FALSE)) { - isl_dim *dim = isl_map_get_dim(map); + isl_space *dim = isl_map_get_space(map); isl_map_free(map); return isl_map_empty(dim); } - return add_constraint(s, v, map); + return add_constraint(s, v, map, rational); error: isl_map_free(map); return NULL; } static __isl_give isl_map *read_conjuncts(struct isl_stream *s, - struct vars *v, __isl_take isl_map *map) + struct vars *v, __isl_take isl_map *map, int rational) { isl_map *res; int negate; negate = isl_stream_eat_if_available(s, ISL_TOKEN_NOT); - res = read_conjunct(s, v, isl_map_copy(map)); + res = read_conjunct(s, v, isl_map_copy(map), rational); if (negate) res = isl_map_subtract(isl_map_copy(map), res); - while (isl_stream_eat_if_available(s, ISL_TOKEN_AND)) { + while (res && isl_stream_eat_if_available(s, ISL_TOKEN_AND)) { isl_map *res_i; negate = isl_stream_eat_if_available(s, ISL_TOKEN_NOT); - res_i = read_conjunct(s, v, isl_map_copy(map)); + res_i = read_conjunct(s, v, isl_map_copy(map), rational); if (negate) res = isl_map_subtract(res, res_i); else @@ -1165,21 +1450,21 @@ static __isl_give isl_map *read_conjuncts(struct isl_stream *s, } static struct isl_map *read_disjuncts(struct isl_stream *s, - struct vars *v, __isl_take isl_map *map) + struct vars *v, __isl_take isl_map *map, int rational) { isl_map *res; if (isl_stream_next_token_is(s, '}')) { - isl_dim *dim = isl_map_get_dim(map); + isl_space *dim = isl_map_get_space(map); isl_map_free(map); return isl_map_universe(dim); } - res = read_conjuncts(s, v, isl_map_copy(map)); + res = read_conjuncts(s, v, isl_map_copy(map), rational); while (isl_stream_eat_if_available(s, ISL_TOKEN_OR)) { isl_map *res_i; - res_i = read_conjuncts(s, v, isl_map_copy(map)); + res_i = read_conjuncts(s, v, isl_map_copy(map), rational); res = isl_map_union(res, res_i); } @@ -1280,8 +1565,7 @@ error: return NULL; } -static __isl_give isl_basic_map *basic_map_read_polylib(struct isl_stream *s, - int nparam) +static __isl_give isl_basic_map *basic_map_read_polylib(struct isl_stream *s) { int i; struct isl_token *tok; @@ -1290,9 +1574,7 @@ static __isl_give isl_basic_map *basic_map_read_polylib(struct isl_stream *s, int on_new_line; unsigned in = 0, out, local = 0; struct isl_basic_map *bmap = NULL; - - if (nparam < 0) - nparam = 0; + int nparam = 0; tok = isl_stream_next_token(s); if (!tok) { @@ -1399,7 +1681,7 @@ error: return NULL; } -static struct isl_map *map_read_polylib(struct isl_stream *s, int nparam) +static struct isl_map *map_read_polylib(struct isl_stream *s) { struct isl_token *tok; struct isl_token *tok2; @@ -1415,7 +1697,7 @@ static struct isl_map *map_read_polylib(struct isl_stream *s, int nparam) if (tok2 && tok2->type == ISL_TOKEN_VALUE) { isl_stream_push_token(s, tok2); isl_stream_push_token(s, tok); - return isl_map_from_basic_map(basic_map_read_polylib(s, nparam)); + return isl_map_from_basic_map(basic_map_read_polylib(s)); } if (tok2) { isl_stream_error(s, tok2, "unexpected token"); @@ -1428,11 +1710,11 @@ static struct isl_map *map_read_polylib(struct isl_stream *s, int nparam) isl_assert(s->ctx, n >= 1, return NULL); - map = isl_map_from_basic_map(basic_map_read_polylib(s, nparam)); + map = isl_map_from_basic_map(basic_map_read_polylib(s)); for (i = 1; map && i < n; ++i) map = isl_map_union(map, - isl_map_from_basic_map(basic_map_read_polylib(s, nparam))); + isl_map_from_basic_map(basic_map_read_polylib(s))); return map; } @@ -1489,8 +1771,9 @@ static __isl_give isl_pw_qpolynomial *read_factor(struct isl_stream *s, pwqp = isl_pw_qpolynomial_pow(pwqp, pow); } else if (tok->type == ISL_TOKEN_VALUE) { struct isl_token *tok2; - tok2 = isl_stream_next_token(s); isl_qpolynomial *qp; + + tok2 = isl_stream_next_token(s); if (tok2 && tok2->type == '/') { isl_token_free(tok2); tok2 = next_token(s); @@ -1500,12 +1783,12 @@ static __isl_give isl_pw_qpolynomial *read_factor(struct isl_stream *s, isl_token_free(tok2); return NULL; } - qp = isl_qpolynomial_rat_cst(isl_map_get_dim(map), + qp = isl_qpolynomial_rat_cst_on_domain(isl_map_get_space(map), tok->u.v, tok2->u.v); isl_token_free(tok2); } else { isl_stream_push_token(s, tok2); - qp = isl_qpolynomial_cst(isl_map_get_dim(map), + qp = isl_qpolynomial_cst_on_domain(isl_map_get_space(map), tok->u.v); } isl_token_free(tok); @@ -1513,12 +1796,12 @@ static __isl_give isl_pw_qpolynomial *read_factor(struct isl_stream *s, } else if (tok->type == ISL_TOKEN_INFTY) { isl_qpolynomial *qp; isl_token_free(tok); - qp = isl_qpolynomial_infty(isl_map_get_dim(map)); + qp = isl_qpolynomial_infty_on_domain(isl_map_get_space(map)); pwqp = isl_pw_qpolynomial_from_qpolynomial(qp); } else if (tok->type == ISL_TOKEN_NAN) { isl_qpolynomial *qp; isl_token_free(tok); - qp = isl_qpolynomial_nan(isl_map_get_dim(map)); + qp = isl_qpolynomial_nan_on_domain(isl_map_get_space(map)); pwqp = isl_pw_qpolynomial_from_qpolynomial(qp); } else if (tok->type == ISL_TOKEN_IDENT) { int n = v->n; @@ -1537,14 +1820,14 @@ static __isl_give isl_pw_qpolynomial *read_factor(struct isl_stream *s, } isl_token_free(tok); pow = optional_power(s); - qp = isl_qpolynomial_var_pow(isl_map_get_dim(map), pos, pow); + qp = isl_qpolynomial_var_pow_on_domain(isl_map_get_space(map), pos, pow); pwqp = isl_pw_qpolynomial_from_qpolynomial(qp); } else if (tok->type == '[') { isl_pw_aff *pwaff; int pow; isl_stream_push_token(s, tok); - pwaff = accept_affine(s, isl_map_get_dim(map), v); + pwaff = accept_div(s, isl_map_get_space(map), v); pow = optional_power(s); pwqp = isl_pw_qpolynomial_from_pw_aff(pwaff); pwqp = isl_pw_qpolynomial_pow(pwqp, pow); @@ -1614,7 +1897,7 @@ static __isl_give isl_pw_qpolynomial *read_term(struct isl_stream *s, } static __isl_give isl_map *read_optional_disjuncts(struct isl_stream *s, - __isl_take isl_map *map, struct vars *v) + __isl_take isl_map *map, struct vars *v, int rational) { struct isl_token *tok; @@ -1626,7 +1909,7 @@ static __isl_give isl_map *read_optional_disjuncts(struct isl_stream *s, if (tok->type == ':' || (tok->type == ISL_TOKEN_OR && !strcmp(tok->u.s, "|"))) { isl_token_free(tok); - map = read_disjuncts(s, v, map); + map = read_disjuncts(s, v, map, rational); } else isl_stream_push_token(s, tok); @@ -1644,7 +1927,7 @@ static struct isl_obj obj_read_poly(struct isl_stream *s, struct isl_set *set; pwqp = read_term(s, map, v); - map = read_optional_disjuncts(s, map, v); + map = read_optional_disjuncts(s, map, v, 0); set = isl_map_range(map); pwqp = isl_pw_qpolynomial_intersect_domain(pwqp, set); @@ -1656,25 +1939,24 @@ static struct isl_obj obj_read_poly(struct isl_stream *s, } static struct isl_obj obj_read_poly_or_fold(struct isl_stream *s, - __isl_take isl_map *map, struct vars *v, int n) + __isl_take isl_set *set, struct vars *v, int n) { struct isl_obj obj = { isl_obj_pw_qpolynomial_fold, NULL }; isl_pw_qpolynomial *pwqp; isl_pw_qpolynomial_fold *pwf = NULL; - isl_set *set; if (!isl_stream_eat_if_available(s, ISL_TOKEN_MAX)) - return obj_read_poly(s, map, v, n); + return obj_read_poly(s, set, v, n); if (isl_stream_eat(s, '(')) goto error; - pwqp = read_term(s, map, v); + pwqp = read_term(s, set, v); pwf = isl_pw_qpolynomial_fold_from_pw_qpolynomial(isl_fold_max, pwqp); while (isl_stream_eat_if_available(s, ',')) { isl_pw_qpolynomial_fold *pwf_i; - pwqp = read_term(s, map, v); + pwqp = read_term(s, set, v); pwf_i = isl_pw_qpolynomial_fold_from_pw_qpolynomial(isl_fold_max, pwqp); pwf = isl_pw_qpolynomial_fold_fold(pwf, pwf_i); @@ -1683,8 +1965,7 @@ static struct isl_obj obj_read_poly_or_fold(struct isl_stream *s, if (isl_stream_eat(s, ')')) goto error; - map = read_optional_disjuncts(s, map, v); - set = isl_map_range(map); + set = read_optional_disjuncts(s, set, v, 0); pwf = isl_pw_qpolynomial_fold_intersect_domain(pwf, set); vars_drop(v, v->n - n); @@ -1692,7 +1973,7 @@ static struct isl_obj obj_read_poly_or_fold(struct isl_stream *s, obj.v = pwf; return obj; error: - isl_map_free(map); + isl_set_free(set); isl_pw_qpolynomial_fold_free(pwf); obj.type = isl_obj_none; return obj; @@ -1722,34 +2003,43 @@ static struct isl_obj obj_read_body(struct isl_stream *s, struct isl_token *tok; struct isl_obj obj = { isl_obj_set, NULL }; int n = v->n; + int rational; - if (is_rational(s)) + rational = is_rational(s); + if (rational) map = isl_map_set_rational(map); + if (isl_stream_next_token_is(s, ':')) { + obj.type = isl_obj_set; + obj.v = read_optional_disjuncts(s, map, v, rational); + return obj; + } + if (!next_is_tuple(s)) return obj_read_poly_or_fold(s, map, v, n); - map = read_tuple(s, map, isl_dim_in, v); + map = read_map_tuple(s, map, isl_dim_in, v, rational, 0); if (!map) goto error; tok = isl_stream_next_token(s); - if (tok && tok->type == ISL_TOKEN_TO) { + if (!tok) + goto error; + if (tok->type == ISL_TOKEN_TO) { obj.type = isl_obj_map; isl_token_free(tok); if (!next_is_tuple(s)) { - map = isl_map_reverse(map); - return obj_read_poly_or_fold(s, map, v, n); + isl_set *set = isl_map_domain(map); + return obj_read_poly_or_fold(s, set, v, n); } - map = read_tuple(s, map, isl_dim_out, v); + map = read_map_tuple(s, map, isl_dim_out, v, rational, 0); if (!map) goto error; } else { - map = isl_map_reverse(map); - if (tok) - isl_stream_push_token(s, tok); + map = isl_map_domain(map); + isl_stream_push_token(s, tok); } - map = read_optional_disjuncts(s, map, v); + map = read_optional_disjuncts(s, map, v, rational); vars_drop(v, v->n - n); @@ -1808,21 +2098,21 @@ static struct isl_obj obj_add(struct isl_ctx *ctx, obj2.type == isl_obj_pw_qpolynomial_fold) obj2 = to_union(ctx, obj2); isl_assert(ctx, obj1.type == obj2.type, goto error); - if (obj1.type == isl_obj_map && !isl_map_has_equal_dim(obj1.v, obj2.v)) { + if (obj1.type == isl_obj_map && !isl_map_has_equal_space(obj1.v, obj2.v)) { obj1 = to_union(ctx, obj1); obj2 = to_union(ctx, obj2); } - if (obj1.type == isl_obj_set && !isl_set_has_equal_dim(obj1.v, obj2.v)) { + if (obj1.type == isl_obj_set && !isl_set_has_equal_space(obj1.v, obj2.v)) { obj1 = to_union(ctx, obj1); obj2 = to_union(ctx, obj2); } if (obj1.type == isl_obj_pw_qpolynomial && - !isl_pw_qpolynomial_has_equal_dim(obj1.v, obj2.v)) { + !isl_pw_qpolynomial_has_equal_space(obj1.v, obj2.v)) { obj1 = to_union(ctx, obj1); obj2 = to_union(ctx, obj2); } if (obj1.type == isl_obj_pw_qpolynomial_fold && - !isl_pw_qpolynomial_fold_has_equal_dim(obj1.v, obj2.v)) { + !isl_pw_qpolynomial_fold_has_equal_space(obj1.v, obj2.v)) { obj1 = to_union(ctx, obj1); obj2 = to_union(ctx, obj2); } @@ -1836,7 +2126,7 @@ error: return obj1; } -static struct isl_obj obj_read(struct isl_stream *s, int nparam) +static struct isl_obj obj_read(struct isl_stream *s) { isl_map *map = NULL; struct isl_token *tok; @@ -1864,12 +2154,15 @@ static struct isl_obj obj_read(struct isl_stream *s, int nparam) } isl_stream_push_token(s, tok2); isl_stream_push_token(s, tok); - map = map_read_polylib(s, nparam); + map = map_read_polylib(s); if (!map) goto error; - if (isl_map_dim(map, isl_dim_in) > 0) + if (isl_map_may_be_set(map)) + obj.v = isl_map_range(map); + else { obj.type = isl_obj_map; - obj.v = map; + obj.v = map; + } return obj; } v = vars_new(s->ctx); @@ -1877,14 +2170,12 @@ static struct isl_obj obj_read(struct isl_stream *s, int nparam) isl_stream_push_token(s, tok); goto error; } - map = isl_map_universe(isl_dim_alloc(s->ctx, 0, 0, 0)); + map = isl_map_universe(isl_space_params_alloc(s->ctx, 0)); if (tok->type == '[') { isl_stream_push_token(s, tok); - map = read_tuple(s, map, isl_dim_param, v); + map = read_map_tuple(s, map, isl_dim_param, v, 0, 0); if (!map) goto error; - if (nparam >= 0) - isl_assert(s->ctx, nparam == v->n, goto error); tok = isl_stream_next_token(s); if (!tok || tok->type != ISL_TOKEN_TO) { isl_stream_error(s, tok, "expecting '->'"); @@ -1894,8 +2185,7 @@ static struct isl_obj obj_read(struct isl_stream *s, int nparam) } isl_token_free(tok); tok = isl_stream_next_token(s); - } else if (nparam > 0) - map = isl_map_add_dims(map, isl_dim_param, nparam); + } if (!tok || tok->type != '{') { isl_stream_error(s, tok, "expecting '{'"); if (tok) @@ -1911,14 +2201,12 @@ static struct isl_obj obj_read(struct isl_stream *s, int nparam) isl_token_free(tok); if (isl_stream_eat(s, '=')) goto error; - map = read_tuple(s, map, isl_dim_param, v); + map = read_map_tuple(s, map, isl_dim_param, v, 0, 1); if (!map) goto error; - if (nparam >= 0) - isl_assert(s->ctx, nparam == v->n, goto error); } else if (tok->type == '}') { obj.type = isl_obj_union_set; - obj.v = isl_union_set_empty(isl_map_get_dim(map)); + obj.v = isl_union_set_empty(isl_map_get_space(map)); isl_token_free(tok); goto done; } else @@ -1971,17 +2259,20 @@ error: struct isl_obj isl_stream_read_obj(struct isl_stream *s) { - return obj_read(s, -1); + return obj_read(s); } -__isl_give isl_map *isl_stream_read_map(struct isl_stream *s, int nparam) +__isl_give isl_map *isl_stream_read_map(struct isl_stream *s) { struct isl_obj obj; - obj = obj_read(s, nparam); + obj = obj_read(s); if (obj.v) isl_assert(s->ctx, obj.type == isl_obj_map || obj.type == isl_obj_set, goto error); + + if (obj.type == isl_obj_set) + obj.v = isl_map_from_range(obj.v); return obj.v; error: @@ -1989,13 +2280,18 @@ error: return NULL; } -__isl_give isl_set *isl_stream_read_set(struct isl_stream *s, int nparam) +__isl_give isl_set *isl_stream_read_set(struct isl_stream *s) { struct isl_obj obj; - obj = obj_read(s, nparam); - if (obj.v) + obj = obj_read(s); + if (obj.v) { + if (obj.type == isl_obj_map && isl_map_may_be_set(obj.v)) { + obj.v = isl_map_range(obj.v); + obj.type = isl_obj_set; + } isl_assert(s->ctx, obj.type == isl_obj_set, goto error); + } return obj.v; error: @@ -2007,7 +2303,7 @@ __isl_give isl_union_map *isl_stream_read_union_map(struct isl_stream *s) { struct isl_obj obj; - obj = obj_read(s, -1); + obj = obj_read(s); if (obj.type == isl_obj_map) { obj.type = isl_obj_union_map; obj.v = isl_union_map_from_map(obj.v); @@ -2016,9 +2312,11 @@ __isl_give isl_union_map *isl_stream_read_union_map(struct isl_stream *s) obj.type = isl_obj_union_set; obj.v = isl_union_set_from_set(obj.v); } - if (obj.v) - isl_assert(s->ctx, obj.type == isl_obj_union_map || - obj.type == isl_obj_union_set, goto error); + if (obj.v && obj.type == isl_obj_union_set && + isl_union_set_is_empty(obj.v)) + obj.type = isl_obj_union_map; + if (obj.v && obj.type != isl_obj_union_map) + isl_die(s->ctx, isl_error_invalid, "invalid input", goto error); return obj.v; error: @@ -2030,7 +2328,7 @@ __isl_give isl_union_set *isl_stream_read_union_set(struct isl_stream *s) { struct isl_obj obj; - obj = obj_read(s, -1); + obj = obj_read(s); if (obj.type == isl_obj_set) { obj.type = isl_obj_union_set; obj.v = isl_union_set_from_set(obj.v); @@ -2044,13 +2342,13 @@ error: return NULL; } -static struct isl_basic_map *basic_map_read(struct isl_stream *s, int nparam) +static __isl_give isl_basic_map *basic_map_read(struct isl_stream *s) { struct isl_obj obj; struct isl_map *map; struct isl_basic_map *bmap; - obj = obj_read(s, nparam); + obj = obj_read(s); map = obj.v; if (!map) return NULL; @@ -2070,108 +2368,115 @@ error: return NULL; } +static __isl_give isl_basic_set *basic_set_read(struct isl_stream *s) +{ + isl_basic_map *bmap; + bmap = basic_map_read(s); + if (!bmap) + return NULL; + if (!isl_basic_map_may_be_set(bmap)) + isl_die(s->ctx, isl_error_invalid, + "input is not a set", goto error); + return isl_basic_map_range(bmap); +error: + isl_basic_map_free(bmap); + return NULL; +} + __isl_give isl_basic_map *isl_basic_map_read_from_file(isl_ctx *ctx, - FILE *input, int nparam) + FILE *input) { struct isl_basic_map *bmap; struct isl_stream *s = isl_stream_new_file(ctx, input); if (!s) return NULL; - bmap = basic_map_read(s, nparam); + bmap = basic_map_read(s); isl_stream_free(s); return bmap; } __isl_give isl_basic_set *isl_basic_set_read_from_file(isl_ctx *ctx, - FILE *input, int nparam) + FILE *input) { - struct isl_basic_map *bmap; - bmap = isl_basic_map_read_from_file(ctx, input, nparam); - if (!bmap) + isl_basic_set *bset; + struct isl_stream *s = isl_stream_new_file(ctx, input); + if (!s) return NULL; - isl_assert(ctx, isl_basic_map_n_in(bmap) == 0, goto error); - return (struct isl_basic_set *)bmap; -error: - isl_basic_map_free(bmap); - return NULL; + bset = basic_set_read(s); + isl_stream_free(s); + return bset; } struct isl_basic_map *isl_basic_map_read_from_str(struct isl_ctx *ctx, - const char *str, int nparam) + const char *str) { struct isl_basic_map *bmap; struct isl_stream *s = isl_stream_new_str(ctx, str); if (!s) return NULL; - bmap = basic_map_read(s, nparam); + bmap = basic_map_read(s); isl_stream_free(s); return bmap; } struct isl_basic_set *isl_basic_set_read_from_str(struct isl_ctx *ctx, - const char *str, int nparam) + const char *str) { - struct isl_basic_map *bmap; - bmap = isl_basic_map_read_from_str(ctx, str, nparam); - if (!bmap) + isl_basic_set *bset; + struct isl_stream *s = isl_stream_new_str(ctx, str); + if (!s) return NULL; - isl_assert(ctx, isl_basic_map_n_in(bmap) == 0, goto error); - return (struct isl_basic_set *)bmap; -error: - isl_basic_map_free(bmap); - return NULL; + bset = basic_set_read(s); + isl_stream_free(s); + return bset; } __isl_give isl_map *isl_map_read_from_file(struct isl_ctx *ctx, - FILE *input, int nparam) + FILE *input) { struct isl_map *map; struct isl_stream *s = isl_stream_new_file(ctx, input); if (!s) return NULL; - map = isl_stream_read_map(s, nparam); + map = isl_stream_read_map(s); isl_stream_free(s); return map; } __isl_give isl_map *isl_map_read_from_str(struct isl_ctx *ctx, - const char *str, int nparam) + const char *str) { struct isl_map *map; struct isl_stream *s = isl_stream_new_str(ctx, str); if (!s) return NULL; - map = isl_stream_read_map(s, nparam); + map = isl_stream_read_map(s); isl_stream_free(s); return map; } __isl_give isl_set *isl_set_read_from_file(struct isl_ctx *ctx, - FILE *input, int nparam) + FILE *input) { - struct isl_map *map; - map = isl_map_read_from_file(ctx, input, nparam); - if (!map) + isl_set *set; + struct isl_stream *s = isl_stream_new_file(ctx, input); + if (!s) return NULL; - isl_assert(ctx, isl_map_n_in(map) == 0, goto error); - return (struct isl_set *)map; -error: - isl_map_free(map); - return NULL; + set = isl_stream_read_set(s); + isl_stream_free(s); + return set; } struct isl_set *isl_set_read_from_str(struct isl_ctx *ctx, - const char *str, int nparam) + const char *str) { - struct isl_map *map; - map = isl_map_read_from_str(ctx, str, nparam); - if (!map) + isl_set *set; + struct isl_stream *s = isl_stream_new_str(ctx, str); + if (!s) return NULL; - isl_assert(ctx, isl_map_n_in(map) == 0, goto error); - return (struct isl_set *)map; -error: - isl_map_free(map); - return NULL; + set = isl_stream_read_set(s); + isl_stream_free(s); + return set; } __isl_give isl_union_map *isl_union_map_read_from_file(isl_ctx *ctx, @@ -2278,7 +2583,7 @@ __isl_give isl_pw_qpolynomial *isl_stream_read_pw_qpolynomial( { struct isl_obj obj; - obj = obj_read(s, -1); + obj = obj_read(s); if (obj.v) isl_assert(s->ctx, obj.type == isl_obj_pw_qpolynomial, goto error); @@ -2312,3 +2617,430 @@ __isl_give isl_pw_qpolynomial *isl_pw_qpolynomial_read_from_file(isl_ctx *ctx, isl_stream_free(s); return pwqp; } + +/* Is the next token an identifer not in "v"? + */ +static int next_is_fresh_ident(struct isl_stream *s, struct vars *v) +{ + int n = v->n; + int fresh; + struct isl_token *tok; + + tok = isl_stream_next_token(s); + if (!tok) + return 0; + fresh = tok->type == ISL_TOKEN_IDENT && vars_pos(v, tok->u.s, -1) >= n; + isl_stream_push_token(s, tok); + + vars_drop(v, v->n - n); + + return fresh; +} + +/* First read the domain of the affine expression, which may be + * a parameter space or a set. + * The tricky part is that we don't know if the domain is a set or not, + * so when we are trying to read the domain, we may actually be reading + * the affine expression itself (defined on a parameter domains) + * If the tuple we are reading is named, we assume it's the domain. + * Also, if inside the tuple, the first thing we find is a nested tuple + * or a new identifier, we again assume it's the domain. + * Otherwise, we assume we are reading an affine expression. + */ +static __isl_give isl_set *read_aff_domain(struct isl_stream *s, + __isl_take isl_set *dom, struct vars *v) +{ + struct isl_token *tok; + + tok = isl_stream_next_token(s); + if (tok && (tok->type == ISL_TOKEN_IDENT || tok->is_keyword)) { + isl_stream_push_token(s, tok); + return read_map_tuple(s, dom, isl_dim_set, v, 1, 0); + } + if (!tok || tok->type != '[') { + isl_stream_error(s, tok, "expecting '['"); + goto error; + } + if (next_is_tuple(s) || next_is_fresh_ident(s, v)) { + isl_stream_push_token(s, tok); + dom = read_map_tuple(s, dom, isl_dim_set, v, 1, 0); + } else + isl_stream_push_token(s, tok); + + return dom; +error: + if (tok) + isl_stream_push_token(s, tok); + isl_set_free(dom); + return NULL; +} + +/* Read an affine expression from "s". + */ +__isl_give isl_aff *isl_stream_read_aff(struct isl_stream *s) +{ + isl_aff *aff; + isl_multi_aff *ma; + + ma = isl_stream_read_multi_aff(s); + if (!ma) + return NULL; + if (isl_multi_aff_dim(ma, isl_dim_out) != 1) + isl_die(s->ctx, isl_error_invalid, + "expecting single affine expression", + goto error); + + aff = isl_multi_aff_get_aff(ma, 0); + isl_multi_aff_free(ma); + return aff; +error: + isl_multi_aff_free(ma); + return NULL; +} + +/* Read a piecewise affine expression from "s" with domain (space) "dom". + */ +static __isl_give isl_pw_aff *read_pw_aff_with_dom(struct isl_stream *s, + __isl_take isl_set *dom, struct vars *v) +{ + isl_pw_aff *pwaff = NULL; + + if (!isl_set_is_params(dom) && isl_stream_eat(s, ISL_TOKEN_TO)) + goto error; + + if (isl_stream_eat(s, '[')) + goto error; + + pwaff = accept_affine(s, isl_set_get_space(dom), v); + + if (isl_stream_eat(s, ']')) + goto error; + + dom = read_optional_disjuncts(s, dom, v, 0); + pwaff = isl_pw_aff_intersect_domain(pwaff, dom); + + return pwaff; +error: + isl_set_free(dom); + isl_pw_aff_free(pwaff); + return NULL; +} + +__isl_give isl_pw_aff *isl_stream_read_pw_aff(struct isl_stream *s) +{ + struct vars *v; + isl_set *dom = NULL; + isl_set *aff_dom; + isl_pw_aff *pa = NULL; + int n; + + v = vars_new(s->ctx); + if (!v) + return NULL; + + dom = isl_set_universe(isl_space_params_alloc(s->ctx, 0)); + if (next_is_tuple(s)) { + dom = read_map_tuple(s, dom, isl_dim_param, v, 1, 0); + if (isl_stream_eat(s, ISL_TOKEN_TO)) + goto error; + } + if (isl_stream_eat(s, '{')) + goto error; + + n = v->n; + aff_dom = read_aff_domain(s, isl_set_copy(dom), v); + pa = read_pw_aff_with_dom(s, aff_dom, v); + vars_drop(v, v->n - n); + + while (isl_stream_eat_if_available(s, ';')) { + isl_pw_aff *pa_i; + + n = v->n; + aff_dom = read_aff_domain(s, isl_set_copy(dom), v); + pa_i = read_pw_aff_with_dom(s, aff_dom, v); + vars_drop(v, v->n - n); + + pa = isl_pw_aff_union_add(pa, pa_i); + } + + if (isl_stream_eat(s, '}')) + goto error; + + vars_free(v); + isl_set_free(dom); + return pa; +error: + vars_free(v); + isl_set_free(dom); + isl_pw_aff_free(pa); + return NULL; +} + +__isl_give isl_aff *isl_aff_read_from_str(isl_ctx *ctx, const char *str) +{ + isl_aff *aff; + struct isl_stream *s = isl_stream_new_str(ctx, str); + if (!s) + return NULL; + aff = isl_stream_read_aff(s); + isl_stream_free(s); + return aff; +} + +__isl_give isl_pw_aff *isl_pw_aff_read_from_str(isl_ctx *ctx, const char *str) +{ + isl_pw_aff *pa; + struct isl_stream *s = isl_stream_new_str(ctx, str); + if (!s) + return NULL; + pa = isl_stream_read_pw_aff(s); + isl_stream_free(s); + return pa; +} + +/* Read an isl_pw_multi_aff from "s". + * We currently read a generic object and if it turns out to be a set or + * a map, we convert that to an isl_pw_multi_aff. + * It would be more efficient if we were to construct the isl_pw_multi_aff + * directly. + */ +__isl_give isl_pw_multi_aff *isl_stream_read_pw_multi_aff(struct isl_stream *s) +{ + struct isl_obj obj; + + obj = obj_read(s); + if (!obj.v) + return NULL; + + if (obj.type == isl_obj_map) + return isl_pw_multi_aff_from_map(obj.v); + if (obj.type == isl_obj_set) + return isl_pw_multi_aff_from_set(obj.v); + + obj.type->free(obj.v); + isl_die(s->ctx, isl_error_invalid, "unexpected object type", + return NULL); +} + +__isl_give isl_pw_multi_aff *isl_pw_multi_aff_read_from_str(isl_ctx *ctx, + const char *str) +{ + isl_pw_multi_aff *pma; + struct isl_stream *s = isl_stream_new_str(ctx, str); + if (!s) + return NULL; + pma = isl_stream_read_pw_multi_aff(s); + isl_stream_free(s); + return pma; +} + +/* Read an isl_union_pw_multi_aff from "s". + * We currently read a generic object and if it turns out to be a set or + * a map, we convert that to an isl_union_pw_multi_aff. + * It would be more efficient if we were to construct + * the isl_union_pw_multi_aff directly. + */ +__isl_give isl_union_pw_multi_aff *isl_stream_read_union_pw_multi_aff( + struct isl_stream *s) +{ + struct isl_obj obj; + + obj = obj_read(s); + if (!obj.v) + return NULL; + + if (obj.type == isl_obj_map || obj.type == isl_obj_set) + obj = to_union(s->ctx, obj); + if (obj.type == isl_obj_union_map) + return isl_union_pw_multi_aff_from_union_map(obj.v); + if (obj.type == isl_obj_union_set) + return isl_union_pw_multi_aff_from_union_set(obj.v); + + obj.type->free(obj.v); + isl_die(s->ctx, isl_error_invalid, "unexpected object type", + return NULL); +} + +/* Read an isl_union_pw_multi_aff from "str". + */ +__isl_give isl_union_pw_multi_aff *isl_union_pw_multi_aff_read_from_str( + isl_ctx *ctx, const char *str) +{ + isl_union_pw_multi_aff *upma; + struct isl_stream *s = isl_stream_new_str(ctx, str); + if (!s) + return NULL; + upma = isl_stream_read_union_pw_multi_aff(s); + isl_stream_free(s); + return upma; +} + +/* Assuming "pa" represents a single affine expression defined on a universe + * domain, extract this affine expression. + */ +static __isl_give isl_aff *aff_from_pw_aff(__isl_take isl_pw_aff *pa) +{ + isl_aff *aff; + + if (!pa) + return NULL; + if (pa->n != 1) + isl_die(isl_pw_aff_get_ctx(pa), isl_error_invalid, + "expecting single affine expression", + goto error); + if (!isl_set_plain_is_universe(pa->p[0].set)) + isl_die(isl_pw_aff_get_ctx(pa), isl_error_invalid, + "expecting universe domain", + goto error); + + aff = isl_aff_copy(pa->p[0].aff); + isl_pw_aff_free(pa); + return aff; +error: + isl_pw_aff_free(pa); + return NULL; +} + +/* Read a multi-affine expression from "s". + * If the multi-affine expression has a domain, then then tuple + * representing this domain cannot involve any affine expressions. + * The tuple representing the actual expressions needs to consist + * of only affine expressions. Moreover, these expressions can + * only depend on parameters and input dimensions and not on other + * output dimensions. + */ +__isl_give isl_multi_aff *isl_stream_read_multi_aff(struct isl_stream *s) +{ + struct vars *v; + isl_set *dom = NULL; + isl_multi_pw_aff *tuple = NULL; + int dim, i, n; + isl_space *space, *dom_space; + isl_multi_aff *ma = NULL; + + v = vars_new(s->ctx); + if (!v) + return NULL; + + dom = isl_set_universe(isl_space_params_alloc(s->ctx, 0)); + if (next_is_tuple(s)) { + dom = read_map_tuple(s, dom, isl_dim_param, v, 1, 0); + if (isl_stream_eat(s, ISL_TOKEN_TO)) + goto error; + } + if (!isl_set_plain_is_universe(dom)) + isl_die(s->ctx, isl_error_invalid, + "expecting universe parameter domain", goto error); + if (isl_stream_eat(s, '{')) + goto error; + + tuple = read_tuple(s, v, 0, 0); + if (!tuple) + goto error; + if (isl_stream_eat_if_available(s, ISL_TOKEN_TO)) { + isl_set *set; + isl_space *space; + int has_expr; + + has_expr = tuple_has_expr(tuple); + if (has_expr < 0) + goto error; + if (has_expr) + isl_die(s->ctx, isl_error_invalid, + "expecting universe domain", goto error); + space = isl_space_range(isl_multi_pw_aff_get_space(tuple)); + set = isl_set_universe(space); + dom = isl_set_intersect_params(set, dom); + isl_multi_pw_aff_free(tuple); + tuple = read_tuple(s, v, 0, 0); + if (!tuple) + goto error; + } + + if (isl_stream_eat(s, '}')) + goto error; + + n = isl_multi_pw_aff_dim(tuple, isl_dim_out); + dim = isl_set_dim(dom, isl_dim_all); + dom_space = isl_set_get_space(dom); + space = isl_space_range(isl_multi_pw_aff_get_space(tuple)); + space = isl_space_align_params(space, isl_space_copy(dom_space)); + if (!isl_space_is_params(dom_space)) + space = isl_space_map_from_domain_and_range( + isl_space_copy(dom_space), space); + isl_space_free(dom_space); + ma = isl_multi_aff_alloc(space); + + for (i = 0; i < n; ++i) { + isl_pw_aff *pa; + isl_aff *aff; + pa = isl_multi_pw_aff_get_pw_aff(tuple, i); + aff = aff_from_pw_aff(pa); + if (!aff) + goto error; + if (isl_aff_involves_dims(aff, isl_dim_in, dim, i + 1)) { + isl_aff_free(aff); + isl_die(s->ctx, isl_error_invalid, + "not an affine expression", goto error); + } + aff = isl_aff_drop_dims(aff, isl_dim_in, dim, n); + space = isl_multi_aff_get_domain_space(ma); + aff = isl_aff_reset_domain_space(aff, space); + ma = isl_multi_aff_set_aff(ma, i, aff); + } + + isl_multi_pw_aff_free(tuple); + vars_free(v); + isl_set_free(dom); + return ma; +error: + isl_multi_pw_aff_free(tuple); + vars_free(v); + isl_set_free(dom); + isl_multi_aff_free(ma); + return NULL; +} + +__isl_give isl_multi_aff *isl_multi_aff_read_from_str(isl_ctx *ctx, + const char *str) +{ + isl_multi_aff *maff; + struct isl_stream *s = isl_stream_new_str(ctx, str); + if (!s) + return NULL; + maff = isl_stream_read_multi_aff(s); + isl_stream_free(s); + return maff; +} + +__isl_give isl_union_pw_qpolynomial *isl_stream_read_union_pw_qpolynomial( + struct isl_stream *s) +{ + struct isl_obj obj; + + obj = obj_read(s); + if (obj.type == isl_obj_pw_qpolynomial) { + obj.type = isl_obj_union_pw_qpolynomial; + obj.v = isl_union_pw_qpolynomial_from_pw_qpolynomial(obj.v); + } + if (obj.v) + isl_assert(s->ctx, obj.type == isl_obj_union_pw_qpolynomial, + goto error); + + return obj.v; +error: + obj.type->free(obj.v); + return NULL; +} + +__isl_give isl_union_pw_qpolynomial *isl_union_pw_qpolynomial_read_from_str( + isl_ctx *ctx, const char *str) +{ + isl_union_pw_qpolynomial *upwqp; + struct isl_stream *s = isl_stream_new_str(ctx, str); + if (!s) + return NULL; + upwqp = isl_stream_read_union_pw_qpolynomial(s); + isl_stream_free(s); + return upwqp; +}