def testExp(self):
# Define the algorithm folder
algoFolderPath = os.path.join(self.modulePath, "configData", "cwfs",
"algo")
# Define the algorithm being used: "exp" or "fft"
useAlgorithm = "exp"
# Define the algorithm to be used.
algo = Algorithm(algoFolderPath)
algo.config(useAlgorithm, self.inst, debugLevel=3)
algo.setDebugLevel(0)
self.assertEqual(algo.debugLevel, 0)
# Test functions: itr0() and nextItr()
algo.itr0(self.inst, self.I1, self.I2, self.opticalModel)
tmp1 = algo.zer4UpNm
algo.nextItr(self.inst, self.I1, self.I2, self.opticalModel, nItr=2)
algo.itr0(self.inst, self.I1, self.I2, self.opticalModel)
tmp2 = algo.zer4UpNm
difference = np.sum(np.abs(tmp1 - tmp2))
self.assertEqual(difference, 0)
# Run it
algo.runIt(self.inst, self.I1, self.I2, self.opticalModel, tol=1e-3)
# Check the value
Zk = algo.zer4UpNm
self.assertEqual(int(Zk[7]), -192)
# Reset and check the calculation again
fieldXY = [self.I1.fieldX, self.I1.fieldY]
self.I1.setImg(fieldXY, image=self.I1.image0, atype=self.I1.atype)
self.I2.setImg(fieldXY, image=self.I2.image0, atype=self.I2.atype)
algo.reset()
algo.runIt(self.inst, self.I1, self.I2, self.opticalModel, tol=1e-3)
Zk = algo.zer4UpNm
self.assertEqual(int(Zk[7]), -192)
class TestAlgorithm(unittest.TestCase):
"""Test the Algorithm class."""
def setUp(self):
# Get the path of module
self.modulePath = getModulePath()
# Define the image folder and image names
# Image data -- Don't know the final image format.
# It is noted that image.readFile inuts is based on the txt file
imageFolderPath = os.path.join(self.modulePath, "tests", "testData",
"testImages", "LSST_NE_SN25")
intra_image_name = "z11_0.25_intra.txt"
extra_image_name = "z11_0.25_extra.txt"
# Define fieldXY: [1.185, 1.185] or [0, 0]
# This is the position of donut on the focal plane in degree
fieldXY = [1.185, 1.185]
# Define the optical model: "paraxial", "onAxis", "offAxis"
self.opticalModel = "offAxis"
# Image files Path
intra_image_file = os.path.join(imageFolderPath, intra_image_name)
extra_image_file = os.path.join(imageFolderPath, extra_image_name)
# Theree is the difference between intra and extra images
# I1: intra_focal images, I2: extra_focal Images
self.I1 = CompensableImage()
self.I2 = CompensableImage()
self.I1.setImg(fieldXY, DefocalType.Intra, imageFile=intra_image_file)
self.I2.setImg(fieldXY, DefocalType.Extra, imageFile=extra_image_file)
# Set up the instrument
cwfsConfigDir = os.path.join(getConfigDir(), "cwfs")
instDir = os.path.join(cwfsConfigDir, "instData")
self.inst = Instrument(instDir)
self.inst.config(CamType.LsstCam,
self.I1.getImgSizeInPix(),
announcedDefocalDisInMm=1.0)
# Set up the algorithm
algoDir = os.path.join(cwfsConfigDir, "algo")
self.algoExp = Algorithm(algoDir)
self.algoExp.config("exp", self.inst)
self.algoFft = Algorithm(algoDir)
self.algoFft.config("fft", self.inst)
def testGetDebugLevel(self):
self.assertEqual(self.algoExp.getDebugLevel(), 0)
def testSetDebugLevel(self):
self.algoExp.config("exp", self.inst, debugLevel=3)
self.assertEqual(self.algoExp.getDebugLevel(), 3)
self.algoExp.setDebugLevel(0)
self.assertEqual(self.algoExp.getDebugLevel(), 0)
def testGetZer4UpInNm(self):
zer4UpNm = self.algoExp.getZer4UpInNm()
self.assertTrue(isinstance(zer4UpNm, np.ndarray))
def testGetPoissonSolverName(self):
self.assertEqual(self.algoExp.getPoissonSolverName(), "exp")
self.assertEqual(self.algoFft.getPoissonSolverName(), "fft")
def testGetNumOfZernikes(self):
self.assertEqual(self.algoExp.getNumOfZernikes(), 22)
self.assertEqual(self.algoFft.getNumOfZernikes(), 22)
def testGetZernikeTerms(self):
zTerms = self.algoExp.getZernikeTerms()
self.assertTrue(type(zTerms[0]), int)
self.assertEqual(len(zTerms), self.algoExp.getNumOfZernikes())
self.assertEqual(zTerms[1], 1)
self.assertEqual(zTerms[-1], self.algoExp.getNumOfZernikes() - 1)
zTerms = self.algoFft.getZernikeTerms()
self.assertTrue(type(zTerms[0]), int)
self.assertEqual(len(zTerms), self.algoExp.getNumOfZernikes())
def testGetObsOfZernikes(self):
self.assertEqual(self.algoExp.getObsOfZernikes(),
self.inst.getObscuration())
self.assertEqual(self.algoFft.getObsOfZernikes(),
self.inst.getObscuration())
def testGetNumOfOuterItr(self):
self.assertEqual(self.algoExp.getNumOfOuterItr(), 14)
self.assertEqual(self.algoFft.getNumOfOuterItr(), 14)
def testGetNumOfInnerItr(self):
self.assertEqual(self.algoFft.getNumOfInnerItr(), 6)
def testGetFeedbackGain(self):
self.assertEqual(self.algoExp.getFeedbackGain(), 0.6)
self.assertEqual(self.algoFft.getFeedbackGain(), 0.6)
def testGetOffAxisPolyOrder(self):
self.assertEqual(self.algoExp.getOffAxisPolyOrder(), 10)
self.assertEqual(self.algoFft.getOffAxisPolyOrder(), 10)
def testGetCompensatorMode(self):
self.assertEqual(self.algoExp.getCompensatorMode(), "zer")
self.assertEqual(self.algoFft.getCompensatorMode(), "zer")
def testGetCompSequence(self):
compSequence = self.algoExp.getCompSequence()
self.assertTrue(isinstance(compSequence, np.ndarray))
self.assertEqual(compSequence.dtype, int)
self.assertEqual(len(compSequence), self.algoExp.getNumOfOuterItr())
self.assertEqual(compSequence[0], 4)
self.assertEqual(compSequence[-1], 22)
compSequence = self.algoFft.getCompSequence()
self.assertEqual(len(compSequence), self.algoFft.getNumOfOuterItr())
def testGetBoundaryThickness(self):
self.assertEqual(self.algoExp.getBoundaryThickness(), 8)
self.assertEqual(self.algoFft.getBoundaryThickness(), 1)
def testGetFftDimension(self):
self.assertEqual(self.algoFft.getFftDimension(), 128)
def testGetSignalClipSequence(self):
sumclipSequence = self.algoFft.getSignalClipSequence()
self.assertTrue(isinstance(sumclipSequence, np.ndarray))
self.assertEqual(len(sumclipSequence),
self.algoExp.getNumOfOuterItr() + 1)
self.assertEqual(sumclipSequence[0], 0.33)
self.assertEqual(sumclipSequence[-1], 0.51)
def testGetMaskScalingFactor(self):
self.assertAlmostEqual(self.algoExp.getMaskScalingFactor(),
1.0939,
places=4)
self.assertAlmostEqual(self.algoFft.getMaskScalingFactor(),
1.0939,
places=4)
def testGetWavefrontMapEstiInIter0(self):
self.assertRaises(RuntimeError, self.algoExp.getWavefrontMapEsti)
def testGetWavefrontMapEstiAndResidual(self):
self.algoExp.runIt(self.I1, self.I2, self.opticalModel, tol=1e-3)
wavefrontMapEsti = self.algoExp.getWavefrontMapEsti()
wavefrontMapEsti[np.isnan(wavefrontMapEsti)] = 0
self.assertGreater(np.sum(np.abs(wavefrontMapEsti)), 4.8e-4)
wavefrontMapResidual = self.algoExp.getWavefrontMapResidual()
wavefrontMapResidual[np.isnan(wavefrontMapResidual)] = 0
self.assertLess(np.sum(np.abs(wavefrontMapResidual)), 2.5e-6)
def testItr0(self):
self.algoExp.itr0(self.I1, self.I2, self.opticalModel)
zer4UpNm = self.algoExp.getZer4UpInNm()
self.assertEqual(
np.sum(np.abs(np.rint(zer4UpNm) - self._getAnsItr0())), 0)
def _getAnsItr0(self):
return [
31,
-69,
-21,
84,
44,
-53,
48,
-146,
6,
10,
13,
-5,
1,
-12,
-8,
7,
0,
-6,
11,
]
def testNextItrWithOneIter(self):
self.algoExp.nextItr(self.I1, self.I2, self.opticalModel, nItr=1)
zer4UpNm = self.algoExp.getZer4UpInNm()
self.assertEqual(
np.sum(np.abs(np.rint(zer4UpNm) - self._getAnsItr0())), 0)
def testNextItrWithTwoIter(self):
self.algoExp.nextItr(self.I1, self.I2, self.opticalModel, nItr=2)
zer4UpNm = self.algoExp.getZer4UpInNm()
ansRint = [
40,
-80,
-18,
92,
44.0,
-52,
54,
-146,
5,
10,
15,
-3,
-0,
-12,
-8,
7,
1,
-3,
12,
]
self.assertEqual(np.sum(np.abs(np.rint(zer4UpNm) - ansRint)), 0)
def testIter0AndNextIterToCheckReset(self):
self.algoExp.itr0(self.I1, self.I2, self.opticalModel)
tmp1 = self.algoExp.getZer4UpInNm()
self.algoExp.nextItr(self.I1, self.I2, self.opticalModel, nItr=2)
# itr0() should reset the images and ignore the effect from nextItr()
self.algoExp.itr0(self.I1, self.I2, self.opticalModel)
tmp2 = self.algoExp.getZer4UpInNm()
difference = np.sum(np.abs(tmp1 - tmp2))
self.assertEqual(difference, 0)
def testRunItOfExp(self):
self.algoExp.runIt(self.I1, self.I2, self.opticalModel, tol=1e-3)
# Check the value
zk = self.algoExp.getZer4UpInNm()
self.assertEqual(int(zk[7]), -192)
def testResetAfterFullCalc(self):
self.algoExp.runIt(self.I1, self.I2, self.opticalModel, tol=1e-3)
# Reset and check the calculation again
fieldXY = [self.I1.fieldX, self.I1.fieldY]
self.I1.setImg(fieldXY,
self.I1.getDefocalType(),
image=self.I1.getImgInit())
self.I2.setImg(fieldXY,
self.I2.getDefocalType(),
image=self.I2.getImgInit())
self.algoExp.reset()
self.algoExp.runIt(self.I1, self.I2, self.opticalModel, tol=1e-3)
zk = self.algoExp.getZer4UpInNm()
self.assertEqual(int(zk[7]), -192)
def testRunItOfFft(self):
self.algoFft.runIt(self.I1, self.I2, self.opticalModel, tol=1e-3)
zk = self.algoFft.getZer4UpInNm()
self.assertEqual(int(zk[7]), -192)
class TestAlgorithm(unittest.TestCase):
"""Test the Algorithm class."""
def setUp(self):
# Get the path of module
self.modulePath = getModulePath()
# Define the image folder and image names
# Image data -- Don't know the final image format.
# It is noted that image.readFile inuts is based on the txt file
imageFolderPath = os.path.join(self.modulePath, "tests", "testData",
"testImages", "LSST_NE_SN25")
intra_image_name = "z11_0.25_intra.txt"
extra_image_name = "z11_0.25_extra.txt"
# Define fieldXY: [1.185, 1.185] or [0, 0]
# This is the position of donut on the focal plane in degree
fieldXY = [1.185, 1.185]
# Define the optical model: "paraxial", "onAxis", "offAxis"
self.opticalModel = "offAxis"
# Image files Path
intra_image_file = os.path.join(imageFolderPath, intra_image_name)
extra_image_file = os.path.join(imageFolderPath, extra_image_name)
# Theree is the difference between intra and extra images
# I1: intra_focal images, I2: extra_focal Images
self.I1 = CompensableImage()
self.I2 = CompensableImage()
self.I1.setImg(fieldXY, DefocalType.Intra, imageFile=intra_image_file)
self.I2.setImg(fieldXY, DefocalType.Extra, imageFile=extra_image_file)
# Set up the instrument
cwfsConfigDir = os.path.join(getConfigDir(), "cwfs")
instDir = os.path.join(cwfsConfigDir, "instData")
self.inst = Instrument(instDir)
self.inst.config(CamType.LsstCam, self.I1.getImgSizeInPix(),
announcedDefocalDisInMm=1.0)
# Set up the algorithm
algoDir = os.path.join(cwfsConfigDir, "algo")
self.algoExp = Algorithm(algoDir)
self.algoExp.config("exp", self.inst)
self.algoFft = Algorithm(algoDir)
self.algoFft.config("fft", self.inst)
def testGetDebugLevel(self):
self.assertEqual(self.algoExp.getDebugLevel(), 0)
def testSetDebugLevel(self):
self.algoExp.config("exp", self.inst, debugLevel=3)
self.assertEqual(self.algoExp.getDebugLevel(), 3)
self.algoExp.setDebugLevel(0)
self.assertEqual(self.algoExp.getDebugLevel(), 0)
def testGetZer4UpInNm(self):
zer4UpNm = self.algoExp.getZer4UpInNm()
self.assertTrue(isinstance(zer4UpNm, np.ndarray))
def testGetPoissonSolverName(self):
self.assertEqual(self.algoExp.getPoissonSolverName(), "exp")
self.assertEqual(self.algoFft.getPoissonSolverName(), "fft")
def testGetNumOfZernikes(self):
self.assertEqual(self.algoExp.getNumOfZernikes(), 22)
self.assertEqual(self.algoFft.getNumOfZernikes(), 22)
def testGetZernikeTerms(self):
zTerms = self.algoExp.getZernikeTerms()
self.assertTrue(zTerms.dtype, int)
self.assertEqual(len(zTerms), self.algoExp.getNumOfZernikes())
self.assertEqual(zTerms[0], 1)
self.assertEqual(zTerms[-1], self.algoExp.getNumOfZernikes())
zTerms = self.algoFft.getZernikeTerms()
self.assertTrue(zTerms.dtype, int)
self.assertEqual(len(zTerms), self.algoExp.getNumOfZernikes())
def testGetObsOfZernikes(self):
self.assertEqual(self.algoExp.getObsOfZernikes(),
self.inst.getObscuration())
self.assertEqual(self.algoFft.getObsOfZernikes(),
self.inst.getObscuration())
def testGetNumOfOuterItr(self):
self.assertEqual(self.algoExp.getNumOfOuterItr(), 14)
self.assertEqual(self.algoFft.getNumOfOuterItr(), 14)
def testGetNumOfInnerItr(self):
self.assertEqual(self.algoFft.getNumOfInnerItr(), 6)
def testGetFeedbackGain(self):
self.assertEqual(self.algoExp.getFeedbackGain(), 0.6)
self.assertEqual(self.algoFft.getFeedbackGain(), 0.6)
def testGetOffAxisPolyOrder(self):
self.assertEqual(self.algoExp.getOffAxisPolyOrder(), 10)
self.assertEqual(self.algoFft.getOffAxisPolyOrder(), 10)
def testGetCompensatorMode(self):
self.assertEqual(self.algoExp.getCompensatorMode(), "zer")
self.assertEqual(self.algoFft.getCompensatorMode(), "zer")
def testGetCompSequence(self):
compSequence = self.algoExp.getCompSequence()
self.assertTrue(isinstance(compSequence, np.ndarray))
self.assertEqual(compSequence.dtype, int)
self.assertEqual(len(compSequence), self.algoExp.getNumOfOuterItr())
self.assertEqual(compSequence[0], 4)
self.assertEqual(compSequence[-1], 22)
compSequence = self.algoFft.getCompSequence()
self.assertEqual(len(compSequence), self.algoFft.getNumOfOuterItr())
def testGetBoundaryThickness(self):
self.assertEqual(self.algoExp.getBoundaryThickness(), 8)
self.assertEqual(self.algoFft.getBoundaryThickness(), 1)
def testGetFftDimension(self):
self.assertEqual(self.algoFft.getFftDimension(), 128)
def testGetSignalClipSequence(self):
sumclipSequence = self.algoFft.getSignalClipSequence()
self.assertTrue(isinstance(sumclipSequence, np.ndarray))
self.assertEqual(len(sumclipSequence), self.algoExp.getNumOfOuterItr()+1)
self.assertEqual(sumclipSequence[0], 0.33)
self.assertEqual(sumclipSequence[-1], 0.51)
def testGetMaskScalingFactor(self):
self.assertAlmostEqual(self.algoExp.getMaskScalingFactor(), 1.0939,
places=4)
self.assertAlmostEqual(self.algoFft.getMaskScalingFactor(), 1.0939,
places=4)
def testRunItOfExp(self):
# Test functions: itr0() and nextItr()
self.algoExp.itr0(self.I1, self.I2, self.opticalModel)
tmp1 = self.algoExp.getZer4UpInNm()
self.algoExp.nextItr(self.I1, self.I2, self.opticalModel, nItr=2)
self.algoExp.itr0(self.I1, self.I2, self.opticalModel)
tmp2 = self.algoExp.getZer4UpInNm()
difference = np.sum(np.abs(tmp1-tmp2))
self.assertEqual(difference, 0)
# Run it
self.algoExp.runIt(self.I1, self.I2, self.opticalModel, tol=1e-3)
# Check the value
Zk = self.algoExp.getZer4UpInNm()
self.assertEqual(int(Zk[7]), -192)
# Reset and check the calculation again
fieldXY = [self.I1.fieldX, self.I1.fieldY]
self.I1.setImg(fieldXY, self.I1.getDefocalType(), image=self.I1.getImgInit())
self.I2.setImg(fieldXY, self.I2.getDefocalType(), image=self.I2.getImgInit())
self.algoExp.reset()
self.algoExp.runIt(self.I1, self.I2, self.opticalModel, tol=1e-3)
Zk = self.algoExp.getZer4UpInNm()
self.assertEqual(int(Zk[7]), -192)
def testRunItOfFft(self):
self.algoFft.runIt(self.I1, self.I2, self.opticalModel, tol=1e-3)
zk = self.algoFft.getZer4UpInNm()
self.assertEqual(int(zk[7]), -192)