teareduce 0.6.0__py3-none-any.whl → 0.6.1__py3-none-any.whl

teareduce/cleanest/cosmicraycleanerapp.py

@@ -285,17 +285,21 @@ class CosmicRayCleanerApp(ImageDisplay):
                 sdum = str(num_cr_pixels_after_dilation)
             else:
                 sdum = str(num_cr_pixels_before_dilation)
-            print("Number of cosmic ray pixels detected..........: " f"{num_cr_pixels_before_dilation:>{len(sdum)}}")
             if dilation > 0:
                 print(
-                    f"Number of cosmic ray pixels after dilation....: " f"{num_cr_pixels_after_dilation:>{len(sdum)}}"
+                    "Number of cosmic ray pixels before dilation.......: "
+                    f"{num_cr_pixels_before_dilation:>{len(sdum)}}"
+                )
+                print(
+                    f"Number of cosmic ray pixels after dilation........: "
+                    f"{num_cr_pixels_after_dilation:>{len(sdum)}}"
                 )
             # Label connected components in the mask; note that by default,
             # structure is a cross [0,1,0;1,1,1;0,1,0], but we want to consider
             # diagonal connections too, so we define a 3x3 square.
             structure = [[1, 1, 1], [1, 1, 1], [1, 1, 1]]
             self.cr_labels, self.num_features = ndimage.label(self.mask_crfound, structure=structure)
-            print(f"Number of cosmic ray features (grouped pixels): {self.num_features:>{len(sdum)}}")
+            print(f"Number of cosmic ray features (grouped pixels)....: {self.num_features:>{len(sdum)}}")
             self.replace_detected_cr_button.config(state=tk.NORMAL)
             self.review_detected_cr_button.config(state=tk.NORMAL)
             self.update_cr_overlay()
@@ -886,10 +890,7 @@ class CosmicRayCleanerApp(ImageDisplay):
                 mask_crfound2 = mask_crfound2 & (usefulmask.astype(bool))
                 # Combine results from both runs
                 if np.any(mask_crfound):
-                    print(f"Number of cosmic ray pixels (run1).......: {np.sum(mask_crfound)}")
-                    print(f"Number of cosmic ray pixels (run2).......: {np.sum(mask_crfound2)}")
-                    mask_crfound = merge_peak_tail_masks(mask_crfound, mask_crfound2)
-                    print(f"Number of cosmic ray pixels (run1 & run2): {np.sum(mask_crfound)}")
+                    mask_crfound = merge_peak_tail_masks(mask_crfound, mask_crfound2, verbose=True)
                 # Use the cleandata from the second run
                 cleandata_lacosmic = cleandata_lacosmic2
             # Select the image region to process
@@ -936,9 +937,6 @@ class CosmicRayCleanerApp(ImageDisplay):
             # recalculate labels and number of features
             structure = [[1, 1, 1], [1, 1, 1], [1, 1, 1]]
             self.cr_labels, self.num_features = ndimage.label(self.mask_crfound, structure=structure)
-            sdum = str(np.sum(self.mask_crfound))
-            print(f"Number of cosmic ray pixels detected by L.A.Cosmic...........: {sdum}")
-            print(f"Number of cosmic rays (grouped pixels) detected by L.A.Cosmic: {self.num_features:>{len(sdum)}}")
             # Define parameters for L.A.Cosmic from default dictionary
             editor_window = tk.Toplevel(self.root)
             center_on_parent(child=editor_window, parent=self.root)

teareduce/cleanest/imagedisplay.py

@@ -11,6 +11,7 @@
 
 from tkinter import simpledialog
 import numpy as np
+from rich import print
 
 from ..sliceregion import SliceRegion2D
 from ..zscale import zscale

teareduce/cleanest/interpolate.py

@@ -18,6 +18,7 @@ except ModuleNotFoundError as e:
         "`pip install teareduce[cleanest]`."
     ) from e
 import numpy as np
+from rich import print
 from scipy import ndimage
 from tqdm import tqdm
 
