23c21affa5b3bfc28003e5db50c7acc7aebb0977
[linpy.git] / pypol / domains.py
1 import ast
2 import functools
3 import re
4 import math
5
6 from fractions import Fraction
7
8 from . import islhelper
9 from .islhelper import mainctx, libisl
10 from .linexprs import Expression, Symbol, Rational
11 from .geometry import GeometricObject, Point, Vector
12
13
14 __all__ = [
15 'Domain',
16 'And', 'Or', 'Not',
17 ]
18
19
20 @functools.total_ordering
21 class Domain(GeometricObject):
22
23 __slots__ = (
24 '_polyhedra',
25 '_symbols',
26 '_dimension',
27 )
28
29 def __new__(cls, *polyhedra):
30 from .polyhedra import Polyhedron
31 if len(polyhedra) == 1:
32 argument = polyhedra[0]
33 if isinstance(argument, str):
34 return cls.fromstring(argument)
35 elif isinstance(argument, GeometricObject):
36 return argument.aspolyhedron()
37 else:
38 raise TypeError('argument must be a string '
39 'or a GeometricObject instance')
40 else:
41 for polyhedron in polyhedra:
42 if not isinstance(polyhedron, Polyhedron):
43 raise TypeError('arguments must be Polyhedron instances')
44 symbols = cls._xsymbols(polyhedra)
45 islset = cls._toislset(polyhedra, symbols)
46 return cls._fromislset(islset, symbols)
47
48 @classmethod
49 def _xsymbols(cls, iterator):
50 """
51 Return the ordered tuple of symbols present in iterator.
52 """
53 symbols = set()
54 for item in iterator:
55 symbols.update(item.symbols)
56 return tuple(sorted(symbols, key=Symbol.sortkey))
57
58 @property
59 def polyhedra(self):
60 return self._polyhedra
61
62 @property
63 def symbols(self):
64 return self._symbols
65
66 @property
67 def dimension(self):
68 return self._dimension
69
70 def disjoint(self):
71 """
72 Returns this set as disjoint.
73 """
74 islset = self._toislset(self.polyhedra, self.symbols)
75 islset = libisl.isl_set_make_disjoint(mainctx, islset)
76 return self._fromislset(islset, self.symbols)
77
78 def isempty(self):
79 """
80 Returns true if this set is an Empty set.
81 """
82 islset = self._toislset(self.polyhedra, self.symbols)
83 empty = bool(libisl.isl_set_is_empty(islset))
84 libisl.isl_set_free(islset)
85 return empty
86
87 def __bool__(self):
88 return not self.isempty()
89
90 def isuniverse(self):
91 """
92 Returns true if this set is the Universe set.
93 """
94 islset = self._toislset(self.polyhedra, self.symbols)
95 universe = bool(libisl.isl_set_plain_is_universe(islset))
96 libisl.isl_set_free(islset)
97 return universe
98
99 def isbounded(self):
100 """
101 Returns true if this set is bounded.
102 """
103 islset = self._toislset(self.polyhedra, self.symbols)
104 bounded = bool(libisl.isl_set_is_bounded(islset))
105 libisl.isl_set_free(islset)
106 return bounded
107
108 def __eq__(self, other):
109 """
110 Returns true if two sets are equal.
111 """
112 symbols = self._xsymbols([self, other])
113 islset1 = self._toislset(self.polyhedra, symbols)
114 islset2 = other._toislset(other.polyhedra, symbols)
115 equal = bool(libisl.isl_set_is_equal(islset1, islset2))
116 libisl.isl_set_free(islset1)
117 libisl.isl_set_free(islset2)
118 return equal
119
120 def isdisjoint(self, other):
121 """
122 Return True if two sets have a null intersection.
123 """
124 symbols = self._xsymbols([self, other])
125 islset1 = self._toislset(self.polyhedra, symbols)
126 islset2 = self._toislset(other.polyhedra, symbols)
127 equal = bool(libisl.isl_set_is_disjoint(islset1, islset2))
128 libisl.isl_set_free(islset1)
129 libisl.isl_set_free(islset2)
130 return equal
131
132 def issubset(self, other):
133 """
134 Report whether another set contains this set.
135 """
136 symbols = self._xsymbols([self, other])
137 islset1 = self._toislset(self.polyhedra, symbols)
138 islset2 = self._toislset(other.polyhedra, symbols)
139 equal = bool(libisl.isl_set_is_subset(islset1, islset2))
140 libisl.isl_set_free(islset1)
141 libisl.isl_set_free(islset2)
142 return equal
143
144 def __le__(self, other):
145 """
146 Returns true if this set is less than or equal to another set.
147 """
148 return self.issubset(other)
149
150 def __lt__(self, other):
151 """
152 Returns true if this set is less than another set.
153 """
154 symbols = self._xsymbols([self, other])
155 islset1 = self._toislset(self.polyhedra, symbols)
156 islset2 = self._toislset(other.polyhedra, symbols)
157 equal = bool(libisl.isl_set_is_strict_subset(islset1, islset2))
158 libisl.isl_set_free(islset1)
159 libisl.isl_set_free(islset2)
160 return equal
161
162 def complement(self):
163 """
164 Returns the complement of this set.
165 """
166 islset = self._toislset(self.polyhedra, self.symbols)
167 islset = libisl.isl_set_complement(islset)
168 return self._fromislset(islset, self.symbols)
169
170 def __invert__(self):
171 """
172 Returns the complement of this set.
173 """
174 return self.complement()
175
176 def simplify(self):
177 """
178 Returns a set without redundant constraints.
179 """
180 islset = self._toislset(self.polyhedra, self.symbols)
181 islset = libisl.isl_set_remove_redundancies(islset)
182 return self._fromislset(islset, self.symbols)
183
184 def aspolyhedron(self):
185 """
186 Returns polyhedral hull of set.
187 """
188 from .polyhedra import Polyhedron
189 islset = self._toislset(self.polyhedra, self.symbols)
190 islbset = libisl.isl_set_polyhedral_hull(islset)
191 return Polyhedron._fromislbasicset(islbset, self.symbols)
192
193 def asdomain(self):
194 return self
195
196 def project(self, dims):
197 """
198 Return new set with given dimensions removed.
199 """
200 islset = self._toislset(self.polyhedra, self.symbols)
201 n = 0
202 for index, symbol in reversed(list(enumerate(self.symbols))):
203 if symbol in dims:
204 n += 1
205 elif n > 0:
206 islset = libisl.isl_set_project_out(islset, libisl.isl_dim_set, index + 1, n)
207 n = 0
208 if n > 0:
209 islset = libisl.isl_set_project_out(islset, libisl.isl_dim_set, 0, n)
210 dims = [symbol for symbol in self.symbols if symbol not in dims]
211 return Domain._fromislset(islset, dims)
212
213 def sample(self):
214 """
215 Returns a single subset of the input.
216 """
217 islset = self._toislset(self.polyhedra, self.symbols)
218 islpoint = libisl.isl_set_sample_point(islset)
219 if bool(libisl.isl_point_is_void(islpoint)):
220 libisl.isl_point_free(islpoint)
221 raise ValueError('domain must be non-empty')
222 point = {}
223 for index, symbol in enumerate(self.symbols):
224 coordinate = libisl.isl_point_get_coordinate_val(islpoint,
225 libisl.isl_dim_set, index)
226 coordinate = islhelper.isl_val_to_int(coordinate)
227 point[symbol] = coordinate
228 libisl.isl_point_free(islpoint)
229 return point
230
231 def intersection(self, *others):
232 """
233 Return the intersection of two sets as a new set.
234 """
235 if len(others) == 0:
236 return self
237 symbols = self._xsymbols((self,) + others)
238 islset1 = self._toislset(self.polyhedra, symbols)
239 for other in others:
240 islset2 = other._toislset(other.polyhedra, symbols)
241 islset1 = libisl.isl_set_intersect(islset1, islset2)
242 return self._fromislset(islset1, symbols)
243
244 def __and__(self, other):
245 """
246 Return the intersection of two sets as a new set.
247 """
248 return self.intersection(other)
249
250 def union(self, *others):
251 """
252 Return the union of sets as a new set.
253 """
254 if len(others) == 0:
255 return self
256 symbols = self._xsymbols((self,) + others)
257 islset1 = self._toislset(self.polyhedra, symbols)
258 for other in others:
259 islset2 = other._toislset(other.polyhedra, symbols)
260 islset1 = libisl.isl_set_union(islset1, islset2)
261 return self._fromislset(islset1, symbols)
262
263 def __or__(self, other):
264 """
265 Return a new set with elements from both sets.
266 """
267 return self.union(other)
268
269 def __add__(self, other):
270 """
271 Return new set containing all elements in both sets.
272 """
273 return self.union(other)
274
275 def difference(self, other):
276 """
277 Return the difference of two sets as a new set.
278 """
279 symbols = self._xsymbols([self, other])
280 islset1 = self._toislset(self.polyhedra, symbols)
281 islset2 = other._toislset(other.polyhedra, symbols)
282 islset = libisl.isl_set_subtract(islset1, islset2)
283 return self._fromislset(islset, symbols)
284
285 def __sub__(self, other):
286 """
287 Return the difference of two sets as a new set.
288 """
289 return self.difference(other)
290
291 def lexmin(self):
292 """
293 Return a new set containing the lexicographic minimum of the elements in the set.
294 """
295 islset = self._toislset(self.polyhedra, self.symbols)
296 islset = libisl.isl_set_lexmin(islset)
297 return self._fromislset(islset, self.symbols)
298
299 def lexmax(self):
300 """
301 Return a new set containing the lexicographic maximum of the elements in the set.
302 """
303 islset = self._toislset(self.polyhedra, self.symbols)
304 islset = libisl.isl_set_lexmax(islset)
305 return self._fromislset(islset, self.symbols)
306
307 def num_parameters(self):
308 """
309 Return the total number of parameters, input, output or set dimensions.
310 """
311 islbset = self._toislbasicset(self.equalities, self.inequalities, self.symbols)
312 num = libisl.isl_basic_set_dim(islbset, libisl.isl_dim_set)
313 return num
314
315 def involves_dims(self, dims):
316 """
317 Returns true if set depends on given dimensions.
318 """
319 islset = self._toislset(self.polyhedra, self.symbols)
320 dims = sorted(dims)
321 symbols = sorted(list(self.symbols))
322 n = 0
323 if len(dims)>0:
324 for dim in dims:
325 if dim in symbols:
326 first = symbols.index(dims[0])
327 n +=1
328 else:
329 first = 0
330 else:
331 return False
332 value = bool(libisl.isl_set_involves_dims(islset, libisl.isl_dim_set, first, n))
333 libisl.isl_set_free(islset)
334 return value
335
336 _RE_COORDINATE = re.compile(r'\((?P<num>\-?\d+)\)(/(?P<den>\d+))?')
337
338 def vertices(self):
339 """
340 Return a list of vertices for this Polygon.
341 """
342 from .polyhedra import Polyhedron
343 if not self.isbounded():
344 raise ValueError('domain must be bounded')
345 islbset = self._toislbasicset(self.equalities, self.inequalities, self.symbols)
346 vertices = libisl.isl_basic_set_compute_vertices(islbset);
347 vertices = islhelper.isl_vertices_vertices(vertices)
348 points = []
349 for vertex in vertices:
350 expr = libisl.isl_vertex_get_expr(vertex)
351 coordinates = []
352 if islhelper.isl_version < '0.13':
353 constraints = islhelper.isl_basic_set_constraints(expr)
354 for constraint in constraints:
355 constant = libisl.isl_constraint_get_constant_val(constraint)
356 constant = islhelper.isl_val_to_int(constant)
357 for index, symbol in enumerate(self.symbols):
358 coefficient = libisl.isl_constraint_get_coefficient_val(constraint,
359 libisl.isl_dim_set, index)
360 coefficient = islhelper.isl_val_to_int(coefficient)
361 if coefficient != 0:
362 coordinate = -Fraction(constant, coefficient)
363 coordinates.append((symbol, coordinate))
364 else:
365 string = islhelper.isl_multi_aff_to_str(expr)
366 matches = self._RE_COORDINATE.finditer(string)
367 for symbol, match in zip(self.symbols, matches):
368 numerator = int(match.group('num'))
369 denominator = match.group('den')
370 denominator = 1 if denominator is None else int(denominator)
371 coordinate = Fraction(numerator, denominator)
372 coordinates.append((symbol, coordinate))
373 points.append(Point(coordinates))
374 return points
375
376 def points(self):
377 """
378 Returns the points contained in the set.
379 """
380 if not self.isbounded():
381 raise ValueError('domain must be bounded')
382 from .polyhedra import Universe, Eq
383 islset = self._toislset(self.polyhedra, self.symbols)
384 islpoints = islhelper.isl_set_points(islset)
385 points = []
386 for islpoint in islpoints:
387 coordinates = {}
388 for index, symbol in enumerate(self.symbols):
389 coordinate = libisl.isl_point_get_coordinate_val(islpoint,
390 libisl.isl_dim_set, index)
391 coordinate = islhelper.isl_val_to_int(coordinate)
392 coordinates[symbol] = coordinate
393 points.append(Point(coordinates))
394 return points
395
396 @classmethod
397 def _polygon_inner_point(cls, points):
398 symbols = points[0].symbols
399 coordinates = {symbol: 0 for symbol in symbols}
400 for point in points:
401 for symbol, coordinate in point.coordinates():
402 coordinates[symbol] += coordinate
403 for symbol in symbols:
404 coordinates[symbol] /= len(points)
405 return Point(coordinates)
406
407 @classmethod
408 def _sort_polygon_2d(cls, points):
409 if len(points) <= 3:
410 return points
411 o = cls._polygon_inner_point(points)
412 angles = {}
413 for m in points:
414 om = Vector(o, m)
415 dx, dy = (coordinate for symbol, coordinate in om.coordinates())
416 angle = math.atan2(dy, dx)
417 angles[m] = angle
418 return sorted(points, key=angles.get)
419
420 @classmethod
421 def _sort_polygon_3d(cls, points):
422 if len(points) <= 3:
423 return points
424 o = cls._polygon_inner_point(points)
425 a = points[0]
426 oa = Vector(o, a)
427 norm_oa = oa.norm()
428 for b in points[1:]:
429 ob = Vector(o, b)
430 u = oa.cross(ob)
431 if not u.isnull():
432 u = u.asunit()
433 break
434 else:
435 raise ValueError('degenerate polygon')
436 angles = {a: 0.}
437 for m in points[1:]:
438 om = Vector(o, m)
439 normprod = norm_oa * om.norm()
440 cosinus = max(oa.dot(om) / normprod, -1.)
441 sinus = u.dot(oa.cross(om)) / normprod
442 angle = math.acos(cosinus)
443 angle = math.copysign(angle, sinus)
444 angles[m] = angle
445 return sorted(points, key=angles.get)
446
447 def faces(self):
448 vertices = self.vertices()
449 faces = []
450 for constraint in self.constraints:
451 face = []
452 for vertex in vertices:
453 if constraint.subs(vertex.coordinates()) == 0:
454 face.append(vertex)
455 faces.append(face)
456 return faces
457
458 def _plot_2d(self, plot=None, **kwargs):
459 import matplotlib.pyplot as plt
460 from matplotlib.patches import Polygon
461 for polyhedron in self.polyhedra:
462 vertices = polyhedron._sort_polygon_2d(polyhedron.vertices())
463 xys = [tuple(vertex.values()) for vertex in vertices]
464 if plot is None:
465 fig = plt.figure()
466 plot = fig.add_subplot(1, 1, 1)
467 xmin, xmax = plot.get_xlim()
468 ymin, ymax = plot.get_xlim()
469 xs, ys = zip(*xys)
470 xmin, xmax = min(xmin, float(min(xs))), max(xmax, float(max(xs)))
471 ymin, ymax = min(ymin, float(min(ys))), max(ymax, float(max(ys)))
472 plot.set_xlim(xmin, xmax)
473 plot.set_ylim(ymin, ymax)
474 plot.add_patch(Polygon(xys, closed=True, **kwargs))
475 return plot
476
477 def _plot_3d(self, plot=None, **kwargs):
478 from .polyhedra import Polyhedron
479 import matplotlib.pyplot as plt
480 from mpl_toolkits.mplot3d import Axes3D
481 from mpl_toolkits.mplot3d.art3d import Poly3DCollection
482 if plot is None:
483 fig = plt.figure()
484 axes = Axes3D(fig)
485 else:
486 axes = plot
487 xmin, xmax = axes.get_xlim()
488 ymin, ymax = axes.get_xlim()
489 zmin, zmax = axes.get_xlim()
490 poly_xyzs = []
491 for polyhedron in self.polyhedra:
492 for vertices in polyhedron.faces():
493 if len(vertices) == 0:
494 continue
495 vertices = Polyhedron._sort_polygon_3d(vertices)
496 vertices.append(vertices[0])
497 face_xyzs = [tuple(vertex.values()) for vertex in vertices]
498 xs, ys, zs = zip(*face_xyzs)
499 xmin, xmax = min(xmin, float(min(xs))), max(xmax, float(max(xs)))
500 ymin, ymax = min(ymin, float(min(ys))), max(ymax, float(max(ys)))
501 zmin, zmax = min(zmin, float(min(zs))), max(zmax, float(max(zs)))
502 poly_xyzs.append(face_xyzs)
503 collection = Poly3DCollection(poly_xyzs, **kwargs)
504 axes.add_collection3d(collection)
505 axes.set_xlim(xmin, xmax)
506 axes.set_ylim(ymin, ymax)
507 axes.set_zlim(zmin, zmax)
508 return axes
509
510 def plot(self, plot=None, **kwargs):
511 """
512 Display plot of this set.
513 """
514 if self.dimension == 2:
515 return self._plot_2d(plot=plot, **kwargs)
516 elif self.dimension == 3:
517 return self._plot_3d(plot=plot, **kwargs)
518 else:
519 raise ValueError('polyhedron must be 2 or 3-dimensional')
520
521 def __contains__(self, point):
522 for polyhedron in self.polyhedra:
523 if point in polyhedron:
524 return True
525 return False
526
527 def subs(self, symbol, expression=None):
528 polyhedra = [polyhedron.subs(symbol, expression)
529 for polyhedron in self.polyhedra]
530 return Domain(*polyhedra)
531
532 @classmethod
533 def _fromislset(cls, islset, symbols):
534 from .polyhedra import Polyhedron
535 islset = libisl.isl_set_remove_divs(islset)
536 islbsets = islhelper.isl_set_basic_sets(islset)
537 libisl.isl_set_free(islset)
538 polyhedra = []
539 for islbset in islbsets:
540 polyhedron = Polyhedron._fromislbasicset(islbset, symbols)
541 polyhedra.append(polyhedron)
542 if len(polyhedra) == 0:
543 from .polyhedra import Empty
544 return Empty
545 elif len(polyhedra) == 1:
546 return polyhedra[0]
547 else:
548 self = object().__new__(Domain)
549 self._polyhedra = tuple(polyhedra)
550 self._symbols = cls._xsymbols(polyhedra)
551 self._dimension = len(self._symbols)
552 return self
553
554 @classmethod
555 def _toislset(cls, polyhedra, symbols):
556 polyhedron = polyhedra[0]
557 islbset = polyhedron._toislbasicset(polyhedron.equalities,
558 polyhedron.inequalities, symbols)
559 islset1 = libisl.isl_set_from_basic_set(islbset)
560 for polyhedron in polyhedra[1:]:
561 islbset = polyhedron._toislbasicset(polyhedron.equalities,
562 polyhedron.inequalities, symbols)
563 islset2 = libisl.isl_set_from_basic_set(islbset)
564 islset1 = libisl.isl_set_union(islset1, islset2)
565 return islset1
566
567 @classmethod
568 def _fromast(cls, node):
569 from .polyhedra import Polyhedron
570 if isinstance(node, ast.Module) and len(node.body) == 1:
571 return cls._fromast(node.body[0])
572 elif isinstance(node, ast.Expr):
573 return cls._fromast(node.value)
574 elif isinstance(node, ast.UnaryOp):
575 domain = cls._fromast(node.operand)
576 if isinstance(node.operand, ast.invert):
577 return Not(domain)
578 elif isinstance(node, ast.BinOp):
579 domain1 = cls._fromast(node.left)
580 domain2 = cls._fromast(node.right)
581 if isinstance(node.op, ast.BitAnd):
582 return And(domain1, domain2)
583 elif isinstance(node.op, ast.BitOr):
584 return Or(domain1, domain2)
585 elif isinstance(node, ast.Compare):
586 equalities = []
587 inequalities = []
588 left = Expression._fromast(node.left)
589 for i in range(len(node.ops)):
590 op = node.ops[i]
591 right = Expression._fromast(node.comparators[i])
592 if isinstance(op, ast.Lt):
593 inequalities.append(right - left - 1)
594 elif isinstance(op, ast.LtE):
595 inequalities.append(right - left)
596 elif isinstance(op, ast.Eq):
597 equalities.append(left - right)
598 elif isinstance(op, ast.GtE):
599 inequalities.append(left - right)
600 elif isinstance(op, ast.Gt):
601 inequalities.append(left - right - 1)
602 else:
603 break
604 left = right
605 else:
606 return Polyhedron(equalities, inequalities)
607 raise SyntaxError('invalid syntax')
608
609 _RE_BRACES = re.compile(r'^\{\s*|\s*\}$')
610 _RE_EQ = re.compile(r'([^<=>])=([^<=>])')
611 _RE_AND = re.compile(r'\band\b|,|&&|/\\|∧|∩')
612 _RE_OR = re.compile(r'\bor\b|;|\|\||\\/|∨|∪')
613 _RE_NOT = re.compile(r'\bnot\b|!|¬')
614 _RE_NUM_VAR = Expression._RE_NUM_VAR
615 _RE_OPERATORS = re.compile(r'(&|\||~)')
616
617 @classmethod
618 def fromstring(cls, string):
619 # remove curly brackets
620 string = cls._RE_BRACES.sub(r'', string)
621 # replace '=' by '=='
622 string = cls._RE_EQ.sub(r'\1==\2', string)
623 # replace 'and', 'or', 'not'
624 string = cls._RE_AND.sub(r' & ', string)
625 string = cls._RE_OR.sub(r' | ', string)
626 string = cls._RE_NOT.sub(r' ~', string)
627 # add implicit multiplication operators, e.g. '5x' -> '5*x'
628 string = cls._RE_NUM_VAR.sub(r'\1*\2', string)
629 # add parentheses to force precedence
630 tokens = cls._RE_OPERATORS.split(string)
631 for i, token in enumerate(tokens):
632 if i % 2 == 0:
633 token = '({})'.format(token)
634 tokens[i] = token
635 string = ''.join(tokens)
636 tree = ast.parse(string, 'eval')
637 return cls._fromast(tree)
638
639 def __repr__(self):
640 assert len(self.polyhedra) >= 2
641 strings = [repr(polyhedron) for polyhedron in self.polyhedra]
642 return 'Or({})'.format(', '.join(strings))
643
644 def _repr_latex_(self):
645 strings = []
646 for polyhedron in self.polyhedra:
647 strings.append('({})'.format(polyhedron._repr_latex_().strip('$')))
648 return '${}$'.format(' \\vee '.join(strings))
649
650 @classmethod
651 def fromsympy(cls, expr):
652 import sympy
653 from .polyhedra import Lt, Le, Eq, Ne, Ge, Gt
654 funcmap = {
655 sympy.And: And, sympy.Or: Or, sympy.Not: Not,
656 sympy.Lt: Lt, sympy.Le: Le,
657 sympy.Eq: Eq, sympy.Ne: Ne,
658 sympy.Ge: Ge, sympy.Gt: Gt,
659 }
660 if expr.func in funcmap:
661 args = [Domain.fromsympy(arg) for arg in expr.args]
662 return funcmap[expr.func](*args)
663 elif isinstance(expr, sympy.Expr):
664 return Expression.fromsympy(expr)
665 raise ValueError('non-domain expression: {!r}'.format(expr))
666
667 def tosympy(self):
668 import sympy
669 polyhedra = [polyhedron.tosympy() for polyhedron in polyhedra]
670 return sympy.Or(*polyhedra)
671
672
673 def And(*domains):
674 """
675 Return the intersection of two sets as a new set.
676 """
677 if len(domains) == 0:
678 from .polyhedra import Universe
679 return Universe
680 else:
681 return domains[0].intersection(*domains[1:])
682
683 def Or(*domains):
684 """
685 Return the union of sets as a new set.
686 """
687 if len(domains) == 0:
688 from .polyhedra import Empty
689 return Empty
690 else:
691 return domains[0].union(*domains[1:])
692
693 def Not(domain):
694 """
695 Returns the complement of this set.
696 """
697 return ~domain