Semapp 1.0.1__py3-none-any.whl → 1.0.3__py3-none-any.whl

semapp/Layout/create_button.py

@@ -5,10 +5,11 @@
 import sys
 import os
 import time
+import re
 from PyQt5.QtWidgets import QApplication 
 from PyQt5.QtWidgets import (
     QWidget, QButtonGroup, QPushButton, QLabel, QGroupBox, QGridLayout,
-    QFileDialog, QProgressDialog, QRadioButton, QSizePolicy)
+    QFileDialog, QProgressDialog, QRadioButton, QSizePolicy, QSlider)
 from PyQt5.QtGui import QFont
 from PyQt5.QtCore import Qt 
 from semapp.Processing.processing import Process
@@ -40,6 +41,10 @@ class ButtonFrame(QWidget):
         self.selected_image = None
         self.table_data = None
         self.table_vars = None
+        self.image_slider = None  # Slider for COMPLUS4T mode
+        self.slider_value = 0  # Current slider value
+        self.image_group_box = None  # Store reference to image type group box
+        self.plot_frame = None  # Reference to plot frame for updates
 
         self.rename = QRadioButton("Rename")
         self.split_rename = QRadioButton("Split .tif and rename (w/ tag)")
@@ -63,7 +68,7 @@ class ButtonFrame(QWidget):
         # Example of adding them to a layout
         self.entries = {}
         self.dirname = None
-        # self.dirname = r"C:\Users\TM273821\Desktop\SEM\Amel"
+        # self.dirname = r"C:\Users\TM273821\Desktop\SEM\TEst_complus"
 
 
         max_characters = 30  # Set character limit
@@ -91,9 +96,6 @@ class ButtonFrame(QWidget):
         """Create the directory if it does not exist."""
         if not os.path.exists(path):
             os.makedirs(path)
-            print(f"Directory created: {path}")
-        else:
-            print(f"Directory already exists: {path}")
 
     def init_ui(self):
         """Initialize the user interface"""
@@ -169,6 +171,7 @@ class ButtonFrame(QWidget):
         """Method to select folder and update checkbuttons"""
         self.select_folder()
         self.update_wafer()
