biosignal-device-interface 0.1.11b0__py3-none-any.whl → 0.1.32a1__py3-none-any.whl

biosignal_device_interface/gui/device_template_widgets/otb/otb_quattrocento_widget.py

@@ -0,0 +1,260 @@
+"""
+Quattrocento Widget class for GUI configuration
+of the Quattrocento from OT Bioelettronica.
+
+Developer: Dominik I. Braun
+Contact: dome.braun@fau.de
+Last Update: 2025-01-15
+"""
+
+from __future__ import annotations
+from functools import partial
+from typing import TYPE_CHECKING
+
+from biosignal_device_interface.gui.device_template_widgets.core.base_device_widget import (
+    BaseDeviceWidget,
+)
+from biosignal_device_interface.gui.ui_compiled.otb_quattrocento_template_widget import (
+    Ui_QuattrocentoForm,
+)
+from biosignal_device_interface.devices.otb.otb_quattrocento import (
+    OTBQuattrocento,
+)
+
+# Constants
+from biosignal_device_interface.constants.devices.otb.otb_quattrocento_constants import (
+    QuattrocentoAcqSettByte,
+    QuattrocentoAcquisitionMode,
+    QuattrocentoDecimMode,
+    QuattrocentoINXConf2Byte,
+    QuattrocentoRecordingMode,
+    QuattrocentoSamplingFrequencyMode,
+    QuattrocentoNumberOfChannelsMode,
+    QuattrocentoLowPassFilterMode,
+    QuattrocentoHighPassFilterMode,
+    QuattrocentoDetectionMode,
+)
+
+if TYPE_CHECKING:
+    from PySide6.QtWidgets import (
+        QMainWindow,
+        QWidget,
+        QGroupBox,
+        QPushButton,
+        QCheckBox,
+        QComboBox,
+        QLabel,
+        QLineEdit,
+    )
+
+
+class OTBQuattrocentoWidget(BaseDeviceWidget):
+    def __init__(self, parent: QWidget | QMainWindow | None = None):
+        super().__init__(parent)
+        self._set_device(OTBQuattrocento(self))
+
+    def _toggle_connection(self):
+        if not self._device.is_connected:
+            self._connect_push_button.setEnabled(False)
+
+        self._device.toggle_connection(
+            (
+                self._connection_ip_line_edit.text(),
+                int(self._connection_port_line_edit.text()),
+            ),
+        )
+
+    def _connection_toggled(self, is_connected: bool) -> None:
+        self._connect_push_button.setEnabled(True)
+        if is_connected:
+            self._connect_push_button.setText("Disconnect")
+            self._connect_push_button.setChecked(True)
+            self._configure_push_button.setEnabled(True)
+            self._connection_group_box.setEnabled(False)
+        else:
+            self._connect_push_button.setText("Connect")
+            self._connect_push_button.setChecked(False)
+            self._configure_push_button.setEnabled(False)
+            self._stream_push_button.setEnabled(False)
+            self._connection_group_box.setEnabled(True)
+
+        self.connect_toggled.emit(is_connected)
+
+    def _toggle_configuration(self):
+        self._device_params["grids"] = [
+            i
+            for i, check_box in enumerate(self._grid_selection_check_box_list)
+            if check_box.isChecked()
+        ]
+
+        # Acquisition settings
+        acq_sett_configuration = QuattrocentoAcqSettByte()
+        acq_sett_configuration.update(
+            decimation_mode=QuattrocentoDecimMode(
+                int(self._acquisition_decimator_check_box.isChecked()) + 1
+            ),
+            recording_mode=QuattrocentoRecordingMode(
+                int(self._acquisition_recording_check_box.isChecked()) + 1
+            ),
+            sampling_frequency_mode=QuattrocentoSamplingFrequencyMode(
+                self._acquisition_sampling_frequency_combo_box.currentIndex() + 1
+            ),
+            number_of_channels_mode=QuattrocentoNumberOfChannelsMode(
+                self._acquisition_number_of_channels_combo_box.currentIndex() + 1
+            ),
+        )
+        self._device_params["acq_sett_configuration"] = acq_sett_configuration
+
+        # Input settings
+        inx_configuration = QuattrocentoINXConf2Byte()
+        inx_configuration.update(
+            high_pass_filter=QuattrocentoHighPassFilterMode(
+                self._input_high_pass_filter_combo_box.currentIndex() + 1
+            ),
+            low_pass_filter=QuattrocentoLowPassFilterMode(
+                self._input_low_pass_filter_combo_box.currentIndex() + 1
+            ),
+            detection_mode=QuattrocentoDetectionMode(
+                self._input_detection_mode_combo_box.currentIndex() + 1
+            ),
+        )
+
+        self._device_params["input_top_left_configuration"] = inx_configuration
+        self._device_params["input_top_right_configuration"] = inx_configuration
+        self._device_params["multiple_input_one_configuration"] = inx_configuration
+        self._device_params["multiple_input_two_configuration"] = inx_configuration
+        self._device_params["multiple_input_three_configuration"] = inx_configuration
+        self._device_params["multiple_input_four_configuration"] = inx_configuration
+
+        self._device.configure_device(self._device_params)
+
+    def _configuration_toggled(self, is_configured: bool) -> None:
+        if is_configured:
+            self._stream_push_button.setEnabled(True)
+
+        self.configure_toggled.emit(is_configured)
+
+    def _toggle_stream(self) -> None:
+        self._stream_push_button.setEnabled(False)
+        self._device.toggle_streaming()
+
+    def _toggle_configuration_group_boxes(self, state: bool) -> None:
+        for group_box in self._configuration_group_boxes:
+            group_box.setEnabled(state)
+
+    def _stream_toggled(self, is_streaming):
+        self._stream_push_button.setEnabled(True)
+        if is_streaming:
+            self._stream_push_button.setText("Stop Streaming")
+            self._stream_push_button.setChecked(True)
+            self._configure_push_button.setEnabled(False)
+            self._toggle_configuration_group_boxes(False)
+        else:
+            self._stream_push_button.setText("Start Streaming")
+            self._stream_push_button.setChecked(False)
+            self._configure_push_button.setEnabled(True)
+            self._toggle_configuration_group_boxes(True)
+
+    def _initialize_device_params(self):
+        self._device_params: dict = {
+            "grids": [],
+            "acq_sett_configuration": QuattrocentoAcqSettByte(),
+            "input_top_left_configuration": QuattrocentoINXConf2Byte(),
+            "input_top_right_configuration": QuattrocentoINXConf2Byte(),
+            "multiple_input_one_configuration": QuattrocentoINXConf2Byte(),
+            "multiple_input_two_configuration": QuattrocentoINXConf2Byte(),
+            "multiple_input_three_configuration": QuattrocentoINXConf2Byte(),
+            "multiple_input_four_configuration": QuattrocentoINXConf2Byte(),
+        }
+
+    def _initialize_ui(self):
+        self.ui = Ui_QuattrocentoForm()
+        self.ui.setupUi(self)
+
+        # Command Push Buttons
+        self._connect_push_button: QPushButton = self.ui.commandConnectionPushButton
+        self._connect_push_button.clicked.connect(self._toggle_connection)
+        self._device.connect_toggled.connect(self._connection_toggled)
+
+        self._configure_push_button: QPushButton = (
+            self.ui.commandConfigurationPushButton
+        )
+        self._configure_push_button.clicked.connect(self._toggle_configuration)
+        self._configure_push_button.setEnabled(False)
+        self._device.configure_toggled.connect(self._configuration_toggled)
+
+        self._stream_push_button: QPushButton = self.ui.commandStreamPushButton
+        self._stream_push_button.clicked.connect(self._toggle_stream)
+        self._stream_push_button.setEnabled(False)
+        self._device.stream_toggled.connect(self._stream_toggled)
+
+        # Connection parameters
+        self._connection_group_box: QGroupBox = self.ui.connectionGroupBox
+        self._connection_ip_line_edit: QLineEdit = self.ui.connectionIPLineEdit
+        self._default_connection_ip: str = self._connection_ip_line_edit.text()
+        self._connection_ip_line_edit.editingFinished.connect(
+            partial(
+                self._check_ip_input,
+                self._connection_ip_line_edit,
+                self._default_connection_ip,
+            )
+        )
+        self._connection_port_line_edit: QLineEdit = self.ui.connectionPortLineEdit
+        self._default_connection_port: str = self._connection_port_line_edit.text()
+        self._connection_port_line_edit.editingFinished.connect(
+            partial(
+                self._check_port_input,
+                self._connection_port_line_edit,
+                self._default_connection_port,
+            )
+        )
+
+        # Acquisition parameters
+        self._acquisition_group_box: QGroupBox = self.ui.acquisitionGroupBox
+        self._acquisition_sampling_frequency_combo_box: QComboBox = (
+            self.ui.acquisitionSamplingFrequencyComboBox
+        )
+        self._acquisition_number_of_channels_combo_box: QComboBox = (
+            self.ui.acquisitionNumberOfChannelsComboBox
+        )
+        self._acquisition_decimator_check_box: QCheckBox = (
+            self.ui.acquisitionDecimatorCheckBox
+        )
+        self._acquisition_recording_check_box: QCheckBox = (
+            self.ui.acquisitionRecordingCheckBox
+        )
+
+        # Grid selection
+        self._grid_selection_group_box: QGroupBox = self.ui.gridSelectionGroupBox
+        self._grid_selection_check_box_list: list[QCheckBox] = [
+            self.ui.gridOneCheckBox,
+            self.ui.gridTwoCheckBox,
+            self.ui.gridThreeCheckBox,
+            self.ui.gridFourCheckBox,
+            self.ui.gridFiveCheckBox,
+            self.ui.gridSixCheckBox,
+        ]
+
+        [
+            check_box.setChecked(False)
+            for check_box in self._grid_selection_check_box_list
+        ]
+
+        # Input parameters
+        self._input_group_box: QGroupBox = self.ui.inputGroupBox
+        self._input_channel_combo_box: QComboBox = self.ui.inputChannelComboBox
+        self._input_low_pass_filter_combo_box: QComboBox = self.ui.inputLowPassComboBox
+        self._input_low_pass_default = self.ui.inputLowPassComboBox.currentIndex()
+        self._input_high_pass_filter_combo_box: QComboBox = (
+            self.ui.inputHighPassComboBox
+        )
+        self._input_high_pass_default = self.ui.inputHighPassComboBox.currentIndex()
+        self._input_detection_mode_combo_box: QComboBox = (
+            self.ui.inputDetectionModeComboBox
+        )
+
+        self._configuration_group_boxes: list[QGroupBox] = [
+            self._acquisition_group_box,
+            self._grid_selection_group_box,
+            self._input_group_box,
+        ]

