From d2ec7dd24a7eeab018d12e3ae27d2d1c7226a7f3 Mon Sep 17 00:00:00 2001
From: Vivien Maisonneuve <v.maisonneuve@gmail.com>
Date: Fri, 4 Jul 2014 21:15:37 +0200
Subject: [PATCH 1/1] Minor refactoring of Domain.points()

---
 pypol/domains.py   | 51 +++++++++++++++++++++++-----------------------
 pypol/islhelper.py |  7 +++++++
 pypol/polyhedra.py |  2 +-
 3 files changed, 33 insertions(+), 27 deletions(-)

diff --git a/pypol/domains.py b/pypol/domains.py
index 05c9c8f..a924b83 100644
--- a/pypol/domains.py
+++ b/pypol/domains.py
@@ -250,34 +250,33 @@ class Domain:
         return value
 
     def vertices(self):
-        if self.isbounded():
-            islbset = self._toislbasicset(self.equalities, self.inequalities, self.symbols)
-            vertices = libisl.isl_basic_set_compute_vertices(islbset);
-            vertexes = islhelper.isl_vertices_vertices(vertices)
-            #vertex = libisl.isl_vertices_get_n_vertices(vertices)
-            for verts in vertexes:
-                expr = libisl.isl_vertex_get_expr(verts);
-                this = islhelper.isl_set_to_str(expr)
-                print(this)
-        else:
-            raise TypeError('set must be bounded')
-        return string
+        islbset = self._toislbasicset(self.equalities, self.inequalities, self.symbols)
+        vertices = libisl.isl_basic_set_compute_vertices(islbset);
+        vertices = islhelper.isl_vertices_vertices(vertices)
+        for vertex in vertices:
+            expr = libisl.isl_vertex_get_expr(vertex);
+            if islhelper.isl_version < '0.13':
+                string = islhelper.isl_set_to_str(expr)
+            else:
+                string = islhelper.isl_multi_aff_to_str(expr)
+            print(string)
 
     def points(self):
-        bounds = {}
-        coordinates = []
-        symbols = self.symbols
-        if self.isbounded():
-            islset = self._toislset(self.polyhedra, self.symbols)
-            points = islhelper.isl_set_points(islset)
-            for sym in symbols:
-                for point in points:
-                    coordinate = libisl.isl_point_get_coordinate_val(point, libisl.isl_dim_set, symbols.index(sym))
-                    coordinate = islhelper.isl_val_to_int(coordinate)
-                    coordinates.append(coordinate)
-        else:
-            raise TypeError('set must be bounded')
-        return coordinates
+        if not self.isbounded():
+            raise ValueError('domain must be unbounded')
+        from .polyhedra import Universe, Eq
+        islset = self._toislset(self.polyhedra, self.symbols)
+        islpoints = islhelper.isl_set_points(islset)
+        points = []
+        for islpoint in islpoints:
+            point = {}
+            for index, symbol in enumerate(self.symbols):
+                coordinate = libisl.isl_point_get_coordinate_val(islpoint,
+                    libisl.isl_dim_set, index)
+                coordinate = islhelper.isl_val_to_int(coordinate)
+                point[symbol] = coordinate
+            points.append(point)
+        return points
 
     @classmethod
     def _fromislset(cls, islset, symbols):
diff --git a/pypol/islhelper.py b/pypol/islhelper.py
index 21bd5b9..a493363 100644
--- a/pypol/islhelper.py
+++ b/pypol/islhelper.py
@@ -13,6 +13,7 @@ __all__ = [
     'isl_set_to_str', 'isl_set_basic_sets',
     'isl_set_points',
     'isl_vertices_vertices',
+    'isl_multi_aff_to_str',
 ]
 
 
@@ -45,3 +46,9 @@ def isl_set_to_str(islset):
     islpr = libisl.isl_printer_print_set(islpr, islset)
     string = libisl.isl_printer_get_str(islpr).decode()
     return string
+
+def isl_multi_aff_to_str(islmaff):
+    islpr = libisl.isl_printer_to_str(mainctx)
+    islpr = libisl.isl_printer_print_multi_aff(islpr, islmaff)
+    string = libisl.isl_printer_get_str(islpr).decode()
+    return string
diff --git a/pypol/polyhedra.py b/pypol/polyhedra.py
index 7181565..f93f31e 100644
--- a/pypol/polyhedra.py
+++ b/pypol/polyhedra.py
@@ -84,7 +84,7 @@ class Polyhedron(Domain):
 
     def polyhedral_hull(self):
         return self
-    
+
     @classmethod
     def _fromislbasicset(cls, islbset, symbols):
         islconstraints = islhelper.isl_basic_set_constraints(islbset)
-- 
2.20.1