remove_joints update bvh rot_order

UPC-ViRVIG · Apr 28, 2024 · 855cd69 · 855cd69
1 parent f12c0da
commit 855cd69
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/pymotion/io/bvh.py b/pymotion/io/bvh.py
@@ -317,11 +317,10 @@ def remove_joints(self, delete_joints: list[int]):
 
         # Update data
         self.data["names"] = [self.data["names"][i] for i in keep_joints]
+        self.data["rot_order"] = self.data["rot_order"][..., keep_joints, :]
         self.data["positions"] = new_pos
         self.data["rotations"] = np.degrees(
-            quat.to_euler(
-                new_rots, order=np.tile(self.data["rot_order"][..., keep_joints, :], (rots.shape[0], 1, 1))
-            )
+            quat.to_euler(new_rots, order=np.tile(self.data["rot_order"], (rots.shape[0], 1, 1)))
         )
         self.data["parents"] = new_parents
         self.data["offsets"] = new_offsets

diff --git a/pyproject.toml b/pyproject.toml
@@ -10,7 +10,7 @@ exclude = ["*test*"]
 
 [project]
 name = "upc-pymotion"
-version = "0.1.4"
+version = "0.1.5"
 description = "A Python library for working with motion data in NumPy or PyTorch."
 readme = "README.md"
 authors = [{ name = "Jose Luis Ponton", email = "[email protected]" }]