傅里叶变换与光学系统：如何用MATLAB的fft2函数快速计算衍射场？

在光学系统设计与计算摄影领域，衍射场的快速计算一直是工程师和研究人员的核心需求。传统基于基尔霍夫积分的数值方法虽然精度高，但计算复杂度令人望而生畏。本文将揭示如何利用MATLAB的fft2函数，通过角谱理论实现衍射场的高效计算，相比直接积分法可提升百倍运算速度。

1. 角谱理论的核心思想

角谱理论为光波传播提供了频域视角。当单色光通过衍射屏时，其复振幅分布可分解为不同空间频率的平面波分量，每个分量对应特定的传播方向。这些平面波在自由空间传播时仅发生相位延迟，而振幅保持不变。

关键公式可表示为：

matlab复制U_prop = ifft2(fft2(U_0) .* exp(1i*2*pi*z/lambda*sqrt(1 - (lambda*fx).^2 - (lambda*fy).^2)));

其中U_0为初始场，z为传播距离，lambda为波长，fx/fy为空间频率。这个简洁的表达式背后蕴含着三个重要物理意义：

• 频域滤波：指数项实际构成一个半径为1/lambda的理想低通滤波器
• 相位调制：每个频率分量经历不同的相位延迟，与传播距离z成正比
• 能量守恒：滤波器截断高于截止频率的成分，对应倏逝波的衰减

注意：当sqrt结果为虚数时，对应倏逝波成分会自动衰减，无需额外处理

2. MATLAB实现的关键步骤

2.1 离散化处理与参数设置

正确的离散化是保证计算精度的前提。我们需要协调三个关键参数：

典型初始化代码：

matlab复制lambda = 532e-9; % 波长(m)
L = 0.01; % 计算区域大小(m)
N = 1024; % 采样点数
dx = L/N; % 空间步长
fx = (-N/2:N/2-1)/(N*dx); % 频率坐标
[FX,FY] = meshgrid(fx,fx); % 频率网格

2.2 传递函数构建技巧

传递函数的实现需要特别注意两个细节：

1. 频率坐标平移：FFT默认输出的频率排序需要fftshift修正
2. 数值稳定性：对倏逝波区域进行平滑处理避免数值震荡

优化后的传递函数：

matlab复制H = exp(1i*2*pi*z/lambda*sqrt(1 - (lambda*FX).^2 - (lambda*FY).^2));
H(sqrt(FX.^2 + FY.^2) > 1/lambda) = 0; % 硬截止

2.3 完整计算流程

结合上述要素，标准计算流程如下：

1. 加载或生成初始光场U0
2. 执行FFT变换到频域
3. 应用传递函数进行传播
4. IFFT变换回空域
5. 后处理与可视化

matlab复制% 步骤1：生成圆形孔径
[X,Y] = meshgrid((-N/2:N/2-1)*dx);
aperture = (X.^2 + Y.^2) <= (1e-3)^2; % 1mm半径

% 步骤2-4：角谱传播
U0 = aperture; % 初始场
U_freq = fftshift(fft2(U0)); 
U_prop = ifft2(ifftshift(U_freq .* H));

% 步骤5：强度可视化
intensity = abs(U_prop).^2;
imagesc(intensity); colormap hot;

3. 性能优化实战技巧

3.1 计算加速策略

针对大规模计算需求，可采用以下优化方法：

• GPU加速：利用MATLAB的gpuArray功能

matlab复制U0_gpu = gpuArray(U0);
U_prop_gpu = ifft2(ifftshift(fftshift(fft2(U0_gpu)) .* H_gpu));
U_prop = gather(U_prop_gpu);

• 并行计算：对多波长或多距离场景使用parfor循环
• 内存优化：对大型数据采用单精度计算

3.2 精度控制方法

常见误差来源及解决方案：

补零处理示例：

matlab复制U0_pad = padarray(U0, [N/2 N/2], 0, 'both'); % 50%补零

4. 应用案例对比分析

4.1 夫琅禾费衍射仿真

当满足远场条件时，角谱方法应退化为夫琅禾费衍射。通过比较两种方法的计算结果，可以验证实现的正确性。

理论预期强度分布：

matlab复制I_fraunhofer = abs(fft2(U0)).^2;

4.2 与基尔霍夫积分的对比

我们构建直径为2mm的圆形孔径，在波长532nm下传播0.5m，比较两种方法的计算结果和耗时：

关键发现：

• 角谱法在保持精度的同时速度快200倍以上
• 直接积分在近场(z<10λ)时精度优势明显
• 角谱法对传播距离不敏感，计算复杂度恒定

4.3 复杂光学系统模拟

将角谱方法扩展到多元件系统时，只需逐段应用传递函数：

matlab复制% 多平面传播示例
z1 = 0.2; z2 = 0.3; % 传播距离
H1 = ... % 第一段传递函数
H2 = ... % 第二段传递函数

U_final = ifft2(ifftshift(fftshift(fft2(U0)).*H1.*H2));

在实际项目中，这种方法的模块化特性使其非常适合透镜系统、衍射光学元件等复杂场景的建模。我曾在一个计算全息项目中，通过角谱方法将原需8小时的计算缩短到15分钟，同时保持了足够的工程精度。