lsst.pipe.tasks.drpAssociationPipe Namespace Reference

Classes
class	DrpAssociationPipeConnections

Functions
	_addDiaObjectCoords (self, objects, sources)
	_trimToPatch (self, cat, innerPatchBox, innerTractSkyRegion, wcs)

Variables
	visitInfo = finalVisitSummary.find(detector).visitInfo
	bbox = finalVisitSummary.find(detector).getBBox()
	wcs = finalVisitSummary.find(detector).wcs
	ssoAssocResult
	associatedSsSources = ssoAssocResult.associatedSsSources
	associatedSsDiaSources = ssoAssocResult.ssoAssocDiaSources
	ssInTractPatch
	assocDiaSrcIds = set(associatedSsSources['diaSourceId'])
list	diaSrcMask = [diaId in assocDiaSrcIds for diaId in associatedSsDiaSources['diaSourceId']]
	unassociatedSsObjects = ssoAssocResult.unassociatedSsObjects
	ssObjInTractPatch
	nSsSrc = ssInTractPatch.sum()
	nSsObj = ssObjInTractPatch.sum()
	diaCat = ssoAssocResult.unAssocDiaSources.to_pandas()
	diaInTractPatch
	nDiaSrc = diaInTractPatch.sum()
	diaSourceHistoryCat = pd.concat(diaSourceHistory)
	ssSourceHistoryCat = tb.vstack(ssSourceHistory)
	unassociatedSsObjectHistoryCat = tb.vstack(unassociatedSsObjectHistory)
	assocResult = self.associator.run(diaSourceHistoryCat, idGenerator=idGenerator)
	assocDiaSources
	diaObjectTable
	assocDiaSourceTable

Detailed Description

Pipeline for running DiaSource association in a DRP context.

Function Documentation

_addDiaObjectCoords()

lsst.pipe.tasks.drpAssociationPipe._addDiaObjectCoords ( self, objects, sources )

protected

Definition at line 361 of file drpAssociationPipe.py.

    def _addDiaObjectCoords(self, objects, sources):
        obj = objects[['ra', 'dec']].rename(columns={"ra": "coord_ra", "dec": "coord_dec"})
        df = pd.merge(sources.reset_index(), obj, left_on='diaObjectId', right_index=True,
                      how='inner').set_index('diaSourceId')
        return df
 

_trimToPatch()

lsst.pipe.tasks.drpAssociationPipe._trimToPatch ( self, cat, innerPatchBox, innerTractSkyRegion, wcs )

protected

Create generator testing if a set of DiaSources are in the
patch/tract.

Parameters
----------
cat : `pandas.DataFrame`
    Catalog of DiaSources to test within patch/tract.
innerPatchBox : `lsst.geom.Box2D`
    Bounding box of the patch.
wcs : `lsst.geom.SkyWcs`
    Wcs of the tract.

Returns
------
isInPatch : `numpy.ndarray`, (N,)
    Booleans representing if the DiaSources are contained within the
    current patch and tract.

Definition at line 367 of file drpAssociationPipe.py.

    def _trimToPatch(self, cat, innerPatchBox, innerTractSkyRegion, wcs):
        """Create generator testing if a set of DiaSources are in the
        patch/tract.
 
        Parameters
        ----------
        cat : `pandas.DataFrame`
            Catalog of DiaSources to test within patch/tract.
        innerPatchBox : `lsst.geom.Box2D`
            Bounding box of the patch.
        wcs : `lsst.geom.SkyWcs`
            Wcs of the tract.
 
        Returns
        ------
        isInPatch : `numpy.ndarray`, (N,)
            Booleans representing if the DiaSources are contained within the
            current patch and tract.
        """
        isInPatch = np.zeros(len(cat), dtype=bool)
 
        for idx, row in cat.iterrows():
            spPoint = geom.SpherePoint(row["ra"], row["dec"], geom.degrees)
            pxCoord = wcs.skyToPixel(spPoint)
            ra_rad = np.deg2rad(row["ra"])
            dec_rad = np.deg2rad(row["dec"])
 
            isInPatch[idx] = innerPatchBox.contains(pxCoord) and innerTractSkyRegion.contains(ra_rad, dec_rad)
 
        return isInPatch

Variable Documentation

assocDiaSources

lsst.pipe.tasks.drpAssociationPipe.assocDiaSources

Definition at line 351 of file drpAssociationPipe.py.

assocDiaSourceTable

lsst.pipe.tasks.drpAssociationPipe.assocDiaSourceTable

Definition at line 356 of file drpAssociationPipe.py.

assocDiaSrcIds

lsst.pipe.tasks.drpAssociationPipe.assocDiaSrcIds = set(associatedSsSources['diaSourceId'])

