mirror of
https://github.com/dcs-liberation/dcs_liberation.git
synced 2025-11-10 14:22:26 +00:00
d09a15a7f3
This PR: Keeps track of time spent in mission Introduces a new "turnless mode" setting, which activates the following: - At the end of a mission, fast forwards through the time spent in the mission, skipping any combat (which has already been tracked through state.json) - Removes killed flights from the ATO - Does not start a new turn, instead allows the player to continue the current turn.
101 lines
3.6 KiB
Python
101 lines
3.6 KiB
Python
from __future__ import annotations
|
|
|
|
from collections.abc import Iterator
|
|
from typing import Optional, TYPE_CHECKING
|
|
|
|
from .flightmember import FlightMember
|
|
from .flightroster import FlightRoster
|
|
from .iflightroster import IFlightRoster
|
|
from .loadouts import Loadout
|
|
|
|
if TYPE_CHECKING:
|
|
from game.squadrons import Pilot
|
|
from .flight import Flight
|
|
|
|
|
|
class FlightMembers(IFlightRoster):
|
|
def __init__(self, flight: Flight, initial_size: int = 0) -> None:
|
|
self.flight = flight
|
|
self.members: list[FlightMember] = []
|
|
self.resize(initial_size)
|
|
|
|
@staticmethod
|
|
def from_roster(flight: Flight, roster: FlightRoster) -> FlightMembers:
|
|
members = FlightMembers(flight)
|
|
loadout = Loadout.default_for(flight)
|
|
members.members = [FlightMember(p, loadout) for p in roster.pilots]
|
|
return members
|
|
|
|
def iter_pilots(self) -> Iterator[Pilot | None]:
|
|
yield from (m.pilot for m in self.members)
|
|
|
|
def pilot_at(self, idx: int) -> Pilot | None:
|
|
return self.members[idx].pilot
|
|
|
|
@property
|
|
def max_size(self) -> int:
|
|
return len(self.members)
|
|
|
|
@property
|
|
def player_count(self) -> int:
|
|
return len([m for m in self.members if m.is_player])
|
|
|
|
@property
|
|
def missing_pilots(self) -> int:
|
|
return len([m for m in self.members if m.pilot is None])
|
|
|
|
def resize(self, new_size: int) -> None:
|
|
if self.max_size > new_size:
|
|
for member in self.members[new_size:]:
|
|
if (pilot := member.pilot) is not None:
|
|
self.flight.squadron.return_pilot(pilot)
|
|
if (code := member.tgp_laser_code) is not None:
|
|
code.release()
|
|
self.members = self.members[:new_size]
|
|
return
|
|
if self.max_size:
|
|
loadout = self.members[0].loadout.clone()
|
|
else:
|
|
loadout = Loadout.default_for(self.flight)
|
|
for _ in range(new_size - self.max_size):
|
|
member = FlightMember(self.flight.squadron.claim_available_pilot(), loadout)
|
|
member.use_custom_loadout = loadout.is_custom
|
|
self.members.append(member)
|
|
|
|
def set_pilot(self, index: int, pilot: Optional[Pilot]) -> None:
|
|
if pilot is not None:
|
|
self.flight.squadron.claim_pilot(pilot)
|
|
if (current_pilot := self.pilot_at(index)) is not None:
|
|
self.flight.squadron.return_pilot(current_pilot)
|
|
self.members[index].pilot = pilot
|
|
|
|
def remove_pilot(self, pilot: Pilot) -> None:
|
|
for i, member in enumerate(self.members):
|
|
if member.pilot is not None and member.pilot.name == pilot.name:
|
|
self.members.pop(i)
|
|
if (code := member.tgp_laser_code) is not None:
|
|
code.release()
|
|
return
|
|
raise ValueError(f"Pilot {pilot.name} not a member")
|
|
|
|
def clear(self) -> None:
|
|
self.flight.squadron.return_pilots(
|
|
[p for p in self.iter_pilots() if p is not None]
|
|
)
|
|
for member in self.members:
|
|
if (code := member.tgp_laser_code) is not None:
|
|
code.release()
|
|
|
|
def use_same_loadout_for_all_members(self) -> None:
|
|
if not self.members:
|
|
return
|
|
loadout = self.members[0].loadout
|
|
for member in self.members[1:]:
|
|
# Do not clone the loadout, we want any changes in the UI to be mirrored
|
|
# across all flight members.
|
|
member.loadout = loadout
|
|
|
|
def use_distinct_loadouts_for_each_member(self) -> None:
|
|
for member in self.members:
|
|
member.loadout = member.loadout.clone()
|