-
Notifications
You must be signed in to change notification settings - Fork 0
/
cp_consistency_test.py
54 lines (36 loc) · 1.12 KB
/
cp_consistency_test.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
# -*- coding: utf-8 -*-
"""
CP-project:
check trial data by overlay
@author: Jussi (jnu@iki.fi)
"""
import gaitutils
from cp_common import get_files
def _onpick(ev):
a = ev.artist
print '%s: cycle %s/%d' % (a._trialname, a._cycle.context, a._cycle.index)
subject = 'TD26'
cond = 'normal'
files = get_files(subject, cond)
pl = gaitutils.Plotter()
vars = [['HipAnglesX', 'KneeAnglesX', 'AnkleAnglesX'],
['PelvisAnglesX', 'PelvisAnglesY', 'PelvisAnglesZ'],
['ThoraxAnglesX', 'ThoraxAnglesY', 'ThoraxAnglesZ'],
['ShoulderAnglesX', 'ShoulderAnglesY', 'ShoulderAnglesZ']]
pl.layout = vars
cant_read = list()
for fn in files[:5]:
pl.open_trial(fn)
# check if model vars can be read
try:
pl.trial['LHipAnglesX']
except (gaitutils.GaitDataError, KeyError):
cant_read.append(fn)
continue
pl.plot_trial(model_cycles='all', show=False, superpose=True,
model_alpha=.5)
pl.set_title('%s: %s (%d files)' % (subject, cond, len(files)))
pl.fig.canvas.mpl_connect('pick_event', _onpick)
pl.show()
if cant_read:
print 'could not read: %s' % cant_read