|
import numpy as np |
|
import pandas as pd |
|
|
|
class BVHWriter(): |
|
def __init__(self): |
|
pass |
|
|
|
def write(self, X, ofile): |
|
|
|
|
|
ofile.write('HIERARCHY\n') |
|
|
|
self.motions_ = [] |
|
self._printJoint(X, X.root_name, 0, ofile) |
|
|
|
|
|
ofile.write('MOTION\n') |
|
ofile.write('Frames: %d\n'%X.values.shape[0]) |
|
ofile.write('Frame Time: %f\n'%X.framerate) |
|
|
|
|
|
self.motions_ = np.asarray(self.motions_).T |
|
lines = [" ".join(item) for item in self.motions_.astype(str)] |
|
ofile.write("".join("%s\n"%l for l in lines)) |
|
|
|
def _printJoint(self, X, joint, tab, ofile): |
|
|
|
if X.skeleton[joint]['parent'] == None: |
|
ofile.write('ROOT %s\n'%joint) |
|
elif len(X.skeleton[joint]['children']) > 0: |
|
ofile.write('%sJOINT %s\n'%('\t'*(tab), joint)) |
|
else: |
|
ofile.write('%sEnd site\n'%('\t'*(tab))) |
|
|
|
ofile.write('%s{\n'%('\t'*(tab))) |
|
|
|
ofile.write('%sOFFSET %3.5f %3.5f %3.5f\n'%('\t'*(tab+1), |
|
X.skeleton[joint]['offsets'][0], |
|
X.skeleton[joint]['offsets'][1], |
|
X.skeleton[joint]['offsets'][2])) |
|
channels = X.skeleton[joint]['channels'] |
|
n_channels = len(channels) |
|
|
|
if n_channels > 0: |
|
for ch in channels: |
|
self.motions_.append(np.asarray(X.values['%s_%s'%(joint, ch)].values)) |
|
|
|
if len(X.skeleton[joint]['children']) > 0: |
|
ch_str = ''.join(' %s'*n_channels%tuple(channels)) |
|
ofile.write('%sCHANNELS %d%s\n' %('\t'*(tab+1), n_channels, ch_str)) |
|
|
|
for c in X.skeleton[joint]['children']: |
|
self._printJoint(X, c, tab+1, ofile) |
|
|
|
ofile.write('%s}\n'%('\t'*(tab))) |
|
|
