clc
clear
lambda=1.064;
NA=0.55;
alpha=asin(NA);
f=4e3;
n1=1;
n2=2.585;    
d_p=590;
d=204;
thetax=linspace(0,alpha,100);
rho=sin(thetax)/sin(alpha);

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%设定参数
for i=1:length(thetax)
    theta=thetax(i);
    syms theta1
    A=d_p*n1*sin(theta1)./sqrt(n2^2-n1^2*sin(theta1).^2)-d*tan(theta1);
    eq=theta1+asin(cos(theta1)/f.*A)-theta==0;
    theta1_ans(i)=double(vpasolve(eq,theta1));
end

%%%%%%%%%%%解方程计算theta1和Theta关系
% 
% AA=d_p*n1*sin(theta1_ans)./sqrt(n2^2-n1^2*sin(theta1_ans).^2)-d*tan(theta1_ans);
% deta=theta1_ans+asin(cos(theta1_ans)/f.*AA)-thetax;
% plot(thetax,deta)
% %%%%%%%%%%%%%%%%%%%%%%%%%%检验方程的解

theta2=asin(n1*sin(theta1_ans)/n2);
phi_sita=n1*(f*cos(thetax)-d)./cos(theta1_ans)+n2*d_p./cos(theta2);
phi_0=n1*(f-d)+n2*d_p;
optical_dif=phi_sita-phi_0;  %光程差
phi_dif=-2*pi*optical_dif/lambda;  %相位差,需要补偿的相位差取相反数
phi_dif=phi_dif-min(phi_dif);   
phi_dif_mod=mod(phi_dif,2*pi);
plot(rho,phi_dif,rho,phi_dif_mod)
p=polyfit(rho,phi_dif,8);
% 
% rho0=linspace(0,1,100);
% phi_dif0=polyval(p,rho0);
% plot(rho,phi_dif,'-',rho0,phi_dif0,'--')
% % % %%%%%%%%%%%%%%%对rho--phi_dif关系用5阶多项式拟合，并检验拟合效果
% % % 
pixelx=1280;
pixely=1024;
F=1.2;
pixel=F*pixely;
%%%%%%%%%%%%%%%%%%%%%%%设置相位图像素数目，设置调制区域线性占比F(填充因子)

[x,y]=meshgrid(1:pixelx,1:pixely);
x=x-(pixelx+1)/2;
y=y-(pixely+1)/2;
D=(pixel-1)/2;
Aperture=(x.^2+y.^2)./D.^2; 
Aperture(Aperture<=1)=1;
Aperture(Aperture>1)=0;
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% 设置约束孔径Aperture
rho1=(sqrt(x.^2+y.^2)/D).*Aperture;
phi_dif1=polyval(p,rho1);   %计算（x，y）坐标对应的相位

Phase=mod(phi_dif1,2*pi).*Aperture;
PhaseHolo=round(Phase*255/(2*pi));

ApertureN=(x.^2+y.^2)./D.^2; 
ApertureN(ApertureN<=1)=0;
ApertureN(ApertureN>1)=1;
Noise=round(255*rand(pixely,pixelx)).*ApertureN;
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%生成0-255间随机噪声，并设置相反约束孔径ApertureN

PhaseHolo=PhaseHolo+Noise;%添加随机噪声后的相位图

figure(1)
imshow(PhaseHolo,[])
figure(2)
plot(linspace(-1,1,pixelx),PhaseHolo(512,:)*2*pi/255,'r')
axis([-1,1,0,2*pi])
filename=['C:\Users\simin\Desktop\球差校正\','F=',num2str(F),'.bmp'];
imwrite(uint8(PhaseHolo),filename);  %%%%%%%%%%%%%%%%%%%%%%%%%%导出8位相位图存储