コード例 #1
0
def set_optics(v=None):
    el = []
    el.append(srwlib.SRWLOptA("r", "a", 0.0002, 0.001, 0.0, 0.0))
    el.append(srwlib.SRWLOptD(6.9))
    ifnMirror1 = "mirror_1d.dat"
    if ifnMirror1:
        hProfDataMirror1 = srwlib.srwl_uti_read_data_cols(ifnMirror1, "\t", 0, 1)
        el.append(srwlib.srwl_opt_setup_surf_height_1d(hProfDataMirror1, _dim="x", _ang=0.0031415926, _amp_coef=1.0, _size_x=0.00094, _size_y=0.001))
    el.append(srwlib.SRWLOptD(2.5))
    el.append(srwlib.SRWLOptA("r", "a", 0.0002, 0.001, 0.0, 0.0))
    el.append(srwlib.SRWLOptD(4.4))
    el.append(srwlib.SRWLOptA("r", "a", 5e-05, 0.001, 0.0, 0.0))
    el.append(srwlib.SRWLOptD(1.1))
    el.append(srwlib.srwl_opt_setup_CRL(2, 4.20756805e-06, 0.00731294, 1, 0.001, 0.0024, 0.0015, 1, 8e-05, 0, 0))
    el.append(srwlib.srwl_opt_setup_CRL(2, 4.20756805e-06, 0.00731294, 1, 0.001, 0.0014, 0.0005, 6, 8e-05, 0, 0))
    el.append(srwlib.SRWLOptD(9.1))
    el.append(srwlib.SRWLOptA("r", "a", 0.0014, 0.0002, 0.0, 0.0))
    el.append(srwlib.SRWLOptL(3.24479, 1e+23, 0.0, 0.0))
    el.append(srwlib.SRWLOptD(3.5))
    el.append(srwlib.SRWLOptA("r", "a", 1e-05, 1e-05, 0.0, 0.0))
    el.append(srwlib.SRWLOptD(0.7))

    pp = []
    pp.append([0, 0, 1.0, 0, 0, 2.5, 5.0, 1.5, 2.5])

    pp.append([0, 0, 1.0, 1, 0, 1.0, 1.0, 1.0, 1.0])
    if ifnMirror1:
        pp.append([0, 0, 1.0, 1, 0, 1.0, 1.0, 1.0, 1.0])

    pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0])
    pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0])

    pp.append([0, 0, 1.0, 1, 0, 1.0, 1.0, 1.0, 1.0])
    pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0])

    pp.append([0, 0, 1.0, 1, 0, 1.0, 1.0, 1.0, 1.0])
    pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0])

    pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0])

    pp.append([0, 0, 1.0, 1, 0, 1.0, 1.0, 1.0, 1.0])
    pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0])

    pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0])

    pp.append([0, 0, 1.0, 1, 0, 1.0, 1.0, 1.0, 1.0])
    pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0])

    pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0])

    pp.append([0, 0, 1.0, 0, 0, 0.3, 2.0, 0.5, 1.0])

    return srwlib.SRWLOptC(el, pp)
コード例 #2
0
def set_optics(v=None):
    el = []
    pp = []
    names = [
        'CRL', 'CRL_Watchpoint', 'Watchpoint', 'Fiber', 'Fiber_Watchpoint2',
        'Watchpoint2'
    ]
    for el_name in names:
        if el_name == 'CRL':
            # CRL: crl 36.0m
            el.append(
                srwlib.srwl_opt_setup_CRL(
                    _foc_plane=v.op_CRL_foc_plane,
                    _delta=v.op_CRL_delta,
                    _atten_len=v.op_CRL_atten_len,
                    _shape=v.op_CRL_shape,
                    _apert_h=v.op_CRL_apert_h,
                    _apert_v=v.op_CRL_apert_v,
                    _r_min=v.op_CRL_r_min,
                    _n=v.op_CRL_n,
                    _wall_thick=v.op_CRL_wall_thick,
                    _xc=v.op_CRL_x,
                    _yc=v.op_CRL_y,
                ))
            pp.append(v.op_CRL_pp)
        elif el_name == 'CRL_Watchpoint':
            # CRL_Watchpoint: drift 36.0m
            el.append(srwlib.SRWLOptD(_L=v.op_CRL_Watchpoint_L, ))
            pp.append(v.op_CRL_Watchpoint_pp)
        elif el_name == 'Watchpoint':
            # Watchpoint: watch 70.1915m
            pass
        elif el_name == 'Fiber':
            # Fiber: fiber 70.1915m
            el.append(
                srwlib.srwl_opt_setup_cyl_fiber(
                    _foc_plane=v.op_Fiber_foc_plane,
                    _delta_ext=v.op_Fiber_delta_ext,
                    _delta_core=v.op_Fiber_delta_core,
                    _atten_len_ext=v.op_Fiber_atten_len_ext,
                    _atten_len_core=v.op_Fiber_atten_len_core,
                    _diam_ext=v.op_Fiber_externalDiameter,
                    _diam_core=v.op_Fiber_diam_core,
                    _xc=v.op_Fiber_xc,
                    _yc=v.op_Fiber_yc,
                ))
            pp.append(v.op_Fiber_pp)
        elif el_name == 'Fiber_Watchpoint2':
            # Fiber_Watchpoint2: drift 70.1915m
            el.append(srwlib.SRWLOptD(_L=v.op_Fiber_Watchpoint2_L, ))
            pp.append(v.op_Fiber_Watchpoint2_pp)
        elif el_name == 'Watchpoint2':
            # Watchpoint2: watch 70.85m
            pass
    pp.append(v.op_fin_pp)
    return srwlib.SRWLOptC(el, pp)
コード例 #3
0
def set_optics(v=None):
    el = []
    pp = []
    names = ['CRL', 'CRL_Watchpoint', 'Watchpoint', 'Fiber', 'Fiber_Watchpoint2', 'Watchpoint2']
    for el_name in names:
        if el_name == 'CRL':
            # CRL: crl 36.0m
            el.append(srwlib.srwl_opt_setup_CRL(
                _foc_plane=v.op_CRL_foc_plane,
                _delta=v.op_CRL_delta,
                _atten_len=v.op_CRL_atten_len,
                _shape=v.op_CRL_shape,
                _apert_h=v.op_CRL_apert_h,
                _apert_v=v.op_CRL_apert_v,
                _r_min=v.op_CRL_r_min,
                _n=v.op_CRL_n,
                _wall_thick=v.op_CRL_wall_thick,
                _xc=v.op_CRL_x,
                _yc=v.op_CRL_y,
            ))
            pp.append(v.op_CRL_pp)
        elif el_name == 'CRL_Watchpoint':
            # CRL_Watchpoint: drift 36.0m
            el.append(srwlib.SRWLOptD(
                _L=v.op_CRL_Watchpoint_L,
            ))
            pp.append(v.op_CRL_Watchpoint_pp)
        elif el_name == 'Watchpoint':
            # Watchpoint: watch 70.1915m
            pass
        elif el_name == 'Fiber':
            # Fiber: fiber 70.1915m
            el.append(srwlib.srwl_opt_setup_cyl_fiber(
                _foc_plane=v.op_Fiber_foc_plane,
                _delta_ext=v.op_Fiber_delta_ext,
                _delta_core=v.op_Fiber_delta_core,
                _atten_len_ext=v.op_Fiber_atten_len_ext,
                _atten_len_core=v.op_Fiber_atten_len_core,
                _diam_ext=v.op_Fiber_externalDiameter,
                _diam_core=v.op_Fiber_diam_core,
                _xc=v.op_Fiber_xc,
                _yc=v.op_Fiber_yc,
            ))
            pp.append(v.op_Fiber_pp)
        elif el_name == 'Fiber_Watchpoint2':
            # Fiber_Watchpoint2: drift 70.1915m
            el.append(srwlib.SRWLOptD(
                _L=v.op_Fiber_Watchpoint2_L,
            ))
            pp.append(v.op_Fiber_Watchpoint2_pp)
        elif el_name == 'Watchpoint2':
            # Watchpoint2: watch 70.85m
            pass
    pp.append(v.op_fin_pp)
    return srwlib.SRWLOptC(el, pp)
