teareduce 0.5.4__tar.gz → 0.5.7__tar.gz

{teareduce-0.5.4/src/teareduce.egg-info → teareduce-0.5.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: teareduce
-Version: 0.5.4
+Version: 0.5.7
 Summary: Utilities for astronomical data reduction
 Author-email: Nicolás Cardiel <cardiel@ucm.es>
 License: GPL-3.0-or-later

{teareduce-0.5.4 → teareduce-0.5.7}/src/teareduce/cleanest/__init__.py

@@ -7,4 +7,6 @@
 # License-Filename: LICENSE.txt
 #
 
-from .cleanest import cleanest
+from .interpolate import interpolate
+from .lacosmicpad import lacosmicpad
+from .mergemasks import merge_peak_tail_masks

teareduce-0.5.7/src/teareduce/cleanest/__main__.py

@@ -0,0 +1,169 @@
+#
+# Copyright 2025 Universidad Complutense de Madrid
+#
+# This file is part of teareduce
+#
+# SPDX-License-Identifier: GPL-3.0+
+# License-Filename: LICENSE.txt
+#
+
+"""Interactive Cosmic Ray cleaning tool."""
+
+import argparse
+from ast import arg
+import tkinter as tk
+from tkinter import filedialog
+from tkinter import simpledialog
+import os
+from pathlib import Path
+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
+from ..version import VERSION
+
+import matplotlib
+
+matplotlib.use("TkAgg")
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        description="Interactive cosmic ray cleaner for FITS images.",
+        formatter_class=RichHelpFormatter,
+    )
+    parser.add_argument("input_fits", nargs="?", default=None, help="Path to the FITS file to be cleaned.")
+    parser.add_argument("--extension", type=str, 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=str,
+        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}).",
+    )
+    parser.add_argument("--version", action="version", version=f"%(prog)s {VERSION}")
+    parser.add_argument("--verbose", "-v", action="store_true", help="Enable verbose output.")
+    args = parser.parse_args()
+
+    # Welcome message
+    print("[bold green]Cosmic Ray Cleaner[/bold green]")
+    print("Interactive tool to clean cosmic rays from FITS images.")
+    print("teareduce version: " + VERSION)
+    print(f"https://nicocardiel.github.io/teareduce-cookbook/docs/cleanest/cleanest.html\n")
+
+    # If input_file is not provided, ask for it using a file dialog
+    if args.input_fits is None:
+        root = tk.Tk()
+        root.withdraw()  # Hide the root window
+        args.input_fits = filedialog.askopenfilename(
+            title="Select FITS file to be cleaned",
+            filetypes=[("FITS files", "*.fits *.fit *.fts"), ("All files", "*.*")],
+        )
+        if not args.input_fits:
+            print("No input FITS file selected. Exiting.")
+            exit(1)
+        print(f"Selected input FITS file: {args.input_fits}")
+        args.extension = simpledialog.askstring(
+                "Select Extension",
+                f"\nEnter extension number or name for file:\n{Path(args.input_fits).name}",
+                initialvalue=args.extension,
+            )
+        # Ask for auxiliary file if not provided
+        if args.auxfile is None:
+            use_auxfile = tk.messagebox.askyesno(
+                "Auxiliary File",
+                "Do you want to use an auxiliary FITS file?",
+                default=tk.messagebox.NO,
+            )
+            if use_auxfile:
+                args.auxfile = filedialog.askopenfilename(
+                    title="Select Auxiliary FITS file",
+                    filetypes=[("FITS files", "*.fits *.fit *.fts"), ("All files", "*.*")],
+                    initialfile=args.auxfile,
+                )
+                if not args.auxfile:
+                    print("No auxiliary FITS file selected. Exiting.")
+                    exit(1)
+        else:
+            use_auxfile = True
+        if use_auxfile:
+            print(f"Selected auxiliary FITS file: {args.auxfile}")
+            args.extension_auxfile = simpledialog.askstring(
+                "Select Extension for Auxiliary File",
+                f"\nEnter extension number or name for auxiliary file:\n{Path(args.auxfile).name}",
+                initialvalue=args.extension_auxfile,
+            )
+        root.destroy()
+
+    # Check that input files, and the corresponding extensions, exist
+    if not os.path.isfile(args.input_fits):
+        print(f"Error: File '{args.input_fits}' does not exist.")
+        exit(1)
+    if args.auxfile is not None and not os.path.isfile(args.auxfile):
+        print(f"Error: Auxiliary file '{args.auxfile}' does not exist.")
+        exit(1)
+
+    # 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(
+        root=root,
+        input_fits=args.input_fits,
+        extension=args.extension,
+        auxfile=args.auxfile,
+        extension_auxfile=args.extension_auxfile,
+        fontfamily=args.fontfamily,
+        fontsize=args.fontsize,
+        width=args.width,
+        height=args.height,
+        verbose=args.verbose,
+    )
+
+    # Execute
+    root.mainloop()
+
+
+if __name__ == "__main__":
+    main()

teareduce-0.5.7/src/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}")