|
import numpy as np |
|
|
|
class Joint(): |
|
def __init__(self, name, parent=None, children=None): |
|
self.name = name |
|
self.parent = parent |
|
self.children = children |
|
|
|
class MocapData(): |
|
def __init__(self): |
|
self.skeleton = {} |
|
self.values = None |
|
self.channel_names = [] |
|
self.framerate = 0.0 |
|
self.root_name = '' |
|
|
|
def traverse(self, j=None): |
|
stack = [self.root_name] |
|
while stack: |
|
joint = stack.pop() |
|
yield joint |
|
for c in self.skeleton[joint]['children']: |
|
stack.append(c) |
|
|
|
def clone(self): |
|
import copy |
|
new_data = MocapData() |
|
new_data.skeleton = copy.copy(self.skeleton) |
|
new_data.values = copy.copy(self.values) |
|
new_data.channel_names = copy.copy(self.channel_names) |
|
new_data.root_name = copy.copy(self.root_name) |
|
new_data.framerate = copy.copy(self.framerate) |
|
return new_data |
|
|
|
def get_all_channels(self): |
|
'''Returns all of the channels parsed from the file as a 2D numpy array''' |
|
|
|
frames = [f[1] for f in self.values] |
|
return np.asarray([[channel[2] for channel in frame] for frame in frames]) |
|
|
|
def get_skeleton_tree(self): |
|
tree = [] |
|
root_key = [j for j in self.skeleton if self.skeleton[j]['parent']==None][0] |
|
|
|
root_joint = Joint(root_key) |
|
|
|
def get_empty_channels(self): |
|
|
|
pass |
|
|
|
def get_constant_channels(self): |
|
|
|
pass |
|
|
