mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
31c80dfd02
refactor to enum typing and many other fixes fix tests attempt to fix some typescript more typescript fixes more typescript test fixes revert all API changes update to pydcs mypy fixes Use properties to check if player is blue/red/neutral update requirements.txt black -_- bump pydcs and fix mypy add opponent property bump pydcs
77 lines
2.7 KiB
Python
77 lines
2.7 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.ground_forces.combat_stance import CombatStance
|
|
from game.theater import FrontLine
|
|
|
|
if TYPE_CHECKING:
|
|
from game.coalition import Coalition
|
|
from game.theater.player import Player
|
|
|
|
|
|
class FrontLineStanceTask(TheaterCommanderTask, ABC):
|
|
def __init__(self, front_line: FrontLine, player: Player) -> 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
|
|
if self.friendly_cp.deployable_front_line_units == 0:
|
|
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
|