def record(recordState,
amp_name,
amp_serial,
record_dir,
eeg_only,
recordLogger=logger,
queue=None):
redirect_stdout_to_queue(recordLogger, queue, 'INFO')
# set data file name
timestamp = time.strftime('%Y%m%d-%H%M%S', time.localtime())
pcl_file = "%s/%s-raw.pcl" % (record_dir, timestamp)
eve_file = '%s/%s-eve.txt' % (record_dir, timestamp)
recordLogger.info('>> Output file: %s' % (pcl_file))
# test writability
try:
qc.make_dirs(record_dir)
open(
pcl_file,
'w').write('The data will written when the recording is finished.')
except:
raise RuntimeError('Problem writing to %s. Check permission.' %
pcl_file)
# start a server for sending out data pcl_file when software trigger is used
outlet = start_server('StreamRecorderInfo', channel_format='string',\
source_id=eve_file, stype='Markers')
# connect to EEG stream server
sr = StreamReceiver(buffer_size=0,
amp_name=amp_name,
amp_serial=amp_serial,
eeg_only=eeg_only)
# start recording
recordLogger.info('\n>> Recording started (PID %d).' % os.getpid())
qc.print_c('\n>> Press Enter to stop recording', 'G')
tm = qc.Timer(autoreset=True)
next_sec = 1
while recordState.value == 1:
sr.acquire()
if sr.get_buflen() > next_sec:
duration = str(datetime.timedelta(seconds=int(sr.get_buflen())))
recordLogger.info('RECORDING %s' % duration)
next_sec += 1
tm.sleep_atleast(0.001)
# record stop
recordLogger.info('>> Stop requested. Copying buffer')
buffers, times = sr.get_buffer()
signals = buffers
events = None
# channels = total channels from amp, including trigger channel
data = {
'signals': signals,
'timestamps': times,
'events': events,
'sample_rate': sr.get_sample_rate(),
'channels': sr.get_num_channels(),
'ch_names': sr.get_channel_names(),
'lsl_time_offset': sr.lsl_time_offset
}
recordLogger.info('Saving raw data ...')
qc.save_obj(pcl_file, data)
recordLogger.info('Saved to %s\n' % pcl_file)
# automatically convert to fif and use event file if it exists (software trigger)
if os.path.exists(eve_file):
recordLogger.info('Found matching event file, adding events.')
else:
eve_file = None
recordLogger.info('Converting raw file into fif.')
pcl2fif(pcl_file, external_event=eve_file)
class RecordingFromHardWare():
def __init__(self):
self.start_recording = False
self.new_data = []
self.sr = StreamReceiver(buffer_size=0)
self.record_dir = '%s/records' % os.getcwd()
self.current_window = np.ndarray([])
self.current_time_stamps = np.ndarray([])
self.MRCP_window_size = 6
print("I RUMNNNNNNNNNNNNNNNNNNNNNNNNNNNNNN")
# =============================================================================
# def runInfinitecording(self):
# while self.start_recording:
# self.sr.acquire()
# if self.sr.get_buflen() > 20:
# duration = str(datetime.timedelta(seconds=int(self.sr.get_buflen())))
# #recordLogger.info('RECORDING %s' % duration)
# #next_sec += 1
# buffers, times = self.sr.get_buffer()
# signals = buffers
# events = None
#
# # channels = total channels from amp, including trigger channel
# data = {'signals':signals, 'timestamps':times, 'events':events,
# 'sample_rate':self.sr.get_sample_rate(), 'channels':self.sr.get_num_channels(),
# 'ch_names':self.sr.get_channel_names(), 'lsl_time_offset':self.sr.lsl_time_offset}
# print(data)
# self.new_data.append(data)
# =============================================================================
def startRecording(self):
# do some stuff
self.start_recording = True
#download_thread = threading.Thread(target=self.runInfinitecording)
recordLogger = logger
thread = threading.Thread(target=self.record)
thread.start()
def stopRecording(self):
self.start_recording = False
#Write into file
#self.new_data = []
def isRecordingIsRunning(self):
return self.start_recording
def set_MRCP_window_size(self, MRCP_window_size):
self.MRCP_window_size = MRCP_window_size
def record(self):
recordLogger = logger
#redirect_stdout_to_queue(recordLogger, 'INFO')
# set data file name
timestamp = time.strftime('%Y%m%d-%H%M%S', time.localtime())
pcl_file = "%s/%s-raw.pcl" % (self.record_dir, timestamp)
eve_file = '%s/%s-eve.txt' % (self.record_dir, timestamp)
recordLogger.info('>> Output file: %s' % (pcl_file))
# test writability
try:
qc.make_dirs(self.record_dir)
open(pcl_file, 'w').write(
'The data will written when the recording is finished.')
except:
raise RuntimeError('Problem writing to %s. Check permission.' %
pcl_file)
# start a server for sending out data pcl_file when software trigger is used
outlet = start_server('StreamRecorderInfo', channel_format='string',\
source_id=eve_file, stype='Markers')
# connect to EEG stream server
#sr = StreamReceiver(buffer_size=0, eeg_only=eeg_only)
# start recording
recordLogger.info('\n>> Recording started (PID %d).' % os.getpid())
qc.print_c('\n>> Press Enter to stop recording', 'G')
tm = qc.Timer(autoreset=True)
next_sec = 1
while self.start_recording:
self.sr.acquire()
if self.sr.get_buflen() > next_sec:
duration = str(
datetime.timedelta(seconds=int(self.sr.get_buflen())))
#recordLogger.info('RECORDING %s' % duration)
next_sec += 1
#current_buffer, current_times = self.sr.get_buffer()
self.sr.set_window_size(self.MRCP_window_size)
self.current_window, self.current_time_stamps = self.sr.get_window(
)
#print("window shape: {} time stamps shape: {}".format(self.current_window.shape, self.current_time_stamps.shape))
tm.sleep_atleast(0.001)
# record stop
recordLogger.info('>> Stop requested. Copying buffer')
buffers, times = self.sr.get_buffer()
signals = buffers
events = None
# channels = total channels from amp, including trigger channel
data = {
'signals': signals,
'timestamps': times,
'events': events,
'sample_rate': self.sr.get_sample_rate(),
'channels': self.sr.get_num_channels(),
'ch_names': self.sr.get_channel_names(),
'lsl_time_offset': self.sr.lsl_time_offset
}
print("data length : {}".format(data['signals'].shape))
self.new_data = data['signals']
recordLogger.info('Saving raw data ...')
qc.save_obj(pcl_file, data)
recordLogger.info('Saved to %s\n' % pcl_file)
# automatically convert to fif and use event file if it exists (software trigger)
if os.path.exists(eve_file):
recordLogger.info('Found matching event file, adding events.')
else:
eve_file = None
recordLogger.info('Converting raw file into fif.')
pcl2fif(pcl_file, external_event=eve_file)
import os
import mne
import numpy as np
import matplotlib.pyplot as plt
from pycnbi.stream_receiver.stream_receiver import StreamReceiver
import pycnbi.utils.pycnbi_utils as pu
import pycnbi.utils.q_common as qc
mne.set_log_level('ERROR')
# actually improves performance for multitaper
os.environ['OMP_NUM_THREADS'] = '1'
amp_name, amp_serial = pu.search_lsl()
stream_receiver = StreamReceiver(
window_size=1, buffer_size=1, amp_serial=amp_serial, eeg_only=False, amp_name=amp_name)
sfreq = stream_receiver.get_sample_rate()
watchdog = qc.Timer()
tm = qc.Timer(autoreset=True)
trg_ch = stream_receiver.get_trigger_channel()
last_ts = 0
qc.print_c('Trigger channel: %d' % trg_ch, 'G')
if SHOW_PSD:
psde = mne.decoding.PSDEstimator(sfreq=sfreq, fmin=1, fmax=50, bandwidth=None,
adaptive=False, low_bias=True, n_jobs=1, normalization='length', verbose=None)
sfreq = 512
interval = 1. / sfreq
fig, ax = plt.subplots(1, 1)
time_range = 1
x = np.arange(0, time_range, interval)
# We have 191 LEDs in the PlasticBrain
leds_values = [0] * 191
leds_values_index_for_test = 0
mne.set_log_level('ERROR')
# actually improves performance for multitaper
os.environ['OMP_NUM_THREADS'] = '1'
# Find a LSL stream to receive raw EEG data
amp_name, amp_serial = pu.search_lsl()
sr = StreamReceiver(window_size=1,
buffer_size=1,
amp_name=amp_name,
amp_serial=amp_serial,
eeg_only=True)
sfreq = sr.get_sample_rate()
watchdog = qc.Timer()
tm = qc.Timer(autoreset=True)
trg_ch = sr.get_trigger_channel()
last_ts = 0
# qc.print_c('Trigger channel: %d' % trg_ch, 'G')
# Frequency band for the band-pass filter
fmin = 1
fmax = 47
# Create an estimator of the power spectrum density in the frequency band
psde = mne.decoding.PSDEstimator(sfreq=sfreq,
fmin=fmin,
fmax=fmax,
bandwidth=None,
adaptive=False,