Mercurial > octave
diff scripts/plot/draw/quiver3.m @ 32074:03fe0b635d2e
quiver/quiver3: Overhaul input processing, validation, and add BISTs.
* scripts/plot/draw/private/__quiver__.m: Overhaul numeric input validation.
Simplify input classification using numeric input count switch statements
and avoid quiver3 miscount due to scale factor. Add error messages for all
valid numeric input combinations including vector x,y,z and scale factor.
Move newplot command from quiver/quiver3 into __quiver__ after numeric input
validation. Add hax as an output argument to return any changes back to
calling function.
* scripts/plot/draw/quiver.m: Remove newplot call. Update __quiver__ call
to include hax as a return variable. Update docstring with note that line
style and name-value pairs can both be provided but linstyle must appear
first. Add BISTs to check standard inputs with single and multiple arrows,
arrowhead shape, vector and array inputs, proper treatment of scaling factor
"off", some simple input styles, and input validation BISTs to cover all
numeric input errors. Added known failing BIST for linestyle+pair
arrowhead showing when it should stay off (bug #64143).
* scripts/plot/draw/quiver3.m: Remove newplot call. Update __quiver__ call
to include hax as a return variable. Update docstring with note that line
style and name-value pairs can both be provided but linstyle must appear
first. Add BISTs to check standard inputs with single and multiple arrows,
vector and array inputs, and input validation BISTs to cover all numeric
input errors.
* etc/NEWS.9.md: Update quiver/quiver3 improvement description under General
Improvements.
| author | Nicholas R. Jankowski <jankowski.nicholas@gmail.com> |
|---|---|
| date | Wed, 03 May 2023 22:52:33 -0400 |
| parents | ada96a467a28 |
| children | 998cba4600e9 |
line wrap: on
line diff
--- a/scripts/plot/draw/quiver3.m Wed May 03 20:45:33 2023 +0200 +++ b/scripts/plot/draw/quiver3.m Wed May 03 22:52:33 2023 -0400 @@ -36,7 +36,7 @@ ## ## Plot the (@var{u}, @var{v}, @var{w}) components of a vector field at the ## grid points defined by (@var{x}, @var{y}, @var{z}). If the grid is uniform -## then @var{x}, @var{y}, and @var{z} can be specified as vectors and +## then @var{x}, @var{y}, and @var{z} can be specified as grid vectors and ## @code{meshgrid} is used to create the 3-D grid. ## ## If @var{x} and @var{y} are not given they are assumed to be @@ -53,7 +53,8 @@ ## then the markers are drawn at the origin of the vectors (which are the grid ## points defined by @var{x}, @var{y}, @var{z}). When a marker is specified, ## the arrowhead is not drawn. If the argument @qcode{"filled"} is given then -## the markers are filled. +## the markers are filled. If name-value plot style properties are used, they +## must appear in pairs and follow any other plot style arguments. ## ## If the first argument @var{hax} is an axes handle, then plot into this axes, ## rather than the current axes returned by @code{gca}. @@ -89,8 +90,7 @@ oldfig = get (0, "currentfigure"); endif unwind_protect - hax = newplot (hax); - htmp = __quiver__ (hax, true, varargin{:}); + [hax, htmp] = __quiver__ (hax, true, varargin{:}); if (! ishold ()) set (hax, "view", [-37.5, 30], @@ -108,7 +108,6 @@ endfunction - %!demo %! clf; %! colormap ("default"); @@ -135,9 +134,238 @@ %! title ({"quiver3() of surface normals to peaks() function"; ... %! 'shading "interp"'}); +## Check standard inputs, single arrow. +%!test +%! hf = figure ("visible", "off"); +%! hax = gca (); +%! unwind_protect +%! +%! h = quiver3 (hax, 0, 1, 2, 3); +%! children = get (h, "children"); +%! childxdata = get (children, "xdata"); +%! childydata = get (children, "ydata"); +%! childzdata = get (children, "zdata"); +%! stemchild = find (cellfun (@numel, childxdata) == 3); +%! arrowheadchild = find (cellfun (@numel, childxdata) == 4); +%! assert (childxdata{stemchild}(1), 1, eps); +%! assert (childxdata{stemchild}(2), 1 + 1*0.9, eps); +%! assert (isnan (childxdata{stemchild}(3))); +%! assert (childxdata{arrowheadchild}(2), 1 + 1*0.9, eps); +%! assert (isnan (childxdata{arrowheadchild}(4))); +%! assert (childydata{stemchild}(1), 1, eps); +%! assert (childydata{stemchild}(2), 1 + 2*0.9, eps); +%! assert (isnan (childydata{stemchild}(3))); +%! assert (childydata{arrowheadchild}(2), 1 + 2*0.9, eps); +%! assert (isnan (childydata{arrowheadchild}(4))); +%! assert (childzdata{stemchild}(1), 0, eps); +%! assert (childzdata{stemchild}(2), 0 + 3*0.9, eps); +%! assert (isnan (childzdata{stemchild}(3))); +%! assert (childzdata{arrowheadchild}(2), 0 + 3*0.9, eps); +%! assert (isnan (childzdata{arrowheadchild}(4))); +%! +%! h = quiver3 (hax, 1, 1, 0, 1, 2, 3); +%! children = get (h, "children"); +%! childxdata = get (children, "xdata"); +%! childydata = get (children, "ydata"); +%! childzdata = get (children, "zdata"); +%! stemchild = find (cellfun (@numel, childxdata) == 3); +%! arrowheadchild = find (cellfun (@numel, childxdata) == 4); +%! assert (childxdata{stemchild}(1), 1, eps); +%! assert (childxdata{stemchild}(2), 1 + 1*0.9, eps); +%! assert (isnan (childxdata{stemchild}(3))); +%! assert (childxdata{arrowheadchild}(2), 1 + 1*0.9, eps); +%! assert (isnan (childxdata{arrowheadchild}(4))); +%! assert (childydata{stemchild}(1), 1, eps); +%! assert (childydata{stemchild}(2), 1 + 2*0.9, eps); +%! assert (isnan (childydata{stemchild}(3))); +%! assert (childydata{arrowheadchild}(2), 1 + 2*0.9, eps); +%! assert (isnan (childydata{arrowheadchild}(4))); +%! assert (childzdata{stemchild}(1), 0, eps); +%! assert (childzdata{stemchild}(2), 0 + 3*0.9, eps); +%! assert (isnan (childzdata{stemchild}(3))); +%! assert (childzdata{arrowheadchild}(2), 0 + 3*0.9, eps); +%! assert (isnan (childzdata{arrowheadchild}(4))); +%! unwind_protect_cleanup +%! close (hf); +%! end_unwind_protect + +## Check standard inputs, multiple arrows. +%!test +%! hf = figure ("visible", "off"); +%! hax = gca (); +%! unwind_protect +%! +%! a = reshape(1:12,2,3,2); +%! x = 1:3; y = 1:2; z = 1:2; +%! [xx,yy,zz] = meshgrid (x,y,z); +%! numpts = 12; +%! sf= sqrt(sumsq([1/3 1/2 11/6])/432); # Actual internal scale factor, z=a. +%! sf2= sqrt(sumsq([1/3 1/2 1/6])/432); # z vector internal scale factor. +%! +%! h = quiver3 (hax, a, a, a, a, 1); # No x,y input. +%! children = get (h, "children"); +%! childxdata = get (children, "xdata"); +%! childydata = get (children, "ydata"); +%! childzdata = get (children, "zdata"); +%! basechild = find (cellfun (@numel, childxdata) == numpts); +%! stemchild = find (cellfun (@numel, childxdata) == numpts*3); +%! arrowheadchild = find (cellfun (@numel, childxdata) == numpts*4); +%! ## Check all bases. +%! assert (childxdata{basechild}, [1, 1, 2, 2, 3, 3, 1, 1, 2, 2, 3, 3]); +%! assert (childydata{basechild}, [1, 2, 1, 2, 1, 2, 1, 2, 1, 2, 1, 2]); +%! assert (childzdata{basechild}, [1:12]); +%! ## Check first arrow. +%! assert (childxdata{stemchild}(1), 1, eps); +%! assert (childxdata{stemchild}(2), 1 + 1*sf, eps); +%! assert (isnan (childxdata{stemchild}(3))); +%! assert (childxdata{arrowheadchild}(2), 1 + 1*sf, eps); +%! assert (isnan (childxdata{arrowheadchild}(4))); +%! assert (childydata{stemchild}(1), 1, eps); +%! assert (childydata{stemchild}(2), 1 + 1*sf, eps); +%! assert (isnan (childydata{stemchild}(3))); +%! assert (childydata{arrowheadchild}(2), 1 + 1*sf, eps); +%! assert (isnan (childydata{arrowheadchild}(4))); +%! assert (childzdata{stemchild}(1), 1, eps); +%! assert (childzdata{stemchild}(2), 1 + 1*sf, eps); +%! assert (isnan (childzdata{stemchild}(3))); +%! assert (childzdata{arrowheadchild}(2), 1 + 1*sf, eps); +%! assert (isnan (childzdata{arrowheadchild}(4))); +%! ## Check last arrow. +%! assert (childxdata{stemchild}(numpts*3-2), 3, eps); +%! assert (childxdata{stemchild}(numpts*3-1), 3 + 12*sf, eps); +%! assert (isnan (childxdata{stemchild}(end))); +%! assert (childxdata{arrowheadchild}(numpts*4-2), 3 + 12*sf, eps); +%! assert (isnan (childxdata{arrowheadchild}(end))); +%! assert (childydata{stemchild}(numpts*3-2), 2, eps); +%! assert (childydata{stemchild}(numpts*3-1), 2 + 12*sf, eps); +%! assert (isnan (childydata{stemchild}(end))); +%! assert (childydata{arrowheadchild}(numpts*4-2), 2 + 12*sf, eps); +%! assert (isnan (childydata{arrowheadchild}(end))); +%! assert (childzdata{stemchild}(numpts*3-2), 12, eps); +%! assert (childzdata{stemchild}(numpts*3-1), 12 + 12*sf, eps); +%! assert (isnan (childzdata{stemchild}(end))); +%! assert (childzdata{arrowheadchild}(numpts*4-2), 12 + 12*sf, eps); +%! assert (isnan (childzdata{arrowheadchild}(end))); +%! +%! h = quiver3 (hax, xx, yy, a, a, a, a, 1); # x,y input as matrices. +%! children = get (h, "children"); +%! childxdata = get (children, "xdata"); +%! childydata = get (children, "ydata"); +%! childzdata = get (children, "zdata"); +%! basechild = find (cellfun (@numel, childxdata) == numpts); +%! stemchild = find (cellfun (@numel, childxdata) == numpts*3); +%! arrowheadchild = find (cellfun (@numel, childxdata) == numpts*4); +%! ## Check all bases. +%! assert (childxdata{basechild}, [1, 1, 2, 2, 3, 3, 1, 1, 2, 2, 3, 3]); +%! assert (childydata{basechild}, [1, 2, 1, 2, 1, 2, 1, 2, 1, 2, 1, 2]); +%! assert (childzdata{basechild}, [1:12]); +%! ## Check first arrow. +%! assert (childxdata{stemchild}(1), 1, eps); +%! assert (childxdata{stemchild}(2), 1 + 1*sf, eps); +%! assert (isnan (childxdata{stemchild}(3))); +%! assert (childxdata{arrowheadchild}(2), 1 + 1*sf, eps); +%! assert (isnan (childxdata{arrowheadchild}(4))); +%! assert (childydata{stemchild}(1), 1, eps); +%! assert (childydata{stemchild}(2), 1 + 1*sf, eps); +%! assert (isnan (childydata{stemchild}(3))); +%! assert (childydata{arrowheadchild}(2), 1 + 1*sf, eps); +%! assert (isnan (childydata{arrowheadchild}(4))); +%! assert (childzdata{stemchild}(1), 1, eps); +%! assert (childzdata{stemchild}(2), 1 + 1*sf, eps); +%! assert (isnan (childzdata{stemchild}(3))); +%! assert (childzdata{arrowheadchild}(2), 1 + 1*sf, eps); +%! assert (isnan (childzdata{arrowheadchild}(4))); +%! ## Check last arrow. +%! assert (childxdata{stemchild}(numpts*3-2), 3, eps); +%! assert (childxdata{stemchild}(numpts*3-1), 3 + 12*sf, eps); +%! assert (isnan (childxdata{stemchild}(end))); +%! assert (childxdata{arrowheadchild}(numpts*4-2), 3 + 12*sf, eps); +%! assert (isnan (childxdata{arrowheadchild}(end))); +%! assert (childydata{stemchild}(numpts*3-2), 2, eps); +%! assert (childydata{stemchild}(numpts*3-1), 2 + 12*sf, eps); +%! assert (isnan (childydata{stemchild}(end))); +%! assert (childydata{arrowheadchild}(numpts*4-2), 2 + 12*sf, eps); +%! assert (isnan (childydata{arrowheadchild}(end))); +%! assert (childzdata{stemchild}(numpts*3-2), 12, eps); +%! assert (childzdata{stemchild}(numpts*3-1), 12 + 12*sf, eps); +%! assert (isnan (childzdata{stemchild}(end))); +%! assert (childzdata{arrowheadchild}(numpts*4-2), 12 + 12*sf, eps); +%! assert (isnan (childzdata{arrowheadchild}(end))); +%! +%! h = quiver3 (hax, x, y, z, a, a, a, 1); # x,y z input as vectors. +%! children = get (h, "children"); +%! childxdata = get (children, "xdata"); +%! childydata = get (children, "ydata"); +%! childzdata = get (children, "zdata"); +%! basechild = find (cellfun (@numel, childxdata) == numpts); +%! stemchild = find (cellfun (@numel, childxdata) == numpts*3); +%! arrowheadchild = find (cellfun (@numel, childxdata) == numpts*4); +%! ## Check all bases. +%! assert (childxdata{basechild}, [1, 1, 2, 2, 3, 3, 1, 1, 2, 2, 3, 3]); +%! assert (childydata{basechild}, [1, 2, 1, 2, 1, 2, 1, 2, 1, 2, 1, 2]); +%! assert (childzdata{basechild}, [1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2]); +%! ## Check first arrow. +%! assert (childxdata{stemchild}(1), 1, eps); +%! assert (childxdata{stemchild}(2), 1 + 1*sf2, eps); +%! assert (isnan (childxdata{stemchild}(3))); +%! assert (childxdata{arrowheadchild}(2), 1 + 1*sf2, eps); +%! assert (isnan (childxdata{arrowheadchild}(4))); +%! assert (childydata{stemchild}(1), 1, eps); +%! assert (childydata{stemchild}(2), 1 + 1*sf2, eps); +%! assert (isnan (childydata{stemchild}(3))); +%! assert (childydata{arrowheadchild}(2), 1 + 1*sf2, eps); +%! assert (isnan (childydata{arrowheadchild}(4))); +%! assert (childzdata{stemchild}(1), 1, eps); +%! assert (childzdata{stemchild}(2), 1 + 1*sf2, eps); +%! assert (isnan (childzdata{stemchild}(3))); +%! assert (childzdata{arrowheadchild}(2), 1 + 1*sf2, eps); +%! assert (isnan (childzdata{arrowheadchild}(4))); +%! ## Check last arrow. +%! assert (childxdata{stemchild}(numpts*3-2), 3, eps); +%! assert (childxdata{stemchild}(numpts*3-1), 3 + 12*sf2, eps); +%! assert (isnan (childxdata{stemchild}(end))); +%! assert (childxdata{arrowheadchild}(numpts*4-2), 3 + 12*sf2, eps); +%! assert (isnan (childxdata{arrowheadchild}(end))); +%! assert (childydata{stemchild}(numpts*3-2), 2, eps); +%! assert (childydata{stemchild}(numpts*3-1), 2 + 12*sf2, eps); +%! assert (isnan (childydata{stemchild}(end))); +%! assert (childydata{arrowheadchild}(numpts*4-2), 2 + 12*sf2, eps); +%! assert (isnan (childydata{arrowheadchild}(end))); +%! assert (childzdata{stemchild}(numpts*3-2), 2, eps); +%! assert (childzdata{stemchild}(numpts*3-1), 2 + 12*sf2, eps); +%! assert (isnan (childzdata{stemchild}(end))); +%! assert (childzdata{arrowheadchild}(numpts*4-2), 2 + 12*sf2, eps); +%! assert (isnan (childzdata{arrowheadchild}(end))); +%! unwind_protect_cleanup +%! close (hf); +%! end_unwind_protect + ##Test input validation %!error <Invalid call> quiver3 () %!error <Invalid call> quiver3 (1.1) %!error <Invalid call> quiver3 (1.1, 2) %!error <Invalid call> quiver3 (1.1, 2, 3) - +%!error <Invalid call> quiver3 (1.1, 2, 3, "foo") +%!error <Invalid call> quiver3 (1.1, 2, 3, 4, 5, 6, 7, 8, "foo") +%!error <U, V, and W must be the same> quiver3 (30, [40 50], 60, 70) +%!error <Z vector length must equal size of> quiver3 ([30 40], eye(3), eye(3), eye(3)) +%!error <Z, U, V, and W must be the same> quiver3 ([30 40], 50, 60, 70) +%!error <Z, U, V, and W must be the same> quiver3 (eye(2), eye(3), eye(2), eye(2)) +%!error <Z, U, V, and W must be the same> quiver3 (eye(2), eye(2), eye(3), eye(2)) +%!error <Z, U, V, and W must be the same> quiver3 (eye(2), eye(2), eye(2), eye(3)) +%!error <U, V, and W must be the same size> quiver3 ([1:2], [1:2], 1, eye(3), eye(2), eye(2)) +%!error <U, V, and W must be the same size> quiver3 ([1:2], [1:2], 1, eye(2), eye(3), eye(2)) +%!error <U, V, and W must be the same size> quiver3 ([1:2], [1:2], 1, eye(2), eye(2), eye(3)) +%!error <X vector length must equal number of> quiver3 ([1:3], [1:2], 1, eye(2), eye(2), eye(2)) +%!error <Y vector length must equal number of> quiver3 ([1:2], [1:3], 1, eye(2), eye(2), eye(2)) +%!error <Z vector length must equal size of> quiver3 ([1:2], [1:2], [1:2], eye(2), eye(2), eye(2)) +%!error <X, Y, Z, U, V, and W must be the same size> quiver3 (eye(3), eye(2), eye(2), eye(2), eye(2), eye(2)) +%!error <X, Y, Z, U, V, and W must be the same size> quiver3 (eye(2), eye(3), eye(2), eye(2), eye(2), eye(2)) +%!error <X, Y, Z, U, V, and W must be the same size> quiver3 (eye(2), eye(2), eye(3), eye(2), eye(2), eye(2)) +%!error <X, Y, Z, U, V, and W must be the same size> quiver3 (eye(2), eye(2), eye(2), eye(3), eye(2), eye(2)) +%!error <X, Y, Z, U, V, and W must be the same size> quiver3 (eye(2), eye(2), eye(2), eye(2), eye(3), eye(2)) +%!error <X, Y, Z, U, V, and W must be the same size> quiver3 (eye(2), eye(2), eye(2), eye(2), eye(2), eye(3)) +%!error <scaling factor must be> quiver3 (10, 20, 30, 40, -5) +%!error <scaling factor must be> quiver3 (10, 20, 30, 40, [1 2]) +%!error <scaling factor must be> quiver3 (10, 20, 30, 40, 50, 60, -5) +%!error <scaling factor must be> quiver3 (10, 20, 30, 40, 50, 60, [1 2])