Mercurial > hg > octave-lyh
annotate scripts/plot/ndgrid.m @ 8455:fd11a08a9b31
disallow invalid {}-indexed assigments
| author | Jaroslav Hajek <highegg@gmail.com> |
|---|---|
| date | Sun, 11 Jan 2009 07:55:44 +0100 |
| parents | a1dbe9d80eee |
| children | 836618fee9d6 |
| rev | line source |
|---|---|
| 7017 | 1 ## Copyright (C) 2006, 2007 Alexander Barth |
| 5837 | 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. | |
| 5837 | 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/>. | |
| 5837 | 18 |
| 19 ## -*- texinfo -*- | |
| 6547 | 20 ## @deftypefn {Function File} {[@var{y1}, @var{y2}, @dots{}, @var{y}n] =} ndgrid (@var{x1}, @var{x2}, @dots{}, @var{x}n) |
| 21 ## @deftypefnx {Function File} {[@var{y1}, @var{y2}, @dots{}, @var{y}n] =} ndgrid (@var{x}) | |
| 6895 | 22 ## Given n vectors @var{x1}, @dots{} @var{x}n, @code{ndgrid} returns |
| 23 ## n arrays of dimension n. The elements of the ith output argument | |
| 24 ## contains the elements of the vector @var{x}i repeated over all | |
| 25 ## dimensions different from the ith dimension. Calling ndgrid with | |
| 26 ## only one input argument @var{x} is equivalent of calling ndgrid with | |
| 5837 | 27 ## all n input arguments equal to @var{x}: |
| 28 ## | |
| 6547 | 29 ## [@var{y1}, @var{y2}, @dots{}, @var{y}n] = ndgrid (@var{x}, @dots{}, @var{x}) |
| 5837 | 30 ## @seealso{meshgrid} |
| 31 ## @end deftypefn | |
| 32 | |
| 33 ## Author: Alexander Barth <abarth@marine.usf.edu> | |
| 34 | |
| 5838 | 35 function varargout = ndgrid (varargin) |
| 5837 | 36 |
| 37 if (nargin == 1) | |
| 5838 | 38 n = max ([nargout, 2]); |
| 5837 | 39 ## If only one input argument is given, repeat it n-times |
|
8455
fd11a08a9b31
disallow invalid {}-indexed assigments
Jaroslav Hajek <highegg@gmail.com>
parents:
7017
diff
changeset
|
40 varargin(1:n) = varargin(1); |
| 5837 | 41 elseif (nargin >= nargout) |
| 5838 | 42 n = max ([nargin, 2]); |
| 5837 | 43 else |
| 44 error ("ndgrid: wrong number of input arguments"); | |
| 45 endif | |
| 46 | |
| 47 ## Determine the size of the output arguments | |
| 48 | |
| 5838 | 49 shape = zeros (1, n); |
| 5837 | 50 |
| 5838 | 51 for i = 1:n |
| 52 if (! isvector (varargin{i})) | |
| 5837 | 53 error ("ndgrid: arguments must be vectors"); |
| 54 endif | |
| 55 | |
| 5838 | 56 shape(i) = length (varargin{i}); |
| 5837 | 57 endfor |
| 58 | |
| 5838 | 59 for i = 1:n |
| 5837 | 60 ## size for reshape |
| 5838 | 61 r = ones (1, n); |
| 5837 | 62 r(i) = shape(i); |
| 63 | |
| 64 ## size for repmat | |
| 65 s = shape; | |
| 66 s(i) = 1; | |
| 67 | |
| 5838 | 68 varargout{i} = repmat (reshape (varargin{i}, r), s); |
| 5837 | 69 endfor |
| 70 | |
| 71 endfunction |