def xtest_find_height_profile_dimension():
from sirepo.template import srw
for dimension in (1, 2):
dat_file = pkresource.filename(
'static/dat/mirror_{}d.dat'.format(dimension), srw)
found_dimension = srw.find_height_profile_dimension(dat_file)
assert found_dimension == dimension
def _mirror_plot(model_data):
mirror = model_data['models']['mirrorReport']
dat_file = mirror['heightProfileFile']
dimension = find_height_profile_dimension(dat_file)
func_name = 'srwl_opt_setup_surf_height_{}d'.format(dimension)
add_args = [0, 1] if dimension == 1 else []
element = getattr(srwlib, func_name)(
srwlib.srwl_uti_read_data_cols(mirror['heightProfileFile'], "\t", *add_args),
_dim=mirror['orientation'],
_ang=float(mirror['grazingAngle']) / 1e3,
_amp_coef=float(mirror['heightAmplification']))
transmission_data = element.get_data(3, 3)
srwlib.srwl_uti_save_intens_ascii(
transmission_data, element.mesh, 'res_mirror.dat', 0,
['', 'Horizontal Position', 'Vertical Position', 'Optical Path Difference'], _arUnits=['', 'm', 'm', ''])
return 'res_mirror.dat'
def _mirror_plot(model_data):
mirror = model_data['models']['mirrorReport']
dat_file = mirror['heightProfileFile']
dimension = find_height_profile_dimension(dat_file)
func_name = 'srwl_opt_setup_surf_height_{}d'.format(dimension)
add_args = [0, 1] if dimension == 1 else []
element = getattr(srwlib, func_name)(
srwlib.srwl_uti_read_data_cols(mirror['heightProfileFile'], "\t",
*add_args),
_dim=mirror['orientation'],
_ang=float(mirror['grazingAngle']) / 1e3,
_amp_coef=float(mirror['heightAmplification']))
transmission_data = element.get_data(3, 3)
srwlib.srwl_uti_save_intens_ascii(
transmission_data,
element.mesh,
'res_mirror.dat',
0, [
'', 'Horizontal Position', 'Vertical Position',
'Optical Path Difference'
],
_arUnits=['', 'm', 'm', ''])
return 'res_mirror.dat'
def xtest_find_height_profile_dimension():
from sirepo.template import srw
for dimension in (1, 2):
dat_file = pkresource.filename('static/dat/mirror_{}d.dat'.format(dimension), srw)
found_dimension = srw.find_height_profile_dimension(dat_file)
assert found_dimension == dimension
