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.
38 lines
1.4 KiB
Python
38 lines
1.4 KiB
Python
from __future__ import annotations
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
from game import Game
|
|
from game.ato import Flight
|
|
from game.flightplan import IpZoneGeometry
|
|
from .config import ENABLE_EXPENSIVE_DEBUG_TOOLS
|
|
from .leaflet import LeafletPoly
|
|
from .shapelyutil import ShapelyUtil
|
|
|
|
|
|
class IpZonesJs(BaseModel):
|
|
home_bubble: LeafletPoly = Field(alias="homeBubble")
|
|
ipBubble: LeafletPoly = Field(alias="ipBubble")
|
|
permissibleZone: LeafletPoly = Field(alias="permissibleZone")
|
|
safeZones: list[LeafletPoly] = Field(alias="safeZones")
|
|
|
|
@classmethod
|
|
def empty(cls) -> IpZonesJs:
|
|
return IpZonesJs(homeBubble=[], ipBubble=[], permissibleZone=[], safeZones=[])
|
|
|
|
@classmethod
|
|
def for_flight(cls, flight: Flight, game: Game) -> IpZonesJs:
|
|
if not ENABLE_EXPENSIVE_DEBUG_TOOLS:
|
|
return IpZonesJs.empty()
|
|
target = flight.package.target
|
|
home = flight.departure
|
|
geometry = IpZoneGeometry(target.position, home.position, game.blue)
|
|
return IpZonesJs(
|
|
homeBubble=ShapelyUtil.poly_to_leaflet(geometry.home_bubble, game.theater),
|
|
ipBubble=ShapelyUtil.poly_to_leaflet(geometry.ip_bubble, game.theater),
|
|
permissibleZone=ShapelyUtil.poly_to_leaflet(
|
|
geometry.permissible_zone, game.theater
|
|
),
|
|
safeZones=ShapelyUtil.polys_to_leaflet(geometry.safe_zones, game.theater),
|
|
)
|