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

Cleaned up pretense/pretenseaircraftgenerator.py

Browse Source 
Moved a lot of logic into proper methods and added comments for all of those methods.
This commit is contained in:
MetalStormGhost 2023-09-16 21:11:55 +03:00
parent 3c139d8038
commit 51462673a2
1 changed files with 129 additions and 71 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

200
game/pretense/pretenseaircraftgenerator.py
View File

@ -98,34 +98,11 @@ class PretenseAircraftGenerator:
for parking_slot in cp.parking_slots: for parking_slot in cp.parking_slots:
parking_slot.unit_id = None parking_slot.unit_id = None
def generate_flights( def find_pretense_cargo_plane_cp(self, cp: ControlPoint) -> ControlPoint:
self,
country: Country,
cp: ControlPoint,
ato: AirTaskingOrder,
dynamic_runways: Dict[str, RunwayData],
) -> None:
from game.squadrons import Squadron
"""Adds aircraft to the mission for every flight in the ATO.
Aircraft generation is done by walking the ATO and spawning each flight in turn.
After the flight is generated the group is added to the UnitMap so aircraft
deaths can be tracked.
Args:
country: The country from the mission to use for this ATO.
ato: The ATO to spawn aircraft for.
dynamic_runways: Runway data for carriers and FARPs.
""" """
Finds the location (ControlPoint) for Pretense off-map transport planes.
num_of_sead = 0 """
num_of_cas = 0 distance_to_flot = 0.0
num_of_strike = 0
num_of_cap = 0
# Find locations for off-map transport planes
distance_to_flot = 0
offmap_transport_cp_id = cp.id offmap_transport_cp_id = cp.id
parking_type = ParkingType( parking_type = ParkingType(
fixed_wing=True, fixed_wing_stol=True, rotary_wing=False fixed_wing=True, fixed_wing_stol=True, rotary_wing=False
@ -144,11 +121,19 @@ class PretenseAircraftGenerator:
front_line.position front_line.position
) )
offmap_transport_cp_id = front_line_cp.id offmap_transport_cp_id = front_line_cp.id
offmap_transport_cp = self.game.theater.find_control_point_by_id( return self.game.theater.find_control_point_by_id(offmap_transport_cp_id)
offmap_transport_cp_id
)
# Ensure that the faction has at least one transport helicopter and one cargo plane squadron def should_generate_pretense_transports(
self, cp: ControlPoint
) -> tuple[bool, bool]:
"""
Returns a tuple of booleans, telling whether transport helicopter and aircraft
squadrons should be generated from the faction squadron definitions.
This helps to ensure that the faction has at least one transport helicopter and one cargo plane squadron.
(autogenerate_transport_helicopter_squadron, autogenerate_cargo_plane_squadron)
"""
autogenerate_transport_helicopter_squadron = True autogenerate_transport_helicopter_squadron = True
autogenerate_cargo_plane_squadron = True autogenerate_cargo_plane_squadron = True
for aircraft_type in cp.coalition.air_wing.squadrons: for aircraft_type in cp.coalition.air_wing.squadrons:
@ -164,48 +149,80 @@ class PretenseAircraftGenerator:
or FlightType.AIR_ASSAULT in mission_types or FlightType.AIR_ASSAULT in mission_types
): ):
autogenerate_cargo_plane_squadron = False autogenerate_cargo_plane_squadron = False
return (
autogenerate_transport_helicopter_squadron,
autogenerate_cargo_plane_squadron,
)
if autogenerate_transport_helicopter_squadron: def generate_pretense_transport_squadron(
flight_type = FlightType.AIR_ASSAULT self,
squadron_def = ( cp: ControlPoint,
cp.coalition.air_wing.squadron_def_generator.generate_for_task( flight_type: FlightType,
flight_type, offmap_transport_cp fixed_wing: bool,
) num_retries: int,
) ) -> None:
squadron = Squadron.create_from( from game.squadrons import Squadron
squadron_def,
flight_type, """
2, Generates a Pretense transport squadron from the faction squadron definitions. Use FlightType AIR_ASSAULT
offmap_transport_cp, for Pretense supply helicopters and TRANSPORT for off-map cargo plane squadrons.
cp.coalition,
self.game, Retribution does not differentiate between fixed wing and rotary wing transport squadron definitions, which
) is why there is a retry mechanism in case the wrong type is returned. Use fixed_wing False
cp.coalition.air_wing.squadrons[squadron.aircraft] = list() for Pretense supply helicopters and fixed_wing True for off-map cargo plane squadrons.
cp.coalition.air_wing.add_squadron(squadron)
if autogenerate_cargo_plane_squadron: TODO: Find out if Pretense can handle rotary wing "cargo planes".
flight_type = FlightType.TRANSPORT """
squadron_def = (
cp.coalition.air_wing.squadron_def_generator.generate_for_task( squadron_def = cp.coalition.air_wing.squadron_def_generator.generate_for_task(
flight_type, offmap_transport_cp flight_type, cp
) )
) print(
for retries in range(PRETENSE_SQUADRON_DEF_RETRIES): f"Generating a squadron definition for fixed-wing {fixed_wing} squadron at {cp}"
if squadron_def.aircraft.helicopter: )
squadron_def = ( for retries in range(num_retries):
cp.coalition.air_wing.squadron_def_generator.generate_for_task( if squadron_def is None or fixed_wing != squadron_def.aircraft.helicopter:
flight_type, offmap_transport_cp squadron_def = (
) cp.coalition.air_wing.squadron_def_generator.generate_for_task(
flight_type, cp
) )
squadron = Squadron.create_from( )
squadron_def,
flight_type, # Failed, stop here
2, if squadron_def is None:
offmap_transport_cp, return
cp.coalition,
self.game, squadron = Squadron.create_from(
) squadron_def,
cp.coalition.air_wing.squadrons[squadron.aircraft] = list() flight_type,
cp.coalition.air_wing.add_squadron(squadron) 2,
cp,
cp.coalition,
self.game,
)
cp.coalition.air_wing.squadrons[squadron.aircraft] = list()
cp.coalition.air_wing.add_squadron(squadron)
return
def generate_pretense_aircraft(
self, cp: ControlPoint, ato: AirTaskingOrder
) -> None:
"""
Plans and generates AI aircraft groups/packages for Pretense.
Aircraft generation is done by walking the control points which will be made into
Pretense "zones" and spawning flights for different missions.
After the flight is generated the package is added to the ATO so the flights
can be configured.
Args:
cp: Control point to generate aircraft for.
ato: The ATO to generate aircraft for.
"""
num_of_sead = 0
num_of_cas = 0
num_of_strike = 0
num_of_cap = 0
for squadron in cp.squadrons: for squadron in cp.squadrons:
# Intentionally don't spawn anything at OffMapSpawns in Pretense # Intentionally don't spawn anything at OffMapSpawns in Pretense
@ -279,6 +296,47 @@ class PretenseAircraftGenerator:
) )
ato.add_package(package) ato.add_package(package)
return
def generate_flights(
self,
country: Country,
cp: ControlPoint,
ato: AirTaskingOrder,
dynamic_runways: Dict[str, RunwayData],
) -> None:
"""Adds aircraft to the mission for every flight in the ATO.
Args:
country: The country from the mission to use for this ATO.
cp: Control point to generate aircraft for.
ato: The ATO to generate aircraft for.
dynamic_runways: Runway data for carriers and FARPs.
"""
offmap_transport_cp = self.find_pretense_cargo_plane_cp(cp)
(
autogenerate_transport_helicopter_squadron,
autogenerate_cargo_plane_squadron,
) = self.should_generate_pretense_transports(cp)
if autogenerate_transport_helicopter_squadron:
self.generate_pretense_transport_squadron(
offmap_transport_cp,
FlightType.AIR_ASSAULT,
False,
PRETENSE_SQUADRON_DEF_RETRIES,
)
if autogenerate_cargo_plane_squadron:
self.generate_pretense_transport_squadron(
offmap_transport_cp,
FlightType.TRANSPORT,
True,
PRETENSE_SQUADRON_DEF_RETRIES,
)
self.generate_pretense_aircraft(cp, ato)
self._reserve_frequencies_and_tacan(ato) self._reserve_frequencies_and_tacan(ato)