def test_boundary_condition():
#beam1 = Beam(10000)
#beam1.set_point(0, 0, 0)
#beam1.set_flat_divergence(5e-3, 5e-2)
shadow_beam = run_shadow_source()
beam1 = Beam(10000)
beam1.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),
0
)
bound1=BoundaryRectangle(xmax=0.005,xmin=-0.005,ymax=0.05,ymin=-0.05)
bound2=BoundaryRectangle(xmax=0.01,xmin=-0.01,ymax=0.1,ymin=-0.1)
plane_mirror=Optical_element.initialize_as_plane_mirror(2,1,65*np.pi/180,0)
parabolic_mirror=Optical_element.initialize_as_surface_conic_paraboloid_from_focal_distances(5,2,28*np.pi/180,90*np.pi/180)
plane_mirror.rectangular_bound(bound1)
parabolic_mirror.rectangular_bound(bound2)
beam1=plane_mirror.trace_optical_element(beam1)
beam1=parabolic_mirror.trace_optical_element(beam1)
beam1.plot_xz()
plt.title("Total points plot")
beam1.plot_good_xz()
plt.title("Good points plot")
print(beam1.flag)
indices=np.where(beam1.flag>0)
print("The good number of ray are: %f" %(beam1.flag[indices].size))
plt.show()
shadow_beam=trace_shadow(shadow_beam)
assert_almost_equal(beam1.x, shadow_beam.getshonecol(1), 8)
assert_almost_equal(beam1.y, shadow_beam.getshonecol(2), 8)
assert_almost_equal(beam1.z, shadow_beam.getshonecol(3), 8)
def initialiaze_as_wolter_2(cls,p1,q1,z0):
#q1 = - q1
focal = q1+2*z0
print("focal=%f" %(focal))
theta1 = 0.
alpha1 = 0.
oe1 = Optical_element.initialize_as_surface_conic_paraboloid_from_focal_distances(p1,0.,theta1,alpha1,"p", focal)
oe2 = Optical_element.initialize_my_hyperboloid(p=0. ,q=-(focal-2*z0), theta=90*np.pi/180, alpha=0, wolter=2, z0=z0, distance_of_focalization=focal)
return CompoundOpticalElement(oe_list=[oe1,oe2],oe_name="Wolter 2")
def initialiaze_as_wolter_1(cls,p1,q1,z0):
theta1 = 0.
alpha1 = 0.
print(q1)
print(2*z0)
print("dof=%f" %(2*z0-q1))
#oe1 = Optical_element.initialize_as_surface_conic_paraboloid_from_focal_distances(p1,0.,theta1,alpha1,"p",2*z0-q1)
oe1 = Optical_element.initialize_as_surface_conic_paraboloid_from_focal_distances(p1, 0., theta1, alpha1, "p", 2*z0-q1)
#oe2 = Optical_element.initialize_my_hyperboloid(p=0.,q=q1,theta=90*np.pi/180,alpha=0,wolter=1, z0=z0, distance_of_focalization=2*z0-q1)
oe2 = Optical_element.initialize_my_hyperboloid(p=0., q=q1, theta=90 * np.pi / 180, alpha=0, wolter=1, z0=z0,distance_of_focalization=2*z0-q1)
return CompoundOpticalElement(oe_list=[oe1,oe2],oe_name="Wolter 1")
def initialize_as_wolter_3(cls, p, q, distance_between_the_foci):
f=-q
oe1 = Optical_element.initialize_as_surface_conic_paraboloid_from_focal_distances(p=p, q=f, theta=0., alpha=0., infinity_location="p")
#c = z0+np.abs(f)
c = distance_between_the_foci/2
z0 = np.abs(c)-np.abs(f)
b = c+100
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])
oe2 = Optical_element.initialize_as_surface_conic_from_coefficients(ccc)
oe2.set_parameters(p=0., q=z0+z0+np.abs(q), theta=90*np.pi/180)
return CompoundOpticalElement(oe_list=[oe1,oe2],oe_name="Wolter 3")
def initialiaze_as_wolter_1_with_two_parameters(cls,p1, R, theta):
cp1 = -2 * R / np.tan(theta)
cp2 = 2 * R * np.tan(theta)
cp = max(cp1, cp2)
f = cp / 4
print("focal=%f" % (f))
oe1 = Optical_element.initialize_as_surface_conic_paraboloid_from_focal_distances(p=p1, q=f, theta=0., alpha=0.,infinity_location="p")
s1 = R / np.tan(2 * theta)
s2 = R / np.tan(4 * theta)
c = (s1 - s2) / 2
z0 = f + c
b1 = np.sqrt(
0.5 * c ** 2 + 0.5 * R ** 2 + 0.5 * R ** 4 / cp ** 2 - R ** 2 * z0 / cp + 0.5 * z0 ** 2 - 0.5 / cp ** 2 * np.sqrt(
(
-c ** 2 * cp ** 2 - cp ** 2 * R ** 2 - R ** 4 + 2 * cp * R ** 2 * z0 - cp ** 2 * z0 ** 2) ** 2 - 4 * cp ** 2 * (
c ** 2 * R ** 4 - 2 * c ** 2 * cp * R ** 2 * z0 + c ** 2 * cp ** 2 * z0 ** 2)))
b2 = np.sqrt(
0.5 * c ** 2 + 0.5 * R ** 2 + 0.5 * R ** 4 / cp ** 2 - R ** 2 * z0 / cp + 0.5 * z0 ** 2 + 0.5 / cp ** 2 * np.sqrt(
(
-c ** 2 * cp ** 2 - cp ** 2 * R ** 2 - R ** 4 + 2 * cp * R ** 2 * z0 - cp ** 2 * z0 ** 2) ** 2 - 4 * cp ** 2 * (
c ** 2 * R ** 4 - 2 * c ** 2 * cp * R ** 2 * z0 + c ** 2 * cp ** 2 * z0 ** 2)))
b = min(b1, b2)
a = np.sqrt(c ** 2 - b ** 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])
oe2 = Optical_element.initialize_as_surface_conic_from_coefficients(ccc)
oe2.set_parameters(p=0., q=z0+c, theta=90*np.pi/180, alpha=0., type="My hyperbolic mirror")
#oe2.type = "My hyperbolic mirror"
#oe2.p = 0.
#oe2.q = z0 + c
#oe2.theta = 90 * np.pi / 180
#oe2.alpha = 0.
return CompoundOpticalElement(oe_list=[oe1,oe2],oe_name="Wolter 1")
def initialize_as_montel_parabolic(cls, p, q, theta, bound1, bound2, distance_of_the_screen=None, angle_of_mismatch=0.):
beta = (90. - angle_of_mismatch)*np.pi/180 #### angle beetween the two mirror, if angle_of_mismatch is >0 the two mirror are closer
oe1 = Optical_element.initialize_as_surface_conic_paraboloid_from_focal_distances(p=p, q=q, theta=theta, alpha=0., infinity_location='p', focal=q, cylindrical=1)
oe1.set_bound(bound1)
oe2 = oe1.duplicate()
oe2.rotation_surface_conic(beta, 'y')
oe2.set_bound(bound2)
if distance_of_the_screen == None:
distance_of_the_screen = q
ccc = np.array([0., 0., 0., 0., 0., 0., 0., 1., 0., -distance_of_the_screen])
screen = Optical_element.initialize_as_surface_conic_from_coefficients(ccc)
screen.set_parameters(p, q, 0., 0., "Surface conical mirror")
return CompoundOpticalElement(oe_list=[oe1, oe2, screen], oe_name="Montel parabolic")
def initialiaze_as_wolter_12(cls,p1,q1,focal_parabola,Rmin):
focal = focal_parabola
d = q1 - focal_parabola
z0 = focal_parabola + d/2
print("focal=%f" %(focal))
theta1 = 0.
alpha1 = 0.
oe1 = Optical_element.initialize_as_surface_conic_paraboloid_from_focal_distances(p1,0.,theta1,alpha1,"p", focal)
ccc = oe1.ccc_object.get_coefficients()
cp = -ccc[8]
print("R=%f, d=%f, cp=%f, z0=%f" %(Rmin,d,cp,z0))
#b1 = np.sqrt(0.125*d**2+Rmin+2*Rmin**4/cp**2-2*Rmin**2*z0/cp+0.5*z0**2-0.125/cp**2*np.sqrt((-cp**2*d**2-8*cp**2*Rmin-16*Rmin**4+16*cp*Rmin**2*z0-4*cp*z0**2)**2-16*cp**2*(4*d**2*Rmin**4-4*cp*d**2*Rmin**2*z0+cp**2*d**2*z0**2)))
#b2 = np.sqrt(0.125*d**2+Rmin+2*Rmin**4/cp**2-2*Rmin**2*z0/cp+0.5*z0**2+0.125/cp**2*np.sqrt((-cp**2*d**2-8*cp**2*Rmin-16*Rmin**4+16*cp*Rmin**2*z0-4*cp*z0**2)**2-16*cp**2*(4*d**2*Rmin**4-4*cp*d**2*Rmin**2*z0+cp**2*d**2*z0**2)))
p1 = -cp ** 2 * d ** 2 - 8 * cp ** 2 * Rmin - 16 * Rmin ** 4 + 16 * cp * Rmin ** 2 * z0 - 4 * cp ** 2 * z0 ** 2
p1 = p1**2
p2 = 16 * cp ** 2 * (4 * d ** 2 * Rmin ** 4 - 4 * cp * d ** 2 * Rmin ** 2 * z0 + cp ** 2 * d ** 2 * z0 ** 2)
sp = 0.125/cp**2*np.sqrt(p1-p2)
sp0 = 0.125*d**2+Rmin+2*Rmin**4/cp**2-2*Rmin**2*z0/cp+0.5*z0**2
b = np.sqrt(sp0-sp)
a = np.sqrt(d**2/4-b**2)
print("a=%f, b=%f" %(a,b))
#oe2 = Optical_element.initialize_my_hyperboloid(p=0. ,q=-(focal-2*z0), theta=90*np.pi/180, alpha=0, wolter=1.1, z0=z0, distance_of_focalization=focal)
cc = np.array([-1/a**2, -1/a**2, 1/b**2, 0., 0., 0., 0., 0., -2*z0/b**2, (z0/b)**2-1])
oe2 = Optical_element.initialize_as_surface_conic_from_coefficients(cc)
oe2.type = "My hyperbolic mirror"
oe2.set_parameters(p=0., q=q1, theta=90.*np.pi/180, alpha=0.)
return CompoundOpticalElement(oe_list=[oe1,oe2],oe_name="Wolter 1.2")
beam=beam1.merge(beam2) beam3=Beam() beam3.set_divergences_collimated() beam3.set_point(0.+100,0.,0.+100) beam3.set_rectangular_spot(5.,-5.,10.,-40.) beam=beam.merge(beam3) beamd=beam.duplicate() p = 10. q = 25. theta = 44 * np.pi / 180 alpha = 90 * np.pi / 180 prova = Optical_element.initialize_as_surface_conic_paraboloid_from_focal_distances(p,q,theta,alpha,"p") beamd=prova.trace_optical_element(beamd) print(prova.ccc_object.get_coefficients()) t = (100-beam.y)/beam.vy beamd.x = beamd.x + beamd.vx * t beamd.y = beamd.y + beamd.vy * t beamd.z = beamd.z + beamd.vz * t beamd.plot_xz() beam.plot_xz() beam.plot_xpzp()
beta = (90. + 0.) * np.pi / 180
alpha = 87. * np.pi / 180
xmax = 0.
xmin = -0.4
ymax = 0.4
ymin = -0.4
zmax = 0.4
# print("zmax = %f" %(zmax))
# zmax = 0.4
zmin = 0.
oe1 = Optical_element.initialize_as_surface_conic_paraboloid_from_focal_distances(
p=p,
q=q,
theta=theta,
alpha=0.,
infinity_location="p",
focal=q,
cylindrical=1)
ccc1 = oe1.ccc_object.get_coefficients()
print(ccc1)
c1 = ccc1[1]
c2 = ccc1[2]
c4 = ccc1[4]
c8 = ccc1[8]
####### rotation of the oe around y #############################################################################
a = np.cos(beta)
b = np.sin(beta)
def test_paraboloid_mirror(self):
print(">>>>>>>>>>>>>>> test_paraboloid_mirror")
#beam1=Beam(5000)
#beam1.set_point(0,0,0)
#beam1.set_flat_divergence(5e-3,5e-2)
shadow_beam = run_shadow_source()
beam1 = Beam()
beam1.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),
)
p = 10.
q = 20.
theta = 72 * np.pi / 180
alpha = 0 * np.pi / 180
spherical_mirror = Optical_element.initialize_as_surface_conic_paraboloid_from_focal_distances(
p, q, theta, alpha)
beam1 = spherical_mirror.trace_optical_element(beam1)
if do_plot:
beam1.plot_xz()
beam1.plot_xpzp()
plt.title("Paraboloid mirror with p=10, q=20, theta=72")
print(spherical_mirror.ccc_object.get_coefficients())
plt.show()
shadow_beam = run_shadow_parabolic_mirror(shadow_beam)
assert_almost_equal(beam1.x, shadow_beam.getshonecol(1), 7)
assert_almost_equal(beam1.y, shadow_beam.getshonecol(2), 7)
assert_almost_equal(beam1.z, shadow_beam.getshonecol(3), 7)
######## This is problematic #######################################################################################
#
#def test_hyperboloid_mirror():
# #beam1=Beam(5000)
# #beam1.set_point(0,0,0)
# #beam1.set_flat_divergence(5e-3,5e-2)
#
# shadow_beam=run_shadow_source()
#
# beam1=Beam(5000)
# beam1.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),
# 0
# )
#
# p=1.
# q=2.
# theta = 76*np.pi/180
# spherical_mirror=Optical_element.initialize_as_hyperboloid_from_focal_distances(p,q,theta)
# beam1=spherical_mirror.trace_surface_conic(beam1)
# beam1.plot_xz()
# beam1.plot_xpzp()
# plt.show()
#
# shadow_beam=run_shadow_hyperbolic_mirror(shadow_beam)
#
#
########################################################################################################################