コード例 #4
0
def set_optics(v=None):
    el = []
    # CRL: crl 36.0m
    el.append(
        srwlib.srwl_opt_setup_CRL(2, 4.716943e-06, 0.006257, 1, 0.001, 0.001,
                                  0.0005, 3, 8e-05, 0.0, 0.0))
    el.append(srwlib.SRWLOptD(34.1915))
    # Watchpoint: watch 70.1915m

    # Fiber: fiber 70.1915m
    el.append(
        srwlib.srwl_opt_setup_cyl_fiber(_foc_plane=2,
                                        _delta_ext=6.228746e-06,
                                        _delta_core=4.129923e-05,
                                        _atten_len_ext=0.002412,
                                        _atten_len_core=3.63751e-06,
                                        _diam_ext=0.0001,
                                        _diam_core=1e-05,
                                        _xc=0.0,
                                        _yc=0.0))
    el.append(srwlib.SRWLOptD(0.6585))
    # Watchpoint: watch 70.85m

    pp = []
    # CRL
    pp.append([
        0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
        0.0
    ])
    pp.append([
        0, 0, 1.0, 1, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
        0.0
    ])
    # Watchpoint
    # Fiber
    pp.append([
        0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
        0.0
    ])
    pp.append([
        0, 0, 1.0, 1, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
        0.0
    ])
    # Watchpoint
    # final post-propagation
    pp.append([
        0, 0, 1.0, 0, 0, 0.7, 2.0, 0.2, 10.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
        0.0, 0.0
    ])
    return srwlib.SRWLOptC(el, pp)
