biosignal-device-interface 0.1.311__py3-none-any.whl → 0.2.1a1__py3-none-any.whl

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

@@ -347,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()
 

biosignal_device_interface/gui/ui/otb_quattrocento_template_widget.ui

@@ -0,0 +1,415 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<ui version="4.0">
+ <class>QuattrocentoForm</class>
+ <widget class="QWidget" name="QuattrocentoForm">
+  <property name="geometry">
+   <rect>
+    <x>0</x>
+    <y>0</y>
+    <width>400</width>
+    <height>638</height>
+   </rect>
+  </property>
+  <property name="windowTitle">
+   <string>Form</string>
+  </property>
+  <layout class="QGridLayout" name="gridLayout_2">
+   <item row="0" column="0">
+    <widget class="QWidget" name="widget" native="true">
+     <layout class="QGridLayout" name="gridLayout_6">
+      <item row="2" column="0">
+       <widget class="QGroupBox" name="inputGroupBox">
+        <property name="title">
+         <string>Input Parameters</string>
+        </property>
+        <layout class="QGridLayout" name="gridLayout_4">
+         <item row="0" column="0">
+          <widget class="QLabel" name="label_3">
+           <property name="text">
+            <string>Select Channel</string>
+           </property>
+          </widget>
+         </item>
+         <item row="3" column="1">
+          <widget class="QComboBox" name="inputDetectionModeComboBox">
+           <item>
+            <property name="text">
+             <string>MONOPOLAR</string>
+            </property>
+           </item>
+           <item>
+            <property name="text">
+             <string>DIFFERENTIAL</string>
+            </property>
+           </item>
+           <item>
+            <property name="text">
+             <string>BIPOLAR</string>
+            </property>
+           </item>
+          </widget>
+         </item>
+         <item row="3" column="0">
+          <widget class="QLabel" name="label_10">
+           <property name="text">
+            <string>Mode</string>
+           </property>
+          </widget>
+         </item>
+         <item row="2" column="0">
+          <widget class="QLabel" name="label_9">
+           <property name="text">
+            <string>Low Pass</string>
+           </property>
+          </widget>
+         </item>
+         <item row="1" column="1">
+          <widget class="QComboBox" name="inputHighPassComboBox">
+           <property name="currentIndex">
+            <number>1</number>
+           </property>
+           <item>
+            <property name="text">
+             <string>0.7 Hz</string>
+            </property>
+           </item>
+           <item>
+            <property name="text">
+             <string>10 Hz</string>
+            </property>
+           </item>
+           <item>
+            <property name="text">
+             <string>100 Hz</string>
+            </property>
+           </item>
+           <item>
+            <property name="text">
+             <string>200 Hz</string>
+            </property>
+           </item>
+          </widget>
+         </item>
+         <item row="1" column="0">
+          <widget class="QLabel" name="label_8">
+           <property name="text">
+            <string>High Pass</string>
+           </property>
+          </widget>
+         </item>
+         <item row="2" column="1">
+          <widget class="QComboBox" name="inputLowPassComboBox">
+           <property name="currentIndex">
+            <number>1</number>
+           </property>
+           <item>
+            <property name="text">
+             <string>130 Hz</string>
+            </property>
+           </item>
+           <item>
+            <property name="text">
+             <string>500 Hz</string>
+            </property>
+           </item>
+           <item>
+            <property name="text">
+             <string>900 Hz</string>
+            </property>
+           </item>
+           <item>
+            <property name="text">
+             <string>4400 Hz</string>
+            </property>
+           </item>
+          </widget>
+         </item>
+         <item row="0" column="1">
+          <widget class="QComboBox" name="inputChannelComboBox">
+           <property name="enabled">
+            <bool>false</bool>
+           </property>
+           <item>
+            <property name="text">
+             <string>IN1-4</string>
+            </property>
+           </item>
+           <item>
+            <property name="text">
+             <string>IN5-8</string>
+            </property>
+           </item>
+           <item>
+            <property name="text">
+             <string>MULTIPLE_IN_1</string>
+            </property>
+           </item>
+           <item>
+            <property name="text">
+             <string>MULTIPLE_IN_2</string>
+            </property>
+           </item>
+           <item>
+            <property name="text">
+             <string>MULTIPLE_IN_3</string>
+            </property>
+           </item>
+           <item>
+            <property name="text">
+             <string>MULTIPLE_IN_4</string>
+            </property>
+           </item>
+          </widget>
+         </item>
+         <item row="4" column="1">
+          <widget class="QPushButton" name="inputConfigurationPushButton">
+           <property name="enabled">
+            <bool>false</bool>
+           </property>
+           <property name="text">
+            <string>Configure Input</string>
+           </property>
+          </widget>
+         </item>
+        </layout>
+       </widget>
+      </item>
+      <item row="5" column="0">
+       <spacer name="verticalSpacer">
+        <property name="orientation">
+         <enum>Qt::Orientation::Vertical</enum>
+        </property>
+        <property name="sizeHint" stdset="0">
+         <size>
+          <width>20</width>
+          <height>40</height>
+         </size>
+        </property>
+       </spacer>
+      </item>
+      <item row="4" column="0">
+       <widget class="QGroupBox" name="commandsGroupBox">
+        <property name="title">
+         <string>Commands</string>
+        </property>
+        <layout class="QGridLayout" name="gridLayout_3">
+         <item row="0" column="0">
+          <widget class="QPushButton" name="commandConnectionPushButton">
+           <property name="text">
+            <string>Connect</string>
+           </property>
+          </widget>
+         </item>
+         <item row="1" column="0">
+          <widget class="QPushButton" name="commandConfigurationPushButton">
+           <property name="text">
+            <string>Configure</string>
+           </property>
+          </widget>
+         </item>
+         <item row="2" column="0">
+          <widget class="QPushButton" name="commandStreamPushButton">
+           <property name="text">
+            <string>Stream</string>
+           </property>
+          </widget>
+         </item>
+        </layout>
+       </widget>
+      </item>
+      <item row="1" column="0">
+       <widget class="QGroupBox" name="acquisitionGroupBox">
+        <property name="title">
+         <string>Acquisiton Parameters</string>
+        </property>
+        <layout class="QGridLayout" name="gridLayout">
+         <item row="1" column="1">
+          <widget class="QComboBox" name="acquisitionNumberOfChannelsComboBox">
+           <property name="currentIndex">
+            <number>3</number>
+           </property>
+           <item>
+            <property name="text">
+             <string>120</string>
+            </property>
+           </item>
+           <item>
+            <property name="text">
+             <string>216</string>
+            </property>
+           </item>
+           <item>
+            <property name="text">
+             <string>312</string>
+            </property>
+           </item>
+           <item>
+            <property name="text">
+             <string>408</string>
+            </property>
+           </item>
+          </widget>
+         </item>
+         <item row="0" column="1">
+          <widget class="QComboBox" name="acquisitionSamplingFrequencyComboBox">
+           <property name="currentIndex">
+            <number>1</number>
+           </property>
+           <item>
+            <property name="text">
+             <string>512</string>
+            </property>
+           </item>
+           <item>
+            <property name="text">
+             <string>2048</string>
+            </property>
+           </item>
+           <item>
+            <property name="text">
+             <string>5120</string>
+            </property>
+           </item>
+           <item>
+            <property name="text">
+             <string>10240</string>
+            </property>
+           </item>
+          </widget>
+         </item>
+         <item row="1" column="0">
+          <widget class="QLabel" name="label_2">
+           <property name="text">
+            <string>Number of Channels</string>
+           </property>
+          </widget>
+         </item>
+         <item row="0" column="0">
+          <widget class="QLabel" name="label">
+           <property name="text">
+            <string>Sampling Frequency</string>
+           </property>
+          </widget>
+         </item>
+         <item row="2" column="0">
+          <widget class="QCheckBox" name="acquisitionDecimatorCheckBox">
+           <property name="text">
+            <string>Decimator</string>
+           </property>
+           <property name="checked">
+            <bool>true</bool>
+           </property>
+          </widget>
+         </item>
+         <item row="2" column="1">
+          <widget class="QCheckBox" name="acquisitionRecordingCheckBox">
+           <property name="enabled">
+            <bool>false</bool>
+           </property>
+           <property name="text">
+            <string>Recording</string>
+           </property>
+          </widget>
+         </item>
+        </layout>
+       </widget>
+      </item>
+      <item row="3" column="0">
+       <widget class="QGroupBox" name="gridSelectionGroupBox">
+        <property name="title">
+         <string>Grid Selection</string>
+        </property>
+        <layout class="QGridLayout" name="gridLayout_5">
+         <item row="1" column="1">
+          <widget class="QCheckBox" name="gridFourCheckBox">
+           <property name="text">
+            <string>Grid 4</string>
+           </property>
+          </widget>
+         </item>
+         <item row="1" column="0">
+          <widget class="QCheckBox" name="gridThreeCheckBox">
+           <property name="text">
+            <string>Grid 3</string>
+           </property>
+          </widget>
+         </item>
+         <item row="1" column="2">
+          <widget class="QCheckBox" name="gridFiveCheckBox">
+           <property name="text">
+            <string>Grid 5</string>
+           </property>
+          </widget>
+         </item>
+         <item row="1" column="3">
+          <widget class="QCheckBox" name="gridSixCheckBox">
+           <property name="text">
+            <string>Grid 6</string>
+           </property>
+          </widget>
+         </item>
+         <item row="0" column="3">
+          <widget class="QCheckBox" name="gridTwoCheckBox">
+           <property name="layoutDirection">
+            <enum>Qt::LayoutDirection::LeftToRight</enum>
+           </property>
+           <property name="text">
+            <string>Grid 2</string>
+           </property>
+          </widget>
+         </item>
+         <item row="0" column="0">
+          <widget class="QCheckBox" name="gridOneCheckBox">
+           <property name="text">
+            <string>Grid 1</string>
+           </property>
+          </widget>
+         </item>
+        </layout>
+       </widget>
+      </item>
+      <item row="0" column="0">
+       <widget class="QGroupBox" name="connectionGroupBox">
+        <property name="title">
+         <string>Connection parameters</string>
+        </property>
+        <layout class="QGridLayout" name="gridLayout_7">
+         <item row="0" column="0">
+          <widget class="QLabel" name="label_4">
+           <property name="text">
+            <string>IP</string>
+           </property>
+          </widget>
+         </item>
+         <item row="0" column="1">
+          <widget class="QLineEdit" name="connectionIPLineEdit">
+           <property name="text">
+            <string>169.254.1.10</string>
+           </property>
+          </widget>
+         </item>
+         <item row="1" column="0">
+          <widget class="QLabel" name="label_5">
+           <property name="text">
+            <string>Port</string>
+           </property>
+          </widget>
+         </item>
+         <item row="1" column="1">
+          <widget class="QLineEdit" name="connectionPortLineEdit">
+           <property name="text">
+            <string>23456</string>
+           </property>
+          </widget>
+         </item>
+        </layout>
+       </widget>
+      </item>
+     </layout>
+    </widget>
+   </item>
+  </layout>
+ </widget>
+ <resources/>
+ <connections/>
+</ui>