lsst.synpipe  15.0-4-g9ee0f43+2
detectOnlyFakes.py
Go to the documentation of this file.
1 import random
2 import lsst.pex.config as pexConfig
3 import lsst.pipe.base as pipeBase
4 import lsst.afw.table as afwTable
5 import lsst.afw.image as afwImage
6 import lsst.meas.algorithms as measAlg
7 import lsst.afw.detection as afwDetect
8 
9 #WARNING: if you want to add configuration variables (maybe how close you are
10 #to a fake object), you will need to deal with the fact that the configuration
11 #for a retargeted subtask (like this one) blows away any overrides in setDefault of
12 #the parent task (processCoadd in this case) and, it seems any camera specific
13 #overrides in $OBS_SUBARU/config/hsc/processCcd.py.
14 #See https://dev.lsstcorp.org/trac/ticket/2282 for more details
15 #I think you need something like cmdLineTask.applyOverrides to deal with this
16 
17 
18 class OnlyFakesDetectionConfig(measAlg.SourceDetectionTask.ConfigClass):
19  dummyVar = pexConfig.Field(doc='Dummy config variable, does nothing',
20  dtype=bool, default=True)
21 
22 
23 class OnlyFakesDetectionTask(measAlg.SourceDetectionTask):
24  """This task serves culls the source list to sources which overlap with fakes"""
25 
26 
29  ConfigClass = measAlg.SourceDetectionConfig
30 
31  def makeSourceCatalog(self, table, exposure, doSmooth=True, sigma=None, clearMask=True):
32  if self.negativeFlagKey is not None and self.negativeFlagKey not in table.getSchema():
33  raise ValueError("Table has incorrect Schema")
34 
35  # detect the footprints as usual
36  fpSets = self.detectFootprints(exposure=exposure, doSmooth=doSmooth, sigma=sigma,
37  clearMask=clearMask)
38 
39  #ignore objects whose footprints do NOT overlap with the 'FAKE' mask
40  mask = exposure.getMaskedImage().getMask()
41  fakebit = mask.getPlaneBitMask('FAKE')
42  fpPos = fpSets.positive.getFootprints()
43  removes = []
44  for i_foot, foot in enumerate(fpPos):
45  footTmp = afwDetect.Footprint(foot)
46  footTmp.intersectMask(mask, fakebit)
47  if footTmp.getArea() == foot.getArea():
48  removes.append(i_foot)
49  removes = sorted(removes, reverse=True)
50  for r in removes:
51  del fpPos[r]
52 
53  self.log.info("Found %d sources near fake footprints" % len(fpPos))
54 
55  fpSets.numPos = len(fpPos)
56  if fpSets.negative:
57  del fpSets.negative.getFootprints()[0:]
58  fpSets.negative = None
59 
60  # make sources
61  sources = afwTable.SourceCatalog(table)
62  table.preallocate(fpSets.numPos)
63  if fpSets.positive:
64  fpSets.positive.makeSources(sources)
65 
66  return pipeBase.Struct(sources=sources, fpSets=fpSets)
def makeSourceCatalog(self, table, exposure, doSmooth=True, sigma=None, clearMask=True)