|
7189
|
1 ## Copyright (C) 2007 David Bateman |
|
|
2 ## |
|
|
3 ## This file is part of Octave. |
|
|
4 ## |
|
|
5 ## Octave is free software; you can redistribute it and/or modify it |
|
|
6 ## under the terms of the GNU General Public License as published by |
|
|
7 ## the Free Software Foundation; either version 3 of the License, or (at |
|
|
8 ## your option) any later version. |
|
|
9 ## |
|
|
10 ## Octave is distributed in the hope that it will be useful, but |
|
|
11 ## WITHOUT ANY WARRANTY; without even the implied warranty of |
|
|
12 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
|
|
13 ## General Public License for more details. |
|
|
14 ## |
|
|
15 ## You should have received a copy of the GNU General Public License |
|
|
16 ## along with Octave; see the file COPYING. If not, see |
|
|
17 ## <http://www.gnu.org/licenses/>. |
|
|
18 |
|
|
19 ## Undocumented internal function |
|
|
20 |
|
|
21 function hlist = __quiver__ (varargin) |
|
7191
|
22 |
|
|
23 h = varargin{1}; |
|
|
24 is3d = varargin{2}; |
|
7189
|
25 |
|
|
26 s = 1; |
|
|
27 arrowsize = 0.33; |
|
|
28 |
|
|
29 firstnonnumeric = Inf; |
|
|
30 for i = 3:nargin |
|
7191
|
31 if (! isnumeric (varargin{i})) |
|
7189
|
32 firstnonnumeric = i; |
|
|
33 break; |
|
|
34 endif |
|
|
35 endfor |
|
|
36 |
|
|
37 ioff = 3; |
|
|
38 if (nargin < (6 + is3d) || firstnonnumeric < (6 + is3d)) |
|
|
39 u = varargin{ioff++}; |
|
|
40 v = varargin{ioff++}; |
|
|
41 if (is3d) |
|
|
42 w = varargin{ioff++} |
|
|
43 [x, y, z] = meshgrid (1:size(u,1), 1:size(u,2), 1:max(size(w))); |
|
|
44 else |
|
|
45 [x, y] = meshgrid (1:size(u,1), 1:size(u,2)); |
|
|
46 endif |
|
7191
|
47 if (nargin >= ioff && isnumeric (varargin{ioff}) |
|
|
48 && isscalar (varargin{ioff})) |
|
7189
|
49 s = varargin{ioff++}; |
|
|
50 endif |
|
|
51 else |
|
|
52 x = varargin{ioff++}; |
|
|
53 y = varargin{ioff++}; |
|
|
54 if (is3d) |
|
|
55 z = varargin{ioff++}; |
|
|
56 endif |
|
|
57 u = varargin{ioff++}; |
|
|
58 v = varargin{ioff++}; |
|
|
59 if (is3d) |
|
|
60 w = varargin{ioff++}; |
|
7191
|
61 if (isvector (x) && isvector (y) && isvector (z) |
|
|
62 && (! isvector (u) || ! isvector (v) || ! isvector(w))) |
|
7189
|
63 [x, y, z] = meshgrid (x, y, z); |
|
|
64 endif |
|
|
65 else |
|
7191
|
66 if (isvector (x) && isvector (y) && (! isvector (u) || ! isvector (v))) |
|
7189
|
67 [x, y] = meshgrid (x, y); |
|
|
68 endif |
|
|
69 endif |
|
7191
|
70 if (nargin >= ioff && isnumeric (varargin{ioff}) |
|
|
71 && isscalar (varargin{ioff})) |
|
7189
|
72 s = varargin{ioff++}; |
|
|
73 endif |
|
|
74 endif |
|
|
75 |
|
|
76 have_filled = false; |
|
|
77 have_line_spec = false; |
|
|
78 while (ioff <= nargin) |
|
7191
|
79 arg = varargin{ioff++}; |
|
7189
|
80 if (ischar (arg) && strncmp (tolower (arg), "filled", 6)) |
|
|
81 have_filled = true; |
|
|
82 elseif ((isstr (arg) || iscell (arg)) |
|
|
83 && ! have_line_spec) |
|
|
84 [linespec, valid] = __pltopt__ ("quiver", arg, false); |
|
|
85 if (valid) |
|
|
86 have_line_spec = true; |
|
|
87 if (strncmp (linespec.linestyle, "none", 4)) |
|
|
88 linespec.linestyle = "-"; |
|
|
89 endif |
|
|
90 else |
|
|
91 error ("quiver: invalid linespec"); |
|
|
92 endif |
|
|
93 else |
|
|
94 error ("quiver: unrecognized argument"); |
|
|
95 endif |
|
|
96 endwhile |
|
|
97 |
|
|
98 if (s) |
|
|
99 ## Scale the arrows to fit in the grid |
|
|
100 dx = (max(x(:)) - min(x(:))) ./ size (x, 2); |
|
|
101 dy = (max(y(:)) - min(y(:))) ./ size (y, 1); |
|
|
102 if (is3d) |
|
|
103 ## What should this be divided by? The below seems right |
|
|
104 dz = (max(z(:)) - min(z(:))) ./ max (size (z)); |
|
|
105 len = max (sqrt (u(:).^2 + dy(:).^2) + dz(:).^2); |
|
|
106 else |
|
|
107 len = max (sqrt (u(:).^2 + dy(:).^2)); |
|
|
108 dz = 0; |
|
|
109 endif |
|
|
110 if (len > 0) |
|
|
111 s = s / sqrt (2) * sqrt (dx.^2 + dy.^2 + dz.^2) / len; |
|
|
112 u = s * u; |
|
|
113 v = s * v; |
|
|
114 if (is3d) |
|
|
115 w = s*w; |
|
|
116 endif |
|
|
117 endif |
|
|
118 endif |
|
|
119 |
|
|
120 x = x(:); |
|
|
121 y = y(:); |
|
|
122 xend = x + u(:); |
|
|
123 yend = y + v(:); |
|
|
124 if (is3d) |
|
|
125 z = z(:); |
|
|
126 zend = z + w(:); |
|
|
127 endif |
|
|
128 |
|
|
129 hstate = get (h, "nextplot"); |
|
|
130 unwind_protect |
|
|
131 if (have_line_spec) |
|
|
132 if (is3d) |
|
|
133 h1 = plot3 ([x.'; xend.'; NaN(1, length (x))](:), |
|
|
134 [y.'; yend.'; NaN(1, length (y))](:), |
|
|
135 [z.'; zend.'; NaN(1, length (z))](:), |
|
|
136 "linestyle", linespec.linestyle); |
|
|
137 else |
|
|
138 h1 = plot ([x.'; xend.'; NaN(1, length (x))](:), |
|
|
139 [y.'; yend.'; NaN(1, length (y))](:), |
|
|
140 "linestyle", linespec.linestyle); |
|
|
141 endif |
|
|
142 else |
|
|
143 if (is3d) |
|
|
144 h1 = plot3 ([x.'; xend.'; NaN(1, length (x))](:), |
|
|
145 [y.'; yend.'; NaN(1, length (y))](:), |
|
|
146 [z.'; zend.'; NaN(1, length (z))](:)); |
|
|
147 else |
|
|
148 h1 = plot ([x.'; xend.'; NaN(1, length (x))](:), |
|
|
149 [y.'; yend.'; NaN(1, length (y))](:)); |
|
|
150 endif |
|
|
151 endif |
|
|
152 hold on; |
|
|
153 |
|
|
154 xtmp = x + u(:) .* (1 - arrowsize); |
|
|
155 ytmp = y + v(:) .* (1 - arrowsize); |
|
|
156 xarrw1 = xtmp + (y - yend) * arrowsize / 3; |
|
|
157 xarrw2 = xtmp - (y - yend) * arrowsize / 3; |
|
|
158 yarrw1 = ytmp - (x - xend) * arrowsize / 3; |
|
|
159 yarrw2 = ytmp + (x - xend) * arrowsize / 3; |
|
|
160 if (is3d) |
|
|
161 zarrw1 = zarrw2 = zend - w(:) * arrowsize / 3; |
|
|
162 endif |
|
|
163 |
|
|
164 if (have_line_spec) |
|
|
165 if (isfield (linespec, "marker") && |
|
|
166 ! strncmp (linespec.marker, "none", 4)) |
|
|
167 if (is3d) |
|
|
168 h2 = plot3 ([xarrw1.'; xend.'; xarrw2.'; NaN(1, length (x))](:), |
|
|
169 [yarrw1.'; yend.'; yarrw2.'; NaN(1, length (y))](:), |
|
|
170 [zarrw1.'; zend.'; zarrw2.'; NaN(1, length (z))](:), |
|
|
171 "linestyle", "none"); |
|
|
172 else |
|
|
173 h2 = plot ([xarrw1.'; xend.'; xarrw2.'; NaN(1, length (x))](:), |
|
|
174 [yarrw1.'; yend.'; yarrw2.'; NaN(1, length (y))](:), |
|
|
175 "linestyle", "none"); |
|
|
176 endif |
|
|
177 else |
|
|
178 if (is3d) |
|
|
179 h2 = plot3 ([xarrw1.'; xend.'; xarrw2.'; NaN(1, length (x))](:), |
|
|
180 [yarrw1.'; yend.'; yarrw2.'; NaN(1, length (y))](:), |
|
|
181 [zarrw1.'; zend.'; zarrw2.'; NaN(1, length (z))](:), |
|
|
182 "linestyle", linespec.linestyle); |
|
|
183 else |
|
|
184 h2 = plot ([xarrw1.'; xend.'; xarrw2.'; NaN(1, length (x))](:), |
|
|
185 [yarrw1.'; yend.'; yarrw2.'; NaN(1, length (y))](:), |
|
|
186 "linestyle", linespec.linestyle); |
|
|
187 endif |
|
|
188 endif |
|
|
189 elseif (is3d) |
|
|
190 h2 = plot3 ([xarrw1.'; xend.'; xarrw2.'; NaN(1, length (x))](:), |
|
|
191 [yarrw1.'; yend.'; yarrw2.'; NaN(1, length (y))](:), |
|
|
192 [zarrw1.'; zend.'; zarrw2.'; NaN(1, length (z))](:)); |
|
|
193 else |
|
|
194 h2 = plot ([xarrw1.'; xend.'; xarrw2.'; NaN(1, length (x))](:), |
|
|
195 [yarrw1.'; yend.'; yarrw2.'; NaN(1, length (y))](:)); |
|
|
196 endif |
|
|
197 |
|
7191
|
198 if (! have_line_spec |
|
|
199 || (isfield (linespec, "marker") |
|
|
200 && strncmp (linespec.marker, "none", 4))) |
|
7189
|
201 if (is3d) |
|
|
202 h3 = plot3 (x, y, z, "linestyle", "none", "marker", "none"); |
|
|
203 else |
|
|
204 h3 = plot (x, y, "linestyle", "none", "marker", "none"); |
|
|
205 endif |
|
|
206 else |
|
|
207 if (is3d) |
|
|
208 h3 = plot3 (x, y, z, "linestyle", "none", "marker", linespec.marker); |
|
|
209 else |
|
|
210 |
|
|
211 h3 = plot (x, y, "linestyle", "none", "marker", linespec.marker); |
|
|
212 endif |
|
|
213 endif |
|
|
214 if (have_filled) |
|
|
215 ## FIXME gnuplot doesn't respect the markerfacecolor field |
|
7191
|
216 set (h3, "markerfacecolor", get (h1, "color")); |
|
7189
|
217 endif |
|
|
218 unwind_protect_cleanup |
|
|
219 set (h, "nextplot", hstate); |
|
|
220 end_unwind_protect |
|
|
221 |
|
|
222 hlist = [h1; h2; h3]; |
|
|
223 |
|
|
224 endfunction |