コード例 #5
0
def set_optics(v=None):
    el = []
    pp = []
    names = [
        'MOAT_1', 'MOAT_1_MOAT_2', 'MOAT_2', 'MOAT_2_HFM', 'HFM', 'HFM_VFM',
        'VFM', 'VFM_VDM', 'VDM', 'VDM_SSA', 'SSA', 'SSA_ES1', 'ES1', 'ES1_CRL',
        'CRL', 'CRL_ES2', 'ES2'
    ]
    for el_name in names:
        if el_name == 'MOAT_1':
            # MOAT_1: crystal 31.94m
            crystal = srwlib.SRWLOptCryst(
                _d_sp=v.op_MOAT_1_d_sp,
                _psi0r=v.op_MOAT_1_psi0r,
                _psi0i=v.op_MOAT_1_psi0i,
                _psi_hr=v.op_MOAT_1_psiHr,
                _psi_hi=v.op_MOAT_1_psiHi,
                _psi_hbr=v.op_MOAT_1_psiHBr,
                _psi_hbi=v.op_MOAT_1_psiHBi,
                _tc=v.op_MOAT_1_tc,
                _ang_as=v.op_MOAT_1_ang_as,
            )
            crystal.set_orient(
                _nvx=v.op_MOAT_1_nvx,
                _nvy=v.op_MOAT_1_nvy,
                _nvz=v.op_MOAT_1_nvz,
                _tvx=v.op_MOAT_1_tvx,
                _tvy=v.op_MOAT_1_tvy,
            )
            el.append(crystal)
            pp.append(v.op_MOAT_1_pp)
            mirror_file = v.op_MOAT_1_hfn
            assert os.path.isfile(mirror_file), \
                'Missing input file {}, required by MOAT_1 beamline element'.format(mirror_file)
            el.append(
                srwlib.srwl_opt_setup_surf_height_1d(
                    srwlib.srwl_uti_read_data_cols(mirror_file, "\t", 0, 1),
                    _dim=v.op_MOAT_1_dim,
                    _ang=abs(v.op_MOAT_1_ang),
                    _amp_coef=v.op_MOAT_1_amp_coef,
                ))
            pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0])
        elif el_name == 'MOAT_1_MOAT_2':
            # MOAT_1_MOAT_2: drift 31.94m
            el.append(srwlib.SRWLOptD(_L=v.op_MOAT_1_MOAT_2_L, ))
            pp.append(v.op_MOAT_1_MOAT_2_pp)
        elif el_name == 'MOAT_2':
            # MOAT_2: crystal 31.99m
            crystal = srwlib.SRWLOptCryst(
                _d_sp=v.op_MOAT_2_d_sp,
                _psi0r=v.op_MOAT_2_psi0r,
                _psi0i=v.op_MOAT_2_psi0i,
                _psi_hr=v.op_MOAT_2_psiHr,
                _psi_hi=v.op_MOAT_2_psiHi,
                _psi_hbr=v.op_MOAT_2_psiHBr,
                _psi_hbi=v.op_MOAT_2_psiHBi,
                _tc=v.op_MOAT_2_tc,
                _ang_as=v.op_MOAT_2_ang_as,
            )
            crystal.set_orient(
                _nvx=v.op_MOAT_2_nvx,
                _nvy=v.op_MOAT_2_nvy,
                _nvz=v.op_MOAT_2_nvz,
                _tvx=v.op_MOAT_2_tvx,
                _tvy=v.op_MOAT_2_tvy,
            )
            el.append(crystal)
            pp.append(v.op_MOAT_2_pp)

        elif el_name == 'MOAT_2_HFM':
            # MOAT_2_HFM: drift 31.99m
            el.append(srwlib.SRWLOptD(_L=v.op_MOAT_2_HFM_L, ))
            pp.append(v.op_MOAT_2_HFM_pp)
        elif el_name == 'HFM':
            # HFM: sphericalMirror 34.88244m
            el.append(
                srwlib.SRWLOptMirSph(
                    _r=v.op_HFM_r,
                    _size_tang=v.op_HFM_size_tang,
                    _size_sag=v.op_HFM_size_sag,
                    _nvx=v.op_HFM_nvx,
                    _nvy=v.op_HFM_nvy,
                    _nvz=v.op_HFM_nvz,
                    _tvx=v.op_HFM_tvx,
                    _tvy=v.op_HFM_tvy,
                    _x=v.op_HFM_x,
                    _y=v.op_HFM_y,
                ))
            pp.append(v.op_HFM_pp)
            mirror_file = v.op_HFM_hfn
            assert os.path.isfile(mirror_file), \
                'Missing input file {}, required by HFM beamline element'.format(mirror_file)
            el.append(
                srwlib.srwl_opt_setup_surf_height_1d(
                    srwlib.srwl_uti_read_data_cols(mirror_file, "\t", 0, 1),
                    _dim=v.op_HFM_dim,
                    _ang=abs(v.op_HFM_ang),
                    _amp_coef=v.op_HFM_amp_coef,
                ))
            pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0])
        elif el_name == 'HFM_VFM':
            # HFM_VFM: drift 34.88244m
            el.append(srwlib.SRWLOptD(_L=v.op_HFM_VFM_L, ))
            pp.append(v.op_HFM_VFM_pp)
        elif el_name == 'VFM':
            # VFM: sphericalMirror 38.30244m
            el.append(
                srwlib.SRWLOptMirSph(
                    _r=v.op_VFM_r,
                    _size_tang=v.op_VFM_size_tang,
                    _size_sag=v.op_VFM_size_sag,
                    _nvx=v.op_VFM_nvx,
                    _nvy=v.op_VFM_nvy,
                    _nvz=v.op_VFM_nvz,
                    _tvx=v.op_VFM_tvx,
                    _tvy=v.op_VFM_tvy,
                    _x=v.op_VFM_x,
                    _y=v.op_VFM_y,
                ))
            pp.append(v.op_VFM_pp)
            mirror_file = v.op_VFM_hfn
            assert os.path.isfile(mirror_file), \
                'Missing input file {}, required by VFM beamline element'.format(mirror_file)
            el.append(
                srwlib.srwl_opt_setup_surf_height_1d(
                    srwlib.srwl_uti_read_data_cols(mirror_file, "\t", 0, 1),
                    _dim=v.op_VFM_dim,
                    _ang=abs(v.op_VFM_ang),
                    _amp_coef=v.op_VFM_amp_coef,
                ))
            pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0])
        elif el_name == 'VFM_VDM':
            # VFM_VDM: drift 38.30244m
            el.append(srwlib.SRWLOptD(_L=v.op_VFM_VDM_L, ))
            pp.append(v.op_VFM_VDM_pp)
        elif el_name == 'VDM':
            # VDM: sphericalMirror 39.0m
            el.append(
                srwlib.SRWLOptMirSph(
                    _r=v.op_VDM_r,
                    _size_tang=v.op_VDM_size_tang,
                    _size_sag=v.op_VDM_size_sag,
                    _nvx=v.op_VDM_nvx,
                    _nvy=v.op_VDM_nvy,
                    _nvz=v.op_VDM_nvz,
                    _tvx=v.op_VDM_tvx,
                    _tvy=v.op_VDM_tvy,
                    _x=v.op_VDM_x,
                    _y=v.op_VDM_y,
                ))
            pp.append(v.op_VDM_pp)
            mirror_file = v.op_VDM_hfn
            assert os.path.isfile(mirror_file), \
                'Missing input file {}, required by VDM beamline element'.format(mirror_file)
            el.append(
                srwlib.srwl_opt_setup_surf_height_1d(
                    srwlib.srwl_uti_read_data_cols(mirror_file, "\t", 0, 1),
                    _dim=v.op_VDM_dim,
                    _ang=abs(v.op_VDM_ang),
                    _amp_coef=v.op_VDM_amp_coef,
                ))
            pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0])
        elif el_name == 'VDM_SSA':
            # VDM_SSA: drift 39.0m
            el.append(srwlib.SRWLOptD(_L=v.op_VDM_SSA_L, ))
            pp.append(v.op_VDM_SSA_pp)
        elif el_name == 'SSA':
            # SSA: aperture 47.00244m
            el.append(
                srwlib.SRWLOptA(
                    _shape=v.op_SSA_shape,
                    _ap_or_ob='a',
                    _Dx=v.op_SSA_Dx,
                    _Dy=v.op_SSA_Dy,
                    _x=v.op_SSA_x,
                    _y=v.op_SSA_y,
                ))
            pp.append(v.op_SSA_pp)
        elif el_name == 'SSA_ES1':
            # SSA_ES1: drift 47.00244m
            el.append(srwlib.SRWLOptD(_L=v.op_SSA_ES1_L, ))
            pp.append(v.op_SSA_ES1_pp)
        elif el_name == 'ES1':
            # ES1: watch 50.9m
            pass
        elif el_name == 'ES1_CRL':
            # ES1_CRL: drift 50.9m
            el.append(srwlib.SRWLOptD(_L=v.op_ES1_CRL_L, ))
            pp.append(v.op_ES1_CRL_pp)
        elif el_name == 'CRL':
            # CRL: crl 57.335m
            el.append(
                srwlib.srwl_opt_setup_CRL(
                    _foc_plane=v.op_CRL_foc_plane,
                    _delta=v.op_CRL_delta,
                    _atten_len=v.op_CRL_atten_len,
                    _shape=v.op_CRL_shape,
                    _apert_h=v.op_CRL_apert_h,
                    _apert_v=v.op_CRL_apert_v,
                    _r_min=v.op_CRL_r_min,
                    _n=v.op_CRL_n,
                    _wall_thick=v.op_CRL_wall_thick,
                    _xc=v.op_CRL_x,
                    _yc=v.op_CRL_y,
                ))
            pp.append(v.op_CRL_pp)
        elif el_name == 'CRL_ES2':
            # CRL_ES2: drift 57.335m
            el.append(srwlib.SRWLOptD(_L=v.op_CRL_ES2_L, ))
            pp.append(v.op_CRL_ES2_pp)
        elif el_name == 'ES2':
            # ES2: watch 59.0m
            pass
    pp.append(v.op_fin_pp)
    return srwlib.SRWLOptC(el, pp)
