98 lines
2.8 KiB
Python
98 lines
2.8 KiB
Python
|
|
from PyQt5.QtWidgets import QWidget, QPushButton, QSizePolicy, QLineEdit,\
|
|
QMessageBox, QVBoxLayout, QHBoxLayout, QGridLayout, QLabel, QFrame,\
|
|
QDoubleSpinBox
|
|
from PyQt5.QtCore import pyqtSlot
|
|
|
|
import numpy as np
|
|
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
|
|
from matplotlib.figure import Figure
|
|
|
|
|
|
class SpikeRedetectGui(QWidget):
|
|
|
|
def __init__(self, data_provider):
|
|
super().__init__()
|
|
self.data_provider = data_provider
|
|
self.title = 'Spike Redetection'
|
|
self.left = 10
|
|
self.top = 10
|
|
self.width = 640
|
|
self.height = 400
|
|
self.initUI()
|
|
|
|
def initUI(self):
|
|
self.setWindowTitle(self.title)
|
|
self.setGeometry(self.left, self.top, self.width, self.height)
|
|
|
|
# Middle:
|
|
middle = QHBoxLayout()
|
|
|
|
# Canvas for matplotlib figure
|
|
m = PlotCanvas(self, width=5, height=4)
|
|
m.move(0, 0)
|
|
middle.addWidget(m)
|
|
middle.addWidget(QVLine())
|
|
|
|
# Side (options) panel
|
|
panel = QFrame()
|
|
panel_layout = QVBoxLayout()
|
|
button = QPushButton('Button!', self)
|
|
button.setToolTip('A nice button!')
|
|
button.clicked.connect(m.plot)
|
|
panel_layout.addWidget(button)
|
|
|
|
threshold_label = QLabel("Threshold:")
|
|
panel_layout.addWidget(threshold_label)
|
|
threshold_spinbox = QDoubleSpinBox(self)
|
|
threshold_spinbox.setValue(2)
|
|
threshold_spinbox.setSingleStep(0.5)
|
|
threshold_spinbox.valueChanged.connect(lambda: m.plot(threshold_spinbox.value()))
|
|
panel_layout.addWidget(threshold_spinbox)
|
|
|
|
panel.setLayout(panel_layout)
|
|
middle.addWidget(panel)
|
|
|
|
self.setLayout(middle)
|
|
self.show()
|
|
|
|
|
|
class PlotCanvas(FigureCanvas):
|
|
|
|
def __init__(self, parent=None, width=5, height=4, dpi=100):
|
|
fig = Figure(figsize=(width, height), dpi=dpi)
|
|
self.axes = fig.add_subplot(111)
|
|
|
|
FigureCanvas.__init__(self, fig)
|
|
self.setParent(parent)
|
|
|
|
FigureCanvas.setSizePolicy(self,
|
|
QSizePolicy.Expanding,
|
|
QSizePolicy.Expanding)
|
|
FigureCanvas.updateGeometry(self)
|
|
self.plot()
|
|
|
|
@pyqtSlot()
|
|
def plot(self, mean=1):
|
|
x = np.arange(0, 1, 0.0001)
|
|
data = np.sin(x*np.pi*2*mean)
|
|
ax = self.axes
|
|
ax.clear()
|
|
ax.plot(data, 'r-')
|
|
ax.set_title('Sinus Example')
|
|
self.draw()
|
|
|
|
|
|
class QHLine(QFrame):
|
|
def __init__(self):
|
|
super(QHLine, self).__init__()
|
|
self.setFrameShape(QFrame.HLine)
|
|
self.setFrameShadow(QFrame.Sunken)
|
|
|
|
|
|
class QVLine(QFrame):
|
|
def __init__(self):
|
|
super(QVLine, self).__init__()
|
|
self.setFrameShape(QFrame.VLine)
|
|
self.setFrameShadow(QFrame.Sunken)
|