def testGaussianMeasureCentroid(self):
"""Test that we can instantiate and play with GAUSSIAN centroids"""
control = measBase.GaussianCentroidControl()
control.doFootprintCheck = False
self.do_testAstrometry(measBase.GaussianCentroidAlgorithm, 10.0,
control)
def showPsfMosaic(exposure, psf=None, nx=7, ny=None,
showCenter=True, showEllipticity=False, showFwhm=False,
stampSize=0, frame=None, title=None):
"""Show a mosaic of Psf images. exposure may be an Exposure (optionally with PSF),
or a tuple (width, height)
If stampSize is > 0, the psf images will be trimmed to stampSize*stampSize
"""
scale = 1.0
if showFwhm:
showEllipticity = True
scale = 2*math.log(2) # convert sigma^2 to HWHM^2 for a Gaussian
mos = displayUtils.Mosaic()
try: # maybe it's a real Exposure
width, height = exposure.getWidth(), exposure.getHeight()
x0, y0 = exposure.getXY0()
if not psf:
psf = exposure.getPsf()
except AttributeError:
try: # OK, maybe a list [width, height]
width, height = exposure[0], exposure[1]
x0, y0 = 0, 0
except TypeError: # I guess not
raise RuntimeError("Unable to extract width/height from object of type %s" % type(exposure))
if not ny:
ny = int(nx*float(height)/width + 0.5)
if not ny:
ny = 1
centroidName = "base_GaussianCentroid"
shapeName = "base_SdssShape"
schema = afwTable.SourceTable.makeMinimalSchema()
schema.getAliasMap().set("slot_Centroid", centroidName)
schema.getAliasMap().set("slot_Centroid_flag", centroidName+"_flag")
control = measBase.GaussianCentroidControl()
centroider = measBase.GaussianCentroidAlgorithm(control, centroidName, schema)
sdssShape = measBase.SdssShapeControl()
shaper = measBase.SdssShapeAlgorithm(sdssShape, shapeName, schema)
table = afwTable.SourceTable.make(schema)
table.defineCentroid(centroidName)
table.defineShape(shapeName)
bbox = None
if stampSize > 0:
w, h = psf.computeImage(afwGeom.PointD(0, 0)).getDimensions()
if stampSize <= w and stampSize <= h:
bbox = afwGeom.BoxI(afwGeom.PointI((w - stampSize)//2, (h - stampSize)//2),
afwGeom.ExtentI(stampSize, stampSize))
centers = []
shapes = []
for iy in range(ny):
for ix in range(nx):
x = int(ix*(width-1)/(nx-1)) + x0
y = int(iy*(height-1)/(ny-1)) + y0
im = psf.computeImage(afwGeom.PointD(x, y)).convertF()
imPeak = psf.computePeak(afwGeom.PointD(x, y))
im /= imPeak
if bbox:
im = im.Factory(im, bbox)
lab = "PSF(%d,%d)" % (x, y) if False else ""
mos.append(im, lab)
exp = afwImage.makeExposure(afwImage.makeMaskedImage(im))
w, h = im.getWidth(), im.getHeight()
centerX = im.getX0() + w//2
centerY = im.getY0() + h//2
src = table.makeRecord()
foot = afwDet.Footprint(exp.getBBox())
foot.addPeak(centerX, centerY, 1)
src.setFootprint(foot)
centroider.measure(src, exp)
centers.append((src.getX() - im.getX0(), src.getY() - im.getY0()))
shaper.measure(src, exp)
shapes.append((src.getIxx(), src.getIxy(), src.getIyy()))
mos.makeMosaic(frame=frame, title=title if title else "Model Psf", mode=nx)
if centers and frame is not None:
with ds9.Buffering():
for i, (cen, shape) in enumerate(zip(centers, shapes)):
bbox = mos.getBBox(i)
xc, yc = cen[0] + bbox.getMinX(), cen[1] + bbox.getMinY()
if showCenter:
ds9.dot("+", xc, yc, ctype=ds9.BLUE, frame=frame)
if showEllipticity:
ixx, ixy, iyy = shape
ixx *= scale
ixy *= scale
iyy *= scale
ds9.dot("@:%g,%g,%g" % (ixx, ixy, iyy), xc, yc, frame=frame, ctype=ds9.RED)
return mos
def showKernelMosaic(bbox, kernel, nx=7, ny=None, frame=None, title=None,
showCenter=True, showEllipticity=True):
"""Show a mosaic of Kernel images.
"""
mos = displayUtils.Mosaic()
x0 = bbox.getBeginX()
y0 = bbox.getBeginY()
width = bbox.getWidth()
height = bbox.getHeight()
if not ny:
ny = int(nx*float(height)/width + 0.5)
if not ny:
ny = 1
schema = afwTable.SourceTable.makeMinimalSchema()
centroidName = "base_GaussianCentroid"
shapeName = "base_SdssShape"
control = measBase.GaussianCentroidControl()
schema.getAliasMap().set("slot_Centroid", centroidName)
schema.getAliasMap().set("slot_Centroid_flag", centroidName+"_flag")
centroider = measBase.GaussianCentroidAlgorithm(control, centroidName, schema)
sdssShape = measBase.SdssShapeControl()
shaper = measBase.SdssShapeAlgorithm(sdssShape, shapeName, schema)
table = afwTable.SourceTable.make(schema)
table.defineCentroid(centroidName)
table.defineShape(shapeName)
centers = []
shapes = []
for iy in range(ny):
for ix in range(nx):
x = int(ix*(width-1)/(nx-1)) + x0
y = int(iy*(height-1)/(ny-1)) + y0
im = afwImage.ImageD(kernel.getDimensions())
ksum = kernel.computeImage(im, False, x, y)
lab = "Kernel(%d,%d)=%.2f" % (x, y, ksum) if False else ""
mos.append(im, lab)
# GaussianCentroidAlgorithm.measure requires an exposure of floats
exp = afwImage.makeExposure(afwImage.makeMaskedImage(im.convertF()))
w, h = im.getWidth(), im.getHeight()
centerX = im.getX0() + w//2
centerY = im.getY0() + h//2
src = table.makeRecord()
foot = afwDet.Footprint(exp.getBBox())
foot.addPeak(centerX, centerY, 1)
src.setFootprint(foot)
centroider.measure(src, exp)
centers.append((src.getX(), src.getY()))
shaper.measure(src, exp)
shapes.append((src.getIxx(), src.getIxy(), src.getIyy()))
mos.makeMosaic(frame=frame, title=title if title else "Model Kernel", mode=nx)
if centers and frame is not None:
i = 0
with ds9.Buffering():
for cen, shape in zip(centers, shapes):
bbox = mos.getBBox(i)
i += 1
xc, yc = cen[0] + bbox.getMinX(), cen[1] + bbox.getMinY()
if showCenter:
ds9.dot("+", xc, yc, ctype=ds9.BLUE, frame=frame)
if showEllipticity:
ixx, ixy, iyy = shape
ds9.dot("@:%g,%g,%g" % (ixx, ixy, iyy), xc, yc, frame=frame, ctype=ds9.RED)
return mos