teareduce 0.5.3__py3-none-any.whl → 0.5.5__py3-none-any.whl

teareduce/cleanest/__main__.py

@@ -12,26 +12,59 @@
 import argparse
 import tkinter as tk
 import os
+import platform
 from rich import print
 from rich_argparse import RichHelpFormatter
 
+from .definitions import DEFAULT_FONT_FAMILY
+from .definitions import DEFAULT_FONT_SIZE
+from .definitions import DEFAULT_TK_WINDOW_SIZE_X
+from .definitions import DEFAULT_TK_WINDOW_SIZE_Y
 from .cosmicraycleanerapp import CosmicRayCleanerApp
 
 import matplotlib
+
 matplotlib.use("TkAgg")
 
 
 def main():
     parser = argparse.ArgumentParser(
         description="Interactive cosmic ray cleaner for FITS images.",
-        formatter_class=RichHelpFormatter)
+        formatter_class=RichHelpFormatter,
+    )
     parser.add_argument("input_fits", help="Path to the FITS file to be cleaned.")
-    parser.add_argument("--extension", type=int, default=0,
-                        help="FITS extension to use (default: 0).")
-    parser.add_argument("--auxfile", type=str, default=None,
-                        help="Auxiliary FITS file")
-    parser.add_argument("--extension_auxfile", type=int, default=0,
-                        help="FITS extension for auxiliary file (default: 0).")
+    parser.add_argument("--extension", type=int, default=0, help="FITS extension to use (default: 0).")
+    parser.add_argument("--auxfile", type=str, default=None, help="Auxiliary FITS file")
+    parser.add_argument(
+        "--extension_auxfile",
+        type=int,
+        default=0,
+        help="FITS extension for auxiliary file (default: 0).",
+    )
+    parser.add_argument(
+        "--fontfamily",
+        type=str,
+        default=DEFAULT_FONT_FAMILY,
+        help=f"Font family for the GUI (default: {DEFAULT_FONT_FAMILY}).",
+    )
+    parser.add_argument(
+        "--fontsize",
+        type=int,
+        default=DEFAULT_FONT_SIZE,
+        help=f"Font size for the GUI (default: {DEFAULT_FONT_SIZE}).",
+    )
+    parser.add_argument(
+        "--width",
+        type=int,
+        default=DEFAULT_TK_WINDOW_SIZE_X,
+        help=f"Width of the GUI window in pixels (default: {DEFAULT_TK_WINDOW_SIZE_X}).",
+    )
+    parser.add_argument(
+        "--height",
+        type=int,
+        default=DEFAULT_TK_WINDOW_SIZE_Y,
+        help=f"Height of the GUI window in pixels (default: {DEFAULT_TK_WINDOW_SIZE_Y}).",
+    )
     args = parser.parse_args()
 
     if not os.path.isfile(args.input_fits):
@@ -43,6 +76,18 @@ def main():
 
     # Initialize Tkinter root
     root = tk.Tk()
+    system = platform.system()
+    if system == "Darwin":  # macOS
+        # Center the window on the screen
+        xoffset = root.winfo_screenwidth() // 2 - args.width // 2
+        yoffset = root.winfo_screenheight() // 2 - args.height // 2
+    else:
+        # Note that geometry("XxY+Xoffset+Yoffset") does not work properly on Fedora Linux
+        xoffset = 0
+        yoffset = 0
+    root.geometry(f"+{xoffset}+{yoffset}")
+    root.focus_force()  # Request focus
+    root.lift()  # Bring to front
 
     # Create and run the application
     CosmicRayCleanerApp(
@@ -50,7 +95,11 @@ def main():
         input_fits=args.input_fits,
         extension=args.extension,
         auxfile=args.auxfile,
-        extension_auxfile=args.extension_auxfile
+        extension_auxfile=args.extension_auxfile,
+        fontfamily=args.fontfamily,
+        fontsize=args.fontsize,
+        width=args.width,
+        height=args.height,
     )
 
     # Execute

teareduce/cleanest/centerchildparent.py

