6702
|
1 ## Copyright (C) 2007 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 |
7016
|
7 ## the Free Software Foundation; either version 3 of the License, or (at |
|
8 ## your option) any later version. |
6702
|
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 |
7016
|
16 ## along with Octave; see the file COPYING. If not, see |
|
17 ## <http://www.gnu.org/licenses/>. |
6702
|
18 |
6945
|
19 ## Undocumented internal function. |
|
20 |
6702
|
21 ## -*- texinfo -*- |
7650
|
22 ## @deftypefn {Function File} {@var{yi} =} __splinen__ (@var{x}, @var{y}, @var{xi}) |
6702
|
23 ## Internal support function for multi-dimensional splines. |
|
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 ("Incorrect number of arguments"); |
|
31 endif |
8506
|
32 ## ND isvector function. |
|
33 isvec = @(x) numel (x) == length (x); |
6721
|
34 if (!iscell (x) || length(x) < ndims(y) || any (! cellfun (isvec, x)) || |
|
35 !iscell (xi) || length(xi) < ndims(y) || any (! cellfun (isvec, xi))) |
6702
|
36 error ("%s: non gridded data or dimensions inconsistent", f); |
|
37 endif |
|
38 yi = y; |
|
39 for i = length(x):-1:1 |
6721
|
40 yi = permute (spline (x{i}, yi, xi{i}), [length(x),1:length(x)-1]); |
6702
|
41 endfor |
|
42 |
7423
|
43 [xi{:}] = ndgrid (cellfun (@(x) x(:), xi, "UniformOutput", false){:}); |
6702
|
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 |