Helmholtz design

.github/workflows/test.yml

@@ -40,7 +40,7 @@ jobs:
           git clone --depth 1 https://github.com/pyvista/gl-ci-helpers.git
           powershell gl-ci-helpers/appveyor/install_opengl.ps1
 
-      - name: Install dependencies
+      - name: Install client dependencies
         run: |
           python -m pip install --upgrade pip
           python -m pip install setuptools  
@@ -52,6 +52,8 @@ jobs:
 
       - name: Install server dependencies (for communication tests)
         run: |
+          pip install numpy
+          pip install wheel
           pip install -e ".[dev]"
         working-directory: src/server
 
@@ -87,7 +89,7 @@ jobs:
         with:
           python-version: ${{ matrix.python-version }}
 
-      - name: Install dependencies
+      - name: Install server dependencies
         run: |
           python -m pip install --upgrade pip
           python -m pip install --upgrade setuptools  

src/client/dcp_client/gui/_custom_qt_helpers.py

@@ -0,0 +1,64 @@
+from PyQt5.QtWidgets import  QFileIconProvider, QStyledItemDelegate
+from PyQt5.QtCore import QSize
+from PyQt5.QtGui import QPixmap, QIcon
+
+from dcp_client.utils import settings
+
+class IconProvider(QFileIconProvider):
+    def __init__(self) -> None:
+        """Initializes the IconProvider with the default icon size."""
+        super().__init__()
+        self.ICON_SIZE = QSize(512, 512)
+
+    def icon(self, type: QFileIconProvider.IconType) -> QIcon:
+        """Returns the icon for the specified file type.
+
+        :param type: The type of the file for which the icon is requested.
+        :type type: QFileIconProvider.IconType
+        :return: The icon for the file type.
+        :rtype: QIcon
+        """
+        try:
+            fn = type.filePath()
+        except AttributeError:
+            return super().icon(type)  # TODO handle exception differently?
+
+        if fn.endswith(settings.accepted_types):
+            a = QPixmap(self.ICON_SIZE)
+            # a = a.scaled(QSize(1024, 1024))
+            a.load(fn)
+            return QIcon(a)
+        else:
+            return super().icon(type)
+
+
+class CustomItemDelegate(QStyledItemDelegate):
+    """
+    A custom item delegate for setting a fixed height for items in a view.
+    This delegate overrides the sizeHint method to set a fixed height for items.
+    """
+    def __init__(self, parent=None):
+        """
+        Initialize the CustomItemDelegate.
+
+        :param parent: The parent QObject. Default is None.
+        :type parent: QObject
+        """
+        super().__init__(parent)
+
+    def sizeHint(self, option, index):
+        """
+        Returns the size hint for the item specified by the given index.
+
+        :param option: The parameters used to draw the item.
+        :type option: QStyleOptionViewItem
+
+        :param index: The model index of the item.
+        :type index: QModelIndex
+
+        :returns: The size hint for the item.
+        :rtype: QSize
+        """
+        size = super().sizeHint(option, index)
+        size.setHeight(100)  
+        return size

src/client/dcp_client/gui/_filesystem_wig.py

@@ -0,0 +1,112 @@
+import os
+import numpy as np
+from skimage.io import imread
+from skimage.color import label2rgb
+
+
+from PyQt5.QtWidgets import QFileSystemModel
+from PyQt5.QtCore import Qt, QVariant, QDir
+from PyQt5.QtGui import QImage
+
+from dcp_client.utils import settings
+
+class MyQFileSystemModel(QFileSystemModel):
+    def __init__(self, app):
+        """
+        Initializes a custom QFileSystemModel
+        """
+        super().__init__()
+        self.app = app
+        self.img_x = 100
+        self.img_y = 100
+
+    def setFilter(self, filters):
+        """
+        Sets filters for the model.
+
+        :param filters: The filters to be applied. (QDir.Filters)
+        """
+        filters |= QDir.NoDotAndDotDot | QDir.AllDirs | QDir.Files
+        super().setFilter(filters)
+
+        # Exclude files containing '_seg' in their names
+        self.addFilter(lambda fileInfo: "_seg" not in fileInfo.fileName())
+
+    def addFilter(self, filterFunc):
+        """
+        Adds a custom filter function to the model.
+
+        :param filterFunc: The filter function to be added. (function)
+        """
+        self.filterFunc = filterFunc
+
+    def headerData(self, section, orientation, role):
+        """
+        Reimplemented method to provide custom header data for the model's headers.
+
+        :param section: The section (column) index. (int)
+        :param orientation: The orientation of the header. (Qt.Orientation)
+        :param role: The role of the header data. (int)
+        :rtype: QVariant
+        """
+        if section == 0 and role == Qt.DisplayRole:
+            return ""
+        else:
+            return super().headerData(section, orientation, role)
+
+
+    def data(self, index, role=Qt.DisplayRole):
+        """
+        Reimplemented method to provide custom data for the model's items.
+
+        :param index: The index of the item. (QModelIndex)
+        :param role: The role of the data. (int)
+        :rtype: QVariant
+        """
+        if not index.isValid():
+            return QVariant()
+
+        if role == Qt.DecorationRole:
+
+            filepath_img = self.filePath(index)
+            # if an image of our dataset
+            if filepath_img.endswith(settings.accepted_types):
+
+                # if a mask make sure it is displayed properly
+                if "_seg" in filepath_img and os.path.exists(filepath_img):
+                    img = imread(filepath_img)
+                    if img.ndim > 2: img = img[0]
+                    img = label2rgb(img)
+                    img = (255 * img.copy()).astype(
+                        np.uint8
+                    )#np.transpose(img, (1, 0, 2))
+                    height, width = img.shape[0], img.shape[1]
+                    img = QImage(img, 
+                                width, 
+                                height, 
+                                3 * width,
+                                QImage.Format_RGB888
+                    )
+                    #img  = img.scaled(self.img_x, (width*self.img_x)//height, Qt.KeepAspectRatio)
+                    img  = img.scaled(self.img_x, self.img_y, Qt.KeepAspectRatio) # yields the same
+                else:
+                    img = QImage(filepath_img).scaled(self.img_x, self.img_y, Qt.KeepAspectRatio)
+                '''
+                # It would be cool if instead of the mask and the image we could show them both merged 
+                # together with label2rgb if the mask exists - would need to remove _seg files from list
+                filepath_mask = '.'.join(filepath_img.split('.')[:-1])+'_seg.tiff'
+                if os.path.exists(filepath_mask):
+                    mask = imread(filepath_mask)
+                    if mask.ndim>2: mask = mask[0]
+                    img = imread(filepath_img, as_gray=True)
+                    img = label2rgb(mask, img)
+                    img = QImage(img,
+                                img.shape[1], 
+                                img.shape[0],
+                                QImage.Format_RGB888
+                                ).scaled(self.img_x, self.img_y, Qt.KeepAspectRatio)
+                '''
+                return img
+
+        return super().data(index, role)
+