Mercurial > hg > octave-lyh

annotate scripts/general/cart2sph.m @ 17535:c12c688a35ed default tip lyh

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Fix warnings
author LYH <lyh.kernel@gmail.com>
date Fri, 27 Sep 2013 17:43:27 +0800
parents f3d52523cde1
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
14138
72c96de7a403 maint: update copyright notices for 2012
John W. Eaton <jwe@octave.org>
parents: 11587
diff changeset
1 ## Copyright (C) 2000-2012 Kai Habel
3803
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
2 ##
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
3 ## This file is part of Octave.
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
4 ##
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
5 ## Octave is free software; you can redistribute it and/or modify it
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
6 ## under the terms of the GNU General Public License as published by
7016
93c65f2a5668 [project @ 2007-10-12 06:40:56 by jwe]
jwe
parents: 6786
diff changeset
7 ## the Free Software Foundation; either version 3 of the License, or (at
93c65f2a5668 [project @ 2007-10-12 06:40:56 by jwe]
jwe
parents: 6786
diff changeset
8 ## your option) any later version.
3803
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
9 ##
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
10 ## Octave is distributed in the hope that it will be useful, but
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
11 ## WITHOUT ANY WARRANTY; without even the implied warranty of
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
12 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
13 ## General Public License for more details.
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
14 ##
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
15 ## You should have received a copy of the GNU General Public License
7016
93c65f2a5668 [project @ 2007-10-12 06:40:56 by jwe]
jwe
parents: 6786
diff changeset
16 ## along with Octave; see the file COPYING. If not, see
93c65f2a5668 [project @ 2007-10-12 06:40:56 by jwe]
jwe
parents: 6786
diff changeset
17 ## <http://www.gnu.org/licenses/>.
3803
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
18
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
19 ## -*- texinfo -*-
10688
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
20 ## @deftypefn {Function File} {[@var{theta}, @var{phi}, @var{r}] =} cart2sph (@var{x}, @var{y}, @var{z})
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
21 ## @deftypefnx {Function File} {[@var{theta}, @var{phi}, @var{r}] =} cart2sph (@var{C})
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
22 ## @deftypefnx {Function File} {@var{S} =} cart2sph (@dots{})
9168
742cf6388a8f Update section 17.7 (Coordinate Transformations) of arith.txi
Rik <rdrider0-list@yahoo.com>
parents: 8920
diff changeset
23 ## Transform Cartesian to spherical coordinates.
10688
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
24 ##
9168
742cf6388a8f Update section 17.7 (Coordinate Transformations) of arith.txi
Rik <rdrider0-list@yahoo.com>
parents: 8920
diff changeset
25 ## @var{theta} describes the angle relative to the positive x-axis.
8491
aaff46fef256 [docs] fix hypenation: x - axis => x-axis, etc
Brian Gough <bjg@gnu.org>
parents: 7017
diff changeset
26 ## @var{phi} is the angle relative to the xy-plane.
10687
a8ce6bdecce5 Improve documentation strings.
Rik <octave@nomad.inbox5.com>
parents: 9168
diff changeset
27 ## @var{r} is the distance to the origin @w{(0, 0, 0)}.
10688
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
28 ## @var{x}, @var{y}, and @var{z} must be the same shape, or scalar.
11587
c792872f8942 all script files: untabify and strip trailing whitespace
John W. Eaton <jwe@octave.org>
parents: 11523
diff changeset
29 ## If called with a single matrix argument then each row of @var{c}
10688
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
30 ## represents the Cartesian coordinate (@var{x}, @var{y}, @var{z}).
11587
c792872f8942 all script files: untabify and strip trailing whitespace
John W. Eaton <jwe@octave.org>
parents: 11523
diff changeset
31 ##
c792872f8942 all script files: untabify and strip trailing whitespace
John W. Eaton <jwe@octave.org>
parents: 11523
diff changeset
32 ## If only a single return argument is requested then return a matrix
c792872f8942 all script files: untabify and strip trailing whitespace
John W. Eaton <jwe@octave.org>
parents: 11523
diff changeset
33 ## @var{s} where each row represents one spherical coordinate
10688
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
34 ## (@var{theta}, @var{phi}, @var{r}).
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
35 ## @seealso{sph2cart, cart2pol, pol2cart}
3803
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
36 ## @end deftypefn
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
37
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
38 ## Author: Kai Habel <kai.habel@gmx.de>
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
39 ## Adapted-by: jwe
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
40
8533
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
41 function [theta, phi, r] = cart2sph (x, y, z)
3803
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
42
10688
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
43 if (nargin != 1 && nargin != 3)
6046
34f96dd5441b [project @ 2006-10-10 16:10:25 by jwe]
jwe
parents: 5642
diff changeset
44 print_usage ();
3803
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
45 endif
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
46
10688
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
47 if (nargin == 1)
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
48 if (ismatrix (x) && columns (x) == 3)
11587
c792872f8942 all script files: untabify and strip trailing whitespace
John W. Eaton <jwe@octave.org>
parents: 11523
diff changeset
49 z = x(:,3);
c792872f8942 all script files: untabify and strip trailing whitespace
John W. Eaton <jwe@octave.org>
parents: 11523
diff changeset
50 y = x(:,2);
c792872f8942 all script files: untabify and strip trailing whitespace
John W. Eaton <jwe@octave.org>
parents: 11523
diff changeset
51 x = x(:,1);
10688
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
52 else
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
53 error ("cart2sph: matrix input must have 3 columns [X, Y, Z]");
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
54 endif
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
55 elseif (nargin == 3)
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
56 if (! ((ismatrix (x) && ismatrix (y) && ismatrix (z))
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
57 && (size_equal (x, y) || isscalar (x) || isscalar (y))
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
58 && (size_equal (x, z) || isscalar (x) || isscalar (z))
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
59 && (size_equal (y, z) || isscalar (y) || isscalar (z))))
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
60 error ("cart2sph: X, Y, Z must be matrices of the same size, or scalar");
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
61 endif
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
62 endif
8533
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
63
10688
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
64 theta = atan2 (y, x);
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
65 phi = atan2 (z, sqrt (x .^ 2 + y .^ 2));
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
66 r = sqrt (x .^ 2 + y .^ 2 + z .^ 2);
8533
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
67
10688
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
68 if (nargout <= 1)
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
69 theta = [theta, phi, r];
3803
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
70 endif
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
71
8533
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
72 endfunction
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
73
14363
f3d52523cde1 Use Octave coding conventions in all m-file %!test blocks
Rik <octave@nomad.inbox5.com>
parents: 14138
diff changeset
74
8533
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
75 %!test
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
76 %! x = [0, 1, 2];
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
77 %! y = [0, 1, 2];
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
78 %! z = [0, 1, 2];
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
79 %! [t, p, r] = cart2sph (x, y, z);
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
80 %! assert (t, [0, pi/4, pi/4], eps);
14363
f3d52523cde1 Use Octave coding conventions in all m-file %!test blocks
Rik <octave@nomad.inbox5.com>
parents: 14138
diff changeset
81 %! assert (p, [0, 1, 1]*atan (sqrt (0.5)), eps);
f3d52523cde1 Use Octave coding conventions in all m-file %!test blocks
Rik <octave@nomad.inbox5.com>
parents: 14138
diff changeset
82 %! assert (r, [0, 1, 2]*sqrt (3), eps);
3803
63c75bc3db82 [project @ 2001-02-28 08:24:40 by jwe]
jwe
parents:
diff changeset
83
8533
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
84 %!test
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
85 %! x = 0;
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
86 %! y = [0, 1, 2];
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
87 %! z = [0, 1, 2];
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
88 %! [t, p, r] = cart2sph (x, y, z);
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
89 %! assert (t, [0, 1, 1] * pi/2, eps);
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
90 %! assert (p, [0, 1, 1] * pi/4, eps);
14363
f3d52523cde1 Use Octave coding conventions in all m-file %!test blocks
Rik <octave@nomad.inbox5.com>
parents: 14138
diff changeset
91 %! assert (r, [0, 1, 2] * sqrt (2), eps);
8533
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
92
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
93 %!test
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
94 %! x = [0, 1, 2];
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
95 %! y = 0;
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
96 %! z = [0, 1, 2];
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
97 %! [t, p, r] = cart2sph (x, y, z);
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
98 %! assert (t, [0, 0, 0]);
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
99 %! assert (p, [0, 1, 1] * pi/4);
14363
f3d52523cde1 Use Octave coding conventions in all m-file %!test blocks
Rik <octave@nomad.inbox5.com>
parents: 14138
diff changeset
100 %! assert (r, [0, 1, 2] * sqrt (2));
8533
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
101
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
102 %!test
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
103 %! x = [0, 1, 2];
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
104 %! y = [0, 1, 2];
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
105 %! z = 0;
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
106 %! [t, p, r] = cart2sph (x, y, z);
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
107 %! assert (t, [0, 1, 1] * pi/4);
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
108 %! assert (p, [0, 0, 0]);
14363
f3d52523cde1 Use Octave coding conventions in all m-file %!test blocks
Rik <octave@nomad.inbox5.com>
parents: 14138
diff changeset
109 %! assert (r, [0, 1, 2] * sqrt (2));
8533
fb1b87ea4af9 Permit scalars when transforming coordinates.
Ben Abbott <bpabbott@mac.com>
parents: 8491
diff changeset
110
10688
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
111 %!test
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
112 %! C = [0, 0, 0; 1, 0, 1; 2, 0, 2];
7357e37f34fa coordinate transforms: add option to operate on column matrix of coordinates.
Rik <octave@nomad.inbox5.com>
parents: 10687
diff changeset
113 %! S = [0, 0, 0; 0, pi/4, sqrt(2); 0, pi/4, 2*sqrt(2)];
14363
f3d52523cde1 Use Octave coding conventions in all m-file %!test blocks
Rik <octave@nomad.inbox5.com>
parents: 14138
diff changeset
114 %! assert (cart2sph (C), S, eps);
f3d52523cde1 Use Octave coding conventions in all m-file %!test blocks
Rik <octave@nomad.inbox5.com>
parents: 14138
diff changeset
115