isl_tab_pip.c: fix typos in comments
[platform/upstream/isl.git] / isl_input.c
index 24f64af..3fcf5e0 100644 (file)
 #include <stdio.h>
 #include <string.h>
 #include <strings.h>
-#include <isl_set.h>
-#include <isl_seq.h>
-#include <isl_div.h>
-#include "isl_stream.h"
+#include <isl/set.h>
+#include <isl/seq.h>
+#include <isl/div.h>
+#include <isl/stream.h>
 #include "isl_map_private.h"
-#include "isl_obj.h"
+#include <isl/obj.h>
 #include "isl_polynomial_private.h"
-#include <isl_union_map.h>
+#include <isl/union_map.h>
 #include <isl_mat_private.h>
 
 struct variable {
@@ -140,24 +140,24 @@ static int vars_add_anon(struct vars *v)
        return 0;
 }
 
-static __isl_give isl_dim *set_name(__isl_take isl_dim *dim,
+static __isl_give isl_basic_map *set_name(__isl_take isl_basic_map *bmap,
        enum isl_dim_type type, unsigned pos, char *name)
 {
        char *prime;
 
-       if (!dim)
+       if (!bmap)
                return NULL;
        if (!name)
-               return dim;
+               return bmap;
 
        prime = strchr(name, '\'');
        if (prime)
                *prime = '\0';
-       dim = isl_dim_set_name(dim, type, pos, name);
+       bmap = isl_basic_map_set_dim_name(bmap, type, pos, name);
        if (prime)
                *prime = '\'';
 
-       return dim;
+       return bmap;
 }
 
 static int accept_cst_factor(struct isl_stream *s, isl_int *f)
@@ -184,6 +184,7 @@ error:
 }
 
 static struct isl_vec *accept_affine(struct isl_stream *s, struct vars *v);
+static int read_div_definition(struct isl_stream *s, struct vars *v);
 
 static __isl_give isl_vec *accept_affine_factor(struct isl_stream *s,
        struct vars *v)
@@ -232,6 +233,18 @@ static __isl_give isl_vec *accept_affine_factor(struct isl_stream *s,
                        goto error;
                if (isl_stream_eat(s, ')'))
                        goto error;
+       } else if (tok->type == '[') {
+               if (vars_add_anon(v) < 0)
+                       goto error;
+               aff = isl_vec_alloc(v->ctx, 1 + v->n);
+               if (!aff)
+                       goto error;
+               isl_seq_clr(aff->el, aff->size);
+               isl_int_set_si(aff->el[1 + v->n - 1], 1);
+               isl_stream_push_token(s, tok);
+               tok = NULL;
+               if (read_div_definition(s, v) < 0)
+                       goto error;
        } else {
                isl_stream_error(s, tok, "expecting factor");
                goto error;
@@ -277,13 +290,15 @@ static struct isl_vec *accept_affine(struct isl_stream *s, struct vars *v)
                        isl_token_free(tok);
                        continue;
                }
-               if (tok->type == '(' || tok->type == ISL_TOKEN_IDENT) {
+               if (tok->type == '(' || tok->type == '[' ||
+                   tok->type == ISL_TOKEN_IDENT) {
                        isl_vec *aff2;
                        isl_stream_push_token(s, tok);
                        tok = NULL;
                        aff2 = accept_affine_factor(s, v);
                        if (sign < 0)
                                aff2 = isl_vec_scale(aff2, s->ctx->negone);
+                       aff = isl_vec_zero_extend(aff, 1 + v->n);
                        aff = isl_vec_add(aff, aff2);
                        if (!aff)
                                goto error;
@@ -296,6 +311,7 @@ static struct isl_vec *accept_affine(struct isl_stream *s, struct vars *v)
                                isl_vec *aff2;
                                aff2 = accept_affine_factor(s, v);
                                aff2 = isl_vec_scale(aff2, tok->u.v);
+                               aff = isl_vec_zero_extend(aff, 1 + v->n);
                                aff = isl_vec_add(aff, aff2);
                                if (!aff)
                                        goto error;
@@ -303,6 +319,11 @@ static struct isl_vec *accept_affine(struct isl_stream *s, struct vars *v)
                                isl_int_add(aff->el[0], aff->el[0], tok->u.v);
                        }
                        sign = 1;
+               } else {
+                       isl_stream_error(s, tok, "unexpected isl_token");
+                       isl_stream_push_token(s, tok);
+                       isl_vec_free(aff);
+                       return NULL;
                }
                isl_token_free(tok);
 