コード例 #6
0
def set_optics(v=None):
    el = []
    # MOAT 1: crystal 31.94m
    opCr = srwlib.SRWLOptCryst(_d_sp=3.13557135638,
                               _psi0r=-2.33400050166e-06,
                               _psi0i=8.59790386417e-09,
                               _psi_hr=-1.22944507993e-06,
                               _psi_hi=6.00282990962e-09,
                               _psi_hbr=-1.22944507993e-06,
                               _psi_hbi=6.00282990962e-09,
                               _tc=0.01,
                               _ang_as=0.0)
    # Set crystal orientation:
    opCr.set_orient(-0.0966554453406, 0.990567587399, -0.0971266167475,
                    -0.00943241252825, 0.0966675192333)
    el.append(opCr)
    ifnCryst1 = "Si_heat204.dat"
    if ifnCryst1:
        assert os.path.isfile(
            ifnCryst1
        ), "Missing input file Si_heat204.dat, required by MOAT 1 beamline element"
        hProfDataCryst1 = srwlib.srwl_uti_read_data_cols(ifnCryst1, "\t", 0, 1)
        el.append(
            srwlib.srwl_opt_setup_surf_height_1d(hProfDataCryst1,
                                                 _dim="y",
                                                 _ang=0.0972679033965,
                                                 _amp_coef=1.0))
    el.append(srwlib.SRWLOptD(0.05))
    # MOAT 2: crystal 31.99m
    opCr = srwlib.SRWLOptCryst(_d_sp=3.13557135638,
                               _psi0r=-2.33400050166e-06,
                               _psi0i=8.59790386417e-09,
                               _psi_hr=-1.22944507993e-06,
                               _psi_hi=6.00282990962e-09,
                               _psi_hbr=-1.22944507993e-06,
                               _psi_hbi=6.00282990962e-09,
                               _tc=0.01,
                               _ang_as=0.0)
    # Set crystal orientation:
    opCr.set_orient(0.0966554453406, 0.990567587399, -0.0971266167475,
                    0.00943241252825, 0.0966675192333)
    el.append(opCr)

    el.append(srwlib.SRWLOptD(2.89244))
    # HFM: sphericalMirror 34.88244m
    el.append(
        srwlib.SRWLOptMirSph(_r=7100.0,
                             _size_tang=0.5,
                             _size_sag=0.04,
                             _nvx=0.999995065202,
                             _nvy=0.0,
                             _nvz=-0.00314158748629,
                             _tvx=0.00314158748629,
                             _tvy=0.0,
                             _x=0.0,
                             _y=0.0))
    ifnSphMirror2 = "HFM_Rh7.6km.dat"
    if ifnSphMirror2:
        assert os.path.isfile(
            ifnSphMirror2
        ), "Missing input file HFM_Rh7.6km.dat, required by HFM beamline element"
        hProfDataSphMirror2 = srwlib.srwl_uti_read_data_cols(
            ifnSphMirror2, "\t", 0, 1)
        el.append(
            srwlib.srwl_opt_setup_surf_height_1d(hProfDataSphMirror2,
                                                 _dim="x",
                                                 _ang=0.003141592654,
                                                 _amp_coef=1.0))
    el.append(srwlib.SRWLOptD(3.42))
    # VFM: sphericalMirror 38.30244m
    el.append(
        srwlib.SRWLOptMirSph(_r=6100.0,
                             _size_tang=0.4,
                             _size_sag=0.04,
                             _nvx=0.0,
                             _nvy=0.999995065202,
                             _nvz=-0.00314158748629,
                             _tvx=0.0,
                             _tvy=0.00314158748629,
                             _x=0.0,
                             _y=0.0))
    ifnSphMirror3 = "VFM_Rh5.4km.dat"
    if ifnSphMirror3:
        assert os.path.isfile(
            ifnSphMirror3
        ), "Missing input file VFM_Rh5.4km.dat, required by VFM beamline element"
        hProfDataSphMirror3 = srwlib.srwl_uti_read_data_cols(
            ifnSphMirror3, "\t", 0, 1)
        el.append(
            srwlib.srwl_opt_setup_surf_height_1d(hProfDataSphMirror3,
                                                 _dim="y",
                                                 _ang=0.003141592654,
                                                 _amp_coef=1.0))
    el.append(srwlib.SRWLOptD(0.69756))
    # VDM: sphericalMirror 39.0m
    el.append(
        srwlib.SRWLOptMirSph(_r=300000.0,
                             _size_tang=0.4,
                             _size_sag=0.04,
                             _nvx=0.0,
                             _nvy=0.999995065202,
                             _nvz=-0.00314158743229,
                             _tvx=0.0,
                             _tvy=0.00314158743229,
                             _x=0.0,
                             _y=0.0))
    ifnSphMirror4 = "VDM.dat"
    if ifnSphMirror4:
        assert os.path.isfile(
            ifnSphMirror4
        ), "Missing input file VDM.dat, required by VDM beamline element"
        hProfDataSphMirror4 = srwlib.srwl_uti_read_data_cols(
            ifnSphMirror4, "\t", 0, 1)
        el.append(
            srwlib.srwl_opt_setup_surf_height_1d(hProfDataSphMirror4,
                                                 _dim="y",
                                                 _ang=0.0031415926,
                                                 _amp_coef=1.0))
    el.append(srwlib.SRWLOptD(8.00244))
    # SSA: aperture 47.00244m
    el.append(srwlib.SRWLOptA("r", "a", 0.0004, 0.0004, 0.0, 0.0))
    el.append(srwlib.SRWLOptD(3.89756))
    # ES1: watch 50.9m

    el.append(srwlib.SRWLOptD(6.435))
    # CRL: crl 57.335m
    el.append(
        srwlib.srwl_opt_setup_CRL(3, 8.211821e-07, 0.028541, 1, 0.001, 0.001,
                                  5e-05, 23, 3.24e-05, 0.0, 0.0))
    el.append(srwlib.SRWLOptD(1.665))
    # ES2: watch 59.0m

    pp = []
    # MOAT 1
    pp.append([
        0, 0, 1.0, 0, 0, 3.0, 1.0, 3.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
        0.0
    ])
    if ifnCryst1:
        pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0])
    pp.append([
        0, 0, 1.0, 1, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
        0.0
    ])
    # MOAT 2
    pp.append([
        0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
        0.0
    ])
    pp.append([
        0, 0, 1.0, 1, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
        0.0
    ])
    # HFM
    pp.append([
        0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
        0.0
    ])
    if ifnSphMirror2:
        pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0])
    pp.append([
        0, 0, 1.0, 1, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
        0.0
    ])
    # VFM
    pp.append([
        0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
        0.0
    ])
    if ifnSphMirror3:
        pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0])
    pp.append([
        0, 0, 1.0, 1, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
        0.0
    ])
    # VDM
    pp.append([
        0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
        0.0
    ])
    if ifnSphMirror4:
        pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0])
    pp.append([
        0, 0, 1.0, 1, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
        0.0
    ])
    # SSA
    pp.append([
        0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
        0.0
    ])
    pp.append([
        0, 0, 1.0, 1, 0, 0.5, 5.0, 0.5, 5.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
        0.0
    ])
    # ES1
    pp.append([
        0, 0, 1.0, 1, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
        0.0
    ])
    # CRL
    pp.append([
        0, 0, 1.0, 2, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
        0.0
    ])
    pp.append([
        0, 0, 1.0, 1, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
        0.0
    ])
    # ES2
    # final post-propagation
    pp.append([
        0, 0, 1.0, 0, 0, 0.4, 3.0, 0.4, 3.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
        0.0
    ])
    return srwlib.SRWLOptC(el, pp)
