py-neuromodulation 0.0.2__py3-none-any.whl → 0.0.3__py3-none-any.whl

packages/tmsi/build/lib/TMSiFileFormats/file_writer.py

@@ -0,0 +1,102 @@
+"""
+(c) 2022 Twente Medical Systems International B.V., Oldenzaal The Netherlands
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+
+#######  #     #   #####   #
+   #     ##   ##  #        
+   #     # # # #  #        #
+   #     #  #  #   #####   #
+   #     #     #        #  #
+   #     #     #        #  #
+   #     #     #  #####    #
+
+/**
+ * @file ${file_writer.py} 
+ * @brief File Writer Interface
+ *
+ */
+
+
+"""
+
+from enum import Enum
+from TMSiSDK.error import TMSiError, TMSiErrorCode
+
+
+class FileFormat(Enum):
+    none = 0
+    poly5 = 1
+    xdf = 2
+    lsl = 3
+
+
+class FileWriter:
+    """<FileWriter> implements a file-writer for writing sample-data, captured
+    during a measurement, to a specific file in a specific data-format.
+
+    Args:
+        data_format_type : <FileFormat> Specifies the data-format of the file.
+        This can be poly5, gdf+ or gdf.
+
+        filename : <string> The path and name of the file, into which the
+        measurement-data must be written.
+    """
+
+    def __init__(
+        self, data_format_type, filename, add_ch_locs=False, download=False
+    ):
+        if data_format_type == FileFormat.poly5:
+            from .file_formats.poly5_file_writer import Poly5Writer
+
+            self._data_format_type = data_format_type
+            self._file_writer = Poly5Writer(filename, download)
+        elif data_format_type == FileFormat.xdf:
+            from .file_formats.xdf_file_writer import XdfWriter
+
+            self._data_format_type = data_format_type
+            self._file_writer = XdfWriter(filename, add_ch_locs)
+        elif data_format_type == FileFormat.lsl:
+            from .file_formats.lsl_stream_writer import LSLWriter
+
+            self._data_format_type = data_format_type
+            self._file_writer = LSLWriter(filename)
+        else:
+            print("Unsupported data format")
+            raise TMSiError(TMSiErrorCode.api_incorrect_argument)
+
+    def open(self, device):
+        """Opens a file-writer session.
+
+        Must be called BEFORE a measurement is started.
+
+        Links a 'Device' to a 'Filewriter'-instance and prepares for the
+        measurement that will start.
+
+        In the open-method the file-writer-object will prepare for the receipt
+        of sample-data from the 'Device' as next:
+            - Retrieve the configuration (e.g. channel-list, sample_rate) from the 'Device'
+              to write to the file and to prepare for the receipt of sample-data.
+            - Register at the 'sample_data_server' for receipt of the sample-data.
+            - Create a dedicated sampling-thread, which is responsible to
+              processes during the measurement the incoming sample-data.
+        """
+        self._file_writer.open(device)
+
+    def close(self):
+        """Closes an ongoing file-writer session.
+
+        Must be called AFTER a measurement is stopped.
+
+        """
+        self._file_writer.close()

packages/tmsi/build/lib/TMSiPlotters/__init__.py

@@ -0,0 +1,2 @@
+from . import gui
+from . import plotters

packages/tmsi/build/lib/TMSiPlotters/gui/__init__.py

@@ -0,0 +1,39 @@
+'''
+(c) 2022 Twente Medical Systems International B.V., Oldenzaal The Netherlands
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+
+#######  #     #   #####   #
+   #     ##   ##  #        
+   #     # # # #  #        #
+   #     #  #  #   #####   #
+   #     #     #        #  #
+   #     #     #        #  #
+   #     #     #  #####    #
+
+/**
+ * @file ${__init__.py} 
+ * @brief Initialisation of the TMSiPlotters gui directory classes.
+ *
+ */
+
+
+'''
+
+import sys
+from os.path import join, dirname, realpath
+Plotters_dir = dirname(realpath(__file__)) # directory of this file
+modules_dir = join(Plotters_dir, '...') # directory with all modules
+sys.path.append(modules_dir)
+
+from .plotting_gui import PlottingGUI

