ntentionally don't spawn anything at OffMapSpawns in Pretense

This commit is contained in:
MetalStormGhost 2023-09-11 21:42:42 +03:00
parent 7b2424c746
commit 9f271cdb91
2 changed files with 9 additions and 2 deletions

View File

@ -32,6 +32,7 @@ from game.theater.controlpoint import (
Airfield,
ControlPoint,
Fob,
OffMapSpawn,
)
from game.unitmap import UnitMap
from game.missiongenerator.aircraft.aircraftpainter import AircraftPainter
@ -121,6 +122,10 @@ class PretenseAircraftGenerator:
num_of_strike = 0
num_of_cap = 0
for squadron in cp.squadrons:
# Intentionally don't spawn anything at OffMapSpawns in Pretense
if isinstance(squadron.location, OffMapSpawn):
continue
squadron.owned_aircraft += 1
squadron.untasked_aircraft += 1
package = Package(cp, squadron.flight_db, auto_asap=False)

View File

@ -17,7 +17,7 @@ from game.ato.starttype import StartType
from game.missiongenerator.aircraft.flightgroupspawner import FlightGroupSpawner
from game.missiongenerator.missiondata import MissionData
from game.naming import NameGenerator
from game.theater import Airfield, ControlPoint, Fob, NavalControlPoint
from game.theater import Airfield, ControlPoint, Fob, NavalControlPoint, OffMapSpawn
from game.utils import feet, meters
@ -67,11 +67,13 @@ class PretenseFlightGroupSpawner(FlightGroupSpawner):
cp = self.flight.departure
name = namegen.next_pretense_aircraft_name(cp, self.flight)
print(name)
try:
if self.start_type is StartType.IN_FLIGHT:
group = self._generate_over_departure(name, cp)
return group
elif isinstance(cp, OffMapSpawn):
# Intentionally don't spawn anything at OffMapSpawns in Pretense
logging.info(f"Skipping flight generation for off-map spawn {cp}.")
elif isinstance(cp, NavalControlPoint):
group_name = cp.get_carrier_group_name()
carrier_group = self.mission.find_group(group_name)