def testNoMask(self):
mask = afwImage.Mask(afwGeom.Extent2I(20, 20))
mask.set(0)
fsb = ipDiffim.FindSetBitsU()
bbox = afwGeom.Box2I(afwGeom.Point2I(0, 10), afwGeom.Point2I(9, 12))
fsb.apply(afwImage.Mask(mask, bbox, afwImage.LOCAL))
self.assertEqual(fsb.getBits(), 0)
def testOneMask(self):
mask = afwImage.Mask(afwGeom.Extent2I(20, 20))
mask.set(0)
bitmaskBad = mask.getPlaneBitMask('BAD')
fsb = ipDiffim.FindSetBitsU()
bbox = afwGeom.Box2I(afwGeom.Point2I(9, 10), afwGeom.Point2I(11, 12))
submask = afwImage.Mask(mask, bbox, afwImage.LOCAL)
submask |= bitmaskBad
bbox2 = afwGeom.Box2I(afwGeom.Point2I(8, 8), afwGeom.Point2I(19, 19))
fsb.apply(afwImage.Mask(mask, bbox2, afwImage.LOCAL))
self.assertEqual(fsb.getBits(), bitmaskBad)
def testManyMask(self):
mask = afwImage.Mask(geom.Extent2I(20, 20))
mask.set(0)
bitmaskBad = mask.getPlaneBitMask('BAD')
bitmaskSat = mask.getPlaneBitMask('SAT')
fsb = ipDiffim.FindSetBitsU()
bbox = geom.Box2I(geom.Point2I(9, 10), geom.Point2I(11, 12))
submask = afwImage.Mask(mask, bbox, afwImage.LOCAL)
submask |= bitmaskBad
bbox2 = geom.Box2I(geom.Point2I(8, 8), geom.Point2I(19, 19))
submask2 = afwImage.Mask(mask, bbox2, afwImage.LOCAL)
submask2 |= bitmaskSat
bbox3 = geom.Box2I(geom.Point2I(0, 0), geom.Point2I(19, 19))
fsb.apply(afwImage.Mask(mask, bbox3, afwImage.LOCAL))
self.assertEqual(fsb.getBits(), bitmaskBad | bitmaskSat)