mirror of
https://github.com/dcs-liberation/dcs_liberation.git
synced 2025-11-10 14:22:26 +00:00
Flights without a meaningful TOT make the code around startup time (and other scheduling behaviors) unnecessarily complicated because they have to handle unpredictable flight plans. We can simplify this by requiring that all flight plans have a waypoint associated with their TOT. For custom flight plans, we can just fall back to the takeoff waypoint. For RTB flight plans (which are only synthetic flight plans injected for aborted flights), we can use the abort point. This also means that all flight plans now have, at the very least, a departure waypoint. Deleting this waypoint is invalid even for custom flights, so that's no a problem.
148 lines
5.1 KiB
Python
148 lines
5.1 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.theater.missiontarget import MissionTarget
|
|
from game.utils import feet
|
|
from .ibuilder import IBuilder
|
|
from .planningerror import PlanningError
|
|
from .standard import StandardFlightPlan, StandardLayout
|
|
from .waypointbuilder import WaypointBuilder
|
|
|
|
if TYPE_CHECKING:
|
|
from ..flightwaypoint import FlightWaypoint
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class AirliftLayout(StandardLayout):
|
|
nav_to_pickup: list[FlightWaypoint]
|
|
pickup: FlightWaypoint | None
|
|
nav_to_drop_off: list[FlightWaypoint]
|
|
drop_off: FlightWaypoint
|
|
stopover: FlightWaypoint | None
|
|
nav_to_home: list[FlightWaypoint]
|
|
|
|
def iter_waypoints(self) -> Iterator[FlightWaypoint]:
|
|
yield self.departure
|
|
yield from self.nav_to_pickup
|
|
if self.pickup is not None:
|
|
yield self.pickup
|
|
yield from self.nav_to_drop_off
|
|
yield self.drop_off
|
|
if self.stopover is not None:
|
|
yield self.stopover
|
|
yield from self.nav_to_home
|
|
yield self.arrival
|
|
if self.divert is not None:
|
|
yield self.divert
|
|
yield self.bullseye
|
|
|
|
|
|
class AirliftFlightPlan(StandardFlightPlan[AirliftLayout]):
|
|
@staticmethod
|
|
def builder_type() -> Type[Builder]:
|
|
return Builder
|
|
|
|
@property
|
|
def tot_waypoint(self) -> FlightWaypoint:
|
|
return self.layout.drop_off
|
|
|
|
def tot_for_waypoint(self, waypoint: FlightWaypoint) -> timedelta | None:
|
|
# TOT planning isn't really useful for transports. They're behind the front
|
|
# lines so no need to wait for escorts or for other missions to complete.
|
|
return None
|
|
|
|
def depart_time_for_waypoint(self, waypoint: FlightWaypoint) -> timedelta | None:
|
|
return None
|
|
|
|
@property
|
|
def mission_departure_time(self) -> timedelta:
|
|
return self.package.time_over_target
|
|
|
|
|
|
class Builder(IBuilder[AirliftFlightPlan, AirliftLayout]):
|
|
def layout(self) -> AirliftLayout:
|
|
cargo = self.flight.cargo
|
|
if cargo is None:
|
|
raise PlanningError(
|
|
"Cannot plan transport mission for flight with no cargo."
|
|
)
|
|
|
|
altitude = feet(1500)
|
|
altitude_is_agl = True
|
|
|
|
builder = WaypointBuilder(self.flight, self.coalition)
|
|
|
|
pickup = None
|
|
stopover = None
|
|
if self.flight.is_helo:
|
|
# Create a pickupzone where the cargo will be spawned
|
|
pickup_zone = MissionTarget(
|
|
"Pickup Zone", cargo.origin.position.random_point_within(1000, 200)
|
|
)
|
|
pickup = builder.pickup(pickup_zone)
|
|
# If The cargo is at the departure controlpoint, the pickup waypoint should
|
|
# only be created for client flights
|
|
pickup.only_for_player = cargo.origin == self.flight.departure
|
|
|
|
# Create a dropoff zone where the cargo should be dropped
|
|
drop_off_zone = MissionTarget(
|
|
"Dropoff zone",
|
|
cargo.next_stop.position.random_point_within(1000, 200),
|
|
)
|
|
drop_off = builder.drop_off(drop_off_zone)
|
|
|
|
# Add an additional stopover point so that the flight can refuel
|
|
stopover = builder.stopover(cargo.next_stop)
|
|
else:
|
|
# Fixed Wing will get stopover points for pickup and dropoff
|
|
if cargo.origin != self.flight.departure:
|
|
pickup = builder.stopover(cargo.origin, "PICKUP")
|
|
drop_off = builder.stopover(cargo.next_stop, "DROP OFF")
|
|
|
|
nav_to_pickup = builder.nav_path(
|
|
self.flight.departure.position,
|
|
cargo.origin.position,
|
|
altitude,
|
|
altitude_is_agl,
|
|
)
|
|
|
|
if self.flight.client_count > 0:
|
|
# Normal Landing Waypoint
|
|
arrival = builder.land(self.flight.arrival)
|
|
else:
|
|
# The AI Needs another Stopover point to actually fly back to the original
|
|
# base. Otherwise the Cargo drop will be the new Landing Waypoint and the
|
|
# AI will end its mission there instead of flying back.
|
|
# https://forum.dcs.world/topic/211775-landing-to-refuel-and-rearm-the-landingrefuar-waypoint/
|
|
arrival = builder.stopover(self.flight.arrival, "LANDING")
|
|
|
|
return AirliftLayout(
|
|
departure=builder.takeoff(self.flight.departure),
|
|
nav_to_pickup=nav_to_pickup,
|
|
pickup=pickup,
|
|
nav_to_drop_off=builder.nav_path(
|
|
cargo.origin.position,
|
|
cargo.next_stop.position,
|
|
altitude,
|
|
altitude_is_agl,
|
|
),
|
|
drop_off=drop_off,
|
|
stopover=stopover,
|
|
nav_to_home=builder.nav_path(
|
|
cargo.origin.position,
|
|
self.flight.arrival.position,
|
|
altitude,
|
|
altitude_is_agl,
|
|
),
|
|
arrival=arrival,
|
|
divert=builder.divert(self.flight.divert),
|
|
bullseye=builder.bullseye(),
|
|
)
|
|
|
|
def build(self) -> AirliftFlightPlan:
|
|
return AirliftFlightPlan(self.flight, self.layout())
|