% -------------------------------------------------------------------------------------------------
% ResultTest.m: Classification of all cases for event A
%               using a linear discriminant function in the (RR, Fm) plane,
%               or in the (HR, Fm) plane.
%
% Copyright (C) 2004 Maurizio Varanini, Clinical Physiology Institute, CNR, Pisa, Italy
%
% This program is free software; you can redistribute it and/or modify it under the terms
% of the GNU General Public License as published by the Free Software Foundation; either
% version 2 of the License, or (at your option) any later version.
%
% This program is distributed "as is" and "as available" in the hope that it will be useful,
% but WITHOUT ANY WARRANTY of any kind; without even the implied warranty of MERCHANTABILITY
% or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
%
% You should have received a copy of the GNU General Public License along with this program;
% if not, write to the Free Software Foundation, Inc.,
% 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
%
% For any comment or bug report, please send e-mail to: maurizio.varanini@ifc.cnr.it
% -------------------------------------------------------------------------------------------------

function ResultTest(MainDataDir)

if(nargin < 1)  MainDataDir='..\..\DatiQc'; end

% List of cases of the test set whose type turned out 
% from the previous classification submissions
casinotiaT={'a19', 'a29'};
casinotiaN={'a10', 'a15', 'a22', 'a26', 'a28'};

casinotibT={'b07', 'b17'};
casinotibS={'b02', 'b10'};

% -----------------------------------------------------------------------------------------
FpTh=5.5;
% line in the plane RR, Fm
% The following points are chosen using all the known cases
RRr(1)=1.61; Fmr(1)=3;
RRr(2)=0.2; Fmr(2)=7.2;
% compute slope and intercept of the line through two points 
% in the plane (RR, Fm).
r_ang= (Fmr(2)-Fmr(1))/(RRr(2)-RRr(1));
r_int= Fmr(1) - r_ang* RRr(1);

% line in the plane HR, Fm 
HRr(1)=40; Fmrh(1)=4.6;
HRr(2)=200; Fmrh(2)=8;
rh_ang= (Fmrh(2)-Fmrh(1))/(HRr(2)-HRr(1));
rh_int= Fmrh(1) - rh_ang* HRr(1);
% -----------------------------------------------------------------------------------------


filepath_out= MainDataDir;
WriteFile=1;
% -----------------------------------------------------------------------------------------
if(1)
MainDataPath=fullfile(MainDataDir,'learning-set');
ResDataPath=fullfile(MainDataPath,'SptRes');
j=1
tipoC='n';    % case type: N not terminated,  T terminated,  S terminated after ....
fprintf('%s\n', ResDataPath);
lista= dir(fullfile(ResDataPath, [tipoC,'*', '.mat']));
for i=1:length(lista)
    [path, nname, ext] = fileparts(lista(i).name);
    fileName= fullfile(ResDataPath,lista(i).name);
%    fprintf('Reading parameters from file: %s\n', fileName);
    par{j}(i)=load(fileName);    %  RRs RRmean RRstd nc crr rp1fn rp2fn
    cname{j}{i}=nname;
end;

j=j+1;
tipoC='t';    % case type: N not terminated,  T terminated,  S terminated after ....
fprintf('%s\n', ResDataPath);
lista= dir(fullfile(ResDataPath, [tipoC,'*', '.mat']));
for i=1:length(lista)
    [path, nname, ext] = fileparts(lista(i).name);
    fileName= fullfile(ResDataPath,lista(i).name);
%    fprintf('Reading parameters from file: %s\n', fileName);
    par{j}(i)=load(fileName);    %  RRs RRmean RRstd nc crr rp1fn rp2fn
    cname{j}{i}=nname;
end;

j=j+1;
tipoC='s';    % case type: N not terminated,  T terminated,  S terminated after ....
fprintf('%s\n', ResDataPath);
lista= dir(fullfile(ResDataPath, [tipoC,'*', '.mat']));
for i=1:length(lista)
    [path, nname, ext] = fileparts(lista(i).name);
    fileName= fullfile(ResDataPath,lista(i).name);
