Enable inference of rowDims from ColDims and conversely

complete.m

@@ -4,8 +4,8 @@ function [LFRef,LFMask] = complete(LFRef,varargin)
 
 LFSize = size(LFRef);
 numDims = ndims(LFRef);
-colDims = 1:min(3,numDims);
-rowDims = 4:numDims;
+colDims = [];
+rowDims = [];
 
 p = inputParser; p.StructExpand = true; p.KeepUnmatched = true;
 p.addParameter('colDims', colDims, @isnumeric);
@@ -15,6 +15,10 @@ p.parse(varargin{:});
 colDims = p.Results.colDims;
 rowDims = p.Results.rowDims;
 
+if (isempty(colDims)&&isempty(rowDims)); colDims = 1:min(3,numDims); end
+if isempty(colDims); colDims = setdiff(1:numDims,rowDims); end
+if isempty(rowDims); rowDims = setdiff(1:numDims,colDims); end
+
 colSize = LFSize(colDims);
 rowSize = LFSize(rowDims);
 

factorize.m

@@ -4,8 +4,8 @@ function [LFB,C,U,S,V] = factorize(LFRef,k,varargin)
 
 LFSize = size(LFRef);
 numDims = ndims(LFRef);
-colDims = 1:min(3,numDims);
-rowDims = 4:numDims;
+colDims = [];
+rowDims = [];
 
 p = inputParser; p.StructExpand = true; p.KeepUnmatched = true;
 p.addParameter('colDims', colDims, @isnumeric);
@@ -15,6 +15,10 @@ p.parse(varargin{:});
 colDims = p.Results.colDims;
 rowDims = p.Results.rowDims;
 
+if (isempty(colDims)&&isempty(rowDims)); colDims = 1:min(3,numDims); end
+if isempty(colDims); colDims = setdiff(1:numDims,rowDims); end
+if isempty(rowDims); rowDims = setdiff(1:numDims,colDims); end
+
 colSize = LFSize(colDims);
 rowSize = LFSize(rowDims);