pymodaq 4.4.7__py3-none-any.whl → 5.0.1__py3-none-any.whl

pymodaq/utils/gui_utils/widgets/tree_layout.py

@@ -1,188 +0,0 @@
-from qtpy import QtWidgets, QtCore, QtGui
-from qtpy.QtCore import QObject, Signal, QLocale
-import sys
-
-
-class TreeLayout(QObject):
-    """
-    qtpy class object based on QtreeWidget
-    The function populate_tree has to be used in order to populate the tree with structure as nested lists of dicts
-
-    """
-    status_sig = Signal(str)
-    item_clicked_sig = Signal(object)
-    item_double_clicked_sig = Signal(object)
-    
-    def __init__(self, parent=None, col_counts=1, labels=None):
-        
-        super().__init__()
-
-        if parent is None:
-            parent = QtWidgets.QWidget()
-        self.parent = parent
-
-        self.setupUi()
-
-        self.tree.setColumnCount(col_counts)
-        if labels is not None:
-            self.tree.setHeaderLabels(labels)
-
-        self.open_tree_pb.clicked.connect(self.expand_all)
-        self.close_tree_pb.clicked.connect(self.collapse_all)
-        self.open_tree_selected_pb.clicked.connect(self.open_tree_selection)
-        
-        self.tree.itemClicked.connect(self.item_clicked_sig.emit)
-        self.tree.itemDoubleClicked.connect(self.item_double_clicked_sig.emit)
-
-    def _current_text(self, col_index: int = 2):
-        return self.tree.currentItem().text(col_index)
-
-    def current_node_path(self):
-        return self._current_text(2)
-
-    def expand_all(self):
-        self.tree.expandAll()
-
-    def collapse_all(self):
-        self.tree.collapseAll()
-
-    @property
-    def treewidget(self):
-        return self.tree
-
-    def setupUi(self):
-        vlayout = QtWidgets.QVBoxLayout()
-        hlayout = QtWidgets.QHBoxLayout()
-
-        self.tree = CustomTree()
-        vlayout.addWidget(self.tree)
-
-        iconopen = QtGui.QIcon()
-        iconopen.addPixmap(QtGui.QPixmap(":/icons/Icon_Library/tree.png"), QtGui.QIcon.Normal,
-                           QtGui.QIcon.Off)
-        self.open_tree_pb = QtWidgets.QPushButton('Open Tree')
-        self.open_tree_pb.setIcon(iconopen)
-
-        iconopensel = QtGui.QIcon()
-        iconopensel.addPixmap(QtGui.QPixmap(":/icons/Icon_Library/tree.png"), QtGui.QIcon.Normal,
-                              QtGui.QIcon.Off)
-        self.open_tree_selected_pb = QtWidgets.QPushButton('Open Selected')
-        self.open_tree_selected_pb.setIcon(iconopensel)
-
-        iconclose = QtGui.QIcon()
-        iconclose.addPixmap(QtGui.QPixmap(":/icons/Icon_Library/CollapseAll.png"), QtGui.QIcon.Normal,
-                            QtGui.QIcon.Off)
-        self.close_tree_pb = QtWidgets.QPushButton('Close Tree')
-        self.close_tree_pb.setIcon(iconclose)
-
-        hlayout.addWidget(self.open_tree_pb)
-        hlayout.addWidget(self.open_tree_selected_pb)
-        hlayout.addWidget(self.close_tree_pb)
-
-        vlayout.addLayout(hlayout)
-
-        self.parent.setLayout(vlayout)
-
-    def open_tree_selection(self):
-        self.tree_open_children(self.tree.selectedIndexes()[0])
-
-    def tree_open_children(self, item_index):
-        try:
-            if not (item_index.isValid()):
-                return
-
-            self.tree.expand(item_index)
-        except Exception as e:
-            self.status_sig.emit(str(e))
-
-    def tree_open_parents(self, item_index):
-        try:
-            if not (item_index.isValid()):
-                return
-            flag = True
-            empty = QtCore.QModelIndex()
-            parent = item_index
-            while flag:
-                parent = parent.parent()
-                if parent != empty:
-                    self.tree.expand(parent)
-                else:
-                    flag = False
-                    break
-        except Exception as e:
-            self.status_sig.emit(str(e))
-
-    def populate_tree(self, data_dict):
-        try:
-            parents = []
-            for data in data_dict:
-                str_list = [str(data['name'])]
-                if 'filename' in data.keys():
-                    str_list.append(data['filename'])
-                if 'info' in data.keys():
-                    str_list.append(data['info'])
-                parent = QtWidgets.QTreeWidgetItem(str_list)
-                Items = self.populate_sub_tree(data['contents'])
-                parent.addChildren(Items)
-                parents.append(parent)
-
-            self.tree.addTopLevelItems(parents)
-        except Exception as e:
-            self.status_sig.emit(str(e))
-
-    def populate_sub_tree(self, datas):
-        try:
-            parents = []
-            for data in datas:
-                str_list = [str(data['name'])]
-                if 'filename' in data.keys():
-                    str_list.append(data['filename'])
-                if 'info' in data.keys():
-                    str_list.append(data['info'])
-                parent = QtWidgets.QTreeWidgetItem(str_list)
-                if 'contents' in data.keys():
-                    if type(data['contents']) == list:
-                        Items = self.populate_sub_tree(data['contents'])
-                        parent.addChildren(Items)
-                parents.append(parent)
-            return parents
-        except Exception as e:
-            self.status_sig.emit(str(e))
-            return parents
-
-
-class CustomTree(QtWidgets.QTreeWidget):
-
-    def __init__(self, parent=None):
-        super().__init__(parent)
-        self.setContextMenuPolicy(QtCore.Qt.ActionsContextMenu)
-
-
-if __name__ == '__main__':
-
-
-    app = QtWidgets.QApplication(sys.argv)
-    Form = QtWidgets.QWidget()
-    prog = TreeLayout(Form, col_counts=2, labels=["Material", "File"])
-
-    # example of actions to add to the tree widget in order to show a context menu
-    detector_action = QtWidgets.QAction("Grab from camera", None)
-    prog.tree.addAction(detector_action)
-    ########################
-
-    Form.show()
-    data = [dict(name='papa', contents=[
-        dict(name='fiston', contents=[dict(name='subfiston', contents='baby', filename='Cest pas sorcier')]),
-        dict(name='fiston1', contents=[dict(name='subfiston', contents='baby', filename='Cest pas malin')]),
-        dict(name='fiston2', contents=[dict(name='subfiston', contents='baby', filename='Cest pas normal')])]),
-        dict(name='maman', contents=[dict(name='fistone', contents=[dict(name='subfistone', contents='baby')])])]
-    prog.populate_tree(data)
-    # filename='C:\\Data\\2019\\20190220\\Dataset_20190220_004\\Dataset_20190220_004.h5'
-    # import tables
-    # h5_file = tables.open_file(filename, mode = "a")
-    # for node in h5_file.walk_nodes():
-    #     print(node)
-    # base_node = h5_file.root
-    # base_tree_item, pixmap_items = h5tree_to_QTree(h5_file, base_node)
-    # prog.ui.Tree.addTopLevelItem(base_tree_item)
-    sys.exit(app.exec_())

