mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
2e901f3586
These aren't as ergonomic as I'd hoped because of https://www.python.org/dev/peps/pep-0647/#narrowing-of-implicit-self-and-cls-parameters. I added a decorator `@self_type_guard` so we can avoid needing to import the descendent types in the typeguard implementation (which wouldn't fix any circular imports, just move them).
47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import timedelta
|
|
from typing import TYPE_CHECKING
|
|
|
|
from dcs import Point
|
|
|
|
from game.ato.flightstate import FlightState, InFlight
|
|
from game.ato.flightstate.navigating import Navigating
|
|
from game.utils import Distance, Speed
|
|
|
|
if TYPE_CHECKING:
|
|
from game.ato.flight import Flight
|
|
from game.settings import Settings
|
|
|
|
|
|
class Loiter(InFlight):
|
|
def __init__(self, flight: Flight, settings: Settings, waypoint_index: int) -> None:
|
|
assert flight.flight_plan.is_loiter(flight.flight_plan)
|
|
self.hold_duration = flight.flight_plan.hold_duration
|
|
super().__init__(flight, settings, waypoint_index)
|
|
|
|
def estimate_position(self) -> Point:
|
|
return self.current_waypoint.position
|
|
|
|
def estimate_altitude(self) -> tuple[Distance, str]:
|
|
return self.current_waypoint.alt, self.current_waypoint.alt_type
|
|
|
|
def estimate_speed(self) -> Speed:
|
|
return self.flight.unit_type.preferred_patrol_speed(self.estimate_altitude()[0])
|
|
|
|
def estimate_fuel(self) -> float:
|
|
# TODO: Estimate loiter consumption per minute?
|
|
return self.estimate_fuel_at_current_waypoint()
|
|
|
|
def next_waypoint_state(self) -> FlightState:
|
|
# Do not automatically advance to the next waypoint. Just proceed from the
|
|
# current one with the normal flying state.
|
|
return Navigating(self.flight, self.settings, self.waypoint_index)
|
|
|
|
def travel_time_between_waypoints(self) -> timedelta:
|
|
return self.hold_duration
|
|
|
|
@property
|
|
def description(self) -> str:
|
|
return f"Loitering for {self.hold_duration - self.elapsed_time}"
|