Add calib_astrometryUsed field to record sources used

python/lsst/meas/astrom/astrometry.py

@@ -22,6 +22,8 @@
 from __future__ import absolute_import, division, print_function
 from builtins import range
 
+import random
+
 import lsst.pex.config as pexConfig
 import lsst.pipe.base as pipeBase
 from .ref_match import RefMatchTask, RefMatchConfig
@@ -55,6 +57,11 @@ class AstrometryConfig(RefMatchConfig):
         default=0.001,
         min=0,
     )
+    doWriteOutput = pexConfig.Field(
+        dtype=bool,
+        default=True,
+        doc="Write used flag to the schema",
+    )
 
 # The following block adds links to this task from the Task Documentation page.
 ## \addtogroup LSST_task_documentation
@@ -147,6 +154,13 @@ def __init__(self, refObjLoader, schema=None, **kwargs):
         @param[in] kwargs  additional keyword arguments for pipe_base Task.\_\_init\_\_
         """
         RefMatchTask.__init__(self, refObjLoader, schema=schema, **kwargs)
+
+        if self.config.doWriteOutput and not schema is None:
+            self.usedKey = schema.addField("calib_astrometryUsed", type="Flag",
+                                               doc="set if source was used in astrometric calibration")
+        else:
+            self.usedKey = None
+
         self.makeSubtask("wcsFitter")
 
     @pipeBase.timeMethod
@@ -269,6 +283,9 @@ def solve(self, exposure, sourceCat):
             (iterNum, len(tryRes.matches), tryMatchDist.distMean.asArcseconds(),
                 tryMatchDist.distStdDev.asArcseconds()))
 
+        for m in res.matches:
+            if self.usedKey:
+                m.second.set(self.usedKey, True)
         exposure.setWcs(res.wcs)
 
         return pipeBase.Struct(