Python光学仿真之对光的干涉理解学习

import numpy as np
import matplotlib.pyplot as plt
#两束光叠加
waveAdd = lambda I1,I2,theta : I1+I2+2*np.sqrt(I1*I2)*np.cos(theta)
#同一束光经过小孔之后的光程差，dSlit为小孔间距,dWave为波长
#point为衍射屏上某点，theta为衍射屏相对小孔所在直线倾角
def interSlit(dSlit,point=(1,1,1),n=1,dWave=1.06e-6):
 pVar = sum(np.array(point)**2) + 0.25*dSlit**2
 delt = 2*np.pi*n/dWave*(
  np.sqrt(pVar+point[0]*dSlit)-
  np.sqrt(pVar-point[0]*dSlit))
 return delt
#杨氏干涉，只考虑x方向
#dSlit：小孔间距；dScreen：衍射屏距离
#nGrid：网格个数；dGrid：网格间距
def interYang(dSlit=1e-3,dScreen=1,n=1,I1=1,I2=1,
  dWave=1.06e-6,nGrid=500,dGrid=1e-5):
 xAxis = np.arange(-nGrid,nGrid+1)*dGrid
 I = [waveAdd(I1,I2,interSlit(
dSlit,(x,0,dScreen),n,dWave)) 
for x in xAxis]
 plt.plot(xAxis,I)
 plt.show()

#alpha为双镜锐角，dSA为光源到劈点距离，dScreen为S像点连线到干涉屏距离
def lensFresnel(alpha=0.1,dSA=1e-3,dScreen=1,
 n=1,I1=1,I2=1,dWave=1.06e-6,
 nGrid=500,dGrid=1e-4):
 dSlit = 2*dSA*np.sin(alpha)
 interYang(dSlit,dScreen,n,I1,I2,dWave,nGrid,dGrid)