mirror of
https://github.com/dcs-liberation/dcs_liberation.git
synced 2025-11-10 14:22:26 +00:00
123 lines
3.5 KiB
Python
123 lines
3.5 KiB
Python
from __future__ import annotations
|
|
import random
|
|
from typing import TYPE_CHECKING
|
|
|
|
from dcs.ships import (
|
|
Corvette_1124_4_Grisha,
|
|
Corvette_1241_1_Molniya,
|
|
Frigate_11540_Neustrashimy,
|
|
Frigate_1135M_Rezky,
|
|
Cruiser_1164_Moskva,
|
|
SSK_877V_Kilo,
|
|
SSK_641B_Tango,
|
|
)
|
|
|
|
from gen.fleet.dd_group import DDGroupGenerator
|
|
from gen.sam.group_generator import ShipGroupGenerator
|
|
from game.factions.faction import Faction
|
|
from game.theater.theatergroundobject import TheaterGroundObject
|
|
|
|
|
|
if TYPE_CHECKING:
|
|
from game.game import Game
|
|
|
|
|
|
class RussianNavyGroupGenerator(ShipGroupGenerator):
|
|
def generate(self):
|
|
|
|
include_frigate = random.choice([True, True, False])
|
|
include_dd = random.choice([True, False])
|
|
|
|
if include_dd:
|
|
include_cc = random.choice([True, False])
|
|
else:
|
|
include_cc = False
|
|
|
|
if not any([include_frigate, include_dd, include_cc]):
|
|
include_frigate = True
|
|
|
|
if include_frigate:
|
|
frigate_type = random.choice(
|
|
[Corvette_1124_4_Grisha, Corvette_1241_1_Molniya]
|
|
)
|
|
self.add_unit(
|
|
frigate_type,
|
|
"FF1",
|
|
self.position.x + 1200,
|
|
self.position.y + 900,
|
|
self.heading,
|
|
)
|
|
self.add_unit(
|
|
frigate_type,
|
|
"FF2",
|
|
self.position.x + 1200,
|
|
self.position.y - 900,
|
|
self.heading,
|
|
)
|
|
|
|
if include_dd:
|
|
dd_type = random.choice([Frigate_11540_Neustrashimy, Frigate_1135M_Rezky])
|
|
self.add_unit(
|
|
dd_type,
|
|
"DD1",
|
|
self.position.x + 2400,
|
|
self.position.y + 900,
|
|
self.heading,
|
|
)
|
|
self.add_unit(
|
|
dd_type,
|
|
"DD2",
|
|
self.position.x + 2400,
|
|
self.position.y - 900,
|
|
self.heading,
|
|
)
|
|
|
|
if include_cc:
|
|
# Only include the Moskva for now, the Pyotry Velikiy is an unkillable monster.
|
|
# See https://github.com/dcs-liberation/dcs_liberation/issues/567
|
|
self.add_unit(
|
|
Cruiser_1164_Moskva,
|
|
"CC1",
|
|
self.position.x,
|
|
self.position.y,
|
|
self.heading,
|
|
)
|
|
|
|
self.get_generated_group().points[0].speed = 20
|
|
|
|
|
|
class GrishaGroupGenerator(DDGroupGenerator):
|
|
def __init__(
|
|
self, game: Game, ground_object: TheaterGroundObject, faction: Faction
|
|
):
|
|
super(GrishaGroupGenerator, self).__init__(
|
|
game, ground_object, faction, Corvette_1124_4_Grisha
|
|
)
|
|
|
|
|
|
class MolniyaGroupGenerator(DDGroupGenerator):
|
|
def __init__(
|
|
self, game: Game, ground_object: TheaterGroundObject, faction: Faction
|
|
):
|
|
super(MolniyaGroupGenerator, self).__init__(
|
|
game, ground_object, faction, Corvette_1241_1_Molniya
|
|
)
|
|
|
|
|
|
class KiloSubGroupGenerator(DDGroupGenerator):
|
|
def __init__(
|
|
self, game: Game, ground_object: TheaterGroundObject, faction: Faction
|
|
):
|
|
super(KiloSubGroupGenerator, self).__init__(
|
|
game, ground_object, faction, SSK_877V_Kilo
|
|
)
|
|
|
|
|
|
class TangoSubGroupGenerator(DDGroupGenerator):
|
|
def __init__(
|
|
self, game: Game, ground_object: TheaterGroundObject, faction: Faction
|
|
):
|
|
super(TangoSubGroupGenerator, self).__init__(
|
|
game, ground_object, faction, SSK_641B_Tango
|
|
)
|