/*

 * Copyright 2008-2009 Katholieke Universiteit Leuven

 * Copyright 2010      INRIA Saclay

 * Copyright 2012-2013 Ecole Normale Superieure

 *

 * 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 <stdlib.h>

#include <string.h>

#include <isl_ctx_private.h>

#include <isl_map_private.h>

#include <isl/set.h>

#include <isl_seq.h>

#include <isl_polynomial_private.h>

#include <isl_printer_private.h>

#include <isl_space_private.h>

#include <isl_mat_private.h>

#include <isl_vec_private.h>

#include <isl/union_map.h>

#include <isl/constraint.h>

#include <isl_local_space_private.h>

#include <isl_aff_private.h>

#include <isl_val_private.h>

#include <isl/ast_build.h>

#include <isl_sort.h>

static const char *s_to[2] = { " -> ", " \\to " };

static const char *s_and[2] = { " and ", " \\wedge " };

static const char *s_or[2] = { " or ", " \\vee " };

static const char *s_le[2] = { "<=", "\\le" };

static const char *s_ge[2] = { ">=", "\\ge" };

static const char *s_open_set[2] = { "{ ", "\\{\\, " };

static const char *s_close_set[2] = { " }", " \\,\\}" };

static const char *s_open_list[2] = { "[", "(" };

static const char *s_close_list[2] = { "]", ")" };

static const char *s_such_that[2] = { " : ", " \\mid " };

static const char *s_open_exists[2] = { "exists (", "\\exists \\, " };

static const char *s_close_exists[2] = { ")", "" };

static const char *s_div_prefix[2] = { "e", "\\alpha_" };

static const char *s_param_prefix[2] = { "p", "p_" };

static const char *s_input_prefix[2] = { "i", "i_" };

static const char *s_output_prefix[2] = { "o", "o_" };

static __isl_give isl_printer *print_constraint_polylib(

	struct isl_basic_map *bmap, int ineq, int n, __isl_take isl_printer *p)

{

	int i;

	unsigned n_in = isl_basic_map_dim(bmap, isl_dim_in);

	unsigned n_out = isl_basic_map_dim(bmap, isl_dim_out);

	unsigned nparam = isl_basic_map_dim(bmap, isl_dim_param);

	isl_int *c = ineq ? bmap->ineq[n] : bmap->eq[n];

	p = isl_printer_start_line(p);

	p = isl_printer_print_int(p, ineq);

	for (i = 0; i < n_out; ++i) {

		p = isl_printer_print_str(p, " ");

		p = isl_printer_print_isl_int(p, c[1+nparam+n_in+i]);

	}

	for (i = 0; i < n_in; ++i) {

		p = isl_printer_print_str(p, " ");

		p = isl_printer_print_isl_int(p, c[1+nparam+i]);

	}

	for (i = 0; i < bmap->n_div; ++i) {

		p = isl_printer_print_str(p, " ");

		p = isl_printer_print_isl_int(p, c[1+nparam+n_in+n_out+i]);

	}

	for (i = 0; i < nparam; ++i) {

		p = isl_printer_print_str(p, " ");

		p = isl_printer_print_isl_int(p, c[1+i]);

	}

	p = isl_printer_print_str(p, " ");

	p = isl_printer_print_isl_int(p, c[0]);

	p = isl_printer_end_line(p);

	return p;

}

static __isl_give isl_printer *print_constraints_polylib(

	struct isl_basic_map *bmap, __isl_take isl_printer *p)

{

	int i;

	p = isl_printer_set_isl_int_width(p, 5);

	for (i = 0; i < bmap->n_eq; ++i)

		p = print_constraint_polylib(bmap, 0, i, p);

	for (i = 0; i < bmap->n_ineq; ++i)

		p = print_constraint_polylib(bmap, 1, i, p);

	return p;

}

static __isl_give isl_printer *bset_print_constraints_polylib(

	struct isl_basic_set *bset, __isl_take isl_printer *p)

{

	return print_constraints_polylib((struct isl_basic_map *)bset, p);

}

static __isl_give isl_printer *isl_basic_map_print_polylib(

	__isl_keep isl_basic_map *bmap, __isl_take isl_printer *p, int ext)

{

	unsigned total = isl_basic_map_total_dim(bmap);

	p = isl_printer_start_line(p);

	p = isl_printer_print_int(p, bmap->n_eq + bmap->n_ineq);

	p = isl_printer_print_str(p, " ");

	p = isl_printer_print_int(p, 1 + total + 1);

	if (ext) {

		p = isl_printer_print_str(p, " ");

		p = isl_printer_print_int(p,

				    isl_basic_map_dim(bmap, isl_dim_out));

		p = isl_printer_print_str(p, " ");

		p = isl_printer_print_int(p,

				    isl_basic_map_dim(bmap, isl_dim_in));

		p = isl_printer_print_str(p, " ");

		p = isl_printer_print_int(p,

				    isl_basic_map_dim(bmap, isl_dim_div));

		p = isl_printer_print_str(p, " ");

		p = isl_printer_print_int(p,

				    isl_basic_map_dim(bmap, isl_dim_param));

	}

	p = isl_printer_end_line(p);

	return print_constraints_polylib(bmap, p);

}

static __isl_give isl_printer *isl_basic_set_print_polylib(

	__isl_keep isl_basic_set *bset, __isl_take isl_printer *p, int ext)

{

	return isl_basic_map_print_polylib((struct isl_basic_map *)bset, p, ext);

}

static __isl_give isl_printer *isl_map_print_polylib(__isl_keep isl_map *map,

	__isl_take isl_printer *p, int ext)

{

	int i;

	p = isl_printer_start_line(p);

	p = isl_printer_print_int(p, map->n);

	p = isl_printer_end_line(p);

	for (i = 0; i < map->n; ++i) {

		p = isl_printer_start_line(p);

		p = isl_printer_end_line(p);

		p = isl_basic_map_print_polylib(map->p[i], p, ext);

	}

	return p;

}

static __isl_give isl_printer *isl_set_print_polylib(__isl_keep isl_set *set,

	__isl_take isl_printer *p, int ext)

{

	return isl_map_print_polylib((struct isl_map *)set, p, ext);

}

static int count_same_name(__isl_keep isl_space *dim,

	enum isl_dim_type type, unsigned pos, const char *name)

{

	enum isl_dim_type t;

	unsigned p, s;

	int count = 0;

	for (t = isl_dim_param; t <= type && t <= isl_dim_out; ++t) {

		s = t == type ? pos : isl_space_dim(dim, t);

		for (p = 0; p < s; ++p) {

			const char *n = isl_space_get_dim_name(dim, t, p);

			if (n && !strcmp(n, name))

				count++;

		}

	}

	return count;

}

static __isl_give isl_printer *print_name(__isl_keep isl_space *dim,

	__isl_take isl_printer *p, enum isl_dim_type type, unsigned pos,

	int latex)

{

	const char *name;

	char buffer[20];

	int primes;

	name = type == isl_dim_div ? NULL : isl_space_get_dim_name(dim, type, pos);

	if (!name) {

		const char *prefix;

		if (type == isl_dim_param)

			prefix = s_param_prefix[latex];

		else if (type == isl_dim_div)

			prefix = s_div_prefix[latex];

		else if (isl_space_is_set(dim) || type == isl_dim_in)

			prefix = s_input_prefix[latex];

		else

			prefix = s_output_prefix[latex];

		snprintf(buffer, sizeof(buffer), "%s%d", prefix, pos);

		name = buffer;

	}

	primes = count_same_name(dim, name == buffer ? isl_dim_div : type,

				 pos, name);

	p = isl_printer_print_str(p, name);

	while (primes-- > 0)

		p = isl_printer_print_str(p, "'");

	return p;

}

static enum isl_dim_type pos2type(__isl_keep isl_space *dim, unsigned *pos)

{

	enum isl_dim_type type;

	unsigned n_in = isl_space_dim(dim, isl_dim_in);

	unsigned n_out = isl_space_dim(dim, isl_dim_out);

	unsigned nparam = isl_space_dim(dim, isl_dim_param);

	if (*pos < 1 + nparam) {

		type = isl_dim_param;

		*pos -= 1;

	} else if (*pos < 1 + nparam + n_in) {

		type = isl_dim_in;

		*pos -= 1 + nparam;

	} else if (*pos < 1 + nparam + n_in + n_out) {

		type = isl_dim_out;

		*pos -= 1 + nparam + n_in;

	} else {

		type = isl_dim_div;

		*pos -= 1 + nparam + n_in + n_out;

	}

	return type;

}

static __isl_give isl_printer *print_div(__isl_keep isl_space *dim,

	__isl_keep isl_mat *div, int pos, __isl_take isl_printer *p);

static __isl_give isl_printer *print_term(__isl_keep isl_space *dim,

	__isl_keep isl_mat *div,

	isl_int c, unsigned pos, __isl_take isl_printer *p, int latex)

{

	enum isl_dim_type type;

	int print_div_def;

	if (pos == 0)

		return isl_printer_print_isl_int(p, c);

	type = pos2type(dim, &pos;;

	print_div_def = type == isl_dim_div && div &&

			!isl_int_is_zero(div->row[pos][0]);

	if (isl_int_is_one(c))

		;

	else if (isl_int_is_negone(c))

		p = isl_printer_print_str(p, "-");

	else {

		p = isl_printer_print_isl_int(p, c);

		if (p->output_format == ISL_FORMAT_C || print_div_def)

			p = isl_printer_print_str(p, "*");

	}

	if (print_div_def)

		p = print_div(dim, div, pos, p);

	else

		p = print_name(dim, p, type, pos, latex);

	return p;

}

static __isl_give isl_printer *print_affine_of_len(__isl_keep isl_space *dim,

	__isl_keep isl_mat *div,

	__isl_take isl_printer *p, isl_int *c, int len)

{

	int i;

	int first;

	for (i = 0, first = 1; i < len; ++i) {

		int flip = 0;

		if (isl_int_is_zero(c[i]))

			continue;

		if (!first) {

			if (isl_int_is_neg(c[i])) {

				flip = 1;

				isl_int_neg(c[i], c[i]);

				p = isl_printer_print_str(p, " - ");

			} else 

				p = isl_printer_print_str(p, " + ");

		}

		first = 0;

		p = print_term(dim, div, c[i], i, p, 0);

		if (flip)

			isl_int_neg(c[i], c[i]);

	}

	if (first)

		p = isl_printer_print_str(p, "0");

	return p;

}

static __isl_give isl_printer *print_affine(__isl_keep isl_basic_map *bmap,

	__isl_keep isl_space *dim, __isl_take isl_printer *p, isl_int *c)

{

	unsigned len = 1 + isl_basic_map_total_dim(bmap);

	return print_affine_of_len(dim, NULL, p, c, len);

}

/* Internal data structure for print_space.

 *

 * latex is set if that is the output format.

 * print_dim (if not NULL) is called on each dimension.

 * user is set by the caller of print_space and may be used inside print_dim.

 *

 * space is the global space that is being printed.  This field is set by

 *	print_space.

 * type is the tuple of the global space that is currently being printed.

 *	This field is set by print_space.

 */

