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

Set correct unit name for carriers

Browse Source
This commit is contained in:
RndName 2022-04-29 13:53:05 +02:00
parent 015103f613
commit 9e3edd7208
No known key found for this signature in database
GPG Key ID: 5EF516FD9537F7C0
1 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
game/theater/start_generator.py
View File

@ -190,7 +190,17 @@ class NoOpGroundObjectGenerator(ControlPointGroundObjectGenerator):
return True
class CarrierGroundObjectGenerator(ControlPointGroundObjectGenerator):
class GenericCarrierGroundObjectGenerator(ControlPointGroundObjectGenerator):
def update_carrier_name(self, carrier_name: str) -> None:
# Set Control Point name
self.control_point.name = carrier_name
# Set UnitName. First unit of the TGO is always the carrier
carrier = next(self.control_point.ground_objects[-1].units)
carrier.name = carrier_name
class CarrierGroundObjectGenerator(GenericCarrierGroundObjectGenerator):
def generate(self) -> bool:
if not super().generate():
return False
@ -215,11 +225,11 @@ class CarrierGroundObjectGenerator(ControlPointGroundObjectGenerator):
self.control_point.heading,
),
)
self.control_point.name = random.choice(carrier_names)
self.update_carrier_name(random.choice(carrier_names))
return True
class LhaGroundObjectGenerator(ControlPointGroundObjectGenerator):
class LhaGroundObjectGenerator(GenericCarrierGroundObjectGenerator):
def generate(self) -> bool:
if not super().generate():
return False
@ -246,7 +256,7 @@ class LhaGroundObjectGenerator(ControlPointGroundObjectGenerator):
self.control_point.heading,
),
)
self.control_point.name = random.choice(lha_names)
self.update_carrier_name(random.choice(lha_names))
return True