def testAssembleCcd(self):
"""Test if we can build a Ccd out of Amps"""
#print >> sys.stderr, "Skipping testRotatedCcd"; return
compImage = afwImage.ImageI(os.path.join(eups.productDir("afw"),
"tests", "test_comp.fits.gz"))
compImageTrimmed = afwImage.ImageI(os.path.join(eups.productDir("afw"), "tests",
"test_comp_trimmed.fits.gz"))
ccdId = cameraGeom.Id(1, "LsstLike")
ccdInfo = {"ampSerial" : CameraGeomTestCase.ampSerial}
ccd = cameraGeomUtils.makeCcd(self.geomPolicy, ccdId, ccdInfo=ccdInfo)
#
# Test assembly of images that require preparation for assembly (like
# LSST images)
#
outImage = cameraGeomUtils.makeImageFromCcd(ccd,
imageSource=LsstLikeImage(),
isTrimmed=False, imageFactory=afwImage.ImageU)
self.assertTrue(outImage==compImage)
if display:
cameraGeomUtils.showCcd(ccd, outImage)
ds9.incrDefaultFrame()
ccdId = cameraGeom.Id(1, "ScLike")
ccdInfo = {"ampSerial" : CameraGeomTestCase.ampSerial}
ccd = cameraGeomUtils.makeCcd(self.geomPolicy, ccdId, ccdInfo=ccdInfo)
outImage = cameraGeomUtils.makeImageFromCcd(ccd,
imageSource=ScLikeImage(),
isTrimmed=False, imageFactory=afwImage.ImageU)
self.assertTrue(outImage==compImage)
if display:
cameraGeomUtils.showCcd(ccd, outImage)
ds9.incrDefaultFrame()
#
# Trim the CCD and try again
#
ccd.setTrimmed(True)
ccdId = cameraGeom.Id(1, "LsstLike Trimmed")
ccdInfo = {"ampSerial" : CameraGeomTestCase.ampSerial}
ccd = cameraGeomUtils.makeCcd(self.geomPolicy, ccdId, ccdInfo=ccdInfo)
outImage = cameraGeomUtils.makeImageFromCcd(ccd,
imageSource=LsstLikeImage(),
isTrimmed=True, imageFactory=afwImage.ImageU)
self.assertTrue(outImage==compImage)
if display:
cameraGeomUtils.showCcd(ccd, outImage)
ds9.incrDefaultFrame()
def testLinearity(self):
"""Test if we can set Linearity parameters"""
for ccdNum, threshold in [(-1, 0), (1234, 10),]:
ccdId = cameraGeom.Id(ccdNum, "")
ccd = cameraGeomUtils.makeCcd(self.geomPolicy, ccdId)
for amp in list(ccd)[0:2]: # only two amps in TestCameraGeom.paf
lin = amp.getElectronicParams().getLinearity()
self.assertEqual(lin.threshold, threshold)
def testDefectBase(self):
"""Test DefectBases"""
#print >> sys.stderr, "Skipping testDefectBase"; return
defectsDict = cameraGeomUtils.makeDefects(self.geomPolicy)
for ccdName in ("Defective", "Defective II"):
ccd = cameraGeomUtils.makeCcd(self.geomPolicy, cameraGeom.Id(ccdName))
ccdImage = cameraGeomUtils.makeImageFromCcd(ccd)
if ccdName == "Defective":
#
# Insert some defects into the Ccd
#
for x0, y0, x1, y1 in [
(34, 0, 35, 80 ),
(34, 81, 34, 100),
(180, 100, 182, 130),
]:
bbox = afwGeom.Box2I(afwGeom.Point2I(x0, y0), afwGeom.Point2I(x1, y1))
bad = ccdImage.Factory(ccdImage, bbox, afwImage.LOCAL)
bad.set(100)
if display:
ds9.mtv(ccdImage, title="Defects")
cameraGeomUtils.showCcd(ccd, None)
defects = [v for (k, v) in defectsDict.items() if k == ccd.getId()]
if len(defects) == 0:
contine
elif len(defects) == 1:
defects = defects[0]
else:
raise RuntimeError, ("Found more than one defect set for CCD %s" % ccd.getId())
ccd.setDefects(defects)
if False:
print "CCD (%s)" % ccd.getId()
for a in ccd:
print " ", a.getId(), [str(d.getBBox()) for d in a.getDefects()]
if ccdName == "Defective" and display:
for d in ccd.getDefects():
displayUtils.drawBBox(d.getBBox(), ctype=ds9.CYAN, borderWidth=1.5)
for a in ccd:
for d in a.getDefects():
displayUtils.drawBBox(d.getBBox(), ctype=ds9.YELLOW, borderWidth=1.0)
ds9.incrDefaultFrame()
def testRotatedCcd(self):
"""Test if we can build a Ccd out of Amps"""
#print >> sys.stderr, "Skipping testRotatedCcd"; return
ccdId = cameraGeom.Id("Rot. CCD")
ccdInfo = {"ampSerial" : CameraGeomTestCase.ampSerial}
ccd = cameraGeomUtils.makeCcd(self.geomPolicy, ccdId, ccdInfo=ccdInfo)
ccd.setOrientation(cameraGeom.Orientation(1, 0.0, 0.0, 0.0))
if display:
cameraGeomUtils.showCcd(ccd)
ds9.incrDefaultFrame()
#
# Trim the CCD and try again
#
trimmedImage = trimCcd(ccd)
if display:
cameraGeomUtils.showCcd(ccd, trimmedImage)
ds9.incrDefaultFrame()
def testRotatedCcd(self):
"""Test if we can build a Ccd out of Amps"""
#print >> sys.stderr, "Skipping testRotatedCcd"; return
ccdId = cameraGeom.Id(1, "Rot. CCD")
ccdInfo = {"ampSerial" : CameraGeomTestCase.ampSerial}
ccd = cameraGeomUtils.makeCcd(self.geomPolicy, ccdId, ccdInfo=ccdInfo)
zero = 0.0*afwGeom.radians
ccd.setOrientation(cameraGeom.Orientation(1, zero, zero, zero))
if display:
cameraGeomUtils.showCcd(ccd)
ds9.incrDefaultFrame()
#
# Trim the CCD and try again
#
trimmedImage = trimCcd(ccd)
if display:
cameraGeomUtils.showCcd(ccd, trimmedImage)
ds9.incrDefaultFrame()
def testAssembleCcd(self):
"""Test if we can build a Ccd out of Amps"""
compImage = afwImage.ImageU(os.path.join(eups.productDir("afw"),
"tests", "test_comp.fits.gz"))
compImageTrimmed = afwImage.ImageU(os.path.join(eups.productDir("afw"), "tests",
"test_comp_trimmed.fits.gz"))
ccdId = cameraGeom.Id(1, "LsstLike")
ccdInfo = {"ampSerial" : CameraGeomTestCase.ampSerial}
ccd = cameraGeomUtils.makeCcd(self.geomPolicy, ccdId, ccdInfo=ccdInfo)
#
# Test assembly of images that require preparation for assembly (like
# LSST images)
#
outImage = cameraGeomUtils.makeImageFromCcd(ccd,
imageSource=LsstLikeImage(),
isTrimmed=False, imageFactory=afwImage.ImageU)
self.assertImagesAreEqual(outImage, compImage)
if display:
cameraGeomUtils.showCcd(ccd, outImage)
ds9.incrDefaultFrame()
#
# Test assembly of images that reside in a pre-assembled state from
# the DAQ (like Suprime-Cam images)
#
ccdId = cameraGeom.Id(1, "ScLike")
ccdInfo = {"ampSerial" : CameraGeomTestCase.ampSerial}
ccd = cameraGeomUtils.makeCcd(self.geomPolicy, ccdId, ccdInfo=ccdInfo)
outImage = cameraGeomUtils.makeImageFromCcd(ccd,
imageSource=ScLikeImage(),
isTrimmed=False, imageFactory=afwImage.ImageU)
self.assertImagesAreEqual(outImage, compImage)
if display:
cameraGeomUtils.showCcd(ccd, outImage)
ds9.incrDefaultFrame()
#
# Do the same tests for trimmed ccds.
#
ccdId = cameraGeom.Id(1, "LsstLike")
ccdInfo = {"ampSerial" : CameraGeomTestCase.ampSerial}
ccd = cameraGeomUtils.makeCcd(self.geomPolicy, ccdId, ccdInfo=ccdInfo)
outImage = cameraGeomUtils.makeImageFromCcd(ccd,
imageSource=LsstLikeImage(),
isTrimmed=True, imageFactory=afwImage.ImageU)
ccd.setTrimmed(True)
self.assertImagesAreEqual(outImage, compImageTrimmed)
if display:
cameraGeomUtils.showCcd(ccd, outImage)
ds9.incrDefaultFrame()
ccdId = cameraGeom.Id(1, "ScLike")
ccdInfo = {"ampSerial" : CameraGeomTestCase.ampSerial}
ccd = cameraGeomUtils.makeCcd(self.geomPolicy, ccdId, ccdInfo=ccdInfo)
outImage = cameraGeomUtils.makeImageFromCcd(ccd,
imageSource=ScLikeImage(),
isTrimmed=True, imageFactory=afwImage.ImageU)
ccd.setTrimmed(True)
self.assertImagesAreEqual(outImage, compImageTrimmed)
if display:
cameraGeomUtils.showCcd(ccd, outImage)
ds9.incrDefaultFrame()
def testCcd(self):
"""Test if we can build a Ccd out of Amps"""
#print >> sys.stderr, "Skipping testCcd"; return
ccdId = cameraGeom.Id("CCD")
ccdInfo = {"ampSerial" : CameraGeomTestCase.ampSerial}
ccd = cameraGeomUtils.makeCcd(self.geomPolicy, ccdId, ccdInfo=ccdInfo)
if display:
cameraGeomUtils.showCcd(ccd)
ds9.incrDefaultFrame()
trimmedImage = cameraGeomUtils.makeImageFromCcd(ccd, isTrimmed=True)
cameraGeomUtils.showCcd(ccd, trimmedImage, isTrimmed=True)
ds9.incrDefaultFrame()
for i in range(2):
self.assertEqual(ccd.getSize().getMm()[i],
ccdInfo["pixelSize"]*ccd.getAllPixels(True).getDimensions()[i])
self.assertEqual(ccd.getId().getName(), ccdInfo["name"])
self.assertEqual(ccd.getAllPixels().getWidth(), ccdInfo["width"])
self.assertEqual(ccd.getAllPixels().getHeight(), ccdInfo["height"])
self.assertEqual([a.getId().getSerial() for a in ccd],
range(ccdInfo["ampIdMin"], ccdInfo["ampIdMax"] + 1))
id = cameraGeom.Id("ID%d" % ccdInfo["ampIdMax"])
self.assertTrue(ccd.findAmp(id), id)
self.assertEqual(ccd.findAmp(afwGeom.Point2I(10, 10)).getId().getSerial(), ccdInfo["ampIdMin"])
self.assertEqual(ccd.getAllPixels().getMin(),
ccd.findAmp(afwGeom.Point2I(10, 10)).getAllPixels().getMin())
self.assertEqual(ccd.getAllPixels().getMax(),
ccd.findAmp(afwGeom.Point2I(ccdInfo["width"] - 1,
ccdInfo["height"] - 1)).getAllPixels().getMax())
ps = ccd.getPixelSize()
#
# Test mapping pixel <--> mm. Use a pixel at the middle of the top of the CCD
#
pix = afwGeom.Point2D(99.5, 203.5) # wrt bottom left
pos = cameraGeom.FpPoint(0.00, 1.02) # pixel center wrt CCD center
posll = cameraGeom.FpPoint(0.00, 1.02) # llc of pixel wrt CCD center
#
# Map pix into untrimmed coordinates
#
amp = ccd.findAmp(afwGeom.Point2I(int(pix[0]), int(pix[1])))
corrI = amp.getDataSec(False).getMin() - amp.getDataSec(True).getMin()
corr = afwGeom.Extent2D(corrI.getX(), corrI.getY())
pix += corr
self.assertEqual(amp.getDiskCoordSys(), cameraGeom.Amp.AMP)
self.assertEqual(ccd.getPixelFromPosition(pos) + corr, pix)
#
# Trim the CCD and try again
#
trimmedImage = trimCcd(ccd)
if display:
ds9.mtv(trimmedImage, title='Trimmed')
cameraGeomUtils.showCcd(ccd, trimmedImage)
ds9.incrDefaultFrame()
a = ccd.findAmp(cameraGeom.Id("ID%d" % ccdInfo["ampIdMin"]))
self.assertEqual(a.getDataSec(), afwGeom.Box2I(afwGeom.Point2I(0, 0),
afwGeom.Extent2I(ccdInfo["ampWidth"], ccdInfo["ampHeight"])))
self.assertEqual(ccd.getSize().getMm()[0], ccdInfo["pixelSize"]*ccdInfo["trimmedWidth"])
self.assertEqual(ccd.getSize().getMm()[1], ccdInfo["pixelSize"]*ccdInfo["trimmedHeight"])
#
# Test mapping pixel <--> mm
#
pix = afwGeom.Point2D(99.5, 203.5) # wrt bottom left
pos = cameraGeom.FpPoint(0.00, 1.02) # pixel center wrt CCD center
posll = cameraGeom.FpPoint(0.00, 1.02) # llc of pixel wrt CCD center
self.assertEqual(ccd.getPixelFromPosition(pos), pix)
self.assertEqual(ccd.getPositionFromPixel(pix).getMm(), posll.getMm())