packages/tmsi/build/lib/TMSiPlotters/gui/_plotter_gui.py

@@ -0,0 +1,234 @@
+'''
+(c) 2022 Twente Medical Systems International B.V., Oldenzaal The Netherlands
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+
+#######  #     #   #####   #
+   #     ##   ##  #        
+   #     # # # #  #        #
+   #     #  #  #   #####   #
+   #     #     #        #  #
+   #     #     #        #  #
+   #     #     #  #####    #
+
+/**
+ * @file ${_plotter_gui.py} 
+ * @brief GUI window used by the different plotter instances.
+ *
+ */
+
+
+'''
+
+from PySide2.QtCore import *
+from PySide2.QtGui import *
+from PySide2.QtWidgets import *
+
+from pyqtgraph import GraphicsLayoutWidget
+
+
+class Ui_MainWindow(object):
+    def setupUi(self, MainWindow):
+        if not MainWindow.objectName():
+            MainWindow.setObjectName(u"MainWindow")
+        MainWindow.resize(1134, 704)
+        self.centralwidget = QWidget(MainWindow)
+        self.centralwidget.setObjectName(u"centralwidget")
+        self.gridLayout = QGridLayout(self.centralwidget)
+        self.gridLayout.setObjectName(u"gridLayout")
+        self.RealTimePlotWidget = GraphicsLayoutWidget(self.centralwidget)
+        self.RealTimePlotWidget.setObjectName(u"RealTimePlotWidget")
+        sizePolicy = QSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
+        sizePolicy.setHorizontalStretch(0)
+        sizePolicy.setVerticalStretch(0)
+        sizePolicy.setHeightForWidth(self.RealTimePlotWidget.sizePolicy().hasHeightForWidth())
+        self.RealTimePlotWidget.setSizePolicy(sizePolicy)
+
+        self.gridLayout.addWidget(self.RealTimePlotWidget, 0, 1, 3, 4)
+
+        self.autoscale_button = QPushButton(self.centralwidget)
+        self.autoscale_button.setObjectName(u"autoscale_button")
+
+        self.gridLayout.addWidget(self.autoscale_button, 3, 3, 1, 1)
+
+        self.increase_time_button = QPushButton(self.centralwidget)
+        self.increase_time_button.setObjectName(u"increase_time_button")
+
+        self.gridLayout.addWidget(self.increase_time_button, 3, 2, 1, 1)
+
+        self.verticalLayout = QVBoxLayout()
+        self.verticalLayout.setObjectName(u"verticalLayout")
+        self.set_range_label = QLabel(self.centralwidget)
+        self.set_range_label.setObjectName(u"set_range_label")
+        font = QFont()
+        font.setBold(True)
+        font.setWeight(75)
+        self.set_range_label.setFont(font)
+
+        self.verticalLayout.addWidget(self.set_range_label)
+
+        self.set_range_box = QComboBox(self.centralwidget)
+        self.set_range_box.addItem("")
+        self.set_range_box.addItem("")
+        self.set_range_box.addItem("")
+        self.set_range_box.addItem("")
+        self.set_range_box.addItem("")
+        self.set_range_box.addItem("")
+        self.set_range_box.addItem("")
+        self.set_range_box.setObjectName(u"set_range_box")
+
+        self.verticalLayout.addWidget(self.set_range_box)
+
+
+        self.gridLayout.addLayout(self.verticalLayout, 3, 4, 1, 1)
+
+        self.gridLayout_2 = QGridLayout()
+        self.gridLayout_2.setObjectName(u"gridLayout_2")
+        self.hide_UNI_button = QPushButton(self.centralwidget)
+        self.hide_UNI_button.setObjectName(u"hide_UNI_button")
+        self.hide_UNI_button.setMaximumSize(QSize(100, 16777215))
+        self.hide_UNI_button.setLayoutDirection(Qt.RightToLeft)
+
+        self.gridLayout_2.addWidget(self.hide_UNI_button, 0, 1, 1, 1)
+
+        self.show_BIP_button = QPushButton(self.centralwidget)
+        self.show_BIP_button.setObjectName(u"show_BIP_button")
+
+        self.gridLayout_2.addWidget(self.show_BIP_button, 1, 0, 1, 1)
+
+        self.show_UNI_button = QPushButton(self.centralwidget)
+        self.show_UNI_button.setObjectName(u"show_UNI_button")
+        self.show_UNI_button.setMaximumSize(QSize(100, 16777215))
+
+        self.gridLayout_2.addWidget(self.show_UNI_button, 0, 0, 1, 1)
+
+        self.hide_BIP_button = QPushButton(self.centralwidget)
+        self.hide_BIP_button.setObjectName(u"hide_BIP_button")
+        self.hide_BIP_button.setLayoutDirection(Qt.RightToLeft)
+
+        self.gridLayout_2.addWidget(self.hide_BIP_button, 1, 1, 1, 1)
+
+        self.show_AUX_button = QPushButton(self.centralwidget)
+        self.show_AUX_button.setObjectName(u"show_AUX_button")
+
+        self.gridLayout_2.addWidget(self.show_AUX_button, 2, 0, 1, 1)
+
+        self.show_DIGI_button = QPushButton(self.centralwidget)
+        self.show_DIGI_button.setObjectName(u"show_DIGI_button")
+
+        self.gridLayout_2.addWidget(self.show_DIGI_button, 3, 0, 1, 1)
+
+        self.hide_AUX_button = QPushButton(self.centralwidget)
+        self.hide_AUX_button.setObjectName(u"hide_AUX_button")
+        self.hide_AUX_button.setLayoutDirection(Qt.RightToLeft)
+
+        self.gridLayout_2.addWidget(self.hide_AUX_button, 2, 1, 1, 1)
+
+        self.hide_DIGI_button = QPushButton(self.centralwidget)
+        self.hide_DIGI_button.setObjectName(u"hide_DIGI_button")
+        self.hide_DIGI_button.setLayoutDirection(Qt.RightToLeft)
+
+        self.gridLayout_2.addWidget(self.hide_DIGI_button, 3, 1, 1, 1)
+
+
+        self.gridLayout.addLayout(self.gridLayout_2, 2, 0, 1, 1)
+
+        self.decrease_time_button = QPushButton(self.centralwidget)
+        self.decrease_time_button.setObjectName(u"decrease_time_button")
+
+        self.gridLayout.addWidget(self.decrease_time_button, 3, 1, 1, 1)
+
+        self.channel_list_groupbox = QGroupBox(self.centralwidget)
+        self.channel_list_groupbox.setObjectName(u"channel_list_groupbox")
+        sizePolicy1 = QSizePolicy(QSizePolicy.Preferred, QSizePolicy.Expanding)
+        sizePolicy1.setHorizontalStretch(0)
+        sizePolicy1.setVerticalStretch(0)
+        sizePolicy1.setHeightForWidth(self.channel_list_groupbox.sizePolicy().hasHeightForWidth())
+        self.channel_list_groupbox.setSizePolicy(sizePolicy1)
+        self.channel_list_groupbox.setMinimumSize(QSize(100, 0))
+        self.channel_list_groupbox.setMaximumSize(QSize(400, 16777215))
+        font1 = QFont()
+        font1.setBold(False)
+        font1.setWeight(50)
+        self.channel_list_groupbox.setFont(font1)
+
+        self.gridLayout.addWidget(self.channel_list_groupbox, 0, 0, 1, 1)
+
+        self.horizontalLayout = QHBoxLayout()
+        self.horizontalLayout.setObjectName(u"horizontalLayout")
+        self.enable_filter_button = QPushButton(self.centralwidget)
+        self.enable_filter_button.setObjectName(u"enable_filter_button")
+
+        self.horizontalLayout.addWidget(self.enable_filter_button)
+
+        self.disable_filter_button = QPushButton(self.centralwidget)
+        self.disable_filter_button.setObjectName(u"disable_filter_button")
+
+        self.horizontalLayout.addWidget(self.disable_filter_button)
+
+
+        self.gridLayout.addLayout(self.horizontalLayout, 3, 0, 1, 1)
+
+        self.table_live_impedance = QTableWidget(self.centralwidget)
+        self.table_live_impedance.setObjectName(u"table_live_impedance")
+        sizePolicy2 = QSizePolicy(QSizePolicy.Preferred, QSizePolicy.Preferred)
+        sizePolicy2.setHorizontalStretch(0)
+        sizePolicy2.setVerticalStretch(0)
+        sizePolicy2.setHeightForWidth(self.table_live_impedance.sizePolicy().hasHeightForWidth())
+        self.table_live_impedance.setSizePolicy(sizePolicy2)
+        self.table_live_impedance.setMinimumSize(QSize(220, 0))
+        self.table_live_impedance.setMaximumSize(QSize(220, 16777215))
+
+        self.gridLayout.addWidget(self.table_live_impedance, 0, 5, 3, 1)
+
+        MainWindow.setCentralWidget(self.centralwidget)
+        self.menubar = QMenuBar(MainWindow)
+        self.menubar.setObjectName(u"menubar")
+        self.menubar.setGeometry(QRect(0, 0, 1134, 21))
+        MainWindow.setMenuBar(self.menubar)
+        self.statusbar = QStatusBar(MainWindow)
+        self.statusbar.setObjectName(u"statusbar")
+        MainWindow.setStatusBar(self.statusbar)
+
+        self.retranslateUi(MainWindow)
+
+        QMetaObject.connectSlotsByName(MainWindow)
+    # setupUi
+
+    def retranslateUi(self, MainWindow):
+        MainWindow.setWindowTitle(QCoreApplication.translate("MainWindow", u"MainWindow", None))
+        self.autoscale_button.setText(QCoreApplication.translate("MainWindow", u"Auto Scale", None))
+        self.increase_time_button.setText(QCoreApplication.translate("MainWindow", u"Increase time range", None))
+        self.set_range_label.setText(QCoreApplication.translate("MainWindow", u"<html><head/><body><p>Set Range ('\u00b5Volt')</p></body></html>", None))
+        self.set_range_box.setItemText(0, QCoreApplication.translate("MainWindow", u"1", None))
+        self.set_range_box.setItemText(1, QCoreApplication.translate("MainWindow", u"20", None))
+        self.set_range_box.setItemText(2, QCoreApplication.translate("MainWindow", u"50", None))
+        self.set_range_box.setItemText(3, QCoreApplication.translate("MainWindow", u"100", None))
+        self.set_range_box.setItemText(4, QCoreApplication.translate("MainWindow", u"250", None))
+        self.set_range_box.setItemText(5, QCoreApplication.translate("MainWindow", u"1000", None))
+        self.set_range_box.setItemText(6, QCoreApplication.translate("MainWindow", u"10000", None))
+
+        self.hide_UNI_button.setText(QCoreApplication.translate("MainWindow", u"Hide UNI", None))
+        self.show_BIP_button.setText(QCoreApplication.translate("MainWindow", u"Show BIP", None))
+        self.show_UNI_button.setText(QCoreApplication.translate("MainWindow", u"Show UNI", None))
+        self.hide_BIP_button.setText(QCoreApplication.translate("MainWindow", u"Hide BIP", None))
+        self.show_AUX_button.setText(QCoreApplication.translate("MainWindow", u"Show AUX", None))
+        self.show_DIGI_button.setText(QCoreApplication.translate("MainWindow", u"Show DIGI", None))
+        self.hide_AUX_button.setText(QCoreApplication.translate("MainWindow", u"Hide AUX", None))
+        self.hide_DIGI_button.setText(QCoreApplication.translate("MainWindow", u"Hide DIGI", None))
+        self.decrease_time_button.setText(QCoreApplication.translate("MainWindow", u"Decrease time range", None))
+        self.channel_list_groupbox.setTitle(QCoreApplication.translate("MainWindow", u"Channel list", None))
+        self.enable_filter_button.setText(QCoreApplication.translate("MainWindow", u"Enable filter", None))
+        self.disable_filter_button.setText(QCoreApplication.translate("MainWindow", u"Disable filter", None))
+    # retranslateUi
+