abreza commited on
Commit
4b2c00a
·
1 Parent(s): bee6993

fix helper_keyframes

Browse files
Files changed (1) hide show
  1. visualization/logger.py +14 -6
visualization/logger.py CHANGED
@@ -143,6 +143,20 @@ class SimulationLogger:
143
  if not helper_keyframes:
144
  return
145
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
146
  for keyframe_idx, helper_keyframe in enumerate(helper_keyframes):
147
  try:
148
  position = vector3_to_numpy(helper_keyframe['position'])
@@ -168,12 +182,6 @@ class SimulationLogger:
168
  timeless=True
169
  )
170
 
171
- rr.log(
172
- f"world/helper_camera_{keyframe_idx}/point",
173
- rr.Points3D(colors=[1.0, 1.0, 0.0, 1.0]),
174
- timeless=True
175
- )
176
-
177
  except Exception as e:
178
  print(
179
  f"Error logging helper keyframe {keyframe_idx}: {str(e)}")
 
143
  if not helper_keyframes:
144
  return
145
 
146
+
147
+ helper_positions = np.array([
148
+ vector3_to_numpy(frame['position']) for frame in helper_keyframes
149
+ ])
150
+ rr.log(
151
+ "world/helper_keyframes",
152
+ rr.Points3D(
153
+ helper_positions,
154
+ colors=np.full((len(helper_positions), 4),
155
+ [1.0, 1.0, 0.0, 1.0])
156
+ ),
157
+ timeless=True
158
+ )
159
+
160
  for keyframe_idx, helper_keyframe in enumerate(helper_keyframes):
161
  try:
162
  position = vector3_to_numpy(helper_keyframe['position'])
 
182
  timeless=True
183
  )
184
 
 
 
 
 
 
 
185
  except Exception as e:
186
  print(
187
  f"Error logging helper keyframe {keyframe_idx}: {str(e)}")