Mercurial > octave
annotate scripts/general/divergence.m @ 23220:092078913d54
maint: Merge stable to default.
author | John W. Eaton <jwe@octave.org> |
---|---|
date | Wed, 22 Feb 2017 12:58:07 -0500 |
parents | ef4d915df748 3ac9f9ecfae5 |
children | 194eb4bd202b |
rev | line source |
---|---|
23219
3ac9f9ecfae5
maint: Update copyright dates.
John W. Eaton <jwe@octave.org>
parents:
23083
diff
changeset
|
1 ## Copyright (C) 2009-2017 Kai Habel |
11428 | 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 | |
22755
3a2b891d0b33
maint: Standardize Copyright formatting.
Rik <rik@octave.org>
parents:
22323
diff
changeset
|
7 ## the Free Software Foundation; either version 3 of the License, or |
3a2b891d0b33
maint: Standardize Copyright formatting.
Rik <rik@octave.org>
parents:
22323
diff
changeset
|
8 ## (at your option) any later version. |
11428 | 9 ## |
10 ## Octave is distributed in the hope that it will be useful, but | |
11 ## WITHOUT ANY WARRANTY; without even the implied warranty of | |
22755
3a2b891d0b33
maint: Standardize Copyright formatting.
Rik <rik@octave.org>
parents:
22323
diff
changeset
|
12 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
3a2b891d0b33
maint: Standardize Copyright formatting.
Rik <rik@octave.org>
parents:
22323
diff
changeset
|
13 ## GNU General Public License for more details. |
11428 | 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 ## -*- texinfo -*- | |
20852
516bb87ea72e
2015 Code Sprint: remove class of function from docstring for all m-files.
Rik <rik@octave.org>
parents:
20711
diff
changeset
|
20 ## @deftypefn {} {@var{div} =} divergence (@var{x}, @var{y}, @var{z}, @var{fx}, @var{fy}, @var{fz}) |
516bb87ea72e
2015 Code Sprint: remove class of function from docstring for all m-files.
Rik <rik@octave.org>
parents:
20711
diff
changeset
|
21 ## @deftypefnx {} {@var{div} =} divergence (@var{fx}, @var{fy}, @var{fz}) |
516bb87ea72e
2015 Code Sprint: remove class of function from docstring for all m-files.
Rik <rik@octave.org>
parents:
20711
diff
changeset
|
22 ## @deftypefnx {} {@var{div} =} divergence (@var{x}, @var{y}, @var{fx}, @var{fy}) |
516bb87ea72e
2015 Code Sprint: remove class of function from docstring for all m-files.
Rik <rik@octave.org>
parents:
20711
diff
changeset
|
23 ## @deftypefnx {} {@var{div} =} divergence (@var{fx}, @var{fy}) |
11575
d6619410e79c
Spellcheck documentation before 3.4 release.
Rik <octave@nomad.inbox5.com>
parents:
11563
diff
changeset
|
24 ## Calculate divergence of a vector field given by the arrays @var{fx}, |
11587
c792872f8942
all script files: untabify and strip trailing whitespace
John W. Eaton <jwe@octave.org>
parents:
11575
diff
changeset
|
25 ## @var{fy}, and @var{fz} or @var{fx}, @var{fy} respectively. |
20158
7503499a252b
doc: Update docstrings to have one sentence summary as first line.
Rik <rik@octave.org>
parents:
19833
diff
changeset
|
26 ## |
11428 | 27 ## @tex |
28 ## $$ | |
29 ## div F(x,y,z) = \partial_x{F} + \partial_y{F} + \partial_z{F} | |
30 ## $$ | |
31 ## @end tex | |
32 ## @ifnottex | |
11563
3c6e8aaa9555
Grammarcheck m-files before 3.4 release.
Rik <octave@nomad.inbox5.com>
parents:
11523
diff
changeset
|
33 ## |
11428 | 34 ## @example |
35 ## @group | |
36 ## d d d | |
37 ## div F(x,y,z) = -- F(x,y,z) + -- F(x,y,z) + -- F(x,y,z) | |
38 ## dx dy dz | |
39 ## @end group | |
40 ## @end example | |
11563
3c6e8aaa9555
Grammarcheck m-files before 3.4 release.
Rik <octave@nomad.inbox5.com>
parents:
11523
diff
changeset
|
41 ## |
11428 | 42 ## @end ifnottex |
11563
3c6e8aaa9555
Grammarcheck m-files before 3.4 release.
Rik <octave@nomad.inbox5.com>
parents:
11523
diff
changeset
|
43 ## The coordinates of the vector field can be given by the arguments @var{x}, |
11587
c792872f8942
all script files: untabify and strip trailing whitespace
John W. Eaton <jwe@octave.org>
parents:
11575
diff
changeset
|
44 ## @var{y}, @var{z} or @var{x}, @var{y} respectively. |
11428 | 45 ## |
12520
ad05e1547398
Add function chop to documentation.
Rik <octave@nomad.inbox5.com>
parents:
11589
diff
changeset
|
46 ## @seealso{curl, gradient, del2, dot} |
11428 | 47 ## @end deftypefn |
48 | |
49 ## Author: Kai Habel <kai.habel@gmx.de> | |
50 | |
51 function retval = divergence (varargin) | |
52 | |
53 fidx = 1; | |
54 if (nargin == 2) | |
55 sz = size (varargin{fidx}); | |
56 dx = (1:sz(2))(:); | |
57 dy = (1:sz(1))(:); | |
58 elseif (nargin == 3) | |
59 sz = size (varargin{fidx}); | |
60 dx = 1:sz(2); | |
61 dy = 1:sz(1); | |
62 dz = 1:sz(3); | |
63 elseif (nargin == 4) | |
64 fidx = 3; | |
65 dx = varargin{1}(1,:); | |
66 dy = varargin{2}(:,1); | |
67 elseif (nargin == 6) | |
68 fidx = 4; | |
69 dx = varargin{1}(1,:,1)(:); | |
70 dy = varargin{2}(:,1,1)(:); | |
71 dz = varargin{3}(1,1,:)(:); | |
72 else | |
14846
460a3c6d8bf1
maint: Use Octave coding convention for cuddled parenthis in function calls with empty argument lists.
Rik <octave@nomad.inbox5.com>
parents:
14363
diff
changeset
|
73 print_usage (); |
11428 | 74 endif |
11587
c792872f8942
all script files: untabify and strip trailing whitespace
John W. Eaton <jwe@octave.org>
parents:
11575
diff
changeset
|
75 |
19833
9fc020886ae9
maint: Clean up m-files to follow Octave coding conventions.
Rik <rik@octave.org>
parents:
19697
diff
changeset
|
76 if (nargin == 4 || nargin == 2) |
9fc020886ae9
maint: Clean up m-files to follow Octave coding conventions.
Rik <rik@octave.org>
parents:
19697
diff
changeset
|
77 if (! size_equal (varargin{fidx},varargin{fidx + 1})) |
11588
d5bd2766c640
style fixes for warning and error messages in script files
John W. Eaton <jwe@octave.org>
parents:
11587
diff
changeset
|
78 error ("divergence: size of X and Y must match"); |
11428 | 79 elseif (ndims (varargin{fidx}) != 2) |
20711
7b608fadc663
Make error messages more specific about the variable and problem encountered.
Rik <rik@octave.org>
parents:
20158
diff
changeset
|
80 error ("divergence: X and Y must be 2-D matrices"); |
19833
9fc020886ae9
maint: Clean up m-files to follow Octave coding conventions.
Rik <rik@octave.org>
parents:
19697
diff
changeset
|
81 elseif (length (dx) != columns (varargin{fidx}) |
9fc020886ae9
maint: Clean up m-files to follow Octave coding conventions.
Rik <rik@octave.org>
parents:
19697
diff
changeset
|
82 || length (dy) != rows (varargin{fidx})) |
11472
1740012184f9
Use uppercase for variable names in error() strings to match Info documentation. Only m-files done.
Rik <octave@nomad.inbox5.com>
parents:
11428
diff
changeset
|
83 error ("divergence: size of dx and dy must match the respective dimension of X and Y"); |
11428 | 84 endif |
85 | |
14868
5d3a684236b0
maint: Use Octave coding conventions for cuddling parentheses in scripts directory
Rik <octave@nomad.inbox5.com>
parents:
14846
diff
changeset
|
86 retval = gradient (varargin{fidx}, dx, dy); |
5d3a684236b0
maint: Use Octave coding conventions for cuddling parentheses in scripts directory
Rik <octave@nomad.inbox5.com>
parents:
14846
diff
changeset
|
87 retval += gradient (varargin{fidx + 1}.', dy, dx).'; |
11587
c792872f8942
all script files: untabify and strip trailing whitespace
John W. Eaton <jwe@octave.org>
parents:
11575
diff
changeset
|
88 |
19833
9fc020886ae9
maint: Clean up m-files to follow Octave coding conventions.
Rik <rik@octave.org>
parents:
19697
diff
changeset
|
89 elseif (nargin == 6 || nargin == 3) |
9fc020886ae9
maint: Clean up m-files to follow Octave coding conventions.
Rik <rik@octave.org>
parents:
19697
diff
changeset
|
90 if (! size_equal (varargin{fidx},varargin{fidx + 1},varargin{fidx + 2})) |
11589
b0084095098e
missing semicolons in script files
John W. Eaton <jwe@octave.org>
parents:
11588
diff
changeset
|
91 error ("divergence: size of X, Y, and Z must match"); |
11428 | 92 elseif (ndims (varargin{fidx}) != 3) |
20711
7b608fadc663
Make error messages more specific about the variable and problem encountered.
Rik <rik@octave.org>
parents:
20158
diff
changeset
|
93 error ("divergence: X, Y, and Z must be 2-D matrices"); |
11428 | 94 elseif ((length (dx) != size (varargin{fidx}, 2)) |
95 || (length (dy) != size (varargin{fidx}, 1)) | |
96 || (length (dz) != size (varargin{fidx}, 3))) | |
11588
d5bd2766c640
style fixes for warning and error messages in script files
John W. Eaton <jwe@octave.org>
parents:
11587
diff
changeset
|
97 error ("divergence: size of dx, dy, and dz must match the respective dimesion of X, Y, and Z"); |
11428 | 98 endif |
99 | |
100 ## x-direction | |
101 retval = gradient (varargin{fidx}, dx, dy, dz); | |
102 ## y-direction | |
103 retval += shiftdim (gradient (shiftdim (varargin{fidx + 1}, 2), dy), 1); | |
104 ## z-direction | |
105 retval += shiftdim (gradient (shiftdim (varargin{fidx + 2}, 1), dz), 2); | |
106 endif | |
107 | |
108 endfunction | |
109 | |
14363
f3d52523cde1
Use Octave coding conventions in all m-file %!test blocks
Rik <octave@nomad.inbox5.com>
parents:
14138
diff
changeset
|
110 |
11428 | 111 %!test |
14363
f3d52523cde1
Use Octave coding conventions in all m-file %!test blocks
Rik <octave@nomad.inbox5.com>
parents:
14138
diff
changeset
|
112 %! [X,Y] = meshgrid (-20:20,-22:22); |
f3d52523cde1
Use Octave coding conventions in all m-file %!test blocks
Rik <octave@nomad.inbox5.com>
parents:
14138
diff
changeset
|
113 %! div = divergence (X-Y,Y); |
f3d52523cde1
Use Octave coding conventions in all m-file %!test blocks
Rik <octave@nomad.inbox5.com>
parents:
14138
diff
changeset
|
114 %! assert (all (div(:) == 2)); |
f3d52523cde1
Use Octave coding conventions in all m-file %!test blocks
Rik <octave@nomad.inbox5.com>
parents:
14138
diff
changeset
|
115 %! assert (size_equal (X,Y,div)); |