% SUBSREF - index eegdata class
%
% Author: Arnaud Delorme, SCCN, INC, UCSD, Nov. 2008

% Copyright (C) 2008 Arnaud Delorme, SCCN, INC, UCSD
%
% 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 b = subsref(a,s)
    
    if s(1).type == '.'
        b = builtin('subsref', struct(a), s); return;
    end
        
    subs = s(1).subs;
    finaldim = cellfun('length', subs);
    
    % one dimension input
    % -------------------
    if length(s(1).subs) == 1 
        if ischar(subs{1})
            subs{1} = [1:size(a,1)];
            subs{2} = [1:size(a,2)];
            if ndims(a) == 3, subs{3} = [1:size(a,3)]; end
            finaldim = prod(size(a));
        end
        
    % two dimension input
    % -------------------
    elseif length(s(1).subs) == 2 
        if ischar(subs{1}), subs{1} = [1:size(a,1)]; end
        
        if ischar(subs{2}),
            subs{2} = [1:size(a,2)];
            if ndims(a) == 3, subs{3} = [1:size(a,3)]; end
        end
        if length(subs) == 3
             finaldim = [ length(subs{1}) length(subs{2})*length(subs{3}) ];
        else finaldim = [ length(subs{1}) length(subs{2}) ];
        end
            
    % three dimension input
    % ---------------------
    elseif length(s(1).subs) == 3
        
        if ischar(subs{1}), subs{1} = [1:size(a,1)]; end
        if ischar(subs{2}), subs{2} = [1:size(a,2)]; end
        if ndims(a) == 2, 
            subs(3) = []; 
        else
            if ischar(subs{3}), subs{3} = [1:size(a,3)]; end
        end
        finaldim = cellfun('length', subs);
     
    end

    % non-transposed data
    % -------------------
    if ~strcmpi(a.fileformat, 'transposed')
        if length(subs) == 1, b = a.data.data.x(subs{1}); end
        if length(subs) == 2, b = a.data.data.x(subs{1}, subs{2}); end
        if length(subs) == 3, b = a.data.data.x(subs{1}, subs{2}, subs{3}); end
    else
        if ndims(a) == 2
            %if length(s) ==         0, b = transpose(a.data.data.x); return; end
            if length(s(1).subs) == 1, b = a.data.data.x(s(1).subs{1})'; end
            if length(s(1).subs) == 2, b = a.data.data.x(s(1).subs{2}, s(1).subs{1})'; end
            if length(s(1).subs) == 3, b = a.data.data.x(s(1).subs{2}, s(1).subs{1})'; end
        else
            %if length(s) ==         0, b = permute(a.data.data.x, [3 1 2]); return; end
            if length(subs) == 1,
                inds1 = mod(subs{1}-1, size(a,1))+1;
                inds2 = mod((subs{1}-inds1)/size(a,1), size(a,2))+1;
                inds3 = ((subs{1}-inds1)/size(a,1)-inds2+1)/size(a,2)+1;
                inds  = (inds1-1)*size(a,2)*size(a,3) + (inds3-1)*size(a,2) + inds2;
                b = a.data.data.x(inds);
            else
                if length(subs) < 2, subs{3} = 1; end
                
                % repmat if several indices in different dimensions
                % -------------------------------------------------
                len = cellfun('length', subs);
                subs{1} = repmat(reshape(subs{1}, [len(1) 1 1]), [1 len(2) len(3)]);
                subs{2} = repmat(reshape(subs{2}, [1 len(2) 1]), [len(1) 1 len(3)]);
                subs{3} = repmat(reshape(subs{3}, [1 1 len(3)]), [len(1) len(2) 1]);
                
                inds = (subs{1}-1)*a.data.Format{2}(1)*a.data.Format{2}(2) + (subs{3}-1)*a.data.Format{2}(1) + subs{2};
                inds = reshape(inds, [1 prod(size(inds))]);
                b = a.data.data.x(inds);
            end
        end
    end
 
    if length(finaldim) == 1, finaldim(2) = 1; end
    b = reshape(b, finaldim);

% 2 dims
%inds1 = mod(myinds-1, size(a,1))+1;
%inds2 = (myinds-inds1)/size(a,1)+1;
%inds  = (inds2-1)*size(a,1) + inds1;

% 3 dims
%inds1 = mod(myinds-1, size(a,1))+1;
%inds2 = mod((myinds-inds1)/size(a,1), size(a,2))+1;
%inds3 = ((myinds-inds1)/size(a,1)-inds2)/size(a,2)+1;
%inds  = (inds3-1)*size(a,1)*size(a,2) + inds2*size(a,1) + inds1;