GS算法MATLAB全息图生成代码

GS算法（Gerchberg-Saxton algorithm），也称为迭代相位恢复法，是一种经典的全息图重构技术，用于从幅度信息恢复出原始物体的三维信息。在MATLAB中，你可以使用内置函数结合自定义循环来实现这个过程。下面是一个简单的示例代码：

function hologram = generate_hologram(obj_amplitude, wavefront_phase)
    % 参数设置
    max_iterations = 50; % 最大迭代次数
    convergence_threshold = 1e-6; % 相似度阈值

    % 初始化全息图和迭代变量
    hologram = obj_amplitude;
    prev_phase = zeros(size(hologram));
    
    for iter = 1:max_iterations
        % 直接波前计算
        direct_wavefront = exp(1i * wavefront_phase);

        % 反射波前计算
        reconstructed_obj = ifftshift(ifft2(hologram .* direct_wavefront));

        % 计算重建物体与原物体的相似度
        similarity = norm(reconstructed_obj - obj_amplitude, 'fro') / norm(obj_amplitude, 'fro');

        % 检查是否达到收敛
        if similarity < convergence_threshold
            break;
        end

        % 更新相位估计
        wavefront_phase = angle(ifftshift(fft2(obj_amplitude ./ reconstructed_obj)));
        
        % 更新全息图
        hologram = abs(ifftshift(fft2(reconstructed_obj)));
    end
end

% 使用示例
obj_amplitude = randn(size([512 512])); % 假设这是物体的幅度数据
wavefront_phase = randn(size(obj_amplitude)); % 初始相位分布
hologram = generate_hologram(obj_amplitude, wavefront_phase);

%