Added option to destripe data upon last iteration when creating sourc…

…e masks

vircampype/fits/images/sky.py

@@ -899,6 +899,12 @@ def build_master_source_mask(self):
                     - 10 * sky_std[:, np.newaxis, np.newaxis]
                 )
 
+                # destripe on last iteration
+                if (
+                    i == self.setup.source_masks_n_iter - 1
+                ) & self.setup.source_masks_destripe:
+                    cube_raw_temp.destripe(masks=mask_additional)
+
                 """ Destriping here is very problematic. Firstly, I can't average,
                 because I have only frames of a single detector and secondly, any
                 extended emission that covers a large part of the image will affect

vircampype/pipeline/setup.py

@@ -110,6 +110,7 @@ class Setup(dict):
     source_mask_closing: bool = True
     source_masks_closing_size: int = 5
     source_masks_closing_iter: int = 2
+    source_masks_destripe: bool = True
     # Noisechisel setup for source masks
     noisechisel_qthresh: float = 0.9
     noisechisel_erode: int = 2