struct isl_print_space_data {

	int latex;

	__isl_give isl_printer *(*print_dim)(__isl_take isl_printer *p,

		struct isl_print_space_data *data, unsigned pos);

	void *user;

	isl_space *space;

	enum isl_dim_type type;

};

/* offset is the offset of local_dim inside data->type of data->space.

 */

static __isl_give isl_printer *print_nested_var_list(__isl_take isl_printer *p,

	__isl_keep isl_space *local_dim, enum isl_dim_type local_type,

	struct isl_print_space_data *data, int offset)

{

	int i;

	if (data->space != local_dim && local_type == isl_dim_out)

		offset += local_dim->n_in;

	for (i = 0; i < isl_space_dim(local_dim, local_type); ++i) {

		if (i)

			p = isl_printer_print_str(p, ", ");

		if (data->print_dim)

			p = data->print_dim(p, data, offset + i);

		else

			p = print_name(data->space, p, data->type, offset + i,

					data->latex);

	}

	return p;

}

static __isl_give isl_printer *print_var_list(__isl_take isl_printer *p,

	__isl_keep isl_space *space, enum isl_dim_type type)

{

	struct isl_print_space_data data = { .space = space, .type = type };

	return print_nested_var_list(p, space, type, &data, 0);

}

static __isl_give isl_printer *print_nested_map_dim(__isl_take isl_printer *p,

	__isl_keep isl_space *local_dim,

	struct isl_print_space_data *data, int offset);

