1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
function [varargout] = ndgrid(varargin)
//
// CALLING SEQUENCES
// [X, Y] = ndgrid(x, y)
// or [X, Y, Z] = ndgrid(x, y, z)
// or [X, Y, Z, T] = ndgrid(x, y, z, t)
// etc ...
//
// PURPOSE
// An utility similar to the Matlab 's one. For instance
// in the 3d case :
//
// [X, Y, Z] = ndgrid(x, y, z)
//
// with x, y, z three vectors of length nx, ny, nz, we have
// X, Y, Z which are hypermatrices of format nx x ny x nz
// and, forall i in [1,nx], j in [1, ny], k in [1, nz]
// X(i,j,k) = x(i)
// Y(i,j,k) = y(j)
// Z(i,j,k) = z(k)
//
// X, Y, Z are the coordinates of the grid points formed by
// the cartesian product of the one dimensionnal grids
// x, y and z.
//
// AUTHOR
// Bruno Pincon <Bruno.Pincon@iecn.u-nancy.fr>
//
nbdim = length(varargin)
if nbdim < 2 then
error(" ndgrid must have at least 2 args")
end
dim = zeros(1,nbdim)
for k = 1:nbdim
if type(varargin(k)) ~= 1 then
error(" ndgrid : bad "+string(k)+" th arg")
end
dim(k) = length(varargin(k))
varargin(k) = matrix(varargin(k),1,dim(k)) // force row form
end
varargout = list()
for k = 1:nbdim
varargout(k) = matrix( ones(1,prod(dim(k+1:$))) ...
.*. varargin(k) ...
.*. ones(1,prod(dim(1:k-1))) , dim )
end
endfunction
|
