mirror of
https://github.com/dcs-liberation/dcs_liberation.git
synced 2025-11-10 14:22:26 +00:00
12 lines
435 B
Python
12 lines
435 B
Python
from collections.abc import Iterator
|
|
|
|
from game.commander.tasks.primitive.refueling import PlanRefueling
|
|
from game.commander.theaterstate import TheaterState
|
|
from game.htn import CompoundTask, Method
|
|
|
|
|
|
class PlanRefuelingSupport(CompoundTask[TheaterState]):
|
|
def each_valid_method(self, state: TheaterState) -> Iterator[Method[TheaterState]]:
|
|
for target in state.refueling_targets:
|
|
yield [PlanRefueling(target)]
|