From: Sven Verdoolaege <skimo@kotnet.org>
Date: Thu, 29 Mar 2012 12:41:23 +0000 (+0200)
Subject: isl_schedule.c: extract out extract_schedule_row
X-Git-Tag: isl-0.13~114
X-Git-Url: https://repo.or.cz/w/isl.git/commitdiff_plain/d22d7142fb77a56f85f23a42a41e68ae56000c29

isl_schedule.c: extract out extract_schedule_row

We will be able to reuse this function when adding support for
conditional validity constraints.

Signed-off-by: Sven Verdoolaege <skimo@kotnet.org>
---

diff --git a/isl_schedule.c b/isl_schedule.c
index 9f0a8ff0..3ada8b6b 100644
--- a/isl_schedule.c
+++ b/isl_schedule.c
@@ -1746,18 +1746,46 @@ error:
 	return -1;
 }
 
+/* Convert row "row" of node->sched into an isl_aff living in "ls"
+ * and return this isl_aff.
+ */
+static __isl_give isl_aff *extract_schedule_row(__isl_take isl_local_space *ls,
+	struct isl_sched_node *node, int row)
+{
+	int j;
+	isl_int v;
+	isl_aff *aff;
+
+	isl_int_init(v);
+
+	aff = isl_aff_zero_on_domain(ls);
+	isl_mat_get_element(node->sched, row, 0, &v);
+	aff = isl_aff_set_constant(aff, v);
+	for (j = 0; j < node->nparam; ++j) {
+		isl_mat_get_element(node->sched, row, 1 + j, &v);
+		aff = isl_aff_set_coefficient(aff, isl_dim_param, j, v);
+	}
+	for (j = 0; j < node->nvar; ++j) {
+		isl_mat_get_element(node->sched, row, 1 + node->nparam + j, &v);
+		aff = isl_aff_set_coefficient(aff, isl_dim_in, j, v);
+	}
+
+	isl_int_clear(v);
+
+	return aff;
+}
+
 /* Convert node->sched into a multi_aff and return this multi_aff.
  */
 static __isl_give isl_multi_aff *node_extract_schedule_multi_aff(
 	struct isl_sched_node *node)
 {
-	int i, j;
+	int i;
 	isl_space *space;
 	isl_local_space *ls;
 	isl_aff *aff;
 	isl_multi_aff *ma;
 	int nrow, ncol;
-	isl_int v;
 
 	nrow = isl_mat_rows(node->sched);
 	ncol = isl_mat_cols(node->sched) - 1;
@@ -1766,26 +1794,11 @@ static __isl_give isl_multi_aff *node_extract_schedule_multi_aff(
 	ma = isl_multi_aff_zero(space);
 	ls = isl_local_space_from_space(isl_space_copy(node->dim));
 
-	isl_int_init(v);
-
 	for (i = 0; i < nrow; ++i) {
-		aff = isl_aff_zero_on_domain(isl_local_space_copy(ls));
-		isl_mat_get_element(node->sched, i, 0, &v);
-		aff = isl_aff_set_constant(aff, v);
-		for (j = 0; j < node->nparam; ++j) {
-			isl_mat_get_element(node->sched, i, 1 + j, &v);
-			aff = isl_aff_set_coefficient(aff, isl_dim_param, j, v);
-		}
-		for (j = 0; j < node->nvar; ++j) {
-			isl_mat_get_element(node->sched,
-					    i, 1 + node->nparam + j, &v);
-			aff = isl_aff_set_coefficient(aff, isl_dim_in, j, v);
-		}
+		aff = extract_schedule_row(isl_local_space_copy(ls), node, i);
 		ma = isl_multi_aff_set_aff(ma, i, aff);
 	}
 
-	isl_int_clear(v);
-
 	isl_local_space_free(ls);
 
 	return ma;