diff --git a/trap/renderer.py b/trap/renderer.py index e08574f..04e0a06 100644 --- a/trap/renderer.py +++ b/trap/renderer.py @@ -342,6 +342,7 @@ class Renderer: def init_labels(self): base_color = (255,)*4 info_color = (255,255,0, 255) + info2_color = (255,0, 255, 255) options = [] for option in ['prediction_horizon','num_samples','full_dist','gmm_mode','z_mode', 'model_dir']: @@ -352,11 +353,11 @@ class Renderer: 'frame_idx': pyglet.text.Label("", x=20, y=self.window.height - 17, color=base_color, batch=self.batch_overlay), 'frame_time': pyglet.text.Label("t", x=120, y=self.window.height - 17, color=base_color, batch=self.batch_overlay), 'pred_idx': pyglet.text.Label("", x=90, y=self.window.height - 17, color=info_color, batch=self.batch_overlay), - 'pred_time': pyglet.text.Label("", x=200, y=self.window.height - 17, color=info_color, batch=self.batch_overlay), + 'frame_latency': pyglet.text.Label("", x=200, y=self.window.height - 17, color=info2_color, batch=self.batch_overlay), + 'pred_time': pyglet.text.Label("", x=300, y=self.window.height - 17, color=info_color, batch=self.batch_overlay), 'track_len': pyglet.text.Label("", x=500, y=self.window.height - 17, color=base_color, batch=self.batch_overlay), 'options1': pyglet.text.Label(options.pop(-1), x=20, y=30, color=base_color, batch=self.batch_overlay), 'options2': pyglet.text.Label(" | ".join(options), x=20, y=10, color=base_color, batch=self.batch_overlay), - } def refresh_labels(self, dt: float): @@ -364,7 +365,8 @@ class Renderer: if self.frame: self.labels['frame_idx'].text = f"{self.frame.index:06d}" - self.labels['frame_time'].text = f"{self.frame.time - self.first_time:.3f}s" + self.labels['frame_time'].text = f"{self.frame.time - self.first_time: >10.2f}s" + self.labels['frame_latency'].text = f"{self.frame.time - time.time():.2f}s" if self.prediction_frame: self.labels['pred_idx'].text = f"{self.prediction_frame.index - self.frame.index}"