Dan Albert fa8c0d9660 Clean up flight plan code.
Split the oversized file into one per plan type. This also moves the
layout responsibility out of the oversized FlightPlanBuilder and into
each flight plan type file.
2022-03-09 02:15:07 -08:00

30 lines
1.0 KiB
Python

from __future__ import annotations
from typing import Type
from game.theater import TheaterGroundObject
from .formationattack import FormationAttackBuilder, FormationAttackFlightPlan
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]):
def build(self) -> FormationAttackFlightPlan:
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(StrikeFlightPlan, FlightWaypointType.INGRESS_STRIKE, targets)