class DemoController(Controller):
iface = Any
timer = Any
snippet_source = Any
snippet_store = Any
plot = Any
container = Any
button = Button
tool = Any
def _button_fired(self):
hoops = self.tool.windows
coeffs = np.zeros((30, 3))
for i, hoop in enumerate(hoops):
x = round(hoop[0]*self.snippet_source.fs)
coeffs[x] = hoop[1], hoop[2], i+1
print x, hoop[1], hoop[2], i+1
#self.iface.set_coefficients('c_spike1', coeffs.ravel())
#self.plot.last_reset = len(self.snippet_store.buffer)
def _container_default(self):
container = OverlayPlotContainer(padding=[50, 50, 50, 50])
index_range = DataRange1D(low=0, high=0.0012)
value_range = DataRange1D(low=-0.00025, high=0.0005)
index_mapper = LinearMapper(range=index_range)
value_mapper = LinearMapper(range=value_range)
plot = SnippetChannelPlot(history=100, channel=self.snippet_store,
value_mapper=value_mapper, index_mapper=index_mapper)
self.plot = plot
# Add the axes
axis = PlotAxis(orientation='left', component=plot)
plot.overlays.append(axis)
axis = PlotAxis(orientation='bottom', component=plot)
plot.overlays.append(axis)
plot.overlays.append(ZoomTool(plot, axis='value'))
self.tool = WindowTool(component=plot)
plot.overlays.append(self.tool)
container.add(plot)
# plot = SnippetChannelPlot(history=100, channel=self.snippet_store,
# value_mapper=value_mapper, index_mapper=index_mapper,
# classifier=1, line_color='red')
# container.add(plot)
#
#plot = SnippetChannelPlot(history=100, channel=self.snippet_store,
# value_mapper=value_mapper, index_mapper=index_mapper,
# classifier=2, line_color='green')
#container.add(plot)
#plot = SnippetChannelPlot(history=5, channel=self.snippet_store,
# value_mapper=value_mapper, index_mapper=index_mapper,
# classifier=2, line_color='blue')
#container.add(plot)
print 'adding container'
return container
def _snippet_store_default(self):
datafile = get_temp_file()
return FileSnippetChannel(node=datafile.root, name='snippet',
snippet_size=30)
def init(self, info):
filename = join(RCX_ROOT, 'physiology')
self.iface = DSPCircuit(filename, 'RZ5')
self.iface.set_tag('a_spike1', 1e-4)
self.snippet_source = self.iface.get_buffer('spike1', 'r',
block_size=32)
self.snippet_store.fs = self.snippet_source.fs
self.iface.start()
self.iface.trigger('A', 'high')
self.timer = Timer(100, self.monitor)
def monitor(self):
data = self.snippet_source.read().reshape((-1, 32))
self.snippet_store.send(data[:,1:-1], data[:,0].view('int32'),
data[:,-1].view('int32'))
