plot_areaerrorbar.m

% ----------------------------------------------------------------------- %
% Function plot_areaerrorbar plots the mean and standard deviation of a   %
% set of data filling the space between the positive and negative mean    %
% error using a semi-transparent background, completely customizable.     %
%                                                                         %
%   Input parameters:                                                     %
%       - data:     Data matrix, with rows corresponding to observations  %
%                   and columns to samples.                               %
%       - options:  (Optional) Struct that contains the customized params.%
%           * options.handle:       Figure handle to plot the result.     %
%           * options.color_area:   RGB color of the filled area.         %
%           * options.color_line:   RGB color of the mean line.           %
%           * options.alpha:        Alpha value for transparency.         %
%           * options.line_width:   Mean line width.                      %
%           * options.x_axis:       X time vector.                        %
%           * options.error:        Type of error to plot (+/-).          %
%                   if 'std',       one standard deviation;               %
%                   if 'sem',       standard error mean;                  %
%                   if 'var',       one variance;                         %
%                   if 'c95',       95% confidence interval.              %
% ----------------------------------------------------------------------- %
%   Example of use:                                                       %
%       data = repmat(sin(1:0.01:2*pi),100,1);                            %
%       data = data + randn(size(data));                                  %
%       plot_areaerrorbar(data);                                          %
% ----------------------------------------------------------------------- %
%   Author:  Victor Martinez-Cagigal                                      %
%   Date:    30/04/2018                                                   %
%   E-mail:  vicmarcag (at) gmail (dot) com                               %
% ----------------------------------------------------------------------- %

% Adapted by M.N. April. 2019
% corrected sem and c95 calculation
% added compatibility for NaNs and if std==0;

function plot_areaerrorbar(data, options)

% Default options
if(nargin<2)
    options.handle     = figure(1);
    options.color_area = [128 193 219]./255;    % Blue theme
    options.color_line = [ 52 148 186]./255;
    %options.color_area = [243 169 114]./255;    % Orange theme
    %options.color_line = [236 112  22]./255;
    options.alpha      = 0.5;
    options.line_width = 2;
    options.error      = 'sem';
end

% Computing the mean and standard deviation of the data matrix
data_mean = nanmean(data,1);
data_std  = nanstd(data,0,1);

% NaN compatibility
data_std(isnan(data_std))   = 0;
data(:,isnan(data_mean))  = [];
data_std(isnan(data_mean))  = [];
data_mean(isnan(data_mean)) = [];

if(isfield(options,'x_axis')==0), options.x_axis = 1:size(data,2); end
options.x_axis = options.x_axis(:);

% Type of error plot
switch(options.error)
    case 'std', error = data_std;
    case 'sem', error = (data_std./sqrt(sum(~isnan(data)))); % M.N.
    case 'var', error = (data_std.^2);
    case 'c95', error = (data_std./sqrt(sum(~isnan(data)))).*1.96; % M.N:
end

% Plotting the result
figure(options.handle);hold all;
x_vector = [options.x_axis', fliplr(options.x_axis')];
patch = fill(x_vector, [data_mean+error,fliplr(data_mean-error)], options.color_area);
set(patch, 'edgecolor', 'none');
set(patch, 'FaceAlpha', options.alpha);
plot(options.x_axis, data_mean, 'color', options.color_line, ...
    'LineWidth', options.line_width);
hold off;
end