def thick_lens_R(R1, R2, d, n, n0=1):
phi=1/R1*(n-n0)+1/R2*(n0-n)+d/n*1/R1*1/R2*(n-n0)**2
return phi
# basic ABCD Matrix
def ABCD(a,b,c,d):
return Matrix([[a,b],[c,d]])
# Some specific examples
def free_space_matrix(d):
return ABCD(1, d, 0, 1)
def flat_interface_matrix(n1,n2):
return ABCD(1,0,0,n1/n2)
def curved_interface_matrix(R, n1, n2):
return ABCD(1, 0, (n1-n2)/(R*n2), n1/n2)
def curved_mirror_matrix(R):
return ABCD(1, 0, -2/R, 1)
def thin_lens_matrix(f):
return ABCD(1, 0, -1/f, 1)
def u_hat_out(u_hat_in, y_k, phi_k ):
return u_hat_in-y_k*phi_k
def y_next(y_k, u_hat_out, d_k):
return y_k+u_hat_out*d_k
d0=symbols('d0')
ray_tracing_table={
"surface":[0,1,2,3],
"phi": [0, 1/8, 0, 1/2],
"d": [8.4, 168, 2, 0],
"y":[0,np.nan,np.nan,np.nan],
"u":[10,np.nan,np.nan,np.nan]
}
ray_tracing_df=pd.DataFrame(ray_tracing_table)
ray_tracing_df
# todo
# fill the nan
def solve_conjugate_condition(matrix):
return partial(solve, matrix[1])
def solve_object_at_front_focal(matrix):
return partial(solve, matrix[3])
def solve_image_at_rear_focal(matrix):
return partial(solve, matrix[0])
def solve_afocal_condition(matrix):
return partial(solve, matrix[2])
def solve_image_location(M, object_distance):
return -(object_distance*M[0]-M[1])/(object_distance*M[2]-M[3])
def one_full_round_trip_of_optical_resonator(r, d):
half_trip= ABCD(1, 0, -2/r, 1)* ABCD(1, d, 0, 1)
return half_trip**2
