clear all;
[y1,fs,bits]=wavread('F:\noise enhancing\wav\5.wav');
%[y1,fs,bits]=wavread('F:\語音降噪\wav\female.wav');
%y1=y1/max(abs(y1));%語音信號歸一化
wavwrite(y1,8000,8,'F:\noise enhancing\voise\5.wav');
%wavwrite(y1,8000,8,'F:\語音降噪\wav\sunshine2.wav');
figure(1);
plot(y1);

[noise,fs1,bits1]=wavread('F:\noise enhancing\wav\5_noise.wav');
%[noise,fs1,bits1]=wavread('F:\語音降噪\wav\pc_noise.wav');
y=mixsig(y1,noise,0);% 混合
y=y/max(abs(y));%歸一化
wavwrite(y,8000,8,'F:\noise enhancing\wav\mymasking_s&w(-10).wav');
figure(2);
plot(y);

frame = 256;    % Defining frame size
shift=64;
win=hamming(256);

for j1 = 1:length(y),
 signal(j1) = y(j1);
end;

ps_noise=zeros(length(signal)/frame,frame);
frame_temp = zeros(length(signal)/frame,frame);

hh = 0; 
   for k = 1 : 5,
       for l = 1 : frame,
          b(l) = signal(hh+l);
      end;
        hh = hh + frame;
        frame_temp(k,1:frame) = abs(fft(b));                      %fft for the first 50 frames
        ps_noise(k,1:frame) = (frame_temp(k,1:frame).*conj(frame_temp(k,1:frame)))/frame;
        %ps_noise(1,1:frame)= (sum(ps_noise(1:k,l))/20);          % Sum of the power spectral densities of samples within a frame

    end;%語音前五真早聲能量
    %ps_noise=zeros(length(signal)/frame,frame);
    %ps_noise(1,1:frame)= sum(A)/20;            % setting the threshold for the noise(frame noise)
    ps_noise(1,1:frame)= (sum(ps_noise(1:5,1:frame))/5);

head = 0; 
mm=1;
nn=1;
% START OF THE NOISE ELIMINATION THROUGH SPECTRAL SUBTRACTION BASED ON THE THRESHOLD SET

   for k = 1 :( length(signal)/frame),
        for m = 1 : frame,
          abc1(m) = signal(head+m);    
        end;
               
        head = head +frame;
        frame_temp(k,1:frame) = abs(fft(abc1));% FFT OF THE SIGNAL + NOISE FRAME BY FRAME
      
        frame_angle(k,1:frame) = angle(fft(abc1));% ANGLE OF FFT OF THE SIGNAL + NOISE FRAME BY FRAME

        ps_signal(k,1:frame) = (frame_temp(k,1:frame).*conj(frame_temp(k,1:frame)))./frame;
        ps_noise(k,1:frame)=ps_noise(1,1:frame);
       
      frame_ps(1,k) = (sum(ps_signal(k,1:frame)));%一陣內信號能量之和
      frame_pn(1,k) = sum(ps_noise(k,1:frame));   
      ps_final(1,k) = frame_ps(1,k)- frame_pn(1,k);      
       
      %ps_final(k,1:frame) = ps_signal(k,1:frame)-ps_noise(k,1:frame);
      %if ps_final(k,1:frame)<0
      %    ps_final(k,1:frame)=zeros(1,frame);
      %end     
      %frame1(k,1:frame)=sqrt(ps_final(k,1:frame));
      
       if ps_final(1,k)<0
           ps_final(1,k)=0;
       end 
      
       aa=1;bb=1;
       h(1,k)=power(ps_final(1,k)/(ps_final(1,k)+aa*frame_pn(1,k)),bb);
       frame_temp(k,1:frame) = h(1,k).*(frame_temp(k,1:frame));%為那濾波
        
       frame_temp(k,1:frame) = frame_temp(k,1:frame).*(exp(i*frame_angle(k,1:frame)));
       frame2(k,1:frame)=ifft(frame_temp(k,1:frame));
 
       signal(1,(((k-1)*frame)+1):(k*frame)) = frame2(k,1:frame); % Retriving back the signal(after spectral subtraction)
    end     
   
 signal((length(y)-1000):(length(y)))=[];   %give up 4 frames in the end
 y1((length(y1)-1000):(length(y1)))=[];
figure(3);
plot(1:length(frame_ps),frame_ps,1:length(ps_final),ps_final);
figure(4);
signal=signal';
signal=signal/max(abs(signal));%歸一化
plot(1:length(signal),signal);

overall_snr = 10*log10(sum(abs(y1).^2)/sum((abs(y1-signal)).^2))
wavwrite(signal,8000,8,'F:\noise enhancing\wav\mymasking21(0db).wav');