コード例 #7
0
def set_optics(v=None):
    el = []
    pp = []
    names = ['S0', 'S0_HDM', 'HDM', 'HDM_S1', 'S1', 'S1_S2', 'S2', 'S2_CRL1', 'CRL1', 'CRL2', 'CRL2_KLA', 'KLA', 'KL', 'KL_S3', 'S3', 'S3_Sample', 'Sample']
    for el_name in names:
        if el_name == 'S0':
            # S0: aperture 20.5m
            el.append(srwlib.SRWLOptA(
                _shape=v.op_S0_shape,
                _ap_or_ob='a',
                _Dx=v.op_S0_Dx,
                _Dy=v.op_S0_Dy,
                _x=v.op_S0_x,
                _y=v.op_S0_y,
            ))
            pp.append(v.op_S0_pp)
        elif el_name == 'S0_HDM':
            # S0_HDM: drift 20.5m
            el.append(srwlib.SRWLOptD(
                _L=v.op_S0_HDM_L,
            ))
            pp.append(v.op_S0_HDM_pp)
        elif el_name == 'HDM':
            # HDM: mirror 27.4m
            mirror_file = v.op_HDM_hfn
            assert os.path.isfile(mirror_file), \
                'Missing input file {}, required by HDM beamline element'.format(mirror_file)
            el.append(srwlib.srwl_opt_setup_surf_height_1d(
                srwlib.srwl_uti_read_data_cols(mirror_file, "\t", 0, 1),
                _dim=v.op_HDM_dim,
                _ang=v.op_HDM_ang,
                _amp_coef=v.op_HDM_amp_coef,
                _size_x=v.op_HDM_size_x,
                _size_y=v.op_HDM_size_y,
            ))
            pp.append(v.op_HDM_pp)
        elif el_name == 'HDM_S1':
            # HDM_S1: drift 27.4m
            el.append(srwlib.SRWLOptD(
                _L=v.op_HDM_S1_L,
            ))
            pp.append(v.op_HDM_S1_pp)
        elif el_name == 'S1':
            # S1: aperture 29.9m
            el.append(srwlib.SRWLOptA(
                _shape=v.op_S1_shape,
                _ap_or_ob='a',
                _Dx=v.op_S1_Dx,
                _Dy=v.op_S1_Dy,
                _x=v.op_S1_x,
                _y=v.op_S1_y,
            ))
            pp.append(v.op_S1_pp)
        elif el_name == 'S1_S2':
            # S1_S2: drift 29.9m
            el.append(srwlib.SRWLOptD(
                _L=v.op_S1_S2_L,
            ))
            pp.append(v.op_S1_S2_pp)
        elif el_name == 'S2':
            # S2: aperture 34.3m
            el.append(srwlib.SRWLOptA(
                _shape=v.op_S2_shape,
                _ap_or_ob='a',
                _Dx=v.op_S2_Dx,
                _Dy=v.op_S2_Dy,
                _x=v.op_S2_x,
                _y=v.op_S2_y,
            ))
            pp.append(v.op_S2_pp)
        elif el_name == 'S2_CRL1':
            # S2_CRL1: drift 34.3m
            el.append(srwlib.SRWLOptD(
                _L=v.op_S2_CRL1_L,
            ))
            pp.append(v.op_S2_CRL1_pp)
        elif el_name == 'CRL1':
            # CRL1: crl 35.4m
            el.append(srwlib.srwl_opt_setup_CRL(
                _foc_plane=v.op_CRL1_foc_plane,
                _delta=v.op_CRL1_delta,
                _atten_len=v.op_CRL1_atten_len,
                _shape=v.op_CRL1_shape,
                _apert_h=v.op_CRL1_apert_h,
                _apert_v=v.op_CRL1_apert_v,
                _r_min=v.op_CRL1_r_min,
                _n=v.op_CRL1_n,
                _wall_thick=v.op_CRL1_wall_thick,
                _xc=v.op_CRL1_x,
                _yc=v.op_CRL1_y,
            ))
            pp.append(v.op_CRL1_pp)
        elif el_name == 'CRL2':
            # CRL2: crl 35.4m
            el.append(srwlib.srwl_opt_setup_CRL(
                _foc_plane=v.op_CRL2_foc_plane,
                _delta=v.op_CRL2_delta,
                _atten_len=v.op_CRL2_atten_len,
                _shape=v.op_CRL2_shape,
                _apert_h=v.op_CRL2_apert_h,
                _apert_v=v.op_CRL2_apert_v,
                _r_min=v.op_CRL2_r_min,
                _n=v.op_CRL2_n,
                _wall_thick=v.op_CRL2_wall_thick,
                _xc=v.op_CRL2_x,
                _yc=v.op_CRL2_y,
            ))
            pp.append(v.op_CRL2_pp)
        elif el_name == 'CRL2_KLA':
            # CRL2_KLA: drift 35.4m
            el.append(srwlib.SRWLOptD(
                _L=v.op_CRL2_KLA_L,
            ))
            pp.append(v.op_CRL2_KLA_pp)
        elif el_name == 'KLA':
            # KLA: aperture 44.5m
            el.append(srwlib.SRWLOptA(
                _shape=v.op_KLA_shape,
                _ap_or_ob='a',
                _Dx=v.op_KLA_Dx,
                _Dy=v.op_KLA_Dy,
                _x=v.op_KLA_x,
                _y=v.op_KLA_y,
            ))
            pp.append(v.op_KLA_pp)
        elif el_name == 'KL':
            # KL: lens 44.5m
            el.append(srwlib.SRWLOptL(
                _Fx=v.op_KL_Fx,
                _Fy=v.op_KL_Fy,
                _x=v.op_KL_x,
                _y=v.op_KL_y,
            ))
            pp.append(v.op_KL_pp)
        elif el_name == 'KL_S3':
            # KL_S3: drift 44.5m
            el.append(srwlib.SRWLOptD(
                _L=v.op_KL_S3_L,
            ))
            pp.append(v.op_KL_S3_pp)
        elif el_name == 'S3':
            # S3: aperture 48.0m
            el.append(srwlib.SRWLOptA(
                _shape=v.op_S3_shape,
                _ap_or_ob='a',
                _Dx=v.op_S3_Dx,
                _Dy=v.op_S3_Dy,
                _x=v.op_S3_x,
                _y=v.op_S3_y,
            ))
            pp.append(v.op_S3_pp)
        elif el_name == 'S3_Sample':
            # S3_Sample: drift 48.0m
            el.append(srwlib.SRWLOptD(
                _L=v.op_S3_Sample_L,
            ))
            pp.append(v.op_S3_Sample_pp)
        elif el_name == 'Sample':
            # Sample: watch 48.7m
            pass
    pp.append(v.op_fin_pp)
    return srwlib.SRWLOptC(el, pp)