pymodaq/utils/gui_utils/widgets/tree_toml.py

@@ -1,110 +0,0 @@
-# -*- coding: utf-8 -*-
-"""
-Created the 19/10/2023
-
-@author: Sebastien Weber
-"""
-import datetime
-
-from qtpy import QtWidgets, QtCore
-from qtpy.QtCore import QObject
-
-from pymodaq.utils.parameter import ParameterTree, Parameter
-from pymodaq.utils.config import Config, create_toml_from_dict
-
-
-class TreeFromToml(QObject):
-    """ Create a ParameterTree from a configuration file"""
-
-    def __init__(self, config: Config = None, capitalize=True):
-        super().__init__()
-
-        if config is None:
-            config = Config()
-        self._config = config
-        params = [{'title': 'Config path', 'name': 'config_path', 'type': 'str',
-                   'value': str(self._config.config_path),
-                   'readonly': True}]
-        params.extend(self.dict_to_param(config.to_dict(), capitalize=capitalize))
-
-        self.settings = Parameter.create(title='settings', name='settings', type='group',
-                                         children=params)
-        self.settings_tree = ParameterTree()
-        self.settings_tree.setParameters(self.settings, showTop=False)
-
-    def show_dialog(self) -> bool:
-
-        self.dialog = QtWidgets.QDialog()
-        self.dialog.setWindowTitle('Please enter new configuration values!')
-        self.dialog.setLayout(QtWidgets.QVBoxLayout())
-        buttonBox = QtWidgets.QDialogButtonBox(parent=self.dialog)
-
-        buttonBox.addButton('Save', buttonBox.AcceptRole)
-        buttonBox.accepted.connect(self.dialog.accept)
-        buttonBox.addButton('Cancel', buttonBox.RejectRole)
-        buttonBox.rejected.connect(self.dialog.reject)
-
-        self.dialog.layout().addWidget(self.settings_tree)
-        self.dialog.layout().addWidget(buttonBox)
-        self.dialog.setWindowTitle('Configuration entries')
-        res = self.dialog.exec()
-
-        if res == self.dialog.Accepted:
-            with open(self._config.config_path, 'w') as f:
-                config_dict = self.param_to_dict(self.settings)
-                config_dict.pop('config_path')
-                create_toml_from_dict(config_dict, self._config.config_path)
-        return res
-
-    @classmethod
-    def param_to_dict(cls, param: Parameter) -> dict:
-        config = dict()
-        for child in param.children():
-            if 'group' in child.opts['type']:
-                config[child.name()] = cls.param_to_dict(child)
-            else:
-                if child.opts['type'] == 'datetime':
-                    config[child.name()] = datetime.fromtimestamp(
-                        child.value().toSecsSinceEpoch())  # convert QDateTime to python datetime
-                elif child.opts['type'] == 'date':
-                    qdt = QtCore.QDateTime()
-                    qdt.setDate(child.value())
-                    pdt = datetime.fromtimestamp(qdt.toSecsSinceEpoch())
-                    config[child.name()] = pdt.date()
-                elif child.opts['type'] == 'list':
-                    config[child.name()] = child.opts['limits']
-                else:
-                    config[child.name()] = child.value()
-        return config
-
-    @classmethod
-    def dict_to_param(cls, config: dict, capitalize=True) -> Parameter:
-        params = []
-        for key in config:
-            if isinstance(config[key], dict):
-                params.append({'title': f'{key.capitalize() if capitalize else key}:',
-                               'name': key, 'type': 'group',
-                               'children': cls.dict_to_param(config[key], capitalize=capitalize),
-                               'expanded': 'user' in key.lower() or 'general' in key.lower()})
-            else:
-                param = {'title': f'{key.capitalize() if capitalize else key}:',
-                         'name': key, 'value': config[key]}
-                if isinstance(config[key], float):
-                    param['type'] = 'float'
-                elif isinstance(config[key], bool):  # placed before int because a bool is an instance of int
-                    param['type'] = 'bool'
-                elif isinstance(config[key], int):
-                    param['type'] = 'int'
-                elif isinstance(config[key], datetime.datetime):
-                    param['type'] = 'datetime'
-                elif isinstance(config[key], datetime.date):
-                    param['type'] = 'date'
-                elif isinstance(config[key], str):
-                    param['type'] = 'str'
-                elif isinstance(config[key], list):
-                    param['type'] = 'list'
-                    param['limits'] = config[key]
-                    param['value'] = config[key][0]
-                    param['show_pb'] = True
-                params.append(param)
-        return params