teareduce 0.5.7__py3-none-any.whl → 0.5.8__py3-none-any.whl

teareduce/cleanest/cosmicraycleanerapp.py

@@ -17,6 +17,7 @@ from tkinter import simpledialog
 import sys
 
 from astropy.io import fits
+from maskfill import maskfill
 import matplotlib.pyplot as plt
 from matplotlib.backend_bases import key_press_handler
 from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg, NavigationToolbar2Tk
@@ -773,6 +774,21 @@ class CosmicRayCleanerApp(ImageDisplay):
                     # upate mask_crfound by eliminating the cleaned pixels
                     self.mask_crfound[mask_crfound_region] = False
                     data_has_been_modified = True
+                elif cleaning_method == "maskfill":
+                    # Replace detected CR pixels with local median values
+                    smoothed_output, _ = maskfill(
+                        input_image=self.data,
+                        mask=mask_crfound_region,
+                        size=3,
+                        operator="median",
+                        smooth=True,
+                    )
+                    self.data[mask_crfound_region] = smoothed_output[mask_crfound_region]
+                    # update mask_fixed to include the newly fixed pixels
+                    self.mask_fixed[mask_crfound_region] = True
+                    # upate mask_crfound by eliminating the cleaned pixels
+                    self.mask_crfound[mask_crfound_region] = False
+                    data_has_been_modified = True
                 elif cleaning_method == "auxdata":
                     if self.auxdata is None:
                         print("No auxiliary data available. Cleaning skipped!")

teareduce/cleanest/definitions.py

@@ -43,7 +43,7 @@ lacosmic_default_dict = {
 }
 
 # Default parameters for cleaning methods
-VALID_CLEANING_METHODS = ["x interp.", "y interp.", "surface interp.", "median", "mean", "lacosmic", "auxdata"]
+VALID_CLEANING_METHODS = ["x interp.", "y interp.", "surface interp.", "median", "mean", "lacosmic", "maskfill", "auxdata"]
 
 # Maximum pixel distance to consider when finding closest CR pixel
 MAX_PIXEL_DISTANCE_TO_CR = 15

teareduce/cleanest/interpolationeditor.py

@@ -97,6 +97,7 @@ class InterpolationEditor:
             "median": "a-median",
             "mean": "a-mean",
             "lacosmic": "lacosmic",
+            "maskfill": "maskfill",
             "auxdata": "auxdata",
         }
         self.check_interp_methods()

teareduce/cleanest/reviewcosmicray.py

@@ -13,6 +13,7 @@ import tkinter as tk
 from tkinter import messagebox
 from tkinter import simpledialog
 
+from maskfill import maskfill
 import matplotlib.pyplot as plt
 from matplotlib.backend_bases import key_press_handler
 from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg, NavigationToolbar2Tk
@@ -150,7 +151,7 @@ class ReviewCosmicRay(ImageDisplay):
             self.root.minsize(1000, 700)
         else:
             # self.root.geometry("800x700+100+100")  # This does not work in Fedora
-            self.root.minsize(800, 700)
+            self.root.minsize(900, 700)
         self.root.update_idletasks()
         self.root.geometry("+100+100")
         self.data = data
@@ -224,6 +225,8 @@ class ReviewCosmicRay(ImageDisplay):
             self.interp_l_button.config(state=tk.DISABLED)
         if self.cleandata_lacosmic is None:
             self.interp_l_button.config(state=tk.DISABLED)
+        self.interp_maskfill_button = tk.Button(self.button_frame2, text="mas[k]fill", command=self.use_maskfill)
+        self.interp_maskfill_button.pack(side=tk.LEFT, padx=5)
         self.interp_aux_button = tk.Button(self.button_frame2, text="[a]ux. data", command=self.use_auxdata)
         self.interp_aux_button.pack(side=tk.LEFT, padx=5)
         if self.auxdata is None:
@@ -246,7 +249,7 @@ class ReviewCosmicRay(ImageDisplay):
         if self.auxdata is not None:
             self.fig, (self.ax_aux, self.ax) = plt.subplots(ncols=2, figsize=(11, 5.5), constrained_layout=True)
         else:
-            self.fig, self.ax = plt.subplots(figsize=(8, 5.5), constrained_layout=True)
+            self.fig, self.ax = plt.subplots(figsize=(9, 5.5), constrained_layout=True)
         self.canvas = FigureCanvasTkAgg(self.fig, master=self.root)
         self.canvas.get_tk_widget().pack(padx=5, pady=5)
         # The next two instructions prevent a segmentation fault when pressing "q"