コード例 #8
0
ファイル: hxn_sample.py プロジェクト: ahebnl/SRW_PROP
def set_optics(v=None):
    el = []
    pp = []
    for el_name in names:
        if el_name == 'S1':
            # S1: aperture 26.62m
            el.append(
                srwlib.SRWLOptA(
                    _shape=v.op_S1_shape,
                    _ap_or_ob='a',
                    _Dx=v.op_S1_Dx,
                    _Dy=v.op_S1_Dy,
                    _x=v.op_S1_x,
                    _y=v.op_S1_y,
                ))
            pp.append(v.op_S1_pp)
        elif el_name == 'S1_HCM':
            # S1_HCM: drift 26.62m
            el.append(srwlib.SRWLOptD(_L=v.op_S1_HCM_L, ))
            pp.append(v.op_S1_HCM_pp)
        elif el_name == 'HCM':
            # HCM: sphericalMirror 28.35m
            el.append(
                srwlib.SRWLOptMirSph(
                    _r=v.op_HCM_r,
                    _size_tang=v.op_HCM_size_tang,
                    _size_sag=v.op_HCM_size_sag,
                    _nvx=v.op_HCM_nvx,
                    _nvy=v.op_HCM_nvy,
                    _nvz=v.op_HCM_nvz,
                    _tvx=v.op_HCM_tvx,
                    _tvy=v.op_HCM_tvy,
                    _x=v.op_HCM_x,
                    _y=v.op_HCM_y,
                ))
            pp.append(v.op_HCM_pp)

        elif el_name == 'HCM_DCM_C1':
            # HCM_DCM_C1: drift 28.35m
            el.append(srwlib.SRWLOptD(_L=v.op_HCM_DCM_C1_L, ))
            pp.append(v.op_HCM_DCM_C1_pp)
        elif el_name == 'DCM_C1':
            # DCM_C1: crystal 30.42m
            crystal = srwlib.SRWLOptCryst(
                _d_sp=v.op_DCM_C1_d_sp,
                _psi0r=v.op_DCM_C1_psi0r,
                _psi0i=v.op_DCM_C1_psi0i,
                _psi_hr=v.op_DCM_C1_psiHr,
                _psi_hi=v.op_DCM_C1_psiHi,
                _psi_hbr=v.op_DCM_C1_psiHBr,
                _psi_hbi=v.op_DCM_C1_psiHBi,
                _tc=v.op_DCM_C1_tc,
                _ang_as=v.op_DCM_C1_ang_as,
            )
            crystal.set_orient(
                _nvx=v.op_DCM_C1_nvx,
                _nvy=v.op_DCM_C1_nvy,
                _nvz=v.op_DCM_C1_nvz,
                _tvx=v.op_DCM_C1_tvx,
                _tvy=v.op_DCM_C1_tvy,
            )
            el.append(crystal)
            pp.append(v.op_DCM_C1_pp)

        elif el_name == 'DCM_C2':
            # DCM_C2: crystal 30.42m
            crystal = srwlib.SRWLOptCryst(
                _d_sp=v.op_DCM_C2_d_sp,
                _psi0r=v.op_DCM_C2_psi0r,
                _psi0i=v.op_DCM_C2_psi0i,
                _psi_hr=v.op_DCM_C2_psiHr,
                _psi_hi=v.op_DCM_C2_psiHi,
                _psi_hbr=v.op_DCM_C2_psiHBr,
                _psi_hbi=v.op_DCM_C2_psiHBi,
                _tc=v.op_DCM_C2_tc,
                _ang_as=v.op_DCM_C2_ang_as,
            )
            crystal.set_orient(
                _nvx=v.op_DCM_C2_nvx,
                _nvy=v.op_DCM_C2_nvy,
                _nvz=v.op_DCM_C2_nvz,
                _tvx=v.op_DCM_C2_tvx,
                _tvy=v.op_DCM_C2_tvy,
            )
            el.append(crystal)
            pp.append(v.op_DCM_C2_pp)

        elif el_name == 'DCM_C2_HFM':
            # DCM_C2_HFM: drift 30.42m
            el.append(srwlib.SRWLOptD(_L=v.op_DCM_C2_HFM_L, ))
            pp.append(v.op_DCM_C2_HFM_pp)
        elif el_name == 'HFM':
            # HFM: sphericalMirror 32.64m
            el.append(
                srwlib.SRWLOptMirSph(
                    _r=v.op_HFM_r,
                    _size_tang=v.op_HFM_size_tang,
                    _size_sag=v.op_HFM_size_sag,
                    _nvx=v.op_HFM_nvx,
                    _nvy=v.op_HFM_nvy,
                    _nvz=v.op_HFM_nvz,
                    _tvx=v.op_HFM_tvx,
                    _tvy=v.op_HFM_tvy,
                    _x=v.op_HFM_x,
                    _y=v.op_HFM_y,
                ))
            pp.append(v.op_HFM_pp)

        elif el_name == 'After_HFM':
            # After_HFM: watch 32.64m
            pass
        elif el_name == 'After_HFM_CRL1':
            # After_HFM_CRL1: drift 32.64m
            el.append(srwlib.SRWLOptD(_L=v.op_After_HFM_CRL1_L, ))
            pp.append(v.op_After_HFM_CRL1_pp)
        elif el_name == 'CRL1':
            # CRL1: crl 34.15m
            el.append(
                srwlib.srwl_opt_setup_CRL(
                    _foc_plane=v.op_CRL1_foc_plane,
                    _delta=v.op_CRL1_delta,
                    _atten_len=v.op_CRL1_atten_len,
                    _shape=v.op_CRL1_shape,
                    _apert_h=v.op_CRL1_apert_h,
                    _apert_v=v.op_CRL1_apert_v,
                    _r_min=v.op_CRL1_r_min,
                    _n=v.op_CRL1_n,
                    _wall_thick=v.op_CRL1_wall_thick,
                    _xc=v.op_CRL1_x,
                    _yc=v.op_CRL1_y,
                ))
            pp.append(v.op_CRL1_pp)
        elif el_name == 'CRL2':
            # CRL2: crl 34.15m
            el.append(
                srwlib.srwl_opt_setup_CRL(
                    _foc_plane=v.op_CRL2_foc_plane,
                    _delta=v.op_CRL2_delta,
                    _atten_len=v.op_CRL2_atten_len,
                    _shape=v.op_CRL2_shape,
                    _apert_h=v.op_CRL2_apert_h,
                    _apert_v=v.op_CRL2_apert_v,
                    _r_min=v.op_CRL2_r_min,
                    _n=v.op_CRL2_n,
                    _wall_thick=v.op_CRL2_wall_thick,
                    _xc=v.op_CRL2_x,
                    _yc=v.op_CRL2_y,
                ))
            pp.append(v.op_CRL2_pp)
        elif el_name == 'CRL2_Before_SSA':
            # CRL2_Before_SSA: drift 34.15m
            el.append(srwlib.SRWLOptD(_L=v.op_CRL2_Before_SSA_L, ))
            pp.append(v.op_CRL2_Before_SSA_pp)
        elif el_name == 'Before_SSA':
            # Before_SSA: watch 94.5m
            pass
        elif el_name == 'SSA':
            # SSA: aperture 94.5m
            el.append(
                srwlib.SRWLOptA(
                    _shape=v.op_SSA_shape,
                    _ap_or_ob='a',
                    _Dx=v.op_SSA_Dx,
                    _Dy=v.op_SSA_Dy,
                    _x=v.op_SSA_x,
                    _y=v.op_SSA_y,
                ))
            pp.append(v.op_SSA_pp)
        elif el_name == 'SSA_Before_FFO':
            # SSA_Before_FFO: drift 94.5m
            el.append(srwlib.SRWLOptD(_L=v.op_SSA_Before_FFO_L, ))
            pp.append(v.op_SSA_Before_FFO_pp)
        elif el_name == 'Before_FFO':
            # Before_FFO: watch 109.0m
            pass
        elif el_name == 'AFFO':
            # AFFO: aperture 109.0m
            el.append(
                srwlib.SRWLOptA(
                    _shape=v.op_AFFO_shape,
                    _ap_or_ob='a',
                    _Dx=v.op_AFFO_Dx,
                    _Dy=v.op_AFFO_Dy,
                    _x=v.op_AFFO_x,
                    _y=v.op_AFFO_y,
                ))
            pp.append(v.op_AFFO_pp)
        elif el_name == 'FFO':
            # FFO: lens 109.0m
            el.append(
                srwlib.SRWLOptL(
                    _Fx=v.op_FFO_Fx,
                    _Fy=v.op_FFO_Fy,
                    _x=v.op_FFO_x,
                    _y=v.op_FFO_y,
                ))
            pp.append(v.op_FFO_pp)
        elif el_name == 'FFO_At_Sample':
            # FFO_At_Sample: drift 109.0m
            el.append(srwlib.SRWLOptD(_L=v.op_FFO_At_Sample_L, ))
            pp.append(v.op_FFO_At_Sample_pp)
        elif el_name == 'At_Sample':
            # At_Sample: watch 109.018163m
            pass
    pp.append(v.op_fin_pp)
    return srwlib.SRWLOptC(el, pp)
