Spaces:
Runtime error
Runtime error
fix helper_keyframes
Browse files- 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)}")
|