|
3980
|
1 ## Copyright (C) 1999, 2000 Paul Kienzle |
|
|
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 2, or (at your option) |
|
|
8 ## 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, write to the Free |
|
|
17 ## Software Foundation, 59 Temple Place - Suite 330, Boston, MA |
|
|
18 ## 02111-1307, USA. |
|
|
19 |
|
|
20 ## -*- texinfo -*- |
|
|
21 ## @deftypefn {Mapping Function} {} mod (@var{x}, @var{y}) |
|
|
22 ## Compute modulo function, using |
|
|
23 ## |
|
|
24 ## @example |
|
|
25 ## x - y .* floor (x ./ y) |
|
|
26 ## @end example |
|
|
27 ## |
|
|
28 ## Note that this handles negative numbers correctly: |
|
|
29 ## @code{mod (-1, 3)} is 2, not -1 as @code{rem (-1, 3)} returns. |
|
|
30 ## Also, @code{mod (@var{x}, 0)} returns @var{x}. |
|
|
31 ## |
|
|
32 ## An error message is printed if the dimensions of the arguments do not |
|
|
33 ## agree, or if either of the arguments is complex. |
|
|
34 ## @end deftypefn |
|
|
35 ## @seealso{rem, round} |
|
|
36 |
|
|
37 ## Author: Paul Kienzle <pkienzle@kienzle.powernet.co.uk> |
|
|
38 ## Modified by: Teemu Ikonen <tpikonen@pcu.helsinki.fi> |
|
|
39 ## Adapted by: jwe |
|
|
40 |
|
|
41 function r = mod (x, y) |
|
|
42 |
|
|
43 if (nargin != 2) |
|
|
44 usage ("r = mod (x, y)"); |
|
|
45 endif |
|
|
46 |
|
|
47 if (any (size (x) != size (y)) && ! (is_scalar (x) || is_scalar (y))) |
|
|
48 error ("mod: argument sizes must agree"); |
|
|
49 endif |
|
|
50 |
|
|
51 ## Matlab allows complex arguments, but as far as I can tell, that's a |
|
|
52 ## bunch of hooey. |
|
|
53 |
|
|
54 if (any (any (imag (x))) || any (any (imag (y)))) |
|
|
55 error ("mod: complex arguments are not allowed"); |
|
|
56 endif |
|
|
57 |
|
|
58 nz = y != 0.0; |
|
|
59 if (all (nz(:))) |
|
|
60 ## No elements of y are zero. |
|
|
61 r = x - y .* floor (x ./ y); |
|
|
62 elseif (is_scalar (y)) |
|
|
63 ## y must be zero. |
|
|
64 r = x; |
|
|
65 else |
|
|
66 ## Some elements of y are zero. |
|
|
67 if (is_scalar (x)) |
|
|
68 r = x * ones (size (y)); |
|
|
69 else |
|
|
70 r = x; |
|
|
71 x = x(nz); |
|
|
72 endif |
|
|
73 r(nz) = x - y .* floor (x ./ y); |
|
|
74 endif |
|
|
75 |
|
|
76 endfunction |
|
|
77 |
|
|
78 ## empty input test |
|
|
79 %!assert (isempty(mod([], []))); |
|
|
80 |
|
|
81 ## x mod y, y != 0 tests |
|
|
82 %!assert (mod(5, 3), 2); |
|
|
83 %!assert (mod(-5, 3), 1); |
|
|
84 %!assert (mod(0, 3), 0); |
|
|
85 %!assert (mod([-5, 5, 0], [3, 3, 3]), [1, 2, 0]); |
|
|
86 %!assert (mod([-5; 5; 0], [3; 3; 3]), [1; 2; 0]); |
|
|
87 %!assert (mod([-5, 5; 0, 3], [3, 3 ; 3, 1]), [1, 2 ; 0, 0]); |
|
|
88 |
|
|
89 ## x mod 0 tests |
|
|
90 %!assert (mod(5, 0), 5); |
|
|
91 %!assert (mod(-5, 0), -5); |
|
|
92 %!assert (mod([-5, 5, 0], [3, 0, 3]), [1, 5, 0]); |
|
|
93 %!assert (mod([-5; 5; 0], [3; 0; 3]), [1; 5; 0]); |
|
|
94 %!assert (mod([-5, 5; 0, 3], [3, 0 ; 3, 1]), [1, 5 ; 0, 0]); |
|
|
95 %!assert (mod([-5, 5; 0, 3], [0, 0 ; 0, 0]), [-5, 5; 0, 3]); |
|
|
96 |
|
|
97 ## mixed scalar/matrix tests |
|
|
98 %!assert (mod([-5, 5; 0, 3], 0), [-5, 5; 0, 3]); |
|
|
99 %!assert (mod([-5, 5; 0, 3], 3), [1, 2; 0, 0]); |
|
|
100 %!assert (mod(-5,[0,0; 0,0]), [-5, -5; -5, -5]); |
|
|
101 %!assert (mod(-5,[3,0; 3,1]), [1, -5; 1, 0]); |
|
|
102 %!assert (mod(-5,[3,2; 3,1]), [1, 1; 1, 0]); |
|
|
103 |
