Mercurial > hg > octave-nkf

annotate scripts/general/mod.m @ 9665:1dba57e9d08d

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
use blas_trans_type for xgemm
author Jaroslav Hajek <highegg@gmail.com>
date Sat, 26 Sep 2009 10:41:07 +0200
parents 16f53d29049f
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
9245
16f53d29049f update copyright notices
John W. Eaton <jwe@octave.org>
parents: 9141
diff changeset
1 ## Copyright (C) 1999, 2000, 2002, 2004, 2005, 2006, 2007, 2008,
16f53d29049f update copyright notices
John W. Eaton <jwe@octave.org>
parents: 9141
diff changeset
2 ## 2009 Paul Kienzle
3980
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
3 ##
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
4 ## This file is part of Octave.
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
5 ##
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
6 ## Octave is free software; you can redistribute it and/or modify it
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
7 ## 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: 6157
diff changeset
8 ## the Free Software Foundation; either version 3 of the License, or (at
93c65f2a5668 [project @ 2007-10-12 06:40:56 by jwe]
jwe
parents: 6157
diff changeset
9 ## your option) any later version.
3980
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
10 ##
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
11 ## Octave is distributed in the hope that it will be useful, but
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
12 ## WITHOUT ANY WARRANTY; without even the implied warranty of
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
13 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
14 ## General Public License for more details.
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
15 ##
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
16 ## 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: 6157
diff changeset
17 ## along with Octave; see the file COPYING. If not, see
93c65f2a5668 [project @ 2007-10-12 06:40:56 by jwe]
jwe
parents: 6157
diff changeset
18 ## <http://www.gnu.org/licenses/>.
3980
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
19
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
20 ## -*- texinfo -*-
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
21 ## @deftypefn {Mapping Function} {} mod (@var{x}, @var{y})
9141
c1fff751b5a8 Update section 17.1 (Utility Functions) of arith.txi
Rik <rdrider0-list@yahoo.com>
parents: 9051
diff changeset
22 ## Compute the modulo of @var{x} and @var{y}. Conceptually this is given by
3980
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
23 ##
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
24 ## @example
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
25 ## x - y .* floor (x ./ y)
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
26 ## @end example
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
27 ##
9141
c1fff751b5a8 Update section 17.1 (Utility Functions) of arith.txi
Rik <rdrider0-list@yahoo.com>
parents: 9051
diff changeset
28 ## and is written such that the correct modulus is returned for
c1fff751b5a8 Update section 17.1 (Utility Functions) of arith.txi
Rik <rdrider0-list@yahoo.com>
parents: 9051
diff changeset
29 ## integer types. This function handles negative values correctly. That
c1fff751b5a8 Update section 17.1 (Utility Functions) of arith.txi
Rik <rdrider0-list@yahoo.com>
parents: 9051
diff changeset
30 ## is, @code{mod (-1, 3)} is 2, not -1, as @code{rem (-1, 3)} returns.
c1fff751b5a8 Update section 17.1 (Utility Functions) of arith.txi
Rik <rdrider0-list@yahoo.com>
parents: 9051
diff changeset
31 ## @code{mod (@var{x}, 0)} returns @var{x}.
3980
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
32 ##
9141
c1fff751b5a8 Update section 17.1 (Utility Functions) of arith.txi
Rik <rdrider0-list@yahoo.com>
parents: 9051
diff changeset
33 ## An error results if the dimensions of the arguments do not agree, or if
c1fff751b5a8 Update section 17.1 (Utility Functions) of arith.txi
Rik <rdrider0-list@yahoo.com>
parents: 9051
diff changeset
34 ## either of the arguments is complex.
c1fff751b5a8 Update section 17.1 (Utility Functions) of arith.txi
Rik <rdrider0-list@yahoo.com>
parents: 9051
diff changeset
35 ## @seealso{rem, fmod}
3980
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
36 ## @end deftypefn
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
37
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
38 ## Author: Paul Kienzle <pkienzle@kienzle.powernet.co.uk>
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
39 ## Modified by: Teemu Ikonen <tpikonen@pcu.helsinki.fi>
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
40 ## Adapted by: jwe
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
41
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
42 function r = mod (x, y)
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
43
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
44 if (nargin != 2)
6046
34f96dd5441b [project @ 2006-10-10 16:10:25 by jwe]
jwe
parents: 5642
diff changeset
45 print_usage ();
3980
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
46 endif
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
47
6157
045038e0108a [project @ 2006-11-13 22:22:53 by jwe]
jwe
parents: 6046
diff changeset
48 if (! size_equal (x, y) && ! (isscalar (x) || isscalar (y)))
3980
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
49 error ("mod: argument sizes must agree");
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
50 endif
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
51
4027
af3a0264eefc [project @ 2002-08-09 07:01:03 by jwe]
jwe
parents: 3980
diff changeset
52 if (isreal (x) && isreal (y))
af3a0264eefc [project @ 2002-08-09 07:01:03 by jwe]
jwe
parents: 3980
diff changeset
53 nz = y != 0.0;
af3a0264eefc [project @ 2002-08-09 07:01:03 by jwe]
jwe
parents: 3980
diff changeset
54 if (all (nz(:)))
af3a0264eefc [project @ 2002-08-09 07:01:03 by jwe]
jwe
parents: 3980
diff changeset
55 ## No elements of y are zero.
7494
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
56 if (isinteger(x) || isinteger(y))
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
57 if (isinteger (x))
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
58 typ = class (x);
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
59 else
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
60 typ = class (y);
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
61 endif
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
62 r = x - y .* cast (floor (double(x) ./ double(y)), typ);
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
63 else
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
64 r = x - y .* floor (x ./ y);
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
65 endif
4030
22bd65326ec1 [project @ 2002-08-09 18:58:13 by jwe]
jwe
parents: 4027
diff changeset
66 elseif (isscalar (y))
4027
af3a0264eefc [project @ 2002-08-09 07:01:03 by jwe]
jwe
parents: 3980
diff changeset
67 ## y must be zero.
af3a0264eefc [project @ 2002-08-09 07:01:03 by jwe]
jwe
parents: 3980
diff changeset
68 r = x;
af3a0264eefc [project @ 2002-08-09 07:01:03 by jwe]
jwe
parents: 3980
diff changeset
69 else
af3a0264eefc [project @ 2002-08-09 07:01:03 by jwe]
jwe
parents: 3980
diff changeset
70 ## Some elements of y are zero.
4030
22bd65326ec1 [project @ 2002-08-09 18:58:13 by jwe]
jwe
parents: 4027
diff changeset
71 if (isscalar (x))
7494
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
72 r = x * ones (size(y), class(y));
4027
af3a0264eefc [project @ 2002-08-09 07:01:03 by jwe]
jwe
parents: 3980
diff changeset
73 else
af3a0264eefc [project @ 2002-08-09 07:01:03 by jwe]
jwe
parents: 3980
diff changeset
74 r = x;
af3a0264eefc [project @ 2002-08-09 07:01:03 by jwe]
jwe
parents: 3980
diff changeset
75 x = x(nz);
af3a0264eefc [project @ 2002-08-09 07:01:03 by jwe]
jwe
parents: 3980
diff changeset
76 endif
af3a0264eefc [project @ 2002-08-09 07:01:03 by jwe]
jwe
parents: 3980
diff changeset
77 y = y(nz);
7494
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
78 if (isinteger(x) || isinteger(y))
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
79 if (isinteger (x))
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
80 typ = class (x);
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
81 else
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
82 typ = class (y);
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
83 endif
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
84 r(nz) = x - y .* floor (double(x) ./ double(y));
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
85 else
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
86 r(nz) = x - y .* floor (x ./ y);
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
87 endif
4027
af3a0264eefc [project @ 2002-08-09 07:01:03 by jwe]
jwe
parents: 3980
diff changeset
88 endif
af3a0264eefc [project @ 2002-08-09 07:01:03 by jwe]
jwe
parents: 3980
diff changeset
89 else
3980
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
90 error ("mod: complex arguments are not allowed");
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
91 endif
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
92
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
93 endfunction
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
94
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
95 ## empty input test
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
96 %!assert (isempty(mod([], [])));
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
97
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
98 ## x mod y, y != 0 tests
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
99 %!assert (mod(5, 3), 2);
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
100 %!assert (mod(-5, 3), 1);
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
101 %!assert (mod(0, 3), 0);
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
102 %!assert (mod([-5, 5, 0], [3, 3, 3]), [1, 2, 0]);
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
103 %!assert (mod([-5; 5; 0], [3; 3; 3]), [1; 2; 0]);
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
104 %!assert (mod([-5, 5; 0, 3], [3, 3 ; 3, 1]), [1, 2 ; 0, 0]);
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
105
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
106 ## x mod 0 tests
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
107 %!assert (mod(5, 0), 5);
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
108 %!assert (mod(-5, 0), -5);
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
109 %!assert (mod([-5, 5, 0], [3, 0, 3]), [1, 5, 0]);
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
110 %!assert (mod([-5; 5; 0], [3; 0; 3]), [1; 5; 0]);
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
111 %!assert (mod([-5, 5; 0, 3], [3, 0 ; 3, 1]), [1, 5 ; 0, 0]);
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
112 %!assert (mod([-5, 5; 0, 3], [0, 0 ; 0, 0]), [-5, 5; 0, 3]);
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
113
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
114 ## mixed scalar/matrix tests
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
115 %!assert (mod([-5, 5; 0, 3], 0), [-5, 5; 0, 3]);
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
116 %!assert (mod([-5, 5; 0, 3], 3), [1, 2; 0, 0]);
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
117 %!assert (mod(-5,[0,0; 0,0]), [-5, -5; -5, -5]);
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
118 %!assert (mod(-5,[3,0; 3,1]), [1, -5; 1, 0]);
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
119 %!assert (mod(-5,[3,2; 3,1]), [1, 1; 1, 0]);
02a959755258 [project @ 2002-07-10 18:48:58 by jwe]
jwe
parents:
diff changeset
120
7494
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
121 ## integer types
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
122 %!assert (mod(uint8(5),uint8(4)),uint8(1))
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
123 %!assert (mod(uint8([1:5]),uint8(4)),uint8([1,2,3,0,1]))
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
124 %!assert (mod(uint8([1:5]),uint8(0)),uint8([1:5]))
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
125 %!error (mod(uint8(5),int8(4)))
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
126
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
127 ## mixed integer/real types
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
128 %!assert (mod(uint8(5),4),uint8(1))
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
129 %!assert (mod(5,uint8(4)),uint8(1))
bd2bd04e68ca Treat integer types for mod/rem correctly
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
130 %!assert (mod(uint8([1:5]),4),uint8([1,2,3,0,1]))