def test_compound_wolter1_with_hole():
print(
">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> test_compound_wolter1_with_hole"
)
p = 100.
beam1 = Beam.initialize_as_person(100000)
beam1.x *= 50.
beam1.z *= 50.
beam1.set_point(p, 0., p)
op_ax = Beam(1)
op_ax.set_point(p, 0., p)
beam = op_ax.merge(beam1)
beam.set_divergences_collimated()
beam.plot_xz()
p = 6 * 1e8
R = 100.
theta = 0.001 * np.pi / 180
wolter = CompoundOpticalElement.initialiaze_as_wolter_1_with_two_parameters(
p1=p, R=R, theta=theta)
#beam = wolter.trace_compound(beam)
beam = wolter.trace_good_rays(beam)
beam.plot_good_xz()
beam.retrace(10.)
beam.plot_good_xz()
plt.show()
def test_clean_wolter3():
p = 50.
beam1 = Beam.initialize_as_person()
beam1.set_point(p, 0., p)
#beam1.set_rectangular_spot(5 / 2 * 1e-5, -5 / 2 * 1e-5, 5 / 2 * 1e-5, -5 / 2 * 1e-5)
op_ax = Beam(1)
op_ax.set_point(p, 0., p)
beam = op_ax.merge(beam1)
beam.set_divergences_collimated()
beam.plot_xz()
distance_between_the_foci = 10.
wolter3 = CompoundOpticalElement.initialize_as_wolter_3(
20., 5., distance_between_the_foci)
print(wolter3.oe1.ccc_object.get_coefficients())
print(wolter3.oe2.ccc_object.get_coefficients())
#beam = wolter3.trace_wolter3(beam, z0)
beam = wolter3.trace_compound(beam)
beam.plot_xz()
beam.retrace(0.1)
beam.plot_xz()
plt.show()
def test_person(self):
print(">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> test_person")
beam = Beam.initialize_as_person()
if do_plot:
beam.plot_xz()
plt.show()
def test_wolter2_good_rays(self):
print(
">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> test_wolter2_good_rays"
)
p = 100. ##### if p=100 the trace_good_ray goes crazy
beam1 = Beam.initialize_as_person(10000)
#beam1 = Beam(100000)
#beam1.set_circular_spot(1.)
# beam1.set_rectangular_spot(5 / 2 * 1e-5, -5 / 2 * 1e-5, 5 / 2 * 1e-5, -5 / 2 * 1e-5)
beam1.x *= 10.
beam1.z *= 10.
beam1.set_point(p, 0., p)
op_ax = Beam(1)
op_ax.set_point(p, 0., p)
beam = op_ax.merge(beam1)
beam.set_divergences_collimated()
beam.plot_xz(0)
p = 20000.
q = 30.
z0 = 5.
focal = 2 * z0 + q
wolter2 = CompoundOpticalElement.initialiaze_as_wolter_2(p1=p,
q1=q,
z0=z0)
beam = wolter2.trace_good_rays(beam)
beam.plot_good_xz()
beam.retrace(10.)
beam.plot_good_xz()
plt.title("test_wolter2_good_rays")
print(beam.flag)
if do_plot:
plt.show()
def test_optimezed_wolter1_good_rays(self):
print(
">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> test_optimezed_wolter1_good_rays"
)
p = 100.
beam1 = Beam.initialize_as_person()
beam1.x *= 50.
beam1.z *= 50.
beam1.set_point(p, 0., p)
op_ax = Beam(1)
op_ax.set_point(p, 0., p)
beam = op_ax.merge(beam1)
beam.set_divergences_collimated()
beam.plot_xz()
p = 1e12
R = 100.
theta = 1e-3 * np.pi / 180
wolter1 = CompoundOpticalElement.initialiaze_as_wolter_1_with_two_parameters(
p1=p, R=R, theta=theta)
beam = wolter1.trace_good_rays(beam)
beam.plot_good_xz()
indices = np.where(beam.flag >= 0)
assert_almost_equal(beam.x[indices], 0., 8)
assert_almost_equal(beam.z[indices], 0., 8)
beam.retrace(100.)
beam.plot_good_xz()
plt.title("optimezed_wolter1_good_rays")
if do_plot:
plt.show()
from Beam import Beam import matplotlib.pyplot as plt import Shadow from Shadow.ShadowTools import plotxy beam = Beam.initialize_as_person() # beam.plot_xz() # # plt.show() # # initialize shadow3 source (oe0) and beam # beam_shadow = Shadow.Beam() oe0 = Shadow.Source() # # Define variables. See meaning of variables in: # https://raw.githubusercontent.com/srio/shadow3/master/docs/source.nml # https://raw.githubusercontent.com/srio/shadow3/master/docs/oe.nml # oe0.FDISTR = 1 oe0.FSOUR = 0 oe0.F_PHOT = 0 oe0.HDIV1 = 1e-12 oe0.HDIV2 = 1e-12 oe0.IDO_VX = 0 oe0.IDO_VZ = 0 oe0.IDO_X_S = 0
def test_wolter_1_microscope():
p = 0.
beam1 = Beam.initialize_as_person()
beam1.set_flat_divergence_with_different_optical_axis(0.005, 0.005)
beam1.set_point(p, 0., p)
op_ax = Beam (1)
op_ax.set_point(p, 0., p)
beam = op_ax.merge(beam1)
beam.x = beam.x
beam.z = beam.z
#beam.set_divergences_collimated()
beam.plot_xz()
distance_of_focalization = 5.
hyp = Optical_element.initialize_my_hyperboloid(p=distance_of_focalization, q=0., theta=0., alpha=0., wolter=1.1, z0=0., distance_of_focalization=distance_of_focalization)
ah = distance_of_focalization/np.sqrt(2)
q = 20.
z0 = 0.5*(q - np.sqrt(2)*ah)
c = z0 + np.sqrt(2)*ah
#b = c + 0.1
#a=np.sqrt(b**2-c**2)
b = c*1.5
a = np.sqrt(b**2-c**2)
ccc = np.array ([1/a**2, 1/a**2, 1/b**2, 0., 0., 0., 0., 0., -2*z0/b**2, z0**2/b**2-1])
ellips = Optical_element.initialize_as_surface_conic_from_coefficients(ccc)
hyp.effect_of_optical_element(beam)
beam.plot_yx(0)
plt.title("footprint %s" % (hyp.type))
ellips.p = 0.
ellips.theta = 90*np.pi/180
ellips.effect_of_optical_element(beam)
beam.plot_yx(0)
plt.title("footprint %s" % (ellips.type))
t = -beam.y/beam.vy
beam.x = beam.x + beam.vx * t
beam.y = beam.y + beam.vy * t
beam.z = beam.z + beam.vz * t
beam.plot_yx(0)
beam.plot_xz()
print("a of ellips is: %f" %(a))
print("b of ellips is: %f" %(b))
print("c of ellips is: %f" %(c))
print(np.mean(beam.z))
beam.histogram()
plt.show()
