modelisationanalysespectrale/rossignol/parametrique/mylevinsondurbin.m

58 lines
1.7 KiB
Mathematica
Raw Normal View History

2019-10-12 21:41:54 +00:00
%%% Algorithme de Levinson-Durbin pour la d\'etermination des param\`etres AR
%%%
%%% entr\'ees :
%%% - xx : signal
%%% - pp : ordre du mod\`ele AR (choisi de mani\`ere ind\'ependante)
%%% - fe : fr\'equence d'\'echantillonnage
%%%
%%% sorties :
%%% - aa : les param\`etres AR
%%% - sigma2 : variance du bruit
%%% - ref : coefficients de r\'eflexion
%%% - ff : fr\'equences auxquelles la dsp a \'et\'e calcul\'ee
%%% - mydsp : la dsp elle-m\^eme
%%%
%%% exemple : fe=32000;f0=440;xx=cos(2*pi*f0/fe*[1:1280]+2*pi*rand(1,1));mylevinsondurbin(xx,4,fe);
%%%
%%% S. Rossignol -- 2012
function [aa, sigma2, ref, ff, mydsp] = mylevinsondurbin (xx, pp, fe)
acf = xcorr(xx, pp+1, 'biased'); %% autocorr\'elation
acf(1:pp+1) = []; %% on enl\`eve la partie n\'egative
acf(1) = real(acf(1)); %% Levinson-Durbin requiert c(1)==conj(c(1))
ref = zeros(pp,1);
gg = -acf(2)/acf(1);
aa = [ gg ];
sigma2 = real( ( 1 - gg*conj(gg)) * acf(1) ); %% real : enl\`eve une \'eventuelle
%% partie imaginaire r\'esiduelle
ref(1) = gg;
for tt = 2 : pp
gg = -(acf(tt+1) + aa * acf(tt:-1:2)') / sigma2;
aa = [ aa + gg*conj(aa(tt-1:-1:1)), gg ];
sigma2 = sigma2 * ( 1 - real(gg*conj(gg)) );
ref(tt) = gg;
end;
aa = [1, aa];
%%% densit\'e spectrale de puissance
interm2=-j*2*pi/fe*[1:pp];
df=0.9765625; %%% la dsp est calcul\'ee tous les df Hz
ff=-fe/2:df:fe/2;
interm3=interm2'*ff;
interm=1.+aa(2:pp+1)*exp(interm3);
mydsp = sigma2./(interm.*conj(interm));
% figure(1);
% clf;
% grid on;
% hold on;
% plot(ff,mydsp,'linewidth',2);
% xlabel('frequency (in Hz)','fontsize',20);
% ylabel('magnitude','fontsize',20);
% hold off;
% drawnow;