%    fprintf('Reading parameters from file: %s\n', fileName);
    par{j}(i)=load(fileName);    %  RRs RRmean RRstd nc crr rp1fn rp2fn
    cname{j}{i}=nname;
end;

for j=1:3
    for i=1:size(par{j},2)
        Pdmaxm{j}(i)=par{j}(i).Pdmaxb;
        Fmaxm{j}(i)=par{j}(i).fPmaxb;
        RRmean{j}(i)=par{j}(i).RRmean;
    end
    HRmean{j}= 60./RRmean{j};
end
%  Classification of all the cases in learning set
for j=1:3
    cTh{j}= 1 + (Fmaxm{j} > FpTh);
    Cl{j} = 1 + (Fmaxm{j} > r_ang* RRmean{j} + r_int); 
%     Cl{j} = 1 + (Fmaxm{j} > rh_ang* HRmean{j} + rh_int); 
end

for j=1:3
    fprintf('case\tcl\tcth\tFmaxm\tRRmean\n');
    for i=1:size(cname{j},2) fprintf(' %s\t%d\t%d\t%6.3f\t%6.3f\n', cname{j}{i}, Cl{j}(i), cTh{j}(i), Fmaxm{j}(i),RRmean{j}(i)); end
    fprintf('n. cases of type 1 = %d,   %d\n', sum(Cl{j}==1), sum(cTh{j}==1));
    fprintf('n. cases of type 2 = %d,   %d\n', sum(Cl{j}==2), sum(cTh{j}==2));
end
fprintf('min freq. for type 1 = %f\n', min(Fmaxm{1}));
fprintf('max freq. for type 2 = %f\n', max(Fmaxm{2}));
fprintf('max freq. for type 3 = %f\n', max(Fmaxm{3}));
fprintf('min freq. for type 3 = %f\n', min(Fmaxm{3}));
% ---------------------------------------------------------------------------------------------------------
if(WriteFile)
    % Write on file the results
    fnameRes= fullfile(filepath_out, ['ResultTab', '.txt']);
    fidw = fopen(fnameRes,'Wt');
    fprintf(fidw,'%s\n', ResDataPath);
    fprintf(fidw,'Date: %s\n', date);
    for j=1:3
        fprintf(fidw,'case\tcl\tcth\tFmaxm\tRRmean\n');
        for i=1:size(cname{j},2) fprintf(fidw,' %s\t%d\t%d\t%6.3f\t%6.3f\n', cname{j}{i}, Cl{j}(i), cTh{j}(i), Fmaxm{j}(i),RRmean{j}(i)); end
        fprintf(fidw,'n. casi tipo 1 = %d,  %d\n', sum(Cl{j}==1), sum(cTh{j}==1));
        fprintf(fidw,'n. casi tipo 2 = %d,  %d\n', sum(Cl{j}==2), sum(cTh{j}==2));
    end
    fprintf(fidw,'min freq. for type 1 = %f\n', min(Fmaxm{1}));
    fprintf(fidw,'max freq. for type 2 = %f\n', max(Fmaxm{2}));
    fprintf(fidw,'max freq. for type 3 = %f\n', max(Fmaxm{3}));
    fprintf(fidw,'min freq. for type 3 = %f\n', min(Fmaxm{3}));
end
% ---------------------------------------------------------------------------------------------------------


figure
plot(RRmean{1}, Fmaxm{1},'xr'); hold on
plot(RRmean{2}, Fmaxm{2},'xb');
plot(RRmean{3}, Fmaxm{3},'xg');
plot(RRr, Fmr,':k'); % draw the decision line