static __isl_give isl_printer *print_nested_tuple(__isl_take isl_printer *p,

	__isl_keep isl_space *local_dim, enum isl_dim_type local_type,

	struct isl_print_space_data *data, int offset)

{

	const char *name = NULL;

	unsigned n = isl_space_dim(local_dim, local_type);

	if ((local_type == isl_dim_in || local_type == isl_dim_out)) {

		name = isl_space_get_tuple_name(local_dim, local_type);

		if (name) {

			if (data->latex)

				p = isl_printer_print_str(p, "\\mathrm{");

			p = isl_printer_print_str(p, name);

			if (data->latex)

				p = isl_printer_print_str(p, "}");

		}

	}

	if (!data->latex || n != 1 || name)

		p = isl_printer_print_str(p, s_open_list[data->latex]);

	if ((local_type == isl_dim_in || local_type == isl_dim_out) &&

	    local_dim->nested[local_type - isl_dim_in]) {

		if (data->space != local_dim && local_type == isl_dim_out)

			offset += local_dim->n_in;

		p = print_nested_map_dim(p,

				local_dim->nested[local_type - isl_dim_in],

				data, offset);

	} else

		p = print_nested_var_list(p, local_dim, local_type, data,

					  offset);

	if (!data->latex || n != 1 || name)

