mirror of
https://github.com/dcs-liberation/dcs_liberation.git
synced 2025-11-10 14:22:26 +00:00
ca96a232f0
This reverts commit bd2ec12e0f039c9500ea0dd94e7b2e4f7d168fef. Country is both the data (name, ID, etc) and the container for groups added to the miz, so it can't be used across multiple mission generations. See https://github.com/pydcs/dcs/issues/314 for potential follow up work that would let us do this. Fixes https://github.com/dcs-liberation/dcs_liberation/issues/2864.
49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
from __future__ import annotations
|
|
|
|
import itertools
|
|
from typing import TYPE_CHECKING
|
|
|
|
from dcs import Mission
|
|
from dcs.ships import HandyWind
|
|
from dcs.unitgroup import ShipGroup
|
|
|
|
from game.transfers import CargoShip
|
|
from game.unitmap import UnitMap
|
|
from game.utils import knots
|
|
|
|
if TYPE_CHECKING:
|
|
from game import Game
|
|
|
|
|
|
class CargoShipGenerator:
|
|
def __init__(self, mission: Mission, game: Game, unit_map: UnitMap) -> None:
|
|
self.mission = mission
|
|
self.game = game
|
|
self.unit_map = unit_map
|
|
self.count = itertools.count()
|
|
|
|
def generate(self) -> None:
|
|
# Reset the count to make generation deterministic.
|
|
for coalition in self.game.coalitions:
|
|
for ship in coalition.transfers.cargo_ships:
|
|
self.generate_cargo_ship(ship)
|
|
|
|
def generate_cargo_ship(self, ship: CargoShip) -> ShipGroup:
|
|
country = self.mission.country(
|
|
self.game.coalition_for(ship.player_owned).country_name
|
|
)
|
|
waypoints = ship.route
|
|
group = self.mission.ship_group(
|
|
country,
|
|
ship.name,
|
|
HandyWind,
|
|
position=waypoints[0],
|
|
group_size=1,
|
|
)
|
|
for waypoint in waypoints[1:]:
|
|
# 12 knots is very slow but it's also nearly the max allowed by DCS for this
|
|
# type of ship.
|
|
group.add_waypoint(waypoint, speed=knots(12).kph)
|
|
self.unit_map.add_cargo_ship(group, ship)
|
|
return group
|