mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
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
|