Mercurial > hg > octave-nkf
annotate scripts/general/shift.m @ 20828:a3b9ee5c040a
Replace bsxfun with broadcasting for performance with complex inputs (bug #38628).
cumtrapz.m, quadgk.m, trapz.m, center.m, zscore.m: Replace bsxfun with
broadcasting for performance where inputs might be complex.
author | Rik <rik@octave.org> |
---|---|
date | Mon, 12 Oct 2015 21:28:32 -0700 |
parents | 7503499a252b |
children |
rev | line source |
---|---|
19898
4197fc428c7d
maint: Update copyright notices for 2015.
John W. Eaton <jwe@octave.org>
parents:
17744
diff
changeset
|
1 ## Copyright (C) 1995-2015 Kurt Hornik |
3426 | 2 ## |
3922 | 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 | |
7016 | 7 ## the Free Software Foundation; either version 3 of the License, or (at |
8 ## your option) any later version. | |
3426 | 9 ## |
3922 | 10 ## Octave is distributed in the hope that it will be useful, but |
2539 | 11 ## WITHOUT ANY WARRANTY; without even the implied warranty of |
12 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
3426 | 13 ## General Public License for more details. |
14 ## | |
2539 | 15 ## You should have received a copy of the GNU General Public License |
7016 | 16 ## along with Octave; see the file COPYING. If not, see |
17 ## <http://www.gnu.org/licenses/>. | |
2539 | 18 |
3369 | 19 ## -*- texinfo -*- |
10690
35adf2a71f3f
Use common code block to find first non-singleton dimension.
Rik <octave@nomad.inbox5.com>
parents:
10549
diff
changeset
|
20 ## @deftypefn {Function File} {} shift (@var{x}, @var{b}) |
4862 | 21 ## @deftypefnx {Function File} {} shift (@var{x}, @var{b}, @var{dim}) |
3369 | 22 ## If @var{x} is a vector, perform a circular shift of length @var{b} of |
23 ## the elements of @var{x}. | |
3426 | 24 ## |
3369 | 25 ## If @var{x} is a matrix, do the same for each column of @var{x}. |
20368
7503499a252b
doc: Update docstrings to have one sentence summary as first line.
Rik <rik@octave.org>
parents:
20038
diff
changeset
|
26 ## |
7503499a252b
doc: Update docstrings to have one sentence summary as first line.
Rik <rik@octave.org>
parents:
20038
diff
changeset
|
27 ## If the optional @var{dim} argument is given, operate along this dimension. |
3369 | 28 ## @end deftypefn |
2539 | 29 |
30 ## Author: AW <Andreas.Weingessel@ci.tuwien.ac.at> | |
31 ## Created: 14 September 1994 | |
32 ## Adapted-By: jwe | |
33 | |
4862 | 34 function y = shift (x, b, dim) |
2539 | 35 |
4862 | 36 if (nargin != 2 && nargin != 3) |
6046 | 37 print_usage (); |
2539 | 38 endif |
39 | |
13278
04edb15d7966
shift.m: Clean up code and add more tests
Rik <octave@nomad.inbox5.com>
parents:
12676
diff
changeset
|
40 if (numel (x) < 1) |
04edb15d7966
shift.m: Clean up code and add more tests
Rik <octave@nomad.inbox5.com>
parents:
12676
diff
changeset
|
41 error ("shift: X must not be empty"); |
04edb15d7966
shift.m: Clean up code and add more tests
Rik <octave@nomad.inbox5.com>
parents:
12676
diff
changeset
|
42 elseif (! (isscalar (b) && b == fix (b))) |
10690
35adf2a71f3f
Use common code block to find first non-singleton dimension.
Rik <octave@nomad.inbox5.com>
parents:
10549
diff
changeset
|
43 error ("shift: B must be an integer"); |
2539 | 44 endif |
45 | |
4862 | 46 nd = ndims (x); |
47 sz = size (x); | |
48 | |
49 if (nargin == 3) | |
13278
04edb15d7966
shift.m: Clean up code and add more tests
Rik <octave@nomad.inbox5.com>
parents:
12676
diff
changeset
|
50 if (!(isscalar (dim) && dim == fix (dim)) |
11149
fe3c3dfc07eb
style fix: break lines before && and ||, not after
John W. Eaton <jwe@octave.org>
parents:
10690
diff
changeset
|
51 || !(1 <= dim && dim <= nd)) |
10690
35adf2a71f3f
Use common code block to find first non-singleton dimension.
Rik <octave@nomad.inbox5.com>
parents:
10549
diff
changeset
|
52 error ("shift: DIM must be an integer and a valid dimension"); |
4862 | 53 endif |
54 else | |
10690
35adf2a71f3f
Use common code block to find first non-singleton dimension.
Rik <octave@nomad.inbox5.com>
parents:
10549
diff
changeset
|
55 ## Find the first non-singleton dimension. |
12674
9493880928c8
Use common idiom in m-files for finding first non-singleton dimension.
Rik <octave@nomad.inbox5.com>
parents:
11587
diff
changeset
|
56 (dim = find (sz > 1, 1)) || (dim = 1); |
4862 | 57 endif |
58 | |
13278
04edb15d7966
shift.m: Clean up code and add more tests
Rik <octave@nomad.inbox5.com>
parents:
12676
diff
changeset
|
59 d = sz(dim); |
4862 | 60 |
12676
2783fa95cab7
Use common code idiom for creating cell array for indexing ND-arrays
Rik <octave@nomad.inbox5.com>
parents:
12674
diff
changeset
|
61 idx = repmat ({':'}, nd, 1); |
13278
04edb15d7966
shift.m: Clean up code and add more tests
Rik <octave@nomad.inbox5.com>
parents:
12676
diff
changeset
|
62 if (b > 0) |
5568 | 63 b = rem (b, d); |
7208 | 64 idx{dim} = [d-b+1:d, 1:d-b]; |
5568 | 65 elseif (b < 0) |
66 b = rem (abs (b), d); | |
7208 | 67 idx{dim} = [b+1:d, 1:b]; |
5568 | 68 endif |
13278
04edb15d7966
shift.m: Clean up code and add more tests
Rik <octave@nomad.inbox5.com>
parents:
12676
diff
changeset
|
69 |
7208 | 70 y = x(idx{:}); |
4369 | 71 |
7411 | 72 endfunction |
2539 | 73 |
13278
04edb15d7966
shift.m: Clean up code and add more tests
Rik <octave@nomad.inbox5.com>
parents:
12676
diff
changeset
|
74 |
7411 | 75 %!test |
76 %! a = [1, 2, 3]; | |
77 %! b = [4, 5, 6]; | |
78 %! c = [7, 8, 9]; | |
11587
c792872f8942
all script files: untabify and strip trailing whitespace
John W. Eaton <jwe@octave.org>
parents:
11523
diff
changeset
|
79 %! |
7411 | 80 %! r = [a, b, c]; |
81 %! m = [a; b; c]; | |
11587
c792872f8942
all script files: untabify and strip trailing whitespace
John W. Eaton <jwe@octave.org>
parents:
11523
diff
changeset
|
82 %! |
14363
f3d52523cde1
Use Octave coding conventions in all m-file %!test blocks
Rik <octave@nomad.inbox5.com>
parents:
14138
diff
changeset
|
83 %! assert (shift (r, 0), r); |
f3d52523cde1
Use Octave coding conventions in all m-file %!test blocks
Rik <octave@nomad.inbox5.com>
parents:
14138
diff
changeset
|
84 %! assert (shift (r, 3), [c, a, b]); |
f3d52523cde1
Use Octave coding conventions in all m-file %!test blocks
Rik <octave@nomad.inbox5.com>
parents:
14138
diff
changeset
|
85 %! assert (shift (r, -6), [c, a, b]); |
f3d52523cde1
Use Octave coding conventions in all m-file %!test blocks
Rik <octave@nomad.inbox5.com>
parents:
14138
diff
changeset
|
86 %! assert (shift (r, -3), [b, c, a]); |
f3d52523cde1
Use Octave coding conventions in all m-file %!test blocks
Rik <octave@nomad.inbox5.com>
parents:
14138
diff
changeset
|
87 %! assert (shift (m, 1), [c; a; b]); |
f3d52523cde1
Use Octave coding conventions in all m-file %!test blocks
Rik <octave@nomad.inbox5.com>
parents:
14138
diff
changeset
|
88 %! assert (shift (m, -2), [c; a; b]); |
7411 | 89 |
20038
9fc020886ae9
maint: Clean up m-files to follow Octave coding conventions.
Rik <rik@octave.org>
parents:
19898
diff
changeset
|
90 ## Test input validation |
13278
04edb15d7966
shift.m: Clean up code and add more tests
Rik <octave@nomad.inbox5.com>
parents:
12676
diff
changeset
|
91 %!error shift () |
04edb15d7966
shift.m: Clean up code and add more tests
Rik <octave@nomad.inbox5.com>
parents:
12676
diff
changeset
|
92 %!error shift (1, 2, 3, 4) |
04edb15d7966
shift.m: Clean up code and add more tests
Rik <octave@nomad.inbox5.com>
parents:
12676
diff
changeset
|
93 %!error shift ([], 1) |
14363
f3d52523cde1
Use Octave coding conventions in all m-file %!test blocks
Rik <octave@nomad.inbox5.com>
parents:
14138
diff
changeset
|
94 %!error shift (ones (2), ones (2)) |
f3d52523cde1
Use Octave coding conventions in all m-file %!test blocks
Rik <octave@nomad.inbox5.com>
parents:
14138
diff
changeset
|
95 %!error shift (ones (2), 1.5) |
13296
27da11c63d98
shift.m: Add validation tests for dimension argument
Rik <octave@nomad.inbox5.com>
parents:
13292
diff
changeset
|
96 %!error shift (1, 1, 1.5) |
27da11c63d98
shift.m: Add validation tests for dimension argument
Rik <octave@nomad.inbox5.com>
parents:
13292
diff
changeset
|
97 %!error shift (1, 1, 0) |
27da11c63d98
shift.m: Add validation tests for dimension argument
Rik <octave@nomad.inbox5.com>
parents:
13292
diff
changeset
|
98 %!error shift (1, 1, 3) |
7411 | 99 |