Break up mapmodel.py monolith.

qt_ui/widgets/map/model/joinzonesjs.py

@@ -0,0 +1,82 @@
+from __future__ import annotations
+
+from PySide2.QtCore import Property, QObject, Signal
+
+from game import Game
+from game.ato import Flight
+from game.flightplan import JoinZoneGeometry
+from .leaflet import LeafletLatLon, LeafletPoly
+from .shapelyutil import ShapelyUtil
+
+
+class JoinZonesJs(QObject):
+    homeBubbleChanged = Signal()
+    targetBubbleChanged = Signal()
+    ipBubbleChanged = Signal()
+    excludedZonesChanged = Signal()
+    permissibleZonesChanged = Signal()
+    preferredLinesChanged = Signal()
+
+    def __init__(
+        self,
+        home_bubble: LeafletPoly,
+        target_bubble: LeafletPoly,
+        ip_bubble: LeafletPoly,
+        excluded_zones: list[LeafletPoly],
+        permissible_zones: list[LeafletPoly],
+        preferred_lines: list[list[LeafletLatLon]],
+    ) -> None:
+        super().__init__()
+        self._home_bubble = home_bubble
+        self._target_bubble = target_bubble
+        self._ip_bubble = ip_bubble
+        self._excluded_zones = excluded_zones
+        self._permissible_zones = permissible_zones
+        self._preferred_lines = preferred_lines
+
+    @Property(list, notify=homeBubbleChanged)
+    def homeBubble(self) -> LeafletPoly:
+        return self._home_bubble
+
+    @Property(list, notify=targetBubbleChanged)
+    def targetBubble(self) -> LeafletPoly:
+        return self._target_bubble
+
+    @Property(list, notify=ipBubbleChanged)
+    def ipBubble(self) -> LeafletPoly:
+        return self._ip_bubble
+
+    @Property(list, notify=excludedZonesChanged)
+    def excludedZones(self) -> list[LeafletPoly]:
+        return self._excluded_zones
+
+    @Property(list, notify=permissibleZonesChanged)
+    def permissibleZones(self) -> list[LeafletPoly]:
+        return self._permissible_zones
+
+    @Property(list, notify=preferredLinesChanged)
+    def preferredLines(self) -> list[list[LeafletLatLon]]:
+        return self._preferred_lines
+
+    @classmethod
+    def empty(cls) -> JoinZonesJs:
+        return JoinZonesJs([], [], [], [], [], [])
+
+    @classmethod
+    def for_flight(cls, flight: Flight, game: Game) -> JoinZonesJs:
+        if not ENABLE_EXPENSIVE_DEBUG_TOOLS:
+            return JoinZonesJs.empty()
+        target = flight.package.target
+        home = flight.departure
+        if flight.package.waypoints is None:
+            return JoinZonesJs.empty()
+        ip = flight.package.waypoints.ingress
+        geometry = JoinZoneGeometry(target.position, home.position, ip, game.blue)
+        return JoinZonesJs(
+            ShapelyUtil.poly_to_leaflet(geometry.home_bubble, game.theater),
+            ShapelyUtil.poly_to_leaflet(geometry.target_bubble, game.theater),
+            ShapelyUtil.poly_to_leaflet(geometry.ip_bubble, game.theater),
+            ShapelyUtil.polys_to_leaflet(geometry.excluded_zones, game.theater),
+            ShapelyUtil.polys_to_leaflet(geometry.permissible_zones, game.theater),
+            ShapelyUtil.lines_to_leaflet(geometry.preferred_lines, game.theater),
+        )