def run_hyperboloid(kind="hyperboloid"):
#
# shadow3
#
if kind == "hyperboloid":
from shadow4tests.oasys_workspaces.mirrors_branch5_hyperboloid import define_source, run_source, define_beamline, run_beamline
else:
raise Exception("Bad input")
oe0 = define_source()
beam3_source = run_source(oe0)
#
# shadow4
#
from shadow4.syned.element_coordinates import ElementCoordinates
oe = define_beamline()[0]
beam4_source = Beam.initialize_from_array(beam3_source.rays)
beam4 = beam4_source
check_congruence(oe)
#
# shadow definitions
#
if oe.F_DEFAULT == 0:
p_focus = oe.SSOUR
q_focus = oe.SIMAG
grazing_angle = numpy.radians(90 - oe.THETA)
elif oe.F_DEFAULT == 1:
p_focus = oe.T_SOURCE
q_focus = oe.T_IMAGE
grazing_angle = numpy.radians(90 - oe.T_INCIDENCE)
is_cylinder = oe.FCYL
if oe.CIL_ANG == 0:
cylinder_direction = Direction.TANGENTIAL
else:
cylinder_direction = Direction.SAGITTAL
if oe.F_CONVEX == 0:
convexity = Convexity.UPWARD
elif oe.F_CONVEX == 1:
convexity = Convexity.DOWNWARD
name = "Hyperboloid Mirror (%s) " % kind
mirror1 = S4HyperboloidMirrorElement(
optical_element=S4HyperboloidMirror(
name=name,
boundary_shape=None,
surface_calculation=SurfaceCalculation.INTERNAL,
is_cylinder=is_cylinder,
cylinder_direction=cylinder_direction,
convexity=convexity,
p_focus=p_focus,
q_focus=q_focus,
grazing_angle=grazing_angle,
# inputs related to mirror reflectivity
f_reflec=oe.
F_REFLEC, # reflectivity of surface: 0=no reflectivity, 1=full polarization
# f_refl=0, # 0=prerefl file, 1=electric susceptibility, 2=user defined file (1D reflectivity vs angle)
# # 3=user defined file (1D reflectivity vs energy), # 4=user defined file (2D reflectivity vs energy and angle)
# file_refl="", # preprocessor file fir f_refl=0,2,3,4
# refraction_index=1.0 # refraction index (complex) for f_refl=1
),
coordinates=ElementCoordinates(
p=oe.T_SOURCE,
q=oe.T_IMAGE,
angle_radial=numpy.radians(oe.T_INCIDENCE),
),
)
print(mirror1.info())
#
# run
#
beam4, mirr4 = mirror1.trace_beam(beam_in=beam4, flag_lost_value=-11000)
#
# compare
#
oe_list = define_beamline()
beam3 = run_beamline(beam3_source, oe_list)
plotxy(beam3, 1, 3, nbins=201, nolost=1, title="%s shadow3" % name)
plotxy(beam4, 1, 3, nbins=201, nolost=1, title="%s shadow4" % name)
from shadow4tests.compatibility.compare_beams import check_six_columns_mean_and_std, check_almost_equal
check_six_columns_mean_and_std(beam3,
beam4,
do_assert=True,
do_plot=False,
assert_value=1e-6)
check_almost_equal(beam3, beam4, do_assert=False, level=3)
bending_magnet_magnetic_structure=bm)
print(ls_bm.info())
beam4 = ls_bm.get_beam(
F_COHER=oe0.F_COHER,
NRAYS=oe0.NPOINT,
SEED=oe0.ISTAR1,
EPSI_DX=oe0.EPSI_DX,
EPSI_DZ=oe0.EPSI_DZ,
psi_interval_in_units_one_over_gamma=None,
psi_interval_number_of_points=1001,
verbose=False,
)
#
# compare
#
plotxy(beam3, 1, 3, nbins=201, title="BM shadow3")
plotxy(beam4, 1, 3, nbins=201, title="BM shadow4")
from shadow4tests.compatibility.compare_beams import check_six_columns_mean_and_std
check_six_columns_mean_and_std(beam3,
beam4,
do_plot=True,
do_assert=True,
assert_value=1e-2,
to_meters=to_meters)
y_bottom=-oe.RLEN2,
y_top=oe.RLEN1,
)),
coordinates=ElementCoordinates(
p=oe.T_SOURCE,
q=oe.T_IMAGE,
angle_radial=numpy.radians(oe.T_INCIDENCE),
),
)
print(mirror1.info())
#
# run
#
beam4, mirr4 = mirror1.trace_beam(beam_in=beam4, flag_lost_value=-11000)
#
# compare
#
beam3 = run_beamline(beam3_source, define_beamline())
plotxy(beam3, 1, 3, nbins=201, nolost=1, title="%s shadow3" % name)
plotxy(beam4, 1, 3, nbins=201, nolost=1, title="%s shadow4" % name)
from shadow4tests.compatibility.compare_beams import check_six_columns_mean_and_std, check_almost_equal
check_six_columns_mean_and_std(beam3, beam4, do_assert=True, do_plot=False)
check_almost_equal(beam3, beam4, do_assert=True, level=2)
def run_conic(kind="conic"):
#
# shadow3
#
if kind == "conic":
from shadow4tests.oasys_workspaces.mirrors_branch5_conic import define_source, run_source, define_beamline, run_beamline
else:
raise Exception("Bad input")
oe0 = define_source()
beam3_source = run_source(oe0)
#
# shadow4
#
from shadow4.syned.element_coordinates import ElementCoordinates
oe = define_beamline()[0]
beam4_source = Beam.initialize_from_array(beam3_source.rays)
beam4 = beam4_source
check_congruence(oe)
#
# shadow definitions
#
name = "Conic Mirror (%s) " % kind
mirror1 = S4ConicMirrorElement(
optical_element=S4ConicMirror(
name=name,
boundary_shape=None,
conic_coefficients=oe.CCC.tolist(),
# inputs related to mirror reflectivity
f_reflec=oe.F_REFLEC, # reflectivity of surface: 0=no reflectivity, 1=full polarization
# f_refl=0, # 0=prerefl file, 1=electric susceptibility, 2=user defined file (1D reflectivity vs angle)
# # 3=user defined file (1D reflectivity vs energy), # 4=user defined file (2D reflectivity vs energy and angle)
# file_refl="", # preprocessor file fir f_refl=0,2,3,4
# refraction_index=1.0 # refraction index (complex) for f_refl=1
),
coordinates=ElementCoordinates(
p=oe.T_SOURCE,
q=oe.T_IMAGE,
angle_radial=numpy.radians(oe.T_INCIDENCE),
),
)
print(mirror1.info())
#
# run
#
beam4, mirr4 = mirror1.trace_beam(beam_in=beam4, flag_lost_value=-11000)
#
# compare
#
oe_list = define_beamline()
beam3 = run_beamline(beam3_source, oe_list)
plotxy(beam3, 1, 3, nbins=201, nolost=1, title="%s shadow3" % name)
plotxy(beam4, 1, 3, nbins=201, nolost=1, title="%s shadow4" % name)
from shadow4tests.compatibility.compare_beams import check_six_columns_mean_and_std, check_almost_equal
check_six_columns_mean_and_std(beam3, beam4, do_assert=True, do_plot=False, assert_value=1e-6)
check_almost_equal(beam3, beam4, do_assert = True, level=3)
