Implemented stats view.

game/db.py

@@ -86,6 +86,10 @@ PRICES = {
     AH_64D: 15,
     OH_58D: 6,
 
+    # Bombers
+    B_52H: 25,
+    B_1B: 50,
+
     # special
     IL_76MD: 13,
     An_26B: 13,
@@ -183,8 +187,8 @@ UNIT_BY_TASK = {
         MiG_23MLD,
         Su_27,
         Su_33,
-        MiG_21Bis,
         MiG_19P,
+        MiG_21Bis,
         MiG_29A,
         MiG_29S,
         FA_18C_hornet,
@@ -215,7 +219,9 @@ UNIT_BY_TASK = {
         Su_24MR,
         AH_64A,
         AH_64D,
-        OH_58D
+        OH_58D,
+        B_52H,
+        B_1B,
     ],
     Transport: [
         IL_76MD,
@@ -642,6 +648,8 @@ FACTIONS = {
             C_130,
             E_3A,
 
+            B_52H,
+
             UH_1H,
 
             Armor.MBT_M60A3_Patton,
@@ -669,6 +677,8 @@ FACTIONS = {
             A_10A,
             AV8BNA,
 
+            B_1B,
+
             KC_135,
             S_3B_Tanker,
             C_130,
@@ -707,6 +717,8 @@ FACTIONS = {
             A_10C,
             AV8BNA,
 
+            B_1B,
+
             KC_135,
             S_3B_Tanker,
             C_130,

game/game.py

@@ -6,6 +6,7 @@ import math
 from dcs.task import *
 from dcs.vehicles import *
 
+from game.game_stats import GameStats
 from gen.conflictgen import Conflict
 from userdata.debriefing import Debriefing
 from theater import *
@@ -93,6 +94,7 @@ class Game:
     pending_transfers = None  # type: typing.Dict[]
     ignored_cps = None  # type: typing.Collection[ControlPoint]
     turn = 0
+    game_stats: GameStats = None
 
     def __init__(self, player_name: str, enemy_name: str, theater: ConflictTheater, start_date: datetime):
         self.settings = Settings()
@@ -104,6 +106,8 @@ class Game:
         self.enemy_country = db.FACTIONS[enemy_name]["country"]
         self.turn = 0
         self.date = datetime(start_date.year, start_date.month, start_date.day)
+        self.game_stats = GameStats()
+        self.game_stats.update(self)
 
     def _roll(self, prob, mult):
         if self.settings.version == "dev":
@@ -297,8 +301,12 @@ class Game:
 
         self.events = []  # type: typing.List[Event]
         self._generate_events()
+
         #self._generate_globalinterceptions()
 
+        # Update statistics
+        self.game_stats.update(self)
+
     @property
     def current_turn_daytime(self):
         return ["dawn", "day", "dusk", "night"][self.turn % 4]

game/game_stats.py

@@ -0,0 +1,58 @@
+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
+    """
+
+    data_per_turn: [GameTurnMetadata] = []
+
+    def __init__(self):
+        self.data_per_turn = []
+
+    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.sam_count += sum(cp.base.aa.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.sam_count += sum(cp.base.aa.values())
+                turn_data.enemy_units.vehicles_count += sum(cp.base.armor.values())
+
+        self.data_per_turn.append(turn_data)
+