mirror of
https://github.com/dcs-liberation/dcs_liberation.git
synced 2025-11-10 14:22:26 +00:00
87441b8939
Create a WaypointAction class that defines the actions taken at a waypoint. These will often map one-to-one with DCS waypoint actions but can also be higher level and generate multiple actions. Once everything has migrated all waypoint-type-specific behaviors of PydcsWaypointBuilder will be gone, and it'll be easier to keep the sim behaviors in sync with the mission generator behaviors. For now only hold has been migrated. This is actually probably the most complicated action we have (starting with this may have been a mistake, but it did find all the rough edges quickly) since it affects waypoint timings and flight position during simulation. That part isn't handled as neatly as I'd like because the FlightState still has to special case LOITER points to avoid simulating the wrong waypoint position. At some point we should probably start tracking real positions in FlightState, and when we do that will be solved.
85 lines
2.7 KiB
Python
85 lines
2.7 KiB
Python
from __future__ import annotations
|
|
|
|
from abc import ABC, abstractmethod
|
|
from typing import Any, Generic, TYPE_CHECKING, TypeVar
|
|
|
|
from game.navmesh import NavMeshError
|
|
from .flightplan import FlightPlan, Layout
|
|
from .planningerror import PlanningError
|
|
from ..packagewaypoints import PackageWaypoints
|
|
|
|
if TYPE_CHECKING:
|
|
from game.coalition import Coalition
|
|
from game.data.doctrine import Doctrine
|
|
from game.theater import ConflictTheater
|
|
from game.threatzones import ThreatZones
|
|
from ..flight import Flight
|
|
from ..package import Package
|
|
|
|
|
|
FlightPlanT = TypeVar("FlightPlanT", bound=FlightPlan[Any])
|
|
LayoutT = TypeVar("LayoutT", bound=Layout)
|
|
|
|
|
|
class IBuilder(ABC, Generic[FlightPlanT, LayoutT]):
|
|
def __init__(self, flight: Flight) -> None:
|
|
self.flight = flight
|
|
self._flight_plan: FlightPlanT | None = None
|
|
|
|
def get_or_build(self) -> FlightPlanT:
|
|
if self._flight_plan is None:
|
|
self.regenerate()
|
|
assert self._flight_plan is not None
|
|
return self._flight_plan
|
|
|
|
def regenerate(self, dump_debug_info: bool = False) -> None:
|
|
try:
|
|
self._generate_package_waypoints_if_needed(dump_debug_info)
|
|
self._flight_plan = self.build(dump_debug_info)
|
|
self._flight_plan.add_waypoint_actions()
|
|
except NavMeshError as ex:
|
|
color = "blue" if self.flight.squadron.player else "red"
|
|
raise PlanningError(
|
|
f"Could not plan {color} {self.flight.flight_type.value} from "
|
|
f"{self.flight.departure} to {self.package.target}"
|
|
) from ex
|
|
|
|
def _generate_package_waypoints_if_needed(self, dump_debug_info: bool) -> None:
|
|
# Package waypoints are only valid for offensive missions. Skip this if the
|
|
# target is friendly.
|
|
if self.package.target.is_friendly(self.is_player):
|
|
return
|
|
|
|
if self.package.waypoints is None or dump_debug_info:
|
|
self.package.waypoints = PackageWaypoints.create(
|
|
self.package, self.coalition, dump_debug_info
|
|
)
|
|
|
|
@property
|
|
def theater(self) -> ConflictTheater:
|
|
return self.flight.departure.theater
|
|
|
|
@abstractmethod
|
|
def build(self, dump_debug_info: bool = False) -> FlightPlanT:
|
|
...
|
|
|
|
@property
|
|
def package(self) -> Package:
|
|
return self.flight.package
|
|
|
|
@property
|
|
def coalition(self) -> Coalition:
|
|
return self.flight.coalition
|
|
|
|
@property
|
|
def is_player(self) -> bool:
|
|
return self.coalition.player
|
|
|
|
@property
|
|
def doctrine(self) -> Doctrine:
|
|
return self.coalition.doctrine
|
|
|
|
@property
|
|
def threat_zones(self) -> ThreatZones:
|
|
return self.coalition.opponent.threat_zone
|