def test_rectangular_shape(self):
beam = Beam(round(1e5))
plane_mirror = Optical_element.initialize_as_surface_conic_plane(
p=10., q=0., theta=0.)
beam.set_flat_divergence(0.02, 0.1)
xmax = 0.01
xmin = -0.0008
ymax = 1.
ymin = -0.29
bound = BoundaryRectangle(xmax=xmax, xmin=xmin, ymax=ymax, ymin=ymin)
plane_mirror.set_bound(bound)
beam = plane_mirror.trace_optical_element(beam)
beam.plot_xz()
beam.plot_good_xz()
indices = np.where(beam.flag > 0)
assert_almost_equal(max(beam.x[indices]) - xmax, 0., 2)
assert_almost_equal(-min(beam.x[indices]) + xmin, 0., 2)
assert_almost_equal(max(beam.z[indices]) + ymin, 0., 2)
assert_almost_equal(-min(beam.z[indices]) - ymax, 0., 2)
print(max(beam.x[indices]), min(beam.x[indices]), max(beam.y[indices]),
min(beam.y[indices]))
if do_plot is True:
plt.show()
def test_kirk_patrick_baez(self):
shadow_beam = shadow_source()
beam = Beam()
beam.initialize_from_arrays(
shadow_beam.getshonecol(1),
shadow_beam.getshonecol(2),
shadow_beam.getshonecol(3),
shadow_beam.getshonecol(4),
shadow_beam.getshonecol(5),
shadow_beam.getshonecol(6),
shadow_beam.getshonecol(10),
)
bound1 = BoundaryRectangle(xmax=2.5, xmin=-2.5, ymax=2.5, ymin=-2.5)
bound2 = BoundaryRectangle(xmax=1., xmin=-1., ymax=1., ymin=-1.)
kirk_patrick_baez = CompoundOpticalElement.initialize_as_kirkpatrick_baez(p=10., q=5., separation=4., theta=89*np.pi/180, bound1=bound1, bound2=bound2)
beam = kirk_patrick_baez.trace_compound(beam)
beam.plot_good_xz(0)
indices = np.where(beam.flag>0)
assert_almost_equal(beam.x[indices], 0., 4)
assert_almost_equal(beam.z[indices], 0., 4)
beam.retrace(50.)
beam.plot_good_xz()
print(kirk_patrick_baez.info())
if do_plot:
plt.show()