コード例 #9
0
def set_optics(v=None):
    el = []
    pp = []
    names = [
        'Crystal', 'Crystal_s2_on_2nd_crystal', 's2_on_2nd_crystal_Crystal2',
        'Crystal2', 'Crystal2_s3_on_1st_mirror', 's3_on_1st_mirror',
        's3_on_1st_mirror_Elliptical_Cylinder', 'CRL', 'CRL_s4_on_slit',
        's4_on_slit', 's4_on_slit_Aperture', 'Aperture',
        'Aperture_s5_on_2nd_mirror', 's5_on_2nd_mirror', 'CRL2',
        'CRL2_s6_on_3rd_crystal', 's6_on_3rd_crystal',
        's6_on_3rd_crystal_Crystal3', 'Crystal3', 'Crystal3_s7_on_4th_crystal',
        's7_on_4th_crystal', 'Crystal4', 'Crystal4_on_detector', 'on_detector'
    ]
    for el_name in names:
        if el_name == 'Crystal':
            # Crystal: crystal 140.1m
            crystal = srwlib.SRWLOptCryst(
                _d_sp=v.op_Crystal_d_sp,
                _psi0r=v.op_Crystal_psi0r,
                _psi0i=v.op_Crystal_psi0i,
                _psi_hr=v.op_Crystal_psiHr,
                _psi_hi=v.op_Crystal_psiHi,
                _psi_hbr=v.op_Crystal_psiHBr,
                _psi_hbi=v.op_Crystal_psiHBi,
                _tc=v.op_Crystal_tc,
                _ang_as=v.op_Crystal_ang_as,
            )
            crystal.set_orient(
                _nvx=v.op_Crystal_nvx,
                _nvy=v.op_Crystal_nvy,
                _nvz=v.op_Crystal_nvz,
                _tvx=v.op_Crystal_tvx,
                _tvy=v.op_Crystal_tvy,
            )
            el.append(crystal)
            pp.append(v.op_Crystal_pp)

        elif el_name == 'Crystal_s2_on_2nd_crystal':
            # Crystal_s2_on_2nd_crystal: drift 140.1m
            el.append(srwlib.SRWLOptD(_L=v.op_Crystal_s2_on_2nd_crystal_L, ))
            pp.append(v.op_Crystal_s2_on_2nd_crystal_pp)
        elif el_name == 's2_on_2nd_crystal_Crystal2':
            # s2_on_2nd_crystal_Crystal2: drift 140.11788537m
            el.append(srwlib.SRWLOptD(_L=v.op_s2_on_2nd_crystal_Crystal2_L, ))
            pp.append(v.op_s2_on_2nd_crystal_Crystal2_pp)
        elif el_name == 'Crystal2':
            # Crystal2: crystal 140.11788538m
            crystal = srwlib.SRWLOptCryst(
                _d_sp=v.op_Crystal2_d_sp,
                _psi0r=v.op_Crystal2_psi0r,
                _psi0i=v.op_Crystal2_psi0i,
                _psi_hr=v.op_Crystal2_psiHr,
                _psi_hi=v.op_Crystal2_psiHi,
                _psi_hbr=v.op_Crystal2_psiHBr,
                _psi_hbi=v.op_Crystal2_psiHBi,
                _tc=v.op_Crystal2_tc,
                _ang_as=v.op_Crystal2_ang_as,
            )
            crystal.set_orient(
                _nvx=v.op_Crystal2_nvx,
                _nvy=v.op_Crystal2_nvy,
                _nvz=v.op_Crystal2_nvz,
                _tvx=v.op_Crystal2_tvx,
                _tvy=v.op_Crystal2_tvy,
            )
            el.append(crystal)
            pp.append(v.op_Crystal2_pp)

        elif el_name == 'Crystal2_s3_on_1st_mirror':
            # Crystal2_s3_on_1st_mirror: drift 140.11788538m
            el.append(srwlib.SRWLOptD(_L=v.op_Crystal2_s3_on_1st_mirror_L, ))
            pp.append(v.op_Crystal2_s3_on_1st_mirror_pp)
        elif el_name == 's3_on_1st_mirror':
            # s3_on_1st_mirror: watch 140.13788537m
            pass
        elif el_name == 's3_on_1st_mirror_Elliptical_Cylinder':
            # s3_on_1st_mirror_Elliptical_Cylinder: drift 140.13788537m
            el.append(
                srwlib.SRWLOptD(
                    _L=v.op_s3_on_1st_mirror_Elliptical_Cylinder_L, ))
            pp.append(v.op_s3_on_1st_mirror_Elliptical_Cylinder_pp)
        elif el_name == 'CRL':
            # CRL: crl 140.13788538m
            el.append(
                srwlib.srwl_opt_setup_CRL(
                    _foc_plane=v.op_CRL_foc_plane,
                    _delta=v.op_CRL_delta,
                    _atten_len=v.op_CRL_atten_len,
                    _shape=v.op_CRL_shape,
                    _apert_h=v.op_CRL_apert_h,
                    _apert_v=v.op_CRL_apert_v,
                    _r_min=v.op_CRL_r_min,
                    _n=v.op_CRL_n,
                    _wall_thick=v.op_CRL_wall_thick,
                    _xc=v.op_CRL_x,
                    _yc=v.op_CRL_y,
                ))
            pp.append(v.op_CRL_pp)
        elif el_name == 'CRL_s4_on_slit':
            # CRL_s4_on_slit: drift 140.13788538m
            el.append(srwlib.SRWLOptD(_L=v.op_CRL_s4_on_slit_L, ))
            pp.append(v.op_CRL_s4_on_slit_pp)
        elif el_name == 's4_on_slit':
            # s4_on_slit: watch 140.437856138m
            pass
        elif el_name == 's4_on_slit_Aperture':
            # s4_on_slit_Aperture: drift 140.437856138m
            el.append(srwlib.SRWLOptD(_L=v.op_s4_on_slit_Aperture_L, ))
            pp.append(v.op_s4_on_slit_Aperture_pp)
        elif el_name == 'Aperture':
            # Aperture: aperture 140.437856138m
            el.append(
                srwlib.SRWLOptA(
                    _shape=v.op_Aperture_shape,
                    _ap_or_ob='a',
                    _Dx=v.op_Aperture_Dx,
                    _Dy=v.op_Aperture_Dy,
                    _x=v.op_Aperture_x,
                    _y=v.op_Aperture_y,
                ))
            pp.append(v.op_Aperture_pp)
        elif el_name == 'Aperture_s5_on_2nd_mirror':
            # Aperture_s5_on_2nd_mirror: drift 140.437856138m
            el.append(srwlib.SRWLOptD(_L=v.op_Aperture_s5_on_2nd_mirror_L, ))
            pp.append(v.op_Aperture_s5_on_2nd_mirror_pp)
        elif el_name == 's5_on_2nd_mirror':
            # s5_on_2nd_mirror: watch 140.737826895m
            pass
        elif el_name == 'CRL2':
            # CRL2: crl 140.737826895m
            el.append(
                srwlib.srwl_opt_setup_CRL(
                    _foc_plane=v.op_CRL2_foc_plane,
                    _delta=v.op_CRL2_delta,
                    _atten_len=v.op_CRL2_atten_len,
                    _shape=v.op_CRL2_shape,
                    _apert_h=v.op_CRL2_apert_h,
                    _apert_v=v.op_CRL2_apert_v,
                    _r_min=v.op_CRL2_r_min,
                    _n=v.op_CRL2_n,
                    _wall_thick=v.op_CRL2_wall_thick,
                    _xc=v.op_CRL2_x,
                    _yc=v.op_CRL2_y,
                ))
            pp.append(v.op_CRL2_pp)
        elif el_name == 'CRL2_s6_on_3rd_crystal':
            # CRL2_s6_on_3rd_crystal: drift 140.737826895m
            el.append(srwlib.SRWLOptD(_L=v.op_CRL2_s6_on_3rd_crystal_L, ))
            pp.append(v.op_CRL2_s6_on_3rd_crystal_pp)
        elif el_name == 's6_on_3rd_crystal':
            # s6_on_3rd_crystal: watch 140.757826895m
            pass
        elif el_name == 's6_on_3rd_crystal_Crystal3':
            # s6_on_3rd_crystal_Crystal3: drift 140.757826895m
            el.append(srwlib.SRWLOptD(_L=v.op_s6_on_3rd_crystal_Crystal3_L, ))
            pp.append(v.op_s6_on_3rd_crystal_Crystal3_pp)
        elif el_name == 'Crystal3':
            # Crystal3: crystal 140.757826895m
            crystal = srwlib.SRWLOptCryst(
                _d_sp=v.op_Crystal3_d_sp,
                _psi0r=v.op_Crystal3_psi0r,
                _psi0i=v.op_Crystal3_psi0i,
                _psi_hr=v.op_Crystal3_psiHr,
                _psi_hi=v.op_Crystal3_psiHi,
                _psi_hbr=v.op_Crystal3_psiHBr,
                _psi_hbi=v.op_Crystal3_psiHBi,
                _tc=v.op_Crystal3_tc,
                _ang_as=v.op_Crystal3_ang_as,
            )
            crystal.set_orient(
                _nvx=v.op_Crystal3_nvx,
                _nvy=v.op_Crystal3_nvy,
                _nvz=v.op_Crystal3_nvz,
                _tvx=v.op_Crystal3_tvx,
                _tvy=v.op_Crystal3_tvy,
            )
            el.append(crystal)
            pp.append(v.op_Crystal3_pp)

        elif el_name == 'Crystal3_s7_on_4th_crystal':
            # Crystal3_s7_on_4th_crystal: drift 140.757826895m
            el.append(srwlib.SRWLOptD(_L=v.op_Crystal3_s7_on_4th_crystal_L, ))
            pp.append(v.op_Crystal3_s7_on_4th_crystal_pp)
        elif el_name == 's7_on_4th_crystal':
            # s7_on_4th_crystal: watch 140.775712275m
            pass
        elif el_name == 'Crystal4':
            # Crystal4: crystal 140.775712275m
            crystal = srwlib.SRWLOptCryst(
                _d_sp=v.op_Crystal4_d_sp,
                _psi0r=v.op_Crystal4_psi0r,
                _psi0i=v.op_Crystal4_psi0i,
                _psi_hr=v.op_Crystal4_psiHr,
                _psi_hi=v.op_Crystal4_psiHi,
                _psi_hbr=v.op_Crystal4_psiHBr,
                _psi_hbi=v.op_Crystal4_psiHBi,
                _tc=v.op_Crystal4_tc,
                _ang_as=v.op_Crystal4_ang_as,
            )
            crystal.set_orient(
                _nvx=v.op_Crystal4_nvx,
                _nvy=v.op_Crystal4_nvy,
                _nvz=v.op_Crystal4_nvz,
                _tvx=v.op_Crystal4_tvx,
                _tvy=v.op_Crystal4_tvy,
            )
            el.append(crystal)
            pp.append(v.op_Crystal4_pp)

        elif el_name == 'Crystal4_on_detector':
            # Crystal4_on_detector: drift 140.775712275m
            el.append(srwlib.SRWLOptD(_L=v.op_Crystal4_on_detector_L, ))
            pp.append(v.op_Crystal4_on_detector_pp)
        elif el_name == 'on_detector':
            # on_detector: watch 150.0m
            pass
    pp.append(v.op_fin_pp)
    return srwlib.SRWLOptC(el, pp)