@@ -33,6 +34,19 @@ def interpolate(data, mask_crfound, dilation=0, interp_method=None, npoints=None
     The original data and mask are not modified. A copy of both
     arrays are created and returned with the interpolated pixels.
 
+    Cosmic-ray pixels are initially dilated by the specified number
+    of pixels. The resulting flagged pixels are grouped into cosmic-ray
+    features. Each cosmic-ray feature is then interpolated using the
+    specified interpolation method.
+
+    Note that the interpolation methods `lacosmic` and `auxfile`,
+    available in the interactive use of **tea-cleanest** are not
+    implemented in this function because both cases are simply
+    an inmediate replacement of the cosmic ray pixels in the data
+    array by the corresponding pixels in another array using the
+    mask array. Therefore, these two methods do not require any
+    interpolation algorithm.
+
     Parameters
     ----------
     data : 2D numpy.ndarray

teareduce/cleanest/interpolation_a.py

@@ -10,6 +10,7 @@
 """Surface interpolation (plane fit) or median interpolation"""
 
 import numpy as np
+from rich import print
 
 from .dilatemask import dilatemask
 

teareduce/cleanest/interpolation_x.py

@@ -10,6 +10,7 @@
 """Polynomial nterpolation in the X direction"""
 
 import numpy as np
+from rich import print
 
 
 def interpolation_x(data, mask_fixed, cr_labels, cr_index, npoints, degree):

teareduce/cleanest/interpolation_y.py

@@ -10,6 +10,7 @@
 """Polynomial nterpolation in the Y direction"""
 
 import numpy as np
+from rich import print
 
 
 def interpolation_y(data, mask_fixed, cr_labels, cr_index, npoints, degree):

teareduce/cleanest/interpolationeditor.py

@@ -13,6 +13,8 @@ import tkinter as tk
 from tkinter import messagebox
 from tkinter import ttk
 
+from rich import print
+
 from .centerchildparent import center_on_parent
 from .definitions import VALID_CLEANING_METHODS
 from .definitions import MASKFILL_OPERATOR_VALUES
@@ -423,7 +425,7 @@ class InterpolationEditor:
     def action_on_method_change(self):
         """Handle changes in the selected cleaning method."""
         selected_method = self.cleaning_method_var.get()
-        print(f"Selected cleaning method: {selected_method}")
+        print(f"Selected cleaning method: [red bold]{selected_method}[/red bold]")
         if selected_method in ["x interp.", "y interp."]:
             self.entry_npoints.config(state="normal")
             self.entry_degree.config(state="normal")

teareduce/cleanest/lacosmicpad.py

@@ -18,6 +18,7 @@ except ModuleNotFoundError as e:
         "`pip install teareduce[cleanest]`."
     ) from e
 import numpy as np
+from rich import print
 
 from .definitions import VALID_LACOSMIC_PSFMODEL_VALUES
 from .gausskernel2d_elliptical import gausskernel2d_elliptical
@@ -28,7 +29,8 @@ def lacosmicpad(pad_width, show_arguments=False, **kwargs):
 
     This function pads the input image array before applying the LACosmic
     cosmic ray cleaning algorithm. After processing, the padding is removed
-    to return an array of the original size.
+    to return an array of the original size. If `inbkg` or `invar` arrays
+    are provided, they are also padded accordingly.
 
     The padding helps to mitigate edge effects that can occur during the
     cosmic ray detection and cleaning process.

teareduce/cleanest/mergemasks.py

@@ -11,9 +11,10 @@
 
 import numpy as np
 from scipy import ndimage
