Set NOT_DEBLENDED_MASK for skipped parents

lsst · Jun 1, 2022 · ebbf47b · ebbf47b
1 parent 851fa13
commit ebbf47b
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/python/lsst/meas/extensions/scarlet/scarletDeblendTask.py b/python/lsst/meas/extensions/scarlet/scarletDeblendTask.py
@@ -1125,6 +1125,13 @@ def deblend(self, mExposure, catalog):
             # Clear the cached values in scarlet to clear out memory
             scarlet.cache.Cache._cache = {}
 
+        # Update the mExposure mask with the footprint of skipped parents
+        if self.config.notDeblendedMask:
+            for mask in mExposure.mask:
+                for parentIndex in skippedParents:
+                    fp = catalog[parentIndex].getFootprint()
+                    fp.spans.setMask(mask, mask.getPlaneBitMask(self.config.notDeblendedMask))
+
         self.log.info("Deblender results: of %d parent sources, %d were deblended, "
                       "creating %d children, for a total of %d sources",
                       nParents, nDeblendedParents, len(catalog)-nParents, len(catalog))