dcs_liberation/game/sim/combat/joinablecombat.py
Dan Albert 2585dcc130 Add (very!) rough simulation of frozen combat.
There are some TODOs here but th behavior is flagged off by default. The
biggest TODO here is that the time spent frozen is not simulated, so
flights that are engaged by SAMs will unfreeze, move slightly, then re-
freeze.

https://github.com/dcs-liberation/dcs_liberation/issues/1680
2022-02-26 13:01:46 -08:00

32 lines
920 B
Python

from __future__ import annotations
from abc import ABC, abstractmethod
from collections.abc import Iterator
from datetime import timedelta
from typing import TYPE_CHECKING
from game.ato.flightstate import InCombat, InFlight
from .frozencombat import FrozenCombat
if TYPE_CHECKING:
from game.ato import Flight
class JoinableCombat(FrozenCombat, ABC):
def __init__(self, freeze_duration: timedelta, flights: list[Flight]) -> None:
super().__init__(freeze_duration)
self.flights = flights
@abstractmethod
def joinable_by(self, flight: Flight) -> bool:
...
def join(self, flight: Flight) -> None:
assert isinstance(flight.state, InFlight)
assert not isinstance(flight.state, InCombat)
self.flights.append(flight)
flight.set_state(InCombat(flight.state, self))
def iter_flights(self) -> Iterator[Flight]:
yield from self.flights