setiastrosuitepro 1.6.7__py3-none-any.whl → 1.7.0__py3-none-any.whl

setiastro/saspro/gui/main_window.py

@@ -129,7 +129,7 @@ from PyQt6.QtGui import (QPixmap, QColor, QIcon, QKeySequence, QShortcut,
 
 # ----- QtCore -----
 from PyQt6.QtCore import (Qt, pyqtSignal, QCoreApplication, QTimer, QSize, QSignalBlocker,  QModelIndex, QThread, QUrl, QSettings, QEvent, QByteArray, QObject,
-    QPropertyAnimation, QEasingCurve, QElapsedTimer
+    QPropertyAnimation, QEasingCurve, QElapsedTimer, QPoint
 )
 
 from PyQt6.QtNetwork import QNetworkAccessManager, QNetworkRequest, QNetworkReply
@@ -193,9 +193,9 @@ from setiastro.saspro.resources import (
     nbtorgb_path, freqsep_path, contsub_path, halo_path, cosmic_path,
     satellite_path, imagecombine_path, wrench_path, eye_icon_path,multiscale_decomp_path,
     disk_icon_path, nuke_path, hubble_path, collage_path, annotated_path,
-    colorwheel_path, font_path, csv_icon_path, spinner_path, wims_path,
+    colorwheel_path, font_path, csv_icon_path, spinner_path, wims_path, narrowbandnormalization_path,
     wimi_path, linearfit_path, debayer_path, aberration_path, acv_icon_path,
-    functionbundles_path, viewbundles_path, selectivecolor_path, rgbalign_path,
+    functionbundles_path, viewbundles_path, selectivecolor_path, rgbalign_path, planetarystacker_path,
     background_path, script_icon_path
 )
 
@@ -361,6 +361,69 @@ class UiStallDetector(QObject):
             print(f"[UI STALL] tick late by {late_ms:.0f} ms (elapsed={elapsed_ms:.0f} ms)", flush=True)
             self._dump_all_threads_print()
 
+def _strip_filename_ext(title: str) -> str:
+    t = (title or "").strip()
+    if not t:
+        return t
+    base, ext = os.path.splitext(t)
+    # treat as extension only if it looks like one: .fit .fits .tif .tiff .xisf etc
+    if ext and 1 <= len(ext) <= 10 and all(ch.isalnum() for ch in ext[1:]):
+        return base
+    return t
+
+
+
+_DECOR_GLYPHS = "■●◆▲▪▫•◼◻◾◽🔗"
+
+def normalize_doc_title(s: str) -> str:
+    s = (s or "").strip()
+
+    # remove our textual prefix too
+    if s.startswith("[LINK] "):
+        s = s[len("[LINK] "):].strip()
+
+    # strip common UI decorations if you already have this helper
+    try:
+        s = _strip_ui_decorations(s)
+    except Exception:
+        pass
+
+    # remove any leading decorator glyphs repeatedly: "🔗 ", "■ ", etc.
+    while len(s) >= 2 and s[0] in _DECOR_GLYPHS and s[1] == " ":
+        s = s[2:].lstrip()
+
+    # also remove any stray decorator glyphs that got embedded (rare but happens)
+    s = re.sub(rf"[{re.escape(_DECOR_GLYPHS)}]", "", s).strip()
+
+    return s
+
+_VIEW_SUFFIX_RE = re.compile(r"\s+\[View\s+\d+\]\s*$")
+
+def _normalize_title_for_compare(t: str) -> str:
+    t = (t or "").strip()
+    if not t:
+        return ""
+
+    # strip UI decorations (🔗, ■, etc)
+    try:
+        t = _strip_ui_decorations(t)
+    except Exception:
+        pass
+
+    # strip trailing "[View N]" if present
+    t = _VIEW_SUFFIX_RE.sub("", t).strip()
+
+    # strip filename-like extension
+    try:
+        t = _strip_filename_ext(t)
+    except Exception:
+        # fallback: only strip if it looks like an ext
+        base, ext = os.path.splitext(t)
+        if ext and len(ext) <= 10:
+            t = base
+
+    return t.strip()
+
 class AstroSuiteProMainWindow(
     DockMixin, MenuMixin, ToolbarMixin, FileMixin,
     ThemeMixin, GeometryMixin, ViewMixin, HeaderMixin, MaskMixin, UpdateMixin,
@@ -541,7 +604,10 @@ class AstroSuiteProMainWindow(
         self.docman.documentAdded.connect(self._on_document_added)
         self.mdi.viewStateDropped.connect(self._on_mdi_viewstate_drop)
         self.mdi.linkViewDropped.connect(self._on_linkview_drop)
-
+        self._mdi_open_batch = 0
+        self._mdi_place_mode = "cascade"   # or "tile"
+        self._mdi_next_pos = None          # QPoint in MDI coords
+        self._mdi_cascade_step = 28
         self.doc_manager.set_mdi_area(self.mdi)
         # Coalesce undo/redo label refreshes
         self._undo_redo_refresh_pending = False
@@ -2597,10 +2663,27 @@ class AstroSuiteProMainWindow(
         return f"{name}{dims}"
 
     def _update_explorer_item_for_doc(self, doc):
-        for i in range(self.explorer.count()):
-            it = self.explorer.item(i)
-            if it.data(Qt.ItemDataRole.UserRole) is doc:
-                it.setText(self._format_explorer_title(doc))
+        # Delegate to DockMixin implementation if present
+        try:
+            return super()._update_explorer_item_for_doc(doc)
+        except Exception:
+            pass
+
+        # Fallback: tree-safe implementation
+        if not hasattr(self, "explorer") or self.explorer is None:
+            return
+        try:
+            n = self.explorer.topLevelItemCount()
+        except Exception:
+            return
+
+        for i in range(n):
+            it = self.explorer.topLevelItem(i)
+            if it.data(0, Qt.ItemDataRole.UserRole) is doc:
+                try:
+                    self._refresh_explorer_row(it, doc)
+                except Exception:
+                    pass
                 return
     #-----------FUNCTIONS----------------
 
@@ -3825,6 +3908,19 @@ class AstroSuiteProMainWindow(
 
         dlg.show()
 
+    def _open_narrowband_normalization_tool(self):
+        # Correct module import
+        from setiastro.saspro.narrowband_normalization import NarrowbandNormalization
+
+        w = NarrowbandNormalization(doc_manager=self.docman, parent=self)
+        w.setAttribute(Qt.WidgetAttribute.WA_DeleteOnClose, True)
+        w.setWindowTitle("Narrowband Normalization")
+        try:
+            w.setWindowIcon(QIcon(narrowbandnormalization_path))
+        except Exception:
+            pass
+        w.show()
+
     def _open_ppp_tool(self):
         from setiastro.saspro.perfect_palette_picker import PerfectPalettePicker
         w = PerfectPalettePicker(doc_manager=self.docman)  # parent gives access to _spawn_subwindow_for
@@ -4176,6 +4272,14 @@ class AstroSuiteProMainWindow(
         dlg.setWindowIcon(QIcon(livestacking_path))
         dlg.show()
 
+    def _open_planetary_stacker(self):
+        # import locally to avoid startup cost / circular imports
+        from setiastro.saspro.serviewer import SERViewer
+        dlg = SERViewer(self)
+        dlg.setWindowFlag(Qt.WindowType.Window, True)
+        dlg.setWindowIcon(QIcon(planetarystacker_path))        
+        dlg.show()
+
     def _open_stacking_suite(self):
         # Reuse if we already have one
         dlg = getattr(self, "_stacking_suite", None)
@@ -4219,6 +4323,115 @@ class AstroSuiteProMainWindow(
             except Exception:
                 pass
 
+    def _convert_mono_to_rgb_active(self):
+        """
+        Convert active mono document to RGB by duplicating the channel.
+        Updates the active document in-place (undoable).
+        """
+        dm = getattr(self, "docman", None)
+        if dm is None:
+            return
+
+        try:
+            doc = dm.get_active_document()
+        except Exception:
+            doc = None
+        if doc is None:
+            return
+
+        img = getattr(doc, "image", None)
+        if img is None:
+            return
+
+        import numpy as np
+
+        x = np.asarray(img)
+
+        # Already RGB?
+        if x.ndim == 3 and x.shape[-1] == 3:
+            try:
+                name = getattr(doc, "display_name", lambda: None)() or getattr(doc, "name", "") or "Active"
+            except Exception:
+                name = "Active"
+            if hasattr(self, "_log"):
+                self._log(f"Mono → RGB: '{name}' is already RGB (shape={getattr(x,'shape',None)}).")
+            return
+
+        # Determine what we're converting FROM
+        src_desc = "unknown"
+        if x.ndim == 2:
+            mono = x
+            src_desc = "mono (H×W)"
+        elif x.ndim == 3 and x.shape[-1] == 1:
+            mono = x[..., 0]
+            src_desc = "mono (H×W×1)"
+        else:
+            # Unknown format (e.g., multi-channel >3)
+            try:
+                name = getattr(doc, "display_name", lambda: None)() or getattr(doc, "name", "") or "Active"
+            except Exception:
+                name = "Active"
+            if hasattr(self, "_log"):
+                self._log(f"Mono → RGB: '{name}' not convertible (shape={getattr(x,'shape',None)}).")
+            return
+
+        before_shape = getattr(x, "shape", None)
+        before_dtype = getattr(x, "dtype", None)
+
+        mono = mono.astype(np.float32, copy=False)
+        rgb = np.stack([mono, mono, mono], axis=-1)
+
+        # metadata: preserve existing, but force "not mono"
+        try:
+            md = dict(getattr(doc, "metadata", None) or {})
+        except Exception:
+            md = {}
+
+        md["is_mono"] = False
+        md["color_model"] = "RGB"
+        md["channels"] = 3
+        md["source"] = (md.get("source") or "Edit")
+
+        # If you track op params for history explorer
+        md["__op_params__"] = {
+            "op": "mono_to_rgb",
+            "mode": "triplicate",
+            "from": str(src_desc),
+            "from_shape": tuple(before_shape) if before_shape is not None else None,
+            "to_shape": tuple(rgb.shape),
+        }
+
+        # name for logging
+        try:
+            name = getattr(doc, "display_name", lambda: None)() or getattr(doc, "name", "") or "Active"
+        except Exception:
+            name = "Active"
+
+        try:
+            dm.update_active_document(
+                rgb,
+                metadata=md,
+                step_name="Mono → RGB",
+                doc=doc,  # explicit is safer
+            )
+
+            if hasattr(self, "_log"):
+                self._log(
+                    f"Mono → RGB: '{name}' converted {src_desc} "
+                    f"(shape={before_shape}, dtype={before_dtype}) → "
+                    f"RGB (shape={rgb.shape}, dtype={rgb.dtype})."
+                )
+
+        except Exception:
+            import traceback
+            try:
+                from PyQt6.QtWidgets import QMessageBox
+                QMessageBox.critical(self, "Mono → RGB", traceback.format_exc())
+            except Exception:
+                pass
+
+
+
     def _on_stackingsuite_relaunch(self, old_dir: str, new_dir: str):
         # Optional: respond to dialog's relaunch request
         try:
@@ -7779,12 +7992,8 @@ class AstroSuiteProMainWindow(
                 return cand
             n += 1
 
+    
     def _doc_window_title(self, doc) -> str:
-        """
-        Best-effort human title for a subwindow.
-        Prefer metadata['display_name'] (what duplication sets),
-        then doc.display_name(), then basename(file_path).
-        """
         md = getattr(doc, "metadata", {}) or {}
 
         t = (md.get("display_name") or "").strip()
@@ -7797,8 +8006,7 @@ class AstroSuiteProMainWindow(
         if not t:
             fp = (md.get("file_path") or "").strip()
             if fp:
-                import os
-                t = os.path.basename(fp)
+                t = os.path.splitext(os.path.basename(fp))[0]   # ✅ strip ext here too
 
         t = t or "Untitled"
 
@@ -7807,8 +8015,48 @@ class AstroSuiteProMainWindow(
             t = _strip_ui_decorations(t)
         except Exception:
             pass
+
+        # ✅ ALWAYS strip filename-like extension at the very end
+        t = _strip_filename_ext(t)
+
         return t
 
+    def _mdi_begin_open_batch(self, mode: str = "cascade"):
+        self._mdi_open_batch += 1
+        self._mdi_place_mode = mode or "cascade"
+        self._mdi_next_pos = None
+
+    def _mdi_end_open_batch(self):
+        self._mdi_open_batch = max(0, self._mdi_open_batch - 1)
+        if self._mdi_open_batch == 0:
+            self._mdi_next_pos = None
+
+    def _mdi_compute_initial_pos(self) -> QPoint:
+        area = (self.mdi.viewport().geometry() if self.mdi.viewport() else self.mdi.contentsRect())
+        # Put first window a bit inset so titlebars don’t clip
+        return QPoint(area.left() + 18, area.top() + 18)
+
+    def _mdi_place_subwindow(self, sw, target_w: int, target_h: int):
+        """Deterministic placement. Uses a stable cursor during batch opens."""
+        vp = self.mdi.viewport()
+        area = vp.geometry() if vp else self.mdi.contentsRect()
+
+        if self._mdi_next_pos is None:
+            self._mdi_next_pos = self._mdi_compute_initial_pos()
+
+        x = self._mdi_next_pos.x()
+        y = self._mdi_next_pos.y()
+
+        # keep inside viewport; reset when we hit edge
+        if (x + target_w > area.right() - 10) or (y + 40 > area.bottom() - 10):
+            x = area.left() + 18
+            y = area.top() + 18
+
+        sw.move(x, y)
+
+        # advance cursor
+        step = int(self._mdi_cascade_step)
+        self._mdi_next_pos = QPoint(x + step, y + step)
 
     def _spawn_subwindow_for(self, doc, *, force_new: bool = False):
         """
@@ -7945,7 +8193,8 @@ class AstroSuiteProMainWindow(
         # We target ~60% of the viewport height, clamped to sane bounds.
         # -------------------------------------------------------------------------
         vp = self.mdi.viewport()
-        area = vp.rect() if vp else self.mdi.rect()
+        # Use viewport geometry in MDI coordinates (NOT viewport-local rect)
+        area = vp.geometry() if vp else self.mdi.contentsRect()
         
         # Determine aspect ratio
         img_w = img_h = None
@@ -7982,54 +8231,22 @@ class AstroSuiteProMainWindow(
         target_h = max(200, target_h)
 
         sw.resize(target_w, target_h)
-        sw.showNormal()  # CRITICAL: clears any "maximized" flag from previous active window
+        sw.showNormal()  # clears any "maximized" flag from previous active window
 
-        # -------------------------------------------------------------------------
-        # Smart Cascade: Position relative to the *currently active* window
-        # (before we make the new one active).
-        # -------------------------------------------------------------------------
-        new_x, new_y = 0, 0
-        
-        # Get dominant/active window *before* we activate the new one
-        active = self.mdi.activeSubWindow()
-        if active and active.isVisible() and not (active.windowState() & Qt.WindowState.WindowMinimized):
-            # Cascade from the active window
-            geo = active.geometry()
-            new_x = geo.x() + 30
-            new_y = geo.y() + 30
-        else:
-            # Fallback: try to find the "last added" visible window to cascade from
-            # (useful if active is None but windows exist)
-            try:
-                subs = [s for s in self.mdi.subWindowList() if s.isVisible() and s is not sw]
-                if subs:
-                    # simplistic "last created" might be at end of list
-                    last = subs[-1]
-                    geo = last.geometry()
-                    new_x = geo.x() + 30
-                    new_y = geo.y() + 30
+        # Deterministic placement (batch-aware)
+        try:
+            self._mdi_place_subwindow(sw, target_w, target_h)
+        except Exception:
+            # absolute fallback: top-left-ish
+            try:
+                sw.move(area.left() + 18, area.top() + 18)
             except Exception:
                 pass
 
-        # Bounds check: don't let it drift completely off-screen
-        # (allow valid title bar to be visible at least)
-        if (new_x + target_w > area.width() + 50) or (new_y + 50 > area.height()):
-            new_x = 0
-            new_y = 0
-        
-        # Clamp to 0 if negative for some reason
-        new_x = max(0, new_x)
-        new_y = max(0, new_y)
-
-        sw.move(new_x, new_y)
-
-        # ❌ removed the "fill MDI viewport" block - we *don't* want full-monitor first window
-
         # Show / activate
         sw.show()
         sw.raise_()
         self.mdi.setActiveSubWindow(sw)
-        # (no second setWindowTitle() here)
 
         # Optional minimize/restore interceptor
         if hasattr(self, "_minimize_interceptor"):
@@ -8114,6 +8331,11 @@ class AstroSuiteProMainWindow(
         except Exception:
             pass
 
+        try:
+            self._fix_mdi_titlebar_emboss("#dcdcdc" if self._theme_mode()=="dark" else "#f0f0f0")
+        except Exception:
+            pass
+
         # -- 11) If this is the first window and it's an image, mimic "Cascade Views"
         try:
             if first_window and not is_table:
@@ -8208,25 +8430,57 @@ class AstroSuiteProMainWindow(
             "autostretch_target": float(getattr(source_view, "autostretch_target", 0.25)),
         }
 
-        # 2) New name (strip UI decorations if any)
-        base_name = ""
+        # 2) New name (normalized: NO decorators like 🔗■●◆▲▪▫•◼◻◾◽)
         try:
-            base_name = base_doc.display_name() or "Untitled"
+            base_name = self._doc_window_title(base_doc)  # might include decorations
         except Exception:
             base_name = "Untitled"
 
+        # Normalize it so uniqueness checks don't miss decorated titles
+        try:
+            base_name = normalize_doc_title(base_name)
+        except Exception:
+            base_name = (base_name or "Untitled").strip()
+
+        # Build a set of existing document names (normalized)
+        existing = set()
         try:
-            base_name = _strip_ui_decorations(base_name)
+            dm = getattr(self, "doc_manager", None) or getattr(self, "docman", None)
+            docs = []
+
+            # Prefer an official accessor if you have one
+            if dm is not None:
+                if hasattr(dm, "documents"):
+                    docs = list(dm.documents())
+                elif hasattr(dm, "_docs"):
+                    docs = list(dm._docs)
+
+            for d in docs:
+                try:
+                    dn = ""
+                    md = getattr(d, "metadata", {}) or {}
+                    dn = (md.get("display_name") or "").strip() or (d.display_name() or "").strip()
+                    dn = normalize_doc_title(dn)
+                    if dn:
+                        existing.add(dn)
+                except Exception:
+                    pass
         except Exception:
-            # minimal fallback: remove our known prefix/glyphs
-            while len(base_name) >= 2 and base_name[1] == " " and base_name[0] in "â- â--â--†â-²â-ªâ-«â€¢â--¼â--»â--¾â--½":
-                base_name = base_name[2:]
-            if base_name.startswith("Active View: "):
-                base_name = base_name[len("Active View: "):]
+            pass
+
+        # Pick a unique duplicate name: base_duplicate, base_duplicate2, ...
+        candidate = f"{base_name}_duplicate"
+        if candidate in existing:
+            n = 2
+            while True:
+                cand = f"{base_name}_duplicate{n}"
+                if cand not in existing:
+                    candidate = cand
+                    break
+                n += 1
 
         # 3) Duplicate the *base* document (not the ROI proxy)
-        #    NOTE: your project uses `self.docman` elsewhere for duplication.
-        new_doc = self.docman.duplicate_document(base_doc, new_name=f"{base_name}_duplicate")
+        new_doc = self.docman.duplicate_document(base_doc, new_name=candidate)
         print(f"  Duplicated document ID {id(base_doc)} -> {id(new_doc)}")
 
         # 4) Ensure the duplicate starts mask-free (so we don't inherit mask UI state)
@@ -8376,26 +8630,21 @@ class AstroSuiteProMainWindow(
 
 
     def _activate_or_open_from_explorer(self, item):
-        doc = item.data(Qt.ItemDataRole.UserRole)
-        base = self._normalize_base_doc(doc)
-
-        # 1) Try to focus an existing view for this base
-        for sw in self.mdi.subWindowList():
-            w = sw.widget()
-            if getattr(w, "base_document", None) is base:
-                try:
-                    sw.show(); w.show()
-                    st = sw.windowState()
-                    if st & Qt.WindowState.WindowMinimized:
-                        sw.setWindowState(st & ~Qt.WindowState.WindowMinimized)
-                    self.mdi.setActiveSubWindow(sw)
-                    sw.raise_()
-                except Exception:
-                    pass
-                return
-
-        # 2) None exists -> open one
-        self._open_subwindow_for_added_doc(base)
+        doc = item.data(0, Qt.ItemDataRole.UserRole)
+        if doc is None:
+            return
+        # you already have logic for this; typically:
+        sw = self._find_subwindow_for_doc(doc)
+        if sw:
+            self.mdi.setActiveSubWindow(sw)
+            sw.show()
+            sw.raise_()
+            return
+        # else open it (if your app supports opening closed docs, otherwise no-op)
+        try:
+            self._open_subwindow_for_added_doc(doc)
+        except Exception:
+            pass
 
     def _set_linked_stretch_from_action(self, checked: bool):
         # persist as the default for *new* views
@@ -8521,7 +8770,10 @@ class AstroSuiteProMainWindow(
             self._refresh_mask_action_states()
         except Exception:
             pass
-
+        try:
+            self._fix_mdi_titlebar_emboss("#dcdcdc" if self._theme_mode()=="dark" else "#f0f0f0")
+        except Exception:
+            pass
 
     def _sync_docman_active(self, doc):
         dm = self.doc_manager