LSSTApplications  20.0.0
LSSTDataManagementBasePackage
Classes | Functions
lsst.meas.algorithms.skyObjects Namespace Reference

Classes

class  SkyObjectsConfig
 
class  SkyObjectsTask
 

Functions

def generateSkyObjects (mask, seed, config)
 

Function Documentation

◆ generateSkyObjects()

def lsst.meas.algorithms.skyObjects.generateSkyObjects (   mask,
  seed,
  config 
)
Generate a list of Footprints of sky objects

Sky objects don't overlap with other objects. This is determined
through the provided `mask` (in which objects are typically flagged
as `DETECTED`).

The algorithm for determining sky objects is random trial and error:
we try up to `nTrialSkySources` random positions to find `nSources`
sky objects.

Parameters
----------
mask : `lsst.afw.image.Mask`
    Input mask plane, which identifies pixels to avoid for the sky
    objects.
seed : `int`
    Random number generator seed.
config : `SkyObjectsConfig`
    Configuration for finding sky objects.

Returns
-------
skyFootprints : `list` of `lsst.afw.detection.Footprint`
    Footprints of sky objects. Each will have a peak at the center
    of the sky object.

Definition at line 29 of file skyObjects.py.

29 def generateSkyObjects(mask, seed, config):
30  """Generate a list of Footprints of sky objects
31 
32  Sky objects don't overlap with other objects. This is determined
33  through the provided `mask` (in which objects are typically flagged
34  as `DETECTED`).
35 
36  The algorithm for determining sky objects is random trial and error:
37  we try up to `nTrialSkySources` random positions to find `nSources`
38  sky objects.
39 
40  Parameters
41  ----------
42  mask : `lsst.afw.image.Mask`
43  Input mask plane, which identifies pixels to avoid for the sky
44  objects.
45  seed : `int`
46  Random number generator seed.
47  config : `SkyObjectsConfig`
48  Configuration for finding sky objects.
49 
50  Returns
51  -------
52  skyFootprints : `list` of `lsst.afw.detection.Footprint`
53  Footprints of sky objects. Each will have a peak at the center
54  of the sky object.
55  """
56  if config.nSources <= 0:
57  return []
58 
59  skySourceRadius = config.sourceRadius
60  nSkySources = config.nSources
61  nTrialSkySources = config.nTrialSources
62  if nTrialSkySources is None:
63  nTrialSkySources = config.nTrialSourcesMultiplier*nSkySources
64 
65  box = mask.getBBox()
66  box.grow(-(int(skySourceRadius) + 1)) # Avoid objects partially off the image
67  xMin, yMin = box.getMin()
68  xMax, yMax = box.getMax()
69 
70  avoid = lsst.afw.geom.SpanSet.fromMask(mask, mask.getPlaneBitMask(config.avoidMask))
71  if config.growMask > 0:
72  avoid = avoid.dilated(config.growMask)
73 
74  rng = lsst.afw.math.Random(seed=seed)
75 
76  skyFootprints = []
77  for _ in range(nTrialSkySources):
78  if len(skyFootprints) == nSkySources:
79  break
80 
81  x = int(rng.flat(xMin, xMax))
82  y = int(rng.flat(yMin, yMax))
83  spans = lsst.afw.geom.SpanSet.fromShape(int(skySourceRadius), offset=(x, y))
84  if spans.overlaps(avoid):
85  continue
86 
87  fp = lsst.afw.detection.Footprint(spans, mask.getBBox())
88  fp.addPeak(x, y, 0)
89  skyFootprints.append(fp)
90 
91  return skyFootprints
92 
93 
lsst::meas::algorithms.skyObjects.generateSkyObjects
def generateSkyObjects(mask, seed, config)
Definition: skyObjects.py:29
lsst::afw::geom::SpanSet::fromMask
static std::shared_ptr< geom::SpanSet > fromMask(image::Mask< T > const &mask, UnaryPredicate comparator=details::AnyBitSetFunctor< T >())
Create a SpanSet from a mask.
Definition: SpanSet.h:644
lsst::afw::geom::SpanSet::fromShape
static std::shared_ptr< geom::SpanSet > fromShape(int r, Stencil s=Stencil::CIRCLE, lsst::geom::Point2I offset=lsst::geom::Point2I())
Factory function for creating SpanSets from a Stencil.
Definition: SpanSet.cc:689
lsst::afw::math::Random
A class that can be used to generate sequences of random numbers according to a number of different a...
Definition: Random.h:57
lsst::afw::detection::Footprint
Class to describe the properties of a detected object from an image.
Definition: Footprint.h:63