Update visualization/logger.py

visualization/logger.py

@@ -17,7 +17,7 @@ def create_subject_box(frame: Dict, dimensions: Dict) -> Dict[str, np.ndarray]:
 class SimulationLogger:
     def __init__(self):
         rr.init("camera_simulation")
-        rr.log("world", rr.ViewCoordinates.RIGHT_HAND_Y_UP, timeless=True)
+        rr.log("world", rr.ViewCoordinates.RIGHT_HAND_Y_UP)
 
         self.K = np.array([
             [500, 0, 960],
@@ -40,7 +40,7 @@ class SimulationLogger:
                 f"  Subject Index: {inst.get('subjectIndex', 'N/A')}"
                 for i, inst in enumerate(instructions)
             ])
-        ), timeless=True)
+        ))
 
     def log_subjects(self, subjects: List[Dict[str, Any]], frame_idx: int = 0) -> None:
         if not subjects:
@@ -90,8 +90,7 @@ class SimulationLogger:
                     camera_positions,
                     colors=np.full((len(camera_positions), 4),
                                    [0.0, 0.8, 0.8, 1.0])
-                ),
-                timeless=True
+                )
             )
 
             if len(camera_positions) > 1:
@@ -102,17 +101,21 @@ class SimulationLogger:
                     rr.LineStrips3D(
                         lines,
                         colors=[(0.0, 0.8, 0.8, 1.0)]
-                    ),
-                    timeless=True
+                    )
                 )
 
         except Exception as e:
             print(f"Error logging camera trajectory: {str(e)}")
 
-    def log_camera_frames(self, camera_frames: List[Dict[str, Any]], subjects: List[Dict[str, Any]]) -> None:
+    def log_simulation(self, camera_frames: List[Dict[str, Any]], subjects: List[Dict[str, Any]]) -> None:
+        """Main method to log both camera frames and subjects"""
         if not camera_frames:
             return
 
+        # First log the camera trajectory
+        self.log_camera_trajectory(camera_frames)
+
+        # Then log each frame with its corresponding subject positions
         for frame_idx, camera_frame in enumerate(camera_frames):
             try:
                 rr.set_time_sequence("frame_idx", frame_idx)
@@ -156,8 +159,7 @@ class SimulationLogger:
                 helper_positions,
                 radii=np.full(len(helper_positions), 0.03),
                 colors=np.full((len(helper_positions), 4), [1.0, 1.0, 0.0, 1.0]),
-            ),
-            timeless=True
+            )
         )
 
         for keyframe_idx, helper_keyframe in enumerate(helper_keyframes):
@@ -171,8 +173,7 @@ class SimulationLogger:
                         translation=position,
                         rotation=rr.Quaternion(xyzw=rotation_q),
                         scale=(.5, .5, .5)
-                    ),
-                    timeless=True
+                    )
                 )
 
                 rr.log(