Draw the finish time + optimize

racer/constants.py

@@ -1,2 +1,3 @@
 framerate = 60
 rounds = 3
+frames_after_finish = 25_000

racer/game_state.py

@@ -8,6 +8,7 @@
 from .car import Car
 from .car_info import CarInfo
 from .cars import car1
+from .constants import rounds
 from .track import Track
 
 
@@ -35,6 +36,9 @@ def reset(self):
     def update(self, dt: float):
         self.frames += 1
         for bot, car_info in self.bots.items():
+            if car_info.round >= rounds:
+                continue
+
             result, cpu = self.get_bot_commands(bot, car_info)
 
             car_info.cpu += cpu

racer/window.py

@@ -12,7 +12,7 @@
 from pygame_widgets.button import Button
 from pygame_widgets.dropdown import Dropdown
 
-from .constants import framerate
+from .constants import framerate, rounds
 from .game_state import GameState
 from .linear_math import Transform
 from .track import Track
@@ -135,7 +135,10 @@ def draw(self, clock):
             self.window.blit(text, (self.window.get_width() - 300, offset_from_top + i * 20))
 
             # Draw behind time
-            text = self.font.render(f'{behind:+.2f}', True, Color('white'))
+            if car_info.round >= rounds:
+                text = self.font.render(f'{car_info.waypoint_timing[-1]:.3f}', True, Color('white'))
+            else:
+                text = self.font.render(f'{behind:+.3f}', True, Color('white'))
             self.window.blit(text, (self.window.get_width() - 50, offset_from_top + i * 20))
 
     def selected_track(self):

tournament.py

@@ -10,7 +10,7 @@
 import pygame
 import tqdm
 
-from racer.constants import framerate, rounds
+from racer.constants import framerate, rounds, frames_after_finish
 from racer.game_state import GameState
 from racer.track import Track
 from racer.tracks import all_tracks
@@ -56,7 +56,6 @@ def main(track):
 
 def single_game(track):
     min_frames = 6000
-    frames_after_finish = 25_000
 
     game_state = GameState(track)
     finishing = False