lsst.meas.base gfc624380f7+a13aefd6d2
|
ConfigClass = ForcedPhotCcdConfig RunnerClass = pipeBase.ButlerInitializedTaskRunner _DefaultName = "forcedPhotCcd" dataPrefix = "" def __init__(self, butler=None, refSchema=None, initInputs=None, **kwds): super().__init__(**kwds) if initInputs is not None: refSchema = initInputs['inputSchema'].schema self.makeSubtask("references", butler=butler, schema=refSchema) if refSchema is None: refSchema = self.references.schema self.makeSubtask("measurement", refSchema=refSchema) # It is necessary to get the schema internal to the forced measurement task until such a time # that the schema is not owned by the measurement task, but is passed in by an external caller if self.config.doApCorr: self.makeSubtask("applyApCorr", schema=self.measurement.schema) self.makeSubtask('catalogCalculation', schema=self.measurement.schema) self.outputSchema = lsst.afw.table.SourceCatalog(self.measurement.schema) def runQuantum(self, butlerQC, inputRefs, outputRefs): inputs = butlerQC.get(inputRefs) tract = butlerQC.quantum.dataId['tract'] skyMap = inputs.pop("skyMap") inputs['refWcs'] = skyMap[tract].getWcs() inputs['refCat'] = self.mergeAndFilterReferences(inputs['exposure'], inputs['refCat'], inputs['refWcs']) inputs['measCat'], inputs['exposureId'] = self.generateMeasCat(inputRefs.exposure.dataId, inputs['exposure'], inputs['refCat'], inputs['refWcs'], "visit_detector") self.attachFootprints(inputs['measCat'], inputs['refCat'], inputs['exposure'], inputs['refWcs']) # TODO: apply external calibrations (DM-17062) outputs = self.run(**inputs) butlerQC.put(outputs, outputRefs) def mergeAndFilterReferences(self, exposure, refCats, refWcs):
# Step 1: Determine bounds of the exposure photometry will # be performed on. expWcs = exposure.getWcs() expRegion = exposure.getBBox(lsst.afw.image.PARENT) expBBox = lsst.geom.Box2D(expRegion) expBoxCorners = expBBox.getCorners() expSkyCorners = [expWcs.pixelToSky(corner).getVector() for corner in expBoxCorners] expPolygon = lsst.sphgeom.ConvexPolygon(expSkyCorners) # Step 2: Filter out reference catalog sources that are # not contained within the exposure boundaries, or whose # parents are not within the exposure boundaries. Note # that within a single input refCat, the parents always # appear before the children. mergedRefCat = None for refCat in refCats: refCat = refCat.get() if mergedRefCat is None: mergedRefCat = lsst.afw.table.SourceCatalog(refCat.table) containedIds = {0} # zero as a parent ID means "this is a parent" for record in refCat: if expPolygon.contains(record.getCoord().getVector()) and record.getParent() in containedIds: record.setFootprint(record.getFootprint()) mergedRefCat.append(record) containedIds.add(record.getId()) if mergedRefCat is None: raise RuntimeError("No reference objects for forced photometry.") mergedRefCat.sort(lsst.afw.table.SourceTable.getParentKey()) return mergedRefCat def generateMeasCat(self, exposureDataId, exposure, refCat, refWcs, idPackerName):
exposureIdInfo = ExposureIdInfo.fromDataId(exposureDataId, idPackerName) idFactory = exposureIdInfo.makeSourceIdFactory() measCat = self.measurement.generateMeasCat(exposure, refCat, refWcs, idFactory=idFactory) return measCat, exposureIdInfo.expId def runDataRef(self, dataRef, psfCache=None):
Definition at line 478 of file forcedPhotCcd.py.