getDatasetMean.m

function dsMean = getDatasetMean(cycleName,dSet,dim,varargin)
    % GETDATASETMEAN
    %
    %
    
    % Created by Megan Schroeder
    % Last Modified 2014-03-18
    
    
    %% Main
    % Main function definition
    
    if nargin == 3
        dsnames = dSet.Properties.VarNames;
        if dim == 1
            newdata = zeros(1,length(dsnames));
            for i = 1:length(dsnames)    
                newdata(:,i) = mean(dSet.(dsnames{i}));
            end
        elseif dim == 2
            newdata = zeros(size(dSet));
            for i = 1:length(dsnames)
                newdata(:,i) = nanmean(dSet.(dsnames{i}),2);
            end
            % Interpolate around data endpoint (Stair Descent to Floor - only for muscle forces)
            if ~isempty(regexp(cycleName,'2F','ONCE')) && size(dSet,2) == 10
                % First column only
                newdataWithNaNs = mean(dSet.(dsnames{1}),2);
                % First NaN index
                nanFirst = find(isnan(newdataWithNaNs),1,'first');
                % Make sure that there was at least 1 trial with data
                if isempty(find(isnan(newdata(:,1)),1,'first'))
                    newdata(nanFirst:nanFirst+5,:) = NaN;
                    nanInd = isnan(newdata(:,1));
                    x = (0:100)';
                    for i = 1:length(dsnames)
                        newdata(nanInd,i) = interp1(x(~nanInd),newdata(~nanInd,i),x(nanInd),'spline');
                    end
                else    
                    newdata = zeros(size(dSet));
                    for i = 1:length(dsnames)
                        newdata(:,i) = mean(dSet.(dsnames{i}),2);
                    end
                end
            % Stair descent to step
            elseif ~isempty(regexp(cycleName,'2S','ONCE')) && size(dSet,2) == 10
                % Subject 20130207APRM doesn't have data at the beginning for any trials
                if all(isnan(newdata(1,:)))
                    newdata = zeros(size(dSet));
                    for i = 1:length(dsnames)
                        newdata(:,i) = mean(dSet.(dsnames{i}),2);
                    end
                % Patella tendon group b/c of subject 20130207APRM - interpolate around data start point
                elseif any(isnan(dSet.(dsnames{1})(1,:)))
                    % First column only
                    newdataWithNaNs = mean(dSet.(dsnames{1}),2);
                    % Last NaN index
                    nanLast = find(isnan(newdataWithNaNs),1,'last');               
                    newdata(nanLast-5:nanLast,:) = NaN;
                    nanInd = isnan(newdata(:,1));
                    x = (0:100)';
                    for i = 1:length(dsnames)
                        newdata(nanInd,i) = interp1(x(~nanInd),newdata(~nanInd,i),x(nanInd),'spline');
                    end
                end
            end
        end
    % Group average
    else
        dsnames = dSet.Properties.VarNames;
        weights = varargin{1};
        weights = repmat(weights',101,1);
        weights(isnan(dSet.(dsnames{1}))) = NaN;
        sumWeights = nansum(weights,2);
        newdata = zeros(size(dSet));
        for i = 1:length(dsnames)
            newdata(:,i) = nansum(dSet.(dsnames{i}).*weights,2)./sumWeights;
        end
        if ~isempty(regexp(cycleName,'2F','ONCE'))
            % Interpolate around date end point
            if any(isnan(dSet.(dsnames{1})(end,:)))
                % First column only
                newdataWithNaNs = mean(dSet.(dsnames{1}),2);
                % First NaN index
                nanFirst = find(isnan(newdataWithNaNs),1,'first');                
                newdata(nanFirst:nanFirst+5,:) = NaN;
                nanInd = isnan(newdata(:,1));
                x = (0:100)';
                for i = 1:length(dsnames)
                    newdata(nanInd,i) = interp1(x(~nanInd),newdata(~nanInd,i),x(nanInd),'spline');
                end
            end            
        elseif ~isempty(regexp(cycleName,'2S','ONCE'))        
            % Patella tendon group b/c of subject 20130207APRM - interpolate around data start point
            if any(isnan(dSet.(dsnames{1})(1,:)))
                % First column only
                newdataWithNaNs = mean(dSet.(dsnames{1}),2);
                % Last NaN index
                nanLast = find(isnan(newdataWithNaNs),1,'last');               
                newdata(nanLast-5:nanLast,:) = NaN;
                nanInd = isnan(newdata(:,1));
                x = (0:100)';
                for i = 1:length(dsnames)
                    newdata(nanInd,i) = interp1(x(~nanInd),newdata(~nanInd,i),x(nanInd),'spline');
                end
            end
        end
    end
    dsMean = dataset({newdata,dsnames{:}});
    
end