diff --git a/game/ato/flightstate/flightstate.py b/game/ato/flightstate/flightstate.py index 73ad7624..ad8d8649 100644 --- a/game/ato/flightstate/flightstate.py +++ b/game/ato/flightstate/flightstate.py @@ -21,6 +21,39 @@ class FlightState(ABC): self.settings = settings self.avoid_further_combat = False + def reinitialize(self, now: datetime) -> None: + from game.ato.flightstate import WaitingForStart + + start_time = self.flight.flight_plan.startup_time() + if start_time <= now: + self._set_active_flight_state(now) + else: + self.flight.set_state( + WaitingForStart(self.flight, self.settings, start_time) + ) + + def _set_active_flight_state(self, now: datetime) -> None: + from game.ato.flightstate import StartUp + from game.ato.flightstate import Taxi + from game.ato.flightstate import Takeoff + from game.ato.flightstate import Navigating + + match self.flight.start_type: + case StartType.COLD: + self.flight.set_state(StartUp(self.flight, self.settings, now)) + case StartType.WARM: + self.flight.set_state(Taxi(self.flight, self.settings, now)) + case StartType.RUNWAY: + self.flight.set_state(Takeoff(self.flight, self.settings, now)) + case StartType.IN_FLIGHT: + self.flight.set_state( + Navigating(self.flight, self.settings, waypoint_index=0) + ) + case _: + raise ValueError( + f"Unknown start type {self.flight.start_type} for {self.flight}" + ) + @property def alive(self) -> bool: return True diff --git a/game/ato/flightstate/uninitialized.py b/game/ato/flightstate/uninitialized.py index d19792f7..4b76f75f 100644 --- a/game/ato/flightstate/uninitialized.py +++ b/game/ato/flightstate/uninitialized.py @@ -20,7 +20,8 @@ class Uninitialized(FlightState): def on_game_tick( self, events: GameUpdateEvents, time: datetime, duration: timedelta ) -> None: - raise RuntimeError("Attempted to simulate flight that is not fully initialized") + self.reinitialize(time) + self.flight.state.on_game_tick(events, time, duration) @property def is_waiting_for_start(self) -> bool: diff --git a/game/sim/aircraftsimulation.py b/game/sim/aircraftsimulation.py index f56dce3f..0c300185 100644 --- a/game/sim/aircraftsimulation.py +++ b/game/sim/aircraftsimulation.py @@ -8,14 +8,8 @@ from datetime import datetime, timedelta from typing_extensions import TYPE_CHECKING from game.ato.flightstate import ( - Navigating, - StartUp, - Takeoff, - Taxi, Uninitialized, - WaitingForStart, ) -from game.ato.starttype import StartType from .combat import CombatInitiator, FrozenCombat from .simulationresults import SimulationResults @@ -73,25 +67,7 @@ class AircraftSimulation: def set_initial_flight_states(self) -> None: now = self.game.conditions.start_time for flight in self.iter_flights(): - start_time = flight.flight_plan.startup_time() - if start_time <= now: - self.set_active_flight_state(flight, now) - else: - flight.set_state( - WaitingForStart(flight, self.game.settings, start_time) - ) - - def set_active_flight_state(self, flight: Flight, now: datetime) -> None: - if flight.start_type is StartType.COLD: - flight.set_state(StartUp(flight, self.game.settings, now)) - elif flight.start_type is StartType.WARM: - flight.set_state(Taxi(flight, self.game.settings, now)) - elif flight.start_type is StartType.RUNWAY: - flight.set_state(Takeoff(flight, self.game.settings, now)) - elif flight.start_type is StartType.IN_FLIGHT: - flight.set_state(Navigating(flight, self.game.settings, waypoint_index=0)) - else: - raise ValueError(f"Unknown start type {flight.start_type} for {flight}") + flight.state.reinitialize(now) def reset(self) -> None: for flight in self.iter_flights():