def simulate(params, msg_callback=lambda _: _):
"""Run a multi-particle (thick) simulation and return results
Args:
params (OrderedMapping): configuration parameters
msg_callback (function): Called at various stages to log output
Returns:
OrderedMapping: results and params (see code for format)
"""
p = pkcollections.OrderedMapping()
for k in 'simulation_kind', 'wavefront':
v = params[k]
p[k] = v.value if hasattr(v, 'value') else v
pkcollections.mapping_merge(
p, srw_params.to_undulator_multi_particle(params.undulator))
p.beam = srw_params.to_beam(params.beam)
p.stkF = srw_params.to_wavefront_multi_particle(p.wavefront)
p.stkP = srw_params.to_wavefront_multi_particle(p.wavefront)
p.ar_prec_f = srw_params.to_flux_precision(params.precision)
p.ar_prec_p = srw_params.to_power_precision(params.precision)
p.arPrecPar = [1] #General Precision parameters for Trajectory calculation:
p.fieldInterpMeth = 4
p.plots = []
msg_callback('Performing trajectory calculation')
_trajectory(p)
if params.simulation_kind == 'E':
msg_callback('Performing Electric Field (spectrum vs photon energy) calculation')
msg_callback('Extracting Intensity from calculated Electric Field')
srwlib.srwl.CalcStokesUR(p.stkF, p.beam, p.und, p.ar_prec_f)
p.plots.append([
uti_plot.uti_plot1d,
p.stkF.arS,
[p.stkF.mesh.eStart, p.stkF.mesh.eFin, p.stkF.mesh.ne],
[
'Photon Energy [eV]',
'Flux [ph/s/.1%bw]',
'Flux through Finite Aperture',
],
])
elif params.simulation_kind == 'X':
msg_callback('Performing Power Density calculation (from field) vs x-coordinate calculation')
srwlib.srwl.CalcPowDenSR(p.stkP, p.beam, 0, p.magFldCnt, p.ar_prec_p)
msg_callback('Extracting Intensity from calculated Electric Field')
p.plotMeshX = [1000*p.stkP.mesh.xStart, 1000*p.stkP.mesh.xFin, p.stkP.mesh.nx]
p.powDenVsX = pkarray.new_float([0]*p.stkP.mesh.nx)
for i in xrange(p.stkP.mesh.nx):
powDenVsX[i] = p.stkP.arS[p.stkP.mesh.nx*int(p.stkP.mesh.ny*0.5) + i]
p.plots.append([
uti_plot.uti_plot1d,
p.powDenVsX,
p.plotMeshX,
[
'Horizontal Position [mm]',
'Power Density [W/mm^2]',
'Power Density\n(horizontal cut at y = 0)',
],
])
elif params.simulation_kind == 'Y':
msg_callback('Performing Power Density calculation (from field) vs x-coordinate calculation')
srwlib.srwl.CalcPowDenSR(p.stkP, p.beam, 0, p.magFldCnt, p.ar_prec_p)
msg_callback('Extracting Intensity from calculated Electric Field')
p.plotMeshY = [1000*p.stkP.mesh.yStart, 1000*p.stkP.mesh.yFin, p.stkP.mesh.ny]
p.powDenVsY = pkarray.new_float([0]*p.stkP.mesh.ny)
for i in xrange(p.stkP.mesh.ny):
p.powDenVsY[i] = p.stkP.arS[p.stkP.mesh.ny*int(p.stkP.mesh.nx*0.5) + i]
p.plots.append([
uti_plot.uti_plot1d,
p.powDenVsY,
p.plotMeshY,
[
'Vertical Position [mm]',
'Power Density [W/mm^2]',
'Power Density\n(vertical cut at x = 0)',
],
])
elif params.simulation_kind == 'X_AND_Y':
msg_callback('Performing Electric Field (intensity vs x- and y-coordinate) calculation')
srwlib.srwl.CalcPowDenSR(p.stkP, p.beam, 0, p.magFldCnt, p.ar_prec_p)
msg_callback('Extracting Intensity from calculated Electric Field')
p.plotMeshX = [1000*p.stkP.mesh.xStart, 1000*p.stkP.mesh.xFin, p.stkP.mesh.nx]
p.plotMeshY = [1000*p.stkP.mesh.yStart, 1000*p.stkP.mesh.yFin, p.stkP.mesh.ny]
p.plots.append([
uti_plot.uti_plot2d,
p.stkP.arS,
p.plotMeshX,
p.plotMeshY,
[
'Horizontal Position [mm]',
'Vertical Position [mm]',
'Power Density',
],
])
else:
raise AssertionError('{}: invalid simulation_kind'.format(params.simulation_kind))
return p
def simulate(params, msg_callback):
"""Run a single-particle (thin) simulation and return results
Args:
params (OrderedMapping): map of parameters
msg_callback (function): Called at various stages to log output
Returns:
OrderedMapping: results and params (see code for format)
"""
p = pkcollections.OrderedMapping()
for k in 'polarization', 'intensity', 'simulation_kind', 'wavefront':
v = params[k]
p[k] = v.value if hasattr(v, 'value') else v
pkcollections.mapping_merge(
p, srw_params.to_undulator_single_particle(params.undulator))
p.arPrecPar = srw_params.to_precision_single_particle(params.precision)
p.wfrE = srw_params.to_wavefront_single_particle(p.wavefront)
p.wfrE.partBeam = srw_params.to_beam(params.beam)
p.wfrXY = srw_params.to_wavefront_single_particle(p.wavefront)
p.wfrXY.partBeam = srw_params.to_beam(params.beam)
p.beam = srw_params.to_beam(params.beam)
p.plots = []
skv = p.simulation_kind
if params.simulation_kind == 'E':
msg_callback('Performing Electric Field (spectrum vs photon energy) calculation')
srwlib.srwl.CalcElecFieldSR(p.wfrE, 0, p.magFldCnt, p.arPrecPar)
msg_callback('Extracting Intensity from calculated Electric Field')
p.arI1 = pkarray.new_float([0]*p.wfrE.mesh.ne)
srwlib.srwl.CalcIntFromElecField(
p.arI1, p.wfrE, p.polarization, p.intensity, skv, p.wfrE.mesh.eStart, 0, 0)
p.plots.append([
uti_plot.uti_plot1d,
p.arI1,
[p.wfrE.mesh.eStart, p.wfrE.mesh.eFin, p.wfrE.mesh.ne],
['Photon energy, eV','Spectral intensity, ph/s/0.1%BW','Intensity vs photon energy'],
])
elif params.simulation_kind == 'X':
msg_callback('Performing Electric Field (intensity vs x-coordinate) calculation')
srwlib.srwl.CalcElecFieldSR(p.wfrXY, 0, p.magFldCnt, p.arPrecPar)
msg_callback('Extracting Intensity from calculated Electric Field')
p.arI1 = pkarray.new_float([0]*p.wfrXY.mesh.nx)
srwlib.srwl.CalcIntFromElecField(p.arI1, p.wfrXY, p.polarization, p.intensity, skv, 0, p.wfrXY.mesh.xStart, 0)
p.plots.append([
uti_plot.uti_plot1d,
p.arI1,
[p.wfrXY.mesh.xStart, p.wfrXY.mesh.xFin, p.wfrXY.mesh.nx],
['Horizontal Position [m]','Spectral intensity, ph/s/0.1%BW','Intensity vs x-coordinate'],
])
elif params.simulation_kind == 'Y':
msg_callback('Performing Electric Field (intensity vs y-coordinate) calculation')
srwlib.srwl.CalcElecFieldSR(p.wfrXY, 0, p.magFldCnt, p.arPrecPar)
msg_callback('Extracting Intensity from calculated Electric Field')
p.arI1 = pkarray.new_float([0]*p.wfrXY.mesh.ny)
srwlib.srwl.CalcIntFromElecField(p.arI1, p.wfrXY, p.polarization, p.intensity, skv, 0, p.wfrXY.mesh.yStart, 0)
p.plots.append([
uti_plot.uti_plot1d,
p.arI1,
[p.wfrXY.mesh.yStart, p.wfrXY.mesh.yFin, p.wfrXY.mesh.ny],
['Vertical Position [m]','Spectral intensity, ph/s/0.1%BW','Intensity vs y-coordinate'],
])
elif params.simulation_kind == 'X_AND_Y':
msg_callback('Performing Electric Field (intensity vs x- and y-coordinate) calculation')
srwlib.srwl.CalcElecFieldSR(p.wfrXY, 0, p.magFldCnt, p.arPrecPar)
msg_callback('Extracting Intensity from calculated Electric Field')
p.arI1 = pkarray.new_float([0]*p.wfrXY.mesh.nx*p.wfrXY.mesh.ny)
srwlib.srwl.CalcIntFromElecField(p.arI1, p.wfrXY, p.polarization, p.intensity, skv, p.wfrXY.mesh.eStart, p.wfrXY.mesh.xStart, p.wfrXY.mesh.yStart)
p.plots.append([
uti_plot.uti_plot2d,
p.arI1,
[1*p.wfrXY.mesh.xStart, 1*p.wfrXY.mesh.xFin, p.wfrXY.mesh.nx],
[1*p.wfrXY.mesh.yStart, 1*p.wfrXY.mesh.yFin, p.wfrXY.mesh.ny],
['Horizontal Position [m]', 'Vertical Position [m]', 'Intensity at ' + str(wfrXY.mesh.eStart) + ' eV'],
]),
elif params.simulation_kind == 'E_AND_X':
msg_callback('Performing Electric Field (intensity vs energy- and x-coordinate) calculation')
srwlib.srwl.CalcElecFieldSR(p.wfrXY, 0, p.magFldCnt, p.arPrecPar)
msg_callback('* Extracting Intensity from calculated Electric Field')
p.arI1 = pkarray.new_float([0]*p.wfrXY.mesh.ne*p.wfrXY.mesh.nx)
srwlib.srwl.CalcIntFromElecField(p.arI1, p.wfrXY, p.polarization, p.intensity, skv, p.wfrXY.mesh.eStart, p.wfrXY.mesh.xStart, p.wfrXY.mesh.yStart)
p.plots.append([
uti_plot.uti_plot1d,
p.arI1,
[1*p.wfrXY.mesh.eStart, 1*p.wfrXY.mesh.eFin, p.wfrXY.mesh.ne],
[1*p.wfrXY.mesh.xStart, 1*p.wfrXY.mesh.xFin, p.wfrXY.mesh.nx],
['Energy [eV]', 'Horizontal Position [m]', 'Intensity integrated from ' + str(p.wfrXY.mesh.yStart) + ' to ' + str(p.wfrXY.mesh.yFin) + ' ,m in y-coordinate'],
])
elif params.simulation_kind == 'E_AND_Y':
msg_callback('Performing Electric Field (intensity vs energy- and y-coordinate) calculation')
srwlib.srwl.CalcElecFieldSR(p.wfrXY, 0, p.magFldCnt, p.arPrecPar)
msg_callback('Extracting Intensity from calculated Electric Field')
p.arI1 = pkarray.new_float([0]*p.wfrXY.mesh.ne*p.wfrXY.mesh.ny)
srwlib.srwl.CalcIntFromElecField(p.arI1, p.wfrXY, p.polarization, p.intensity, skv, p.wfrXY.mesh.eStart, p.wfrXY.mesh.xStart, p.wfrXY.mesh.yStart)
p.plots.append([
uti_plot.uti_plot1d,
p.arI1,
[1*p.wfrXY.mesh.eStart, 1*p.wfrXY.mesh.eFin, p.wfrXY.mesh.ne],
[1*p.wfrXY.mesh.yStart, 1*p.wfrXY.mesh.yFin, p.wfrXY.mesh.ny],
['Energy [eV]', 'Vertical Position [m]', 'Intensity integrated from ' + str(p.wfrXY.mesh.xStart) + ' to ' + str(p.wfrXY.mesh.xFin)+ ' ,m in x-coordinate'],
])
else:
raise AssertionError('{}: invalid p.simulation_kind'.format(params.simulation_kind))
return p