@@ -330,32 +351,89 @@ error:
        return NULL;
 }
 
-static __isl_give isl_mat *read_var_def(struct isl_stream *s,
-       __isl_take isl_mat *eq, enum isl_dim_type type, struct vars *v)
+/* Add any variables in the variable list "v" that are not already in "bmap"
+ * as existentially quantified variables in "bmap".
+ */
+static __isl_give isl_basic_map *add_divs(__isl_take isl_basic_map *bmap,
+       struct vars *v)
+{
+       int i;
+       int extra;
+       struct variable *var;
+
+       extra = v->n - isl_basic_map_total_dim(bmap);
+
+       if (extra == 0)
+               return bmap;
+
+       bmap = isl_basic_map_extend_dim(bmap, isl_basic_map_get_dim(bmap),
+                                       extra, 0, 2 * extra);
+
+       for (i = 0; i < extra; ++i)
+               if (isl_basic_map_alloc_div(bmap) < 0)
+                       goto error;
+
+       for (i = 0, var = v->v; i < extra; ++i, var = var->next) {
+               int k = bmap->n_div - 1 - i;
+
+               isl_seq_cpy(bmap->div[k], var->def->el, var->def->size);
+               isl_seq_clr(bmap->div[k] + var->def->size,
+                           2 + v->n - var->def->size);
+
+               if (isl_basic_map_add_div_constraints(bmap, k) < 0)
+                       goto error;
+       }
+
+       return bmap;
+error:
+       isl_basic_map_free(bmap);
+       return NULL;
+}
+
+static __isl_give isl_basic_map *read_var_def(struct isl_stream *s,
+       __isl_take isl_basic_map *bmap, enum isl_dim_type type, struct vars *v)
 {
+       isl_dim *dim;
+       isl_basic_map *def = NULL;
        struct isl_vec *vec;
+       int k;
+       int n;
+
+       if (vars_add_anon(v) < 0)
+               goto error;
+       n = v->n;
 
        vec = accept_affine(s, v);
        if (!vec)
                goto error;
-       if (vars_add_anon(v) < 0)
-               goto error;
-       eq = isl_mat_add_rows(eq, 1);
-       if (eq) {
-               isl_seq_cpy(eq->row[eq->n_row - 1], vec->el, vec->size);
-               isl_int_set_si(eq->row[eq->n_row - 1][vec->size], -1);
+
+       dim = isl_basic_map_get_dim(bmap);
+       def = isl_basic_map_universe(dim);
+       def = add_divs(def, v);
+       def = isl_basic_map_extend_constraints(def, 1, 0);
+       k = isl_basic_map_alloc_equality(def);
+       if (k >= 0) {
+               isl_seq_cpy(def->eq[k], vec->el, vec->size);
+               isl_int_set_si(def->eq[k][1 + n - 1], -1);
        }
        isl_vec_free(vec);
+       if (k < 0)
+               goto error;
+
+       vars_drop(v, v->n - n);
 
-       return eq;
+       def = isl_basic_map_simplify(def);
+       def = isl_basic_map_finalize(def);
+       bmap = isl_basic_map_intersect(bmap, def);
+       return bmap;
 error:
-       isl_mat_free(eq);
+       isl_basic_map_free(bmap);
+       isl_basic_map_free(def);
        return NULL;
 }
 
-static __isl_give isl_dim *read_var_list(struct isl_stream *s,
-       __isl_take isl_dim *dim, enum isl_dim_type type, struct vars *v,
-       __isl_keep isl_mat **eq)
+static __isl_give isl_basic_map *read_var_list(struct isl_stream *s,
+       __isl_take isl_basic_map *bmap, enum isl_dim_type type, struct vars *v)
 {
        int i = 0;
        struct isl_token *tok;
@@ -372,14 +450,13 @@ static __isl_give isl_dim *read_var_list(struct isl_stream *s,
                }
 
                if (new_name) {
-                       dim = isl_dim_add(dim, type, 1);
-                       if (eq)
-                               *eq = isl_mat_add_zero_cols(*eq, 1);
-                       dim = set_name(dim, type, i, v->v->name);
+                       bmap = isl_basic_map_add(bmap, type, 1);
+                       bmap = set_name(bmap, type, i, v->v->name);
                        isl_token_free(tok);
                } else if (tok->type == ISL_TOKEN_IDENT ||
                           tok->type == ISL_TOKEN_VALUE ||
-                          tok->type == '-') {
+                          tok->type == '-' ||
+                          tok->type == '(') {
                        if (type == isl_dim_param) {
                                isl_stream_error(s, tok,
                                                "expecting unique identifier");
@@ -387,9 +464,8 @@ static __isl_give isl_dim *read_var_list(struct isl_stream *s,
                        }
                        isl_stream_push_token(s, tok);
                        tok = NULL;
-                       dim = isl_dim_add(dim, type, 1);
-                       *eq = isl_mat_add_zero_cols(*eq, 1);
-                       *eq = read_var_def(s, *eq, type, v);
+                       bmap = isl_basic_map_add(bmap, type, 1);
+                       bmap = read_var_def(s, bmap, type, v);
                } else
                        break;
 
@@ -403,10 +479,10 @@ static __isl_give isl_dim *read_var_list(struct isl_stream *s,
        if (tok)
                isl_stream_push_token(s, tok);
 
-       return dim;
+       return bmap;
 error:
        isl_token_free(tok);
-       isl_dim_free(dim);
+       isl_basic_map_free(bmap);
        return NULL;
 }
 
@@ -435,6 +511,7 @@ static __isl_give isl_mat *accept_affine_list(struct isl_stream *s,
                isl_token_free(tok);
 
                vec = accept_affine(s, v);
+               mat = isl_mat_add_zero_cols(mat, 1 + v->n - isl_mat_cols(mat));
                mat = isl_mat_vec_concat(mat, vec);
                if (!mat)
                        return NULL;
@@ -451,10 +528,7 @@ static int read_div_definition(struct isl_stream *s, struct vars *v)
        struct isl_token *tok;
        int seen_paren = 0;
        struct isl_vec *aff;
-
-       v->v->def = isl_vec_alloc(s->ctx, 2 + v->n);
-       if (!v->v->def)
-               return -1;
+       struct variable *var;
 
        if (isl_stream_eat(s, '['))
                return -1;
@@ -468,11 +542,19 @@ static int read_div_definition(struct isl_stream *s, struct vars *v)
        } else
                isl_stream_push_token(s, tok);
 
+       var = v->v;
+
        aff = accept_affine(s, v);
        if (!aff)
                return -1;
 
-       isl_seq_cpy(v->v->def->el + 1, aff->el, aff->size);
+       var->def = isl_vec_alloc(s->ctx, 2 + v->n);
+       if (!var->def) {
+               isl_vec_free(aff);
+               return -1;
+       }
+
+       isl_seq_cpy(var->def->el + 1, aff->el, aff->size);
 
        isl_vec_free(aff);
 
@@ -489,7 +571,7 @@ static int read_div_definition(struct isl_stream *s, struct vars *v)
                isl_stream_push_token(s, tok);
                return -1;
        }
-       isl_int_set(v->v->def->el[0], tok->u.v);
+       isl_int_set(var->def->el[0], tok->u.v);
        isl_token_free(tok);
 
        if (isl_stream_eat(s, ']'))
@@ -501,10 +583,12 @@ static int read_div_definition(struct isl_stream *s, struct vars *v)
 static struct isl_basic_map *add_div_definition(struct isl_stream *s,
        struct vars *v, struct isl_basic_map *bmap, int k)
 {
+       struct variable *var = v->v;
+
        if (read_div_definition(s, v) < 0)
                goto error;
 
-       isl_seq_cpy(bmap->div[k], v->v->def->el, 2 + v->n);
+       isl_seq_cpy(bmap->div[k], var->def->el, 2 + v->n);
 
        if (isl_basic_map_add_div_constraints(bmap, k) < 0)
                goto error;
@@ -592,27 +676,25 @@ static int next_is_tuple(struct isl_stream *s)
        return is_tuple;
 }
 
-static __isl_give isl_dim *read_tuple(struct isl_stream *s,
-       __isl_take isl_dim *dim, enum isl_dim_type type, struct vars *v,
-       __isl_keep isl_mat **eq);
+static __isl_give isl_basic_map *read_tuple(struct isl_stream *s,
+       __isl_take isl_basic_map *bmap, enum isl_dim_type type, struct vars *v);
 
-static __isl_give isl_dim *read_nested_tuple(struct isl_stream *s,
-       __isl_take isl_dim *dim, struct vars *v, __isl_keep isl_mat **eq)
+static __isl_give isl_basic_map *read_nested_tuple(struct isl_stream *s,
+       __isl_take isl_basic_map *bmap, struct vars *v)
 {
-       dim = read_tuple(s, dim, isl_dim_in, v, eq);
+       bmap = read_tuple(s, bmap, isl_dim_in, v);
        if (isl_stream_eat(s, ISL_TOKEN_TO))
                goto error;
-       dim = read_tuple(s, dim, isl_dim_out, v, eq);
-       dim = isl_dim_wrap(dim);
-       return dim;
+       bmap = read_tuple(s, bmap, isl_dim_out, v);
+       bmap = isl_basic_map_from_range(isl_basic_map_wrap(bmap));
+       return bmap;
 error:
-       isl_dim_free(dim);
+       isl_basic_map_free(bmap);
        return NULL;
 }
 
-static __isl_give isl_dim *read_tuple(struct isl_stream *s,
-       __isl_take isl_dim *dim, enum isl_dim_type type, struct vars *v,
-       __isl_keep isl_mat **eq)
+static __isl_give isl_basic_map *read_tuple(struct isl_stream *s,
+       __isl_take isl_basic_map *bmap, enum isl_dim_type type, struct vars *v)
 {
        struct isl_token *tok;
        char *name = NULL;
@@ -631,20 +713,29 @@ static __isl_give isl_dim *read_tuple(struct isl_stream *s,
        }
        isl_token_free(tok);
        if (type != isl_dim_param && next_is_tuple(s)) {
-               isl_dim *nested = isl_dim_copy(dim);
-               nested = isl_dim_drop(nested, isl_dim_in, 0,
-                                       isl_dim_size(nested, isl_dim_in));
-               nested = isl_dim_drop(nested, isl_dim_out, 0,
-                                       isl_dim_size(nested, isl_dim_out));
-               nested = read_nested_tuple(s, nested, v, eq);
+               isl_dim *dim = isl_basic_map_get_dim(bmap);
+               int nparam = isl_dim_size(dim, isl_dim_param);
+               int n_in = isl_dim_size(dim, isl_dim_in);
+               isl_basic_map *nested;
+               if (type == isl_dim_out)
+                       dim = isl_dim_move(dim, isl_dim_param, nparam,
+                                               isl_dim_in, 0, n_in);
+               nested = isl_basic_map_alloc_dim(dim, 0, 0, 0);
+               nested = read_nested_tuple(s, nested, v);
                if (type == isl_dim_in) {
-                       isl_dim_free(dim);
-                       dim = isl_dim_reverse(nested);
+                       nested = isl_basic_map_reverse(nested);
+                       bmap = isl_basic_map_intersect(nested, bmap);
                } else {
-                       dim = isl_dim_join(dim, nested);
+                       isl_basic_set *bset;
+                       dim = isl_dim_range(isl_basic_map_get_dim(nested));
+                       dim = isl_dim_drop(dim, isl_dim_param, nparam, n_in);
+                       dim = isl_dim_join(isl_basic_map_get_dim(bmap), dim);
+                       bset = isl_basic_map_domain(bmap);
+                       nested = isl_basic_map_reset_dim(nested, dim);
+                       bmap = isl_basic_map_intersect_domain(nested, bset);
                }
        } else
-               dim = read_var_list(s, dim, type, v, eq);
+               bmap = read_var_list(s, bmap, type, v);
        tok = isl_stream_next_token(s);
        if (!tok || tok->type != ']') {
                isl_stream_error(s, tok, "expecting ']'");
@@ -653,15 +744,15 @@ static __isl_give isl_dim *read_tuple(struct isl_stream *s,
        isl_token_free(tok);
 
        if (name) {
-               dim = isl_dim_set_tuple_name(dim, type, name);
+               bmap = isl_basic_map_set_tuple_name(bmap, type, name);
                free(name);
        }
 
-       return dim;
+       return bmap;
 error:
        if (tok)
                isl_token_free(tok);
-       isl_dim_free(dim);
+       isl_basic_map_free(bmap);
        return NULL;
 }
 
@@ -770,7 +861,6 @@ static struct isl_basic_map *add_constraint(struct isl_stream *s,
        struct vars *v, struct isl_basic_map *bmap)
 {
        int i, j;
-       unsigned total = isl_basic_map_total_dim(bmap);
        struct isl_token *tok = NULL;
        struct isl_mat *aff1 = NULL, *aff2 = NULL;
 
@@ -797,13 +887,15 @@ static struct isl_basic_map *add_constraint(struct isl_stream *s,
                tok = NULL;
                goto error;
        }
-       isl_assert(aff1->ctx, aff1->n_col == 1 + total, goto error);
        for (;;) {
                aff2 = accept_affine_list(s, v);
                if (!aff2)
                        goto error;
-               isl_assert(aff2->ctx, aff2->n_col == 1 + total, goto error);
 
+               aff1 = isl_mat_add_zero_cols(aff1, aff2->n_col - aff1->n_col);
+               if (!aff1)
+                       goto error;
+               bmap = add_divs(bmap, v);
                bmap = isl_basic_map_extend_constraints(bmap, 0,
                                                aff1->n_row * aff2->n_row);
                for (i = 0; i < aff1->n_row; ++i)
@@ -1054,6 +1146,13 @@ static __isl_give isl_basic_map *basic_map_read_polylib(struct isl_stream *s,
                isl_stream_error(s, NULL, "unexpected EOF");
                return NULL;
        }
+       if (tok->type != ISL_TOKEN_VALUE || tok2->type != ISL_TOKEN_VALUE) {
+               isl_stream_push_token(s, tok2);
+               isl_stream_push_token(s, tok);
+               isl_stream_error(s, NULL,
+                                "expecting constraint matrix dimensions");
+               return NULL;
+       }
        n_row = isl_int_get_si(tok->u.v);
        n_col = isl_int_get_si(tok2->u.v);
        on_new_line = tok2->on_new_line;
@@ -1120,6 +1219,7 @@ static __isl_give isl_basic_map *basic_map_read_polylib(struct isl_stream *s,
                int k = isl_basic_map_alloc_div(bmap);
                if (k < 0)
                        goto error;
+               isl_seq_clr(bmap->div[k], 1 + 1 + nparam + in + out + local);
        }
 
        for (i = 0; i < n_row; ++i)
@@ -1153,11 +1253,17 @@ static struct isl_map *map_read_polylib(struct isl_stream *s, int nparam)
                return NULL;
        }
        tok2 = isl_stream_next_token_on_same_line(s);
-       if (tok2) {
+       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));
        }
+       if (tok2) {
+               isl_stream_error(s, tok2, "unexpected token");
+               isl_stream_push_token(s, tok2);
+               isl_stream_push_token(s, tok);
+               return NULL;
+       }
        n = isl_int_get_si(tok->u.v);
        isl_token_free(tok);
 
@@ -1165,7 +1271,7 @@ static struct isl_map *map_read_polylib(struct isl_stream *s, int nparam)
 
        map = isl_map_from_basic_map(basic_map_read_polylib(s, nparam));
 
-       for (i = 1; i < n; ++i)
+       for (i = 1; map && i < n; ++i)
                map = isl_map_union(map,
                        isl_map_from_basic_map(basic_map_read_polylib(s, nparam)));
 
@@ -1198,26 +1304,25 @@ static int optional_power(struct isl_stream *s)
 }
 
 static __isl_give isl_div *read_div(struct isl_stream *s,
-       __isl_keep isl_basic_map *bmap, struct vars *v)
+       __isl_take isl_dim *dim, struct vars *v)
 {
-       unsigned total = isl_basic_map_total_dim(bmap);
-       int k;
-
-       bmap = isl_basic_map_copy(bmap);
-       bmap = isl_basic_map_cow(bmap);
-       bmap = isl_basic_map_extend_dim(bmap, isl_dim_copy(bmap->dim),
-                                       1, 0, 2);
+       int n;
+       isl_basic_map *bmap;
 
-       if ((k = isl_basic_map_alloc_div(bmap)) < 0)
-               goto error;
-       isl_seq_clr(bmap->div[k], 1 + 1 + total);
+       n = v->n;
+       bmap = isl_basic_map_universe(dim);
 
        if (vars_add_anon(v) < 0)
                goto error;
-       bmap = add_div_definition(s, v, bmap, k);
-       vars_drop(v, 1);
+       if (read_div_definition(s, v) < 0)
+               goto error;
+       bmap = add_divs(bmap, v);
+       bmap = isl_basic_map_order_divs(bmap);
+       if (!bmap)
+               goto error;
+       vars_drop(v, v->n - n);
 
-       return isl_basic_map_div(bmap, k);
+       return isl_basic_map_div(bmap, bmap->n_div - 1);
 error:
        isl_basic_map_free(bmap);
        return NULL;
@@ -1238,12 +1343,16 @@ static __isl_give isl_qpolynomial *read_factor(struct isl_stream *s,
                return NULL;
        }
        if (tok->type == '(') {
+               int pow;
+
                isl_token_free(tok);
                qp = read_term(s, bmap, v);
                if (!qp)
                        return NULL;
                if (isl_stream_eat(s, ')'))
                        goto error;
+               pow = optional_power(s);
+               qp = isl_qpolynomial_pow(qp, pow);
        } else if (tok->type == ISL_TOKEN_VALUE) {
                struct isl_token *tok2;
                tok2 = isl_stream_next_token(s);
@@ -1286,13 +1395,13 @@ static __isl_give isl_qpolynomial *read_factor(struct isl_stream *s,
                }
                isl_token_free(tok);
                pow = optional_power(s);
-               qp = isl_qpolynomial_pow(isl_basic_map_get_dim(bmap), pos, pow);
+               qp = isl_qpolynomial_var_pow(isl_basic_map_get_dim(bmap), pos, pow);
        } else if (tok->type == '[') {
                isl_div *div;
                int pow;
 
                isl_stream_push_token(s, tok);
-               div = read_div(s, bmap, v);
+               div = read_div(s, isl_basic_map_get_dim(bmap), v);
                pow = optional_power(s);
                qp = isl_qpolynomial_div_pow(div, pow);
        } else if (tok->type == '-') {
@@ -1457,77 +1566,37 @@ error:
        return obj;
 }
 
-static __isl_give isl_basic_map *add_equalities(__isl_take isl_basic_map *bmap,
-       isl_mat *eq)
-{
-       int i, k;
-       unsigned total = 1 + isl_basic_map_total_dim(bmap);
-
-       if (!bmap || !eq)
-               goto error;
-
-       bmap = isl_basic_map_extend_constraints(bmap, eq->n_row, 0);
-
-       for (i = 0; i < eq->n_row; ++i) {
-               k = isl_basic_map_alloc_equality(bmap);
-               if (k < 0)
-                       goto error;
-               isl_seq_cpy(bmap->eq[k], eq->row[i], eq->n_col);
-               isl_seq_clr(bmap->eq[k] + eq->n_col, total - eq->n_col);
-       }
-
-       isl_mat_free(eq);
-       bmap = isl_basic_map_gauss(bmap, NULL);
-       bmap = isl_basic_map_finalize(bmap);
-       return bmap;
-error:
-       isl_mat_free(eq);
-       isl_basic_map_free(bmap);
-       return NULL;
-}
-
 static struct isl_obj obj_read_body(struct isl_stream *s,
-       __isl_take isl_dim *dim, struct vars *v)
+       __isl_take isl_basic_map *bmap, struct vars *v)
 {
        struct isl_map *map = NULL;
        struct isl_token *tok;
        struct isl_obj obj = { isl_obj_set, NULL };
        int n = v->n;
-       isl_mat *eq = NULL;
-       isl_basic_map *bmap;
 
-       if (!next_is_tuple(s)) {
-               bmap = isl_basic_map_alloc_dim(dim, 0, 0, 0);
+       if (!next_is_tuple(s))
                return obj_read_poly_or_fold(s, bmap, v, n);
-       }
 
-       eq = isl_mat_alloc(s->ctx, 0, 1 + n);
-
-       dim = read_tuple(s, dim, isl_dim_in, v, &eq);
-       if (!dim)
+       bmap = read_tuple(s, bmap, isl_dim_in, v);
+       if (!bmap)
                goto error;
        tok = isl_stream_next_token(s);
        if (tok && tok->type == ISL_TOKEN_TO) {
                obj.type = isl_obj_map;
                isl_token_free(tok);
                if (!next_is_tuple(s)) {
-                       bmap = isl_basic_map_alloc_dim(dim, 0, 0, 0);
-                       bmap = add_equalities(bmap, eq);
                        bmap = isl_basic_map_reverse(bmap);
                        return obj_read_poly_or_fold(s, bmap, v, n);
                }
-               dim = read_tuple(s, dim, isl_dim_out, v, &eq);
-               if (!dim)
+               bmap = read_tuple(s, bmap, isl_dim_out, v);
+               if (!bmap)
                        goto error;
        } else {
-               dim = isl_dim_reverse(dim);
+               bmap = isl_basic_map_reverse(bmap);
                if (tok)
                        isl_stream_push_token(s, tok);
        }
 
-       bmap = isl_basic_map_alloc_dim(dim, 0, 0, 0);
-       bmap = add_equalities(bmap, eq);
-
        map = read_optional_disjuncts(s, bmap, v);
 
        vars_drop(v, v->n - n);
@@ -1535,8 +1604,7 @@ static struct isl_obj obj_read_body(struct isl_stream *s,
        obj.v = map;
        return obj;
 error:
-       isl_mat_free(eq);
-       isl_dim_free(dim);
+       isl_basic_map_free(bmap);
        obj.type = isl_obj_none;
        return obj;
 }
@@ -1618,7 +1686,7 @@ error:
 
 static struct isl_obj obj_read(struct isl_stream *s, int nparam)
 {
-       isl_dim *dim = NULL;
+       isl_basic_map *bmap = NULL;
        struct isl_token *tok;
        struct vars *v = NULL;
        struct isl_obj obj = { isl_obj_set, NULL };
@@ -1644,11 +1712,11 @@ static struct isl_obj obj_read(struct isl_stream *s, int nparam)
                isl_stream_push_token(s, tok);
                goto error;
        }
-       dim = isl_dim_alloc(s->ctx, 0, 0, 0);
+       bmap = isl_basic_map_alloc(s->ctx, 0, 0, 0, 0, 0, 0);
        if (tok->type == '[') {
                isl_stream_push_token(s, tok);
-               dim = read_tuple(s, dim, isl_dim_param, v, NULL);
-               if (!dim)
+               bmap = read_tuple(s, bmap, isl_dim_param, v);
+               if (!bmap)
                        goto error;
                if (nparam >= 0)
                        isl_assert(s->ctx, nparam == v->n, goto error);
@@ -1662,7 +1730,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)
-               dim = isl_dim_add(dim, isl_dim_param, nparam);
+               bmap = isl_basic_map_add(bmap, isl_dim_param, nparam);
        if (!tok || tok->type != '{') {
                isl_stream_error(s, tok, "expecting '{'");
                if (tok)
@@ -1678,14 +1746,14 @@ static struct isl_obj obj_read(struct isl_stream *s, int nparam)
                isl_token_free(tok);
                if (isl_stream_eat(s, '='))
                        goto error;
-               dim = read_tuple(s, dim, isl_dim_param, v, NULL);
-               if (!dim)
+               bmap = read_tuple(s, bmap, isl_dim_param, v);
+               if (!bmap)
                        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_dim_copy(dim));
+               obj.v = isl_union_set_empty(isl_basic_map_get_dim(bmap));
                isl_token_free(tok);
                goto done;
        } else
@@ -1694,15 +1762,15 @@ static struct isl_obj obj_read(struct isl_stream *s, int nparam)
        for (;;) {
                struct isl_obj o;
                tok = NULL;
-               o = obj_read_body(s, isl_dim_copy(dim), v);
-               if (o.type == isl_obj_none)
-                       break;
+               o = obj_read_body(s, isl_basic_map_copy(bmap), v);
+               if (o.type == isl_obj_none || !o.v)
+                       goto error;
                if (!obj.v)
                        obj = o;
                else {
                        obj = obj_add(s->ctx, obj, o);
-                       if (obj.type == isl_obj_none)
-                               break;
+                       if (obj.type == isl_obj_none || !obj.v)
+                               goto error;
                }
                tok = isl_stream_next_token(s);
                if (!tok || tok->type != ';')
@@ -1720,11 +1788,11 @@ static struct isl_obj obj_read(struct isl_stream *s, int nparam)
        }
 done:
        vars_free(v);
-       isl_dim_free(dim);
+       isl_basic_map_free(bmap);
 
        return obj;
 error:
-       isl_dim_free(dim);
+       isl_basic_map_free(bmap);
        obj.type->free(obj.v);
        if (v)
                vars_free(v);
@@ -1768,6 +1836,49 @@ error:
        return NULL;
 }
 
