mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
57 lines
1.4 KiB
Python
57 lines
1.4 KiB
Python
from typing import List
|
|
|
|
|
|
class FactionTurnMetadata:
|
|
"""
|
|
Store metadata about a faction
|
|
"""
|
|
|
|
aircraft_count: int = 0
|
|
vehicles_count: int = 0
|
|
sam_count: int = 0
|
|
|
|
def __init__(self):
|
|
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):
|
|
self.allied_units = FactionTurnMetadata()
|
|
self.enemy_units = FactionTurnMetadata()
|
|
|
|
|
|
class GameStats:
|
|
"""
|
|
Store statistics for the current game
|
|
"""
|
|
|
|
def __init__(self):
|
|
self.data_per_turn: List[GameTurnMetadata] = []
|
|
|
|
def update(self, game):
|
|
"""
|
|
Save data for current turn
|
|
:param game: Game we want to save the data about
|
|
"""
|
|
|
|
turn_data = GameTurnMetadata()
|
|
|
|
for cp in game.theater.controlpoints:
|
|
if cp.captured:
|
|
turn_data.allied_units.aircraft_count += sum(cp.base.aircraft.values())
|
|
turn_data.allied_units.vehicles_count += sum(cp.base.armor.values())
|
|
else:
|
|
turn_data.enemy_units.aircraft_count += sum(cp.base.aircraft.values())
|
|
turn_data.enemy_units.vehicles_count += sum(cp.base.armor.values())
|
|
|
|
self.data_per_turn.append(turn_data)
|