mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
A possible explanation for the infrequent CTDs we've been seeing since adding fast forward is that QWebChannel doesn't keep a reference to the python objects that it passes to js, so if the object is GC'd before the front end is done with it, it crashes. We don't really like QWebChannel anyway, so this begins replacing that with FastAPI.
127 lines
4.8 KiB
Python
127 lines
4.8 KiB
Python
from __future__ import annotations
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
from game import Game
|
|
from game.ato import Flight
|
|
from game.flightplan import HoldZoneGeometry, IpZoneGeometry, JoinZoneGeometry
|
|
from ..leaflet import LeafletPoly, ShapelyUtil
|
|
|
|
|
|
class HoldZonesJs(BaseModel):
|
|
home_bubble: LeafletPoly = Field(alias="homeBubble")
|
|
target_bubble: LeafletPoly = Field(alias="targetBubble")
|
|
join_bubble: LeafletPoly = Field(alias="joinBubble")
|
|
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) -> HoldZonesJs:
|
|
return HoldZonesJs(
|
|
homeBubble=[],
|
|
targetBubble=[],
|
|
joinBubble=[],
|
|
excludedZones=[],
|
|
permissibleZones=[],
|
|
preferredLines=[],
|
|
)
|
|
|
|
@classmethod
|
|
def for_flight(cls, flight: Flight, game: Game) -> HoldZonesJs:
|
|
target = flight.package.target
|
|
home = flight.departure
|
|
if flight.package.waypoints is None:
|
|
return HoldZonesJs.empty()
|
|
ip = flight.package.waypoints.ingress
|
|
join = flight.package.waypoints.join
|
|
geometry = HoldZoneGeometry(
|
|
target.position, home.position, ip, join, game.blue, game.theater
|
|
)
|
|
return HoldZonesJs(
|
|
homeBubble=ShapelyUtil.poly_to_leaflet(geometry.home_bubble, game.theater),
|
|
targetBubble=ShapelyUtil.poly_to_leaflet(
|
|
geometry.target_bubble, game.theater
|
|
),
|
|
joinBubble=ShapelyUtil.poly_to_leaflet(geometry.join_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
|
|
),
|
|
)
|
|
|
|
|
|
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:
|
|
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),
|
|
)
|
|
|
|
|
|
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:
|
|
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
|
|
),
|
|
)
|