Merge pull request #17 from yanchxx/yanchxx-patch-1

fix time_col length bug for converting to csv
OlafHaag · Sep 19, 2022 · 83b23d0 · 83b23d0
2 parents 35451c8 + 21a461d
commit 83b23d0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/bvhtoolbox/convert/bvh2csv.py b/src/bvhtoolbox/convert/bvh2csv.py
@@ -52,7 +52,7 @@ def write_joint_rotations(bvh_tree, filepath):
     :return: If the write process was successful or not.
     :rtype: bool
     """
-    time_col = np.arange(0, bvh_tree.nframes*bvh_tree.frame_time, bvh_tree.frame_time)[:, None]
+    time_col = np.arange(0, (bvh_tree.nframes - 0.5)*bvh_tree.frame_time, bvh_tree.frame_time)[:, None]
     data_list = [time_col]
     header = ['time']
     for joint in bvh_tree.get_joints():
@@ -84,7 +84,7 @@ def write_joint_positions(bvh_tree, filepath, scale=1.0, end_sites=False):
     :return: If the write process was successful or not.
     :rtype: bool
     """
-    time_col = np.arange(0, bvh_tree.nframes * bvh_tree.frame_time, bvh_tree.frame_time)[:, None]
+    time_col = np.arange(0, (bvh_tree.nframes - 0.5) * bvh_tree.frame_time, bvh_tree.frame_time)[:, None]
     data_list = [time_col]
     header = ['time']
     root = next(bvh_tree.root.filter('ROOT'))