Capon算法信号方向估计MATLAB代码

clear all
close all
derad = pi/180;            %角度->弧度
radeg = 180/pi;            %弧度->角度
twpi=2*pi;
kelm = 8;                  %阵元数
dd=0.5;                    %阵元间距
d=0:dd:(kelm-1)*dd;        
iwave = 5;                 %信源数
theta =[10 20 30 40 50];   %波达方向
snr = [30, 25, 20, 15, 5]; %信噪比
n=500;                     %采样数（快拍）
A=exp(-1i*twpi*d.'*sin(theta*derad));  %方向矢量
S=randn(iwave,n );                     %信源信号

for isnr=1:5
    X0=A*S;                              %接收信号
    X=awgn(X0,snr(isnr),'measured') ;     %添加噪声
    Rxx=X*X'/n;                          %计算协方差矩阵
    estimates = capon(Rxx,dd,iwave);     %调用子程序
    doaes(isnr,:)=sort(estimates);
end

disp(doaes);

figure(1)
polarplot(doaes(1,1)*pi/180,1,'*',doaes(1,2)*pi/180,1,'square',doaes(1,3)*pi/180,1,'d',doaes(1,4)*pi/180,1,'o',doaes(1,5)*pi/180,1,'+');
grid on;
title('Capon-SNR=30');
hold on;
drawnow;

function estimate = capon(cr,dd,Le)
twpi =2.0*pi;
derad = pi / 180.0;
radeg = 180.0 / pi;
%计算空间谱
[K,~] = size(cr);
G=zeros(K,Le);
for k=1:K
    for l=1:Le
        G(k,l) = exp(-1i*twpi*dd*(l-1)*sin((k-1)*derad));
    end
end
S=G*inv(cr)*G';
%对空间谱取逆变换得到权值系数
h=ifft(S);
%取权值系数的反转序列，估计信号方向
[~,I]=max(abs(h));
estimate=(I-1)*radeg;
end

代码中首先定义了几个参数，包括阵元数、阵元间距、信源数、波达方向、信噪比、采样数等。然后根据这些参数生成了接收信号，并通过调用 capon 函数进行信号方向估计。

capon 函数根据接收信号的协方差矩阵计算空间谱，然后对空间谱进行逆变换得到权值系数，最后通过对权值系数进行反转得到信号方向估计结果。

最后，代码将估计结果绘制在极坐标图上，并根据信噪比不同进行分类显示。

Capon算法是一种基于最小方差准则的信号方向估计方法，它能够有效地抑制噪声和干扰的影响，提高信号方向估计的精度。

代码使用说明:

将代码保存为 .m 文件，例如 capon_doa.m
在MATLAB中运行该文件
观察结果，可通过修改代码中的参数调整实验条件

代码改进方向:

可以增加对不同信噪比条件下的性能分析
可以将代码封装成更易用的函数
可以引入更多先进的信号方向估计算法，例如 MUSIC 算法

希望这份代码能够帮助您学习并理解 Capon 算法。