#############################################################################
# SRWLib for Python Utility module for Synchrotron Sources (electron beams and ID parameters)
# Author: O.C.
# v 0.02
#############################################################################
try: #OC15112022
from .srwlib import *
except:
from srwlib import *
#from srwlib import *
#****************************************************************************
#def srwl_uti_src_e_beams_predef():
[docs]
def srwl_uti_src_e_beam_predef(): #OC25012017
#E-Beam params in the order: _Iavg, _e, _sig_e, _emit_x, _beta_x, _alpha_x, _eta_x, _eta_x_pr, _emit_y, _beta_y, _alpha_y
allBeams = [
#['NSLS-II Low Beta Day 1', [0.5, 3, 0.89e-03, 0.9e-09, 2.02, 0, 0, 0, 8e-12, 1.06, 0]],
#['NSLS-II Low Beta Final', [0.5, 3, 0.89e-03, 0.55e-09, 2.02, 0, 0, 0, 8e-12, 1.06, 0]],
['NSLS-II Low Beta Day 1', [0.5, 3, 0.89e-03, 0.9e-09, 1.84, 0, 0, 0, 8e-12, 1.17, 0]],
['NSLS-II Low Beta All IDs', [0.5, 3, 0.8e-03, 0.76e-09, 1.84, 0, 0, 0, 7e-12, 1.17, 0]],
['NSLS-II Low Beta Final', [0.5, 3, 0.89e-03, 0.55e-09, 1.84, 0, 0, 0, 8e-12, 1.17, 0]],
#['NSLS-II High Beta Day 1', [0.5, 3, 0.89e-03, 0.9e-09, 20.85, 0, 0, 0, 8e-12, 3.4, 0]],
#['NSLS-II High Beta Final', [0.5, 3, 0.89e-03, 0.55e-09, 20.85, 0, 0, 0, 8e-12, 3.4, 0]],
['NSLS-II High Beta Day 1', [0.5, 3, 0.89e-03, 0.9e-09, 20.1, 0, 0, 0, 8e-12, 3.4, 0]],
['NSLS-II High Beta All IDs', [0.5, 3, 0.8e-03, 0.76e-09, 20.1, 0, 0, 0, 7e-12, 3.4, 0]],
['NSLS-II High Beta Final', [0.5, 3, 0.89e-03, 0.55e-09, 20.1, 0, 0, 0, 8e-12, 3.4, 0]],
['NSLS-II 3PW Day 1', [0.5, 3, 0.89e-03, 0.9e-09, 2.956, 1.932, 0.137, -0.105, 8e-12, 19.653, -0.806]],
['NSLS-II 3PW Final', [0.5, 3, 0.89e-03, 0.55e-09, 2.956, 1.932, 0.137, -0.105, 8e-12, 19.653, -0.806]],
['NSLS-II BM Day 1', [0.5, 3, 0.89e-03, 0.9e-09, 1.5, 0, 0.137, -0.1, 8e-12, 22.5, -0.9]],
['NSLS-II BM Final', [0.5, 3, 0.89e-03, 0.55e-09, 1.5, 0, 0.137, -0.1, 8e-12, 22.5, -0.9]],
['DIAMOND Low Beta', [0.3, 3, 1.0e-03, 2.797e-09, 2.283, 0, 0, 0, 24.32e-12, 2.514, 0]],
['SOLEIL Short', [0.5, 2.75, 1.016e-03, 3.73e-09, 17.78, 0, 0.28, 0, 37e-12, 1.75, 0]],
['SOLEIL Medium', [0.5, 2.75, 1.016e-03, 3.73e-09, 4.0, 0, 0.13, 0, 37e-12, 1.77, 0]],
['SOLEIL Long', [0.5, 2.75, 1.016e-03, 3.73e-09, 10.09, 0, 0.2, 0, 37e-12, 8.01, 0]],
['SOLEIL BM 1 deg.', [0.5, 2.75, 1.016e-03, 3.73e-09, 0.603, 0.776, 0.039, -0.088, 37e-12, 16.53, 0.931]],
['SOLEIL BM 4 deg.', [0.5, 2.75, 1.016e-03, 3.73e-09, 0.375, 0.024, 0.021, -0.037, 37e-12, 16.01, 0.899]],
['ESRF Low Beta', [0.2, 6.04, 1.1e-03, 4.e-09, 0.35, 0, 0.031, 0, 25e-12, 2.97, 0]],
['ESRF High Beta', [0.2, 6.04, 1.1e-03, 4.e-09, 37.6, 0, 0.134, 0, 25e-12, 2.95, 0]],
['ESRF BM', [0.2, 6.04, 1.1e-03, 4.e-09, 2.2, 1.43, 0.045, 0, 25e-12, 34.9, 0]], #to correct _alpha_x, _alpha_y, _eta_x_pr
['ESRF-U BM', [0.2, 6., 0.95e-03, 0.13e-09, 1.8138, -2.02, 0.018, 0.016, 5e-12, 2.5685, 1.9307]],
['APS', [0.1, 7, 0.96e-03, 2.79e-09, 22.7, 0, 0.206, 0, 8.4e-12, 3.1, 0]],
['SPring8 High Beta', [0.1, 8, 1.1e-03, 3.4e-09, 22.6, 0, 0.107, 0, 6.8e-12, 5.6, 0]]
]#add more beams
return allBeams
#****************************************************************************
[docs]
def srwl_uti_src_e_beam(_nm, _Iavg=None, _e=None, _sig_e=None, _emit_x=None, _beta_x=None, _alpha_x=None, _eta_x=None, _eta_x_pr=None, _emit_y=None, _beta_y=None, _alpha_y=None):
#def srwl_uti_src_e_beam(_nm):
"""Instantiates electron beam structures describing different existing sources
:param _nm: string identifying a source
:return: SRWLPartBeam object
"""
#allBeams = srwl_uti_src_e_beams_predef()
allBeams = srwl_uti_src_e_beam_predef() #OC25012017
sTest = _nm.replace(' ', '')
sTest = sTest.replace('-', '')
sTest = sTest.capitalize()
resBeam = SRWLPartBeam()
nBeams = len(allBeams)
for i in range(nBeams):
curInf = allBeams[i]
curStr = curInf[0]
curStr = curStr.replace(' ', '')
curStr = curStr.replace('-', '')
curStr = curStr.capitalize()
if sTest == curStr:
ar = curInf[1]
if(_Iavg is not None): ar[0] = _Iavg
if(_e is not None): ar[1] = _e
if(_sig_e is not None): ar[2] = _sig_e
if(_emit_x is not None): ar[3] = _emit_x
if(_beta_x is not None): ar[4] = _beta_x
if(_alpha_x is not None): ar[5] = _alpha_x
if(_eta_x is not None): ar[6] = _eta_x
if(_eta_x_pr is not None): ar[7] = _eta_x_pr
if(_emit_y is not None): ar[8] = _emit_y
if(_beta_y is not None): ar[9] = _beta_y
if(_alpha_y is not None): ar[10] = _alpha_y
resBeam.from_Twiss(_Iavg=ar[0], _e=ar[1], _sig_e=ar[2], _emit_x=ar[3], _beta_x=ar[4], _alpha_x=ar[5], _eta_x=ar[6], _eta_x_pr=ar[7], _emit_y=ar[8], _beta_y=ar[9], _alpha_y=ar[10])
return resBeam
return None
#****************************************************************************
[docs]
def srwl_uti_src_sph_wave(_wfr, _pol, _norm=1): #Move it to different place (C part?)
"""Auxiliary function: sets up spherical wavefront of a point source
:param _wfr: wavefront structure to be set up, using its input params
:param _pol: electric field polarization to set: =0 -Linear Horizontal; =1 -Linear Vertical;
:param _norm: normalizing coeficient
"""
R = _wfr.mesh.zStart #0.5*(_wfr.Rx + _wfr.Ry)
Re2 = R*R
arE = _wfr.arEx if(_pol == 0) else _wfr.arEy
multE = _norm #to update, assuming _norm defining power of the source
eStep = 0 if(_wfr.mesh.ne <= 1) else (_wfr.mesh.eFin - _wfr.mesh.eStart)/(_wfr.mesh.ne - 1)
xStep = (_wfr.mesh.xFin - _wfr.mesh.xStart)/(_wfr.mesh.nx - 1)
yStep = (_wfr.mesh.yFin - _wfr.mesh.yStart)/(_wfr.mesh.ny - 1)
i = 0
y = _wfr.mesh.yStart - yStep - _wfr.yc
for iy in range(_wfr.mesh.ny):
y += yStep
ye2 = y*y
x = _wfr.mesh.xStart - xStep - _wfr.xc
for ix in range(_wfr.mesh.nx):
x += xStep
xe2 = x*x
curR = sqrt(Re2 + xe2 + ye2)
mult = multE/curR
phEn = _wfr.mesh.eFin - eStep
for ie in range(_wfr.mesh.ne):
phEn += eStep
k = (5.067730652e+06)*phEn
ph = k*curR
arE[i] = mult*cos(ph)
arE[i + 1] = mult*sin(ph)
i += 2
