-
Notifications
You must be signed in to change notification settings - Fork 0
/
Myo_scope_edit.py
195 lines (179 loc) · 6.58 KB
/
Myo_scope_edit.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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
# -*- coding: utf-8 -*-
from traits.api import *
from traitsui.api import *
from chaco.api import Plot, AbstractPlotData, ArrayPlotData, VPlotContainer, Legend
from enable.api import Component, ComponentEditor
from pyface.timer.api import Timer
import time
import numpy as np
from myo import init, Hub, Feed, StreamEmg
import Myos_angle
class AnimationHandler(Handler):
def init(self, info):
super(AnimationHandler,self).init(info)
info.object.timer = Timer(120,info.object.on_timer)
def closed(self, info, is_ok):
super(AnimationHandler,self).closed(info,is_ok)
info.object.timer.Stop()
class MYOGrapher(HasTraits):
sEMG_length = Range(100,500,100)
Angle_length = Range(80,300,100)
sEMG_names = List(Str)
Angle_names = List(Str)
Which_Myo_sEMG = Enum(['myo1','myo2'])
select_emg_names = List(Str)
select_angle_names = List(Str)
plot_EMG = Instance(Plot)
plot_Angle = Instance(Plot)
data_EMG = Instance(ArrayPlotData)
data_Angle = Instance(ArrayPlotData)
stream_emg = List(Array)
stream_Angle = List(Array)
EMG_StreamWin = Array
Angle_StreamWin = Array
window_size = Range(5,25,8)
emg_posit = Int
angle_posit = Int
traits_view=View(
HSplit(# HSplit分为左右两个区域，中间有可调节宽度比例的调节手柄
# 左边再分为上下两个区域
VSplit(
VGroup(
Item("Which_Myo_sEMG", label=u"肌电信号源 "),
Item("sEMG_length", label=u"肌电横轴范围 " ),
Item("select_emg_names",style="custom", label=u"肌电通道选择 ",
editor=CheckListEditor(name="object.sEMG_names",
cols=2, format_str=u"%s")),
show_border = True, # 显示组的边框
scrollable = True, # 组中的控件过多时，采用滚动条
),
VGroup(
Item("Angle_length", label=u"角度横轴范围 "),
Item("select_angle_names",style="custom", label=u"选择手环 ",
editor=CheckListEditor(name="object.Angle_names",
cols=2, format_str=u"%s")),
show_border = True, # 显示组的边框
scrollable = True, # 组中的控件过多时，采用滚动条
),
Item("window_size", label=u"活动窗尺寸")
),
VGroup(
Item("plot_EMG", editor = ComponentEditor(),show_label=False,width=700, height=100),
Item("plot_Angle", editor = ComponentEditor(),show_label=False,width=700, height=100),
show_border = True, # 显示组的边框
)
),
resizable = True,
height = 500,
width = 1000,
title = u"MYO数据信号示波器",
handler = AnimationHandler()
)
def __init__(self,**traits):
super(MYOGrapher, self).__init__(**traits)
data_EMG = ArrayPlotData(x=[0],sEMG1=[0],sEMG2=[0],sEMG3=[0],
sEMG4=[0],sEMG5=[0],sEMG6=[0],sEMG7=[0],sEMG8=[0])
data_Angle = ArrayPlotData(x=[0],myo1=[0],myo2=[0])
plot_Angle = Plot(data_Angle)
plot_EMG = Plot(data_EMG)
plot_EMG.plot(("x","sEMG1"),type="line")
plot_EMG.title = "sEMG"
plot_EMG.value_scale = 'linear'
plot_Angle.plot(("x","myo1"),type="line")
plot_Angle.title = "Angle"
plot_Angle.value_scale = 'linear'
self.plot_EMG = plot_EMG
self.data_EMG = data_EMG
self.plot_Angle = plot_Angle
self.data_Angle = data_Angle
self.sEMG_names = ['sEMG1','sEMG2','sEMG3',
'sEMG4','sEMG5','sEMG6','sEMG7','sEMG8']
self.Angle_names = ['myo1','myo2']
self.emg_posit = 0
self.angle_posit = 0
_stream_emg = []
_stream_angle = []
for chanel in range(8):
_stream_emg.append(np.zeros(self.sEMG_length))
self.stream_emg = _stream_emg
for _myos_in in range(2):
_stream_angle.append(np.zeros(self.Angle_length))
self.stream_emg = _stream_emg
self.stream_Angle = _stream_angle
self.EMG_StreamWin = np.zeros(self.window_size)
self.Angle_StreamWin = np.zeros(self.window_size)
def _select_emg_names_changed(self):
for chanel in range(8):
self.stream_emg[chanel] = np.zeros(self.sEMG_length)
self.plot_EMG = Plot(self.data_EMG)
self.plot_EMG.auto_colors = ['green', 'lightgreen','blue',
'lightblue', 'red', 'pink', 'darkgray', 'silver']
for pic in self.select_emg_names:
self.plot_EMG.plot(('x',)+(pic,),name = pic,color = 'auto')
self.plot_EMG.title='Multiple sEMG'
legend = Legend(padding=10,align="ur")
legend.plots = self.plot_EMG.plots
self.plot_EMG.overlays.append(legend)
def _window_size_changed(self):
self.EMG_StreamWin = np.zeros(self.window_size)
self.Angle_StreamWin = np.zeros(self.window_size)
def _select_angle_names_changed(self):
for _myos_in in range(2):
self.stream_Angle[_myos_in] = np.zeros(self.Angle_length)
self.plot_Angle = Plot(self.data_Angle)
self.plot_Angle.auto_colors = ['blue','red']
for _pic in self.select_angle_names:
self.plot_Angle.plot(('x',)+(_pic,),name = _pic,color = 'auto')
self.plot_Angle.title='Multiple Angle'
_legend = Legend(padding=10,align="ur")
_legend.plots = self.plot_Angle.plots
self.plot_Angle.overlays.append(_legend)
@on_trait_change("sEMG_length, Which_Myo_sEMG")
def init_semgdata(self):
for chanel in range(8):
self.stream_emg[chanel] = np.zeros(self.sEMG_length)
@on_trait_change("Angle_length")
def init_angledata(self):
for in_myo in range(2):
self.stream_Angle[in_myo] = np.zeros(self.Angle_length)
def on_timer(self):
global myo_device
__myos = 0
if self.Which_Myo_sEMG == 'myo2':
__myos = 1
if len(self.select_emg_names)!= 0:
if self.emg_posit == self.sEMG_length - self.window_size:
self.emg_posit = 0
for name in self.select_emg_names:
index = self.sEMG_names.index(name)
self.EMG_StreamWin[0] = myo_device[__myos].emg[index]
self.stream_emg[index][self.emg_posit : self.emg_posit + self.window_size] = self.EMG_StreamWin
self.data_EMG[name] = self.stream_emg[index]
self.emg_posit += 1
self.data_EMG['x'] = range(self.sEMG_length)
if len(self.select_angle_names)!= 0:
if self.angle_posit == self.Angle_length - self.window_size:
self.angle_posit = 0
for _name1 in self.select_angle_names:
_index1 = self.Angle_names.index(_name1)
self.Angle_StreamWin[0] = Myos_angle.angle_init(myo_device,[0,0],[_index1])[0]
self.stream_Angle[_index1][self.angle_posit : self.angle_posit + self.window_size] = self.Angle_StreamWin
self.data_Angle[_name1] = self.stream_Angle[_index1]
self.angle_posit += 1
self.data_Angle['x'] = range(self.Angle_length)
if __name__ == "__main__":
init()
feed = Feed()
hub = Hub()
times = 0
hub.run(1000, feed)
try:
myo_device = feed.get_devices()
print(myo_device)
time.sleep(1)
myo_device[0].set_stream_emg(StreamEmg.enabled)
myo_device[1].set_stream_emg(StreamEmg.enabled)
MYO_Grapher = MYOGrapher()
MYO_Grapher.configure_traits()
finally:
hub.shutdown()