From eb1b6a89720a4ae455da05228c7634613e22797b Mon Sep 17 00:00:00 2001
From: Sven Verdoolaege <skimo@kotnet.org>
Date: Sun, 21 Mar 2010 16:36:46 +0100
Subject: [PATCH] add isl_pw_qpolynomial_add_dims

---
 include/isl_polynomial.h |  3 ++
 isl_polynomial.c         | 76 ++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 79 insertions(+)

diff --git a/include/isl_polynomial.h b/include/isl_polynomial.h
index 88e2257..39c932a 100644
--- a/include/isl_polynomial.h
+++ b/include/isl_polynomial.h
@@ -107,6 +107,9 @@ __isl_give isl_pw_qpolynomial *isl_pw_qpolynomial_mul(
 	__isl_take isl_pw_qpolynomial *pwqp1,
 	__isl_take isl_pw_qpolynomial *pwqp2);
 
+__isl_give isl_pw_qpolynomial *isl_pw_qpolynomial_add_dims(
+	__isl_take isl_pw_qpolynomial *pwqp,
+	enum isl_dim_type type, unsigned n);
 __isl_give isl_pw_qpolynomial *isl_pw_qpolynomial_move(
 	__isl_take isl_pw_qpolynomial *pwqp,
 	enum isl_dim_type dst_type, unsigned dst_pos,
diff --git a/isl_polynomial.c b/isl_polynomial.c
index b80ac87..d393836 100644
--- a/isl_polynomial.c
+++ b/isl_polynomial.c
@@ -1959,6 +1959,82 @@ error:
 	return NULL;
 }
 
+__isl_give isl_qpolynomial *isl_qpolynomial_add_dims(
+	__isl_take isl_qpolynomial *qp, enum isl_dim_type type, unsigned n)
+{
+	unsigned total;
+	unsigned g_pos;
+	int *exp;
+
+	if (n == 0)
+		return qp;
+
+	qp = isl_qpolynomial_cow(qp);
+	if (!qp)
+		return NULL;
+
+	g_pos = pos(qp->dim, type) + isl_dim_size(qp->dim, type);
+
+	qp->div = isl_mat_insert_cols(qp->div, 2 + g_pos, n);
+	if (!qp->div)
+		goto error;
+
+	total = qp->div->n_col - 2;
+	if (total > g_pos) {
+		int i;
+		exp = isl_alloc_array(qp->div->ctx, int, total - g_pos);
+		if (!exp)
+			goto error;
+		for (i = 0; i < total - g_pos; ++i)
+			exp[i] = i + n;
+		qp->upoly = expand(qp->upoly, exp, g_pos);
+		free(exp);
+		if (!qp->upoly)
+			goto error;
+	}
+
+	qp->dim = isl_dim_add(qp->dim, type, n);
+	if (!qp->dim)
+		goto error;
+
+	return qp;
+error:
+	isl_qpolynomial_free(qp);
+	return NULL;
+}
+
+__isl_give isl_pw_qpolynomial *isl_pw_qpolynomial_add_dims(
+	__isl_take isl_pw_qpolynomial *pwqp,
+	enum isl_dim_type type, unsigned n)
+{
+	int i;
+
+	if (n == 0)
+		return pwqp;
+
+	pwqp = isl_pw_qpolynomial_cow(pwqp);
+	if (!pwqp)
+		return NULL;
+
+	pwqp->dim = isl_dim_add(pwqp->dim, type, n);
+	if (!pwqp->dim)
+		goto error;
+
+	for (i = 0; i < pwqp->n; ++i) {
+		pwqp->p[i].set = isl_set_add(pwqp->p[i].set, type, n);
+		if (!pwqp->p[i].set)
+			goto error;
+		pwqp->p[i].qp = isl_qpolynomial_add_dims(pwqp->p[i].qp, type, n);
+		if (!pwqp->p[i].qp)
+			goto error;
+	}
+
+	return pwqp;
+error:
+	isl_pw_qpolynomial_free(pwqp);
+	return NULL;
+}
+
 static int *reordering_move(isl_ctx *ctx,
 	unsigned len, unsigned dst, unsigned src, unsigned n)
 {
-- 
2.7.4