mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
90ad1f4a61
https://github.com/dcs-liberation/dcs_liberation/issues/1145 Currently this is fixed at the start of the campaign. The squadron locations are defined by the campaign file. Follow up work: * Track aircraft ownership per-squadron rather than per-airbase. * UI for relocating squadrons. * Ferry missions for squadrons that are relocating. * Auto-relocation (probably only for retreat handling). Fixes https://github.com/dcs-liberation/dcs_liberation/issues/1138
69 lines
2.2 KiB
Python
69 lines
2.2 KiB
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
from collections import defaultdict
|
|
from dataclasses import dataclass
|
|
from typing import Any, TYPE_CHECKING, Union
|
|
|
|
from gen.flights.flight import FlightType
|
|
from game.theater.controlpoint import ControlPoint
|
|
|
|
if TYPE_CHECKING:
|
|
from game.theater import ConflictTheater
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class SquadronConfig:
|
|
primary: FlightType
|
|
secondary: list[FlightType]
|
|
aircraft: list[str]
|
|
|
|
@property
|
|
def auto_assignable(self) -> set[FlightType]:
|
|
return set(self.secondary) | {self.primary}
|
|
|
|
@classmethod
|
|
def from_data(cls, data: dict[str, Any]) -> SquadronConfig:
|
|
secondary_raw = data.get("secondary")
|
|
if secondary_raw is None:
|
|
secondary = []
|
|
elif isinstance(secondary_raw, str):
|
|
secondary = cls.expand_secondary_alias(secondary_raw)
|
|
else:
|
|
secondary = [FlightType(s) for s in secondary_raw]
|
|
|
|
return SquadronConfig(
|
|
FlightType(data["primary"]), secondary, data.get("aircraft", [])
|
|
)
|
|
|
|
@staticmethod
|
|
def expand_secondary_alias(alias: str) -> list[FlightType]:
|
|
if alias == "any":
|
|
return list(FlightType)
|
|
elif alias == "air-to-air":
|
|
return [t for t in FlightType if t.is_air_to_air]
|
|
elif alias == "air-to-ground":
|
|
return [t for t in FlightType if t.is_air_to_ground]
|
|
raise KeyError(f"Unknown secondary mission type: {alias}")
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class CampaignAirWingConfig:
|
|
by_location: dict[ControlPoint, list[SquadronConfig]]
|
|
|
|
@classmethod
|
|
def from_campaign_data(
|
|
cls, data: dict[Union[str, int], Any], theater: ConflictTheater
|
|
) -> CampaignAirWingConfig:
|
|
by_location: dict[ControlPoint, list[SquadronConfig]] = defaultdict(list)
|
|
for base_id, squadron_configs in data.items():
|
|
if isinstance(base_id, int):
|
|
base = theater.find_control_point_by_id(base_id)
|
|
else:
|
|
base = theater.control_point_named(base_id)
|
|
|
|
for squadron_data in squadron_configs:
|
|
by_location[base].append(SquadronConfig.from_data(squadron_data))
|
|
|
|
return CampaignAirWingConfig(by_location)
|