melage 0.0.65__py3-none-any.whl → 1.0.0__py3-none-any.whl

melage/{widgets/fileDialog_widget.py → dialogs/helpers/FileDialog.py}

@@ -10,6 +10,231 @@ from PyQt5 import QtWidgets
 from PyQt5.QtCore import pyqtSignal
 from melage.utils.utils import help_dialogue_open_image
 
+import os
+import numpy as np
+from PyQt5 import QtWidgets, QtCore, QtGui
+from PyQt5.QtWidgets import QFileDialog, QVBoxLayout, QHBoxLayout, QLabel, QFrame
+from PyQt5.QtCore import Qt
+from PyQt5.QtGui import QPixmap, QImage
+import re
+
+class QFileDialogPreview(QFileDialog):
+    """
+    Custom File Dialog that parses the input filter string to automatically
+    create an "All Supported Files" option.
+    """
+
+    def __init__(self, parent=None, caption="", directory="", filter="", options=None, index=0, last_state=False):
+        # 1. Initialize the base QFileDialog
+        # We pass the parent, caption, and directory.
+        # We do NOT pass 'filter' yet, because we want to modify it first.
+        super().__init__(parent, caption, directory, "")
+
+        # 2. Auto-Detect Logic: Create "All Supported Files"
+        if filter:
+            # Step A: Find everything inside parentheses e.g. "*.dcm **"
+            # r'\(([^)]+)\)' captures all text between ( and )
+            raw_groups = re.findall(r'\(([^)]+)\)', filter)
+
+            valid_extensions = []
+
+            for group in raw_groups:
+                # Step B: Split the group into individual items
+                # " *.dcm  ** " -> ["*.dcm", "**"]
+                exts = group.split()
+
+                for ext in exts:
+                    # Step C: STRICTLY EXCLUDE '**'
+                    if ext.strip() == '**':
+                        continue
+
+                    valid_extensions.append(ext)
+
+            # Step D: Remove duplicates and join
+            # Using set() removes duplicates, sorted() keeps the order consistent
+            unique_exts = sorted(list(set(valid_extensions)))
+            all_string = " ".join(unique_exts)
+
+            if all_string:
+                final_filter = f"All Supported Files ({all_string});;{filter}"
+                self.setNameFilter(final_filter)
+            else:
+                # Fallback if no valid extensions found
+                self.setNameFilter(filter)
+
+        # 3. Handle Options (passed as kwarg in your call)
+        if options:
+            self.setOptions(options)
+            # Ensure we keep specific flags like DontUseNativeDialog if options overwrote them
+            self.setOption(QFileDialog.DontUseNativeDialog, True)
+        else:
+            self.setOption(QFileDialog.DontUseNativeDialog, True)
+
+        # 4. Standard Setup
+        self.setFileMode(QFileDialog.ExistingFile)
+        self.setAcceptMode(QFileDialog.AcceptOpen)
+
+        # 3. UI Setup (Preview Panel)
+        self._init_preview_ui(last_state, index)
+
+        # 4. Connections
+        self.currentChanged.connect(self.onChange)
+        self.fileSelected.connect(self.onFileSelected)
+        self.filesSelected.connect(self.onFilesSelected)
+
+        # Internal state
+        self._fileSelected = None
+        self._filesSelected = None
+
+        for combo in self.findChildren(QtWidgets.QComboBox):
+            if "All Supported" in combo.currentText():
+                sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Ignored, QtWidgets.QSizePolicy.Fixed)
+                sizePolicy.setHorizontalStretch(1)
+                combo.setSizePolicy(sizePolicy)
+                combo.setMinimumWidth(150)  # Stop it from disappearing
+
+    def _init_preview_ui(self, last_state, index):
+        """Helper to build the side panel UI"""
+
+        # Container Layout for the side panel
+        self.side_panel_layout = QHBoxLayout()
+        self.side_panel_layout.setContentsMargins(0, 0, 0, 0)
+
+
+
+        # C. Advanced Options Group
+        adv_layout = QHBoxLayout()
+
+        self.checkBox_advanced = QtWidgets.QCheckBox('Advanced', self)
+        self.checkBox_advanced.setObjectName("checkBox")
+        self.checkBox_advanced.stateChanged.connect(self.activate_combobox)
+
+        self._combobox_type = QtWidgets.QComboBox(self)
+        self.systems = ['US (neonatal)', 'US (fetal)', 'MRI']
+        for ss in self.systems:
+            self._combobox_type.addItem(f"   {ss}   ")
+        self._combobox_type.setCurrentIndex(index)
+        self._combobox_type.setEnabled(False)  # Default disabled until checked
+
+        adv_layout.addWidget(self.checkBox_advanced)
+        adv_layout.addWidget(self._combobox_type)
+
+        self.side_panel_layout.addLayout(adv_layout)
+
+        # A. Preview Label
+        self.mpPreview = QLabel("Preview", self)
+        self.mpPreview.setMinimumSize(300, 300)
+        self.mpPreview.setAlignment(Qt.AlignCenter)
+        self.mpPreview.setStyleSheet("border: 1px solid #aaa; background-color: #eee;")  # Visual cue
+        self.mpPreview.setScaledContents(False)  # We handle scaling manually for aspect ratio
+
+
+        self.mpPreview.setVisible(last_state)
+
+        # B. Preview Checkbox
+        self.checkBox_preview = QtWidgets.QCheckBox('Show Preview', self)
+        self.checkBox_preview.setChecked(last_state)
+
+        self.checkBox_preview.stateChanged.connect(self.on_toggle_preview)
+
+        self.side_panel_layout.addWidget(self.checkBox_preview)
+        self.side_panel_layout.addWidget(self.mpPreview)
+
+
+        self.side_panel_layout.addStretch()  # Push everything up
+
+        # D. Inject into QFileDialog Grid
+        # QFileDialog uses a QGridLayout. (1, 3) places it to the right of the file list.
+        # We assume the internal layout is a QGridLayout (standard in PyQt5).
+        layout = self.layout()
+        if layout:
+            layout.addLayout(self.side_panel_layout, 4, 1, 1, 1)
+
+    def on_toggle_preview(self, state):
+        """
+        Slot called when 'Show Preview' is checked/unchecked.
+        state: 0 (Unchecked) or 2 (Checked)
+        """
+        is_checked = (state == Qt.Checked)
+
+        # 1. Hide/Show the actual label widget
+        self.mpPreview.setVisible(is_checked)
+
+        # 2. If turning ON, force an update immediately
+        # (Otherwise the preview would be empty until the user clicks a new file)
+        if is_checked:
+            # Get the file currently highlighted in the dialog
+            current_files = self.selectedFiles()
+            if current_files:
+                self.onChange(current_files[0])
+            else:
+                self.mpPreview.setText("No file selected")
+
+
+
+    def onChange(self, path):
+        if not self.checkBox_preview.isChecked():
+            return
+        """Handle file selection change for preview"""
+        if not self.checkBox_preview.isChecked() or not path or not os.path.isfile(path):
+            self.mpPreview.setText("Preview")
+            return
+
+        try:
+            # CALL YOUR EXTERNAL LOADING FUNCTION
+            # Assuming help_dialogue_open_image returns a numpy array-like object
+            s = help_dialogue_open_image(path)
+
+            if s is None:
+                raise ValueError("Image data is None")
+
+            s = help_dialogue_open_image(path)
+            height, width,_ = s.shape
+
+            bytesPerLine = 3 * width
+            #qImg = QImage(normal.data, width, height, bytesPerLine, QImage.Format_RGB888)
+            #gray_color_table = [qRgb(i, i, i) for i in range(256)]
+            qImg = QImage(s.data, s.shape[1]//3, s.shape[0]//3, bytesPerLine, QImage.Format_RGB888)
+            #qImg.setColorTable(gray_color_table)
+            #from PyQt5.QtCore import QByteArray
+            #qb = QByteArray(np.ndarray.tobytes(s))
+            #qImg = QImage.fromData(qb)
+            pixmap01 = QPixmap.fromImage(qImg)
+            pixmap = QPixmap(pixmap01)
+
+            if pixmap.isNull():
+                self.mpPreview.setText("Preview unavailable")
+            else:
+                # Scale smoothly to fit the label area
+                scaled_pix = pixmap.scaled(
+                    self.mpPreview.width(),
+                    self.mpPreview.height(),
+                    Qt.KeepAspectRatio,
+                    Qt.SmoothTransformation
+                )
+                self.mpPreview.setPixmap(scaled_pix)
+
+        except Exception as e:
+            # Fail silently or show error text so dialogue doesn't crash
+            # print(f"Preview Error: {e}")
+            self.mpPreview.setText("No Preview")
+
+    def activate_combobox(self, state):
+        # state is integer (0: Unchecked, 2: Checked)
+        self._combobox_type.setEnabled(state == Qt.Checked)
+
+    def onFileSelected(self, file):
+        self._fileSelected = file
+
+    def onFilesSelected(self, files):
+        self._filesSelected = files
+
+    def getFileSelected(self):
+        return [self._fileSelected, '']
+
+    def getFilesSelected(self):
+        return self._filesSelected
+
 class QFileSaveDialogPreview(QFileDialog):
     """
     Customizing save dialogue
@@ -132,7 +357,7 @@ class QFileSaveDialogPreview(QFileDialog):
 
 
 
-class QFileDialogPreview(QFileDialog):
+class QFileDialogPreview_or(QFileDialog):
     """
     Open dialogue preview
     """

melage/dialogs/helpers/__init__.py

@@ -0,0 +1,5 @@
+from .ColorDialog import NewDialog
+from .CustomScrollbar import custom_qscrollbar
+from .FileDialog import QFileSaveDialogPreview, QFileDialogPreview
+from .RepeatDialog import repeatN
+from .ScreenshotDialog import screenshot

melage/main.py

@@ -7,10 +7,9 @@ sys.path.append("../")
 #print(sys.path)
 from PyQt5 import QtWidgets, QtCore, QtGui
 #from PyQt5.QtWidgets import QWidget
-from melage.widgets.mainwindow_widget import Ui_Main
-from melage.widgets.activation import activation_dialog
-
-from melage.utils.source_folder import source_folder, desktop
+from melage.mainwindow_widget import Ui_Main
+from melage.dialogs import activation_dialog
+from melage.config import settings
 
 ## verifyClass
 class verifyClass(QtCore.QObject):
@@ -33,11 +32,11 @@ class MainWindow(QtWidgets.QMainWindow, Ui_Main):
         QtWidgets.QWidget.__init__(self)
         self.app = app
         self.setupUi(self)
-        if os.path.isdir(desktop):
-            self.source_dir = desktop
+        #if os.path.isdir(SOURCE_DIR):
+            #self.source_dir = SOURCE_DIR
         #self.settings.setPath(self.settings.IniFormat, self.settings.UserScope, '.')
 
-        self.setWindowIcon(QtGui.QIcon(source_folder+'/main.ico'))
+        self.setWindowIcon(QtGui.QIcon(settings.RESOURCE_DIR+'/main.ico'))
         self._key_picke = b'PPQ0ByoMsieWGv6bMEyJ9rSYXQDoa5D4ldAkwaNNpw0='
 
         if not self.settings.value("geometry") == None:
@@ -52,7 +51,7 @@ class MainWindow(QtWidgets.QMainWindow, Ui_Main):
 
 
             self.setEnabled(False)
-            self.activation_dialog = activation_dialog(self,source_folder)
+            self.activation_dialog = activation_dialog(self,settings.RESOURCE_DIR)
             self.activation_dialog.correct_key.connect(self.EnableAll)
             verifyInstance = verifyClass()
             verifyInstance.sig.connect(self.verify_func)
@@ -239,11 +238,12 @@ def main():
     fmt.setProfile(QtGui.QSurfaceFormat.CoreProfile)  # Set the desired profile
     QtGui.QSurfaceFormat.setDefaultFormat(fmt)
     """
-    root = os.path.dirname(os.path.abspath(__file__))
-    QtCore.QDir.addSearchPath('resource', os.path.join(root, 'resource'))
-    QtCore.QDir.addSearchPath('theme', os.path.join(root, 'resource/theme'))
-    QtCore.QDir.addSearchPath('rc', os.path.join(root, 'resource/theme/rc'))
-    QtCore.QDir.addSearchPath('color', os.path.join(root, 'resource/color'))
+    #root = os.path.dirname(os.path.abspath(__file__))
+    root = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
+    QtCore.QDir.addSearchPath('resource', os.path.join(root, "assets", 'resource'))
+    QtCore.QDir.addSearchPath('theme', os.path.join(root, "assets",'resource', 'theme'))
+    QtCore.QDir.addSearchPath('rc', os.path.join(root, "assets",'resource','theme', 'rc'))
+    QtCore.QDir.addSearchPath('color', os.path.join(root, "assets",'resource', 'color'))
     file = QtCore.QFile("theme:style.qss")
     file.open(QtCore.QFile.ReadOnly | QtCore.QFile.Text)
     stream = QtCore.QTextStream(file)