* Bug #14025 fixed - Extend ndgrid() to any homogeneous data type
[scilab.git] / scilab / modules / elementary_functions / macros / ndgrid.sci
index 5120739..c0f4383 100644 (file)
@@ -1,6 +1,6 @@
 // Scilab ( http://www.scilab.org/ ) - This file is part of Scilab
 // Copyright (C) Bruno Pincon
-//
+// Copyright (C) 2015 - Samuel GOUGEON
 // Copyright (C) 2012 - 2016 - Scilab Enterprises
 //
 // This file is hereby licensed under the terms of the GNU GPL v2.0,
 // along with this program.
 
 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 dimensional grids
-    //     x, y and z.
-    //
-    //  AUTHOR
-    //     Bruno Pincon <Bruno.Pincon@iecn.u-nancy.fr>
-    //
     nbdim = length(varargin)
     if nbdim < 2 then
-        error(msprintf(gettext("%s: Wrong number of input arguments: At least %d expected.\n"),"ndgrid",2));
+        msg = _("%s: Wrong number of input arguments: At least %d expected.\n")
+        error(msprintf(msg, "ndgrid", 2))
     end
+
     dim = zeros(1,nbdim)
     for k = 1:nbdim
-        if type(varargin(k)) ~= 1 then
-            error(msprintf(gettext("%s: Wrong type for argument #%d: Real expected.\n"),"ndgrid",k));
+        if and(type(varargin(k)) ~= [1 2 4 8 10]) & typeof(varargin(k))~="rational"
+            msg = _("%s: Wrong type for argument #%d: Booleans, Integers, Decimals, Complexes, Polynomials, Rationals or Texts expected.\n")
+            error(msprintf(msg, "ndgrid", k))
         end
-        dim(k) = length(varargin(k))
-        varargin(k) = matrix(varargin(k),1,dim(k)) // force row form
+        dim(k) = size(varargin(k),"*")
+        varargin(k) = matrix(varargin(k),1,-1) // force row form
     end
 
     varargout = list()
+    // Manage input []
+    for v = varargin
+        if v==[]
+            for i=1:nbdim
+                varargout(i) = []
+            end
+            return
+        end
+    end
+
+    // Processing:
     for k = 1:nbdim
-        varargout(k) = matrix(     ones(1,prod(dim(k+1:$))) ...
-        .*. varargin(k)              ...
-        .*. ones(1,prod(dim(1:k-1))) , dim )
+        tmp = varargin(k)
+        ind = ones(1,prod(dim(k+1:$))) .*. (1:size(tmp,"*")) ..
+        .*. ones(1,prod(dim(1:k-1)))
+        if typeof(tmp)~="rational"
+            varargout(k) = matrix( tmp(ind) , dim )
+        else
+            varargout(k) = rlist(matrix(tmp(2)(ind), dim), matrix(tmp(3)(ind), dim), [])
+        end
     end
 endfunction