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