		p = isl_printer_print_str(p, s_close_list[data->latex]);

	return p;

}

static __isl_give isl_printer *print_tuple(__isl_keep isl_space *dim,

	__isl_take isl_printer *p, enum isl_dim_type type,

	struct isl_print_space_data *data)

{

	data->space = dim;

	data->type = type;

	return print_nested_tuple(p, dim, type, data, 0);

}

static __isl_give isl_printer *print_nested_map_dim(__isl_take isl_printer *p,

	__isl_keep isl_space *local_dim,

	struct isl_print_space_data *data, int offset)

{

	p = print_nested_tuple(p, local_dim, isl_dim_in, data, offset);

	p = isl_printer_print_str(p, s_to[data->latex]);

	p = print_nested_tuple(p, local_dim, isl_dim_out, data, offset);

	return p;

}

static __isl_give isl_printer *print_space(__isl_keep isl_space *dim,

	__isl_take isl_printer *p, int rational,

	struct isl_print_space_data *data)

{

	if (rational && !data->latex)

		p = isl_printer_print_str(p, "rat: ");

	if (isl_space_is_params(dim))

		;

	else if (isl_space_is_set(dim))

		p = print_tuple(dim, p, isl_dim_set, data);

	else {

		p = print_tuple(dim, p, isl_dim_in, data);

		p = isl_printer_print_str(p, s_to[data->latex]);

		p = print_tuple(dim, p, isl_dim_out, data);

	}

