X-Git-Url: https://scm.cri.ensmp.fr/git/linpy.git/blobdiff_plain/5fa0e7932a1621bee505991475291e43225a4b49..7b93cea1daf2889e9ee10ca9c22a1b5124404937:/pypol/domains.py diff --git a/pypol/domains.py b/pypol/domains.py deleted file mode 100644 index a34b75b..0000000 --- a/pypol/domains.py +++ /dev/null @@ -1,277 +0,0 @@ -import functools - -from . import islhelper - -from .islhelper import mainctx, libisl, isl_set_basic_sets - - -__all__ = [ - 'Domain', - 'And', 'Or', 'Not', -] - - -@functools.total_ordering -class Domain: - - __slots__ = ( - '_polyhedra', - '_symbols', - '_dimension', - ) - - def __new__(cls, *polyhedra): - from .polyhedra import Polyhedron - if len(polyhedra) == 1: - polyhedron = polyhedra[0] - if isinstance(polyhedron, str): - return cls.fromstring(polyhedron) - elif isinstance(polyhedron, Polyhedron): - return polyhedron - else: - raise TypeError('argument must be a string ' - 'or a Polyhedron instance') - else: - for polyhedron in polyhedra: - if not isinstance(polyhedron, Polyhedron): - raise TypeError('arguments must be Polyhedron instances') - symbols = cls._xsymbols(polyhedra) - islset = cls._toislset(polyhedra, symbols) - return cls._fromislset(islset, symbols) - - @classmethod - def _xsymbols(cls, iterator): - """ - Return the ordered tuple of symbols present in iterator. - """ - symbols = set() - for item in iterator: - symbols.update(item.symbols) - return tuple(sorted(symbols)) - - @property - def polyhedra(self): - return self._polyhedra - - @property - def symbols(self): - return self._symbols - - @property - def dimension(self): - return self._dimension - - def disjoint(self): - islset = self._toislset(self.polyhedra, self.symbols) - islset = libisl.isl_set_make_disjoint(mainctx, islset) - return self._fromislset(islset, self.symbols) - - def isempty(self): - islset = self._toislset(self.polyhedra, self.symbols) - empty = bool(libisl.isl_set_is_empty(islset)) - libisl.isl_set_free(islset) - return empty - - def __bool__(self): - return not self.isempty() - - def isuniverse(self): - islset = self._toislset(self.polyhedra, self.symbols) - universe = bool(libisl.isl_set_plain_is_universe(islset)) - libisl.isl_set_free(islset) - return universe - - def isbounded(self): - islset = self._toislset(self.polyhedra, self.symbols) - bounded = bool(libisl.isl_set_is_bounded(islset)) - libisl.isl_set_free(islset) - return bounded - - def __eq__(self, other): - symbols = self._xsymbols([self, other]) - islset1 = self._toislset(self.polyhedra, symbols) - islset2 = other._toislset(other.polyhedra, symbols) - equal = bool(libisl.isl_set_is_equal(islset1, islset2)) - libisl.isl_set_free(islset1) - libisl.isl_set_free(islset2) - return equal - - def isdisjoint(self, other): - symbols = self._xsymbols([self, other]) - islset1 = self._toislset(self.polyhedra, symbols) - islset2 = self._toislset(other.polyhedra, symbols) - equal = bool(libisl.isl_set_is_disjoint(islset1, islset2)) - libisl.isl_set_free(islset1) - libisl.isl_set_free(islset2) - return equal - - def issubset(self, other): - symbols = self._xsymbols([self, other]) - islset1 = self._toislset(self.polyhedra, symbols) - islset2 = self._toislset(other.polyhedra, symbols) - equal = bool(libisl.isl_set_is_subset(islset1, islset2)) - libisl.isl_set_free(islset1) - libisl.isl_set_free(islset2) - return equal - - def __le__(self, other): - return self.issubset(other) - - def __lt__(self, other): - symbols = self._xsymbols([self, other]) - islset1 = self._toislset(self.polyhedra, symbols) - islset2 = self._toislset(other.polyhedra, symbols) - equal = bool(libisl.isl_set_is_strict_subset(islset1, islset2)) - libisl.isl_set_free(islset1) - libisl.isl_set_free(islset2) - return equal - - def complement(self): - islset = self._toislset(self.polyhedra, self.symbols) - islset = libisl.isl_set_complement(islset) - return self._fromislset(islset, self.symbols) - - def __invert__(self): - return self.complement() - - def simplify(self): - # see isl_set_coalesce, isl_set_detect_equalities, - # isl_set_remove_redundancies - # which ones? in which order? - raise NotImplementedError - - def polyhedral_hull(self): - # several types of hull are available - # polyhedral seems to be the more appropriate, to be checked - from .polyhedra import Polyhedron - islset = self._toislset(self.polyhedra, self.symbols) - islbset = libisl.isl_set_polyhedral_hull(islset) - return Polyhedron._fromislbasicset(islbset, self.symbols) - - def project(self, symbols): - # not sure what isl_set_project_out actually does… - # use isl_set_drop_constraints_involving_dims instead? - raise NotImplementedError - - def sample(self): - from .polyhedra import Polyhedron - islset = self._toislset(self.polyhedra, self.symbols) - islbset = libisl.isl_set_sample(islset) - return Polyhedron._fromislbasicset(islbset, self.symbols) - - def intersection(self, *others): - if len(others) == 0: - return self - symbols = self._xsymbols((self,) + others) - islset1 = self._toislset(self.polyhedra, symbols) - for other in others: - islset2 = other._toislset(other.polyhedra, symbols) - islset1 = libisl.isl_set_intersect(islset1, islset2) - return self._fromislset(islset1, symbols) - - def __and__(self, other): - return self.intersection(other) - - def union(self, *others): - if len(others) == 0: - return self - symbols = self._xsymbols((self,) + others) - islset1 = self._toislset(self.polyhedra, symbols) - for other in others: - islset2 = other._toislset(other.polyhedra, symbols) - islset1 = libisl.isl_set_union(islset1, islset2) - return self._fromislset(islset1, symbols) - - def __or__(self, other): - return self.union(other) - - def __add__(self, other): - return self.union(other) - - def difference(self, other): - symbols = self._xsymbols([self, other]) - islset1 = self._toislset(self.polyhedra, symbols) - islset2 = other._toislset(other.polyhedra, symbols) - islset = libisl.isl_set_subtract(islset1, islset2) - return self._fromislset(islset, symbols) - - def __sub__(self, other): - return self.difference(other) - - def lexmin(self): - islset = self._toislset(self.polyhedra, self.symbols) - islset = libisl.isl_set_lexmin(islset) - return self._fromislset(islset, self.symbols) - - def lexmax(self): - islset = self._toislset(self.polyhedra, self.symbols) - islset = libisl.isl_set_lexmax(islset) - return self._fromislset(islset, self.symbols) - - @classmethod - def _fromislset(cls, islset, symbols): - from .polyhedra import Polyhedron - islset = libisl.isl_set_remove_divs(islset) - islbsets = isl_set_basic_sets(islset) - libisl.isl_set_free(islset) - polyhedra = [] - for islbset in islbsets: - polyhedron = Polyhedron._fromislbasicset(islbset, symbols) - polyhedra.append(polyhedron) - if len(polyhedra) == 0: - from .polyhedra import Empty - return Empty - elif len(polyhedra) == 1: - return polyhedra[0] - else: - self = object().__new__(Domain) - self._polyhedra = tuple(polyhedra) - self._symbols = cls._xsymbols(polyhedra) - self._dimension = len(self._symbols) - return self - - def _toislset(cls, polyhedra, symbols): - polyhedron = polyhedra[0] - islbset = polyhedron._toislbasicset(polyhedron.equalities, - polyhedron.inequalities, symbols) - islset1 = libisl.isl_set_from_basic_set(islbset) - for polyhedron in polyhedra[1:]: - islbset = polyhedron._toislbasicset(polyhedron.equalities, - polyhedron.inequalities, symbols) - islset2 = libisl.isl_set_from_basic_set(islbset) - islset1 = libisl.isl_set_union(islset1, islset2) - return islset1 - - @classmethod - def fromstring(cls, string): - raise NotImplementedError - - def __repr__(self): - assert len(self.polyhedra) >= 2 - strings = [repr(polyhedron) for polyhedron in self.polyhedra] - return 'Or({})'.format(', '.join(strings)) - - @classmethod - def fromsympy(cls, expr): - raise NotImplementedError - - def tosympy(self): - raise NotImplementedError - - -def And(*domains): - if len(domains) == 0: - from .polyhedra import Universe - return Universe - else: - return domains[0].intersection(*domains[1:]) - -def Or(*domains): - if len(domains) == 0: - from .polyhedra import Empty - return Empty - else: - return domains[0].union(*domains[1:]) - -def Not(domain): - return ~domain