mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
18 lines
781 B
Python
18 lines
781 B
Python
from collections.abc import Iterator
|
|
|
|
from game.commander.tasks.primitive.armedrecon import PlanArmedRecon
|
|
from game.commander.tasks.primitive.bai import PlanBai
|
|
from game.commander.theaterstate import TheaterState
|
|
from game.htn import CompoundTask, Method
|
|
|
|
|
|
class AttackBattlePositions(CompoundTask[TheaterState]):
|
|
def each_valid_method(self, state: TheaterState) -> Iterator[Method[TheaterState]]:
|
|
for battle_positions in state.enemy_battle_positions.values():
|
|
for battle_position in battle_positions.in_priority_order:
|
|
yield [PlanBai(battle_position)]
|
|
# Only plan against the 2 most important CPs
|
|
for cp in state.control_point_priority_queue[:2]:
|
|
if not cp.is_fleet:
|
|
yield [PlanArmedRecon(cp)]
|