	return p;

}

static __isl_give isl_printer *print_omega_parameters(__isl_keep isl_space *dim,

	__isl_take isl_printer *p)

{

	if (isl_space_dim(dim, isl_dim_param) == 0)

		return p;

	p = isl_printer_start_line(p);

	p = isl_printer_print_str(p, "symbolic ");

	p = print_var_list(p, dim, isl_dim_param);

	p = isl_printer_print_str(p, ";");

	p = isl_printer_end_line(p);

	return p;

}

static __isl_give isl_printer *print_constraint(struct isl_basic_map *bmap,

	__isl_keep isl_space *dim, __isl_take isl_printer *p,

	isl_int *c, int last, const char *op, int first_constraint, int latex)

{

	if (!first_constraint)

		p = isl_printer_print_str(p, s_and[latex]);

	isl_int_abs(c[last], c[last]);

	p = print_term(dim, NULL, c[last], last, p, latex);

	p = isl_printer_print_str(p, " ");

	p = isl_printer_print_str(p, op);

	p = isl_printer_print_str(p, " ");

	isl_int_set_si(c[last], 0);

	p = print_affine(bmap, dim, p, c);

	return p;

}

static __isl_give isl_printer *print_constraints(__isl_keep isl_basic_map *bmap,

	__isl_keep isl_space *dim, __isl_take isl_printer *p, int latex)

{

	int i;

	struct isl_vec *c;

	unsigned total = isl_basic_map_total_dim(bmap);

	c = isl_vec_alloc(bmap->ctx, 1 + total);

	if (!c)

		goto error;

	for (i = bmap->n_eq - 1; i >= 0; --i) {

		int l = isl_seq_last_non_zero(bmap->eq[i], 1 + total);

		if (l < 0) {

			if (i != bmap->n_eq - 1)

				p = isl_printer_print_str(p, s_and[latex]);

			p = isl_printer_print_str(p, "0 = 0");

			continue;

		}

		if (isl_int_is_neg(bmap->eq[i][l]))

			isl_seq_cpy(c->el, bmap->eq[i], 1 + total);

		else

			isl_seq_neg(c->el, bmap->eq[i], 1 + total);

		p = print_constraint(bmap, dim, p, c->el, l,

				    "=", i == bmap->n_eq - 1, latex);

	}

	for (i = 0; i < bmap->n_ineq; ++i) {

		int l = isl_seq_last_non_zero(bmap->ineq[i], 1 + total);

		int s;

		const char *op;

		if (l < 0)

			continue;

		s = isl_int_sgn(bmap->ineq[i][l]);

		if (s < 0)

			isl_seq_cpy(c->el, bmap->ineq[i], 1 + total);

		else

			isl_seq_neg(c->el, bmap->ineq[i], 1 + total);

		op = s < 0 ? s_le[latex] : s_ge[latex];

		p = print_constraint(bmap, dim, p, c->el, l,

					op, !bmap->n_eq && !i, latex);

	}

	isl_vec_free(c);

	return p;

error:

	isl_vec_free(c);

	isl_printer_free(p);

	return NULL;

}

static __isl_give isl_printer *print_div(__isl_keep isl_space *dim,

	__isl_keep isl_mat *div, int pos, __isl_take isl_printer *p)

{

	int c = p->output_format == ISL_FORMAT_C;

	p = isl_printer_print_str(p, c ? "floord(" : "floor((");

	p = print_affine_of_len(dim, div, p,

				div->row[pos] + 1, div->n_col - 1);

	p = isl_printer_print_str(p, c ? ", " : ")/");

	p = isl_printer_print_isl_int(p, div->row[pos][0]);

	p = isl_printer_print_str(p, ")");

	return p;

}

/* Print a comma separated list of div names, with their definitions

 * (provided that they have a definition and we are printing in isl format).

 */

