def test_normalisations_real(self):
log = logging.getLogger('TestZern.test_normalisations_real')
n_alpha = 6
L, K = 400, 357
# polar grid
pol = RZern(n_alpha)
fitAlpha = FitZern(pol, L, K)
t1 = time()
pol.make_pol_grid(fitAlpha.rho_j, fitAlpha.theta_i)
t2 = time()
log.debug('make pol grid {:.6f}'.format(t2 - t1))
# cartesian grid
cart = RZern(n_alpha)
dd = np.linspace(-1.0, 1.0, max(L, K))
xx, yy = np.meshgrid(dd, dd)
t1 = time()
cart.make_cart_grid(xx, yy)
t2 = time()
log.debug('make cart grid {:.6f}'.format(t2 - t1))
smap = np.isfinite(cart.eval_grid(np.zeros(cart.nk)))
scale = (1.0 / np.sum(smap))
log.debug('')
log.debug('{} modes, {} x {} grid'.format(n_alpha, L, K))
for i in range(pol.nk):
a = np.zeros(pol.nk)
a[i] = 1.0
Phi_a = cart.eval_grid(a)
for j in range(pol.nk):
b = np.zeros(pol.nk)
b[j] = 1.0
Phi_b = cart.eval_grid(b)
ip = scale * np.sum(Phi_a[smap] * Phi_b[smap])
if i == j:
eip = 1.0
else:
eip = 0.0
iperr = abs(ip - eip)
log.debug('<{:02},{:02}> = {:+e} {:+e}'.format(
i + 1, j + 1, ip, iperr))
self.assertTrue(iperr < self.max_ip_err)
def test_normalisations_real(self):
log = logging.getLogger('TestZern.test_normalisations_real')
n_alpha = 6
L, K = 400, 357
# polar grid
pol = RZern(n_alpha)
fitAlpha = FitZern(pol, L, K)
t1 = time()
pol.make_pol_grid(fitAlpha.rho_j, fitAlpha.theta_i)
t2 = time()
log.debug('make pol grid {:.6f}'.format(t2 - t1))
# cartesian grid
cart = RZern(n_alpha)
dd = np.linspace(-1.0, 1.0, max(L, K))
xx, yy = np.meshgrid(dd, dd)
t1 = time()
cart.make_cart_grid(xx, yy)
t2 = time()
log.debug('make cart grid {:.6f}'.format(t2 - t1))
smap = np.isfinite(cart.eval_grid(np.zeros(cart.nk)))
scale = (1.0/np.sum(smap))
log.debug('')
log.debug('{} modes, {} x {} grid'.format(n_alpha, L, K))
for i in range(pol.nk):
a = np.zeros(pol.nk)
a[i] = 1.0
Phi_a = cart.eval_grid(a)
for j in range(pol.nk):
b = np.zeros(pol.nk)
b[j] = 1.0
Phi_b = cart.eval_grid(b)
ip = scale*np.sum(Phi_a[smap]*Phi_b[smap])
if i == j:
eip = 1.0
else:
eip = 0.0
iperr = abs(ip - eip)
log.debug('<{:02},{:02}> = {:+e} {:+e}'.format(
i + 1, j + 1, ip, iperr))
self.assertTrue(iperr < self.max_ip_err)
def __init__(self, unparsed):
super().__init__()
args = self.do_cmdline(unparsed)
# plot objects
phaseplot = PhasePlot(n=args.n_alpha) # to plot beta and the PSF
betaplot = BetaPlot(args) # to plot the phase
# complex-valued Zernike polynomials for the GPF
ip = FitZern(CZern(args.n_beta), args.fit_L, args.fit_K)
# real-valued Zernike polynomials for the phase
phase_pol = RZern(args.n_alpha)
phase_pol.make_pol_grid(ip.rho_j, ip.theta_i) # make a polar grid
# real-valued Zernike coefficients
alpha = np.zeros(phase_pol.nk)
# set the alpha coefficients randomly
if args.random:
alpha1 = normal(size=alpha.size - 1)
alpha1 = (args.rms / norm(alpha1)) * alpha1
alpha[1:] = alpha1
del alpha1
self.rms = args.rms
self.alpha = alpha
self.phase_pol = phase_pol
self.ip = ip
self.betaplot = betaplot
self.phaseplot = phaseplot
# fit beta coefficients from alpha coefficients
self.alpha2beta()
# make gui
self.make_gui()
def __init__(self, unparsed):
super().__init__()
args = self.do_cmdline(unparsed)
# plot objects
phaseplot = PhasePlot(n=args.n_alpha) # to plot beta and the PSF
betaplot = BetaPlot(args) # to plot the phase
# complex-valued Zernike polynomials for the GPF
ip = FitZern(CZern(args.n_beta), args.fit_L, args.fit_K)
# real-valued Zernike polynomials for the phase
phase_pol = RZern(args.n_alpha)
phase_pol.make_pol_grid(ip.rho_j, ip.theta_i) # make a polar grid
# real-valued Zernike coefficients
alpha = np.zeros(phase_pol.nk)
# set the alpha coefficients randomly
if args.random:
alpha1 = normal(size=alpha.size-1)
alpha1 = (args.rms/norm(alpha1))*alpha1
alpha[1:] = alpha1
del alpha1
self.rms = args.rms
self.alpha = alpha
self.phase_pol = phase_pol
self.ip = ip
self.betaplot = betaplot
self.phaseplot = phaseplot
# fit beta coefficients from alpha coefficients
self.alpha2beta()
# make gui
self.make_gui()
help='Make a random alpha aberration.')
parser.add_argument(
'--fit-L', type=int, default=95, metavar='L',
help='Grid size for the inner products.')
parser.add_argument(
'--fit-K', type=int, default=105, metavar='K',
help='Grid size for the inner products.')
args = parser.parse_args()
# complex-valued Zernike polynomials for the GPF
ip = FitZern(CZern(args.n_beta), args.fit_L, args.fit_K)
# real-valued Zernike polynomials for the phase
phase_pol = RZern(args.n_alpha)
phase_pol.make_pol_grid(ip.rho_j, ip.theta_i) # make a polar grid
# real-valued Zernike coefficients
alpha = np.zeros(phase_pol.nk)
# nm to linear index conversion
nmlist = list(zip(phase_pol.ntab, phase_pol.mtab))
# set an alpha coefficient using the (n, m) indeces
if args.nm[0] != -1 and args.nm[0] != -1:
try:
k = nmlist.index(tuple(args.nm))
alpha[k] = args.rms
except:
print('Cannot find indeces ' + str(args.nm))
print('Possible [n, m] indeces are:')
help='Make a random alpha aberration.')
parser.add_argument(
'--fit-L', type=int, default=95, metavar='L',
help='Grid size for the inner products.')
parser.add_argument(
'--fit-K', type=int, default=105, metavar='K',
help='Grid size for the inner products.')
args = parser.parse_args()
# complex-valued Zernike polynomials for the GPF
ip = FitZern(CZern(args.n_beta), args.fit_L, args.fit_K)
# real-valued Zernike polynomials for the phase
phase_pol = RZern(args.n_alpha)
phase_pol.make_pol_grid(ip.rho_j, ip.theta_i) # make a polar grid
# real-valued Zernike coefficients
alpha = np.zeros(phase_pol.nk)
# nm to linear index conversion
nmlist = list(zip(phase_pol.ntab, phase_pol.mtab))
# set an alpha coefficient using the (n, m) indeces
if args.nm[0] != -1 and args.nm[0] != -1:
try:
k = nmlist.index(tuple(args.nm))
alpha[k] = args.rms
except BaseException:
print('Cannot find indeces ' + str(args.nm))
print('Possible [n, m] indeces are:')
class Config:
def __init__(self):
pass
def make(
self, wavelength, aperture_radius, focal_length, pixel_size,
image_width, image_height,
n_alpha, n_beta,
fit_L, fit_K,
focus_positions):
self.wavelength = wavelength
self.aperture_radius = aperture_radius
self.focal_length = focal_length
self.image_width = image_width
self.image_height = image_height
self.pixel_size = pixel_size
self.n_alpha, self.n_beta = n_alpha, n_beta
self.fit_L, self.fit_K = fit_L, fit_K
self.focus_positions = np.array(focus_positions)
fu = enz.get_field_unit(
wavelength,
aperture_radius, focal_length)
def make_space(w, p, fu):
if w % 2 == 0:
return np.linspace(-(w/2 - 0.5), w/2 - 0.5, w)*p/fu
else:
return np.linspace(-(w - 1)/2, (w - 1)/2, w)*p/fu
# image side space
xspace = make_space(image_width, pixel_size, fu)
yspace = make_space(image_height, pixel_size, fu)
self.xspace, self.yspace = xspace, yspace
# phase
self.phase_grid = RZern(n_alpha)
self.phase_fit = FitZern(self.phase_grid, self.fit_L, self.fit_K)
print('phase: n_alpha = {}, N_alpha = {}'.format(
self.n_alpha, self.phase_grid.nk))
# complex psf
self.cpsf = CPsf(n_beta)
self.gpf_fit = FitZern(self.cpsf.czern, self.fit_L, self.fit_K)
print('cpsf: n_beta = {}, N_beta = {}, N_f = {}'.format(
n_beta, self.cpsf.czern.nk, self.focus_positions.size))
# make phase polar grid
t1 = time.time()
self.phase_grid.make_pol_grid(
self.phase_fit.rho_j,
self.phase_fit.theta_i)
t2 = time.time()
print('make phase pol grid {:.6f}'.format(t2 - t1))
# make gpf polar grid (Zernike approximation)
t1 = time.time()
self.cpsf.czern.make_pol_grid(
self.phase_fit.rho_j,
self.phase_fit.theta_i)
t2 = time.time()
print('make gpf pol grid {:.6f}'.format(t2 - t1))
# make cpsf cart grid
t1 = time.time()
self.cpsf.make_cart_grid(
x_sp=xspace, y_sp=yspace, f_sp=focus_positions)
t2 = time.time()
print('make cpsf cart grid {:.6f}'.format(t2 - t1))
def save(self, filename, prepend=None, libver='latest'):
"""Save object into an HDF5 file."""
f = h5py.File(filename, 'w', libver=libver)
self.save_h5py(f, prepend=prepend)
f.close()
print('saved <{}>'.format(filename))
def save_h5py(self, f, prepend=None):
"""Dump object contents into an opened HDF5 file object."""
prefix = 'Config/'
if prepend is not None:
prefix = prepend + prefix
params = {
'chunks': True,
'shuffle': True,
'fletcher32': True,
'compression': 'gzip',
'compression_opts': 9,
}
f.create_dataset(
prefix + 'wavelength',
data=np.array([self.wavelength], dtype=np.float))
f.create_dataset(
prefix + 'aperture_radius',
data=np.array([self.aperture_radius], dtype=np.float))
f.create_dataset(
prefix + 'focal_length',
data=np.array([self.focal_length], dtype=np.float))
f.create_dataset(
prefix + 'image_width',
data=np.array([self.image_width], dtype=np.int))
f.create_dataset(
prefix + 'image_height',
data=np.array([self.image_height], dtype=np.int))
f.create_dataset(
prefix + 'pixel_size',
data=np.array([self.pixel_size], dtype=np.float))
f.create_dataset(
prefix + 'n_alpha',
data=np.array([self.n_alpha], dtype=np.int))
f.create_dataset(
prefix + 'n_beta',
data=np.array([self.n_beta], dtype=np.int))
f.create_dataset(
prefix + 'fit_L',
data=np.array([self.fit_L], dtype=np.int))
f.create_dataset(
prefix + 'fit_K',
data=np.array([self.fit_K], dtype=np.int))
params['data'] = self.focus_positions
f.create_dataset(prefix + 'focus_positions', **params)
params['data'] = self.xspace
f.create_dataset(prefix + 'xspace', **params)
params['data'] = self.yspace
f.create_dataset(prefix + 'yspace', **params)
self.phase_fit.save_h5py(f, prepend=prefix+'phase_fit/')
self.cpsf.save_h5py(f, prepend=prefix+'cpsf/')
self.gpf_fit.save_h5py(f, prepend=prefix+'gpf_fit/')
@classmethod
def load(cls, filename, prepend=None):
"""Load object from an HDF5 file."""
f = h5py.File(filename, 'r')
print('load <{}>'.format(filename))
z = cls.load_h5py(f, prepend=prepend)
f.close()
return z
@classmethod
def load_h5py(cls, f, prepend=None):
"""Load object contents from an opened HDF5 file object."""
sc = cls()
prefix = 'Config/'
if prepend is not None:
prefix = prepend + prefix
sc.wavelength = float(f[prefix + 'wavelength'].value[0])
sc.aperture_radius = float(f[prefix + 'aperture_radius'].value[0])
sc.focal_length = float(f[prefix + 'focal_length'].value[0])
sc.image_width = int(f[prefix + 'image_width'].value[0])
sc.image_height = int(f[prefix + 'image_height'].value[0])
sc.pixel_size = float(f[prefix + 'pixel_size'].value[0])
sc.n_alpha = int(f[prefix + 'n_alpha'].value[0])
sc.n_beta = int(f[prefix + 'n_beta'].value[0])
sc.fit_L = int(f[prefix + 'fit_L'].value[0])
sc.fit_K = int(f[prefix + 'fit_K'].value[0])
sc.focus_positions = f[prefix + 'focus_positions'].value
sc.xspace = f[prefix + 'xspace'].value
sc.yspace = f[prefix + 'yspace'].value
sc.phase_fit = FitZern.load_h5py(f, prepend=prefix+'phase_fit/')
sc.phase_grid = sc.phase_fit.z
sc.cpsf = CPsf.load_h5py(f, prepend=prefix+'cpsf/')
sc.gpf_fit = FitZern.load_h5py(f, prepend=prefix+'gpf_fit/')
sc.gpf_fit.z = sc.cpsf.czern
print('phase: n_alpha = {}, N_alpha = {}'.format(
sc.n_alpha, sc.phase_grid.nk))
print('cpsf: n_beta = {}, N_beta = {}, N_f = {}'.format(
sc.n_beta, sc.cpsf.czern.nk, sc.focus_positions.size))
return sc
class Config:
def __init__(self):
pass
def make(self, wavelength, aperture_radius, focal_length, pixel_size,
image_width, image_height, n_alpha, n_beta, fit_L, fit_K,
focus_positions):
self.wavelength = wavelength
self.aperture_radius = aperture_radius
self.focal_length = focal_length
self.image_width = image_width
self.image_height = image_height
self.pixel_size = pixel_size
self.n_alpha, self.n_beta = n_alpha, n_beta
self.fit_L, self.fit_K = fit_L, fit_K
self.focus_positions = np.array(focus_positions)
fu = enz.get_field_unit(wavelength, aperture_radius, focal_length)
def make_space(w, p, fu):
if w % 2 == 0:
return np.linspace(-(w / 2 - 0.5), w / 2 - 0.5, w) * p / fu
else:
return np.linspace(-(w - 1) / 2, (w - 1) / 2, w) * p / fu
# image side space
xspace = make_space(image_width, pixel_size, fu)
yspace = make_space(image_height, pixel_size, fu)
self.xspace, self.yspace = xspace, yspace
# phase
self.phase_grid = RZern(n_alpha)
self.phase_fit = FitZern(self.phase_grid, self.fit_L, self.fit_K)
print('phase: n_alpha = {}, N_alpha = {}'.format(
self.n_alpha, self.phase_grid.nk))
# complex psf
self.cpsf = CPsf(n_beta)
self.gpf_fit = FitZern(self.cpsf.czern, self.fit_L, self.fit_K)
print('cpsf: n_beta = {}, N_beta = {}, N_f = {}'.format(
n_beta, self.cpsf.czern.nk, self.focus_positions.size))
# make phase polar grid
t1 = time.time()
self.phase_grid.make_pol_grid(self.phase_fit.rho_j,
self.phase_fit.theta_i)
t2 = time.time()
print('make phase pol grid {:.6f}'.format(t2 - t1))
# make gpf polar grid (Zernike approximation)
t1 = time.time()
self.cpsf.czern.make_pol_grid(self.phase_fit.rho_j,
self.phase_fit.theta_i)
t2 = time.time()
print('make gpf pol grid {:.6f}'.format(t2 - t1))
# make cpsf cart grid
t1 = time.time()
self.cpsf.make_cart_grid(x_sp=xspace,
y_sp=yspace,
f_sp=focus_positions)
t2 = time.time()
print('make cpsf cart grid {:.6f}'.format(t2 - t1))
def save(self, filename, prepend=None, libver='latest'):
"""Save object into an HDF5 file."""
f = h5py.File(filename, 'w', libver=libver)
self.save_h5py(f, prepend=prepend)
f.close()
print('saved <{}>'.format(filename))
def save_h5py(self, f, prepend=None):
"""Dump object contents into an opened HDF5 file object."""
prefix = 'Config/'
if prepend is not None:
prefix = prepend + prefix
params = {
'chunks': True,
'shuffle': True,
'fletcher32': True,
'compression': 'gzip',
'compression_opts': 9,
}
f.create_dataset(prefix + 'wavelength',
data=np.array([self.wavelength], dtype=np.float))
f.create_dataset(prefix + 'aperture_radius',
data=np.array([self.aperture_radius], dtype=np.float))
f.create_dataset(prefix + 'focal_length',
data=np.array([self.focal_length], dtype=np.float))
f.create_dataset(prefix + 'image_width',
data=np.array([self.image_width], dtype=np.int))
f.create_dataset(prefix + 'image_height',
data=np.array([self.image_height], dtype=np.int))
f.create_dataset(prefix + 'pixel_size',
data=np.array([self.pixel_size], dtype=np.float))
f.create_dataset(prefix + 'n_alpha',
data=np.array([self.n_alpha], dtype=np.int))
f.create_dataset(prefix + 'n_beta',
data=np.array([self.n_beta], dtype=np.int))
f.create_dataset(prefix + 'fit_L',
data=np.array([self.fit_L], dtype=np.int))
f.create_dataset(prefix + 'fit_K',
data=np.array([self.fit_K], dtype=np.int))
params['data'] = self.focus_positions
f.create_dataset(prefix + 'focus_positions', **params)
params['data'] = self.xspace
f.create_dataset(prefix + 'xspace', **params)
params['data'] = self.yspace
f.create_dataset(prefix + 'yspace', **params)
self.phase_fit.save_h5py(f, prepend=prefix + 'phase_fit/')
self.cpsf.save_h5py(f, prepend=prefix + 'cpsf/')
self.gpf_fit.save_h5py(f, prepend=prefix + 'gpf_fit/')
@classmethod
def load(cls, filename, prepend=None):
"""Load object from an HDF5 file."""
f = h5py.File(filename, 'r')
print('load <{}>'.format(filename))
z = cls.load_h5py(f, prepend=prepend)
f.close()
return z
@classmethod
def load_h5py(cls, f, prepend=None):
"""Load object contents from an opened HDF5 file object."""
sc = cls()
prefix = 'Config/'
if prepend is not None:
prefix = prepend + prefix
sc.wavelength = float(f[prefix + 'wavelength'].value[0])
sc.aperture_radius = float(f[prefix + 'aperture_radius'].value[0])
sc.focal_length = float(f[prefix + 'focal_length'].value[0])
sc.image_width = int(f[prefix + 'image_width'].value[0])
sc.image_height = int(f[prefix + 'image_height'].value[0])
sc.pixel_size = float(f[prefix + 'pixel_size'].value[0])
sc.n_alpha = int(f[prefix + 'n_alpha'].value[0])
sc.n_beta = int(f[prefix + 'n_beta'].value[0])
sc.fit_L = int(f[prefix + 'fit_L'].value[0])
sc.fit_K = int(f[prefix + 'fit_K'].value[0])
sc.focus_positions = f[prefix + 'focus_positions'].value
sc.xspace = f[prefix + 'xspace'].value
sc.yspace = f[prefix + 'yspace'].value
sc.phase_fit = FitZern.load_h5py(f, prepend=prefix + 'phase_fit/')
sc.phase_grid = sc.phase_fit.z
sc.cpsf = CPsf.load_h5py(f, prepend=prefix + 'cpsf/')
sc.gpf_fit = FitZern.load_h5py(f, prepend=prefix + 'gpf_fit/')
sc.gpf_fit.z = sc.cpsf.czern
print('phase: n_alpha = {}, N_alpha = {}'.format(
sc.n_alpha, sc.phase_grid.nk))
print('cpsf: n_beta = {}, N_beta = {}, N_f = {}'.format(
sc.n_beta, sc.cpsf.czern.nk, sc.focus_positions.size))
return sc