biosignal_device_interface/gui/device_template_widgets/otb/otb_syncstation_widget.py

@@ -0,0 +1,262 @@
+"""
+Device class for real-time interfacing the OTB Syncstation device.
+Developer: Dominik I. Braun
+Contact: dome.braun@fau.de
+Last Update: 2025-01-09
+"""
+
+from __future__ import annotations
+from functools import partial
+from typing import TYPE_CHECKING, Dict
+
+from biosignal_device_interface.gui.device_template_widgets.core.base_device_widget import (
+    BaseDeviceWidget,
+)
+
+# Local Libraries
+from biosignal_device_interface.gui.ui_compiled.otb_syncstation_template_widget import (
+    Ui_SyncStationForm,
+)
+from biosignal_device_interface.devices.otb.otb_syncstation import OTBSyncStation
+
+# Constants
+from biosignal_device_interface.constants.devices.otb.otb_syncstation_constants import (
+    SyncStationDetectionMode,
+    SyncStationProbeConfigMode,
+    SyncStationRecOnMode,
+    SyncStationWorkingMode,
+)
+
+if TYPE_CHECKING:
+    from PySide6.QtWidgets import (
+        QMainWindow,
+        QWidget,
+        QGroupBox,
+        QPushButton,
+        QCheckBox,
+        QComboBox,
+        QLabel,
+        QTabWidget,
+    )
+
+
+class OTBSyncStationWidget(BaseDeviceWidget):
+    def __init__(self, parent: QWidget | QMainWindow | None = None):
+        super().__init__(parent)
+        self._set_device(OTBSyncStation(self))
+
+    def _toggle_connection(self):
+        if not self._device.is_connected:
+            self._command_connect_push_button.setEnabled(False)
+
+        self._device.toggle_connection(
+            (self._connection_ip_label.text(), int(self._connection_port_label.text())),
+        )
+
+    def _connection_toggled(self, is_connected: bool) -> None:
+        self._command_connect_push_button.setEnabled(True)
+        if is_connected:
+            self._command_connect_push_button.setText("Disconnect")
+            self._command_connect_push_button.setChecked(True)
+            self._command_configure_push_button.setEnabled(True)
+            self._connection_group_box.setEnabled(False)
+        else:
+            self._command_connect_push_button.setText("Connect")
+            self._command_connect_push_button.setChecked(False)
+            self._command_configure_push_button.setEnabled(False)
+            self._command_stream_push_button.setEnabled(False)
+            self._connection_group_box.setEnabled(True)
+
+        self.connect_toggled.emit(is_connected)
+
+    def _toggle_configuration(self) -> None:
+        self._device_params["working_mode"] = SyncStationWorkingMode(
+            self._input_working_mode_combo_box.currentIndex() + 1
+        )
+
+        count_enabled = 0
+        for key, value in self._probes_dict.items():
+            is_enabled = value["probe_status"].isChecked()
+            self._device_params["bytes_configuration_A"][key][
+                "probe_status"
+            ] = is_enabled
+            self._device_params["bytes_configuration_A"][key]["detection_mode"] = (
+                SyncStationDetectionMode(value["detection_mode"].currentIndex() + 1)
+            )
+
+            if is_enabled:
+                count_enabled += 1
+
+        self._device_params["number_of_probes"] = count_enabled
+
+        self._device.configure_device(self._device_params)
+
+    def _configuration_toggled(self, is_configured: bool) -> None:
+        if is_configured:
+            self._command_stream_push_button.setEnabled(True)
+
+        self.configure_toggled.emit(is_configured)
+
+    def _toggle_stream(self) -> None:
+        self._command_stream_push_button.setEnabled(False)
+        self._device.toggle_streaming()
+
+    def _stream_toggled(self, is_streaming: bool) -> None:
+        self._command_stream_push_button.setEnabled(True)
+        if is_streaming:
+            self._command_stream_push_button.setText("Stop Streaming")
+            self._command_stream_push_button.setChecked(True)
+            self._command_configure_push_button.setEnabled(False)
+            self._input_group_box.setEnabled(False)
+        else:
+            self._command_stream_push_button.setText("Start Streaming")
+            self._command_stream_push_button.setChecked(False)
+            self._command_configure_push_button.setEnabled(True)
+            self._input_group_box.setEnabled(True)
+
+        self.stream_toggled.emit(is_streaming)
+
+    def _update_probe_params(self, probe: SyncStationProbeConfigMode, state) -> None:
+        self._probes_dict[probe]["detection_mode"].setEnabled(state == 2)
+
+    def _initialize_device_params(self) -> None:
+        self._device_params = {
+            "rec_on_mode": SyncStationRecOnMode.OFF,
+            "working_mode": SyncStationWorkingMode.EMG,
+            "number_of_probes": 0,
+            "bytes_configuration_A": {
+                SyncStationProbeConfigMode(i): {
+                    "probe_status": False,
+                    "detection_mode": SyncStationDetectionMode.MONOPOLAR_GAIN_LOW,
+                }
+                for i in range(1, len(SyncStationProbeConfigMode))
+            },
+        }
+
+    def _set_default_probe_params(self) -> None:
+        for values in self._probes_dict.values():
+            values["detection_mode"].setCurrentIndex(1)
+            values["probe_status"].setChecked(True)
+            values["probe_status"].setChecked(False)
+
+        self._probes_dict[SyncStationProbeConfigMode.MUOVI_PROBE_ONE][
+            "probe_status"
+        ].setChecked(True)
+
+        self._probes_dict[SyncStationProbeConfigMode.MUOVI_PROBE_TWO][
+            "probe_status"
+        ].setChecked(True)
+
+    def _initialize_ui(self):
+        self.ui = Ui_SyncStationForm()
+        self.ui.setupUi(self)
+
+        # Command Push Buttons
+        self._command_connect_push_button: QPushButton = (
+            self.ui.commandConnectionPushButton
+        )
+        self._command_connect_push_button.clicked.connect(self._toggle_connection)
+        self._device.connect_toggled.connect(self._connection_toggled)
+
+        self._command_configure_push_button: QPushButton = (
+            self.ui.commandConfigurationPushButton
+        )
+        self._command_configure_push_button.clicked.connect(self._toggle_configuration)
+        self._command_configure_push_button.setEnabled(False)
+        self._device.configure_toggled.connect(self._configuration_toggled)
+
+        self._command_stream_push_button: QPushButton = self.ui.commandStreamPushButton
+        self._command_stream_push_button.clicked.connect(self._toggle_stream)
+        self._command_stream_push_button.setEnabled(False)
+        self._device.stream_toggled.connect(self._stream_toggled)
+
+        # Connection Paramters
+        self._connection_group_box: QGroupBox = self.ui.connectionGroupBox
+        self._connection_ip_label: QLabel = self.ui.connectionIPAddressLabel
+        self._connection_port_label: QLabel = self.ui.connectionPortLabel
+
+        # Input Parameters
+        self._input_group_box: QGroupBox = self.ui.inputGroupBox
+        self._input_working_mode_combo_box: QComboBox = self.ui.inputWorkingModeComboBox
+        self._input_working_mode_combo_box.setCurrentIndex(1)
+
+        self._probes_tab_widget: QTabWidget = self.ui.probesTabWidget
+        self._probes_tab_widget.setCurrentIndex(0)
+        self._probes_dict: Dict[
+            SyncStationProbeConfigMode, dict[str, QComboBox | QCheckBox]
+        ] = self._configure_probes_dict()
+
+        for key, value in self._probes_dict.items():
+            value["probe_status"].stateChanged.connect(
+                partial(self._update_probe_params, key)
+            )
+
+        self._set_default_probe_params()
+
+    def _configure_probes_dict(self) -> None:
+        return {
+            SyncStationProbeConfigMode.MUOVI_PROBE_ONE: {
+                "probe_status": self.ui.muoviProbeOneEnableCheckBox,
+                "detection_mode": self.ui.muoviProbeOneDetectionModeComboBox,
+            },
+            SyncStationProbeConfigMode.MUOVI_PROBE_TWO: {
+                "probe_status": self.ui.muoviProbeTwoEnableCheckBox,
+                "detection_mode": self.ui.muoviProbeTwoDetectionModeComboBox,
+            },
+            SyncStationProbeConfigMode.MUOVI_PROBE_THREE: {
+                "probe_status": self.ui.muoviProbeThreeEnableCheckBox,
+                "detection_mode": self.ui.muoviProbeThreeDetectionModeComboBox,
+            },
+            SyncStationProbeConfigMode.MUOVI_PROBE_FOUR: {
+                "probe_status": self.ui.muoviProbeFourEnableCheckBox,
+                "detection_mode": self.ui.muoviProbeFourDetectionModeComboBox,
+            },
+            SyncStationProbeConfigMode.MUOVI_PLUS_PROBE_ONE: {
+                "probe_status": self.ui.muoviPlusProbeOneEnableCheckBox,
+                "detection_mode": self.ui.muoviPlusProbeOneDetectionModeComboBox,
+            },
+            SyncStationProbeConfigMode.MUOVI_PLUS_PROBE_TWO: {
+                "probe_status": self.ui.muoviPlusProbeTwoEnableCheckBox,
+                "detection_mode": self.ui.muoviPlusProbeTwoDetectionModeComboBox,
+            },
+            SyncStationProbeConfigMode.DUE_PLUS_PROBE_ONE: {
+                "probe_status": self.ui.duePlusProbeOneEnableCheckBox,
+                "detection_mode": self.ui.duePlusProbeOneDetectionModeComboBox,
+            },
+            SyncStationProbeConfigMode.DUE_PLUS_PROBE_TWO: {
+                "probe_status": self.ui.duePlusProbeTwoEnableCheckBox,
+                "detection_mode": self.ui.duePlusProbeTwoDetectionModeComboBox,
+            },
+            SyncStationProbeConfigMode.DUE_PLUS_PROBE_THREE: {
+                "probe_status": self.ui.duePlusProbeThreeEnableCheckBox,
+                "detection_mode": self.ui.duePlusProbeThreeDetectionModeComboBox,
+            },
+            SyncStationProbeConfigMode.DUE_PLUS_PROBE_FOUR: {
+                "probe_status": self.ui.duePlusProbeFourEnableCheckBox,
+                "detection_mode": self.ui.duePlusProbeFourDetectionModeComboBox,
+            },
+            SyncStationProbeConfigMode.DUE_PLUS_PROBE_FIVE: {
+                "probe_status": self.ui.duePlusProbeFiveEnableCheckBox,
+                "detection_mode": self.ui.duePlusProbeFiveDetectionModeComboBox,
+            },
+            SyncStationProbeConfigMode.DUE_PLUS_PROBE_SIX: {
+                "probe_status": self.ui.duePlusProbeSixEnableCheckBox,
+                "detection_mode": self.ui.duePlusProbeSixDetectionModeComboBox,
+            },
+            SyncStationProbeConfigMode.DUE_PLUS_PROBE_SEVEN: {
+                "probe_status": self.ui.duePlusProbeSevenEnableCheckBox,
+                "detection_mode": self.ui.duePlusProbeSevenDetectionModeComboBox,
+            },
+            SyncStationProbeConfigMode.DUE_PLUS_PROBE_EIGHT: {
+                "probe_status": self.ui.duePlusProbeEightEnableCheckBox,
+                "detection_mode": self.ui.duePlusProbeEightDetectionModeComboBox,
+            },
+            SyncStationProbeConfigMode.DUE_PLUS_PROBE_NINE: {
+                "probe_status": self.ui.duePlusProbeNineEnableCheckBox,
+                "detection_mode": self.ui.duePlusProbeNineDetectionModeComboBox,
+            },
+            SyncStationProbeConfigMode.DUE_PLUS_PROBE_TEN: {
+                "probe_status": self.ui.duePlusProbeTenEnableCheckBox,
+                "detection_mode": self.ui.duePlusProbeTenDetectionModeComboBox,
+            },
+        }