static __isl_give isl_printer *print_div_list(__isl_take isl_printer *p,

	__isl_keep isl_space *space, __isl_keep isl_mat *div, int latex)

{

	int i;

	unsigned n_div;

	if (!p || !space || !div)

		return isl_printer_free(p);

	n_div = isl_mat_rows(div);

	for (i = 0; i < n_div; ++i) {

		if (i)

			p = isl_printer_print_str(p, ", ");

		p = print_name(space, p, isl_dim_div, i, latex);

		if (p->output_format != ISL_FORMAT_ISL ||

		    isl_int_is_zero(div->row[i][0]))

			continue;

		p = isl_printer_print_str(p, " = ");

		p = print_div(space, div, i, p);

	}

	return p;

}

static __isl_give isl_printer *print_disjunct(__isl_keep isl_basic_map *bmap,

	__isl_keep isl_space *dim, __isl_take isl_printer *p, int latex)

{

	if (bmap->n_div > 0) {

		isl_space *space;

		isl_mat *div;

		space = isl_basic_map_get_space(bmap);

		div = isl_basic_map_get_divs(bmap);

		p = isl_printer_print_str(p, s_open_exists[latex]);

		p = print_div_list(p, space, div, latex);

		isl_space_free(space);

		isl_mat_free(div);

		p = isl_printer_print_str(p, ": ");

	}

	p = print_constraints(bmap, dim, p, latex);

	if (bmap->n_div > 0)

		p = isl_printer_print_str(p, s_close_exists[latex]);

	return p;

}

/* Print a colon followed by the constraints of "bmap"

 * to "p", provided there are any constraints.

 * The names of the variables are taken from "space".

 * "latex" is set if the constraints should be printed in LaTeX format.

 */

static __isl_give isl_printer *print_optional_disjunct(

	__isl_keep isl_basic_map *bmap, __isl_keep isl_space *space,

	__isl_take isl_printer *p, int latex)

{

	if (isl_basic_map_is_universe(bmap))

		return p;

	p = isl_printer_print_str(p, ": ");

	p = print_disjunct(bmap, space, p, latex);

	return p;

}

static __isl_give isl_printer *basic_map_print_omega(

	__isl_keep isl_basic_map *bmap, __isl_take isl_printer *p)

{

	p = isl_printer_print_str(p, "{ [");

	p = print_var_list(p, bmap->dim, isl_dim_in);

	p = isl_printer_print_str(p, "] -> [");

	p = print_var_list(p, bmap->dim, isl_dim_out);

	p = isl_printer_print_str(p, "] ");

	p = print_optional_disjunct(bmap, bmap->dim, p, 0);

	p = isl_printer_print_str(p, " }");

	return p;

}

static __isl_give isl_printer *basic_set_print_omega(

	__isl_keep isl_basic_set *bset, __isl_take isl_printer *p)

{

	p = isl_printer_print_str(p, "{ [");

	p = print_var_list(p, bset->dim, isl_dim_set);

	p = isl_printer_print_str(p, "] ");

	p = print_optional_disjunct(bset, bset->dim, p, 0);

	p = isl_printer_print_str(p, " }");

	return p;

}

static __isl_give isl_printer *isl_map_print_omega(__isl_keep isl_map *map,

	__isl_take isl_printer *p)

{

	int i;

	for (i = 0; i < map->n; ++i) {

		if (i)

			p = isl_printer_print_str(p, " union ");

		p = basic_map_print_omega(map->p[i], p);

	}

	return p;

}

static __isl_give isl_printer *isl_set_print_omega(__isl_keep isl_set *set,

	__isl_take isl_printer *p)

{

	int i;

	for (i = 0; i < set->n; ++i) {

		if (i)

			p = isl_printer_print_str(p, " union ");

		p = basic_set_print_omega(set->p[i], p);

	}

	return p;

}

static __isl_give isl_printer *isl_basic_map_print_isl(

	__isl_keep isl_basic_map *bmap, __isl_take isl_printer *p,

	int latex)

