mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
769fe12159
During package planning we don't care about the details of the flight plan, just the layout (to check if the layout is threatened and we need escorts). Splitting these will allow us to reduce the amount of work that must be done in each loop of the planning phase, potentially caching attempted flight plans between loops.
91 lines
2.8 KiB
Python
91 lines
2.8 KiB
Python
from __future__ import annotations
|
|
|
|
from collections.abc import Iterator
|
|
from dataclasses import dataclass
|
|
from datetime import timedelta
|
|
from typing import TYPE_CHECKING, Type
|
|
|
|
from game.utils import feet
|
|
from .ibuilder import IBuilder
|
|
from .standard import StandardFlightPlan, StandardLayout
|
|
from .waypointbuilder import WaypointBuilder
|
|
from ..flightstate import InFlight
|
|
|
|
if TYPE_CHECKING:
|
|
from ..flightwaypoint import FlightWaypoint
|
|
|
|
|
|
class Builder(IBuilder):
|
|
def build(self) -> RtbLayout:
|
|
if not isinstance(self.flight.state, InFlight):
|
|
raise RuntimeError(f"Cannot abort {self} because it is not in flight")
|
|
|
|
current_position = self.flight.state.estimate_position()
|
|
current_altitude, altitude_reference = self.flight.state.estimate_altitude()
|
|
|
|
altitude_is_agl = self.flight.unit_type.dcs_unit_type.helicopter
|
|
altitude = (
|
|
feet(1500)
|
|
if altitude_is_agl
|
|
else self.flight.unit_type.preferred_patrol_altitude
|
|
)
|
|
builder = WaypointBuilder(self.flight, self.flight.coalition)
|
|
abort_point = builder.nav(
|
|
current_position, current_altitude, altitude_reference == "RADIO"
|
|
)
|
|
abort_point.name = "ABORT AND RTB"
|
|
abort_point.pretty_name = "Abort and RTB"
|
|
abort_point.description = "Abort mission and return to base"
|
|
return RtbLayout(
|
|
departure=builder.takeoff(self.flight.departure),
|
|
abort_location=abort_point,
|
|
nav_to_destination=builder.nav_path(
|
|
current_position,
|
|
self.flight.arrival.position,
|
|
altitude,
|
|
altitude_is_agl,
|
|
),
|
|
arrival=builder.land(self.flight.arrival),
|
|
divert=builder.divert(self.flight.divert),
|
|
bullseye=builder.bullseye(),
|
|
)
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class RtbLayout(StandardLayout):
|
|
abort_location: FlightWaypoint
|
|
nav_to_destination: list[FlightWaypoint]
|
|
|
|
def iter_waypoints(self) -> Iterator[FlightWaypoint]:
|
|
yield self.departure
|
|
yield self.abort_location
|
|
yield from self.nav_to_destination
|
|
yield self.arrival
|
|
if self.divert is not None:
|
|
yield self.divert
|
|
yield self.bullseye
|
|
|
|
|
|
class RtbFlightPlan(StandardFlightPlan[RtbLayout]):
|
|
@staticmethod
|
|
def builder_type() -> Type[Builder]:
|
|
return Builder
|
|
|
|
@property
|
|
def abort_index(self) -> int:
|
|
return 1
|
|
|
|
@property
|
|
def tot_waypoint(self) -> FlightWaypoint | None:
|
|
return None
|
|
|
|
def tot_for_waypoint(self, waypoint: FlightWaypoint) -> timedelta | None:
|
|
return None
|
|
|
|
def depart_time_for_waypoint(self, waypoint: FlightWaypoint) -> timedelta | None:
|
|
return None
|
|
|
|
@property
|
|
def mission_departure_time(self) -> timedelta:
|
|
return timedelta()
|