Created convenience function

Created convenience function for encoding light fields and corresponding example script

encodeLF.m

-function [frames,framesConv,framesConvRec] = encodeLF(LF,name,ConfigFile,varargin)
+function [frames,framesRef,framesRec] = encodeLF(LF,name,varargin)
 %ENCODELF Summary of this function goes here
 %   Detailed explanation goes here
 
 colSpaces = {'yuv','ycbcr','lab','hsv','rgb'};
 subSamps  = {'420','444','422','400'};
-precisions = {'uint8','uint16','uint32','uint64'...
-              'int8','int16','int32','int64',...
+precisions = {'uint8','uint16','uint32','uint64',...
+               'int8', 'int16', 'int32', 'int64',...
               'single','double'};
 
-p = inputParser;
-p.addRequired ('LF'                     ,@isnumeric);
-p.addRequired ('name'                   ,@ischar);
-p.addRequired ('ConfigFile'             ,@ischar);
-p.addParameter('colSpace' ,colSpaces{1} ,@(x) any(validatestring(x,colSpaces)));
-p.addParameter('subSamp'  ,subSamps{1}  ,@(x) any(validatestring(x,subSamps) ));
-p.addParameter('precision',precisions{1},@(x) any(validatestring(x,precisions)));
-p.addParameter('subName'  ,'_rec'       ,@ischar);
+% Create input parser scheme
+params = inputParser;
+params.StructExpand = true;
+params.KeepUnmatched = true;
+params.addParameter('colSpace'  ,colSpaces{1} ,@(x) any(validatestring(x,colSpaces )));
+params.addParameter('subSamp'   ,subSamps{1}  ,@(x) any(validatestring(x,subSamps  )));
+params.addParameter('precision' ,precisions{1},@(x) any(validatestring(x,precisions)));
 
-p.parse(LF,name,ConfigFile,varargin{:});
-%LF         = p.Results.LF;
-%name       = p.Results.name;
-%ConfigFile = p.Results.name;
-colSpace   = p.Results.colSpace;
-subSamp    = p.Results.subSamp;
-precision  = p.Results.precision;
-subName    = p.Results.subName;
-        
-frames = LFToFrames(LF);
-[imgRes,imgSize,framesConv] = yuv.write(frames,name,...
-    'colSpace',colSpace,...
-    'subSamp',subSamp,...
-    'precision',precision);
+% Parse arguments
+params.parse(varargin{:});
 
-[InputFile,BitstreamFile,ReconFile,LogFile,...
-    SourceWidth,SourceHeight,...
-    FramesToBeEncoded,InputChromaFormat]...
-    = extractHEVCParams(name,colSpace,subSamp,subName);
+yuvParams = params.Results;
+extraParams = params.Unmatched;
 
-HEVC.encode(ConfigFile,...
-'InputFile',InputFile,...
-'BitstreamFile',BitstreamFile,...
-'ReconFile',ReconFile,...
-'LogFile',LogFile,...
-'SourceWidth',SourceWidth,...
-'SourceHeight',SourceHeight,...
-'FramesToBeEncoded',FramesToBeEncoded,...
-'InputChromaFormat',InputChromaFormat);
+if isfield(extraParams,'InputChromaFormat')
+    ICF = extraParams.InputChromaFormat;
+    SS  = yuvParams.subSamp;
+    if ~strcmp(ICF,SS)
+        warning(['Conflicting SubSamp and InputChromaFormat parameters',...
+            ' , choosing InputChromaFormat']);
+        yuvParams.subSamp = ICF;
+    end
+end
 
-framesConvRec = yuv.read(name,imgRes,imgSize,...
-    'colSpace',colSpace,...
-    'subSamp',subSamp,...
-    'precision',precision,...
-    'subName',subName);
+% Change 5D LF to a collection of 2D frames
+frames = LFToFrames(LF);
 
+% Convert each frame according to given parameters
+framesRef = yuv.convert(frames,yuvParams);
 
+% Write frames as a YUV sequence
+[imgRes,imgSize] = yuv.write(framesRef,name,yuvParams,'subName','_ref');
 
+% Convert inputs for HEVC
+HEVCparams = extractHEVCParams(name,yuvParams);
 
+% Encode frames using HEVC
+HEVC.encode(HEVCparams,extraParams);
 
+% Recover frames
+framesRec = yuv.read(name,imgRes,imgSize,yuvParams,'subName','_rec');
 
-%%
+%% Auxiliary functions
     function frames = LFToFrames(LF)
         s = size(LF);
         frames = mat2cell(LF,ones(1,s(1)),ones(1,s(2)),s(3),s(4),s(5));
     end
 
-    function LF = frames2LF(frames)
-    end
-
-    function [InputFile,BitstreamFile,ReconFile,LogFile,...
-            SourceWidth,SourceHeight,...
-            FramesToBeEncoded,InputChromaFormat] = extractHEVCParams...
-            (name,colSpace,subSamp,subName)
-        
+    function params = extractHEVCParams(name,yuvParams)
+        colSpace = yuvParams.colSpace;
+        subSamp = yuvParams.subSamp;
         resCell = num2cell(imgRes);
         sizeCell = num2cell(imgSize);
         fullName = [resCell,sizeCell];
         fullName = cellfun(@num2str,fullName,'UniformOutput',false);
         fullName = strjoin([name fullName subSamp],'_');
-        InputFile =     [fullName,        '.',colSpace];
-        BitstreamFile = [fullName,        '.','hevc'  ];
-        ReconFile =     [fullName,subName,'.',colSpace];
-        LogFile =       [fullName,        '.','rtf'   ];
-        SourceWidth  = num2str(imgSize(2));
-        SourceHeight = num2str(imgSize(1));
-        FramesToBeEncoded = num2str(prod(imgRes));
-        InputChromaFormat = num2str(subSamp);
+        params.InputFile =     [fullName,'_ref','.',colSpace];
+        params.BitstreamFile = [fullName,       '.','hevc'  ];
+        params.ReconFile =     [fullName,'_rec','.',colSpace];
+        params.LogFile =       [fullName,       '.','rtf'   ];
+        params.SourceWidth  = num2str(imgSize(2));
+        params.SourceHeight = num2str(imgSize(1));
+        params.FramesToBeEncoded = num2str(prod(imgRes));
+        params.InputChromaFormat = num2str(subSamp);
     end
 end

example.m

+clear all, close all, clc
+
+%%
+LFDir = 'C:\Users\edib\Light Fields\Heidelberg\full_data\test';
+name = 'bedroom';
+
+%%
+load(fullfile(LFDir,name,'LF.mat'));
+LF = LF.LF;
+
+%%
+s = size(LF);
+colSpace = 'yuv';
+subSamp = '400';
+subName = '_rec';
+precision = 'uint8';
+
+params.FramesToBeEncoded = num2str(1);
+params.InputChromaFormat = num2str(subSamp);
+params.QP = num2str(0);
+params.colSpace = colSpace;
+params.subSamp = num2str(subSamp);
+params.precision = precision;
+
+%%
+[frames,convFramesRef,subFramesRec] = HEVC.encodeLF(LF,name,params);
+
+%%
+HEVC.encode(params);
+
+%%
+testHEVC = HEVC.HEVC(params);
+