+__isl_give isl_union_map *isl_stream_read_union_map(struct isl_stream *s)
+{
+       struct isl_obj obj;
+       isl_union_map *umap;
+
+       obj = obj_read(s, -1);
+       if (obj.type == isl_obj_map) {
+               obj.type = isl_obj_union_map;
+               obj.v = isl_union_map_from_map(obj.v);
+       }
+       if (obj.type == isl_obj_set) {
+               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);
+
+       return obj.v;
+error:
+       obj.type->free(obj.v);
+       return NULL;
+}
+
+__isl_give isl_union_set *isl_stream_read_union_set(struct isl_stream *s)
+{
+       struct isl_obj obj;
+       isl_union_set *uset;
+
+       obj = obj_read(s, -1);
+       if (obj.type == isl_obj_set) {
+               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_set, goto error);
+
+       return obj.v;
+error:
+       obj.type->free(obj.v);
+       return NULL;
+}
+
 static struct isl_basic_map *basic_map_read(struct isl_stream *s, int nparam)
 {
        struct isl_obj obj;
@@ -1898,60 +2009,83 @@ error:
        return NULL;
 }
 
-static char *next_line(FILE *input, char *line, unsigned len)
+__isl_give isl_union_map *isl_union_map_read_from_str(struct isl_ctx *ctx,
+               const char *str)
 {
-       char *p;
-
-       do {
-               if (!(p = fgets(line, len, input)))
-                       return NULL;
-               while (isspace(*p) && *p != '\n')
-                       ++p;
-       } while (*p == '#' || *p == '\n');
+       isl_union_map *umap;
+       struct isl_stream *s = isl_stream_new_str(ctx, str);
+       if (!s)
+               return NULL;
+       umap = isl_stream_read_union_map(s);
+       isl_stream_free(s);
+       return umap;
+}
 
