def trace_wavefront(self, fld, wvl, foc, num_rays=32):
def wave(p, ray_pkg, fld, wvl, foc):
x = p[0]
y = p[1]
if ray_pkg is not None:
fod = self.opt_model.optical_spec.parax_data.fod
opd = analyses.wave_abr_full_calc(fod, fld, wvl, foc, ray_pkg,
fld.chief_ray,
fld.ref_sphere)
opd = opd/self.opt_model.nm_to_sys_units(wvl)
else:
opd = 0.0
return np.array([x, y, opd])
rs_pkg, cr_pkg = trace.setup_pupil_coords(self.opt_model,
fld, wvl, foc)
fld.chief_ray = cr_pkg
fld.ref_sphere = rs_pkg
grid_start = np.array([-1., -1.])
grid_stop = np.array([1., 1.])
grid_def = (grid_start, grid_stop, num_rays)
grid = trace.trace_grid(self.opt_model, grid_def, fld, wvl, foc,
img_filter=lambda p, ray_pkg:
wave(p, ray_pkg, fld, wvl, foc), form='grid')
return grid
def trace_grid(self, fct, fi, wl=None, num_rays=21, form='grid',
append_if_none=True):
""" fct is applied to the raw grid and returned as a grid """
osp = self.opt_model.optical_spec
wvls = osp.spectral_region
wvl = self.central_wavelength()
wv_list = wvls.wavelengths if wl is None else [wvl]
fld = osp.field_of_view.fields[fi]
foc = osp.defocus.get_focus()
rs_pkg, cr_pkg = trace.setup_pupil_coords(self.opt_model,
fld, wvl, foc)
fld.chief_ray = cr_pkg
fld.ref_sphere = rs_pkg
grids = []
grid_start = np.array([-1., -1.])
grid_stop = np.array([1., 1.])
grid_def = [grid_start, grid_stop, num_rays]
for wi, wvl in enumerate(wv_list):
grid = trace.trace_grid(self.opt_model, grid_def, fld, wvl, foc,
form=form, append_if_none=append_if_none,
img_filter=lambda p, ray_pkg:
fct(p, wi, ray_pkg, fld, wvl, foc))
grids.append(grid)
rc = wvls.render_colors
return grids, rc