+from rich import print
 
 
-def merge_peak_tail_masks(mask_peaks, mask_tails):
+def merge_peak_tail_masks(mask_peaks, mask_tails, verbose=False):
     """Merge peak and tail masks for cosmic ray cleaning.
 
     Tail pixels are preserved only if they correspond to CR features
@@ -25,6 +26,8 @@ def merge_peak_tail_masks(mask_peaks, mask_tails):
         Boolean array indicating the pixels identified as cosmic ray peaks.
     mask_tails : ndarray
         Boolean array indicating the pixels identified as cosmic ray tails.
+    verbose: bool
+        If True, print additional information during processing.
 
     Returns
     -------
@@ -41,6 +44,10 @@ def merge_peak_tail_masks(mask_peaks, mask_tails):
     if mask_peaks.dtype != bool or mask_tails.dtype != bool:
         raise TypeError("Input masks must be boolean arrays.")
 
+    if verbose:
+        ndigits = np.max([len(str(np.sum(mask_peaks))), len(str(np.sum(mask_tails)))])
+        print(f"Number of cosmic ray pixels (peaks)...............: {np.sum(mask_peaks):{ndigits}d}")
+        print(f"Number of cosmic ray pixels (tails)...............: {np.sum(mask_tails):{ndigits}d}")
     # find structures in tail mask
     structure = [[1, 1, 1], [1, 1, 1], [1, 1, 1]]
     cr_labels_tails, num_crs_tails = ndimage.label(mask_tails, structure=structure)
@@ -55,4 +62,7 @@ def merge_peak_tail_masks(mask_peaks, mask_tails):
         if icr > 0:
             merged_mask[cr_labels_tails == icr] = True
 
+    if verbose:
+        print(f"Number of cosmic ray pixels (merged peaks & tails): {np.sum(merged_mask):{ndigits}d}")
+
     return merged_mask

teareduce/version.py

@@ -9,7 +9,7 @@
 #
 """Module to define the version of the teareduce package."""
 
-VERSION = '0.6.0'
+VERSION = '0.6.1'
 
 
 def main():

{teareduce-0.6.0.dist-info → teareduce-0.6.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: teareduce
-Version: 0.6.0
+Version: 0.6.1
 Summary: Utilities for astronomical data reduction
 Author-email: Nicolás Cardiel <cardiel@ucm.es>
 License: GPL-3.0-or-later
@@ -51,7 +51,7 @@ practical for use in Master’s level classes.
 ## Documentation
 
 The documentation for this package is available at [this
-link](https://guaix.fis.ucm.es/~tea/teareduce-cookbook/_build/html/intro.html).
+link](https://nicocardiel.github.io/teareduce-cookbook/intro.html).
 It includes Juypter notebooks that can be easily downloaded and demonstrate the
 practical use of the defined functionality.
 

{teareduce-0.6.0.dist-info → teareduce-0.6.1.dist-info}/RECORD

@@ -15,7 +15,7 @@ teareduce/sdistortion.py,sha256=5ZsZn4vD5Sw2aoqO8-NIOH7H89Zmh7ZDkow6YbAotHU,5916
 teareduce/simulateccdexposure.py,sha256=cdbpca6GVuM3d7R1LGzlIZZvjTq_jzrlkk_Cli7aouQ,24636
 teareduce/sliceregion.py,sha256=Jdf8XvmGaY_vaY1cneTaRtSOYPxpUsJm9cXJDDMa0YM,18626
 teareduce/statsummary.py,sha256=VTNAnBV8z6suqiyB2Lhw3YjUUOjlmwUPX3enkOKRF54,5422
-teareduce/version.py,sha256=E1uQr8Mx5jXWQL_pejqYQN5pSGxyQT__LkXZ7yPdegY,419
+teareduce/version.py,sha256=RJWCscyH4cJAAqke3NP_StsgFVOL0CjPEjXdsqc_8KI,419
 teareduce/wavecal.py,sha256=2MewWz5Y3ms41c305UtWOM_LaLNdk1mugDXCtzf-fSw,68586
 teareduce/write_array_to_fits.py,sha256=kWDrEH9coJ1yIu56oQJpWtDqJL4c8HGmssE9jle4e94,617
 teareduce/zscale.py,sha256=SDgmcDD2N5GvDn46JADCgTQJPBF_N_jSKMHoeTz9Nsw,1152
@@ -23,19 +23,19 @@ teareduce/cleanest/__init__.py,sha256=1rcfu97uqj3R-OZbU9HHLYaTzWkS-poRFBBy8nuH7p
 teareduce/cleanest/__main__.py,sha256=KhFJSekko0mz_pVz3gxNnEmqvv4nuRBdhaXB2Nt4aL0,6145
 teareduce/cleanest/askextension.py,sha256=Q6mGLA6ajUcDU4gsVA4P78GsPei0c7iZK1f7SfK9iVo,3014
 teareduce/cleanest/centerchildparent.py,sha256=wHxOvNrrQ-KBLZAbtQ9bJAxYhGOzqYBF4LgdIQk7UF8,1285
-teareduce/cleanest/cosmicraycleanerapp.py,sha256=MJPdHmEpID4ykGu3o89dLUD0ebhjw3ZlxFUBQDJ_HcY,61768
+teareduce/cleanest/cosmicraycleanerapp.py,sha256=OpBBhgvvGiC8rO_C26Zwc0gIlQ6phDVjSarQNMEfpTY,61327
 teareduce/cleanest/definitions.py,sha256=Ryreer1ztkk4W3muWGI_SSfm3ZAhyAOQJbZLtdfTJsc,4692
 teareduce/cleanest/dilatemask.py,sha256=I5tHAv5VEO6V0Wed8Ar20uLt4F9P-tgjmLL5BAaFvgM,1276
 teareduce/cleanest/find_closest_true.py,sha256=mWdIvhipzAXDRKfePDrP7f0lP4U48cckpHiKwiB4jHI,1320
 teareduce/cleanest/gausskernel2d_elliptical.py,sha256=f6AT0ZHmLCd83NAIyyJ8ODlihMeeTkmvNnUSK3vjP9I,1606
-teareduce/cleanest/imagedisplay.py,sha256=820Vn-Q0bJyHicOBsxDmfAZxuGOFepEEsm0LTxlPJjc,5848
-teareduce/cleanest/interpolate.py,sha256=upk6OxIIfoV1PetogwI8PUmdwM6TiKvHu9XNUwG1Dac,6380
-teareduce/cleanest/interpolation_a.py,sha256=zE4VIrC41vapf0Vx9qmh1oacw2qkJwcuMnV3JpSDW8Y,4007
-teareduce/cleanest/interpolation_x.py,sha256=D5hKbobT6lJk18XktP3PXhzEN12zqed6U18yfQ-reLQ,3744
-teareduce/cleanest/interpolation_y.py,sha256=O6yw5nKKlTdV6qsP1Ku6CwGhXB6o3j0_YQirXyILi8c,3759
-teareduce/cleanest/interpolationeditor.py,sha256=ZcqQl_iG-0eH-1PdtnJppRI9zKiSK6S7OCQz7XUvAno,19888
-teareduce/cleanest/lacosmicpad.py,sha256=aZp-2NZJKRnu1NmjC_SHnL0EVy0jDXWULgo0Pe7tq3U,6817
-teareduce/cleanest/mergemasks.py,sha256=1Vq6Wqn6DClxSAvHwy6QrJGszA8nkmoMHITyprSykHI,2072
+teareduce/cleanest/imagedisplay.py,sha256=m8wmQ7CSDywABxkra2G4dFD06CVP1KvIYIdnZqoVdrI,5871
+teareduce/cleanest/interpolate.py,sha256=EZGxiRDh8i5jL5_fK_iTLDt96TaK1ILsiFXV5m6g2J4,7068
+teareduce/cleanest/interpolation_a.py,sha256=iyAEbh0Bf74OfAVhZjSvP3cO7pDC8eVDrVnDUd18FZQ,4030
+teareduce/cleanest/interpolation_x.py,sha256=MZhtlZZuMLvLnOK0EsLkJM3_dpyNHsxQL_BU1vLt9Lw,3767
+teareduce/cleanest/interpolation_y.py,sha256=ZX1Uv3FOxoLHtuR_yspcSA5B81CK_J0oNJ1ztzp0l5g,3782
+teareduce/cleanest/interpolationeditor.py,sha256=mIy-25dylQnqC91tLny4iZWS4d_bJ4sDDs-kNOWM-yE,19933
+teareduce/cleanest/lacosmicpad.py,sha256=IXwHAHeh4rQlQjMw2lfKSzjzmoo3HdSt0Qj0nhK_nHU,6921
+teareduce/cleanest/mergemasks.py,sha256=TdbfQl5ETxtyFfod0WCD7Czzmgt-WxYOk3hpLqGNOTQ,2620
 teareduce/cleanest/modalprogressbar.py,sha256=uwd-p92PvOVJnbXd-B8DRcBZ--keKpr4ZN9PLeqm1Ws,6449
 teareduce/cleanest/parametereditor.py,sha256=V8twAQzR0vjd68PB92PfdQBmTRDR04xx2mb4SLSV68Y,24530
 teareduce/cleanest/reviewcosmicray.py,sha256=8jUaJCHVvZqlQ57A9eCDNIfBCKQ0IqFADq8C1H7T7ko,36518
@@ -46,9 +46,9 @@ teareduce/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 teareduce/tests/test_cleanest.py,sha256=6dRqkw1RQMKsFrC8cEweMvTD6wXhiDv3P4PS57-HEqI,5598
 teareduce/tests/test_sliceregion.py,sha256=S7Zoh2eEBFIEbfsXgWBEMCf7pottjw2oLhqlZJQkAwg,3785
 teareduce/tests/test_version.py,sha256=mKLnbXyvVNc1pATq5PxR8qeoFMPAFL_GilFV6IHLOi0,172
-teareduce-0.6.0.dist-info/licenses/LICENSE.txt,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-teareduce-0.6.0.dist-info/METADATA,sha256=ZL_ioffCU8ICr6DsuohcYHR3tvJLWkG1gJePxH61Pgk,3640
-teareduce-0.6.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-teareduce-0.6.0.dist-info/entry_points.txt,sha256=6yBvig5jTL2ugqz5SF767AiszzrHKGRASsX1II84kqA,66
-teareduce-0.6.0.dist-info/top_level.txt,sha256=7OkwtX9zNRkGJ7ACgjk4ESgC74qUYcS5O2qcO0v-Si4,10
-teareduce-0.6.0.dist-info/RECORD,,
+teareduce-0.6.1.dist-info/licenses/LICENSE.txt,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+teareduce-0.6.1.dist-info/METADATA,sha256=tYpScu331TcUdNsSurzVythnm0_NHdEQe9CijpaOiCw,3628
+teareduce-0.6.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+teareduce-0.6.1.dist-info/entry_points.txt,sha256=6yBvig5jTL2ugqz5SF767AiszzrHKGRASsX1II84kqA,66
+teareduce-0.6.1.dist-info/top_level.txt,sha256=7OkwtX9zNRkGJ7ACgjk4ESgC74qUYcS5O2qcO0v-Si4,10
+teareduce-0.6.1.dist-info/RECORD,,