@@ -376,6 +379,7 @@ class ReviewCosmicRay(ImageDisplay):
         self.interp_d_button.config(state=tk.DISABLED)
         self.interp_m_button.config(state=tk.DISABLED)
         self.interp_l_button.config(state=tk.DISABLED)
+        self.interp_maskfill_button.config(state=tk.DISABLED)
         self.interp_aux_button.config(state=tk.DISABLED)
 
     def interp_x(self):
@@ -461,6 +465,27 @@ class ReviewCosmicRay(ImageDisplay):
         self.set_buttons_after_cleaning_cr()
         self.update_display(cleaned=True)
 
+    def use_maskfill(self):
+        """Use maskfill cleaned data to clean a cosmic ray."""
+        print(f"Maskfill interpolation of cosmic ray {self.cr_index}")
+        ycr_list, xcr_list = np.where(self.cr_labels == self.cr_index)
+        mask = np.zeros(self.data.shape, dtype=bool)
+        for iy, ix in zip(ycr_list, xcr_list):
+            mask[iy, ix] = True
+        smoothed_output, _ = maskfill(
+            input_image=self.data,
+            mask=mask,
+            size=3,
+            operator="median",
+            smooth=True,
+        )
+        for iy, ix in zip(ycr_list, xcr_list):
+            self.data[iy, ix] = smoothed_output[iy, ix]
+            self.mask_fixed[iy, ix] = True
+        self.num_cr_cleaned += 1
+        self.set_buttons_after_cleaning_cr()
+        self.update_display(cleaned=True)
+
     def use_auxdata(self):
         """Use auxiliary data to clean a cosmic ray."""
         if self.auxdata is None:
@@ -505,6 +530,7 @@ class ReviewCosmicRay(ImageDisplay):
             if self.cleandata_lacosmic is not None:
                 if self.last_dilation is None or self.last_dilation == 0:
                     self.interp_l_button.config(state=tk.NORMAL)
+            self.interp_maskfill_button.config(state=tk.NORMAL)
             if self.auxdata is not None:
                 self.interp_aux_button.config(state=tk.NORMAL)
         print(f"Restored all pixels of cosmic ray {self.cr_index}")
@@ -566,6 +592,9 @@ class ReviewCosmicRay(ImageDisplay):
         elif event.key == "l":
             if self.interp_l_button.cget("state") != "disabled":
                 self.use_lacosmic()
+        elif event.key == "k":
+            if self.interp_maskfill_button.cget("state") != "disabled":
+                self.use_maskfill()
         elif event.key == "a":
             if self.interp_aux_button.cget("state") != "disabled":
                 self.use_auxdata()

teareduce/version.py

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

