%%?選擇衍射類型
yup=input（‘請選擇衍射屏類型：1-矩孔；2-圓孔；3-正弦條紋\n‘）;
R=input（‘請輸入半徑（可取50單位um）\n‘）;
N=input（‘請輸入衍射屏長寬（可取400，單位um）\n‘）;
Z=input（‘請輸入衍射屏與像屏距離（可取0.02，單位m）\n‘）;
switch（yup）
????case?（1）????%?繪制矩孔??
????????T=zeros（NN）;
????????[xy]=meshgrid（linspace（-N/2N/2N））;
????????T（find（x>-R&x-R&y????????
????case（2）??%?繪制圓孔
????????????T=zeros（NN）;
????????????[xy]=meshgrid（linspace（-N/2N/2N））;
????????????D=（（x）.^2+（y）.^2）.^（1/2）;
????????????xx=D<=R;
????????????T（xx）=1;

????case（3）??%?繪制正弦條紋刻線
????????????T=zeros（NN）;
????????????[xy]=meshgrid（linspace（-N/2N/2N））;
????????????T=1+sin（x*pi/20）;%條紋周期為40um
end

%%?定義初始量
syms?lamda?x0?y0?z?A
lamda=0.6328;??%確定波長um
z=Z*1000000;???%確定衍射屏與像屏距離um
k=2*pi/lamda;??????
A=1;??%定義單色平面波的強度?

%%?出射光場及其頻譜
figu