end;
% -----------------------------------------------------------------------------------------
%if(0)
MainDataPath=[MainDataDir,'\test-set-a'];
ResDataPath=fullfile(MainDataPath,'SptRes');
tipoC='a';
fprintf('%s\n', ResDataPath);
lista= dir(fullfile(ResDataPath, [tipoC,'*', '.mat']));
j=j+1;
for i=1:length(lista)
    [path, nname, ext] = fileparts(lista(i).name);
    fileName= fullfile(ResDataPath,lista(i).name);
%    fprintf('Reading parameters from file: %s\n', fileName);
    par{j}(i)=load(fileName);    %  RRs RRmean RRstd nc crr rp1fn rp2fn
    cname{j}{i}=nname;
end;
for i=1:size(par{j},2)
    Pdmaxm{j}(i)=par{j}(i).Pdmaxb;
    Fmaxm{j}(i)=par{j}(i).fPmaxb;
    RRmean{j}(i)=par{j}(i).RRmean;
end
HRmean{j}= 60./RRmean{j};

cTh{j}= 1 + (Fmaxm{j} > FpTh);
Cl{j}= 1 + (Fmaxm{j}> r_ang* RRmean{j} + r_int); 
% Cl{j}= 1 + (Fmaxm{j}> rh_ang* HRmean{j} + rh_int); 

fprintf('case\tcl\tcth\tFmaxm\tRRmean\n');
for i=1:size(cname{j},2) fprintf(' %s\t%d\t%d\t%6.3f\t%6.3f\n', cname{j}{i}, Cl{j}(i), cTh{j}(i), Fmaxm{j}(i),RRmean{j}(i)); end
fprintf('n. cases tipo 1 = %d,  %d\n', sum(Cl{j}==1), sum(cTh{j}==1));
fprintf('n. cases tipo 2 = %d,  %d\n', sum(Cl{j}==2), sum(cTh{j}==2));

if(WriteFile)
    % Write on file the results
    fprintf(fidw,'%s\n', ResDataPath);
    fprintf(fidw,'case\tcl\tcth\tFmaxm\tRRmean\n');
    for i=1:size(cname{j},2) fprintf(fidw,' %s\t%d\t%d\t%6.3f\t%6.3f\n', cname{j}{i}, Cl{j}(i), cTh{j}(i), Fmaxm{j}(i),RRmean{j}(i)); end
    
    fprintf(fidw,'n. cases tipo 1 = %d,  %d\n', sum(Cl{j}==1), sum(cTh{j}==1));
    fprintf(fidw,'n. cases tipo 2 = %d,  %d\n', sum(Cl{j}==2), sum(cTh{j}==2));
end

figure
plot(RRmean{j}, Fmaxm{j},'om'); hold on

for icn=1:length(casinotiaT)
    icasiknaT(icn)=find(strcmp(casinotiaT(icn),cname{j}));
end
for icn=1:length(casinotiaN)
    icasiknaN(icn)=find(strcmp(casinotiaN(icn),cname{j}));
end
plot(RRmean{j}(icasiknaT), Fmaxm{j}(icasiknaT),'xb'); hold on
plot(RRmean{j}(icasiknaN), Fmaxm{j}(icasiknaN),'xr'); hold on

plot(RRr, Fmr,':k'); % draw the decision line
%end

MainDataPath=[MainDataDir,'\test-set-b'];
ResDataPath=fullfile(MainDataPath,'SptRes');
tipoC='b';
fprintf('%s\n', ResDataPath);
lista= dir(fullfile(ResDataPath, [tipoC,'*', '.mat']));
if(~isempty(lista))
    j=j+1;
    for i=1:length(lista)
        [path, nname, ext] = fileparts(lista(i).name);
        fileName= fullfile(ResDataPath,lista(i).name);
