def singleTestInstance(self, filename, distortFunc, doPlot=False):
sourceCat = self.loadSourceCatalog(self.filename)
refCat = self.computePosRefCatalog(sourceCat)
distortedCat = distort.distortList(sourceCat, distortFunc)
if doPlot:
import matplotlib.pyplot as plt
undistorted = [self.wcs.skyToPixel(self.distortedWcs.pixelToSky(ss.getCentroid())) for
ss in distortedCat]
refs = [self.wcs.skyToPixel(ss.getCoord()) for ss in refCat]
def plot(catalog, symbol):
plt.plot([ss.getX() for ss in catalog], [ss.getY() for ss in catalog], symbol)
#plot(sourceCat, 'k+') # Original positions: black +
plot(distortedCat, 'b+') # Distorted positions: blue +
plot(undistorted, 'g+') # Undistorted positions: green +
plot(refs, 'rx') # Reference catalog: red x
# The green + should overlap with the red x, because that's how matchOptimisticB does it.
# The black + happens to overlap with those also, but that's beside the point.
plt.show()
sourceCat = distortedCat
matchRes = self.matchOptimisticB.matchObjectsToSources(
refCat = refCat,
sourceCat = sourceCat,
wcs = self.distortedWcs,
refFluxField = "r_flux",
)
matches = matchRes.matches
if doPlot:
measAstrom.plotAstrometry(matches=matches, refCat=refCat, sourceCat=sourceCat)
self.assertEqual(len(matches), 183)
refCoordKey = afwTable.CoordKey(refCat.schema["coord"])
srcCoordKey = afwTable.CoordKey(sourceCat.schema["coord"])
refCentroidKey = afwTable.Point2DKey(refCat.getSchema()["centroid"])
maxDistErr = afwGeom.Angle(0)
for refObj, source, distRad in matches:
sourceCoord = source.get(srcCoordKey)
refCoord = refObj.get(refCoordKey)
predDist = sourceCoord.angularSeparation(refCoord)
distErr = abs(predDist - distRad*afwGeom.radians)
maxDistErr = max(distErr, maxDistErr)
if refObj.getId() != source.getId():
refCentroid = refObj.get(refCentroidKey)
sourceCentroid = source.getCentroid()
radius = math.hypot(*(refCentroid - sourceCentroid))
self.fail("ID mismatch: %s at %s != %s at %s; error = %0.1f pix" %
(refObj.getId(), refCentroid, source.getId(), sourceCentroid, radius))
self.assertLess(maxDistErr.asArcseconds(), 1e-7)
def singleTestInstance(self, filename, distortFunc, doPlot=False):
sourceCat = self.loadSourceCatalog(self.filename)
refCat = self.computePosRefCatalog(sourceCat)
distortedCat = distort.distortList(sourceCat, distortFunc)
if doPlot:
import matplotlib.pyplot as plt
undistorted = [
self.wcs.skyToPixel(
self.distortedWcs.pixelToSky(ss.getCentroid()))
for ss in distortedCat
]
refs = [self.wcs.skyToPixel(ss.getCoord()) for ss in refCat]
def plot(catalog, symbol):
plt.plot([ss.getX() for ss in catalog],
[ss.getY() for ss in catalog], symbol)
plot(distortedCat, 'b+') # Distorted positions: blue +
plot(undistorted, 'g+') # Undistorted positions: green +
plot(refs, 'rx') # Reference catalog: red x
# The green + should overlap with the red x, because that's how
# MatchPessimisticB does it.
plt.show()
sourceCat = distortedCat
matchRes = self.MatchPessimisticB.matchObjectsToSources(
refCat=refCat,
sourceCat=sourceCat,
wcs=self.distortedWcs,
refFluxField="r_flux",
)
matches = matchRes.matches
if doPlot:
measAstrom.plotAstrometry(matches=matches,
refCat=refCat,
sourceCat=sourceCat)
self.assertEqual(len(matches), 183)
refCoordKey = afwTable.CoordKey(refCat.schema["coord"])
srcCoordKey = afwTable.CoordKey(sourceCat.schema["coord"])
refCentroidKey = afwTable.Point2DKey(refCat.getSchema()["centroid"])
maxDistErr = afwGeom.Angle(0)
for refObj, source, distRad in matches:
sourceCoord = source.get(srcCoordKey)
refCoord = refObj.get(refCoordKey)
predDist = sourceCoord.angularSeparation(refCoord)
distErr = abs(predDist - distRad * afwGeom.radians)
maxDistErr = max(distErr, maxDistErr)
if refObj.getId() != source.getId():
refCentroid = refObj.get(refCentroidKey)
sourceCentroid = source.getCentroid()
radius = math.hypot(*(refCentroid - sourceCentroid))
self.fail(
"ID mismatch: %s at %s != %s at %s; error = %0.1f pix" %
(refObj.getId(), refCentroid, source.getId(),
sourceCentroid, radius))
self.assertLess(maxDistErr.asArcseconds(), 1e-7)
def singleTestInstance(self, filename, distortFunc, doPlot=False):
sourceCat = self.loadSourceCatalog(self.filename)
refCat = self.computePosRefCatalog(sourceCat)
# Apply source selector to sourceCat, using the astrometry config defaults
tempConfig = measAstrom.AstrometryTask.ConfigClass()
tempConfig.matcher.retarget(measAstrom.MatchOptimisticBTask)
tempConfig.sourceSelector["matcher"].excludePixelFlags = False
tempSolver = measAstrom.AstrometryTask(config=tempConfig,
refObjLoader=None)
sourceSelection = tempSolver.sourceSelector.run(sourceCat)
distortedCat = distort.distortList(sourceSelection.sourceCat,
distortFunc)
if doPlot:
import matplotlib.pyplot as plt
undistorted = [
self.wcs.skyToPixel(
self.distortedWcs.pixelToSky(ss.getCentroid()))
for ss in distortedCat
]
refs = [self.wcs.skyToPixel(ss.getCoord()) for ss in refCat]
def plot(catalog, symbol):
plt.plot([ss.getX() for ss in catalog],
[ss.getY() for ss in catalog], symbol)
# plot(sourceCat, 'k+') # Original positions: black +
plot(distortedCat, 'b+') # Distorted positions: blue +
plot(undistorted, 'g+') # Undistorted positions: green +
plot(refs, 'rx') # Reference catalog: red x
# The green + should overlap with the red x, because that's how matchOptimisticB does it.
# The black + happens to overlap with those also, but that's beside the point.
plt.show()
sourceCat = distortedCat
matchRes = self.matchOptimisticB.matchObjectsToSources(
refCat=refCat,
sourceCat=sourceCat,
wcs=self.distortedWcs,
sourceFluxField='slot_ApFlux_instFlux',
refFluxField="r_flux",
)
matches = matchRes.matches
if doPlot:
measAstrom.plotAstrometry(matches=matches,
refCat=refCat,
sourceCat=sourceCat)
self.assertEqual(len(matches), 183)
refCoordKey = afwTable.CoordKey(refCat.schema["coord"])
srcCoordKey = afwTable.CoordKey(sourceCat.schema["coord"])
refCentroidKey = afwTable.Point2DKey(refCat.getSchema()["centroid"])
maxDistErr = 0 * lsst.geom.radians
for refObj, source, distRad in matches:
sourceCoord = source.get(srcCoordKey)
refCoord = refObj.get(refCoordKey)
predDist = sourceCoord.separation(refCoord)
distErr = abs(predDist - distRad * lsst.geom.radians)
maxDistErr = max(distErr, maxDistErr)
if refObj.getId() != source.getId():
refCentroid = refObj.get(refCentroidKey)
sourceCentroid = source.getCentroid()
radius = math.hypot(*(refCentroid - sourceCentroid))
self.fail(
"ID mismatch: %s at %s != %s at %s; error = %0.1f pix" %
(refObj.getId(), refCentroid, source.getId(),
sourceCentroid, radius))
self.assertLess(maxDistErr.asArcseconds(), 1e-7)
def singleTestInstance(self, filename, distortFunc, doPlot=False):
sourceCat = self.loadSourceCatalog(self.filename)
refCat = self.computePosRefCatalog(sourceCat)
# Apply source selector to sourceCat, using the astrometry config defaults
tempConfig = measAstrom.AstrometryTask.ConfigClass()
tempSolver = measAstrom.AstrometryTask(config=tempConfig, refObjLoader=None)
sourceSelection = tempSolver.sourceSelector.run(sourceCat)
distortedCat = distort.distortList(sourceSelection.sourceCat, distortFunc)
if doPlot:
import matplotlib.pyplot as plt
undistorted = [self.wcs.skyToPixel(self.distortedWcs.pixelToSky(ss.getCentroid()))
for ss in distortedCat]
refs = [self.wcs.skyToPixel(ss.getCoord()) for ss in refCat]
def plot(catalog, symbol):
plt.plot([ss.getX() for ss in catalog],
[ss.getY() for ss in catalog], symbol)
plot(distortedCat, 'b+') # Distorted positions: blue +
plot(undistorted, 'g+') # Undistorted positions: green +
plot(refs, 'rx') # Reference catalog: red x
# The green + should overlap with the red x, because that's how
# MatchPessimisticB does it.
plt.show()
sourceCat = distortedCat
matchRes = self.MatchPessimisticB.matchObjectsToSources(
refCat=refCat,
sourceCat=sourceCat,
wcs=self.distortedWcs,
sourceFluxField='slot_ApFlux_instFlux',
refFluxField="r_flux",
)
matches = matchRes.matches
if doPlot:
measAstrom.plotAstrometry(matches=matches, refCat=refCat,
sourceCat=sourceCat)
self.assertEqual(len(matches), self.expectedMatches)
refCoordKey = afwTable.CoordKey(refCat.schema["coord"])
srcCoordKey = afwTable.CoordKey(sourceCat.schema["coord"])
refCentroidKey = afwTable.Point2DKey(refCat.getSchema()["centroid"])
maxDistErr = 0*lsst.geom.radians
for refObj, source, distRad in matches:
sourceCoord = source.get(srcCoordKey)
refCoord = refObj.get(refCoordKey)
predDist = sourceCoord.separation(refCoord)
distErr = abs(predDist - distRad*lsst.geom.radians)
maxDistErr = max(distErr, maxDistErr)
if refObj.getId() != source.getId():
refCentroid = refObj.get(refCentroidKey)
sourceCentroid = source.getCentroid()
radius = math.hypot(*(refCentroid - sourceCentroid))
self.fail(
"ID mismatch: %s at %s != %s at %s; error = %0.1f pix" %
(refObj.getId(), refCentroid, source.getId(),
sourceCentroid, radius))
self.assertLess(maxDistErr.asArcseconds(), 1e-7)