From: Vivien Maisonneuve <v.maisonneuve@gmail.com>
Date: Sun, 6 Jul 2014 10:22:41 +0000 (+0200)
Subject: Add methods Domain.subs(), Polyhedron.subs()
X-Git-Tag: 1.0~138
X-Git-Url: https://scm.cri.mines-paristech.fr/git/linpy.git/commitdiff_plain/17f12d8f63d696f09dbb44be3dc24b9aba7b88d3?ds=inline

Add methods Domain.subs(), Polyhedron.subs()
---

diff --git a/pypol/domains.py b/pypol/domains.py
index fdb0ab0..2fbd544 100644
--- a/pypol/domains.py
+++ b/pypol/domains.py
@@ -301,6 +301,11 @@ class Domain:
             points.append(point)
         return points
 
+    def subs(self, symbol, expression=None):
+        polyhedra = [polyhedron.subs(symbol, expression)
+            for polyhedron in self.polyhedra]
+        return Domain(*polyhedra)
+
     @classmethod
     def _fromislset(cls, islset, symbols):
         from .polyhedra import Polyhedron
@@ -323,6 +328,7 @@ class Domain:
             self._dimension = len(self._symbols)
             return self
 
+    @classmethod
     def _toislset(cls, polyhedra, symbols):
         polyhedron = polyhedra[0]
         islbset = polyhedron._toislbasicset(polyhedron.equalities,
diff --git a/pypol/polyhedra.py b/pypol/polyhedra.py
index a6adeda..63ecb64 100644
--- a/pypol/polyhedra.py
+++ b/pypol/polyhedra.py
@@ -85,6 +85,13 @@ class Polyhedron(Domain):
     def aspolyhedron(self):
         return self
 
+    def subs(self, symbol, expression=None):
+        equalities = [equality.subs(symbol, expression)
+            for equality in self.equalities]
+        inequalities = [inequality.subs(symbol, expression)
+            for inequality in self.inequalities]
+        return Polyhedron(equalities, inequalities)
+
     @classmethod
     def _fromislbasicset(cls, islbset, symbols):
         islconstraints = islhelper.isl_basic_set_constraints(islbset)