{teareduce-0.5.7.dist-info → teareduce-0.5.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: teareduce
-Version: 0.5.7
+Version: 0.5.8
 Summary: Utilities for astronomical data reduction
 Author-email: Nicolás Cardiel <cardiel@ucm.es>
 License: GPL-3.0-or-later
@@ -23,6 +23,7 @@ Requires-Dist: astropy
 Requires-Dist: ccdproc
 Requires-Dist: lmfit
 Requires-Dist: matplotlib
+Requires-Dist: maskfill
 Requires-Dist: numpy>=1.22
 Requires-Dist: requests
 Requires-Dist: rich

{teareduce-0.5.7.dist-info → teareduce-0.5.8.dist-info}/RECORD

@@ -15,15 +15,15 @@ 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=oqn_gabhd81ghdha2yfonHK150M_SY20r86f1CEW9bg,419
+teareduce/version.py,sha256=G40b1Oo_cwqV4sHca8L26xY3VpGjlKvOFDTnZdytrtg,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
 teareduce/cleanest/__init__.py,sha256=hHFtV6uBx6HAI8LFUDli0urznTUzMotCW9J_4EKGCXc,293
 teareduce/cleanest/__main__.py,sha256=cMzvNWNe2fuTGHLtT3hOc9-NAcMXDZUqHN0SSHWWDnc,5977
 teareduce/cleanest/centerchildparent.py,sha256=wHxOvNrrQ-KBLZAbtQ9bJAxYhGOzqYBF4LgdIQk7UF8,1285
-teareduce/cleanest/cosmicraycleanerapp.py,sha256=FKeQYCUhvod6HDSOqZN79QhVoujhC2_iQsyP1tF3P9o,49726
-teareduce/cleanest/definitions.py,sha256=HLv41cuhUOUyEicRHea0LRTv4Wagm24miSxFt4cZN3g,2546
+teareduce/cleanest/cosmicraycleanerapp.py,sha256=gugC79tdQFhT7plOWKqm5uTtjJCiWIeDpNmaCiE7aA4,50584
+teareduce/cleanest/definitions.py,sha256=L7VXT0SAjv96zgZ_VhMMV1vcf-oxdwefvlmPclBA-ug,2558
 teareduce/cleanest/dilatemask.py,sha256=I5tHAv5VEO6V0Wed8Ar20uLt4F9P-tgjmLL5BAaFvgM,1276
 teareduce/cleanest/find_closest_true.py,sha256=mWdIvhipzAXDRKfePDrP7f0lP4U48cckpHiKwiB4jHI,1320
 teareduce/cleanest/imagedisplay.py,sha256=820Vn-Q0bJyHicOBsxDmfAZxuGOFepEEsm0LTxlPJjc,5848
@@ -31,21 +31,21 @@ teareduce/cleanest/interpolate.py,sha256=qlz4SHw89ahBRx1VHBMkd0IEpFkd96P6FVfsw_L
 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=EksuNwKOM79MyGlAgxb9j-v7pIFXV8QOcyZDLVQlQjU,13221
+teareduce/cleanest/interpolationeditor.py,sha256=J5vrioLTwUSMbUeY0ezqNqm3CvLJgKFe7gvZ44Z62sc,13257
 teareduce/cleanest/lacosmicpad.py,sha256=xWaxifoF_B82Gm5czuJmL7Y4Cv0W___LlvcpH633bAo,2047
 teareduce/cleanest/mergemasks.py,sha256=1Vq6Wqn6DClxSAvHwy6QrJGszA8nkmoMHITyprSykHI,2072
 teareduce/cleanest/modalprogressbar.py,sha256=uwd-p92PvOVJnbXd-B8DRcBZ--keKpr4ZN9PLeqm1Ws,6449
 teareduce/cleanest/parametereditor.py,sha256=_0baFbkIUcPyF-teFIRMt1MpA4IzDZc7VHNh1h0qN4A,14229
-teareduce/cleanest/reviewcosmicray.py,sha256=_qO9ifkAMqB4Wuvq2wIYZ8_DUniB2lWcxn9l404hXnY,28210
+teareduce/cleanest/reviewcosmicray.py,sha256=yTY9vNyYSuh_AZsIJAEtJ50mgIKhjgTe-LQob4xQx9U,29497
 teareduce/cookbook/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 teareduce/cookbook/get_cookbook_file.py,sha256=vde-iNii2lm1QII8GmLRsFsKNxkdsd7njCBE-8Z7io0,1088
 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.5.7.dist-info/licenses/LICENSE.txt,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-teareduce-0.5.7.dist-info/METADATA,sha256=q725fGixcsrRvW2iS9-QRgBPi54dZqjOILxmzEDBShs,3113
-teareduce-0.5.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-teareduce-0.5.7.dist-info/entry_points.txt,sha256=6yBvig5jTL2ugqz5SF767AiszzrHKGRASsX1II84kqA,66
-teareduce-0.5.7.dist-info/top_level.txt,sha256=7OkwtX9zNRkGJ7ACgjk4ESgC74qUYcS5O2qcO0v-Si4,10
-teareduce-0.5.7.dist-info/RECORD,,
+teareduce-0.5.8.dist-info/licenses/LICENSE.txt,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+teareduce-0.5.8.dist-info/METADATA,sha256=Xm0x4wiGEoHvS_3tIk7gmaLHZbN0HDuN_Iec_wdQg84,3137
+teareduce-0.5.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+teareduce-0.5.8.dist-info/entry_points.txt,sha256=6yBvig5jTL2ugqz5SF767AiszzrHKGRASsX1II84kqA,66
+teareduce-0.5.8.dist-info/top_level.txt,sha256=7OkwtX9zNRkGJ7ACgjk4ESgC74qUYcS5O2qcO0v-Si4,10
+teareduce-0.5.8.dist-info/RECORD,,