class Config(SignalNode.Config):
"""Config widget displayed for LSLInput."""
def __init__(self, parent=None):
super().__init__(parent=parent)
self.smoothing_factor = QDoubleSpinBox()
self.smoothing_factor.setMinimum(0)
self.smoothing_factor.setMaximum(1)
self.smoothing_factor.setSingleStep(0.1)
self.smoothing_factor.setPrefix("x")
self.smoothing_factor.valueChanged.connect(self.updateModel)
self.method = QComboBox()
self.method.addItem("Rectification")
self.method.addItem("Fourier Transform")
self.method.addItem("Hilbert Transform")
self.method.addItem("cFIR")
self.method.currentTextChanged.connect(self.updateModel)
self.smoother_type = QComboBox()
for name in EnvelopeDetector.smoother_name_to_type:
self.smoother_type.addItem(name)
self.smoother_type.currentTextChanged.connect(self.updateModel)
layout = QFormLayout()
self.setLayout(layout)
layout.addRow("Smoothing factor", self.smoothing_factor)
layout.addRow("Method", self.method)
layout.addRow("Smoother type", self.smoother_type)
def updateModel(self):
n = self.node()
if n is None:
return
smoothing_factor = self.smoothing_factor.value()
method = self.method.currentText()
smoother_type = n.smoother_name_to_type[self.smoother_type.currentText()]
n.setSmoothingFactor(smoothing_factor)
n.setMethod(method)
n.setSmootherType(smoother_type)
def updateView(self):
n = self.node()
if n is None:
return
self.smoothing_factor.blockSignals(True)
self.method.blockSignals(True)
self.smoother_type.blockSignals(True)
self.smoothing_factor.setValue(n.smoothingFactor())
self.method.setCurrentText(n.method())
self.smoother_type.setCurrentText(n.smoother_type_to_name[n.smootherType()])
self.smoothing_factor.blockSignals(False)
self.method.blockSignals(False)
self.smoother_type.blockSignals(False)
class Config(SignalNode.Config):
"""Config widget displayed for LSLInput."""
def __init__(self, parent=None):
super().__init__(parent=parent)
# Add a new layout
self.average = QDoubleSpinBox()
self.average.setMaximum(sys.float_info.max) # TODO: proper max
self.average.valueChanged.connect(self.updateModel)
self.standard_deviation = QDoubleSpinBox()
self.standard_deviation.setMaximum(
sys.float_info.max) # TODO: proper max
self.standard_deviation.valueChanged.connect(self.updateModel)
layout = QFormLayout()
self.setLayout(layout)
layout.addRow("Average", self.average)
layout.addRow("Std. Deviation", self.standard_deviation)
def updateModel(self):
n = self.node()
if n is None:
return
average = self.average.value()
standard_deviation = self.standard_deviation.value()
n.setAverage(average)
n.setStandardDeviation(standard_deviation)
def updateView(self):
n = self.node()
if n is None:
return
self.average.blockSignals(True)
self.standard_deviation.blockSignals(True)
self.average.setValue(n.average())
self.standard_deviation.setValue(n.standardDeviation())
self.average.blockSignals(False)
self.standard_deviation.blockSignals(False)
class Config(SignalNode.Config):
"""Config widget displayed for BandpassFilter."""
def __init__(self, parent=None):
super().__init__(parent=parent)
# Upper bound ----------------------------------------------------------------------------------------------
self.lower_bound_enable = QCheckBox()
self.lower_bound_enable.setChecked(True)
self.lower_bound_enable.stateChanged.connect(self.updateModel)
self.lower_bound = QDoubleSpinBox()
self.lower_bound.valueChanged.connect(self.updateModel)
self.lower_bound.setMinimum(0)
self.lower_bound.setMaximum(250)
self.lower_bound.setSuffix(" Hz")
layout = QHBoxLayout()
layout.setContentsMargins(0, 0, 0, 0)
lower_bound_widget = QWidget()
lower_bound_widget.setContentsMargins(0, 0, 0, 0)
lower_bound_widget.setLayout(layout)
layout.addWidget(self.lower_bound_enable)
layout.addWidget(self.lower_bound)
# Lower bound ----------------------------------------------------------------------------------------------
self.upper_bound_enable = QCheckBox()
self.upper_bound_enable.setChecked(True)
self.upper_bound_enable.stateChanged.connect(self.updateModel)
self.upper_bound = QDoubleSpinBox()
self.upper_bound.valueChanged.connect(self.updateModel)
self.upper_bound.setMinimum(0)
self.upper_bound.setMaximum(250)
self.upper_bound.setSuffix(" Hz")
layout = QHBoxLayout()
layout.setContentsMargins(0, 0, 0, 0)
upper_bound_widget = QWidget()
upper_bound_widget.setContentsMargins(0, 0, 0, 0)
upper_bound_widget.setLayout(layout)
layout.addWidget(self.upper_bound_enable)
layout.addWidget(self.upper_bound)
# Filter type and length -----------------------------------------------------------------------------------
self.filter_type = QComboBox()
for name in BandpassFilter.filter_name_to_type:
self.filter_type.addItem(name)
self.filter_type.currentTextChanged.connect(self.updateModel)
self.filter_length = QSpinBox()
self.filter_length.setMinimum(2)
self.filter_length.setMaximum(1000000)
self.filter_length.setValue(1000)
self.filter_length.valueChanged.connect(self.updateModel)
self.filter_order = QSpinBox()
self.filter_order.setRange(1, 4)
self.filter_order.valueChanged.connect(self.updateModel)
# ----------------------------------------------------------------------------------------------------------
layout = QFormLayout()
layout.addRow("Lower bound:", lower_bound_widget)
layout.addRow("Upper bound:", upper_bound_widget)
layout.addRow("Filter type:", self.filter_type)
layout.addRow("Filter order:", self.filter_order)
layout.addRow("Filter length:", self.filter_length)
self.setLayout(layout)
def updateModel(self):
n = self.node()
if n is None:
return
if self.lower_bound_enable.isChecked():
lower_bound = self.lower_bound.value()
else:
lower_bound = None
if self.upper_bound_enable.isChecked():
upper_bound = self.upper_bound.value()
else:
upper_bound = None
filter_type = n.filter_name_to_type[self.filter_type.currentText()]
filter_length = self.filter_length.value()
filter_order = self.filter_order.value()
n.setLowerBound(lower_bound)
n.setUpperBound(upper_bound)
n.setFilterType(filter_type)
n.setFilterLength(filter_length)
n.setFilterOrder(filter_order)
def updateView(self):
n = self.node()
if n is None:
return
# Prevent view fields from emitting signals while they are updated
self.lower_bound.blockSignals(True)
self.upper_bound.blockSignals(True)
self.filter_type.blockSignals(True)
self.filter_length.blockSignals(True)
self.filter_order.blockSignals(True)
if n.upperBound() is None:
self.upper_bound_enable.setChecked(False)
else:
self.upper_bound_enable.setChecked(True)
self.upper_bound.setValue(n.upperBound())
if n.lowerBound() is None:
self.lower_bound_enable.setChecked(False)
else:
self.lower_bound_enable.setChecked(True)
self.lower_bound.setValue(n.lowerBound())
self.filter_type.setCurrentText(
n.filter_type_to_name[n.filterType()])
self.filter_length.setValue(n.filterLength())
self.filter_order.setValue(n.filterOrder())
# Release the block and call adjust
self.lower_bound.blockSignals(False)
self.upper_bound.blockSignals(False)
self.filter_type.blockSignals(False)
self.filter_length.blockSignals(False)
self.filter_order.blockSignals(False)
self._adjust()
def _adjust(self):
"""Adjust displayed values and limits in response to changes."""
# Enable spinbox widgets based on their checkbox
self.lower_bound.setEnabled(self.lower_bound_enable.isChecked())
self.upper_bound.setEnabled(self.upper_bound_enable.isChecked())
# Adjust min and max so that lower_bound is never higher than upper_bound
if self.lower_bound_enable.isChecked():
self.upper_bound.setMinimum(self.lower_bound.value())
else:
self.upper_bound.setMinimum(0)
if self.upper_bound_enable.isChecked():
self.lower_bound.setMaximum(self.upper_bound.value())
else:
self.lower_bound.setMaximum(250)
if self.filter_type.currentText() == "Butterworth":
self.filter_order.setEnabled(True)
else:
self.filter_order.setEnabled(False)
