analysis/shift_bvh.py

from analysis.pymo.parsers import BVHParser
from analysis.pymo.data import Joint, MocapData
from analysis.pymo.preprocessing import *
from analysis.pymo.viz_tools import *
from analysis.pymo.writers import *
from sklearn.pipeline import Pipeline
from pathlib import Path
import sys
from feature_extraction.utils import distribute_tasks

p = BVHParser()
datas = []
filename = sys.argv[1]
shift_amount = float(sys.argv[2])
data = p.parse(filename)

data.values["Hips_Yposition"] += shift_amount

writer = BVHWriter()

with open(filename,'w') as f:
    writer.write(data, f)