Mercurial > hg > octave-lyh
annotate scripts/plot/__ezplot__.m @ 7650:eb7bdde776f2
Texinfo fixes
| author | John W. Eaton <jwe@octave.org> |
|---|---|
| date | Wed, 26 Mar 2008 16:27:14 -0400 |
| parents | 3422f39573b1 |
| children | c4482fc30c7f |
| rev | line source |
|---|---|
| 7337 | 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 | |
|
7540
3422f39573b1
strcat.m: Matlab compatibility, with cstrcat.m replacing conventional strcat.m.
Ben Abbott <bpabbott@mac.com>
parents:
7337
diff
changeset
|
23 func = cstrcat ("ez", pfunc); |
| 7337 | 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 '\./', '/'), '[\.]*\*', ''); | |
|
7540
3422f39573b1
strcat.m: Matlab compatibility, with cstrcat.m replacing conventional strcat.m.
Ben Abbott <bpabbott@mac.com>
parents:
7337
diff
changeset
|
257 fstr = cstrcat ("[",fstrx,",",fstry,",",fstrz,"]"); |
| 7337 | 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 |