function output=capacity_water(SNR,M)

%**************************************************************************
%This program implements equation (2.26) using the waterfilling
%algorithm.The equation has been implemented in its determinant form as it
%is more convenient.
% output=capacity_water(SNR,M)
% SNR is signal-to-noise ratio
% M -> number of antennas for a M x M system
% output -> the ergodic capacity in case this program is required to work
% with a script file to plot Fig 2.8.
%**************************************************************************

%Monte-Carlo runs
% SNR=10;
% M=2;
for K=1:10000
      T=randn(M,M)+j*randn(M,M);
      T=0.707*T;
         I=eye(M);
         eigen=eig(T*T'); %extract eigenvalues
         gamma=zeros(M,1);
         r=M;%set rank = number of antennas (full rank)
         p=1;%initial count
     for i=1:r
         mu=getmu(r,SNR,T,p,M);%determine mu value (see eqn. (2.29))
         gamma(i)=mu-(M/SNR)*(1/eigen(i)); %calculate gamma (see eqn (2.30))
         if gamma(i)<0
             gamma(i)=0;% if gamma < 0, set it to zero i.e. discard it
             p=p+1;% increment count
             mu=0;%clear register
         else
            mu=0; % if gamma >0 store it and clear register
         end
    end
%solve eqn (1.26)using the determinant form
    a=I+(SNR/M)*diag(gamma).*diag(eigen);
     a=det(a);
	 y(K)=log2(a);   
end
    [n1 x1]=hist(y,40);
    n1_N=n1/max(K);
    a=cumsum(n1_N);
%plot CDF    
   plot(abs(x1),a,'b');

% uncomment the following lines for ergodic capacity   
%    b=abs(x1);
%    output=interp1q(a,b',0.5);