def start_rec(pos):
global is_started
is_started +=1
if is_started!=5:
return
print 'start_rec', pos, is_started
streams = [dev.streams[0] for dev in devs]
bound_indexes = {dev0.streams[0] : (pos-1000, pos+1000),
dev1.streams[0] : (pos-1000, pos+1000),
}
rec = RawDataRecording(streams, dirname, bound_indexes = bound_indexes)
rec.start()
def start_rec(self, dirname, now, bound_indexes={}):
assert self.recording == False
streams = []
for dev in self.devices:
streams.extend(dev.streams)
self.rec_engine = RawDataRecording(streams,
dirname,
annotations=self.get_annotations(),
bound_indexes=bound_indexes)
self.rec_engine.start()
self.actionPlay.setEnabled(False)
self.recording = True
if self.setup['options']['show_file_list']:
name = os.path.basename(dirname)
self.reclist_widget.add_rec(name, dirname, now, state='recording')
def start_rec(self, dirname, now,bound_indexes = {}):
assert self.recording==False
streams = [ ]
for dev in self.devices:
streams.extend(dev.streams)
self.rec_engine = RawDataRecording(streams, dirname,
annotations = self.get_annotations(),
bound_indexes = bound_indexes)
self.rec_engine.start()
self.actionPlay.setEnabled(False)
self.recording = True
if self.setup['options']['show_file_list']:
name = os.path.basename(dirname)
self.reclist_widget.add_rec(name, dirname, now, state = 'recording')
class MainWindow(QtGui.QMainWindow):
def __init__(self, parent = None, setup = None):
QtGui.QMainWindow.__init__(self, parent = parent)
self.settings = PickleSettings(applicationname=applicationname)
if setup is None:
setup = default_setup
filename = self.settings['last_setup_filname']
#~ print filename
#~ setup = json.load(open(filename))
if filename is not None and os.path.exists(filename):
try:
setup = json.load(open(filename))
except:
self.warning_setup(filename)
self.setWindowTitle(u'Simple acquisition system')
self.setWindowIcon(QtGui.QIcon(':/neurolabscope.png'))
self.setDockNestingEnabled(True)
self.setAnimated(False)
self.createActions()
#~ self.createMenus()
self.createToolBars()
self.statusbar = QtGui.QStatusBar()
self.setStatusBar(self.statusbar)
self.devices_conf = None
self.devices = [ ]
self.docks = [ ]
self.playing = False
self.recording = False
self.streamhandler = StreamHandler()
# patch options in setup in case of seomthing new:
d = setup['options']
setup['options'] = copy.deepcopy(default_setup['options'])
setup['options'].update(d)
#~ self.apply_setup(setup, filename = filename)
try:
self.apply_setup(setup, filename = filename)
except:
self.warning_setup(filename)
setup = default_setup
self.apply_setup(setup)
def createActions(self):
self.actionConf = QtGui.QAction(u'&Configure', self,
checkable = False,
icon =QtGui.QIcon(':/configure.png'),
)
self.actionConf.triggered.connect(self.open_configure)
self.actionPlay = QtGui.QAction(u'Play/Pause', self,
checkable = True,
icon =QtGui.QIcon(':/media-playback-start.png'),
enabled = False,
)
self.actionPlay.triggered.connect(self.play_pause)
self.actionRec = QtGui.QAction(u'&Rec', self,
checkable = True,
icon =QtGui.QIcon(':/media-record.png'),
enabled = False,
)
self.actionRec.triggered.connect(self.on_rec_button_pushed)
#~ def createMenus(self):
#~ pass
def createToolBars(self):
self.toolbar = QtGui.QToolBar()
self.toolbar.setToolButtonStyle(QtCore.Qt.ToolButtonTextUnderIcon)
self.addToolBar(self.toolbar)
self.toolbar.setIconSize(QtCore.QSize(60, 40))
self.toolbar.addAction(self.actionConf, )
self.toolbar.addAction(self.actionPlay, )
self.toolbar.addAction(self.actionRec, )
def closeEvent(self, event):
if self.recording:
event.ignore()
else:
if self.playing:
self.play_pause(play = False)
event.accept()
def warning_setup(self, filename = None):
text = u'Setup file contains errors.'
if filename is not None:
text += u' ({})'.format(filename)
self.warn(u'Setup file problem',text, )
def warn(self, title, text):
mb = QtGui.QMessageBox.warning(self, title,text,
QtGui.QMessageBox.Ok , QtGui.QMessageBox.Default | QtGui.QMessageBox.Escape,
QtGui.QMessageBox.NoButton)
def apply_setup(self, setup, filename = None):
# close old one
for dev in self.devices:
dev.close()
for dock in self.docks:
# do something in pyacq to close properly
if hasattr(dock.widget(), 'stop'):
print dock.widget()
dock.widget().stop()
self.removeDockWidget(dock)
self.streamhandler.reset()
# Devices
self.devices = [ ]
for dev_info in setup['devices']:
_class = dict_device_classes[dev_info['class']]
dev = _class(streamhandler = self.streamhandler)
dev.configure(subdevices = dev_info['subdevices'],
**dev_info['global_params'])
dev.initialize()
self.devices.append(dev)
# Views
self.docks = [ ]
for view in setup['views']:
_class = views_dict[view['class']]
widget = _class(stream = self.devices[view['device_num']].streams[view['subdevice_num']])
widget.set_params(**view['params'])
dock = QtGui.QDockWidget(view['name'])
dock.setWidget(widget)
self.docks.append(dock)
self.addDockWidget(QtCore.Qt.RightDockWidgetArea, dock)
self.actionPlay.setEnabled(True)
# Global options
self.setup = setup
# consistency
if self.setup['options']['recording_mode'] == 'on_trig':
print 'force generate with annotation'
self.setup['options']['filename_mode'] = 'Generate with annotations'
if self.setup['options']['filename_mode'] == 'Generate with annotations':
print 'force show_annotations'
self.setup['options']['show_annotations'] = True
# recording mode
if self.setup['options']['recording_mode'] == 'on_trig' and len(self.devices)!=1:
self.setup['options']['recording_mode'] == 'continuous'
self.warn('recording_mode', 'recording_mode = on_trig work only for 1 device')
if self.setup['options']['recording_mode'] == 'continuous':
pass
elif self.setup['options']['recording_mode'] == 'on_trig':
pass
if self.setup['options']['show_annotations']:
param_annotations = self.setup.get('param_annotations', None)
self.annotation_widget = AnnotationsWidget(param_annotations = param_annotations)
dock = QtGui.QDockWidget('Annotations')
dock.setWidget(self.annotation_widget)
self.docks.append(dock)
self.dock_annotations = dock
self.addDockWidget(QtCore.Qt.LeftDockWidgetArea, dock)
self.annotation_widget.setMaximumWidth(250) # FIXME : do better
for dock in self.docks:
if hasattr(dock.widget(), 'annotation_changed'):
dock.widget().annotation_changed.connect(self.annotation_widget.set_annotations)
else:
self.dock_annotations = None
self.annotation_widget= None
if self.setup['options']['show_file_list']:
self.reclist_widget = RecordingList()
dock = QtGui.QDockWidget('RecordingList')
dock.setWidget(self.reclist_widget)
self.docks.append(dock)
self.dock_reclist = dock
#~ self.addDockWidget(QtCore.Qt.BottomDockWidgetArea, dock)
self.addDockWidget(QtCore.Qt.LeftDockWidgetArea, dock)
p = self.setup['options']['recording_directory']
QtCore.QDir(p).mkpath(p)
if filename is not None:
self.statusbar.showMessage(filename)
else:
self.statusbar.showMessage('')
self.setup_filename = filename
def save_setup(self, filename):
if self.setup['options']['show_annotations']:
self.setup['param_annotations'] = self.annotation_widget.get_param_annotations()
for i, view in enumerate(self.setup['views']):
view['params'] = self.docks[i].widget().get_params()
try:
json.dump(self.setup,open(filename, 'wb'), indent=4, separators=(',', ': '))
except:
print 'erreur save setup'
def open_configure(self):
if self.setup['options']['show_annotations']:
self.setup['param_annotations'] = self.annotation_widget.get_param_annotations()
w = ConfigWindow(parent = self, setup = self.setup, setup_filename = self.setup_filename)
if w.exec_():
self.apply_setup(w.get_setup(), filename = w.setup_filename)
if w.setup_filename is not None:
self.settings['last_setup_filname'] = w.setup_filename
def play_pause(self, play = None):
if play is None:
self.playing = self.actionPlay.isChecked()
else:
self.playing = play
self.actionPlay.setChecked(play)
if self.playing:
# start devices
self.actionRec.setEnabled(True)
self.actionConf.setEnabled(False)
for dev in self.devices:
dev.start()
else:
# stop devices
self.actionRec.setEnabled(False)
self.actionConf.setEnabled(True)
if self.setup['options']['auto_save_setup_on_exit'] and self.settings['last_setup_filname'] is not None:
self.save_setup(self.settings['last_setup_filname'])
for dev in self.devices:
dev.stop()
def get_new_dirname(self, now = None):
basename = self.setup['options']['recording_directory']
if self.setup['options']['filename_mode'] == 'Ask on record':
fd = QtGui.QFileDialog( fileMode= QtGui.QFileDialog.AnyFile,
acceptMode = QtGui.QFileDialog.AcceptSave,
options = QtGui.QFileDialog.ShowDirsOnly)
fd.setDirectory(QtCore.QString(basename))
if fd.exec_():
dirname = unicode(fd.selectedFiles()[0])
os.mkdir(dirname)
else:
dirname = None
elif self.setup['options']['filename_mode'] == 'Generate with annotations':
name = now.strftime('%Y-%m-%d-%Hh%Mm%S,%fs')
name = name +'_'+ '_'.join([ '{}={}'.format(k,v) for k, v in self.annotation_widget.get_annotations().items()])
dirname = os.path.join(basename, name)
os.mkdir(dirname)
return dirname
def get_annotations(self):
if self.annotation_widget is None:
annotations = { }
else:
annotations = self.annotation_widget.get_annotations()
return annotations
def start_rec(self, dirname, now,bound_indexes = {}):
assert self.recording==False
streams = [ ]
for dev in self.devices:
streams.extend(dev.streams)
self.rec_engine = RawDataRecording(streams, dirname,
annotations = self.get_annotations(),
bound_indexes = bound_indexes)
self.rec_engine.start()
self.actionPlay.setEnabled(False)
self.recording = True
if self.setup['options']['show_file_list']:
name = os.path.basename(dirname)
self.reclist_widget.add_rec(name, dirname, now, state = 'recording')
def continuous_stop_rec(self):
assert self.recording==True
if self.setup['options']['auto_save_setup_on_exit'] and self.settings['last_setup_filname'] is not None:
self.save_setup(self.settings['last_setup_filname'])
self.rec_engine.stop()
self.actionPlay.setEnabled(True)
self.recording = False
if self.setup['options']['show_file_list']:
self.reclist_widget.list[-1]['state'] = 'finished'
def on_trigger_start_rec(self, pos):
if self.recording : return
now = datetime.datetime.now()
dirname = self.get_new_dirname(now = now)
l1, l2 = self.recontrig_limits
bound_indexes = { stream: (pos+l1, pos+l2) for stream in self.devices[0].streams }
self.start_rec(dirname, now,bound_indexes = bound_indexes)
# Patch for the moment because rec engine do not send signal
self.thread_waitstop = ThreadWaitStop(rec_engine = self.rec_engine)
self.thread_waitstop.rec_terminated.connect(self.on_rec_terminated)
self.thread_waitstop.start()
def on_rec_terminated(self):
self.thread_waitstop.wait()
self.thread_waitstop = None
self.rec_engine.stop()
self.recording = False
if self.setup['options']['show_file_list']:
self.reclist_widget.list[-1]['state'] = 'finished'
def on_rec_button_pushed(self):
# enable or start rec
rec_mode =self.setup['options']['recording_mode']
if self.actionRec.isChecked():
now = datetime.datetime.now()
if rec_mode == 'continuous':
self.setFocus()# patch tout force annotations parmas
dirname = self.get_new_dirname(now = now)
if dirname is None:
self.actionRec.setChecked(False)
return
else:
self.start_rec(dirname, now)
elif rec_mode == 'on_trig':
#TODO checks stream_num on apply_setup
kargs = dict(self.setup['rec_on_trigger'])
assert len(self.devices) == 1, 'Rec on trig support only one device for the moment'
dev = self.devices[0]
subdevice_num = kargs.pop('subdevice_num')
assert subdevice_num<=len(dev.streams), 'Rec on trig : wrong subdevice_num'
stream = dev.streams[subdevice_num]
assert kargs['channel']<dev.nb_channel, 'Rec on trig : wrong channel'
self.recontrig_limits = (np.array([kargs.pop('rec_start_time'), kargs.pop('rec_stop_time')])*stream.sampling_rate).astype(int)
if type(stream).__name__ == 'AnalogSignalSharedMemStream':
Trigger = AnalogTrigger
elif type(stream).__name__ == 'DigitalSignalSharedMemStream':
Trigger = DigitalTrigger
self.rec_engine = None
self.trigger_rec = Trigger(stream = stream,
callbacks = [ self.on_trigger_start_rec, ],
**kargs)
else:
# disable or stop rec
if rec_mode == 'continuous':
self.continuous_stop_rec()
elif rec_mode == 'on_trig':
self.trigger_rec.stop()
self.trigger_rec = None
if self.rec_engine is not None:
self.rec_engine.stop()
self.actionPlay.setEnabled(True)
else:
pass
def closeEvent (self, event):
if self.playing:
event.ignore()
else:
if self.setup['options']['auto_save_setup_on_exit'] and self.settings['last_setup_filname'] is not None:
self.save_setup(self.settings['last_setup_filname'])
event.accept()
def test1():
streamhandler = StreamHandler()
dev0 = FakeMultiSignals(streamhandler = streamhandler)
dev0.configure( #name = 'Analog fake 0',
nb_channel = 16,
sampling_rate =1000.,
buffer_length = 64,
packet_size = 10,
)
dev1 = FakeDigital(streamhandler = streamhandler)
dev1.configure( #name = u'Digital fake 1 éà',
nb_channel = 10,
sampling_rate =1000.,
buffer_length = 10.,
packet_size = 128,
)
devs = [dev0, dev1]
for dev in devs:
dev.initialize()
streams = [dev.streams[0] for dev in devs]
dirname_base = os.path.join(user.home, 'test_pyacq_recording')
if not os.path.exists(dirname_base):
os.mkdir(dirname_base)
dirname = os.path.join(dirname_base, 'rec {}'.format(datetime.datetime.now()))
if not os.path.exists(dirname):
os.mkdir(dirname)
rec = RawDataRecording(streams, dirname)
rec.start()
for dev in devs:
dev.start()
time.sleep(10.)
rec.stop()
# Stope and release the device
for dev in devs:
dev.stop()
dev.close()
# Read the files
reader = neo.RawBinarySignalIO(filename = os.path.join(dirname, 'fake 16 analog input.raw'))
seg = reader.read_segment(sampling_rate = 1.*pq.kHz,
t_start = 0.*pq.s,
unit = pq.V,
nbchannel = 16,
dtype = np.float32,
)
app = QtGui.QApplication([])
viewer = SegmentViewer(segment = seg)
#~ viewer = SignalViewer(analogsignals = seg.analogsignals, with_time_seeker = True)
viewer.show()
print seg.analogsignals[0].shape
analogsignals2 = [ ]
arr = np.memmap(filename = os.path.join(dirname, u'fake 10 digital input.raw'), mode = 'r', dtype = np.uint8)
arr = arr.reshape(-1, 2)
print arr.shape
for chan in range(10):
b = chan//8
mask = 1<<(chan%8)
sig = (arr[:,b]&mask>0).astype(float)
analogsignals2.append(neo.AnalogSignal(sig*pq.V, t_start = 0*pq.s, sampling_rate = 1*pq.kHz))
viewer2 = SignalViewer(analogsignals = analogsignals2, with_time_seeker = True)
viewer2.show()
app.exec_()
class MainWindow(QtGui.QMainWindow):
def __init__(self, parent=None, setup=None):
QtGui.QMainWindow.__init__(self, parent=parent)
self.settings = PickleSettings(applicationname=applicationname)
if setup is None:
setup = default_setup
filename = self.settings['last_setup_filname']
#~ print filename
#~ setup = json.load(open(filename))
if filename is not None and os.path.exists(filename):
try:
setup = json.load(open(filename))
except:
self.warning_setup(filename)
self.setWindowTitle(u'Simple acquisition system')
self.setWindowIcon(QtGui.QIcon(':/neurolabscope.png'))
self.setDockNestingEnabled(True)
self.setAnimated(False)
self.createActions()
#~ self.createMenus()
self.createToolBars()
self.statusbar = QtGui.QStatusBar()
self.setStatusBar(self.statusbar)
self.devices_conf = None
self.devices = []
self.docks = []
self.playing = False
self.recording = False
self.streamhandler = StreamHandler()
# patch options in setup in case of seomthing new:
d = setup['options']
setup['options'] = copy.deepcopy(default_setup['options'])
setup['options'].update(d)
#~ self.apply_setup(setup, filename = filename)
try:
self.apply_setup(setup, filename=filename)
except:
self.warning_setup(filename)
setup = default_setup
self.apply_setup(setup)
def createActions(self):
self.actionConf = QtGui.QAction(
u'&Configure',
self,
checkable=False,
icon=QtGui.QIcon(':/configure.png'),
)
self.actionConf.triggered.connect(self.open_configure)
self.actionPlay = QtGui.QAction(
u'Play/Pause',
self,
checkable=True,
icon=QtGui.QIcon(':/media-playback-start.png'),
enabled=False,
)
self.actionPlay.triggered.connect(self.play_pause)
self.actionRec = QtGui.QAction(
u'&Rec',
self,
checkable=True,
icon=QtGui.QIcon(':/media-record.png'),
enabled=False,
)
self.actionRec.triggered.connect(self.on_rec_button_pushed)
#~ def createMenus(self):
#~ pass
def createToolBars(self):
self.toolbar = QtGui.QToolBar()
self.toolbar.setToolButtonStyle(QtCore.Qt.ToolButtonTextUnderIcon)
self.addToolBar(self.toolbar)
self.toolbar.setIconSize(QtCore.QSize(60, 40))
self.toolbar.addAction(self.actionConf, )
self.toolbar.addAction(self.actionPlay, )
self.toolbar.addAction(self.actionRec, )
def closeEvent(self, event):
if self.recording:
event.ignore()
else:
if self.playing:
self.play_pause(play=False)
event.accept()
def warning_setup(self, filename=None):
text = u'Setup file contains errors.'
if filename is not None:
text += u' ({})'.format(filename)
self.warn(
u'Setup file problem',
text,
)
def warn(self, title, text):
mb = QtGui.QMessageBox.warning(
self, title, text, QtGui.QMessageBox.Ok,
QtGui.QMessageBox.Default | QtGui.QMessageBox.Escape,
QtGui.QMessageBox.NoButton)
def apply_setup(self, setup, filename=None):
# close old one
for dev in self.devices:
dev.close()
for dock in self.docks:
# do something in pyacq to close properly
if hasattr(dock.widget(), 'stop'):
print dock.widget()
dock.widget().stop()
self.removeDockWidget(dock)
self.streamhandler.reset()
# Devices
self.devices = []
for dev_info in setup['devices']:
_class = dict_device_classes[dev_info['class']]
dev = _class(streamhandler=self.streamhandler)
dev.configure(subdevices=dev_info['subdevices'],
**dev_info['global_params'])
dev.initialize()
self.devices.append(dev)
# Views
self.docks = []
for view in setup['views']:
_class = views_dict[view['class']]
widget = _class(stream=self.devices[view['device_num']].streams[
view['subdevice_num']])
widget.set_params(**view['params'])
dock = QtGui.QDockWidget(view['name'])
dock.setWidget(widget)
self.docks.append(dock)
self.addDockWidget(QtCore.Qt.RightDockWidgetArea, dock)
self.actionPlay.setEnabled(True)
# Global options
self.setup = setup
# consistency
if self.setup['options']['recording_mode'] == 'on_trig':
print 'force generate with annotation'
self.setup['options'][
'filename_mode'] = 'Generate with annotations'
if self.setup['options'][
'filename_mode'] == 'Generate with annotations':
print 'force show_annotations'
self.setup['options']['show_annotations'] = True
# recording mode
if self.setup['options']['recording_mode'] == 'on_trig' and len(
self.devices) != 1:
self.setup['options']['recording_mode'] == 'continuous'
self.warn('recording_mode',
'recording_mode = on_trig work only for 1 device')
if self.setup['options']['recording_mode'] == 'continuous':
pass
elif self.setup['options']['recording_mode'] == 'on_trig':
pass
if self.setup['options']['show_annotations']:
param_annotations = self.setup.get('param_annotations', None)
self.annotation_widget = AnnotationsWidget(
param_annotations=param_annotations)
dock = QtGui.QDockWidget('Annotations')
dock.setWidget(self.annotation_widget)
self.docks.append(dock)
self.dock_annotations = dock
self.addDockWidget(QtCore.Qt.LeftDockWidgetArea, dock)
self.annotation_widget.setMaximumWidth(250) # FIXME : do better
for dock in self.docks:
if hasattr(dock.widget(), 'annotation_changed'):
dock.widget().annotation_changed.connect(
self.annotation_widget.set_annotations)
else:
self.dock_annotations = None
self.annotation_widget = None
if self.setup['options']['show_file_list']:
self.reclist_widget = RecordingList()
dock = QtGui.QDockWidget('RecordingList')
dock.setWidget(self.reclist_widget)
self.docks.append(dock)
self.dock_reclist = dock
#~ self.addDockWidget(QtCore.Qt.BottomDockWidgetArea, dock)
self.addDockWidget(QtCore.Qt.LeftDockWidgetArea, dock)
p = self.setup['options']['recording_directory']
QtCore.QDir(p).mkpath(p)
if filename is not None:
self.statusbar.showMessage(filename)
else:
self.statusbar.showMessage('')
self.setup_filename = filename
def save_setup(self, filename):
if self.setup['options']['show_annotations']:
self.setup[
'param_annotations'] = self.annotation_widget.get_param_annotations(
)
for i, view in enumerate(self.setup['views']):
view['params'] = self.docks[i].widget().get_params()
try:
json.dump(self.setup,
open(filename, 'wb'),
indent=4,
separators=(',', ': '))
except:
print 'erreur save setup'
def open_configure(self):
if self.setup['options']['show_annotations']:
self.setup[
'param_annotations'] = self.annotation_widget.get_param_annotations(
)
w = ConfigWindow(parent=self,
setup=self.setup,
setup_filename=self.setup_filename)
if w.exec_():
self.apply_setup(w.get_setup(), filename=w.setup_filename)
if w.setup_filename is not None:
self.settings['last_setup_filname'] = w.setup_filename
def play_pause(self, play=None):
if play is None:
self.playing = self.actionPlay.isChecked()
else:
self.playing = play
self.actionPlay.setChecked(play)
if self.playing:
# start devices
self.actionRec.setEnabled(True)
self.actionConf.setEnabled(False)
for dev in self.devices:
dev.start()
else:
# stop devices
self.actionRec.setEnabled(False)
self.actionConf.setEnabled(True)
if self.setup['options'][
'auto_save_setup_on_exit'] and self.settings[
'last_setup_filname'] is not None:
self.save_setup(self.settings['last_setup_filname'])
for dev in self.devices:
dev.stop()
def get_new_dirname(self, now=None):
basename = self.setup['options']['recording_directory']
if self.setup['options']['filename_mode'] == 'Ask on record':
fd = QtGui.QFileDialog(fileMode=QtGui.QFileDialog.AnyFile,
acceptMode=QtGui.QFileDialog.AcceptSave,
options=QtGui.QFileDialog.ShowDirsOnly)
fd.setDirectory(QtCore.QString(basename))
if fd.exec_():
dirname = unicode(fd.selectedFiles()[0])
os.mkdir(dirname)
else:
dirname = None
elif self.setup['options'][
'filename_mode'] == 'Generate with annotations':
name = now.strftime('%Y-%m-%d-%Hh%Mm%S,%fs')
name = name + '_' + '_'.join([
'{}={}'.format(k, v)
for k, v in self.annotation_widget.get_annotations().items()
])
dirname = os.path.join(basename, name)
os.mkdir(dirname)
return dirname
def get_annotations(self):
if self.annotation_widget is None:
annotations = {}
else:
annotations = self.annotation_widget.get_annotations()
return annotations
def start_rec(self, dirname, now, bound_indexes={}):
assert self.recording == False
streams = []
for dev in self.devices:
streams.extend(dev.streams)
self.rec_engine = RawDataRecording(streams,
dirname,
annotations=self.get_annotations(),
bound_indexes=bound_indexes)
self.rec_engine.start()
self.actionPlay.setEnabled(False)
self.recording = True
if self.setup['options']['show_file_list']:
name = os.path.basename(dirname)
self.reclist_widget.add_rec(name, dirname, now, state='recording')
def continuous_stop_rec(self):
assert self.recording == True
if self.setup['options']['auto_save_setup_on_exit'] and self.settings[
'last_setup_filname'] is not None:
self.save_setup(self.settings['last_setup_filname'])
self.rec_engine.stop()
self.actionPlay.setEnabled(True)
self.recording = False
if self.setup['options']['show_file_list']:
self.reclist_widget.list[-1]['state'] = 'finished'
def on_trigger_start_rec(self, pos):
if self.recording: return
now = datetime.datetime.now()
dirname = self.get_new_dirname(now=now)
l1, l2 = self.recontrig_limits
bound_indexes = {
stream: (pos + l1, pos + l2)
for stream in self.devices[0].streams
}
self.start_rec(dirname, now, bound_indexes=bound_indexes)
# Patch for the moment because rec engine do not send signal
self.thread_waitstop = ThreadWaitStop(rec_engine=self.rec_engine)
self.thread_waitstop.rec_terminated.connect(self.on_rec_terminated)
self.thread_waitstop.start()
def on_rec_terminated(self):
self.thread_waitstop.wait()
self.thread_waitstop = None
self.rec_engine.stop()
self.recording = False
if self.setup['options']['show_file_list']:
self.reclist_widget.list[-1]['state'] = 'finished'
def on_rec_button_pushed(self):
# enable or start rec
rec_mode = self.setup['options']['recording_mode']
if self.actionRec.isChecked():
now = datetime.datetime.now()
if rec_mode == 'continuous':
self.setFocus() # patch tout force annotations parmas
dirname = self.get_new_dirname(now=now)
if dirname is None:
self.actionRec.setChecked(False)
return
else:
self.start_rec(dirname, now)
elif rec_mode == 'on_trig':
#TODO checks stream_num on apply_setup
kargs = dict(self.setup['rec_on_trigger'])
assert len(
self.devices
) == 1, 'Rec on trig support only one device for the moment'
dev = self.devices[0]
subdevice_num = kargs.pop('subdevice_num')
assert subdevice_num <= len(
dev.streams), 'Rec on trig : wrong subdevice_num'
stream = dev.streams[subdevice_num]
assert kargs[
'channel'] < dev.nb_channel, 'Rec on trig : wrong channel'
self.recontrig_limits = (np.array(
[kargs.pop('rec_start_time'),
kargs.pop('rec_stop_time')]) *
stream.sampling_rate).astype(int)
if type(stream).__name__ == 'AnalogSignalSharedMemStream':
Trigger = AnalogTrigger
elif type(stream).__name__ == 'DigitalSignalSharedMemStream':
Trigger = DigitalTrigger
self.rec_engine = None
self.trigger_rec = Trigger(stream=stream,
callbacks=[
self.on_trigger_start_rec,
],
**kargs)
else:
# disable or stop rec
if rec_mode == 'continuous':
self.continuous_stop_rec()
elif rec_mode == 'on_trig':
self.trigger_rec.stop()
self.trigger_rec = None
if self.rec_engine is not None:
self.rec_engine.stop()
self.actionPlay.setEnabled(True)
else:
pass
def closeEvent(self, event):
if self.playing:
event.ignore()
else:
if self.setup['options'][
'auto_save_setup_on_exit'] and self.settings[
'last_setup_filname'] is not None:
self.save_setup(self.settings['last_setup_filname'])
event.accept()