コード例 #10
0
def set_optics(v=None):
    el = []
    # S0: aperture 20.5m
    el.append(srwlib.SRWLOptA("r", "a", 0.0002, 0.001, 0.0, 0.0))
    el.append(srwlib.SRWLOptD(6.9))
    # HDM: mirror 27.4m
    ifnMirror1 = "mirror_1d.dat"
    if ifnMirror1:
        assert os.path.isfile(ifnMirror1), "Missing input file mirror_1d.dat, required by HDM beamline element"
        hProfDataMirror1 = srwlib.srwl_uti_read_data_cols(ifnMirror1, "\t", 0, 1)
        el.append(srwlib.srwl_opt_setup_surf_height_1d(hProfDataMirror1, _dim="x", _ang=0.0031415926, _amp_coef=1.0, _size_x=0.00094, _size_y=0.001))
    el.append(srwlib.SRWLOptD(2.5))
    # S1: aperture 29.9m
    el.append(srwlib.SRWLOptA("r", "a", 0.0002, 0.001, 0.0, 0.0))
    el.append(srwlib.SRWLOptD(4.4))
    # S2: aperture 34.3m
    el.append(srwlib.SRWLOptA("r", "a", 5e-05, 0.001, 0.0, 0.0))
    el.append(srwlib.SRWLOptD(1.1))
    # CRL1: crl 35.4m
    el.append(srwlib.srwl_opt_setup_CRL(2, 4.207568e-06, 0.007313, 1, 0.001, 0.0024, 0.0015, 1, 8e-05, 0.0, 0.0))
    # CRL2: crl 35.4m
    el.append(srwlib.srwl_opt_setup_CRL(2, 4.207568e-06, 0.007313, 1, 0.001, 0.0014, 0.0005, 6, 8e-05, 0.0, 0.0))
    el.append(srwlib.SRWLOptD(9.1))
    # KLA: aperture 44.5m
    el.append(srwlib.SRWLOptA("r", "a", 0.0014, 0.0002, 0.0, 0.0))
    # KL: lens 44.5m
    el.append(srwlib.SRWLOptL(3.24479, 1e+23, 0.0, 0.0))
    el.append(srwlib.SRWLOptD(3.5))
    # S3: aperture 48.0m
    el.append(srwlib.SRWLOptA("r", "a", 1e-05, 1e-05, 0.0, 0.0))
    el.append(srwlib.SRWLOptD(0.7))
    # Sample: watch 48.7m

    pp = []
    # S0
    pp.append([0, 0, 1.0, 0, 0, 2.5, 5.0, 1.5, 2.5, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0])
    pp.append([0, 0, 1.0, 1, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0])
    # HDM
    if ifnMirror1:
        pp.append([0, 0, 1.0, 1, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0])
    pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0])
    # S1
    pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0])
    pp.append([0, 0, 1.0, 1, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0])
    # S2
    pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0])
    pp.append([0, 0, 1.0, 1, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0])
    # CRL1
    pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0])
    # CRL2
    pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0])
    pp.append([0, 0, 1.0, 1, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0])
    # KLA
    pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0])
    # KL
    pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0])
    pp.append([0, 0, 1.0, 1, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0])
    # S3
    pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0])
    pp.append([0, 0, 1.0, 0, 0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0])
    # Sample
    # final post-propagation
    pp.append([0, 0, 1.0, 0, 0, 0.3, 2.0, 0.5, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0])
    return srwlib.SRWLOptC(el, pp)