mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
48938fc529
Squashing 8 commits by DanAlbert: - Track theater in ControlPoint. Simplifies finding the owning theater of a control point. Not used yet. - Clean some cruft out of FlightPlanBuilder. - Clean up silly some exception handling. - Move FlightPlan instantiation into the builder. I'm working on moving the builder to be owned by the Flight, which will simplify callers that need to create (or recreate) flight plans for a flight. - Simplify IBuilder constructor. We have access to the theater via the flight's departure airbase now. - Move FlightPlan creation into Flight. For now this is just a callsite cleanup. Later, this will make it easier to separate unscheduled and scheduled flights into different classes without complicating the layout/scheduling. - Remove superfluous constructors. - Remove unused Package field.
94 lines
2.9 KiB
Python
94 lines
2.9 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
|
|
|
|
|
|
@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()
|
|
|
|
|
|
class Builder(IBuilder[RtbFlightPlan, RtbLayout]):
|
|
def layout(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(),
|
|
)
|
|
|
|
def build(self) -> RtbFlightPlan:
|
|
return RtbFlightPlan(self.flight, self.layout())
|