mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
6ebda41922
This requires less duplication and paves the way for us replacing QWebChannel with FastAPI, which in turn gets us closer to not needing Qt.
59 lines
2.1 KiB
Python
59 lines
2.1 KiB
Python
from __future__ import annotations
|
|
|
|
from pydantic import Field
|
|
from pydantic.main import BaseModel
|
|
|
|
from game import Game
|
|
from game.ato import Flight
|
|
from game.flightplan import JoinZoneGeometry
|
|
from .config import ENABLE_EXPENSIVE_DEBUG_TOOLS
|
|
from .leaflet import LeafletPoly
|
|
from .shapelyutil import ShapelyUtil
|
|
|
|
|
|
class JoinZonesJs(BaseModel):
|
|
home_bubble: LeafletPoly = Field(alias="homeBubble")
|
|
target_bubble: LeafletPoly = Field(alias="targetBubble")
|
|
ip_bubble: LeafletPoly = Field(alias="ipBubble")
|
|
excluded_zones: list[LeafletPoly] = Field(alias="excludedZones")
|
|
permissible_zones: list[LeafletPoly] = Field(alias="permissibleZones")
|
|
preferred_lines: list[LeafletPoly] = Field(alias="preferredLines")
|
|
|
|
@classmethod
|
|
def empty(cls) -> JoinZonesJs:
|
|
return JoinZonesJs(
|
|
homeBubble=[],
|
|
targetBubble=[],
|
|
ipBubble=[],
|
|
excludedZones=[],
|
|
permissibleZones=[],
|
|
preferredLines=[],
|
|
)
|
|
|
|
@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(
|
|
homeBubble=ShapelyUtil.poly_to_leaflet(geometry.home_bubble, game.theater),
|
|
targetBubble=ShapelyUtil.poly_to_leaflet(
|
|
geometry.target_bubble, game.theater
|
|
),
|
|
ipBubble=ShapelyUtil.poly_to_leaflet(geometry.ip_bubble, game.theater),
|
|
excludedZones=ShapelyUtil.polys_to_leaflet(
|
|
geometry.excluded_zones, game.theater
|
|
),
|
|
permissibleZones=ShapelyUtil.polys_to_leaflet(
|
|
geometry.permissible_zones, game.theater
|
|
),
|
|
preferredLines=ShapelyUtil.lines_to_leaflet(
|
|
geometry.preferred_lines, game.theater
|
|
),
|
|
)
|