mirror of
https://github.com/dcs-liberation/dcs_liberation.git
synced 2025-11-10 14:22:26 +00:00
4521053804
I'm working on moving the builder to be owned by the Flight, which will simplify callers that need to create (or recreate) flight plans for a flight.
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Type
|
|
|
|
from game.theater import TheaterGroundObject
|
|
from .formationattack import (
|
|
FormationAttackBuilder,
|
|
FormationAttackFlightPlan,
|
|
FormationAttackLayout,
|
|
)
|
|
from .invalidobjectivelocation import InvalidObjectiveLocation
|
|
from .waypointbuilder import StrikeTarget
|
|
from ..flightwaypointtype import FlightWaypointType
|
|
|
|
|
|
class StrikeFlightPlan(FormationAttackFlightPlan):
|
|
@staticmethod
|
|
def builder_type() -> Type[Builder]:
|
|
return Builder
|
|
|
|
|
|
class Builder(FormationAttackBuilder[StrikeFlightPlan, FormationAttackLayout]):
|
|
def layout(self) -> FormationAttackLayout:
|
|
location = self.package.target
|
|
|
|
if not isinstance(location, TheaterGroundObject):
|
|
raise InvalidObjectiveLocation(self.flight.flight_type, location)
|
|
|
|
targets: list[StrikeTarget] = []
|
|
for idx, unit in enumerate(location.strike_targets):
|
|
targets.append(StrikeTarget(f"{unit.type.id} #{idx}", unit))
|
|
|
|
return self._build(FlightWaypointType.INGRESS_STRIKE, targets)
|
|
|
|
def build(self) -> StrikeFlightPlan:
|
|
return StrikeFlightPlan(self.flight, self.layout())
|