Mercurial > hg > octave-lyh

comparison scripts/plot/__ezplot__.m @ 7337:e70789e0cd92

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
[project @ 2007-12-28 21:04:44 by jwe]
author jwe
date Fri, 28 Dec 2007 21:06:43 +0000
parents
children 3422f39573b1
comparison
equal deleted inserted replaced
7336:745a8299c2b5 7337:e70789e0cd92
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 [h, needusage] = __ezplot__ (pfunc, varargin)
22
23 func = strcat ("ez", pfunc);
24 if (strncmp (pfunc, "contour", 7))
25 iscontour = true;
26 else
27 iscontour = false;
28 endif
29 if (strcmp (pfunc, "plot3"))
30 isplot3 = true;
31 ispolar = false;
32 nargs = 1
33 elseif (strcmp (pfunc, "polar"))
34 isplot3 = false;
35 ispolar = true;
36 nargs = 1;
37 else
38 isplot3 = false;
39 ispolar = false;
40 nargs = 2;
41 endif
42
43 [ax, varargin, nargin] = __plt_get_axis_arg__ (func, varargin{:});
44
45 needusage = false;
46 if (nargin < 1)
47 needusage = true;
48 return;
49 endif
50
51 parametric = false;
52 fun = varargin {1};
53 if (ischar (fun))
54 fun = vectorize (inline (fun));
55 if (length (argnames (fun)) != nargs)
56 error ("%s: excepting a function of %d arguments", func, nargs);
57 endif
58 fstr = formula (fun);
59 if (isplot3)
60 xarg = "x";
61 yarg = "y";
62 elseif (ispolar)
63 xarg = "";
64 yarg = "";
65 else
66 xarg = argnames(fun){1};
67 yarg = argnames(fun){2};
68 endif
69 elseif (strcmp (typeinfo (fun), "inline function"))
70 if (length (argnames (fun)) != nargs)
71 error ("%s: excepting a function of %d arguments", func, nargs);
72 endif
73 fun = vectorize (fun);
74 fstr = formula (fun);
75 if (isplot3)
76 xarg = "x";
77 yarg = "y";
78 elseif (ispolar)
79 xarg = "";
80 yarg = "";
81 else
82 xarg = argnames(fun)(1);
83 yarg = argnames(fun)(2);
84 endif
85 elseif (isa (fun, "function_handle"))
86 fstr = func2str (fun);
87 args = regexp (substr (fstr, 3, findstr (fstr, ")")(1) - 3),
88 '(\w[\w\d]*)', 'tokens');
89 if (length (args) != nargs)
90 error ("%s: excepting a function of %d arguments", func, nargs);
91 endif
92 fstr = substr (fstr, findstr (fstr, ")")(1) + 1);
93 if (isplot3)
94 xarg = "x";
95 yarg = "y";
96 elseif (ispolar)
97 xarg = "";
98 yarg = "";
99 else
100 xarg = args{1}{1};
101 yarg = args{2}{1};
102 endif
103 else
104 error ("%s: expecting string, inline function or function handle", func);
105 endif
106
107 if (nargin > 2)
108 funx = fun;
109 fstrx = fstr;
110 funy = varargin {2};
111 if (ischar (funy) && ! strcmp (funy, "circ") && ! strcmp (funy, "animate"))
112 parametric = true;
113 funy = vectorize (inline (funy));
114 if (length (argnames (funy)) != nargs)
115 error ("%s: excepting a function of %d arguments", func, nargs);
116 endif
117 fstry = formula (funy);
118 elseif (strcmp (typeinfo (funy), "inline function"))
119 parametric = true;
120 if (length (argnames (funy)) != nargs)
121 error ("%s: excepting a function of %d arguments", func, nargs);
122 endif
123 funy = vectorize (funy);
124 fstry = formula (funy);
125 elseif (isa (funy, "function_handle"))
126 parametric = true;
127 fstry = func2str (funy);
128 args = regexp (substr (fstry, 3, findstr (fstry, ")")(1) - 3),
129 '(\w[\w\d]*)', 'tokens');
130 if (length (args) != nargs)
131 error ("%s: excepting a function of %d arguments", func, nargs);
132 endif
133 fstry = substr (fstry, findstr (fstry, ")")(1) + 1);
134 endif
135
136 if (parametric)
137 funz = varargin {3};
138 if (ischar (funz) && ! strcmp (funz, "circ") &&
139 ! strcmp (funy, "animate"))
140 funz = vectorize (inline (funz));
141 if (length (argnames (funz)) != nargs)
142 error ("%s: excepting a function of %d arguments", func, nargs);
143 endif
144 fstrz = formula (funz);
145 elseif (strcmp (typeinfo (funz), "inline function"))
146 if (length (argnames (funz)) != nargs)
147 error ("%s: excepting a function of %d arguments", func, nargs);
148 endif
149 funz = vectorize (funz);
150 fstrz = formula (funz);
151 elseif (isa (funz, "function_handle"))
152 fstrz = func2str (funz);
153 args = regexp (substr (fstrz, 3, findstr (fstrz, ")")(1) - 3),
154 '(\w[\w\d]*)', 'tokens');
155 if (length (args) != nargs)
156 error ("%s: excepting a function of %d arguments", func, nargs);
157 endif
158 fstrz = substr (fstrz, findstr (fstrz, ")")(1) + 1);
159 else
160 error ("%s: parametric plots expect 3 functions", func);
161 endif
162 endif
163 endif
164
165 n = 60;
166 domain = [];
167 circ = false;
168 animate = false;
169 if (parametric)
170 iarg = 4;
171 else
172 iarg = 2;
173 endif
174 while (iarg <= nargin)
175 arg = varargin{iarg++};
176 if (ischar (arg) && strcmp (arg, "circ"))
177 circ = true;
178 elseif (ischar (arg) && strcmp (arg, "animate"))
179 animate = true;
180 elseif (isscalar (arg))
181 n = arg;
182 elseif (numel (arg) == 2)
183 domain = [arg(:).' arg(:).'];
184 elseif (numel (arg) == 4)
185 domain = arg(:).';
186 else
187 error ("%s: expecting scalar, 2 or 4 element vector", func);
188 endif
189 endwhile
190
191 if (isempty (domain))
192 if (isplot3 || ispolar)
193 domain = [0, 2*pi, 0, 2*pi];
194 else
195 domain = [-2*pi, 2*pi, -2*pi, 2*pi];
196 endif
197 endif
198
199 if (circ)
200 if (iscontour || isplot3)
201 needusage = true;
202 return;
203 endif
204 if (parametric)
205 error ("%s: can not have both circular domain and parametric function",
206 func);
207 endif
208 cent = [domain(1) + domain(2), domain(3) + domain(4)] / 2;
209 funx = @(r,t) r .* cos (t) + cent (1);
210 funy = @(r,t) r .* sin (t) + cent (2);
211 domain = [0, sqrt((domain(2) - cent(1))^2 + (domain(4) - cent(2))^2), ...
212 -pi, pi];
213 funz = fun;
214 parametric = true;
215 endif
216
217 if (animate)
218 if (!isplot3)
219 error ("%s: animated graphs only valid with plot3", func);
220 endif
221 error ("%s: animated graphs not implemented", func);
222 endif
223
224 if (isplot3 || ispolar)
225 X = linspace (domain (1), domain (2), n);
226 else
227 x = linspace (domain (1), domain (2), n);
228 y = linspace (domain (3), domain (4), n);
229 [X, Y] = meshgrid (x, y);
230 endif
231 if (parametric)
232 if (isplot3)
233 Z = feval (funz, X);
234 XX = feval (funx, X);
235 YY = feval (funy, X);
236 X = XX;
237 Y = YY;
238 else
239 Z = feval (funz, X, Y);
240 XX = feval (funx, X, Y);
241 YY = feval (funy, X, Y);
242 X = XX;
243 Y = YY;
244
245 ## Eliminate the singularities
246 X = __eliminate_sing__ (X);
247 Y = __eliminate_sing__ (Y);
248 Z = __eliminate_sing__ (Z);
249 endif
250
251 fstrx = regexprep (regexprep (regexprep (fstrx,'\.\^\s*','^'),
252 '\./', '/'), '[\.]*\*', '');
253 fstry = regexprep (regexprep (regexprep (fstry,'\.\^\s*','^'),
254 '\./', '/'), '[\.]*\*', '');
255 fstrz = regexprep (regexprep (regexprep (fstrz,'\.\^\s*','^'),
256 '\./', '/'), '[\.]*\*', '');
257 fstr = strcat ("[",fstrx,",",fstry,",",fstrz,"]");
258 else
259 if (isplot3)
260 needusage = true;
261 return;
262 endif
263
264 if (ispolar)
265 Z = feval (fun, X);
266 else
267 Z = feval (fun, X, Y);
268
269 ## Eliminate the singularities
270 Z = __eliminate_sing__ (Z);
271 endif
272
273 fstr = regexprep (regexprep (regexprep (fstr,'\.\^\s*','^'), '\./', '/'),
274 '[\.]*\*', '');
275 endif
276
277 oldax = gca ();
278 unwind_protect
279 axes (ax);
280 if (iscontour)
281 [clev, h] = feval (pfunc, X, Y, Z);
282 elseif (ispolar)
283 h = feval (pfunc, X, Z);
284 else
285 h = feval (pfunc, X, Y, Z);
286 endif
287 xlabel (xarg);
288 ylabel (yarg);
289 title (fstr);
290 unwind_protect_cleanup
291 axes (oldax);
292 end_unwind_protect
293
294 endfunction
295
296 function x = __eliminate_sing__ (x)
297 x (isinf (x)) = NaN;
298 x (abs (del2 (x)) > 0.2 * (max(x(:)) - min(x(:)))) = NaN;
299 endfunction