annotate main/plot/quiver.m @ 0:6b33357c7561 octave-forge

Initial revision
author pkienzle
date Wed, 10 Oct 2001 19:54:49 +0000
parents
children 2ac2777b30bc
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
1 ## Copyright (C) 2000 John W. Eaton
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
2 ##
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
3 ## This program is free software; you can redistribute it and/or modify it
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
4 ## under the terms of the GNU General Public License as published by
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
5 ## the Free Software Foundation; either version 2, or (at your option)
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
6 ## any later version.
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
7 ##
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
8 ## This program is distributed in the hope that it will be useful, but
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
9 ## WITHOUT ANY WARRANTY; without even the implied warranty of
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
10 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
11 ## General Public License for more details.
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
12 ##
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
13 ## You should have received a copy of the GNU General Public License
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
14 ## along with this program; see the file COPYING. If not, write to the Free
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
15 ## Software Foundation, 59 Temple Place - Suite 330, Boston, MA
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
16 ## 02111-1307, USA.
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
17
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
18 ## usage: quiver (x, y, u, v)
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
19 ## usage: quiver (u, v)
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
20 ##
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
21 ## Plot the (u,v) components of a vector field in a (x,y) meshgrid.
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
22 ##
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
23 ## You can try:
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
24 ##
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
25 ## a = b = 1:10;
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
26 ## [x,y] = meshgrid(a,b);
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
27 ## u = rand (10,10);
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
28 ## v = rand (10,10);
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
29 ## quiver(x,y,u,v)
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
30 ##
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
31 ## See also: plot, semilogx, semilogy, loglog, polar, mesh, contour,
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
32 ## bar, stairs, gplot, gsplot, replot, xlabel, ylabel, title
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
33
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
34 ## Author: Roberto A. F. Almeida (roberto@calvin.ocfis.furg.br)
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
35 ## 2001-02-25 Paul Kienzle
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
36 ## * change parameter order to correspond to matlab
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
37 ## * allow vector x,y, defaulting to 1:n, 1:m for mxn u
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
38 ## * vectorize
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
39
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
40 ## TODO: use gnuplot 'vector' style instead of setting arrows one at a time
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
41 function quiver (x, y, u, v)
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
42
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
43 if (nargin != 2 && nargin != 4)
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
44 usage ("quiver (x, y, u, v)");
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
45 endif
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
46
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
47 if (nargin == 2)
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
48 u = x; v = y; x = 1:columns(u); y= 1:rows(u);
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
49 endif
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
50
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
51 if (any(size(u)!=size(v)))
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
52 error ("quiver: u, v must have the same shape.");
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
53 endif
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
54
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
55 if is_vector(x) && is_vector(y)
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
56 [nr, nc] = size(u);
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
57 if (length(x) != nc || length(y) != nr)
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
58 error ("quiver: x, y vectors must have correct length");
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
59 endif
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
60 [x,y] = meshgrid(x,y);
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
61 endif
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
62
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
63 if (any (size(x) != size(u) | size(y) != size(u)))
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
64 error ("quiver: x, y must have the same shape as u, v");
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
65 endif
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
66
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
67
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
68 ## Calculating the grid limits.
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
69
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
70 minx = min (min (x));
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
71 maxx = max (max (x));
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
72 miny = min (min (y));
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
73 maxy = max (max (y));
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
74
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
75 max_arrow = max ( [max(max (u)) , max(max (v)) , ...
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
76 abs(min (min (u))) , abs(min (min (v)))] );
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
77 border = max_arrow * 1.2;
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
78
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
79 limits = [ minx - border, maxx + border, miny - border, maxy + border ];
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
80
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
81 axis (limits);
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
82
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
83
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
84 ## Ok, now plot the arrows.
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
85
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
86 gset noarrow;
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
87 gplot 0 title "";
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
88
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
89 command = sprintf ("gset arrow from %f,%f to %f,%f\n", \
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
90 [ x(:)'; y(:)'; x(:)'+u(:)'; y(:)'+v(:)' ]);
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
91 eval ( [ "if 1\n", command, "\nendif" ] );
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
92
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
93 replot;
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
94
6b33357c7561 Initial revision
pkienzle
parents:
diff changeset
95 endfunction