wellton/bci-course
1
0
Fork 0
Código Issues Pull Requests Ações Pacotes Projetos Versões Wiki Atividade

Updated lsl-viewer.py to match new version in muse-lsl

Ver Código-Fonte
Esse commit está contido em:
jdpigeon
2017-06-30 09:51:42 -04:00
pai 4a8646a138
commit bff637a3a3
5 arquivos alterados com 115 adições e 85 exclusões
Baixar Arquivo de Patch Baixar Arquivo de Diff Expandir todos os arquivos Colapsar todos os arquivos
lab4/lsl-viewer.py
+23 -17
Ver Arquivo
@@ -1,17 +1,15 @@
#!/usr/bin/env python #!/usr/bin/env python
## code by Alexandre Barachant
import numpy as np import numpy as np
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from scipy.signal import butter, filtfilt from scipy.signal import butter, lfilter, lfilter_zi
from time import time, sleep from time import sleep
from pylsl import StreamInlet, resolve_byprop from pylsl import StreamInlet, resolve_byprop
from optparse import OptionParser
import seaborn as sns import seaborn as sns
from threading import Thread from threading import Thread
sns.set(style="whitegrid") sns.set(style="whitegrid")
from optparse import OptionParser
parser = OptionParser() parser = OptionParser()
parser.add_option("-w", "--window", parser.add_option("-w", "--window",
@@ -55,6 +53,7 @@ class LSLViewer():
self.dejitter = dejitter self.dejitter = dejitter
self.inlet = StreamInlet(stream, max_chunklen=buf) self.inlet = StreamInlet(stream, max_chunklen=buf)
self.filt = True self.filt = True
info = self.inlet.info() info = self.inlet.info()
description = info.desc() description = info.desc()
@@ -105,6 +104,9 @@ class LSLViewer():
self.bf, self.af = butter(4, np.array([1, 40])/(self.sfreq/2.), self.bf, self.af = butter(4, np.array([1, 40])/(self.sfreq/2.),
'bandpass') 'bandpass')
zi = lfilter_zi(self.bf, self.af)
self.filt_state = np.tile(zi, (self.n_chan, 1)).transpose()
self.data_f = np.zeros((self.n_samples, self.n_chan))
def update_plot(self): def update_plot(self):
k = 0 k = 0
@@ -112,33 +114,37 @@ class LSLViewer():
samples, timestamps = self.inlet.pull_chunk(timeout=1.0, samples, timestamps = self.inlet.pull_chunk(timeout=1.0,
max_samples=12) max_samples=12)
if timestamps: if timestamps:
self.data = np.vstack([self.data, samples])
if self.dejitter: if self.dejitter:
timestamps = np.float64(np.arange(len(timestamps))) timestamps = np.float64(np.arange(len(timestamps)))
timestamps /= self.sfreq timestamps /= self.sfreq
timestamps += self.times[-1] + 1./self.sfreq timestamps += self.times[-1] + 1./self.sfreq
self.times = np.concatenate([self.times, timestamps]) self.times = np.concatenate([self.times, timestamps])
self.n_samples = int(self.sfreq * self.window) self.n_samples = int(self.sfreq * self.window)
self.data = self.data[-self.n_samples:]
self.times = self.times[-self.n_samples:] self.times = self.times[-self.n_samples:]
self.data = np.vstack([self.data, samples])
self.data = self.data[-self.n_samples:]
filt_samples, self.filt_state = lfilter(
self.bf, self.af,
samples,
axis=0, zi=self.filt_state)
self.data_f = np.vstack([self.data_f, filt_samples])
self.data_f = self.data_f[-self.n_samples:]
k += 1 k += 1
if k == self.display_every: if k == self.display_every:
if self.filt:
data_f = filtfilt(self.bf, self.af, self.data, axis=0)
else:
data_f = self.data
data_f -= data_f.mean(axis=0)
if self.filt:
plot_data = self.data_f
elif not self.filt:
plot_data = self.data - self.data.mean(axis=0)
for ii in range(self.n_chan): for ii in range(self.n_chan):
self.lines[ii].set_xdata(self.times[::subsample] - self.lines[ii].set_xdata(self.times[::subsample] -
self.times[-1]) self.times[-1])
self.lines[ii].set_ydata(data_f[::subsample, ii] / self.lines[ii].set_ydata(plot_data[::subsample, ii] /
self.scale - ii) self.scale - ii)
impedances = np.std(plot_data, axis=0)
impedances = np.std(data_f, axis=0) ticks_labels = ['%s - %.2f' % (self.ch_names[ii],
ticks_labels = ['%s - %.2f' % impedances[ii])
(self.ch_names[ii], impedances[ii])
for ii in range(self.n_chan)] for ii in range(self.n_chan)]
self.axes.set_yticklabels(ticks_labels) self.axes.set_yticklabels(ticks_labels)
self.axes.set_xlim(-self.window, 0) self.axes.set_xlim(-self.window, 0)
lab5/lsl-viewer.py
+23 -17
Ver Arquivo
@@ -1,17 +1,15 @@
#!/usr/bin/env python #!/usr/bin/env python
## code by Alexandre Barachant
import numpy as np import numpy as np
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from scipy.signal import butter, filtfilt from scipy.signal import butter, lfilter, lfilter_zi
from time import time, sleep from time import sleep
from pylsl import StreamInlet, resolve_byprop from pylsl import StreamInlet, resolve_byprop
from optparse import OptionParser
import seaborn as sns import seaborn as sns
from threading import Thread from threading import Thread
sns.set(style="whitegrid") sns.set(style="whitegrid")
from optparse import OptionParser
parser = OptionParser() parser = OptionParser()
parser.add_option("-w", "--window", parser.add_option("-w", "--window",
@@ -55,6 +53,7 @@ class LSLViewer():
self.dejitter = dejitter self.dejitter = dejitter
self.inlet = StreamInlet(stream, max_chunklen=buf) self.inlet = StreamInlet(stream, max_chunklen=buf)
self.filt = True self.filt = True
info = self.inlet.info() info = self.inlet.info()
description = info.desc() description = info.desc()
@@ -105,6 +104,9 @@ class LSLViewer():
self.bf, self.af = butter(4, np.array([1, 40])/(self.sfreq/2.), self.bf, self.af = butter(4, np.array([1, 40])/(self.sfreq/2.),
'bandpass') 'bandpass')
zi = lfilter_zi(self.bf, self.af)
self.filt_state = np.tile(zi, (self.n_chan, 1)).transpose()
self.data_f = np.zeros((self.n_samples, self.n_chan))
def update_plot(self): def update_plot(self):
k = 0 k = 0
@@ -112,33 +114,37 @@ class LSLViewer():
samples, timestamps = self.inlet.pull_chunk(timeout=1.0, samples, timestamps = self.inlet.pull_chunk(timeout=1.0,
max_samples=12) max_samples=12)
if timestamps: if timestamps:
self.data = np.vstack([self.data, samples])
if self.dejitter: if self.dejitter:
timestamps = np.float64(np.arange(len(timestamps))) timestamps = np.float64(np.arange(len(timestamps)))
timestamps /= self.sfreq timestamps /= self.sfreq
timestamps += self.times[-1] + 1./self.sfreq timestamps += self.times[-1] + 1./self.sfreq
self.times = np.concatenate([self.times, timestamps]) self.times = np.concatenate([self.times, timestamps])
self.n_samples = int(self.sfreq * self.window) self.n_samples = int(self.sfreq * self.window)
self.data = self.data[-self.n_samples:]
self.times = self.times[-self.n_samples:] self.times = self.times[-self.n_samples:]
self.data = np.vstack([self.data, samples])
self.data = self.data[-self.n_samples:]
filt_samples, self.filt_state = lfilter(
self.bf, self.af,
samples,
axis=0, zi=self.filt_state)
self.data_f = np.vstack([self.data_f, filt_samples])
self.data_f = self.data_f[-self.n_samples:]
k += 1 k += 1
if k == self.display_every: if k == self.display_every:
if self.filt:
data_f = filtfilt(self.bf, self.af, self.data, axis=0)
else:
data_f = self.data
data_f -= data_f.mean(axis=0)
if self.filt:
plot_data = self.data_f
elif not self.filt:
plot_data = self.data - self.data.mean(axis=0)
for ii in range(self.n_chan): for ii in range(self.n_chan):
self.lines[ii].set_xdata(self.times[::subsample] - self.lines[ii].set_xdata(self.times[::subsample] -
self.times[-1]) self.times[-1])
self.lines[ii].set_ydata(data_f[::subsample, ii] / self.lines[ii].set_ydata(plot_data[::subsample, ii] /
self.scale - ii) self.scale - ii)
impedances = np.std(plot_data, axis=0)
impedances = np.std(data_f, axis=0) ticks_labels = ['%s - %.2f' % (self.ch_names[ii],
ticks_labels = ['%s - %.2f' % impedances[ii])
(self.ch_names[ii], impedances[ii])
for ii in range(self.n_chan)] for ii in range(self.n_chan)]
self.axes.set_yticklabels(ticks_labels) self.axes.set_yticklabels(ticks_labels)
self.axes.set_xlim(-self.window, 0) self.axes.set_xlim(-self.window, 0)
lab6/lsl-viewer.py
+23 -17
Ver Arquivo
@@ -1,17 +1,15 @@
#!/usr/bin/env python #!/usr/bin/env python
## code by Alexandre Barachant
import numpy as np import numpy as np
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from scipy.signal import butter, filtfilt from scipy.signal import butter, lfilter, lfilter_zi
from time import time, sleep from time import sleep
from pylsl import StreamInlet, resolve_byprop from pylsl import StreamInlet, resolve_byprop
from optparse import OptionParser
import seaborn as sns import seaborn as sns
from threading import Thread from threading import Thread
sns.set(style="whitegrid") sns.set(style="whitegrid")
from optparse import OptionParser
parser = OptionParser() parser = OptionParser()
parser.add_option("-w", "--window", parser.add_option("-w", "--window",
@@ -55,6 +53,7 @@ class LSLViewer():
self.dejitter = dejitter self.dejitter = dejitter
self.inlet = StreamInlet(stream, max_chunklen=buf) self.inlet = StreamInlet(stream, max_chunklen=buf)
self.filt = True self.filt = True
info = self.inlet.info() info = self.inlet.info()
description = info.desc() description = info.desc()
@@ -105,6 +104,9 @@ class LSLViewer():
self.bf, self.af = butter(4, np.array([1, 40])/(self.sfreq/2.), self.bf, self.af = butter(4, np.array([1, 40])/(self.sfreq/2.),
'bandpass') 'bandpass')
zi = lfilter_zi(self.bf, self.af)
self.filt_state = np.tile(zi, (self.n_chan, 1)).transpose()
self.data_f = np.zeros((self.n_samples, self.n_chan))
def update_plot(self): def update_plot(self):
k = 0 k = 0
@@ -112,33 +114,37 @@ class LSLViewer():
samples, timestamps = self.inlet.pull_chunk(timeout=1.0, samples, timestamps = self.inlet.pull_chunk(timeout=1.0,
max_samples=12) max_samples=12)
if timestamps: if timestamps:
self.data = np.vstack([self.data, samples])
if self.dejitter: if self.dejitter:
timestamps = np.float64(np.arange(len(timestamps))) timestamps = np.float64(np.arange(len(timestamps)))
timestamps /= self.sfreq timestamps /= self.sfreq
timestamps += self.times[-1] + 1./self.sfreq timestamps += self.times[-1] + 1./self.sfreq
self.times = np.concatenate([self.times, timestamps]) self.times = np.concatenate([self.times, timestamps])
self.n_samples = int(self.sfreq * self.window) self.n_samples = int(self.sfreq * self.window)
self.data = self.data[-self.n_samples:]
self.times = self.times[-self.n_samples:] self.times = self.times[-self.n_samples:]
self.data = np.vstack([self.data, samples])
self.data = self.data[-self.n_samples:]
filt_samples, self.filt_state = lfilter(
self.bf, self.af,
samples,
axis=0, zi=self.filt_state)
self.data_f = np.vstack([self.data_f, filt_samples])
self.data_f = self.data_f[-self.n_samples:]
k += 1 k += 1
if k == self.display_every: if k == self.display_every:
if self.filt:
data_f = filtfilt(self.bf, self.af, self.data, axis=0)
else:
data_f = self.data
data_f -= data_f.mean(axis=0)
if self.filt:
plot_data = self.data_f
elif not self.filt:
plot_data = self.data - self.data.mean(axis=0)
for ii in range(self.n_chan): for ii in range(self.n_chan):
self.lines[ii].set_xdata(self.times[::subsample] - self.lines[ii].set_xdata(self.times[::subsample] -
self.times[-1]) self.times[-1])
self.lines[ii].set_ydata(data_f[::subsample, ii] / self.lines[ii].set_ydata(plot_data[::subsample, ii] /
self.scale - ii) self.scale - ii)
impedances = np.std(plot_data, axis=0)
impedances = np.std(data_f, axis=0) ticks_labels = ['%s - %.2f' % (self.ch_names[ii],
ticks_labels = ['%s - %.2f' % impedances[ii])
(self.ch_names[ii], impedances[ii])
for ii in range(self.n_chan)] for ii in range(self.n_chan)]
self.axes.set_yticklabels(ticks_labels) self.axes.set_yticklabels(ticks_labels)
self.axes.set_xlim(-self.window, 0) self.axes.set_xlim(-self.window, 0)
lab7/recording/lsl-viewer.py
+23 -17
Ver Arquivo
@@ -1,17 +1,15 @@
#!/usr/bin/env python #!/usr/bin/env python
## code by Alexandre Barachant
import numpy as np import numpy as np
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from scipy.signal import butter, filtfilt from scipy.signal import butter, lfilter, lfilter_zi
from time import time, sleep from time import sleep
from pylsl import StreamInlet, resolve_byprop from pylsl import StreamInlet, resolve_byprop
from optparse import OptionParser
import seaborn as sns import seaborn as sns
from threading import Thread from threading import Thread
sns.set(style="whitegrid") sns.set(style="whitegrid")
from optparse import OptionParser
parser = OptionParser() parser = OptionParser()
parser.add_option("-w", "--window", parser.add_option("-w", "--window",
@@ -55,6 +53,7 @@ class LSLViewer():
self.dejitter = dejitter self.dejitter = dejitter
self.inlet = StreamInlet(stream, max_chunklen=buf) self.inlet = StreamInlet(stream, max_chunklen=buf)
self.filt = True self.filt = True
info = self.inlet.info() info = self.inlet.info()
description = info.desc() description = info.desc()
@@ -105,6 +104,9 @@ class LSLViewer():
self.bf, self.af = butter(4, np.array([1, 40])/(self.sfreq/2.), self.bf, self.af = butter(4, np.array([1, 40])/(self.sfreq/2.),
'bandpass') 'bandpass')
zi = lfilter_zi(self.bf, self.af)
self.filt_state = np.tile(zi, (self.n_chan, 1)).transpose()
self.data_f = np.zeros((self.n_samples, self.n_chan))
def update_plot(self): def update_plot(self):
k = 0 k = 0
@@ -112,33 +114,37 @@ class LSLViewer():
samples, timestamps = self.inlet.pull_chunk(timeout=1.0, samples, timestamps = self.inlet.pull_chunk(timeout=1.0,
max_samples=12) max_samples=12)
if timestamps: if timestamps:
self.data = np.vstack([self.data, samples])
if self.dejitter: if self.dejitter:
timestamps = np.float64(np.arange(len(timestamps))) timestamps = np.float64(np.arange(len(timestamps)))
timestamps /= self.sfreq timestamps /= self.sfreq
timestamps += self.times[-1] + 1./self.sfreq timestamps += self.times[-1] + 1./self.sfreq
self.times = np.concatenate([self.times, timestamps]) self.times = np.concatenate([self.times, timestamps])
self.n_samples = int(self.sfreq * self.window) self.n_samples = int(self.sfreq * self.window)
self.data = self.data[-self.n_samples:]
self.times = self.times[-self.n_samples:] self.times = self.times[-self.n_samples:]
self.data = np.vstack([self.data, samples])
self.data = self.data[-self.n_samples:]
filt_samples, self.filt_state = lfilter(
self.bf, self.af,
samples,
axis=0, zi=self.filt_state)
self.data_f = np.vstack([self.data_f, filt_samples])
self.data_f = self.data_f[-self.n_samples:]
k += 1 k += 1
if k == self.display_every: if k == self.display_every:
if self.filt:
data_f = filtfilt(self.bf, self.af, self.data, axis=0)
else:
data_f = self.data
data_f -= data_f.mean(axis=0)
if self.filt:
plot_data = self.data_f
elif not self.filt:
plot_data = self.data - self.data.mean(axis=0)
for ii in range(self.n_chan): for ii in range(self.n_chan):
self.lines[ii].set_xdata(self.times[::subsample] - self.lines[ii].set_xdata(self.times[::subsample] -
self.times[-1]) self.times[-1])
self.lines[ii].set_ydata(data_f[::subsample, ii] / self.lines[ii].set_ydata(plot_data[::subsample, ii] /
self.scale - ii) self.scale - ii)
impedances = np.std(plot_data, axis=0)
impedances = np.std(data_f, axis=0) ticks_labels = ['%s - %.2f' % (self.ch_names[ii],
ticks_labels = ['%s - %.2f' % impedances[ii])
(self.ch_names[ii], impedances[ii])
for ii in range(self.n_chan)] for ii in range(self.n_chan)]
self.axes.set_yticklabels(ticks_labels) self.axes.set_yticklabels(ticks_labels)
self.axes.set_xlim(-self.window, 0) self.axes.set_xlim(-self.window, 0)
lab9/lsl-viewer.py
+23 -17
Ver Arquivo
@@ -1,17 +1,15 @@
#!/usr/bin/env python #!/usr/bin/env python
## code by Alexandre Barachant
import numpy as np import numpy as np
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from scipy.signal import butter, filtfilt from scipy.signal import butter, lfilter, lfilter_zi
from time import time, sleep from time import sleep
from pylsl import StreamInlet, resolve_byprop from pylsl import StreamInlet, resolve_byprop
from optparse import OptionParser
import seaborn as sns import seaborn as sns
from threading import Thread from threading import Thread
sns.set(style="whitegrid") sns.set(style="whitegrid")
from optparse import OptionParser
parser = OptionParser() parser = OptionParser()
parser.add_option("-w", "--window", parser.add_option("-w", "--window",
@@ -55,6 +53,7 @@ class LSLViewer():
self.dejitter = dejitter self.dejitter = dejitter
self.inlet = StreamInlet(stream, max_chunklen=buf) self.inlet = StreamInlet(stream, max_chunklen=buf)
self.filt = True self.filt = True
info = self.inlet.info() info = self.inlet.info()
description = info.desc() description = info.desc()
@@ -105,6 +104,9 @@ class LSLViewer():
self.bf, self.af = butter(4, np.array([1, 40])/(self.sfreq/2.), self.bf, self.af = butter(4, np.array([1, 40])/(self.sfreq/2.),
'bandpass') 'bandpass')
zi = lfilter_zi(self.bf, self.af)
self.filt_state = np.tile(zi, (self.n_chan, 1)).transpose()
self.data_f = np.zeros((self.n_samples, self.n_chan))
def update_plot(self): def update_plot(self):
k = 0 k = 0
@@ -112,33 +114,37 @@ class LSLViewer():
samples, timestamps = self.inlet.pull_chunk(timeout=1.0, samples, timestamps = self.inlet.pull_chunk(timeout=1.0,
max_samples=12) max_samples=12)
if timestamps: if timestamps:
self.data = np.vstack([self.data, samples])
if self.dejitter: if self.dejitter:
timestamps = np.float64(np.arange(len(timestamps))) timestamps = np.float64(np.arange(len(timestamps)))
timestamps /= self.sfreq timestamps /= self.sfreq
timestamps += self.times[-1] + 1./self.sfreq timestamps += self.times[-1] + 1./self.sfreq
self.times = np.concatenate([self.times, timestamps]) self.times = np.concatenate([self.times, timestamps])
self.n_samples = int(self.sfreq * self.window) self.n_samples = int(self.sfreq * self.window)
self.data = self.data[-self.n_samples:]
self.times = self.times[-self.n_samples:] self.times = self.times[-self.n_samples:]
self.data = np.vstack([self.data, samples])
self.data = self.data[-self.n_samples:]
filt_samples, self.filt_state = lfilter(
self.bf, self.af,
samples,
axis=0, zi=self.filt_state)
self.data_f = np.vstack([self.data_f, filt_samples])
self.data_f = self.data_f[-self.n_samples:]
k += 1 k += 1
if k == self.display_every: if k == self.display_every:
if self.filt:
data_f = filtfilt(self.bf, self.af, self.data, axis=0)
else:
data_f = self.data
data_f -= data_f.mean(axis=0)
if self.filt:
plot_data = self.data_f
elif not self.filt:
plot_data = self.data - self.data.mean(axis=0)
for ii in range(self.n_chan): for ii in range(self.n_chan):
self.lines[ii].set_xdata(self.times[::subsample] - self.lines[ii].set_xdata(self.times[::subsample] -
self.times[-1]) self.times[-1])
self.lines[ii].set_ydata(data_f[::subsample, ii] / self.lines[ii].set_ydata(plot_data[::subsample, ii] /
self.scale - ii) self.scale - ii)
impedances = np.std(plot_data, axis=0)
impedances = np.std(data_f, axis=0) ticks_labels = ['%s - %.2f' % (self.ch_names[ii],
ticks_labels = ['%s - %.2f' % impedances[ii])
(self.ch_names[ii], impedances[ii])
for ii in range(self.n_chan)] for ii in range(self.n_chan)]
self.axes.set_yticklabels(ticks_labels) self.axes.set_yticklabels(ticks_labels)
self.axes.set_xlim(-self.window, 0) self.axes.set_xlim(-self.window, 0)