log_fps

puterbot/replay.py

@@ -1,18 +1,8 @@
-from pprint import pformat
-import importlib
-import time
-
 from loguru import logger
-from pynput import keyboard, mouse
 import fire
 
-from puterbot.crud import (
-    get_latest_recording,
-)
-from puterbot.utils import (
-    configure_logging,
-    get_strategy_class_by_name,
-)
+from puterbot.crud import get_latest_recording
+from puterbot.utils import configure_logging, get_strategy_class_by_name
 
 
 LOG_LEVEL = "INFO"

puterbot/strategies/base.py

@@ -3,24 +3,33 @@
 """
 
 from abc import ABC, abstractmethod
+import time
 
+from loguru import logger
 from pynput import keyboard, mouse
 import mss.base
+import numpy as np
 
 from puterbot.models import Recording, InputEvent
 from puterbot.playback import play_input_event
 from puterbot.utils import get_screenshot
 
 
+MAX_FRAME_TIMES = 1000
+
+
 class BaseReplayStrategy(ABC):
 
     def __init__(
         self,
         recording: Recording,
+        max_frame_times: int = MAX_FRAME_TIMES,
     ):
         self.recording = recording
+        self.max_frame_times = max_frame_times
         self.input_events = []
         self.screenshots = []
+        self.frame_times = []
 
     @abstractmethod
     def get_next_input_event(
@@ -29,9 +38,7 @@ def get_next_input_event(
     ) -> InputEvent:
         pass
 
-    def run(
-        self,
-    ):
+    def run(self):
         keyboard_controller = keyboard.Controller()
         mouse_controller = mouse.Controller()
         while True:
@@ -41,10 +48,21 @@ def run(
                 input_event = self.get_next_input_event(screenshot)
             except StopIteration:
                 break
+            self.log_fps()
             self.input_events.append(input_event)
             if input_event:
                 play_input_event(
                     input_event,
                     mouse_controller,
                     keyboard_controller,
                 )
+
+    def log_fps(self):
+        t = time.time()
+        self.frame_times.append(t)
+        dts = np.diff(self.frame_times)
+        mean_dt = np.mean(dts)
+        fps = len(dts) / mean_dt
+        logger.info(f"{fps=:.2f}")
+        if len(self.frame_times) > self.max_frame_times:
+            self.frame_times.pop(0)