Add streak mask propagation to DRP tasks

lsst · Oct 25, 2023 · 24d3ab6 · 24d3ab6
1 parent dee613c
commit 24d3ab6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/python/lsst/pipe/tasks/finalizeCharacterization.py b/python/lsst/pipe/tasks/finalizeCharacterization.py
@@ -215,6 +215,11 @@ def setDefaults(self):
         self.measurement.slots.shape = 'ext_shapeHSM_HsmSourceMoments'
         self.measurement.slots.psfShape = 'ext_shapeHSM_HsmPsfMoments'
         self.measurement.plugins['ext_shapeHSM_HsmShapeRegauss'].deblendNChild = ""
+
+        # Keep track of which footprints contain streaks
+        self.measurement.plugins['base_PixelFlags'].masksFpAnywhere = ['STREAK']
+        self.measurement.plugins['base_PixelFlags'].masksFpCenter = ['STREAK']
+
         # Turn off slot setting for measurement for centroid and shape
         # (for which we use the input src catalog measurements)
         self.measurement.slots.centroid = None

diff --git a/python/lsst/pipe/tasks/multiBand.py b/python/lsst/pipe/tasks/multiBand.py
@@ -451,6 +451,9 @@ def setDefaults(self):
                                                                        'INEXACT_PSF']
         self.measurement.plugins['base_PixelFlags'].masksFpCenter = ['CLIPPED', 'SENSOR_EDGE',
                                                                      'INEXACT_PSF']
+        # Keep track of which footprints contain streaks
+        self.measurement.plugins['base_PixelFlags'].masksFpAnywhere = ['STREAK']
+        self.measurement.plugins['base_PixelFlags'].masksFpCenter = ['STREAK']
 
 
 class MeasureMergedCoaddSourcesTask(PipelineTask):