biosignal_device_interface/gui/plot_widgets/biosignal_plot_widget.py

@@ -95,12 +95,16 @@ class BiosignalPlotWidget(QWidget):
     def configure(
         self,
         lines: int,
-        sampling_freuqency: int = 2000,
+        sampling_frequency: int = 2000,
+        plot_sampling_frequency: int | None = None,
         display_time: int = 10,
+        line_height: int = 50,
         background_color: np.ndarray = np.array([18.0, 18.0, 18.0, 1]),
     ):
         self.number_of_lines = lines
-        self.external_sampling_frequency = sampling_freuqency
+        self.external_sampling_frequency = sampling_frequency
+        if plot_sampling_frequency is not None:
+            self.internal_sampling_frequency_threshold = plot_sampling_frequency
 
         if (
             self.external_sampling_frequency
@@ -129,7 +133,7 @@ class BiosignalPlotWidget(QWidget):
         )
 
         self.space_for_each_line = min(
-            int(self.max_texture_size // self.number_of_lines // 1.5), 50
+            int(self.max_texture_size // self.number_of_lines // 1.5), line_height
         )
 
         self.canvas.configure(
@@ -343,7 +347,8 @@ class VispyFastPlotCanvas(app.Canvas):
             self.line_data[:, :-input_vertices] = self.line_data[:, input_vertices:]
             self.line_data[:, -input_vertices:] = input_data
 
-        self.program["a_position"].set_data(self.line_data.ravel().astype(np.float32))
+        plot_data = self.line_data.ravel().astype(np.float32)
+        self.program["a_position"].set_data(plot_data)
         self.update()
         self.context.flush()