%       fprintf('Reading parameters from file: %s\n', fileName);
        par{j}(i)=load(fileName);    %  RRs RRmean RRstd nc crr rp1fn rp2fn
        cname{j}{i}=nname;
    end;
    for i=1:size(par{j},2)
        Pdmaxm{j}(i)=par{j}(i).Pdmaxb;
        Fmaxm{j}(i)=par{j}(i).fPmaxb;
        RRmean{j}(i)=par{j}(i).RRmean;
    end
    HRmean{j}= 60./RRmean{j};
   
    cTh{j}= 1 + (Fmaxm{j} > FpTh);
    Cl{j}= 1 + (Fmaxm{j}> r_ang* RRmean{j} + r_int); 
%    Cl{j}= 1 + (Fmaxm{j}> rh_ang* HRmean{j} + rh_int); 

    fprintf('case\tcl\tFmaxm\tRRmean\n');
    for i=1:size(cname{j},2) fprintf(' %s\t%d\t%d\t%6.3f\t%6.3f\n', cname{j}{i}, Cl{j}(i), cTh{j}(i), Fmaxm{j}(i),RRmean{j}(i)); end
    fprintf('n. cases tipo 1 = %d,  %d\n', sum(Cl{j}==1), sum(cTh{j}==1));
    fprintf('n. cases tipo 2 = %d,  %d\n', sum(Cl{j}==2), sum(cTh{j}==2));
    
    if(WriteFile)
        % Write on file the results
        fprintf(fidw,'%s\n', ResDataPath);
        fprintf(fidw,'case\tcl\tcth\tFmaxm\tRRmean\n');
        for i=1:size(cname{j},2) fprintf(fidw,' %s\t%d\t%d\t%6.3f\t%6.3f\n', cname{j}{i}, cTh{j}(i), Cl{j}(i), Fmaxm{j}(i),RRmean{j}(i)); end
        
        fprintf(fidw,'n. cases tipo 1 = %d,  %d\n', sum(Cl{j}==1), sum(cTh{j}==1));
        fprintf(fidw,'n. cases tipo 2 = %d,  %d\n', sum(Cl{j}==2), sum(cTh{j}==2));
    end
    figure
    plot(RRmean{j}, Fmaxm{j},'og'); hold on
    for icn=1:length(casinotibT)
        icasiknbT(icn)=find(strcmp(casinotibT(icn),cname{j}));
    end
    for icn=1:length(casinotibS)
        icasiknbS(icn)=find(strcmp(casinotibS(icn),cname{j}));
    end
    plot(RRmean{j}(icasiknbT), Fmaxm{j}(icasiknbT),'xb'); hold on
    plot(RRmean{j}(icasiknbS), Fmaxm{j}(icasiknbS),'xg'); hold on

    plot(RRr, Fmr,':k'); % draw the decision line

end

nj=j;
figure; hold on;
marktype={'xr','xb','xg','om','og'};
for j=1:nj
    plot(RRmean{j}, Fmaxm{j},marktype{j}); 
end
j=4;
plot(RRmean{j}(icasiknaT), Fmaxm{j}(icasiknaT),'*b'); hold on
plot(RRmean{j}(icasiknaN), Fmaxm{j}(icasiknaN),'*r'); hold on
j=5;
plot(RRmean{j}(icasiknbT), Fmaxm{j}(icasiknbT),'*b'); hold on
plot(RRmean{j}(icasiknbS), Fmaxm{j}(icasiknbS),'*g'); hold on
plot(RRr, Fmr,':k'); % draw the decision line

figure; hold on;
for j=1:nj
    plot(HRmean{j}, Fmaxm{j},marktype{j}); 
end
j=4;
plot(HRmean{j}(icasiknaT), Fmaxm{j}(icasiknaT),'*b'); hold on
plot(HRmean{j}(icasiknaN), Fmaxm{j}(icasiknaN),'*r'); hold on
j=5;
plot(HRmean{j}(icasiknbT), Fmaxm{j}(icasiknbT),'*b'); hold on
plot(HRmean{j}(icasiknbS), Fmaxm{j}(icasiknbS),'*g'); hold on

plot(HRr, Fmrh,':k'); % draw the decision line

fclose(fidw);