-       return p;
+__isl_give isl_union_set *isl_union_set_read_from_str(struct isl_ctx *ctx,
+               const char *str)
+{
+       isl_union_set *uset;
+       struct isl_stream *s = isl_stream_new_str(ctx, str);
+       if (!s)
+               return NULL;
+       uset = isl_stream_read_union_set(s);
+       isl_stream_free(s);
+       return uset;
 }
 
-static struct isl_vec *isl_vec_read_from_file_polylib(struct isl_ctx *ctx,
-               FILE *input)
+static __isl_give isl_vec *isl_vec_read_polylib(struct isl_stream *s)
 {
        struct isl_vec *vec = NULL;
-       char line[1024];
-       char val[1024];
-       char *p;
+       struct isl_token *tok;
        unsigned size;
        int j;
-       int n;
-       int offset;
 
-       isl_assert(ctx, next_line(input, line, sizeof(line)), return NULL);
-       isl_assert(ctx, sscanf(line, "%u", &size) == 1, return NULL);
+       tok = isl_stream_next_token(s);
+       if (!tok || tok->type != ISL_TOKEN_VALUE) {
+               isl_stream_error(s, tok, "expecting vector length");
+               goto error;
+       }
 
-       vec = isl_vec_alloc(ctx, size);
+       size = isl_int_get_si(tok->u.v);
+       isl_token_free(tok);
 
-       p = next_line(input, line, sizeof(line));
-       isl_assert(ctx, p, goto error);
+       vec = isl_vec_alloc(s->ctx, size);
 
        for (j = 0; j < size; ++j) {
-               n = sscanf(p, "%s%n", val, &offset);
-               isl_assert(ctx, n != 0, goto error);
-               isl_int_read(vec->el[j], val);
-               p += offset;
+               tok = isl_stream_next_token(s);
+               if (!tok || tok->type != ISL_TOKEN_VALUE) {
+                       isl_stream_error(s, tok, "expecting constant value");
+                       goto error;
+               }
+               isl_int_set(vec->el[j], tok->u.v);
+               isl_token_free(tok);
        }
 
        return vec;
 error:
+       isl_token_free(tok);
        isl_vec_free(vec);
        return NULL;
 }
 
+static __isl_give isl_vec *vec_read(struct isl_stream *s,
+       unsigned input_format)
+{
+       if (input_format == ISL_FORMAT_POLYLIB)
+               return isl_vec_read_polylib(s);
+       isl_assert(s->ctx, 0, return NULL);
+}
+
 struct isl_vec *isl_vec_read_from_file(struct isl_ctx *ctx,
                FILE *input, unsigned input_format)
 {
-       if (input_format == ISL_FORMAT_POLYLIB)
-               return isl_vec_read_from_file_polylib(ctx, input);
-       else
-               isl_assert(ctx, 0, return NULL);
+       isl_vec *v;
+       struct isl_stream *s = isl_stream_new_file(ctx, input);
+       if (!s)
+               return NULL;
+       v = vec_read(s, input_format);
+       isl_stream_free(s);
+       return v;
 }
 
 __isl_give isl_pw_qpolynomial *isl_stream_read_pw_qpolynomial(