From d6444456ceaeed6fcf1d44386edefb5b1fb8ec66 Mon Sep 17 00:00:00 2001 From: Vivien Maisonneuve Date: Tue, 1 Jul 2014 19:05:07 +0200 Subject: [PATCH] Overloading for Domains.project_out(), to be improved --- examples/diamond.py | 2 +- pypol/domains.py | 11 ++++++++++- 2 files changed, 11 insertions(+), 2 deletions(-) diff --git a/examples/diamond.py b/examples/diamond.py index ad0942b..1681054 100755 --- a/examples/diamond.py +++ b/examples/diamond.py @@ -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])) diff --git a/pypol/domains.py b/pypol/domains.py index c844e55..6b47fe8 100644 --- a/pypol/domains.py +++ b/pypol/domains.py @@ -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))): -- 2.20.1