mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
107 lines
3.5 KiB
Python
107 lines
3.5 KiB
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
import random
|
|
from typing import TYPE_CHECKING, Optional
|
|
|
|
from dcs.unitgroup import ShipGroup
|
|
|
|
from game import db
|
|
from game.theater.theatergroundobject import (
|
|
LhaGroundObject,
|
|
CarrierGroundObject,
|
|
ShipGroundObject,
|
|
)
|
|
from gen.fleet.carrier_group import CarrierGroupGenerator
|
|
from gen.fleet.cn_dd_group import ChineseNavyGroupGenerator, Type54GroupGenerator
|
|
from gen.fleet.dd_group import (
|
|
ArleighBurkeGroupGenerator,
|
|
OliverHazardPerryGroupGenerator,
|
|
)
|
|
from gen.fleet.lacombattanteII import LaCombattanteIIGroupGenerator
|
|
from gen.fleet.lha_group import LHAGroupGenerator
|
|
from gen.fleet.ru_dd_group import (
|
|
RussianNavyGroupGenerator,
|
|
GrishaGroupGenerator,
|
|
MolniyaGroupGenerator,
|
|
KiloSubGroupGenerator,
|
|
TangoSubGroupGenerator,
|
|
)
|
|
from gen.fleet.schnellboot import SchnellbootGroupGenerator
|
|
from gen.fleet.uboat import UBoatGroupGenerator
|
|
from gen.fleet.ww2lst import WW2LSTGroupGenerator
|
|
|
|
if TYPE_CHECKING:
|
|
from game import Game
|
|
|
|
|
|
SHIP_MAP = {
|
|
"SchnellbootGroupGenerator": SchnellbootGroupGenerator,
|
|
"WW2LSTGroupGenerator": WW2LSTGroupGenerator,
|
|
"UBoatGroupGenerator": UBoatGroupGenerator,
|
|
"OliverHazardPerryGroupGenerator": OliverHazardPerryGroupGenerator,
|
|
"ArleighBurkeGroupGenerator": ArleighBurkeGroupGenerator,
|
|
"RussianNavyGroupGenerator": RussianNavyGroupGenerator,
|
|
"ChineseNavyGroupGenerator": ChineseNavyGroupGenerator,
|
|
"GrishaGroupGenerator": GrishaGroupGenerator,
|
|
"MolniyaGroupGenerator": MolniyaGroupGenerator,
|
|
"KiloSubGroupGenerator": KiloSubGroupGenerator,
|
|
"TangoSubGroupGenerator": TangoSubGroupGenerator,
|
|
"Type54GroupGenerator": Type54GroupGenerator,
|
|
"LaCombattanteIIGroupGenerator": LaCombattanteIIGroupGenerator,
|
|
}
|
|
|
|
|
|
def generate_ship_group(
|
|
game: Game, ground_object: ShipGroundObject, faction_name: str
|
|
) -> Optional[ShipGroup]:
|
|
"""
|
|
This generate a ship group
|
|
:return: The generated group, or None if this faction does not support ships.
|
|
"""
|
|
faction = db.FACTIONS[faction_name]
|
|
if len(faction.navy_generators) > 0:
|
|
gen = random.choice(faction.navy_generators)
|
|
if gen in SHIP_MAP.keys():
|
|
generator = SHIP_MAP[gen](game, ground_object, faction)
|
|
print(generator.position)
|
|
generator.generate()
|
|
return generator.get_generated_group()
|
|
else:
|
|
logging.info(
|
|
"Unable to generate ship group, generator : "
|
|
+ str(gen)
|
|
+ "does not exists"
|
|
)
|
|
return None
|
|
|
|
|
|
def generate_carrier_group(
|
|
faction: str, game: Game, ground_object: CarrierGroundObject
|
|
) -> ShipGroup:
|
|
"""Generates a carrier group.
|
|
|
|
:param faction: The faction the TGO belongs to.
|
|
:param game: The Game the group is being generated for.
|
|
:param ground_object: The ground object which will own the ship group
|
|
:return: The generated group.
|
|
"""
|
|
generator = CarrierGroupGenerator(game, ground_object, db.FACTIONS[faction])
|
|
generator.generate()
|
|
return generator.get_generated_group()
|
|
|
|
|
|
def generate_lha_group(
|
|
faction: str, game: Game, ground_object: LhaGroundObject
|
|
) -> ShipGroup:
|
|
"""Generate an LHA group.
|
|
|
|
:param faction: The faction the TGO belongs to.
|
|
:param game: The Game the group is being generated for.
|
|
:param ground_object: The ground object which will own the ship group
|
|
:return: The generated group.
|
|
"""
|
|
generator = LHAGroupGenerator(game, ground_object, db.FACTIONS[faction])
|
|
generator.generate()
|
|
return generator.get_generated_group()
|