LSST Applications  21.0.0+04719a4bac,21.0.0-1-ga51b5d4+f5e6047307,21.0.0-11-g2b59f77+a9c1acf22d,21.0.0-11-ga42c5b2+86977b0b17,21.0.0-12-gf4ce030+76814010d2,21.0.0-13-g1721dae+760e7a6536,21.0.0-13-g3a573fe+768d78a30a,21.0.0-15-g5a7caf0+f21cbc5713,21.0.0-16-g0fb55c1+b60e2d390c,21.0.0-19-g4cded4ca+71a93a33c0,21.0.0-2-g103fe59+bb20972958,21.0.0-2-g45278ab+04719a4bac,21.0.0-2-g5242d73+3ad5d60fb1,21.0.0-2-g7f82c8f+8babb168e8,21.0.0-2-g8f08a60+06509c8b61,21.0.0-2-g8faa9b5+616205b9df,21.0.0-2-ga326454+8babb168e8,21.0.0-2-gde069b7+5e4aea9c2f,21.0.0-2-gecfae73+1d3a86e577,21.0.0-2-gfc62afb+3ad5d60fb1,21.0.0-25-g1d57be3cd+e73869a214,21.0.0-3-g357aad2+ed88757d29,21.0.0-3-g4a4ce7f+3ad5d60fb1,21.0.0-3-g4be5c26+3ad5d60fb1,21.0.0-3-g65f322c+e0b24896a3,21.0.0-3-g7d9da8d+616205b9df,21.0.0-3-ge02ed75+a9c1acf22d,21.0.0-4-g591bb35+a9c1acf22d,21.0.0-4-g65b4814+b60e2d390c,21.0.0-4-gccdca77+0de219a2bc,21.0.0-4-ge8a399c+6c55c39e83,21.0.0-5-gd00fb1e+05fce91b99,21.0.0-6-gc675373+3ad5d60fb1,21.0.0-64-g1122c245+4fb2b8f86e,21.0.0-7-g04766d7+cd19d05db2,21.0.0-7-gdf92d54+04719a4bac,21.0.0-8-g5674e7b+d1bd76f71f,master-gac4afde19b+a9c1acf22d,w.2021.13
LSST Data Management Base Package
makePsfCandidates.py
Go to the documentation of this file.
1 #
2 # LSST Data Management System
3 #
4 # Copyright 2008-2017 AURA/LSST.
5 #
6 # This product includes software developed by the
7 # LSST Project (http://www.lsst.org/).
8 #
9 # This program is free software: you can redistribute it and/or modify
10 # it under the terms of the GNU General Public License as published by
11 # the Free Software Foundation, either version 3 of the License, or
12 # (at your option) any later version.
13 #
14 # This program is distributed in the hope that it will be useful,
15 # but WITHOUT ANY WARRANTY; without even the implied warranty of
16 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 # GNU General Public License for more details.
18 #
19 # You should have received a copy of the LSST License Statement and
20 # the GNU General Public License along with this program. If not,
21 # see <https://www.lsstcorp.org/LegalNotices/>.
22 #
23 __all__ = ["MakePsfCandidatesConfig", "MakePsfCandidatesTask"]
24 
25 import numpy as np
26 
27 from lsst.afw.table import SourceCatalog
28 import lsst.afw.math as afwMath
29 import lsst.pex.config as pexConfig
31 import lsst.pipe.base as pipeBase
32 from . import makePsfCandidate
33 
34 
35 class MakePsfCandidatesConfig(pexConfig.Config):
36  kernelSize = pexConfig.Field(
37  doc="size of the kernel to create",
38  dtype=int,
39  default=21,
40  )
41  borderWidth = pexConfig.Field(
42  doc="number of pixels to ignore around the edge of PSF candidate postage stamps",
43  dtype=int,
44  default=0,
45  )
46 
47 
48 class MakePsfCandidatesTask(pipeBase.Task):
49  """Create PSF candidates given an input catalog.
50  """
51  ConfigClass = MakePsfCandidatesConfig
52  _DefaultName = "makePsfCandidates"
53 
54  def __init__(self, **kwds):
55  pipeBase.Task.__init__(self, **kwds)
56 
57  def run(self, starCat, exposure, psfCandidateField=None):
58  """Make a list of PSF candidates from a star catalog.
59 
60  Parameters
61  ----------
62  starCat : `lsst.afw.table.SourceCatalog`
63  Catalog of stars, as returned by
64  ``lsst.meas.algorithms.starSelector.run()``.
65  exposure : `lsst.afw.image.Exposure`
66  The exposure containing the sources.
67  psfCandidateField : `str` or None
68  Name of flag field to set True for PSF candidates, or None to not
69  set a field; the field is left unchanged for non-candidates.
70 
71  Returns
72  -------
73  struct : `lsst.pipe.base.Struct`
74  Results struct containing:
75 
76  - ``psfCandidates`` : List of PSF candidates
77  (`list` of `lsst.meas.algorithms.PsfCandidate`).
78  - ``goodStarCat`` : Subset of ``starCat`` that was successfully made
79  into PSF candidates (`lsst.afw.table.SourceCatalog`).
80 
81  """
82  psfResult = self.makePsfCandidatesmakePsfCandidates(starCat, exposure)
83 
84  if psfCandidateField is not None:
85  isStarKey = starCat.schema[psfCandidateField].asKey()
86  for star in psfResult.goodStarCat:
87  star.set(isStarKey, True)
88 
89  return psfResult
90 
91  def makePsfCandidates(self, starCat, exposure):
92  """Make a list of PSF candidates from a star catalog.
93 
94  Parameters
95  ----------
96  starCat : `lsst.afw.table.SourceCatalog`
97  Catalog of stars, as returned by
98  ``lsst.meas.algorithms.starSelector.run()``.
99  exposure : `lsst.afw.image.Exposure`
100  The exposure containing the sources.
101 
102  Returns
103  -------
104  struct : `lsst.pipe.base.Struct`
105  Results struct containing:
106 
107  - ``psfCandidates`` : List of PSF candidates
108  (`list` of `lsst.meas.algorithms.PsfCandidate`).
109  - ``goodStarCat`` : Subset of ``starCat`` that was successfully made
110  into PSF candidates (`lsst.afw.table.SourceCatalog`).
111  """
112  goodStarCat = SourceCatalog(starCat.schema)
113 
114  psfCandidateList = []
115  didSetSize = False
116  for star in starCat:
117  try:
118  psfCandidate = makePsfCandidate(star, exposure)
119 
120  # The setXXX methods are class static, but it's convenient to call them on
121  # an instance as we don't know exposures's pixel type
122  # (and hence psfCandidate's exact type)
123  if not didSetSize:
124  psfCandidate.setBorderWidth(self.config.borderWidth)
125  psfCandidate.setWidth(self.config.kernelSize + 2*self.config.borderWidth)
126  psfCandidate.setHeight(self.config.kernelSize + 2*self.config.borderWidth)
127  didSetSize = True
128 
129  im = psfCandidate.getMaskedImage().getImage()
130  except lsst.pex.exceptions.Exception as err:
131  self.log.warn("Failed to make a psfCandidate from star %d: %s", star.getId(), err)
132  continue
133 
134  vmax = afwMath.makeStatistics(im, afwMath.MAX).getValue()
135  if not np.isfinite(vmax):
136  continue
137  psfCandidateList.append(psfCandidate)
138  goodStarCat.append(star)
139 
140  return pipeBase.Struct(
141  psfCandidates=psfCandidateList,
142  goodStarCat=goodStarCat,
143  )
def run(self, starCat, exposure, psfCandidateField=None)
Provides consistent interface for LSST exceptions.
Definition: Exception.h:107
Statistics makeStatistics(lsst::afw::image::Image< Pixel > const &img, lsst::afw::image::Mask< image::MaskPixel > const &msk, int const flags, StatisticsControl const &sctrl=StatisticsControl())
Handle a watered-down front-end to the constructor (no variance)
Definition: Statistics.h:354
std::shared_ptr< PsfCandidate< PixelT > > makePsfCandidate(boost::shared_ptr< afw::table::SourceRecord > const &source, boost::shared_ptr< afw::image::Exposure< PixelT > > image)
Return a PsfCandidate of the right sort.
Definition: PsfCandidate.h:176