def teste_initPyrWfs(self):
config = confParse.Configurator("../conf/sh_8x8.py")
config.loadSimParams()
mask = aoSimLib.circle(config.sim.pupilSize/2., config.sim.simSize)
wfs = WFS.Pyramid(config.sim, config.wfss[0], config.atmos, config.lgss[0], mask)
def teste_initPyrWfs(self):
config = confParse.Configurator(os.path.join(CONFIG_PATH, "sh_8x8.py"))
config.loadSimParams()
mask = aoSimLib.circle(config.sim.pupilSize/2., config.sim.simSize)
wfs = WFS.Pyramid(config.sim, config.wfss[0], config.atmos, config.lgss[0], mask)
def testd_GradWfsFrame(self):
config = confParse.Configurator(os.path.join(CONFIG_PATH, "sh_8x8.py"))
config.loadSimParams()
mask = aoSimLib.circle(config.sim.pupilSize/2., config.sim.simSize)
wfs = WFS.Gradient(config, mask=mask)
wfs.frame(numpy.zeros((config.sim.simSize, config.sim.simSize)))
def testa_initWfs(self):
config = confParse.Configurator(os.path.join(CONFIG_PATH, "sh_8x8.py"))
config.loadSimParams()
mask = aoSimLib.circle(config.sim.pupilSize/2., config.sim.simSize)
wfs = WFS.WFS(config, mask=mask)
def test_sciInit(self):
config = confParse.Configurator(os.path.join(CONFIG_PATH, "sh_8x8.py"))
config.loadSimParams()
mask = aoSimLib.circle(config.sim.pupilSize / 2., config.sim.simSize)
sci = SCI.PSF(config, 0, mask)
def testa_initLgs(self):
config = confParse.Configurator(os.path.join(CONFIG_PATH, "sh_8x8_lgs-uplink.py"))
config.loadSimParams()
mask = aoSimLib.circle(config.sim.pupilSize/2., config.sim.simSize)
lgs = LGS.LGS(config.wfss[1], config)
def testa_initWfs(self):
config = confParse.Configurator(os.path.join(CONFIG_PATH, "sh_8x8.py"))
config.loadSimParams()
mask = aoSimLib.circle(config.sim.pupilSize / 2., config.sim.simSize)
wfs = WFS.WFS(config, mask=mask)
def testd_initPhysLgs(self):
config = confParse.Configurator(os.path.join(CONFIG_PATH, "sh_8x8_lgs-uplink.py"))
config.loadSimParams()
config.wfss[1].lgs.propagationMode = "Physical"
mask = aoSimLib.circle(config.sim.pupilSize/2., config.sim.simSize)
lgs = LGS.LGS_Physical(config.wfss[1], config)
def testd_GradWfsFrame(self):
config = confParse.Configurator(os.path.join(CONFIG_PATH, "sh_8x8.py"))
config.loadSimParams()
mask = aoSimLib.circle(config.sim.pupilSize / 2., config.sim.simSize)
wfs = WFS.Gradient(config, mask=mask)
wfs.frame(numpy.zeros((config.sim.simSize, config.sim.simSize)))
def testf_PyrWfsFrame(self):
config = confParse.Configurator("../conf/sh_8x8.py")
config.loadSimParams()
mask = aoSimLib.circle(config.sim.pupilSize/2., config.sim.simSize)
wfs = WFS.Pyramid(config.sim, config.wfss[0], config.atmos, config.lgss[0], mask)
wfs.frame(numpy.zeros((config.sim.simSize, config.sim.simSize)))
def testd_SHWfsFrame(self):
config = confParse.Configurator(os.path.join(CONFIG_PATH, "sh_8x8.py"))
config.loadSimParams()
mask = aoSimLib.circle(config.sim.pupilSize/2., config.sim.simSize)
wfs = WFS.ShackHartmann(config.sim, config.wfss[0], config.atmos, config.lgss[0], mask)
wfs.frame(numpy.zeros((config.sim.simSize, config.sim.simSize)))
def loadConfig(self):
"""
Load the Soapy config file
"""
self.config = confParse.YAML_Configurator(self.configfile)
self.config.loadSimParams()
self.mask = aoSimLib.circle(
self.config.sim.pupilSize/2., self.config.sim.simSize)
def testa_initDM(self):
config = confParse.Configurator(os.path.join(CONFIG_PATH, "sh_8x8.py"))
config.loadSimParams()
mask = aoSimLib.circle(config.sim.pupilSize/2., config.sim.simSize)
wfs = WFS.ShackHartmann(config.sim, config.wfss[0], config.atmos, config.lgss[0], mask)
dm = DM.DM(config.sim, config.dms[0], [wfs], mask)
def test_sciInit(self):
config = confParse.Configurator(os.path.join(CONFIG_PATH, "sh_8x8.py"))
config.loadSimParams()
mask = aoSimLib.circle(config.sim.pupilSize/2., config.sim.simSize)
sci = SCI.scienceCam(
config.sim, config.tel, config.atmos, config.scis[0], mask)
def testc_geoLgsPsf(self):
config = confParse.Configurator(os.path.join(CONFIG_PATH, "sh_8x8_lgs-uplink.py"))
config.loadSimParams()
mask = aoSimLib.circle(config.sim.pupilSize/2., config.sim.simSize)
config.wfss[1].lgs.propagationMode = "Geometric"
lgs = LGS.LGS_Geometric(config.wfss[1], config)
psf = lgs.getLgsPsf(
[numpy.zeros((config.sim.simSize, config.sim.simSize))])
def testa_initLgs(self):
config = confParse.Configurator(
os.path.join(CONFIG_PATH, "sh_8x8_lgs-uplink.py"))
config.loadSimParams()
mask = aoSimLib.circle(config.sim.pupilSize / 2., config.sim.simSize)
lgs = LGS.LGS(config.wfss[1], config)
def loadConfig(self):
"""
Load the Soapy config file
"""
self.config = confParse.YAML_Configurator(self.configfile)
self.config.loadSimParams()
self.mask = aoSimLib.circle(self.config.sim.pupilSize / 2.,
self.config.sim.simSize)
def testd_initPhysLgs(self):
config = confParse.Configurator(
os.path.join(CONFIG_PATH, "sh_8x8_lgs-uplink.py"))
config.loadSimParams()
config.wfss[1].lgs.propagationMode = "Physical"
mask = aoSimLib.circle(config.sim.pupilSize / 2., config.sim.simSize)
lgs = LGS.LGS_Physical(config.wfss[1], config)
def testf_initFastPiezo(self):
config = confParse.Configurator(os.path.join(CONFIG_PATH, "sh_8x8.py"))
config.loadSimParams()
mask = aoSimLib.circle(config.sim.pupilSize/2., config.sim.simSize)
wfs = WFS.ShackHartmann(config, mask=mask)
dm = DM.FastPiezo(config, wfss=[wfs], mask=mask)
def testa_initDM(self):
config = confParse.Configurator(os.path.join(CONFIG_PATH, "sh_8x8.py"))
config.loadSimParams()
mask = aoSimLib.circle(config.sim.pupilSize / 2., config.sim.simSize)
wfs = WFS.ShackHartmann(config, mask=mask)
dm = DM.DM(config, wfss=[wfs], mask=mask)
def teste_physLgsPsf(self):
config = confParse.Configurator(os.path.join(CONFIG_PATH, "sh_8x8_lgs-uplink.py"))
config.loadSimParams()
config.wfss[1].lgs.propagationMode = "Physical"
mask = aoSimLib.circle(config.sim.pupilSize/2., config.sim.simSize)
lgs = LGS.LGS_Physical(config.wfss[1], config, nOutPxls=10)
psf = lgs.getLgsPsf(
[numpy.zeros((config.sim.simSize, config.sim.simSize))])
def testc_geoLgsPsf(self):
config = confParse.Configurator(
os.path.join(CONFIG_PATH, "sh_8x8_lgs-uplink.py"))
config.loadSimParams()
mask = aoSimLib.circle(config.sim.pupilSize / 2., config.sim.simSize)
config.wfss[1].lgs.propagationMode = "Geometric"
lgs = LGS.LGS_Geometric(config.wfss[1], config)
psf = lgs.getLgsPsf(
[numpy.zeros((config.sim.simSize, config.sim.simSize))])
def teste_iMatGauss(self):
config = confParse.Configurator(os.path.join(CONFIG_PATH, "sh_8x8.py"))
config.loadSimParams()
mask = aoSimLib.circle(config.sim.pupilSize/2., config.sim.simSize)
wfs = WFS.ShackHartmann(config, mask=mask)
dm = DM.GaussStack(config, wfss=[wfs], mask=mask)
dm.makeIMat()
def test_sciStrehl(self):
config = confParse.Configurator(os.path.join(CONFIG_PATH, "sh_8x8.py"))
config.loadSimParams()
mask = aoSimLib.circle(config.sim.pupilSize / 2., config.sim.simSize)
sci = SCI.PSF(config, 0, mask)
sci.frame(numpy.ones((config.sim.simSize, config.sim.simSize)))
self.assertTrue(numpy.allclose(sci.instStrehl, 1.))
def teste_physLgsPsf(self):
config = confParse.Configurator(
os.path.join(CONFIG_PATH, "sh_8x8_lgs-uplink.py"))
config.loadSimParams()
config.wfss[1].lgs.propagationMode = "Physical"
mask = aoSimLib.circle(config.sim.pupilSize / 2., config.sim.simSize)
lgs = LGS.LGS_Physical(config.wfss[1], config, nOutPxls=10)
psf = lgs.getLgsPsf(
[numpy.zeros((config.sim.simSize, config.sim.simSize))])
def test_PhysWfs(self):
config = confParse.Configurator(os.path.join(CONFIG_PATH, "sh_8x8.py"))
config.loadSimParams()
config.wfss[0].propagationMode = "Physical"
mask = aoSimLib.circle(config.sim.pupilSize/2., config.sim.simSize)
wfs = WFS.WFS(config, mask=mask)
wfs.frame([numpy.zeros((config.sim.scrnSize,)*2)]*config.atmos.scrnNo)
def loadConfig(self):
"""
Load the Soapy config file
"""
self.config = confParse.Configurator(self.configfile)
self.config.loadSimParams()
self.config.wfss[0].pxlsPerSubap = PXLS_PER_SUBAP
self.mask = aoSimLib.circle(
self.config.sim.pupilSize/2., self.config.sim.simSize)
def test_PhysWfs(self):
config = confParse.Configurator(os.path.join(CONFIG_PATH, "sh_8x8.py"))
config.loadSimParams()
config.wfss[0].propagationMode = "Physical"
mask = aoSimLib.circle(config.sim.pupilSize / 2., config.sim.simSize)
wfs = WFS.WFS(config, mask=mask)
wfs.frame([numpy.zeros(
(config.sim.scrnSize, ) * 2)] * config.atmos.scrnNo)
def test_sciStrehl(self):
config = confParse.Configurator(os.path.join(CONFIG_PATH, "sh_8x8.py"))
config.loadSimParams()
mask = aoSimLib.circle(config.sim.pupilSize/2., config.sim.simSize)
sci = SCI.scienceCam(
config.sim, config.tel, config.atmos, config.scis[0], mask)
sci.frame(numpy.ones((config.sim.simSize, config.sim.simSize)))
self.assertTrue(numpy.allclose(sci.instStrehl, 1.))
