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

Use any (and only) transport aircraft for airlift.

Browse Source 
https://github.com/Khopa/dcs_liberation/issues/825
This commit is contained in:
Dan Albert 2021-04-23 22:05:49 -07:00
parent dac2271084
commit c8b4fd1690
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
game/transfers.py
View File

@ -10,6 +10,7 @@ from dcs.mapping import Point
from dcs.unittype import FlyingType, VehicleType from dcs.unittype import FlyingType, VehicleType
from gen.ato import Package from gen.ato import Package
from gen.flights.ai_flight_planner_db import TRANSPORT_CAPABLE
from gen.flights.flightplan import FlightPlanBuilder from gen.flights.flightplan import FlightPlanBuilder
from game.theater import ControlPoint, MissionTarget from game.theater import ControlPoint, MissionTarget
from game.theater.supplyroutes import SupplyRoute from game.theater.supplyroutes import SupplyRoute
@ -147,15 +148,23 @@ class AirliftPlanner:
self.for_player = transfer.destination.captured self.for_player = transfer.destination.captured
self.package = Package(target=transfer.destination, auto_asap=True) self.package = Package(target=transfer.destination, auto_asap=True)
def is_airlift_capable(self, unit_type: Type[FlyingType]) -> bool:
return (
unit_type in TRANSPORT_CAPABLE
and self.transfer.origin.can_operate(unit_type)
and self.transfer.destination.can_operate(unit_type)
)
def create_package_for_airlift(self) -> None: def create_package_for_airlift(self) -> None:
for cp in self.game.theater.player_points(): for cp in self.game.theater.player_points():
inventory = self.game.aircraft_inventory.for_control_point(cp) inventory = self.game.aircraft_inventory.for_control_point(cp)
for unit_type, available in inventory.all_aircraft: for unit_type, available in inventory.all_aircraft:
if unit_type.helicopter: if self.is_airlift_capable(unit_type):
while available and self.transfer.transport is None: while available and self.transfer.transport is None:
flight_size = self.create_airlift_flight(unit_type, inventory) flight_size = self.create_airlift_flight(unit_type, inventory)
available -= flight_size available -= flight_size
self.game.ato_for(self.for_player).add_package(self.package) if self.package.flights:
self.game.ato_for(self.for_player).add_package(self.package)
def create_airlift_flight( def create_airlift_flight(
self, unit_type: Type[FlyingType], inventory: ControlPointAircraftInventory self, unit_type: Type[FlyingType], inventory: ControlPointAircraftInventory