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

setiastro/saspro/numba_utils.py

@@ -317,61 +317,6 @@ def invert_image_numba(image):
                     output[y, x, c] = 1.0 - image[y, x, c]
         return output
 
-
-
-@njit(parallel=True, fastmath=True, cache=True)
-def apply_flat_division_numba_2d(image, master_flat, master_bias=None):
-    """
-    Mono version: image.shape == (H,W)
-    """
-    if master_bias is not None:
-        master_flat = master_flat - master_bias
-        image = image - master_bias
-
-    median_flat = np.mean(master_flat)
-    height, width = image.shape
-
-    for y in prange(height):
-        for x in range(width):
-            image[y, x] /= (master_flat[y, x] / median_flat)
-
-    return image
-
-
-@njit(parallel=True, fastmath=True, cache=True)
-def apply_flat_division_numba_3d(image, master_flat, master_bias=None):
-    """
-    Color version: image.shape == (H,W,C)
-    """
-    if master_bias is not None:
-        master_flat = master_flat - master_bias
-        image = image - master_bias
-
-    median_flat = np.mean(master_flat)
-    height, width, channels = image.shape
-
-    for y in prange(height):
-        for x in range(width):
-            for c in range(channels):
-                image[y, x, c] /= (master_flat[y, x, c] / median_flat)
-
-    return image
-
-def apply_flat_division_numba(image, master_flat, master_bias=None):
-    """
-    Dispatcher that calls the correct Numba function
-    depending on whether 'image' is 2D or 3D.
-    """
-    if image.ndim == 2:
-        # Mono
-        return apply_flat_division_numba_2d(image, master_flat, master_bias)
-    elif image.ndim == 3:
-        # Color
-        return apply_flat_division_numba_3d(image, master_flat, master_bias)
-    else:
-        raise ValueError(f"apply_flat_division_numba: expected 2D or 3D, got shape {image.shape}")
-
-
 @njit(parallel=True, cache=True)
 def subtract_dark_3d(frames, dark_frame):
     """

setiastro/saspro/ops/script_editor.py

@@ -15,6 +15,11 @@ from PyQt6.QtWidgets import (
     QLineEdit, QToolButton, QCheckBox, QTextEdit
 )
 
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from setiastro.saspro.ops.scripts import ScriptEntry
+
 # -----------------------------------------------------------------------------
 # Code editor with line numbers (QPlainTextEdit subclass)
 # -----------------------------------------------------------------------------

setiastro/saspro/ops/scripts.py

@@ -294,6 +294,125 @@ class ScriptContext:
         # ✅ Normal run: let DocManager decide (ROI preview vs full)
         dm.update_active_document(img, metadata={}, step_name=step_name)
 
+    def _find_subwindow_for_doc(self, base_doc):
+        """Return (sw, widget) for the first subwindow showing base_doc."""
+        for sw, w in self._iter_open_subwindows():
+            d = self._base_doc_for_widget(w)
+            if d is base_doc:
+                return sw, w
+        return None, None
+
+    def rename_active_view(self, new_title: str) -> bool:
+        """Rename only the active MDI view title (this window)."""
+        w = self.active_view()
+        if w is None:
+            return False
+
+        t = (new_title or "").strip()
+        if not t:
+            return False
+
+        try:
+            # ImageSubWindow convention
+            setattr(w, "_view_title_override", t)
+            if hasattr(w, "_sync_host_title"):
+                w._sync_host_title()
+            return True
+        except Exception:
+            return False
+
+    def rename_active_document(self, new_name: str) -> bool:
+        """Rename the underlying document display name (affects explorer + other views)."""
+        doc = self.active_document()
+        if doc is None:
+            return False
+
+        n = (new_name or "").strip()
+        if not n:
+            return False
+
+        try:
+            old = ""
+            try:
+                old = str(doc.display_name() or "")
+            except Exception:
+                old = str(getattr(doc, "metadata", {}).get("display_name", "") or "")
+
+            doc.metadata["display_name"] = n
+            try:
+                doc.changed.emit()
+            except Exception:
+                pass
+
+            # If the active view had an override equal to the old doc name, drop it (matches your UI behavior)
+            w = self.active_view()
+            if w is not None:
+                try:
+                    if getattr(w, "_view_title_override", None) == old:
+                        setattr(w, "_view_title_override", None)
+                except Exception:
+                    pass
+                try:
+                    if hasattr(w, "_sync_host_title"):
+                        w._sync_host_title()
+                except Exception:
+                    pass
+
+            return True
+        except Exception:
+            return False
+
+    def rename_view(self, view_name_or_uid: str, new_title: str) -> bool:
+        """Rename a specific *view/window* by name/title/uid (first match)."""
+        doc = self.get_document(view_name_or_uid)
+        if doc is None:
+            return False
+
+        sw, w = self._find_subwindow_for_doc(doc)
+        if w is None:
+            return False
+
+        t = (new_title or "").strip()
+        if not t:
+            return False
+
+        try:
+            setattr(w, "_view_title_override", t)
+            if hasattr(w, "_sync_host_title"):
+                w._sync_host_title()
+            return True
+        except Exception:
+            return False
+
+    def rename_document(self, view_name_or_uid: str, new_name: str) -> bool:
+        """Rename a specific *document* by view name/title/uid."""
+        doc = self.get_document(view_name_or_uid)
+        if doc is None:
+            return False
+        n = (new_name or "").strip()
+        if not n:
+            return False
+
+        try:
+            doc.metadata["display_name"] = n
+            try:
+                doc.changed.emit()
+            except Exception:
+                pass
+
+            # resync any window currently showing it
+            sw, w = self._find_subwindow_for_doc(doc)
+            if w is not None and hasattr(w, "_sync_host_title"):
+                try:
+                    w._sync_host_title()
+                except Exception:
+                    pass
+
+            return True
+        except Exception:
+            return False
+
+
     # ---- convenience wrappers into main window ----
     def run_command(self, command_id: str, preset=None, **kwargs):
         return _run_command(self, command_id, preset, **kwargs)

setiastro/saspro/remove_green.py

@@ -164,7 +164,7 @@ class RemoveGreenDialog(QDialog):
 
     def _build_ui(self):
         lay = QVBoxLayout(self)
-        lay.addWidget(QLabel(self.tr("Select the amount to remove green noise:")))
+        lay.addWidget(QLabel(self.tr("Select the amount to remove green:")))
 
         # amount
         self.slider = QSlider(Qt.Orientation.Horizontal)

setiastro/saspro/resources.py

@@ -244,6 +244,7 @@ class Icons:
     STACKING = property(lambda self: _resource_path('stacking.png'))
     LIVE_STACKING = property(lambda self: _resource_path('livestacking.png'))
     IMAGE_COMBINE = property(lambda self: _resource_path('imagecombine.png'))
+    PLANETARY_STACKER = property(lambda self: _resource_path('planetarystacker.png'))
     
     # Moon phase (WIMS)
     MOON_NEW            = property(lambda self: _resource_path('new_moon.png'))
@@ -301,6 +302,7 @@ class Icons:
     COLOR_WHEEL = property(lambda self: _resource_path('colorwheel.png'))
     SELECTIVE_COLOR = property(lambda self: _resource_path('selectivecolor.png'))
     NB_TO_RGB = property(lambda self: _resource_path('nbtorgb.png'))
+    NARROWBANDNORMALIZATION = property(lambda self: _resource_path('narrowbandnormalization.png'))
     
     # Stretching
     STAT_STRETCH = property(lambda self: _resource_path('statstretch.png'))
@@ -531,6 +533,7 @@ def _init_legacy_paths():
         'collage_path': get_icon_path('collage.png'),
         'annotated_path': get_icon_path('annotated.png'),
         'colorwheel_path': get_icon_path('colorwheel.png'),
+        'narrowbandnormalization_path': get_icon_path('narrowbandnormalization.png'),
         'font_path': get_icon_path('font.png'),
         'csv_icon_path': get_icon_path('cvs.png'),
         'spinner_path': get_data_path('spinner.gif'),
@@ -540,6 +543,7 @@ def _init_legacy_paths():
         'debayer_path': get_icon_path('debayer.png'),
         'aberration_path': get_icon_path('aberration.png'),
         'functionbundles_path': get_icon_path('functionbundle.png'),
+        'planetarystacker_path': get_icon_path('planetarystacker.png'),
         'viewbundles_path': get_icon_path('viewbundle.png'),
         'selectivecolor_path': get_icon_path('selectivecolor.png'),
         'rgbalign_path': get_icon_path('rgbalign.png'),

setiastro/saspro/ser_stack_config.py

@@ -0,0 +1,68 @@
+# src/setiastro/saspro/ser_stack_config.py
+from __future__ import annotations
+from dataclasses import dataclass
+from typing import Optional, Tuple, Literal, Union, Sequence
+
+from setiastro.saspro.imageops.serloader import PlanetaryFrameSource
+
+TrackMode = Literal["off", "planetary", "surface"]
+PlanetarySource = Union[str, Sequence[str], PlanetaryFrameSource]
+
+@dataclass
+class SERStackConfig:
+    source: PlanetarySource
+    roi: Optional[Tuple[int, int, int, int]] = None
+    track_mode: TrackMode = "planetary"
+    surface_anchor: Optional[Tuple[int, int, int, int]] = None
+    keep_percent: float = 20.0
+
+    # AP / alignment
+    ap_size: int = 64
+    ap_spacing: int = 48
+    ap_min_mean: float = 0.03
+    ap_multiscale: bool = False
+    ssd_refine_bruteforce: bool = False
+
+    # ✅ Drizzle
+    drizzle_scale: float = 1.0          # 1.0 = off, 1.5, 2.0
+    drizzle_pixfrac: float = 0.80       # "drop shrink" in output pixels (roughly)
+    drizzle_kernel: str = "gaussian"    # "square" | "circle" | "gaussian"
+    drizzle_sigma: float = 0.0          # only used for gaussian; 0 => auto from pixfrac
+
+    def __init__(self, source: PlanetarySource, **kwargs):
+        # Allow deprecated/ignored kwargs without crashing
+        kwargs.pop("multipoint", None)  # accept but ignore
+
+        self.source = source
+        self.roi = kwargs.pop("roi", None)
+        self.track_mode = kwargs.pop("track_mode", "planetary")
+        self.surface_anchor = kwargs.pop("surface_anchor", None)
+        self.keep_percent = float(kwargs.pop("keep_percent", 20.0))
+
+        self.ap_size = int(kwargs.pop("ap_size", 64))
+        self.ap_spacing = int(kwargs.pop("ap_spacing", 48))
+        self.ap_min_mean = float(kwargs.pop("ap_min_mean", 0.03))
+        self.ap_multiscale = bool(kwargs.pop("ap_multiscale", False))
+        self.ssd_refine_bruteforce = bool(kwargs.pop("ssd_refine_bruteforce", False))
+
+        # ✅ NEW: Drizzle params
+        self.drizzle_scale = float(kwargs.pop("drizzle_scale", 1.0))
+        if self.drizzle_scale not in (1.0, 1.5, 2.0):
+            self.drizzle_scale = 1.0
+
+        self.drizzle_pixfrac = float(kwargs.pop("drizzle_pixfrac", 0.80))
+        self.drizzle_kernel = str(kwargs.pop("drizzle_kernel", "gaussian")).strip().lower()
+        self.drizzle_sigma = float(kwargs.pop("drizzle_sigma", 0.0))
+
+        # sanitize a bit
+        if self.drizzle_scale < 1.0:
+            self.drizzle_scale = 1.0
+        if self.drizzle_pixfrac <= 0.0:
+            self.drizzle_pixfrac = 0.01
+        if self.drizzle_kernel not in ("square", "circle", "gaussian"):
+            self.drizzle_kernel = "gaussian"
+        if self.drizzle_sigma < 0.0:
+            self.drizzle_sigma = 0.0
+
+        if kwargs:
+            raise TypeError(f"Unexpected config keys: {sorted(kwargs.keys())}")