Definition at line 273 of file drpAssociationPipe.py.

associatedSsDiaSources

lsst.pipe.tasks.drpAssociationPipe.associatedSsDiaSources = ssoAssocResult.ssoAssocDiaSources

Definition at line 267 of file drpAssociationPipe.py.

associatedSsSources

lsst.pipe.tasks.drpAssociationPipe.associatedSsSources = ssoAssocResult.associatedSsSources

Definition at line 266 of file drpAssociationPipe.py.

assocResult

lsst.pipe.tasks.drpAssociationPipe.assocResult = self.associator.run(diaSourceHistoryCat, idGenerator=idGenerator)

Definition at line 345 of file drpAssociationPipe.py.

bbox

lsst.pipe.tasks.drpAssociationPipe.bbox = finalVisitSummary.find(detector).getBBox()

Definition at line 256 of file drpAssociationPipe.py.

diaCat

lsst.pipe.tasks.drpAssociationPipe.diaCat = ssoAssocResult.unAssocDiaSources.to_pandas()

Definition at line 286 of file drpAssociationPipe.py.

diaInTractPatch

lsst.pipe.tasks.drpAssociationPipe.diaInTractPatch

Initial value:

=  self._trimToPatch(diaCat,
                                                innerPatchBox,
                                                innerTractSkyRegion,
                                                skyInfo.wcs)

Definition at line 290 of file drpAssociationPipe.py.

diaObjectTable

lsst.pipe.tasks.drpAssociationPipe.diaObjectTable

Definition at line 355 of file drpAssociationPipe.py.

diaSourceHistoryCat

lsst.pipe.tasks.drpAssociationPipe.diaSourceHistoryCat = pd.concat(diaSourceHistory)

Definition at line 316 of file drpAssociationPipe.py.

diaSrcMask

list lsst.pipe.tasks.drpAssociationPipe.diaSrcMask = [diaId in assocDiaSrcIds for diaId in associatedSsDiaSources['diaSourceId']]

Definition at line 274 of file drpAssociationPipe.py.

nDiaSrc

lsst.pipe.tasks.drpAssociationPipe.nDiaSrc = diaInTractPatch.sum()

Definition at line 296 of file drpAssociationPipe.py.

nSsObj

lsst.pipe.tasks.drpAssociationPipe.nSsObj = ssObjInTractPatch.sum()

Definition at line 284 of file drpAssociationPipe.py.

nSsSrc

lsst.pipe.tasks.drpAssociationPipe.nSsSrc = ssInTractPatch.sum()

Definition at line 283 of file drpAssociationPipe.py.

ssInTractPatch

lsst.pipe.tasks.drpAssociationPipe.ssInTractPatch

Initial value:

=  self._trimToPatch(associatedSsSources.to_pandas(),
                                                   innerPatchBox,
                                                   innerTractSkyRegion,
                                                   skyInfo.wcs)

Definition at line 268 of file drpAssociationPipe.py.

ssoAssocResult

lsst.pipe.tasks.drpAssociationPipe.ssoAssocResult

Initial value:

=  self.solarSystemAssociator.run(
                    tb.Table.from_pandas(diaCat),
                    ssCat,
                    visitInfo,
                    bbox,
                    wcs,
                )

Definition at line 258 of file drpAssociationPipe.py.

ssObjInTractPatch

lsst.pipe.tasks.drpAssociationPipe.ssObjInTractPatch

Initial value:

=  self._trimToPatch(unassociatedSsObjects.to_pandas(),
                                                      innerPatchBox,
                                                      innerTractSkyRegion,
                                                      skyInfo.wcs)

Definition at line 278 of file drpAssociationPipe.py.

ssSourceHistoryCat

lsst.pipe.tasks.drpAssociationPipe.ssSourceHistoryCat = tb.vstack(ssSourceHistory)

Definition at line 322 of file drpAssociationPipe.py.

unassociatedSsObjectHistoryCat

lsst.pipe.tasks.drpAssociationPipe.unassociatedSsObjectHistoryCat = tb.vstack(unassociatedSsObjectHistory)

Definition at line 328 of file drpAssociationPipe.py.

unassociatedSsObjects

lsst.pipe.tasks.drpAssociationPipe.unassociatedSsObjects = ssoAssocResult.unassociatedSsObjects

Definition at line 277 of file drpAssociationPipe.py.

visitInfo

lsst.pipe.tasks.drpAssociationPipe.visitInfo = finalVisitSummary.find(detector).visitInfo

Definition at line 255 of file drpAssociationPipe.py.

wcs

lsst.pipe.tasks.drpAssociationPipe.wcs = finalVisitSummary.find(detector).wcs

Definition at line 257 of file drpAssociationPipe.py.