Overloading for Domains.project_out(), to be improved
authorVivien Maisonneuve <v.maisonneuve@gmail.com>
Tue, 1 Jul 2014 17:05:07 +0000 (19:05 +0200)
committerVivien Maisonneuve <v.maisonneuve@gmail.com>
Tue, 1 Jul 2014 17:05:31 +0000 (19:05 +0200)
examples/diamond.py
pypol/domains.py

index ad0942b..1681054 100755 (executable)
@@ -5,4 +5,4 @@ from pypol import *
 x, y = symbols('x y')
 diam = Ge(y, x - 1) & Le(y, x + 1) & Ge(y, -x - 1) & Le(y, -x + 1)
 print('diamond:', diam)
-print('projected on x:', diam.drop_dims('y'))
+print('projected on x:', diam.project_out([y]))
index c844e55..6b47fe8 100644 (file)
@@ -5,7 +5,7 @@ import re
 from . import islhelper
 
 from .islhelper import mainctx, libisl, isl_set_basic_sets
-from .linexprs import Expression
+from .linexprs import Expression, Symbol
 
 
 __all__ = [
@@ -154,6 +154,15 @@ class Domain:
 
     def project_out(self, symbols):
         # use to remove certain variables
+        if isinstance(symbols, str):
+            symbols = symbols.replace(',', ' ').split()
+        else:
+            symbols = list(symbols)
+            for i, symbol in enumerate(symbols):
+                if isinstance(symbol, Symbol):
+                    symbols[i] = symbol.name
+                elif not isinstance(symbol, str):
+                    raise TypeError('symbols must be strings or Symbol instances')
         islset = self._toislset(self.polyhedra, self.symbols)
         # the trick is to walk symbols in reverse order, to avoid index updates
         for index, symbol in reversed(list(enumerate(self.symbols))):