Merge pull request #52 from Ipuch/chunks

fix: in any case stl name
pyomeca · Feb 7, 2025 · a7a40cf · a7a40cf
2 parents f9af2dc + fc7fffb
commit a7a40cf
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/pyorerun/biorbd_components/mesh.py b/pyorerun/biorbd_components/mesh.py
@@ -13,7 +13,10 @@ class TransformableMeshUpdater(Component):
     """
 
     def __init__(self, name: str, mesh: Trimesh, transform_callable: callable):
-        self.__name = name + "/" + mesh.metadata["file_name"]
+        filename = (
+            mesh.metadata["file_name"] if "file_name" in mesh.metadata else mesh.metadata["header"].replace(" ", "")
+        )
+        self.__name = name + "/" + filename
         self.__mesh = mesh
 
         self.transformed_mesh = mesh.copy()

diff --git a/pyorerun/biorbd_components/model_markers.py b/pyorerun/biorbd_components/model_markers.py
@@ -38,7 +38,7 @@ def compute_markers(self, q: np.ndarray) -> np.ndarray:
         nb_frames = q.shape[1]
         markers = np.zeros((3, self.nb_markers, nb_frames))
         for f in range(q.shape[1]):
-            markers[:, :, f] = self.callable_markers(q[:, f]).T
+            markers[:, :, f] = self.callable_markers(q[:, f]).squeeze().T
 
         return markers
 

diff --git a/pyorerun/multi_phase_rerun.py b/pyorerun/multi_phase_rerun.py
@@ -105,4 +105,4 @@ def rerun(self, server_name: str = "multi_phase_animation", notebook=False) -> N
                 )
 
                 more_phases_after_this_one = i < self.nb_phase - 1
-                rr_phase.rerun(init=False, clear_last_node=True)
+                rr_phase.rerun(init=False, clear_last_node=more_phases_after_this_one)