mirror of
https://github.com/dcs-liberation/dcs_liberation.git
synced 2025-11-10 14:22:26 +00:00
fa8c0d9660
Split the oversized file into one per plan type. This also moves the layout responsibility out of the oversized FlightPlanBuilder and into each flight plan type file.
100 lines
3.1 KiB
Python
100 lines
3.1 KiB
Python
from __future__ import annotations
|
|
|
|
from collections.abc import Iterator
|
|
from datetime import timedelta
|
|
from typing import TYPE_CHECKING, Type
|
|
|
|
from game.utils import feet
|
|
from .ibuilder import IBuilder
|
|
from .standard import StandardFlightPlan
|
|
from .waypointbuilder import WaypointBuilder
|
|
from ..flightstate import InFlight
|
|
|
|
if TYPE_CHECKING:
|
|
from ..flight import Flight
|
|
from ..flightwaypoint import FlightWaypoint
|
|
|
|
|
|
class Builder(IBuilder):
|
|
def build(self) -> RtbFlightPlan:
|
|
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 RtbFlightPlan(
|
|
flight=self.flight,
|
|
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(),
|
|
)
|
|
|
|
|
|
class RtbFlightPlan(StandardFlightPlan):
|
|
def __init__(
|
|
self,
|
|
flight: Flight,
|
|
departure: FlightWaypoint,
|
|
arrival: FlightWaypoint,
|
|
divert: FlightWaypoint | None,
|
|
bullseye: FlightWaypoint,
|
|
abort_location: FlightWaypoint,
|
|
nav_to_destination: list[FlightWaypoint],
|
|
) -> None:
|
|
super().__init__(flight, departure, arrival, divert, bullseye)
|
|
self.abort_location = abort_location
|
|
self.nav_to_destination = nav_to_destination
|
|
|
|
@staticmethod
|
|
def builder_type() -> Type[Builder]:
|
|
return Builder
|
|
|
|
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
|
|
|
|
@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()
|