% FILLCURVES - fill the space between 2 curves
%
% Usage:
%   h=fillcurves( Y1, Y2);
%   h=fillcurves( X, Y1, Y2, color, transparent[0 to 1]);
%
% Example:
%   a = rand(1, 50);
%   b = rand(1, 50)+2; b(10) = NaN;
%   figure; fillcurves([51:100], a, b);
%
% Author: A. Delorme, SCCN, INC, UCSD/CERCO, CNRS

% Copyright (C) Arnaud Delorme, arno@salk.edu
%
% This file is part of EEGLAB, see http://www.eeglab.org
% for the documentation and details.
%
% Redistribution and use in source and binary forms, with or without
% modification, are permitted provided that the following conditions are met:
%
% 1. Redistributions of source code must retain the above copyright notice,
% this list of conditions and the following disclaimer.
%
% 2. Redistributions in binary form must reproduce the above copyright notice,
% this list of conditions and the following disclaimer in the documentation
% and/or other materials provided with the distribution.
%
% THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
% AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
% IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
% ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
% LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
% CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
% SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
% INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
% CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
% ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
% THE POSSIBILITY OF SUCH DAMAGE.

function h = fillcurves(X, Y1, Y2, color, transparent, legends)
    
    if nargin < 2
        help fillcurves;
        return;
    end
    
    if nargin < 3
        Y2 = Y1;
        Y1 = X;
        X = [1:length(Y1)];
    end
    if nargin < 4 || isempty(color)
        color = { 'r' 'b' 'g' 'c' };
    elseif ~iscell(color)
        color = { color };
    end
    if nargin < 5
        transparent = 0.5;
    end
    X1 = X(:)';
    X2 = X(:)';
    
    % remove NaNs
    tmpnan1 = find(isnan(Y1));
    tmpnan2 = find(isnan(Y2));
    Y1(tmpnan1) = []; X1(tmpnan1) = [];
    Y2(tmpnan2) = []; X2(tmpnan2) = [];
    
    % remove 0s if log scale
    if strcmpi(get(gca, 'yscale'), 'log')
        tmp1 = find(~Y1);
        tmp2 = find(~Y2);
        Y1(tmp1) = []; X1(tmp1) = [];
        Y2(tmp2) = []; X2(tmp2) = [];
    end
    
    % multiple curve plot
    % -------------------
    if size(Y1,1) ~= 1 && size(Y1,2) ~= 1
        for index = 1:size(Y1,2)
            fillcurves(X, Y1(:,index)', Y2(:,index)', color{index}, transparent);
            hold on;
        end
%         yl = ylim;
%         xl = xlim;
%         line([xl(1) xl(1)]+(xl(2)-xl(1))/2000, yl, 'color', 'k');
%         line(xl, [yl(1) yl(1)]+(yl(2)-yl(1))/2000, 'color', 'k');
        
        % write legend and add transparency to it
        % ---------------------------------------
        if nargin > 5
            h = legend(legends);
            hh = get(h, 'children');
            for index = 1:length(hh)
                fields = get(hh(index));
                if isfield(fields, 'FaceAlpha');
                    numfaces = size(get(hh(index), 'Vertices'),1);
                    set(hh(index), 'FaceVertexCData', repmat([1 1 1], [numfaces 1]), 'Cdatamapping', 'direct', 'facealpha', transparent, 'edgecolor', 'none');
                end
            end
        end
        return;
    end
    
    % plot
    % ----
    allpointsx = [X1 X2(end:-1:1)]';
    allpointsy = [Y1 Y2(end:-1:1)]';
    h = fill(allpointsx, allpointsy, color{1}); 
    set(h, 'edgecolor', color{1});
    xlim([ X(1) X(end) ]);
    if transparent
        numfaces = size(get(h, 'Vertices'),1);
        set(h, 'FaceVertexCData', repmat([1 1 1], [numfaces 1]), 'Cdatamapping', 'direct', 'facealpha', transparent, 'edgecolor', 'none');
    end

    % replot lines at boundaries
    % --------------------------
%     parent = dbstack;
%     if length(parent) == 1 || ~strcmpi(parent(2).name, 'fillcurves')
%         yl = ylim;
%         xl = xlim;
%         line([xl(1) xl(1)]+(xl(2)-xl(1))/2000, yl, 'color', 'k');
%         line(xl, [yl(1) yl(1)]+(yl(2)-yl(1))/2000, 'color', 'k');
%     end