{

	struct isl_print_space_data data = { .latex = latex };

	int rational = ISL_F_ISSET(bmap, ISL_BASIC_MAP_RATIONAL);

	if (isl_basic_map_dim(bmap, isl_dim_param) > 0) {

		p = print_tuple(bmap->dim, p, isl_dim_param, &data);

		p = isl_printer_print_str(p, " -> ");

	}

	p = isl_printer_print_str(p, "{ ");

	p = print_space(bmap->dim, p, rational, &data);

	p = isl_printer_print_str(p, " : ");

	p = print_disjunct(bmap, bmap->dim, p, latex);

	p = isl_printer_print_str(p, " }");

	return p;

}

static __isl_give isl_printer *print_disjuncts(__isl_keep isl_map *map,

	__isl_take isl_printer *p, int latex)

{

	int i;

	if (isl_map_plain_is_universe(map))

		return p;

	p = isl_printer_print_str(p, s_such_that[latex]);

	if (map->n == 0)

		p = isl_printer_print_str(p, "1 = 0");

	for (i = 0; i < map->n; ++i) {

		if (i)

			p = isl_printer_print_str(p, s_or[latex]);

		if (map->n > 1 && map->p[i]->n_eq + map->p[i]->n_ineq > 1)

			p = isl_printer_print_str(p, "(");

		p = print_disjunct(map->p[i], map->dim, p, latex);

		if (map->n > 1 && map->p[i]->n_eq + map->p[i]->n_ineq > 1)

			p = isl_printer_print_str(p, ")");

	}

	return p;

}

/* Print the disjuncts of a map (or set).

 * If the map turns out to be a universal parameter domain, then

 * we need to print the colon.  Otherwise, the output looks identical

 * to the empty set.

 */

static __isl_give isl_printer *print_disjuncts_map(__isl_keep isl_map *map,

	__isl_take isl_printer *p, int latex)

{

	if (isl_map_plain_is_universe(map) && isl_space_is_params(map->dim))

		return isl_printer_print_str(p, s_such_that[latex]);

	else

		return print_disjuncts(map, p, latex);

}

struct isl_aff_split {

	isl_basic_map *aff;

	isl_map *map;

};

static void free_split(__isl_take struct isl_aff_split *split, int n)

{

	int i;

	if (!split)

		return;

	for (i = 0; i < n; ++i) {

		isl_basic_map_free(split[i].aff);

		isl_map_free(split[i].map);

	}

	free(split);

}

static __isl_give isl_basic_map *get_aff(__isl_take isl_basic_map *bmap)

{

	int i, j;

	unsigned nparam, n_in, n_out, total;

	bmap = isl_basic_map_cow(bmap);

	if (!bmap)

		return NULL;

	if (isl_basic_map_free_inequality(bmap, bmap->n_ineq) < 0)

		goto error;

	nparam = isl_basic_map_dim(bmap, isl_dim_param);

	n_in = isl_basic_map_dim(bmap, isl_dim_in);

	n_out = isl_basic_map_dim(bmap, isl_dim_out);

	total = isl_basic_map_dim(bmap, isl_dim_all);

	for (i = bmap->n_eq - 1; i >= 0; --i) {

		j = isl_seq_last_non_zero(bmap->eq[i] + 1, total);

		if (j >= nparam && j < nparam + n_in + n_out &&

		    (isl_int_is_one(bmap->eq[i][1 + j]) ||

		     isl_int_is_negone(bmap->eq[i][1 + j])))

			continue;

		if (isl_basic_map_drop_equality(bmap, i) < 0)

			goto error;

	}

	bmap = isl_basic_map_finalize(bmap);

	return bmap;

error:

	isl_basic_map_free(bmap);

	return NULL;

}

static int aff_split_cmp(const void *p1, const void *p2, void *user)

