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
68 lines
1.8 KiB
Python
68 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import List, TYPE_CHECKING
|
|
|
|
if TYPE_CHECKING:
|
|
from game import Game
|
|
|
|
|
|
class FactionTurnMetadata:
|
|
"""
|
|
Store metadata about a faction
|
|
"""
|
|
|
|
aircraft_count: int = 0
|
|
vehicles_count: int = 0
|
|
sam_count: int = 0
|
|
|
|
def __init__(self) -> None:
|
|
self.aircraft_count = 0
|
|
self.vehicles_count = 0
|
|
self.sam_count = 0
|
|
|
|
|
|
class GameTurnMetadata:
|
|
"""
|
|
Store metadata about a game turn
|
|
"""
|
|
|
|
allied_units: FactionTurnMetadata
|
|
enemy_units: FactionTurnMetadata
|
|
|
|
def __init__(self) -> None:
|
|
self.allied_units = FactionTurnMetadata()
|
|
self.enemy_units = FactionTurnMetadata()
|
|
|
|
|
|
class GameStats:
|
|
"""
|
|
Store statistics for the current game
|
|
"""
|
|
|
|
def __init__(self) -> None:
|
|
self.data_per_turn: List[GameTurnMetadata] = []
|
|
|
|
def update(self, game: Game) -> None:
|
|
"""
|
|
Save data for current turn
|
|
:param game: Game we want to save the data about
|
|
"""
|
|
|
|
# Remove the current turn if its just an update for this turn
|
|
if 0 < game.turn < len(self.data_per_turn):
|
|
del self.data_per_turn[-1]
|
|
|
|
turn_data = GameTurnMetadata()
|
|
|
|
for cp in game.theater.controlpoints:
|
|
if cp.captured.is_blue:
|
|
for squadron in cp.squadrons:
|
|
turn_data.allied_units.aircraft_count += squadron.owned_aircraft
|
|
turn_data.allied_units.vehicles_count += sum(cp.base.armor.values())
|
|
else:
|
|
for squadron in cp.squadrons:
|
|
turn_data.enemy_units.aircraft_count += squadron.owned_aircraft
|
|
turn_data.enemy_units.vehicles_count += sum(cp.base.armor.values())
|
|
|
|
self.data_per_turn.append(turn_data)
|