Source code for biosignal_device_interface.gui.device_template_widgets.all_devices_widget

"""
Template for the QWidget that enables the user to interact with all available devices.
Developer: Dominik I. Braun
Contact: dome.braun@fau.de
Last Update: 2024-06-05
"""

from __future__ import annotations
from typing import TYPE_CHECKING, Dict

from biosignal_device_interface.gui.device_template_widgets.core.base_multiple_devices_widget import (
    BaseMultipleDevicesWidget,
)
from biosignal_device_interface.constants.devices.core.base_device_constants import (
    DeviceType,
)
from biosignal_device_interface.gui.device_template_widgets.otb.otb_muovi_plus_widget import (
    OTBMuoviPlusWidget,
)
from biosignal_device_interface.gui.device_template_widgets.otb.otb_muovi_widget import (
    OTBMuoviWidget,
)
from biosignal_device_interface.gui.device_template_widgets.otb.otb_quattrocento_light_widget import (
    OTBQuattrocentoLightWidget,
)
from biosignal_device_interface.gui.device_template_widgets.otb.otb_quattrocento_widget import (
    OTBQuattrocentoWidget,
)
from biosignal_device_interface.gui.device_template_widgets.otb.otb_syncstation_widget import (
    OTBSyncStationWidget,
)
from biosignal_device_interface.gui.device_template_widgets.otb.otb_sessantaquattro_widget import (
    OTBSessantaquattroWidget,
)

if TYPE_CHECKING:
    from PySide6.QtWidgets import QWidget, QMainWindow
    from biosignal_device_interface.gui.device_template_widgets.core.base_device_widget import (
        BaseDeviceWidget,
    )




class AllDevicesWidget(BaseMultipleDevicesWidget):
    def __init__(self, parent: QWidget | QMainWindow | None = None):
        super().__init__(parent)

        self._device_selection: Dict[DeviceType, BaseDeviceWidget] = {
            DeviceType.OTB_QUATTROCENTO: OTBQuattrocentoWidget(self),
            DeviceType.OTB_QUATTROCENTO_LIGHT: OTBQuattrocentoLightWidget(self),
            DeviceType.OTB_MUOVI: OTBMuoviWidget(self),
            DeviceType.OTB_MUOVI_PLUS: OTBMuoviPlusWidget(self),
            DeviceType.OTB_SYNCSTATION: OTBSyncStationWidget(self),
            DeviceType.OTB_SESSANTAQUATTRO: OTBSessantaquattroWidget(self),
        }
        self._set_devices(self._device_selection)