{

	const struct isl_aff_split *s1, *s2;

	s1 = (const struct isl_aff_split *) p1;

	s2 = (const struct isl_aff_split *) p2;

	return isl_basic_map_plain_cmp(s1->aff, s2->aff);

}

static __isl_give isl_basic_map *drop_aff(__isl_take isl_basic_map *bmap,

	__isl_keep isl_basic_map *aff)

{

	int i, j;

	unsigned total;

	if (!bmap || !aff)

		goto error;

	total = isl_space_dim(bmap->dim, isl_dim_all);

	for (i = bmap->n_eq - 1; i >= 0; --i) {

		if (isl_seq_first_non_zero(bmap->eq[i] + 1 + total,

					    bmap->n_div) != -1)

			continue;

		for (j = 0; j < aff->n_eq; ++j) {

			if (!isl_seq_eq(bmap->eq[i], aff->eq[j], 1 + total) &&

			    !isl_seq_is_neg(bmap->eq[i], aff->eq[j], 1 + total))

				continue;

			if (isl_basic_map_drop_equality(bmap, i) < 0)

				goto error;

			break;

		}

	}

	return bmap;

error:

	isl_basic_map_free(bmap);

	return NULL;

}

static __isl_give struct isl_aff_split *split_aff(__isl_keep isl_map *map)

{

	int i, n;

	struct isl_aff_split *split;

	isl_ctx *ctx;

	ctx = isl_map_get_ctx(map);

	split = isl_calloc_array(ctx, struct isl_aff_split, map->n);

	if (!split)

		return NULL;

	for (i = 0; i < map->n; ++i) {

		isl_basic_map *bmap;

		split[i].aff = get_aff(isl_basic_map_copy(map->p[i]));

		bmap = isl_basic_map_copy(map->p[i]);

		bmap = isl_basic_map_cow(bmap);

		bmap = drop_aff(bmap, split[i].aff);

		split[i].map = isl_map_from_basic_map(bmap);

		if (!split[i].aff || !split[i].map)

			goto error;

	}

	if (isl_sort(split, map->n, sizeof(struct isl_aff_split),

			&aff_split_cmp, NULL) < 0)

		goto error;

	n = map->n;

	for (i = n - 1; i >= 1; --i) {

		if (!isl_basic_map_plain_is_equal(split[i - 1].aff,

						 split[i].aff))

			continue;

		isl_basic_map_free(split[i].aff);

		split[i - 1].map = isl_map_union(split[i - 1].map,

						 split[i].map);

		if (i != n - 1)

			split[i] = split[n - 1];

		split[n - 1].aff = NULL;

		split[n - 1].map = NULL;

		--n;

	}

	return split;

error:

	free_split(split, map->n);

	return NULL;

}

static int defining_equality(__isl_keep isl_basic_map *eq,

	__isl_keep isl_space *dim, enum isl_dim_type type, int pos)

{

	int i;

	unsigned total;

	if (!eq)

		return -1;

	pos += isl_space_offset(dim, type);

	total = isl_basic_map_total_dim(eq);

	for (i = 0; i < eq->n_eq; ++i) {

		if (isl_seq_last_non_zero(eq->eq[i] + 1, total) != pos)

			continue;

		if (isl_int_is_one(eq->eq[i][1 + pos]))

			isl_seq_neg(eq->eq[i], eq->eq[i], 1 + total);

		return i;

	}

	return -1;

}

/* Print dimension "pos" of data->space to "p".

 *

 * data->user is assumed to be an isl_basic_map keeping track of equalities.

 *

 * If the current dimension is defined by these equalities, then print

 * the corresponding expression.  Otherwise, print the name of the dimension.

 */

static __isl_give isl_printer *print_dim_eq(__isl_take isl_printer *p,

	struct isl_print_space_data *data, unsigned pos)

{

	isl_basic_map *eq = data->user;

	int j;

	j = defining_equality(eq, data->space, data->type, pos);

	if (j >= 0) {