Add logged_duration context manager for profiling.

game/profiling.py

@@ -0,0 +1,13 @@
+import logging
+import timeit
+from contextlib import contextmanager
+from datetime import timedelta
+from typing import Iterator
+
+
+@contextmanager
+def logged_duration(event: str) -> Iterator[None]:
+    start = timeit.default_timer()
+    yield
+    end = timeit.default_timer()
+    logging.debug("%s took %s", event, timedelta(seconds=end - start))

game/theater/conflicttheater.py

@@ -56,6 +56,7 @@ from .frontline import FrontLine
 from .landmap import Landmap, load_landmap, poly_contains
 from .projections import TransverseMercator
 from ..point_with_heading import PointWithHeading
+from ..profiling import logged_duration
 from ..utils import Distance, meters, nautical_miles
 
 SIZE_TINY = 150
@@ -114,7 +115,8 @@ class MizCampaignLoader:
     def __init__(self, miz: Path, theater: ConflictTheater) -> None:
         self.theater = theater
         self.mission = Mission()
-        self.mission.load_file(str(miz))
+        with logged_duration("Loading miz"):
+            self.mission.load_file(str(miz))
         self.control_point_id = itertools.count(1000)
 
         # If there are no red carriers there usually aren't red units. Make sure
@@ -682,7 +684,9 @@ class ConflictTheater:
             raise RuntimeError(
                 "Old format (non-miz) campaigns are no longer supported."
             )
-        MizCampaignLoader(directory / miz, t).populate_theater()
+
+        with logged_duration("Importing miz data"):
+            MizCampaignLoader(directory / miz, t).populate_theater()
         return t
 
     @property

game/theater/start_generator.py

@@ -46,6 +46,7 @@ from . import (
     Fob,
     OffMapSpawn,
 )
+from ..profiling import logged_duration
 from ..settings import Settings
 
 GroundObjectTemplates = Dict[str, Dict[str, Any]]
@@ -91,20 +92,21 @@ class GameGenerator:
         self.generator_settings = generator_settings
 
     def generate(self) -> Game:
-        # Reset name generator
-        namegen.reset()
-        self.prepare_theater()
-        game = Game(
-            player_name=self.player,
-            enemy_name=self.enemy,
-            theater=self.theater,
-            start_date=self.generator_settings.start_date,
-            settings=self.settings,
-            player_budget=self.generator_settings.player_budget,
-            enemy_budget=self.generator_settings.enemy_budget,
-        )
+        with logged_duration("TGO population"):
+            # Reset name generator
+            namegen.reset()
+            self.prepare_theater()
+            game = Game(
+                player_name=self.player,
+                enemy_name=self.enemy,
+                theater=self.theater,
+                start_date=self.generator_settings.start_date,
+                settings=self.settings,
+                player_budget=self.generator_settings.player_budget,
+                enemy_budget=self.generator_settings.enemy_budget,
+            )
 
-        GroundObjectGenerator(game, self.generator_settings).generate()
+            GroundObjectGenerator(game, self.generator_settings).generate()
         game.settings.version = VERSION
         game.begin_turn_0()
         return game