Mercurial > hg > octave-nkf
diff scripts/plot/meshz.m @ 7207:71c03c7239fb
[project @ 2007-11-28 02:19:25 by jwe]
author | jwe |
---|---|
date | Wed, 28 Nov 2007 02:20:48 +0000 |
parents | |
children | a730e47fda4d |
line wrap: on
line diff
new file mode 100644 --- /dev/null +++ b/scripts/plot/meshz.m @@ -0,0 +1,100 @@ +## Copyright (C) 2007 David Bateman +## +## This file is part of Octave. +## +## Octave is free software; you can redistribute it and/or modify it +## under the terms of the GNU General Public License as published by +## the Free Software Foundation; either version 3 of the License, or (at +## your option) any later version. +## +## Octave is distributed in the hope that it will be useful, but +## WITHOUT ANY WARRANTY; without even the implied warranty of +## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +## General Public License for more details. +## +## You should have received a copy of the GNU General Public License +## along with Octave; see the file COPYING. If not, see +## <http://www.gnu.org/licenses/>. + +## -*- texinfo -*- +## @deftypefn {Function File} {} meshz (@var{x}, @var{y}, @var{z}) +## Plot a curtain mesh given matrices @var{x}, and @var{y} from +## @code{meshgrid} and a matrix @var{z} corresponding to the @var{x} and +## @var{y} coordinates of the mesh. If @var{x} and @var{y} are vectors, +## then a typical vertex is (@var{x}(j), @var{y}(i), @var{z}(i,j)). Thus, +## columns of @var{z} correspond to different @var{x} values and rows of +## @var{z} correspond to different @var{y} values. +## @seealso{meshgrid, mesh, contour} +## @end deftypefn + +function retval = meshz (varargin) + + if (isscalar (varargin{1}) && ishandle (varargin{1})) + h = varargin {1}; + if (! strcmp (get (h, "type"), "axes")) + error ("meshz: expecting first argument to be an axes object"); + endif + oldh = gca (); + unwind_protect + axes (h); + tmp = __meshz__ (varargin{2:end}); + unwind_protect_cleanup + axes (oldh); + end_unwind_protect + else + tmp = __meshz__ (varargin{:}); + endif + + if (nargout > 0) + retval = tmp; + endif + +endfunction + +function h = __meshz__ (varargin) + + ioff = nargin + 1; + for i = 1 : nargin + if (ischar (varargin {i})) + ioff = i; + break; + endif + endfor + + ## Bundle C matrix back into varargin + if (ioff == 3 || ioff == 5) + ioff --; + endif + + if (ioff == 2) + z = varargin {1}; + [m, n] = size (z); + x = 1 : n; + y = (1 : m).'; + else + x = varargin {1}; + y = varargin {2}; + z = varargin {3}; + endif + + + if (isvector (x) && isvector (y)) + x = [x(1), x(:).', x(end)]; + y = [y(1); y(:); y(end)]; + else + x = [x(1, 1), x(1, :), x(1, end); + x(:, 1), x, x(:, end); + x(end, 1), x(end, :), x(end, end)]; + y = [y(1, 1), y(1, :), y(1, end); + y(:, 1), y, y(:, end); + y(end, 1), y(end, :), y(end, end)]; + endif + + zref = min(z(isfinite(z))); + z = [zref .* ones(1, size(z, 2) + 2); + zref .* ones(size(z, 1), 1), z, zref .* ones(size(z, 1), 1); + zref.* ones(1, size(z, 2) + 2)]; + + h = mesh (x, y, z, varargin{ioff:end}); + +endfunction