octave_packages/m/general/private/__splinen__.m

   1 ## Copyright (C) 2007-2012 David Bateman
   2 ##
   3 ## This file is part of Octave.
   4 ##
   5 ## Octave is free software; you can redistribute it and/or modify it
   6 ## under the terms of the GNU General Public License as published by
   7 ## the Free Software Foundation; either version 3 of the License, or (at
   8 ## your option) any later version.
   9 ##
  10 ## Octave is distributed in the hope that it will be useful, but
  11 ## WITHOUT ANY WARRANTY; without even the implied warranty of
  12 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13 ## General Public License for more details.
  14 ##
  15 ## You should have received a copy of the GNU General Public License
  16 ## along with Octave; see the file COPYING.  If not, see
  17 ## <http://www.gnu.org/licenses/>.
  18
  19 ## Undocumented internal function.
  20
  21 ## -*- texinfo -*-
  22 ## @deftypefn {Function File} {@var{yi} =} __splinen__ (@var{x}, @var{y}, @var{xi})
  23 ## Undocumented internal function.
  24 ## @end deftypefn
  25
  26 ## FIXME: Allow arbitrary grids..
  27
  28 function yi = __splinen__ (x, y, xi, extrapval, f)
  29   if (nargin != 5)
  30     error ("__splinen__: Incorrect number of arguments");
  31   endif
  32   ## ND isvector function.
  33   isvec = @(x) numel (x) == length (x);
  34   if (!iscell (x) || length(x) < ndims(y) || any (! cellfun (isvec, x))
  35       || !iscell (xi) || length(xi) < ndims(y) || any (! cellfun (isvec, xi)))
  36     error ("__splinen__: %s: non gridded data or dimensions inconsistent", f);
  37   endif
  38   yi = y;
  39   for i = length(x):-1:1
  40     yi = permute (spline (x{i}, yi, xi{i}(:)), [length(x),1:length(x)-1]);
  41   endfor
  42
  43   [xi{:}] = ndgrid (cellfun (@(x) x(:), xi, "uniformoutput", false){:});
  44   idx = zeros (size(xi{1}));
  45   for i = 1 : length(x)
  46     idx |= xi{i} < min (x{i}(:)) | xi{i} > max (x{i}(:));
  47   endfor
  48   yi(idx) = extrapval;
  49 endfunction