mirror of
https://github.com/dcs-liberation/dcs_liberation.git
synced 2025-11-10 14:22:26 +00:00
ccf6b6ef5f
This makes it so that the mission planning effects are applied only if the package can be fulfilled. For example, breakthrough will be used only if all the BAI missions were fulfilled, not if they will *attempt* to be fulfilled.
76 lines
2.6 KiB
Python
76 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
import math
|
|
from abc import ABC, abstractmethod
|
|
from typing import TYPE_CHECKING
|
|
|
|
from game.commander.tasks.theatercommandertask import TheaterCommanderTask
|
|
from game.commander.theaterstate import TheaterState
|
|
from game.theater import FrontLine
|
|
from gen.ground_forces.combat_stance import CombatStance
|
|
|
|
if TYPE_CHECKING:
|
|
from game.coalition import Coalition
|
|
|
|
|
|
class FrontLineStanceTask(TheaterCommanderTask, ABC):
|
|
def __init__(self, front_line: FrontLine, player: bool) -> None:
|
|
self.front_line = front_line
|
|
self.friendly_cp = self.front_line.control_point_friendly_to(player)
|
|
self.enemy_cp = self.front_line.control_point_hostile_to(player)
|
|
|
|
@property
|
|
@abstractmethod
|
|
def stance(self) -> CombatStance:
|
|
...
|
|
|
|
@staticmethod
|
|
def management_allowed(state: TheaterState) -> bool:
|
|
return (
|
|
not state.context.coalition.player
|
|
or state.context.settings.automate_front_line_stance
|
|
)
|
|
|
|
def better_stance_already_set(self, state: TheaterState) -> bool:
|
|
current_stance = state.front_line_stances[self.front_line]
|
|
if current_stance is None:
|
|
return False
|
|
preference = (
|
|
CombatStance.RETREAT,
|
|
CombatStance.DEFENSIVE,
|
|
CombatStance.AMBUSH,
|
|
CombatStance.AGGRESSIVE,
|
|
CombatStance.ELIMINATION,
|
|
CombatStance.BREAKTHROUGH,
|
|
)
|
|
current_rating = preference.index(current_stance)
|
|
new_rating = preference.index(self.stance)
|
|
return current_rating >= new_rating
|
|
|
|
@property
|
|
@abstractmethod
|
|
def have_sufficient_front_line_advantage(self) -> bool:
|
|
...
|
|
|
|
@property
|
|
def ground_force_balance(self) -> float:
|
|
# TODO: Planned CAS missions should reduce the expected opposing force size.
|
|
friendly_forces = self.friendly_cp.deployable_front_line_units
|
|
enemy_forces = self.enemy_cp.deployable_front_line_units
|
|
if enemy_forces == 0:
|
|
return math.inf
|
|
return friendly_forces / enemy_forces
|
|
|
|
def preconditions_met(self, state: TheaterState) -> bool:
|
|
if not self.management_allowed(state):
|
|
return False
|
|
if self.better_stance_already_set(state):
|
|
return False
|
|
return self.have_sufficient_front_line_advantage
|
|
|
|
def apply_effects(self, state: TheaterState) -> None:
|
|
state.front_line_stances[self.front_line] = self.stance
|
|
|
|
def execute(self, coalition: Coalition) -> None:
|
|
self.friendly_cp.stances[self.enemy_cp.id] = self.stance
|