setiastrosuitepro 1.6.0__py3-none-any.whl → 1.6.4.post1__py3-none-any.whl

setiastro/saspro/abe.py

@@ -75,15 +75,28 @@ def _fit_poly_on_small(small: np.ndarray, points: np.ndarray, degree: int, patch
 
     if small.ndim == 3 and small.shape[2] == 3:
         bg_small = np.zeros_like(small, dtype=np.float32)
+        
+        # Batch collect samples: (num_samples, 3)
+        # We need N samples. z will be list of (3,) arrays
+        
+        # Pre-allocate Z: (N, 3)
+        Z = np.zeros((len(xs), 3), dtype=np.float32)
+        
+        for k, (x, y) in enumerate(zip(xs, ys)):
+            x0, x1 = max(0, x - half), min(W, x + half + 1)
+            y0, y1 = max(0, y - half), min(H, y + half + 1)
+            # Efficiently compute median for all channels in this patch
+            patch = small[y0:y1, x0:x1, :]
+            Z[k] = np.median(patch, axis=(0, 1))
+
+        # Solve once: A is (N, terms), Z is (N, 3) -> coeffs is (terms, 3)
+        coeffs_all, *_ = np.linalg.lstsq(A, Z, rcond=None)
+        
+        # Evaluate per channel
         for c in range(3):
-            z = []
-            for x, y in zip(xs, ys):
-                x0, x1 = max(0, x - half), min(W, x + half + 1)
-                y0, y1 = max(0, y - half), min(H, y + half + 1)
-                z.append(np.median(small[y0:y1, x0:x1, c]))
-            z = np.asarray(z, dtype=np.float32)
-            coeffs, *_ = np.linalg.lstsq(A, z, rcond=None)
-            bg_small[..., c] = evaluate_polynomial(H, W, coeffs.astype(np.float32), degree)
+            # coeffs_all[:, c] gives the terms for channel c
+            bg_small[..., c] = evaluate_polynomial(H, W, coeffs_all[:, c].astype(np.float32), degree)
+            
         return bg_small
     else:
         z = []
@@ -471,17 +484,22 @@ class ABEDialog(QDialog):
     """
     def __init__(self, parent, document: ImageDocument):
         super().__init__(parent)
-        self.setWindowTitle("Automatic Background Extraction (ABE)")
+        self.setWindowTitle(self.tr("Automatic Background Extraction (ABE)"))
 
         # IMPORTANT: avoid “attached modal sheet” behavior on some Linux WMs
         self.setWindowFlag(Qt.WindowType.Window, True)
-        # keep it blocking if you want, but as a top-level window
-        self.setWindowModality(Qt.WindowModality.ApplicationModal)
+        # Non-modal: allow user to switch between images while dialog is open
+        self.setWindowModality(Qt.WindowModality.NonModal)
         self.setModal(False)
         #self.setAttribute(Qt.WidgetAttribute.WA_DeleteOnClose, True)
 
+        self._main = parent
         self.doc = document
 
+        # Connect to active document change signal
+        if hasattr(self._main, "currentDocumentChanged"):
+            self._main.currentDocumentChanged.connect(self._on_active_doc_changed)
+
         self._preview_scale = 1.0
         self._preview_qimg = None
         self._last_preview = None  # backing ndarray for QImage lifetime
@@ -500,10 +518,10 @@ class ABEDialog(QDialog):
         self.sp_samples = QSpinBox(); self.sp_samples.setRange(20, 10000); self.sp_samples.setSingleStep(20); self.sp_samples.setValue(120)
         self.sp_down = QSpinBox(); self.sp_down.setRange(1, 32); self.sp_down.setValue(4)
         self.sp_patch = QSpinBox(); self.sp_patch.setRange(5, 151); self.sp_patch.setSingleStep(2); self.sp_patch.setValue(15)
-        self.chk_use_rbf = QCheckBox("Enable RBF refinement (after polynomial)"); self.chk_use_rbf.setChecked(True)
+        self.chk_use_rbf = QCheckBox(self.tr("Enable RBF refinement (after polynomial)")); self.chk_use_rbf.setChecked(True)
         self.sp_rbf = QSpinBox(); self.sp_rbf.setRange(0, 1000); self.sp_rbf.setValue(100)  # shown as ×0.01 below
-        self.chk_make_bg_doc = QCheckBox("Create background document"); self.chk_make_bg_doc.setChecked(False)
-        self.chk_preview_bg   = QCheckBox("Preview background instead of corrected"); self.chk_preview_bg.setChecked(False)
+        self.chk_make_bg_doc = QCheckBox(self.tr("Create background document")); self.chk_make_bg_doc.setChecked(False)
+        self.chk_preview_bg   = QCheckBox(self.tr("Preview background instead of corrected")); self.chk_preview_bg.setChecked(False)
 
         # Preview area
         self.preview_label = QLabel(alignment=Qt.AlignmentFlag.AlignCenter)
@@ -516,10 +534,10 @@ class ABEDialog(QDialog):
         self.preview_scroll.setVerticalScrollBarPolicy(Qt.ScrollBarPolicy.ScrollBarAsNeeded)
 
         # Buttons
-        self.btn_preview = QPushButton("Preview")
-        self.btn_apply   = QPushButton("Apply")
-        self.btn_close   = QPushButton("Close")
-        self.btn_clear   = QPushButton("Clear Exclusions")
+        self.btn_preview = QPushButton(self.tr("Preview"))
+        self.btn_apply   = QPushButton(self.tr("Apply"))
+        self.btn_close   = QPushButton(self.tr("Close"))
+        self.btn_clear   = QPushButton(self.tr("Clear Exclusions"))
         self.btn_preview.clicked.connect(self._do_preview)
         self.btn_apply.clicked.connect(self._do_apply)
         self.btn_close.clicked.connect(self.close)
@@ -527,15 +545,15 @@ class ABEDialog(QDialog):
 
         # Layout
         params = QFormLayout()
-        params.addRow("Polynomial degree:", self.sp_degree)
-        params.addRow("# sample points:",   self.sp_samples)
-        params.addRow("Downsample factor:", self.sp_down)
-        params.addRow("Patch size (px):",   self.sp_patch)
+        params.addRow(self.tr("Polynomial degree:"), self.sp_degree)
+        params.addRow(self.tr("# sample points:"),   self.sp_samples)
+        params.addRow(self.tr("Downsample factor:"), self.sp_down)
+        params.addRow(self.tr("Patch size (px):"),   self.sp_patch)
 
-        rbf_box = QGroupBox("RBF Refinement")
+        rbf_box = QGroupBox(self.tr("RBF Refinement"))
         rbf_form = QFormLayout()
         rbf_form.addRow(self.chk_use_rbf)
-        rbf_form.addRow("Smooth (×0.01):", self.sp_rbf)
+        rbf_form.addRow(self.tr("Smooth (x0.01):"), self.sp_rbf)
         rbf_box.setLayout(rbf_form)
 
         opts = QVBoxLayout()
@@ -612,6 +630,17 @@ class ABEDialog(QDialog):
         bar.addWidget(self.btn_autostr)
         return bar
 
+    # ----- active document change -----
+    def _on_active_doc_changed(self, doc):
+        """Called when user clicks a different image window."""
+        if doc is None or getattr(doc, "image", None) is None:
+            return
+        self.doc = doc
+        self._polygons.clear()
+        self._drawing_poly = None
+        self._preview_source_f01 = None
+        self._populate_initial_preview()
+
     # ----- data helpers -----
     def _get_source_float(self) -> np.ndarray | None:
         src = np.asarray(self.doc.image)
@@ -813,12 +842,34 @@ class ABEDialog(QDialog):
                 pass
 
             self._set_status("Done")
-            self.accept()
+            # Dialog stays open so user can apply to other images
+            # Refresh to use the now-active document for next operation
+            self._refresh_document_from_active()
 
         except Exception as e:
             self._set_status("Error")
             QMessageBox.critical(self, "Apply failed", str(e))
 
+    def _refresh_document_from_active(self):
+        """
+        Refresh the dialog's document reference to the currently active document.
+        This allows reusing the same dialog on different images.
+        """
+        try:
+            main = self.parent()
+            if main and hasattr(main, "_active_doc"):
+                new_doc = main._active_doc()
+                if new_doc is not None and new_doc is not self.doc:
+                    self.doc = new_doc
+                    # Reset preview state for new document
+                    self._preview_source_f01 = None
+                    self._last_preview = None
+                    self._preview_qimg = None
+                    # Clear polygons since they were for old image
+                    self._clear_polys()
+        except Exception:
+            pass
+
 
     # ----- exclusion polygons & mask -----
     def _clear_polys(self):

setiastro/saspro/aberration_ai.py

@@ -292,14 +292,14 @@ class _ONNXWorker(QThread):
 class AberrationAIDialog(QDialog):
     def __init__(self, parent, docman, get_active_doc_callable, icon: QIcon | None = None):
         super().__init__(parent)
-        self.setWindowTitle("R.A.'s Aberration Correction (AI)")
+        self.setWindowTitle(self.tr("R.A.'s Aberration Correction (AI)"))
         if icon is not None:
             self.setWindowIcon(icon)
 
         # Normalize window behavior across platforms
         self.setWindowFlag(Qt.WindowType.Window, True)
-        # This is a “big operation” tool; app-modal is usually fine here
-        self.setWindowModality(Qt.WindowModality.ApplicationModal)
+        # Non-modal: allow user to switch between images while dialog is open
+        self.setWindowModality(Qt.WindowModality.NonModal)
         self.setModal(False)
         #self.setAttribute(Qt.WidgetAttribute.WA_DeleteOnClose, True)
 
@@ -312,40 +312,40 @@ class AberrationAIDialog(QDialog):
 
         # Model row
         row = QHBoxLayout()
-        row.addWidget(QLabel("Model:"))
+        row.addWidget(QLabel(self.tr("Model:")))
         self.model_label = QLabel("—")
         self.model_label.setToolTip("")
-        btn_browse = QPushButton("Browse…"); btn_browse.clicked.connect(self._browse_model)
+        btn_browse = QPushButton(self.tr("Browse…")); btn_browse.clicked.connect(self._browse_model)
         row.addWidget(self.model_label, 1)
         row.addWidget(btn_browse)
         v.addLayout(row)
 
         # Providers row
         row2 = QHBoxLayout()
-        self.chk_auto = QCheckBox("Auto GPU (if available)")
+        self.chk_auto = QCheckBox(self.tr("Auto GPU (if available)"))
         self.chk_auto.setChecked(True)
         row2.addWidget(self.chk_auto)
         self.cmb_provider = QComboBox()
-        row2.addWidget(QLabel("Provider:"))
+        row2.addWidget(QLabel(self.tr("Provider:")))
         row2.addWidget(self.cmb_provider, 1)
         v.addLayout(row2)
 
         # Params row
         row3 = QHBoxLayout()
-        row3.addWidget(QLabel("Patch"))
+        row3.addWidget(QLabel(self.tr("Patch")))
         self.spin_patch = QSpinBox(minimum=128, maximum=2048); self.spin_patch.setValue(512)
         row3.addWidget(self.spin_patch)
-        row3.addWidget(QLabel("Overlap"))
+        row3.addWidget(QLabel(self.tr("Overlap")))
         self.spin_overlap = QSpinBox(minimum=16, maximum=512); self.spin_overlap.setValue(64)
         row3.addWidget(self.spin_overlap)
         v.addLayout(row3)
 
         # Download / Open folder
         row4 = QHBoxLayout()
-        btn_latest = QPushButton("Download latest model…")
+        btn_latest = QPushButton(self.tr("Download latest model…"))
         btn_latest.clicked.connect(self._download_latest_model)
         row4.addWidget(btn_latest)
-        btn_openfolder = QPushButton("Open model folder")
+        btn_openfolder = QPushButton(self.tr("Open model folder"))
         btn_openfolder.clicked.connect(self._open_model_folder)
         row4.addWidget(btn_openfolder)
         row4.addStretch(1)
@@ -354,8 +354,8 @@ class AberrationAIDialog(QDialog):
         # Progress + actions
         self.progress = QProgressBar(); self.progress.setRange(0, 100); v.addWidget(self.progress)
         row5 = QHBoxLayout()
-        self.btn_run = QPushButton("Run"); self.btn_run.clicked.connect(self._run)
-        btn_close = QPushButton("Close"); btn_close.clicked.connect(self.reject)
+        self.btn_run = QPushButton(self.tr("Run")); self.btn_run.clicked.connect(self._run)
+        btn_close = QPushButton(self.tr("Close")); btn_close.clicked.connect(self.reject)
         row5.addStretch(1); row5.addWidget(self.btn_run); row5.addWidget(btn_close)
         v.addLayout(row5)
 
@@ -680,7 +680,7 @@ class AberrationAIDialog(QDialog):
         )
 
         self.progress.setValue(100)
-        self.accept()
+        # Dialog stays open so user can apply to other images
 
     def _on_worker_finished(self):
         # If dialog is already gone, this method is never called because the receiver (self)

setiastro/saspro/add_stars.py

@@ -224,10 +224,11 @@ class AddStarsDialog(QDialog):
     stars_added = pyqtSignal(object, object)
     def __init__(self, main, parent=None):
         super().__init__(parent)
-        self.setWindowTitle("Add Stars to Image")
+        self.setWindowTitle(self.tr("Add Stars to Image"))
 
         self.setWindowFlag(Qt.WindowType.Window, True)
-        self.setWindowModality(Qt.WindowModality.ApplicationModal)
+        # Non-modal: allow user to switch between images while dialog is open
+        self.setWindowModality(Qt.WindowModality.NonModal)
         self.setModal(False)
         #self.setAttribute(Qt.WidgetAttribute.WA_DeleteOnClose, True)
 
@@ -277,27 +278,27 @@ class AddStarsDialog(QDialog):
         grid = QGridLayout()
 
         # Blend type
-        grid.addWidget(QLabel("Blend Type:"), 0, 0)
+        grid.addWidget(QLabel(self.tr("Blend Type:")), 0, 0)
         self.cmb_blend = QComboBox(); self.cmb_blend.addItems(["Screen", "Add"])
         self.cmb_blend.currentIndexChanged.connect(self.update_preview)
         grid.addWidget(self.cmb_blend, 0, 1)
 
         # Starless source
-        grid.addWidget(QLabel("Starless View:"), 1, 0)
+        grid.addWidget(QLabel(self.tr("Starless View:")), 1, 0)
         self.cmb_starless = QComboBox(); grid.addWidget(self.cmb_starless, 1, 1)
-        btn_sless_file = QPushButton("Load from File"); btn_sless_file.clicked.connect(lambda: self._load_from_file('starless'))
+        btn_sless_file = QPushButton(self.tr("Load from File")); btn_sless_file.clicked.connect(lambda: self._load_from_file('starless'))
         grid.addWidget(btn_sless_file, 1, 2)
 
         # Stars-only source
-        grid.addWidget(QLabel("Stars-Only View:"), 2, 0)
+        grid.addWidget(QLabel(self.tr("Stars-Only View:")), 2, 0)
         self.cmb_stars   = QComboBox(); grid.addWidget(self.cmb_stars, 2, 1)
-        btn_stars_file = QPushButton("Load from File"); btn_stars_file.clicked.connect(lambda: self._load_from_file('stars'))
+        btn_stars_file = QPushButton(self.tr("Load from File")); btn_stars_file.clicked.connect(lambda: self._load_from_file('stars'))
         grid.addWidget(btn_stars_file, 2, 2)
 
         layout.addLayout(grid)
 
         refresh_row = QHBoxLayout()
-        btn_refresh = QPushButton("Refresh Views")
+        btn_refresh = QPushButton(self.tr("Refresh Views"))
         btn_refresh.clicked.connect(self._populate_doc_combos)
         refresh_row.addStretch(1)
         refresh_row.addWidget(btn_refresh)
@@ -305,7 +306,7 @@ class AddStarsDialog(QDialog):
 
         # Ratio slider
         row = QHBoxLayout()
-        row.addWidget(QLabel("Blend Ratio (Screen/Add Intensity):"))
+        row.addWidget(QLabel(self.tr("Blend Ratio (Screen/Add Intensity):")))
         self.slider_ratio = QSlider(Qt.Orientation.Horizontal)
         self.slider_ratio.setRange(0, 100); self.slider_ratio.setValue(100)
         self.slider_ratio.setTickInterval(10); self.slider_ratio.setTickPosition(QSlider.TickPosition.TicksBelow)
@@ -315,8 +316,8 @@ class AddStarsDialog(QDialog):
 
         # Buttons
         brow = QHBoxLayout(); brow.addStretch(1)
-        btn_apply = QPushButton("Apply"); btn_apply.clicked.connect(self._apply)
-        btn_cancel= QPushButton("Cancel"); btn_cancel.clicked.connect(self.reject)
+        btn_apply = QPushButton(self.tr("Apply")); btn_apply.clicked.connect(self._apply)
+        btn_cancel= QPushButton(self.tr("Cancel")); btn_cancel.clicked.connect(self.reject)
         brow.addWidget(btn_apply); brow.addWidget(btn_cancel)
         layout.addLayout(brow)
 
@@ -574,7 +575,9 @@ class AddStarsDialog(QDialog):
 
         # Emit (target_doc, blended_image)
         self.stars_added.emit(target_doc, self.blended_image.astype(np.float32, copy=False))
-        self.accept()
+        # Dialog stays open so user can apply to other images
+        # Refresh combo boxes for next operation
+        self._populate_doc_combos()
 
 
     # Ensure initial fit once shown