mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
from __future__ import annotations
|
|
|
|
from copy import deepcopy
|
|
from dataclasses import dataclass, field
|
|
from datetime import datetime
|
|
from typing import Literal, TYPE_CHECKING, Any, Dict, Optional
|
|
|
|
from dcs import Point
|
|
|
|
from game.ato.flightwaypointtype import FlightWaypointType
|
|
from game.theater.theatergroup import TheaterUnit
|
|
from game.utils import Distance, meters
|
|
|
|
if TYPE_CHECKING:
|
|
from game.theater import ControlPoint
|
|
|
|
AltitudeReference = Literal["BARO", "RADIO"]
|
|
|
|
|
|
@dataclass
|
|
class FlightWaypoint:
|
|
name: str
|
|
waypoint_type: FlightWaypointType
|
|
position: Point
|
|
alt: Distance = meters(0)
|
|
alt_type: AltitudeReference = "BARO"
|
|
control_point: ControlPoint | None = None
|
|
|
|
# TODO: Merge with pretty_name.
|
|
# Only used in the waypoint list in the flight edit page. No sense
|
|
# having three names. A short and long form is enough.
|
|
description: str = ""
|
|
|
|
targets: list[TheaterUnit] = field(default_factory=list)
|
|
obj_name: str = ""
|
|
pretty_name: str = ""
|
|
only_for_player: bool = False
|
|
flyover: bool = False
|
|
|
|
# The minimum amount of fuel remaining at this waypoint in pounds.
|
|
min_fuel: float | None = None
|
|
|
|
# These are set very late by the air conflict generator (part of mission
|
|
# generation). We do it late so that we don't need to propagate changes
|
|
# to waypoint times whenever the player alters the package TOT or the
|
|
# flight's offset in the UI.
|
|
tot: datetime | None = None
|
|
departure_time: datetime | None = None
|
|
|
|
@property
|
|
def x(self) -> float:
|
|
return self.position.x
|
|
|
|
@property
|
|
def y(self) -> float:
|
|
return self.position.y
|
|
|
|
def __hash__(self) -> int:
|
|
return hash(id(self))
|
|
|
|
def __deepcopy__(self, memo: Optional[Dict[int, Any]] = None) -> FlightWaypoint:
|
|
obj = FlightWaypoint(self.name, self.waypoint_type, self.position)
|
|
for attr in dir(self):
|
|
if attr == "control_point":
|
|
obj.control_point = self.control_point
|
|
elif attr == "targets":
|
|
obj.targets = self.targets
|
|
elif "__" in attr or attr not in obj.__dataclass_fields__:
|
|
continue
|
|
else:
|
|
attr_copy = deepcopy(getattr(self, attr))
|
|
setattr(obj, attr, attr_copy)
|
|
return obj
|