matlab：光场传输算法 快速傅里叶变换算法(FFT)代码

以下是一个简单的 MATLAB 代码示例，演示如何使用FFT算法进行光场传输模拟：

% 定义光波长和像平面尺寸
wavelength = 0.6328e-6;
L = 0.05;

% 定义像平面上采样点数
N = 256;

% 生成像平面坐标
x = linspace(-L/2, L/2, N);
y = linspace(-L/2, L/2, N);
[X, Y] = meshgrid(x, y);

% 定义物平面上的光强分布
u = exp(-(X.^2 + Y.^2)/(2*(0.01)^2));

% 计算傅里叶变换
U = fftshift(fft2(u));

% 计算频域坐标
dx = x(2) - x(1);
fx = linspace(-1/(2*dx), 1/(2*dx), N);
fy = fx;
[Fx, Fy] = meshgrid(fx, fy);

% 计算衍射函数
H = exp(1i*pi*wavelength*(Fx.^2 + Fy.^2));

% 计算像平面光强分布
Uout = ifft2(ifftshift(U.*H));

% 绘制结果
figure;
subplot(1,2,1); imshow(u, []); title('物平面');
subplot(1,2,2); imshow(abs(Uout).^2, []); title('像平面');

该代码先定义了光波长和像平面尺寸，然后生成像平面坐标和物平面上的光强分布。接着使用FFT算法计算光强分布的傅里叶变换，并计算频域坐标和衍射函数。最后，使用逆FFT算法计算像平面光强分布，并绘制结果。

需要注意的是，这只是一个简单的示例，实际的光场传输模拟可能需要更复杂的算法和更高的计算精度。