mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
from __future__ import annotations
|
|
|
|
from abc import ABC, abstractmethod
|
|
from dataclasses import dataclass
|
|
from datetime import datetime, timedelta
|
|
from typing import Any, TYPE_CHECKING, TypeGuard
|
|
from typing import Optional
|
|
|
|
from game.typeguard import self_type_guard
|
|
from .flightplan import FlightPlan
|
|
from .standard import StandardFlightPlan, StandardLayout
|
|
|
|
if TYPE_CHECKING:
|
|
from ..flightwaypoint import FlightWaypoint
|
|
|
|
|
|
@dataclass
|
|
class LoiterLayout(StandardLayout, ABC):
|
|
hold: Optional[FlightWaypoint]
|
|
|
|
|
|
class LoiterFlightPlan(StandardFlightPlan[Any], ABC):
|
|
@property
|
|
def hold_duration(self) -> timedelta:
|
|
return timedelta(minutes=5)
|
|
|
|
@property
|
|
@abstractmethod
|
|
def push_time(self) -> datetime:
|
|
...
|
|
|
|
def depart_time_for_waypoint(self, waypoint: FlightWaypoint) -> datetime | None:
|
|
if waypoint == self.layout.hold:
|
|
return self.push_time
|
|
return None
|
|
|
|
def total_time_between_waypoints(
|
|
self, a: FlightWaypoint, b: FlightWaypoint
|
|
) -> timedelta:
|
|
travel_time = super().total_time_between_waypoints(a, b)
|
|
if a != self.layout.hold:
|
|
return travel_time
|
|
return travel_time + self.hold_duration
|
|
|
|
@self_type_guard
|
|
def is_loiter(self, flight_plan: FlightPlan[Any]) -> TypeGuard[LoiterFlightPlan]:
|
|
return True
|