class TFR_Viewer(): def __init__(self, power, baseline=None, mode='mean', tmin=None, tmax=None, fmin=None, fmax=None, vmin=None, vmax=None, cmap=None, dB=False, colorbar=True, x_label=None, y_label=None, picker=True,): app = QtWidgets.QApplication(sys.argv) window = QtWidgets.QMainWindow() self.ui = Ui_TFR_Viewer() self.ui.setupUi(window) self.connect_slots() if cmap is None: self.cmap = plt.cm.jet else: self.cmap = cmap self.tmin = tmin self.tmax = tmax self.colorbar = colorbar self.x_label = x_label self.y_label = y_label self.picker = picker times, self.freqs = power.times.copy(), power.freqs.copy() self.data = power.data self.tmin, self.tmax = times[0], times[-1] self.data, self.times, self.freqs, self.vmin, self.vmax = \ mne.time_frequency.tfr._preproc_tfr(self.data, times, self.freqs, tmin, tmax, fmin, fmax, mode, baseline, vmin, vmax, dB) self.ui.sb_channel.setMaximum(len(power.ch_names)-1) self.on_sb_channel_valueChanged(0) window.show() app.exec_() @QtCore.pyqtSlot(int) def on_sb_channel_valueChanged(self, chan): self.ui.mplg_tfr_viewer.all_sp_axes[0].clear() extent = (self.tmin, self.tmax, self.freqs[0], self.freqs[-1]) self.ui.mplg_tfr_viewer.all_sp_axes[0].imshow(self.data[chan, :, :], extent=extent, aspect="auto", origin="lower", vmin=self.vmin, vmax=self.vmax, picker=self.picker, cmap=self.cmap) if self.x_label is not None: plt.xlabel(self.x_label) if self.y_label is not None: plt.ylabel(self.y_label) #if self.colorbar: # plt.colorbar() self.ui.mplg_tfr_viewer.canvas.draw() def connect_slots(self): self.ui.sb_channel.valueChanged.connect(self.on_sb_channel_valueChanged)
def __init__( self, power, baseline=None, mode='mean', tmin=None, tmax=None, fmin=None, fmax=None, vmin=None, vmax=None, cmap=None, dB=False, colorbar=True, x_label=None, y_label=None, picker=True, ): app = QtGui.QApplication(sys.argv) window = QtGui.QMainWindow() self.ui = Ui_TFR_Viewer() self.ui.setupUi(window) self.connect_slots() if cmap is None: self.cmap = plt.cm.jet else: self.cmap = cmap self.tmin = tmin self.tmax = tmax self.colorbar = colorbar self.x_label = x_label self.y_label = y_label self.picker = picker times, self.freqs = power.times.copy(), power.freqs.copy() self.data = power.data self.tmin, self.tmax = times[0], times[-1] self.data, self.times, self.freqs, self.vmin, self.vmax = \ mne.time_frequency.tfr._preproc_tfr(self.data, times, self.freqs, tmin, tmax, fmin, fmax, mode, baseline, vmin, vmax, dB) self.ui.sb_channel.setMaximum(len(power.ch_names) - 1) self.on_sb_channel_valueChanged(0) window.show() app.exec_()
def __init__(self, power, baseline=None, mode='mean', tmin=None, tmax=None, fmin=None, fmax=None, vmin=None, vmax=None, cmap=None, dB=False, colorbar=True, x_label=None, y_label=None, picker=True,): app = QtWidgets.QApplication(sys.argv) window = QtWidgets.QMainWindow() self.ui = Ui_TFR_Viewer() self.ui.setupUi(window) self.connect_slots() if cmap is None: self.cmap = plt.cm.jet else: self.cmap = cmap self.tmin = tmin self.tmax = tmax self.colorbar = colorbar self.x_label = x_label self.y_label = y_label self.picker = picker times, self.freqs = power.times.copy(), power.freqs.copy() self.data = power.data self.tmin, self.tmax = times[0], times[-1] self.data, self.times, self.freqs, self.vmin, self.vmax = \ mne.time_frequency.tfr._preproc_tfr(self.data, times, self.freqs, tmin, tmax, fmin, fmax, mode, baseline, vmin, vmax, dB) self.ui.sb_channel.setMaximum(len(power.ch_names)-1) self.on_sb_channel_valueChanged(0) window.show() app.exec_()
class TFR_Viewer(): def __init__( self, power, baseline=None, mode='mean', tmin=None, tmax=None, fmin=None, fmax=None, vmin=None, vmax=None, cmap=None, dB=False, colorbar=True, x_label=None, y_label=None, picker=True, ): app = QtGui.QApplication(sys.argv) window = QtGui.QMainWindow() self.ui = Ui_TFR_Viewer() self.ui.setupUi(window) self.connect_slots() if cmap is None: self.cmap = plt.cm.jet else: self.cmap = cmap self.tmin = tmin self.tmax = tmax self.colorbar = colorbar self.x_label = x_label self.y_label = y_label self.picker = picker times, self.freqs = power.times.copy(), power.freqs.copy() self.data = power.data self.tmin, self.tmax = times[0], times[-1] self.data, self.times, self.freqs, self.vmin, self.vmax = \ mne.time_frequency.tfr._preproc_tfr(self.data, times, self.freqs, tmin, tmax, fmin, fmax, mode, baseline, vmin, vmax, dB) self.ui.sb_channel.setMaximum(len(power.ch_names) - 1) self.on_sb_channel_valueChanged(0) window.show() app.exec_() @QtCore.pyqtSlot(int) def on_sb_channel_valueChanged(self, chan): self.ui.mplg_tfr_viewer.all_sp_axes[0].clear() extent = (self.tmin, self.tmax, self.freqs[0], self.freqs[-1]) self.ui.mplg_tfr_viewer.all_sp_axes[0].imshow(self.data[chan, :, :], extent=extent, aspect="auto", origin="lower", vmin=self.vmin, vmax=self.vmax, picker=self.picker, cmap=self.cmap) if self.x_label is not None: plt.xlabel(self.x_label) if self.y_label is not None: plt.ylabel(self.y_label) #if self.colorbar: # plt.colorbar() self.ui.mplg_tfr_viewer.canvas.draw() def connect_slots(self): self.ui.sb_channel.valueChanged.connect( self.on_sb_channel_valueChanged)