function [Status, FileList] = EnumPRefDir( RefDir )
%EnumPRefDir 枚舉某一保存參考圖片的目錄獲取其中的所有參考圖像文件
% 枚舉目錄（排除子目錄包括.或..）
% 輸入參數
% refdir 被枚舉的目錄
% 輸出參數
% Status 為0表示文件列表為空，1非空
% FileList 文件列表
% 調用示例
% [Status, FileList] = EnumPRefDir( 'RefDir' );



%FileList{1} = 'a111.txt';
%FileList{2} = 'a112.txt';

if ~exist(RefDir, 'dir')
    strerr = ['目錄', RefDir, '不存在'];
    disp(strerr);
end    

DirList=dir(RefDir);
FileListIndex = 1;
Status=0;
FileList{FileListIndex}=char('empty');
for DirListIndex=1:length(DirList)
    if ~DirList(DirListIndex).isdir
        FileList{FileListIndex} = char(DirList(DirListIndex).name);
        FileListIndex = FileListIndex + 1;
        Status=1;
    end
end

end

function Status = FSIMBetweenTwoFileFolder( RefDir, DecodeDir )
%FSIMBetweenTwoFileFolder 調用FSIM計算兩個文件夾中的圖片從而生成相似度列表文件
% 基於參考與解碼文件夾計算這些文件夾中相應圖片的相似度
% Tip:兩個文件夾中的文件命名方式相同，比如RefDir/P000001與RefDir/P000001
% 輸入參數
% RefDir 參數圖片文件夾
% DecodeDir 解碼圖片文件夾
% 輸出參數
% Status 為0出現錯誤(比如文件夾為空或兩個文件夾圖片命方式不同或數量不同) 
% 為1全部文件都經過了正確的相似度計算
% 輸出相似度列表文件
% RefDir+DecodeDir+'FSIM.txt',其行格式為：圖片文件名 + ' ' + 相似度值 + ' ' + 相似度值c
% 調用示例
%  Status = FSIMBetweenTwoFileFolder('RefDir', 'DecodeDir1');

Status = 1;
if ~exist(RefDir, 'dir') ||  ~exist(DecodeDir, 'dir')
    strerr = ['目錄', RefDir,'或',DecodeDir, '不存在'];
    disp(strerr);        
    Status = 0;
    return;
end    




SListFileName = strcat(RefDir, '_');
SListFileName = strcat(SListFileName, DecodeDir);
SListFileName = strcat(SListFileName, '_FSIM.txt');

try
    [FStatus, FileList] = EnumPRefDir( RefDir );
    
    if(FStatus == 0)
        Status = 0;
        return;
    end   
    
    SListFileID = fopen(SListFileName, 'w');
    if(SListFileID < 0)
        Status = 0;
        return;
    end
        
     
      
    for FIndex = 1 : length(FileList)
        OneRefFileName =['', RefDir, '/', char(FileList(FIndex))];        
        OneDecodeFileName = ['', DecodeDir, '/', char(FileList(FIndex))];
        
        if ~exist(OneRefFileName, 'file') || ~exist(OneDecodeFileName, 'file')
            Status = 0;
            break;
        else
            try
            OneRefFileHandle = imread(OneRefFileName);
            OneDecodeFileHandle = imread(OneDecodeFileName);            
            [FSIM, FSIMc] = FeatureSIM(OneRefFileHandle, OneDecodeFileHandle);            
            fprintf(SListFileID, '%s %f %f\r\n', char(FileList(FIndex)), FSIM, FSIMc);
            catch rme
                disp(rme.message);
                Status = 0;
                break;
            end
        end
    end   

    fclose(SListFileID);    
catch eme     
    disp(eme.message);
    Status = 0;
end
    if ~Status
        delete(SListFileName);
    end
end

function FSIMDriver( RefDir,DecodeDir1, DecodeDir2 )
%FSIMDriver FSIM相似度計算主驅動程序
% FSIM相似度計算主驅動程序
% 説明
% 跟據目前的需求，主要基於編碼前圖像目錄(RefDir)與兩種不同編碼參數相應的
% 兩類解碼圖像目錄(DecodeDir1與DecodeDir2)計算圖像特徵相似度，從而比較不同的編碼參數的效果
% 輸入參數
% RefDir 保存編碼前圖像文件的文件夾名稱，默認值是當前目錄下的‘RefDir’
% DecodeDir1 保存解碼後圖像文件的文件夾名稱，默認值是當前目錄下的‘DecodeDir1’
% DecodeDir2 保存解碼後圖像文件的文件夾名稱（DecodeDir1與DecodeDir2中的圖像文件編碼參數不同），默認值是當前目錄下的‘DecodeDir2’

% 輸出參數
% 無
% 調用示例
% FSIMDriver();

	narginchk(0, 3);
    nargoutchk(0, 0);


    if nargin < 3
        DecodeDir2='DecodeDir2';
    end

    if nargin < 2
       DecodeDir1='DecodeDir1';
    end

    if nargin < 1
        RefDir='RefDir';
    end    
    
    try
        ErrorCode = FSIMBetweenTwoFileFolder(RefDir, DecodeDir1);  
        disp(['基於目錄',RefDir,'與', DecodeDir1, '計算圖像特徵相似度', GetErrorDisp( ErrorCode )]);
        
        ErrorCode = FSIMBetweenTwoFileFolder(RefDir, DecodeDir2);
        disp(['基於目錄',RefDir,'與', DecodeDir2, '計算圖像特徵相似度', GetErrorDisp( ErrorCode )]);
    catch tme
        disp(tme.message);
    end
end

function ErrorDisp = GetErrorDisp( ErrorCode )
    if ErrorCode
        ErrorDisp = '成功';
    else
        ErrorDisp = '失敗';
    end    
end