arrayfun.m

[Bill Denney]

Having recently discovered cellfun, I thought that I'd write arrayfun 
for fun.

Bill

scripts/Changelog:

2006-10-09  Bill Denney  <address@hidden>
   * miscellaneous/arrayfun.m: New function, a complementary function 
to cellfun.

## Copyright (C) 2006  Bill Denney  <address@hidden>
##
## This program is free software; you can redistribute it and/or modify
## it under the terms of the GNU General Public License as published by
## the Free Software Foundation; either version 2 of the License, or
## (at your option) any later version.
##
## This program is distributed in the hope that it will be useful,
## but WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
## GNU General Public License for more details.
##
## You should have received a copy of the GNU General Public License
## along with this program; if not, write to the Free Software
## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA

## -*- texinfo -*-
## @deftypefn {Command} @var{a} = arrayfun(@var{name}, @var{c})
## @deftypefnx {Command} @var{a} = arrayfun(@var{func}, @var{c})
## @deftypefnx {Command} @var{a} = arrayfun(@var{func}, @var{c}, @var{d})
## @deftypefnx {Command} @var{a} = arrayfun(@var{func}, @var{c}, @var{options})
## @deftypefnx {Command} address@hidden, @var{b}, @dots{}] = 
arrayfun(@var{func}, @var{c}, @dots{})
## Execute a function on each element of an array.
##
## See cellfun for complete usage instructions.
## @seealso{cellfun}
## @end deftypefn

function [varargout] = arrayfun(func, varargin)

  if (nargin < 2)
    print_usage();
  endif

  ## convert everything to cells and call cellfun (let cellfun error
  ## check the options in case more options come available)
  sizetomatch = size(varargin{1});
  m2cargs{1} = ones (size (varargin{1}, 1), 1);
  m2cargs{2} = ones (size (varargin{1}, 2), 1);
  cfarg{1} = mat2cell (varargin{1}, m2cargs{:});
  stillmatches = true;
  idx = 1;
  while stillmatches && (idx < length(varargin))
    idx++;
    thissize = size (varargin{idx});
    if (length (thissize) == length (sizetomatch)) && \
          all (thissize == sizetomatch)
      if ischar (varargin{idx}) && \
            (strcmpi (varargin{idx}, "UniformOutput") || \
             strcmpi (varargin{idx}, "ErrorHandler"))
        ## catch these strings just in case they happen to be the same
        ## size as the other input.
        stillmatches = false;
      else
        cfarg{idx} = mat2cell (varargin{idx}, m2cargs{:});
      endif
    else
      stillmatches = false;
    endif
  endwhile

  varargout = cell (max ([nargout, 1]), 1);
  [varargout{:}] = cellfun (func, cfarg{:}, varargin{idx+1:length(varargin)});
endfunction