|
5837
|
1 ## Copyright (C) 2006, Alexander Barth |
|
|
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 2, or (at your option) |
|
|
8 ## 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, write to the Free |
|
|
17 ## Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA |
|
|
18 ## 02110-1301, USA. |
|
|
19 |
|
|
20 ## -*- texinfo -*- |
|
6547
|
21 ## @deftypefn {Function File} {[@var{y1}, @var{y2}, @dots{}, @var{y}n] =} ndgrid (@var{x1}, @var{x2}, @dots{}, @var{x}n) |
|
|
22 ## @deftypefnx {Function File} {[@var{y1}, @var{y2}, @dots{}, @var{y}n] =} ndgrid (@var{x}) |
|
6895
|
23 ## Given n vectors @var{x1}, @dots{} @var{x}n, @code{ndgrid} returns |
|
|
24 ## n arrays of dimension n. The elements of the ith output argument |
|
|
25 ## contains the elements of the vector @var{x}i repeated over all |
|
|
26 ## dimensions different from the ith dimension. Calling ndgrid with |
|
|
27 ## only one input argument @var{x} is equivalent of calling ndgrid with |
|
5837
|
28 ## all n input arguments equal to @var{x}: |
|
|
29 ## |
|
6547
|
30 ## [@var{y1}, @var{y2}, @dots{}, @var{y}n] = ndgrid (@var{x}, @dots{}, @var{x}) |
|
5837
|
31 ## @seealso{meshgrid} |
|
|
32 ## @end deftypefn |
|
|
33 |
|
|
34 ## Author: Alexander Barth <abarth@marine.usf.edu> |
|
|
35 |
|
5838
|
36 function varargout = ndgrid (varargin) |
|
5837
|
37 |
|
|
38 if (nargin == 1) |
|
5838
|
39 n = max ([nargout, 2]); |
|
5837
|
40 ## If only one input argument is given, repeat it n-times |
|
|
41 varargin{1:n} = varargin{1}; |
|
|
42 elseif (nargin >= nargout) |
|
5838
|
43 n = max ([nargin, 2]); |
|
5837
|
44 else |
|
|
45 error ("ndgrid: wrong number of input arguments"); |
|
|
46 endif |
|
|
47 |
|
|
48 ## Determine the size of the output arguments |
|
|
49 |
|
5838
|
50 shape = zeros (1, n); |
|
5837
|
51 |
|
5838
|
52 for i = 1:n |
|
|
53 if (! isvector (varargin{i})) |
|
5837
|
54 error ("ndgrid: arguments must be vectors"); |
|
|
55 endif |
|
|
56 |
|
5838
|
57 shape(i) = length (varargin{i}); |
|
5837
|
58 endfor |
|
|
59 |
|
5838
|
60 for i = 1:n |
|
5837
|
61 ## size for reshape |
|
5838
|
62 r = ones (1, n); |
|
5837
|
63 r(i) = shape(i); |
|
|
64 |
|
|
65 ## size for repmat |
|
|
66 s = shape; |
|
|
67 s(i) = 1; |
|
|
68 |
|
5838
|
69 varargout{i} = repmat (reshape (varargin{i}, r), s); |
|
5837
|
70 endfor |
|
|
71 |
|
|
72 endfunction |
