def propagate_single_mode(af, i, beamline):
from wofry.propagator.propagator import PropagationManager, PropagationElements, PropagationParameters
from syned.beamline.beamline_element import BeamlineElement
from syned.beamline.element_coordinates import ElementCoordinates
from wofry.propagator.propagators2D.fresnel_zoom_xy import FresnelZoomXY2D
from wofry.propagator.wavefront2D.generic_wavefront import GenericWavefront2D
from wofry.beamline.optical_elements.ideal_elements.screen import WOScreen
mi = af.coherentMode(i)
evi = af.eigenvalue(i)
print("propagating mode index", i, evi, mi.shape)
input_wavefront = GenericWavefront2D.initialize_wavefront_from_arrays(
x_array=af.xCoordinates(),
y_array=af.yCoordinates(),
z_array=mi * numpy.sqrt(evi),
)
i0 = input_wavefront.get_integrated_intensity()
input_wavefront.set_photon_energy(17226.0)
propagator = PropagationManager.Instance()
try:
propagator.add_propagator(FresnelZoomXY2D())
except:
pass
wfp = beamline.propagate(input_wavefront, propagator)
i1 = wfp[-1].get_integrated_intensity()
return wfp, i1 / i0
def initialize_default_propagator_2D():
propagator = PropagationManager.Instance()
propagator.add_propagator(Fraunhofer2D())
propagator.add_propagator(Fresnel2D())
propagator.add_propagator(FresnelConvolution2D())
propagator.add_propagator(Integral2D())
propagator.add_propagator(FresnelZoomXY2D())
def slit_schermo(input_wavefront):
#
# ===== Example of python code to create propagate current element =====
#
#
# Import section
#
import numpy
from wofry.propagator.propagator import PropagationManager, PropagationElements, PropagationParameters
from syned.beamline.beamline_element import BeamlineElement
from syned.beamline.element_coordinates import ElementCoordinates
from wofry.propagator.propagators2D.fresnel_zoom_xy import FresnelZoomXY2D
#
# info on current oe
#
#
# -------WOSlit---------
# -------Rectangle---------
# x_left : -0.5 m # x (width) minimum (signed)
# x_right : 0.5 m # x (width) maximum (signed)
# y_bottom : -0.5 m # y (length) minimum (signed)
# y_top : 0.5 m # y (length) maximum (signed)
#
#
# define current oe
#
from syned.beamline.shape import Rectangle
boundary_shape = Rectangle(x_left=-0.500000,
x_right=0.500000,
y_bottom=-0.500000,
y_top=0.500000)
from wofry.beamline.optical_elements.absorbers.slit import WOSlit
optical_element = WOSlit(boundary_shape=boundary_shape)
#
# propagating
#
#
propagation_elements = PropagationElements()
beamline_element = BeamlineElement(
optical_element=optical_element,
coordinates=ElementCoordinates(
p=0.050000,
q=0.000000,
angle_radial=numpy.radians(0.000000),
angle_azimuthal=numpy.radians(0.000000)))
propagation_elements.add_beamline_element(beamline_element)
propagation_parameters = PropagationParameters(
wavefront=input_wavefront.duplicate(),
propagation_elements=propagation_elements)
#self.set_additional_parameters(propagation_parameters)
#
propagation_parameters.set_additional_parameters('shift_half_pixel', 1)
propagation_parameters.set_additional_parameters('magnification_x',
0.000070)
propagation_parameters.set_additional_parameters('magnification_y',
0.000090)
#
propagator = PropagationManager.Instance()
try:
propagator.add_propagator(FresnelZoomXY2D())
except:
pass
output_wavefront = propagator.do_propagation(
propagation_parameters=propagation_parameters,
handler_name='FRESNEL_ZOOM_XY_2D')
return output_wavefront, beamline_element
def ideal_lens_KBh(input_wavefront):
#
# ===== Example of python code to create propagate current element =====
#
#
# Import section
#
import numpy
from wofry.propagator.propagator import PropagationManager, PropagationElements, PropagationParameters
from syned.beamline.beamline_element import BeamlineElement
from syned.beamline.element_coordinates import ElementCoordinates
from wofry.propagator.propagators2D.fresnel_zoom_xy import FresnelZoomXY2D
#
# info on current oe
#
#
# -------WOIdealLens---------
# focal_x: 0.049982758620701014 m # Focal length in x [horizontal]
# focal_y: 100000000.0 m # Focal length in y [vertical]
#
#
# define current oe
#
from wofry.beamline.optical_elements.ideal_elements.lens import WOIdealLens
optical_element = WOIdealLens(name='',
focal_x=0.049983,
focal_y=100000000.000000)
#
# propagating
#
#
propagation_elements = PropagationElements()
beamline_element = BeamlineElement(
optical_element=optical_element,
coordinates=ElementCoordinates(
p=0.000000,
q=0.000000,
angle_radial=numpy.radians(0.000000),
angle_azimuthal=numpy.radians(0.000000)))
propagation_elements.add_beamline_element(beamline_element)
propagation_parameters = PropagationParameters(
wavefront=input_wavefront.duplicate(),
propagation_elements=propagation_elements)
#self.set_additional_parameters(propagation_parameters)
#
propagation_parameters.set_additional_parameters('shift_half_pixel', 1)
propagation_parameters.set_additional_parameters('magnification_x',
1.000000)
propagation_parameters.set_additional_parameters('magnification_y',
1.000000)
#
propagator = PropagationManager.Instance()
try:
propagator.add_propagator(FresnelZoomXY2D())
except:
pass
output_wavefront = propagator.do_propagation(
propagation_parameters=propagation_parameters,
handler_name='FRESNEL_ZOOM_XY_2D')
return output_wavefront, beamline_element
xtitle="X [um]",
ytitle="Y [um]")
bl = load_from_json_file("beamline.json")
#
# propagate elements using Fresnel Zoom propagator
#
magnification_x = [1, 1, 0.01, 440, 1, 1, 1, 0.00007]
magnification_y = [1, 1, 1, 5, 1, 0.5, 1, 0.00009]
# define propagator to be used
propagator = PropagationManager.Instance()
try:
propagator.add_propagator(FresnelZoomXY2D())
except:
pass
for i, beamline_element in enumerate(
bl.get_beamline_elements()): #bl.get_beamline_elements_number()):
#
# propagating single element
#
print(">> Propagating element %d of %d" %
(i + 1, bl.get_beamline_elements_number()))
propagation_elements = PropagationElements()
propagation_elements.add_beamline_element(beamline_element)
propagation_parameters = PropagationParameters(
wavefront=wfr.duplicate(),
propagation_elements=propagation_elements)
def run_wofry():
#
# create/import your input_wavefront (THIS IS A PLACEHOLDER - REPLACE WITH YOUR SOURCE)
#
#
from wofry.propagator.wavefront2D.generic_wavefront import GenericWavefront2D
input_wavefront = GenericWavefront2D.load_h5_file(
"/users/srio/Working/paper-hierarchical/WORKSPACES/tmp.h5", "wfr")
# initialize_wavefront_from_range(-10e-6,10e-6,-100e-6,100e-6,(200,100),1e-10)
#
# info on current oe
#
#
# -------WOScreen---------
# -------BoundaryShape---------
#
#
# define current oe
#
from wofry.beamline.optical_elements.ideal_elements.screen import WOScreen
optical_element = WOScreen()
#
# propagating (*** ONLY THE ZOOM PROPAGATOR IS IMPLEMENTED ***)
#
#
propagation_elements = PropagationElements()
beamline_element = BeamlineElement(
optical_element=optical_element,
coordinates=ElementCoordinates(
p=0.000000,
q=28.300000,
angle_radial=numpy.radians(0.000000),
angle_azimuthal=numpy.radians(0.000000)))
propagation_elements.add_beamline_element(beamline_element)
propagation_parameters = PropagationParameters(
wavefront=input_wavefront.duplicate(),
propagation_elements=propagation_elements)
#self.set_additional_parameters(propagation_parameters)
#
propagation_parameters.set_additional_parameters('shift_half_pixel', 1)
propagation_parameters.set_additional_parameters('magnification_x',
8.000000)
propagation_parameters.set_additional_parameters('magnification_y',
10.000000)
#
propagator = PropagationManager.Instance()
try:
propagator.add_propagator(FresnelZoomXY2D())
except:
pass
output_wavefront = propagator.do_propagation(
propagation_parameters=propagation_parameters,
handler_name='FRESNEL_ZOOM_XY_2D')
input_wavefront = output_wavefront
#
# info on current oe
#
#
# -------WOSlit---------
# -------Rectangle---------
# x_left: -0.0018849 m # x (width) minimum (signed)
# x_right: 0.0018849 m # x (width) maximum (signed)
# y_bottom: -0.0018849 m # y (length) minimum (signed)
# y_top: 0.0018849 m # y (length) maximum (signed)
#
#
# define current oe
#
from syned.beamline.shape import Rectangle
boundary_shape = Rectangle(x_left=-0.0018849,
x_right=0.0018849,
y_bottom=-0.0018849,
y_top=0.0018849)
from wofry.beamline.optical_elements.absorbers.slit import WOSlit
optical_element = WOSlit(boundary_shape=boundary_shape)
#
# propagating (*** ONLY THE ZOOM PROPAGATOR IS IMPLEMENTED ***)
#
#
propagation_elements = PropagationElements()
beamline_element = BeamlineElement(
optical_element=optical_element,
coordinates=ElementCoordinates(
p=0.000000,
q=0.000000,
angle_radial=numpy.radians(0.000000),
angle_azimuthal=numpy.radians(0.000000)))
propagation_elements.add_beamline_element(beamline_element)
propagation_parameters = PropagationParameters(
wavefront=input_wavefront.duplicate(),
propagation_elements=propagation_elements)
#self.set_additional_parameters(propagation_parameters)
#
propagation_parameters.set_additional_parameters('shift_half_pixel', 1)
propagation_parameters.set_additional_parameters('magnification_x',
1.000000)
propagation_parameters.set_additional_parameters('magnification_y',
1.000000)
#
propagator = PropagationManager.Instance()
try:
propagator.add_propagator(FresnelZoomXY2D())
except:
pass
output_wavefront = propagator.do_propagation(
propagation_parameters=propagation_parameters,
handler_name='FRESNEL_ZOOM_XY_2D')
input_wavefront = output_wavefront
#
# info on current oe
#
#
# -------WOIdealLens---------
# focal_x: 8.319 m # Focal length in x [horizontal]
# focal_y: 99999999999999.0 m # Focal length in y [vertical]
#
#
# define current oe
#
from wofry.beamline.optical_elements.ideal_elements.lens import WOIdealLens
optical_element = WOIdealLens(name='',
focal_x=8.319000,
focal_y=99999999999999.000000)
#
# propagating (*** ONLY THE ZOOM PROPAGATOR IS IMPLEMENTED ***)
#
#
propagation_elements = PropagationElements()
beamline_element = BeamlineElement(
optical_element=optical_element,
coordinates=ElementCoordinates(
p=0.000000,
q=0.000000,
angle_radial=numpy.radians(0.000000),
angle_azimuthal=numpy.radians(0.000000)))
propagation_elements.add_beamline_element(beamline_element)
propagation_parameters = PropagationParameters(
wavefront=input_wavefront.duplicate(),
propagation_elements=propagation_elements)
#self.set_additional_parameters(propagation_parameters)
#
propagation_parameters.set_additional_parameters('shift_half_pixel', 1)
propagation_parameters.set_additional_parameters('magnification_x',
1.000000)
propagation_parameters.set_additional_parameters('magnification_y',
1.000000)
#
propagator = PropagationManager.Instance()
try:
propagator.add_propagator(FresnelZoomXY2D())
except:
pass
output_wavefront = propagator.do_propagation(
propagation_parameters=propagation_parameters,
handler_name='FRESNEL_ZOOM_XY_2D')
input_wavefront = output_wavefront
#
# info on current oe
#
#
# -------WOSlit---------
# -------Rectangle---------
# x_left: -2.5e-05 m # x (width) minimum (signed)
# x_right: 2.5e-05 m # x (width) maximum (signed)
# y_bottom: -0.5 m # y (length) minimum (signed)
# y_top: 0.5 m # y (length) maximum (signed)
#
#
# define current oe
#
from syned.beamline.shape import Rectangle
boundary_shape = Rectangle(x_left=-2.5e-05,
x_right=2.5e-05,
y_bottom=-0.5,
y_top=0.5)
from wofry.beamline.optical_elements.absorbers.slit import WOSlit
optical_element = WOSlit(boundary_shape=boundary_shape)
#
# propagating (*** ONLY THE ZOOM PROPAGATOR IS IMPLEMENTED ***)
#
#
propagation_elements = PropagationElements()
beamline_element = BeamlineElement(
optical_element=optical_element,
coordinates=ElementCoordinates(
p=11.700000,
q=0.000000,
angle_radial=numpy.radians(0.000000),
angle_azimuthal=numpy.radians(0.000000)))
propagation_elements.add_beamline_element(beamline_element)
propagation_parameters = PropagationParameters(
wavefront=input_wavefront.duplicate(),
propagation_elements=propagation_elements)
#self.set_additional_parameters(propagation_parameters)
#
propagation_parameters.set_additional_parameters('shift_half_pixel', 1)
propagation_parameters.set_additional_parameters('magnification_x',
0.010000)
propagation_parameters.set_additional_parameters('magnification_y',
1.000000)
#
propagator = PropagationManager.Instance()
try:
propagator.add_propagator(FresnelZoomXY2D())
except:
pass
output_wavefront = propagator.do_propagation(
propagation_parameters=propagation_parameters,
handler_name='FRESNEL_ZOOM_XY_2D')
input_wavefront = output_wavefront
#
#
# ######################### KB #########################################
#
#
#
# info on current oe
#
#
# -------WOSlit---------
# -------Rectangle---------
# x_left: -0.025 m # x (width) minimum (signed)
# x_right: 0.025 m # x (width) maximum (signed)
# y_bottom: -0.00045 m # y (length) minimum (signed)
# y_top: 0.00045 m # y (length) maximum (signed)
#
#
# define current oe
#
from syned.beamline.shape import Rectangle
boundary_shape = Rectangle(x_left=-0.025,
x_right=0.025,
y_bottom=-0.00045,
y_top=0.00045)
from wofry.beamline.optical_elements.absorbers.slit import WOSlit
optical_element = WOSlit(boundary_shape=boundary_shape)
#
# propagating (*** ONLY THE ZOOM PROPAGATOR IS IMPLEMENTED ***)
#
#
propagation_elements = PropagationElements()
beamline_element = BeamlineElement(
optical_element=optical_element,
coordinates=ElementCoordinates(
p=144.900000,
q=0.000000,
angle_radial=numpy.radians(0.000000),
angle_azimuthal=numpy.radians(0.000000)))
propagation_elements.add_beamline_element(beamline_element)
propagation_parameters = PropagationParameters(
wavefront=input_wavefront.duplicate(),
propagation_elements=propagation_elements)
# self.set_additional_parameters(propagation_parameters)
#
propagation_parameters.set_additional_parameters('shift_half_pixel', 1)
propagation_parameters.set_additional_parameters('magnification_x',
440.000000)
propagation_parameters.set_additional_parameters('magnification_y',
5.000000)
#
propagator = PropagationManager.Instance()
try:
propagator.add_propagator(FresnelZoomXY2D())
except:
pass
output_wavefront = propagator.do_propagation(
propagation_parameters=propagation_parameters,
handler_name='FRESNEL_ZOOM_XY_2D')
input_wavefront = output_wavefront
#
# info on current oe
#
#
# -------WOIdealLens---------
# focal_x: 100000000.0 m # Focal length in x [horizontal]
# focal_y: 0.09994594594594594 m # Focal length in y [vertical]
#
#
# define current oe
#
from wofry.beamline.optical_elements.ideal_elements.lens import WOIdealLens
optical_element = WOIdealLens(name='',
focal_x=100000000.000000,
focal_y=0.099946)
#
# propagating (*** ONLY THE ZOOM PROPAGATOR IS IMPLEMENTED ***)
#
#
propagation_elements = PropagationElements()
beamline_element = BeamlineElement(
optical_element=optical_element,
coordinates=ElementCoordinates(
p=0.000000,
q=0.000000,
angle_radial=numpy.radians(0.000000),
angle_azimuthal=numpy.radians(0.000000)))
propagation_elements.add_beamline_element(beamline_element)
propagation_parameters = PropagationParameters(
wavefront=input_wavefront.duplicate(),
propagation_elements=propagation_elements)
# self.set_additional_parameters(propagation_parameters)
#
propagation_parameters.set_additional_parameters('shift_half_pixel', 1)
propagation_parameters.set_additional_parameters('magnification_x',
1.000000)
propagation_parameters.set_additional_parameters('magnification_y',
1.000000)
#
propagator = PropagationManager.Instance()
try:
propagator.add_propagator(FresnelZoomXY2D())
except:
pass
output_wavefront = propagator.do_propagation(
propagation_parameters=propagation_parameters,
handler_name='FRESNEL_ZOOM_XY_2D')
input_wavefront = output_wavefront
#
# info on current oe
#
#
# -------WOSlit---------
# -------Rectangle---------
# x_left: -0.000195 m # x (width) minimum (signed)
# x_right: 0.000195 m # x (width) maximum (signed)
# y_bottom: -0.0065 m # y (length) minimum (signed)
# y_top: 0.0065 m # y (length) maximum (signed)
#
#
# define current oe
#
from syned.beamline.shape import Rectangle
boundary_shape = Rectangle(x_left=-0.000195,
x_right=0.000195,
y_bottom=-0.0065,
y_top=0.0065)
from wofry.beamline.optical_elements.absorbers.slit import WOSlit
optical_element = WOSlit(boundary_shape=boundary_shape)
#
# propagating (*** ONLY THE ZOOM PROPAGATOR IS IMPLEMENTED ***)
#
#
propagation_elements = PropagationElements()
beamline_element = BeamlineElement(
optical_element=optical_element,
coordinates=ElementCoordinates(
p=0.050000,
q=0.000000,
angle_radial=numpy.radians(0.000000),
angle_azimuthal=numpy.radians(0.000000)))
propagation_elements.add_beamline_element(beamline_element)
propagation_parameters = PropagationParameters(
wavefront=input_wavefront.duplicate(),
propagation_elements=propagation_elements)
# self.set_additional_parameters(propagation_parameters)
#
propagation_parameters.set_additional_parameters('shift_half_pixel', 1)
propagation_parameters.set_additional_parameters('magnification_x',
1.000000)
propagation_parameters.set_additional_parameters('magnification_y',
0.500000)
#
propagator = PropagationManager.Instance()
try:
propagator.add_propagator(FresnelZoomXY2D())
except:
pass
output_wavefront = propagator.do_propagation(
propagation_parameters=propagation_parameters,
handler_name='FRESNEL_ZOOM_XY_2D')
input_wavefront = output_wavefront
#
# info on current oe
#
#
# -------WOIdealLens---------
# focal_x: 0.049982758620701014 m # Focal length in x [horizontal]
# focal_y: 100000000.0 m # Focal length in y [vertical]
#
#
# define current oe
#
from wofry.beamline.optical_elements.ideal_elements.lens import WOIdealLens
optical_element = WOIdealLens(name='',
focal_x=0.049983,
focal_y=100000000.000000)
#
# propagating (*** ONLY THE ZOOM PROPAGATOR IS IMPLEMENTED ***)
#
#
propagation_elements = PropagationElements()
beamline_element = BeamlineElement(
optical_element=optical_element,
coordinates=ElementCoordinates(
p=0.000000,
q=0.000000,
angle_radial=numpy.radians(0.000000),
angle_azimuthal=numpy.radians(0.000000)))
propagation_elements.add_beamline_element(beamline_element)
propagation_parameters = PropagationParameters(
wavefront=input_wavefront.duplicate(),
propagation_elements=propagation_elements)
# self.set_additional_parameters(propagation_parameters)
#
propagation_parameters.set_additional_parameters('shift_half_pixel', 1)
propagation_parameters.set_additional_parameters('magnification_x',
1.000000)
propagation_parameters.set_additional_parameters('magnification_y',
1.000000)
#
propagator = PropagationManager.Instance()
try:
propagator.add_propagator(FresnelZoomXY2D())
except:
pass
output_wavefront = propagator.do_propagation(
propagation_parameters=propagation_parameters,
handler_name='FRESNEL_ZOOM_XY_2D')
input_wavefront = output_wavefront
#
# info on current oe
#
#
# -------WOSlit---------
# -------Rectangle---------
# x_left: -0.5 m # x (width) minimum (signed)
# x_right: 0.5 m # x (width) maximum (signed)
# y_bottom: -0.5 m # y (length) minimum (signed)
# y_top: 0.5 m # y (length) maximum (signed)
#
#
# define current oe
#
from syned.beamline.shape import Rectangle
boundary_shape = Rectangle(x_left=-0.5,
x_right=0.5,
y_bottom=-0.5,
y_top=0.5)
from wofry.beamline.optical_elements.absorbers.slit import WOSlit
optical_element = WOSlit(boundary_shape=boundary_shape)
#
# propagating (*** ONLY THE ZOOM PROPAGATOR IS IMPLEMENTED ***)
#
#
propagation_elements = PropagationElements()
beamline_element = BeamlineElement(
optical_element=optical_element,
coordinates=ElementCoordinates(
p=0.050000,
q=0.000000,
angle_radial=numpy.radians(0.000000),
angle_azimuthal=numpy.radians(0.000000)))
propagation_elements.add_beamline_element(beamline_element)
propagation_parameters = PropagationParameters(
wavefront=input_wavefront.duplicate(),
propagation_elements=propagation_elements)
# self.set_additional_parameters(propagation_parameters)
#
propagation_parameters.set_additional_parameters('shift_half_pixel', 1)
propagation_parameters.set_additional_parameters('magnification_x',
0.000070)
propagation_parameters.set_additional_parameters('magnification_y',
0.000090)
#
propagator = PropagationManager.Instance()
try:
propagator.add_propagator(FresnelZoomXY2D())
except:
pass
output_wavefront = propagator.do_propagation(
propagation_parameters=propagation_parameters,
handler_name='FRESNEL_ZOOM_XY_2D')
input_wavefront = output_wavefront
return input_wavefront