make BAI plannable against vehicles only

This commit is contained in:
zhexu14 2023-05-03 23:48:16 +10:00 committed by Raffson
parent 2251018de2
commit aa64f28a4b
No known key found for this signature in database
GPG Key ID: B0402B2C9B764D99

View File

@ -163,7 +163,6 @@ class TheaterGroundObject(MissionTarget, SidcDescribable, ABC):
else: else:
yield from [ yield from [
FlightType.STRIKE, FlightType.STRIKE,
FlightType.BAI,
FlightType.REFUELING, FlightType.REFUELING,
] ]
yield from super().mission_types(for_player) yield from super().mission_types(for_player)
@ -602,6 +601,13 @@ class VehicleGroupGroundObject(TheaterGroundObject):
def should_head_to_conflict(self) -> bool: def should_head_to_conflict(self) -> bool:
return True return True
def mission_types(self, for_player: bool) -> Iterator[FlightType]:
from game.ato import FlightType
if not self.is_friendly(for_player):
yield FlightType.BAI
yield from super().mission_types(for_player)
class EwrGroundObject(IadsGroundObject): class EwrGroundObject(IadsGroundObject):
def __init__( def __init__(