@@ -0,0 +1,46 @@
+#
+# Copyright 2025 Universidad Complutense de Madrid
+#
+# This file is part of teareduce
+#
+# SPDX-License-Identifier: GPL-3.0+
+# License-Filename: LICENSE.txt
+#
+
+"""Function to center a tk child window on its parent."""
+
+
+def center_on_parent(child, parent, offset_x=0, offset_y=0):
+    """Center child window on parent window.
+
+    Parameters
+    ----------
+    child : tk.Toplevel or tk.Tk
+        The child window to be centered.
+    parent : tk.Toplevel or tk.Tk
+        The parent window.
+    offset_x : int, optional
+        Horizontal offset from center position (default is 0).
+    offset_y : int, optional
+        Vertical offset from center position (default is 0).
+    """
+    # Update to get accurate dimensions
+    child.update_idletasks()
+    parent.update_idletasks()
+
+    # Get parent position and size
+    parent_x = parent.winfo_x()
+    parent_y = parent.winfo_y()
+    parent_width = parent.winfo_width()
+    parent_height = parent.winfo_height()
+
+    # Get child size
+    child_width = child.winfo_width()
+    child_height = child.winfo_height()
+
+    # Calculate center position
+    x = parent_x + (parent_width - child_width) // 2
+    y = parent_y + (parent_height - child_height) // 2
+
+    # Set child position
+    child.geometry(f"+{x + offset_x}+{y + offset_y}")

teareduce/cleanest/cleanest.py

@@ -18,9 +18,7 @@ from .interpolation_y import interpolation_y
 from .interpolation_a import interpolation_a
 
 
-def cleanest(data, mask_crfound, dilation=0,
-             interp_method=None, npoints=None, degree=None,
-             debug=False):
+def cleanest(data, mask_crfound, dilation=0, interp_method=None, npoints=None, degree=None, debug=False):
     """Interpolate pixels identified in a mask.
 
     The original data and mask are not modified. A copy of both
@@ -68,11 +66,11 @@ def cleanest(data, mask_crfound, dilation=0,
     """
     if interp_method is None:
         raise ValueError("interp_method must be specified.")
-    if interp_method not in ['x', 'y', 's', 'd', 'm']:
+    if interp_method not in ["x", "y", "s", "d", "m"]:
         raise ValueError(f"Unknown interp_method: {interp_method}")
     if npoints is None:
         raise ValueError("npoints must be specified.")
-    if degree is None and interp_method in ['x', 'y']:
+    if degree is None and interp_method in ["x", "y"]:
         raise ValueError("degree must be specified for the chosen interp_method.")
 
     # Apply dilation to the cosmic ray mask if needed
@@ -85,9 +83,7 @@ def cleanest(data, mask_crfound, dilation=0,
     mask_fixed = np.zeros_like(mask_crfound, dtype=bool)
 
     # Determine number of CR features
-    structure = [[1, 1, 1],
-                 [1, 1, 1],
-                 [1, 1, 1]]
+    structure = [[1, 1, 1], [1, 1, 1], [1, 1, 1]]
     cr_labels, num_features = ndimage.label(updated_mask_crfound, structure=structure)
     if debug:
         print(f"Number of cosmic ray pixels to be cleaned: {np.sum(updated_mask_crfound)}")
@@ -97,10 +93,10 @@ def cleanest(data, mask_crfound, dilation=0,
     cleaned_data = data.copy()
     num_cr_cleaned = 0
     for cr_index in range(1, num_features + 1):
-        if interp_method in ['x', 'y']:
+        if interp_method in ["x", "y"]:
             if 2 * npoints <= degree:
                 raise ValueError("2*npoints must be greater than degree for polynomial interpolation.")
-            if interp_method == 'x':
+            if interp_method == "x":
                 interp_func = interpolation_x
             else:
                 interp_func = interpolation_y
@@ -110,23 +106,24 @@ def cleanest(data, mask_crfound, dilation=0,
                 cr_labels=cr_labels,
                 cr_index=cr_index,
                 npoints=npoints,
-                degree=degree)
+                degree=degree,
+            )
             if interpolation_performed:
                 num_cr_cleaned += 1
-        elif interp_method in ['s', 'd', 'm']:
-            if interp_method == 's':
-                method = 'surface'
-            elif interp_method == 'd':
-                method = 'median'
-            elif interp_method == 'm':
-                method = 'mean'
+        elif interp_method in ["s", "d", "m"]:
+            if interp_method == "s":
+                method = "surface"
+            elif interp_method == "d":
+                method = "median"
+            elif interp_method == "m":
+                method = "mean"
             interpolation_performed, _, _ = interpolation_a(
                 data=cleaned_data,
                 mask_fixed=mask_fixed,
                 cr_labels=cr_labels,
                 cr_index=cr_index,
                 npoints=npoints,
-                method=method
+                method=method,
             )
             if interpolation_performed:
                 num_cr_cleaned += 1