def testg_initCustomShape(): config = confParse.loadSoapyConfig(os.path.join(CONFIG_PATH, "sh_8x8.yaml")) config.sim.pupilSize = 64 config.sim.simSize = 66 mask = aotools.circle(config.sim.pupilSize / 2., config.sim.simSize) wfs = WFS.ShackHartmann(config, mask=mask) dm = DM.KarhunenLoeve(config, n_dm=1, wfss=[wfs], mask=mask) customShape = dm.iMatShapes # saving temporary shapes fname = os.path.dirname( os.path.abspath(__file__)) + '/tmp_CustomDmShapes.fits' fits.writeto(fname, customShape, overwrite=True) # change size to ensure it tests interpolation config.sim.pupilSize = 100 config.sim.simSize = 104 mask = aotools.circle(config.sim.pupilSize / 2., config.sim.simSize) wfs = WFS.ShackHartmann(config, mask=mask) config.dms[1].dmShapesFilename = fname dm = DM.CustomShapes(config, n_dm=1, wfss=[wfs], mask=mask) # remove temporary shapes os.remove(fname)
def testg_initKL(): config = confParse.loadSoapyConfig(os.path.join(CONFIG_PATH, "sh_8x8.yaml")) mask = aotools.circle(config.sim.pupilSize / 2., config.sim.simSize) wfs = WFS.ShackHartmann(config, mask=mask) dm = DM.KarhunenLoeve(config, n_dm=1, wfss=[wfs], mask=mask)