SimpleAssociationTask¶
- class lsst.pipe.tasks.simpleAssociation.SimpleAssociationTask(config: Config | None = None, *, name: str | None = None, parentTask: Task | None = None, log: logging.Logger | lsst.utils.logging.LsstLogAdapter | None = None)¶
Bases:
TaskConstruct DiaObjects from a DataFrame of DIASources by spatially associating the sources.
Represents a simple, brute force algorithm, 2-way matching of DiaSources into. DiaObjects. Algorithm picks the nearest, first match within the matching radius of a DiaObject to associate a source to for simplicity.
Methods Summary
addNewDiaObject(diaSrc, diaSources, ...)Create a new DiaObject and append its data.
createDiaObject(objId, ra, dec)Create a simple empty DiaObject with location and id information.
Empty (clear) the metadata for this Task and all sub-Tasks.
findMatches(src_ra, src_dec, tol, hpIndices, ...)Search healPixels around DiaSource locations for DiaObjects.
Get metadata for all tasks.
Get the task name as a hierarchical name including parent task names.
getName()Get the name of the task.
Get a dictionary of all tasks as a shallow copy.
makeField(doc)Make a
lsst.pex.config.ConfigurableFieldfor this task.makeSubtask(name, **keyArgs)Create a subtask as a new instance as the
nameattribute of this task.run(diaSources[, idGenerator])Associate DiaSources into a collection of DiaObjects using a brute force matching algorithm.
timer(name[, logLevel])Context manager to log performance data for an arbitrary block of code.
updateCatalogs(matchIndex, diaSrc, ...)Update DiaObject and DiaSource values after an association.
Methods Documentation
- addNewDiaObject(diaSrc, diaSources, ccdVisit, diaSourceId, diaObjCat, idCat, diaObjCoords, healPixIndices)¶
Create a new DiaObject and append its data.
- Parameters:
- diaSrc
pandas.Series Full unassociated DiaSource to create a DiaObject from.
- diaSources
pandas.DataFrame DiaSource catalog to update information in. The catalog is modified in place.
- ccdVisit
int Unique identifier of the ccdVisit where
diaSrcwas observed.- diaSourceId
int Unique identifier of the DiaSource.
- diaObjectCat
listof `dict`s Catalog of diaObjects to append the new object o.
- idCat
lsst.afw.table.SourceCatalog Catalog with the IdFactory used to generate unique DiaObject identifiers.
- diaObjectCoords
listof `list`s of `lsst.geom.SpherePoint`s Set of coordinates of DiaSource locations that make up the DiaObject average coordinate.
- healPixIndices
listof `int`s HealPix indices representing the locations of each currently existing DiaObject.
- diaSrc
- createDiaObject(objId, ra, dec)¶
Create a simple empty DiaObject with location and id information.
- findMatches(src_ra, src_dec, tol, hpIndices, diaObjs)¶
Search healPixels around DiaSource locations for DiaObjects.
- Parameters:
- src_ra
float DiaSource RA location.
- src_dec
float DiaSource Dec location.
- tol
float Size of annulus to convert to covering healPixels and search for DiaObjects.
- hpIndices
listof `int`s List of heal pix indices containing the DiaObjects in
diaObjs.- diaObjs
listof `dict`s Catalog diaObjects to with full location information for comparing to DiaSources.
- src_ra
- Returns:
- results
lsst.pipe.base.Struct Results struct containing
distsArray of distances between the current DiaSource diaObjects. (
numpy.ndarrayorNone)matchesArray of array indices of diaObjects this DiaSource matches to. (
numpy.ndarrayorNone)
- results
- getFullMetadata() TaskMetadata¶
Get metadata for all tasks.
- Returns:
- metadata
TaskMetadata The keys are the full task name. Values are metadata for the top-level task and all subtasks, sub-subtasks, etc.
- metadata
Notes
The returned metadata includes timing information (if
@timer.timeMethodis used) and any metadata set by the task. The name of each item consists of the full task name with.replaced by:, followed by.and the name of the item, e.g.:topLevelTaskName:subtaskName:subsubtaskName.itemName
using
:in the full task name disambiguates the rare situation that a task has a subtask and a metadata item with the same name.
- getFullName() str¶
Get the task name as a hierarchical name including parent task names.
- Returns:
- fullName
str The full name consists of the name of the parent task and each subtask separated by periods. For example:
The full name of top-level task “top” is simply “top”.
The full name of subtask “sub” of top-level task “top” is “top.sub”.
The full name of subtask “sub2” of subtask “sub” of top-level task “top” is “top.sub.sub2”.
- fullName
- getName() str¶
Get the name of the task.
- Returns:
- taskName
str Name of the task.
- taskName
See also
getFullNameGet the full name of the task.
- getTaskDict() dict[str, weakref.ReferenceType[lsst.pipe.base.task.Task]]¶
Get a dictionary of all tasks as a shallow copy.
- Returns:
- taskDict
dict Dictionary containing full task name: task object for the top-level task and all subtasks, sub-subtasks, etc.
- taskDict
- classmethod makeField(doc: str) ConfigurableField¶
Make a
lsst.pex.config.ConfigurableFieldfor this task.- Parameters:
- doc
str Help text for the field.
- doc
- Returns:
- configurableField
lsst.pex.config.ConfigurableField A
ConfigurableFieldfor this task.
- configurableField
Examples
Provides a convenient way to specify this task is a subtask of another task.
Here is an example of use:
class OtherTaskConfig(lsst.pex.config.Config): aSubtask = ATaskClass.makeField("brief description of task")
- makeSubtask(name: str, **keyArgs: Any) None¶
Create a subtask as a new instance as the
nameattribute of this task.- Parameters:
- name
str Brief name of the subtask.
- **keyArgs
Extra keyword arguments used to construct the task. The following arguments are automatically provided and cannot be overridden:
config.parentTask.
- name
Notes
The subtask must be defined by
Task.config.name, an instance ofConfigurableFieldorRegistryField.
- run(diaSources, idGenerator=None)¶
Associate DiaSources into a collection of DiaObjects using a brute force matching algorithm.
Reproducible is for the same input data is assured by ordering the DiaSource data by ccdVisit ordering.
- Parameters:
- diaSources
pandas.DataFrame DiaSources grouped by CcdVisitId to spatially associate into DiaObjects.
- idGenerator
lsst.meas.base.IdGenerator, optional Object that generates Object IDs and random number generator seeds.
- diaSources
- Returns:
- results
lsst.pipe.base.Struct Results struct with attributes:
assocDiaSourcesTable of DiaSources with updated values for the DiaObjects they are spatially associated to (
pandas.DataFrame).diaObjectsTable of DiaObjects from matching DiaSources (
pandas.DataFrame).
- results
- timer(name: str, logLevel: int = 10) Iterator[None]¶
Context manager to log performance data for an arbitrary block of code.
- Parameters:
See also
lsst.utils.timer.logInfoImplementation function.
Examples
Creating a timer context:
with self.timer("someCodeToTime"): pass # code to time
- updateCatalogs(matchIndex, diaSrc, diaSources, ccdVisit, diaSourceId, diaObjCat, diaObjCoords, healPixIndices)¶
Update DiaObject and DiaSource values after an association.
- Parameters:
- matchIndex
int Array index location of the DiaObject that
diaSrcwas associated to.- diaSrc
pandas.Series Full unassociated DiaSource to create a DiaObject from.
- diaSources
pandas.DataFrame DiaSource catalog to update information in. The catalog is modified in place.
- ccdVisit
int Unique identifier of the ccdVisit where
diaSrcwas observed.- diaSourceId
int Unique identifier of the DiaSource.
- diaObjectCat
listof `dict`s Catalog of diaObjects to append the new object o.
- diaObjectCoords
listof `list`s of `lsst.geom.SpherePoint`s Set of coordinates of DiaSource locations that make up the DiaObject average coordinate.
- healPixIndices
listof `int`s HealPix indices representing the locations of each currently existing DiaObject.
- matchIndex