weyne/dcs-retribution
1
0
Fork 0
mirror of https://github.com/dcs-retribution/dcs-retribution.git synced 2025-11-10 15:41:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Ground unit presets are now generated from the coalition/faction definitions.

Browse Source
This commit is contained in:
MetalStormGhost 2023-12-26 17:23:15 +02:00
parent 200812a39b
commit 2edc0a7305
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
game/pretense/pretensetgogenerator.py
View File

@ -17,6 +17,7 @@ from dcs.country import Country
from dcs.unitgroup import StaticGroup, VehicleGroup from dcs.unitgroup import StaticGroup, VehicleGroup
from dcs.unittype import VehicleType from dcs.unittype import VehicleType
from game.coalition import Coalition
from game.data.units import UnitClass from game.data.units import UnitClass
from game.dcs.groundunittype import GroundUnitType from game.dcs.groundunittype import GroundUnitType
from game.missiongenerator.groundforcepainter import ( from game.missiongenerator.groundforcepainter import (
@ -128,7 +129,10 @@ class PretenseGroundObjectGenerator(GroundObjectGenerator):
def culled(self) -> bool: def culled(self) -> bool:
return self.game.iads_considerate_culling(self.ground_object) return self.game.iads_considerate_culling(self.ground_object)
def ground_unit_of_class(self, unit_class: UnitClass) -> Optional[GroundUnitType]: @staticmethod
def ground_unit_of_class(
coalition: Coalition, unit_class: UnitClass
) -> Optional[GroundUnitType]:
""" """
Returns a GroundUnitType of the specified class that belongs to the Returns a GroundUnitType of the specified class that belongs to the
TheaterGroundObject faction. TheaterGroundObject faction.
@ -137,12 +141,13 @@ class PretenseGroundObjectGenerator(GroundObjectGenerator):
are removed based on a pre-defined list. are removed based on a pre-defined list.
Args: Args:
coalition: Coalition to return the unit for.
unit_class: Class of unit to return. unit_class: Class of unit to return.
""" """
faction_units = ( faction_units = (
set(self.coalition.faction.frontline_units) set(coalition.faction.frontline_units)
| set(self.coalition.faction.artillery_units) | set(coalition.faction.artillery_units)
| set(self.coalition.faction.logistics_units) | set(coalition.faction.logistics_units)
) )
of_class = list({u for u in faction_units if u.unit_class is unit_class}) of_class = list({u for u in faction_units if u.unit_class is unit_class})
@ -186,7 +191,7 @@ class PretenseGroundObjectGenerator(GroundObjectGenerator):
""" """
if self.coalition.faction.has_access_to_unit_class(unit_class): if self.coalition.faction.has_access_to_unit_class(unit_class):
unit_type = self.ground_unit_of_class(unit_class) unit_type = self.ground_unit_of_class(self.coalition, unit_class)
if unit_type is not None and len(vehicle_units) < max_num: if unit_type is not None and len(vehicle_units) < max_num:
unit_id = self.game.next_unit_id() unit_id = self.game.next_unit_id()
unit_name = f"{cp_name}-{group_role}-{unit_id}" unit_name = f"{cp_name}-{group_role}-{unit_id}"