%陷波器實驗 
%輸入信號：x  正弦信號：s   噪聲信號： v  輸出信號：e
%輸入信號功率：Px  正弦信號功率：Ps  噪聲信號功率：Pv

clear;
close all;

N=20;                           %FIR濾波器長度
M=400;                          %信號長度
a=1;                            %噪聲幅值
W0=0.1*pi;                      %正弦信號頻率

v=a*[rand(1,M)-rand(1,M)];     
Pv=a^2/3;
Ps=10*Pv;
s=sin(W0*(1:M));   
A=sqrt(M*Ps/(s*s'));
s=A*s;
x=s+v;
Px=Ps+Pv;
u=1/(10*N*Px*1.1);
x1=[0,x(1:M-1)];               %延時信號
[w,y]=lms(x1,x,u,N);           %調用LMS算法
e=x-y;
figure(1);
subplot(211);plot(s);title('正弦信號s[n]');
subplot(212);plot(x);title('正弦加噪信號x[n]');

% 求系統的頻率響應
fftn=1024;
H=fft(w,fftn);
n=[1:fftn/2]./(fftn/2);
figure(2);
plot(n,1-abs(H([1:fftn/2]))),title('幅頻響應');grid on;

figure(3);
plot(v,'g-');
hold on;
plot(e,'m-');
legend('白噪聲v[n]','系統輸出信號e[n]');