- def __init__(self, start, end, vertices):
- self.start = start
- self.end = end
- self.vertices = vertices
-
- def poly(self, body, filter):
- shape = pymunk.Poly(body, self.vertices)
- shape.filter = filter
- return shape
+ def __init__(self, position, vertices):
+ self.position = position # pointy end of the conical polygon
+ self.vertices = vertices # all vertices in the polygon
+
+ def _between(self, v, start, end):
+ if start < end:
+ return start <= v <= end
+ return (start <= v) or (v <= end)
+
+ def poly(self, start, end):
+ trial = pymunk.Poly(None, self.vertices)
+ trial.update(pymunk.Transform.identity())
+
+ if start is None or end is None:
+ return trial # no limits
+
+ start_info = trial.segment_query(
+ self.position + 1250 * start, self.position + 0.1 * start, 0)
+ end_info = trial.segment_query(
+ self.position + 1250 * end, self.position + 0.1 * end, 0)
+
+ vertices = self.vertices[:]
+ vertices = [
+ v for v in vertices
+ if self._between((v - self.position).angle, start.angle, end.angle)
+ ]
+ if start_info.shape is not None:
+ vertices.append(start_info.point)
+ if end_info.shape is not None:
+ vertices.append(end_info.point)
+ vertices.append(self.position)
+
+ poly = pymunk.Poly(None, vertices)
+ if len(poly.get_vertices()) < 3:
+ return None
+ return poly