gs 光束整形算法的示例代码matlab 做一个实例

以下是一个简单的 Matlab 实现示例：

% 定义光束整形函数
function output = beam_shaping(input, mask)
    % input: 输入光束
    % mask: 光束整形掩模
    
    % 将掩模按照输入光束的大小进行缩放
    mask_resized = imresize(mask, size(input));
    
    % 对输入光束进行傅里叶变换
    input_fft = fft2(input);
    
    % 将掩模应用于傅里叶变换后的光束
    shaped_fft = input_fft .* mask_resized;
    
    % 对应用掩模后的光束进行逆傅里叶变换
    output = ifft2(shaped_fft);
end

% 测试代码
% 创建一个 128x128 的随机输入光束
input = rand(128);

% 创建一个 128x128 的矩形掩模
mask = ones(128);
mask(50:80, 30:60) = 0;

% 应用光束整形函数
output = beam_shaping(input, mask);

% 显示结果
figure;
subplot(1, 2, 1);
imshow(input, []);
title('输入光束');
subplot(1, 2, 2);
imshow(output, []);
title('整形后的光束');

在这个示例中，我们定义了一个名为 `beam_shaping` 的函数，该函数接受两个参数：输入光束和光束整形掩模。函数将输入光束进行傅里叶变换，并将掩模应用于傅里叶变换后的光束。然后，函数对应用掩模后的光束进行逆傅里叶变换，并返回整形后的光束。最后，我们通过一个简单的测试代码来测试 `beam_shaping` 函数，并显示输入光束和整形后的光束。