+        self.image_radiobuttons()  # Refresh image type widget (radio buttons or slider)
 
     def update_wafer(self):
         """Update the appearance of radio buttons based on the existing
@@ -178,11 +181,16 @@ class ButtonFrame(QWidget):
             subdirs = [d for d in os.listdir(self.dirname) if
                        os.path.isdir(os.path.join(self.dirname, d))]
 
+            # Si il n'y a pas de sous-dossiers, chercher les wafers dans les fichiers KLARF
+            wafer_ids = []
+            if not subdirs:
+                wafer_ids = self.extract_wafer_ids_from_klarf()
+
             # Update the style of radio buttons based on the subdirectory presence
             for number in range(1, 27):
                 radio_button = self.radio_vars.get(number)
                 if radio_button:
-                    if str(number) in subdirs:
+                    if str(number) in subdirs or number in wafer_ids:
                         radio_button.setStyleSheet(WAFER_BUTTON_EXISTING_STYLE)
                     else:
                         radio_button.setStyleSheet(WAFER_BUTTON_MISSING_STYLE)
@@ -192,6 +200,44 @@ class ButtonFrame(QWidget):
                 radio_button = self.radio_vars.get(number)
                 radio_button.setStyleSheet(WAFER_BUTTON_MISSING_STYLE)
 
+    def extract_wafer_ids_from_klarf(self):
+        """Extract wafer IDs from KLARF files (.001) that contain COMPLUS4T."""
+        wafer_ids = []
+        
+        if not self.dirname:
+            return wafer_ids
+        
+        # Chercher les fichiers .001
+        try:
+            files = [f for f in os.listdir(self.dirname) 
+                    if f.endswith('.001') and os.path.isfile(os.path.join(self.dirname, f))]
+            
+            for file in files:
+                file_path = os.path.join(self.dirname, file)
+                try:
+                    with open(file_path, 'r', encoding='utf-8', errors='ignore') as f:
+                        content = f.read()
+                        
+                        # Check if file contains "COMPLUS4T"
+                        if 'COMPLUS4T' in content:
+                            # Search for all lines with WaferID
+                            # Pattern to extract number in quotes after WaferID
+                            pattern = r'WaferID\s+"@(\d+)"'
+                            matches = re.findall(pattern, content)
+                            
+                            # Add found IDs (converted to int)
+                            for match in matches:
+                                wafer_id = int(match)
+                                if wafer_id not in wafer_ids and 1 <= wafer_id <= 26:
+                                    wafer_ids.append(wafer_id)
+                except Exception as e:
+                    pass  # Error reading file
+        
+        except Exception as e:
+            pass  # Error listing files
+        
+        return wafer_ids
+
     def create_wafer(self):
         """Create a grid of radio buttons for wafer slots with exclusive selection."""
         group_box = QGroupBox("Wafer Slots")  # Add a title to the group
@@ -237,59 +283,125 @@ class ButtonFrame(QWidget):
             return self.selected_option
 
     def image_radiobuttons(self):
-        """Create a grid of radio buttons for wafer slots with exclusive selection."""
-        self.table_data = self.settings_window.get_table_data()
-        print(self.table_data)
-        number = len(self.table_data)
-
-        group_box = QGroupBox("Image type")  # Add a title to the group
-        group_box.setStyleSheet(GROUP_BOX_STYLE)  # Style the title
-
+        """Create a grid of radio buttons or slider for image type selection."""
+        # Remove old widget if it exists
+        if self.image_group_box is not None:
+            self.layout.removeWidget(self.image_group_box)
+            self.image_group_box.deleteLater()
+            self.image_group_box = None
+        
+        # Reset variables
+        self.image_slider = None
+        self.table_vars = None
+        self.slider_value = 0
+        
+        # Check if COMPLUS4T mode
+        is_complus4t = self._check_complus4t_in_dirname()
+        
+        # Change title based on mode
+        title = "Defect Size (nm)" if is_complus4t else "Image type"
+        group_box = QGroupBox(title)
+        group_box.setStyleSheet(GROUP_BOX_STYLE)
+        self.image_group_box = group_box
+        
         wafer_layout = QGridLayout()
-        wafer_layout.setContentsMargins(2, 20, 2, 2)  # Reduce internal margins
-        wafer_layout.setSpacing(5)  # Reduce spacing between widgets
-
-        self.table_vars = {}  # Store references to radio buttons
-
-        # Add radio buttons from 1 to 24, with 12 radio buttons per row
-        for i in range(number):
-            label = str(self.table_data[i]["Scale"]) + " - " + str(
-                self.table_data[i]["Image Type"])
-            radio_button = QRadioButton(label)
-            radio_button.setStyleSheet(WAFER_BUTTON_DEFAULT_STYLE)
-
-            # Connect the radio button to a handler for exclusive selection
-            radio_button.toggled.connect(self.get_selected_image)
-            self.table_vars[i] = radio_button
-
-            # Calculate the row and column for each radio button in the layout
-            row = (i) // 3  # Row starts at 1 after the label
-            col = (i) % 3  # Column ranges from 0 to 11
-
-            wafer_layout.addWidget(radio_button, row, col)
-
+        wafer_layout.setContentsMargins(2, 20, 2, 2)
+        wafer_layout.setSpacing(5)
+        
+        if is_complus4t:
+            # COMPLUS4T mode: create slider from 0 to 100 nm
+            self.image_slider = QSlider(Qt.Horizontal)
+            self.image_slider.setMinimum(0)
+            self.image_slider.setMaximum(100)
+            self.image_slider.setValue(0)
+            self.image_slider.setTickPosition(QSlider.TicksBelow)
+            self.image_slider.setTickInterval(10)
+            self.image_slider.valueChanged.connect(self.on_slider_changed)
+            
+            self.slider_value_label = QLabel("0 nm")
+            self.slider_value_label.setStyleSheet("color: black; font-size: 20px; font-weight: bold;")
+            
+            wafer_layout.addWidget(self.image_slider, 0, 0, 1, 2)
+            wafer_layout.addWidget(self.slider_value_label, 1, 0, 1, 2)
+        else:
+            # Normal mode: create radio buttons
+            self.table_data = self.settings_window.get_table_data()
+            number = len(self.table_data)
+            
+            self.table_vars = {}
+            
+            for i in range(number):
+                label = str(self.table_data[i]["Scale"]) + " - " + str(
+                    self.table_data[i]["Image Type"])
+                radio_button = QRadioButton(label)
+                radio_button.setStyleSheet(WAFER_BUTTON_DEFAULT_STYLE)
+                radio_button.toggled.connect(self.get_selected_image)
+                self.table_vars[i] = radio_button
+                
+                row = (i) // 3
+                col = (i) % 3
+                wafer_layout.addWidget(radio_button, row, col)
+        
         group_box.setLayout(wafer_layout)
-
-        # Add the QGroupBox to the main layout
         self.layout.addWidget(group_box, 1, 4, 1, 1)
+    
+    def _check_complus4t_in_dirname(self):
+        """Check if COMPLUS4T is present in .001 files in dirname."""
+        if not self.dirname or not os.path.exists(self.dirname):
+            return False
+        
+        try:
+            files = [f for f in os.listdir(self.dirname) 
+                    if f.endswith('.001') and os.path.isfile(os.path.join(self.dirname, f))]
+            
+            for file in files:
+                file_path = os.path.join(self.dirname, file)
+                try:
+                    with open(file_path, 'r', encoding='utf-8', errors='ignore') as f:
+                        content = f.read()
+                        if 'COMPLUS4T' in content:
+                            return True
+                except Exception as e:
+                    pass  # Error reading file
+        except Exception as e:
+            pass  # Error listing files
+        
+        return False
+    
+    def on_slider_changed(self, value):
+        """Handle slider value changes and trigger plot update."""
+        self.slider_value = value
+        self.slider_value_label.setText(f"{value} nm")
+        
+        # Trigger plot refresh if plot_frame exists
+        if self.plot_frame is not None:
+            self.plot_frame._update_plot()
 
     def refresh_radiobuttons(self):
         """Recreates the radio buttons after updating the data in Settings."""
         self.image_radiobuttons()  # Call your method to recreate the radio buttons
 
     def get_selected_image(self):
-        """Track the selected radio button."""
-        selected_number = None  # Variable to store the selected radio button number
-        n_types= len(self.table_vars.items())
-        # Iterate over all radio buttons
-        for number, radio_button in self.table_vars.items():
-
-            if radio_button.isChecked():
-                selected_number = number  # Track the currently selected radio button
-
-        if selected_number is not None:
-            self.selected_image = selected_number  # Store the selected option for further use
-            return self.selected_image, n_types
+        """Track the selected radio button or slider value."""
+        # Check if in COMPLUS4T mode (slider)
+        if self.image_slider is not None:
+            # Return slider value (0-100) as image type
+            return self.slider_value, 1
+        
+        # Normal mode: radio buttons
+        selected_number = None
+        if self.table_vars:
+            n_types = len(self.table_vars.items())
+            # Iterate over all radio buttons
+            for number, radio_button in self.table_vars.items():
+                if radio_button.isChecked():
+                    selected_number = number
+            
+            if selected_number is not None:
+                self.selected_image = selected_number
+                return self.selected_image, n_types
+        
+        return None
 
     def create_radiobuttons(self):
         """Create radio buttons for tools and a settings button."""
@@ -444,7 +556,7 @@ class ButtonFrame(QWidget):
             QApplication.processEvents()  # Ensures the interface is updated
             task_function(*args, **kwargs)
             elapsed_time = time.time() - start_time
-            print(f"{task_name} completed in {elapsed_time:.2f} seconds.")
+            pass  # Task completed
 
         if self.split_rename.isChecked():
             execute_with_timer("Cleaning of folders", sem_class.clean)

semapp/Plot/__init__.py

@@ -1,8 +1,8 @@
 """
 Plot package initialization.
-Contains SEM image processing functionality.
+Contains plotting and visualization functionality for SEM data.
 """
 
-from .utils import*
+from .utils import *
 
-__all__ = ['Process'] 
+__all__ = []