%****************************************************************
% 內容概述：TURBO譯碼性能AWGN信道測試（菜單式）
%          達到誤幀限即可停止當前SNR點的測試，節省計算量
% 創 建 人：朱殿榮/QQ:235347/MSN:njzdr@msn.com
% 單    位：南京郵電大學，通信工程系
% 創建時間：2005年9月12日
% 修改時間：2005年10月26日
% 參考文獻：《數字通信－－基礎與應用》
%          《High performace parallelised 3GPP Turbo Decoder》
%          《改進的Turbo碼算法及其FPGA實現過程的研究》,天津大學，張寧，趙雅興
%       	K.K.Loo, T.Alukaidey, S.A.Jimaa “High Performance Parallelized
%           3GPP Turbo Decoder”, Personal Mobile Communications
%       	Conference 2003. 5th European (Conf. Publ. No. 492)
%       	3GPP TS 25.212 V6.6.0 (2005-09)
%       	3GPP TS 25.222 V6.2.0 (2004-12) 
%       	劉東華。Turbo碼原理與應用技術。電子工業出版社，2004.1
% 版權聲明：任何人均可復制、傳播、修改此文件，同時需保留原始版權信息。
%****************************************************************
clc;
clear all;

algorithm = input('譯碼算法【1:LOG-MAP,2:MAX-LOG-MAP(缺省),3:TH-LOG-MAP)】:');
if isempty(algorithm)
   algorithm =2;
end

length_interleave = input('交織長度=幀長－尾比特長度【1024】:');
if isempty(length_interleave)
   length_interleave = 1024;
end

iter = input('迭代次數【[1,2,3]】:');
if isempty(iter)
   iter =[1 2 3];
end

ferrlim = input('誤幀限(達到此限即可停止當前SNR點的測試)【10】:');
if isempty(ferrlim)
   ferrlim =10;
end

max_EbNo = input('測試最大Eb/No（dB）【2.4】:');
if isempty(max_EbNo)
   max_EbNo =2.4;
end

step_EbNo = input('測試Eb/No步長（dB）【0.2】:');
if isempty(step_EbNo)
   step_EbNo=0.2;
end

save_mat = input('是否保存仿真結果到MAT文件 【1-保存（缺?。?0-不保存】:');
if isempty(save_mat)
   save_mat=1;
end

if save_mat==1
    matFileName = input('MAT文件名 【''臨時測試數據.mat''】:');
    if isempty(matFileName)
        matFileName='臨時測試數據.mat';
    end
end

fprintf('----------------------------------------------------\n'); 
fprintf(' 交織長度=%4dbit；迭代次數=%2d\n',length_interleave,iter);
fprintf(' 測試最大Eb/No = %2.1fdB；測試Eb/No步長 = %2.1fdB\n',max_EbNo,step_EbNo);
switch algorithm
    case 1
        fprintf(' 譯碼算法：LOG-MAP\n');
    case 2
        fprintf(' 譯碼算法：MAX-LOG-MAP\n');
    case 3
        fprintf(' 譯碼算法：門限MAX-LOG-MAP\n');
end
if save_mat==1
    fprintf(' 保存仿真結果到 = %4s\n',matFileName);
end    
fprintf('----------------------------------------------------\n'); 

time_begin=datestr(now);
rate=1/3;           %碼率
m=3;                    %尾比特數
fading_a=1;             %Fading amplitude
EbNo=0:step_EbNo:max_EbNo;                            %EbNo的采樣點
EbNoLinear=10.^(EbNo.*0.1);
num_block_size=length_interleave+m;     %測試的塊尺寸，指包含尾比特的軟輸入系統系列長度
err_counter=zeros(max(iter),length(EbNo));        %初始化錯誤比特計數器
nferr= zeros(max(iter),length(EbNo));             %初始化錯誤幀計數器
ber=zeros(max(iter),length(EbNo));                 %初始化錯誤比特率

random_in=round(rand(1,length_interleave));  %隨機數
[turbod_out,alphain]=turbo(random_in);      %編碼

for ii=1:length(iter)
    for nEN=1:length(EbNo)
        L_c=4*fading_a*EbNoLinear(nEN)*rate;
        sigma=1/sqrt(2*rate*EbNoLinear(nEN));
        nframe = 0;    % clear counter of transmitted frames
        if nEN==1 | ber(iter(ii),nEN-1)>9.0e-6
            while nferr(iter(ii),nEN)<ferrlim        %nferr:當前迭代次數、EbNo點的錯誤幀數
                nframe = nframe + 1; 
                noice=randn(4,num_block_size);    %噪聲
                soft_in=L_c*(turbod_out+sigma*noice);            %信息噪聲疊加
                [hard_out,soft_out]=decoder_all_algorithm(soft_in,alphain,iter(ii),algorithm); %譯碼
                errs=length(find(hard_out(1:length_interleave)~=random_in));%當前點錯誤bit數
                
                if errs>0 
                    err_counter(iter(ii),nEN)=err_counter(iter(ii),nEN)+errs;
                    nferr(iter(ii),nEN)=nferr(iter(ii),nEN)+1;
                end
                fprintf('當前EbNo點：%1.2fdB；已計算：%2.0f幀；其中：%2.0f誤幀\n',...
                    EbNo(nEN),nframe,nferr(iter(ii),nEN));
            end
            ber(iter(ii),nEN) = err_counter(iter(ii),nEN)/nframe/(length_interleave);%誤比特率
            %fer(iter,nEN) = nferr(iter,nEN)/nframe; %誤幀率
        else
            ber(iter(ii),nEN)=NaN;
        end
        fprintf('迭代次數：%1.0f；EbNo：%1.2fdB；誤碼率：%8.4e；\n',...
            iter(ii),EbNo(nEN),ber(iter(ii),nEN));
        if save_mat==1
            save (matFileName,'EbNo','ber');
        end
    end
end
%semilogy(EbNo,ber(1,:),EbNo,ber(2,:),EbNo,ber(3,:));
%xlabel('E_b/N_0 (dB)');
%ylabel('Bit Error Rate');
%title('3GPP標準 Max-Log-MAP譯碼算法,1024交織長度,1/3碼率');
%legend('1次迭代','2次迭代','3次迭代');

time_end=datestr(now);

fprintf('------------------恭喜你！測試完成！--------------------\n'); 
disp([' 仿真起始時間:',time_begin,'=>',time_end])
fprintf(' 交織長度=%4dbit；迭代次數=%2d\n',length_interleave,iter);
fprintf(' 測試最大Eb/No = %2.1fdB；測試Eb/No步長 = %2.1fdB\n',max_EbNo,step_EbNo);
switch algorithm
    case 1
        fprintf(' 譯碼算法：LOG-MAP\n');
    case 2
        fprintf(' 譯碼算法：MAX-LOG-MAP\n');
    case 3
        fprintf(' 譯碼算法：門限MAX-LOG-MAP\n');
end
if save_mat==1
    fprintf(' 保存仿真結果到 = %4s\n',matFileName);
end    
fprintf('-------------------------------------------------------\n');