Mercurial > hg > octave-nkf
annotate scripts/general/repmat.m @ 11032:c9b0a75b02e8
Make all regexp in Octave compatible with both POSIX and PCRE.
| author | Rik <octave@nomad.inbox5.com> |
|---|---|
| date | Tue, 28 Sep 2010 09:25:14 -0700 |
| parents | a40e32927b3a |
| children | d7ea780b036f |
| rev | line source |
|---|---|
| 8920 | 1 ## Copyright (C) 2000, 2002, 2004, 2005, 2006, 2007, 2009 Paul Kienzle |
|
8390
49901b624316
optimize repmat for scalar & matrix case
Jaroslav Hajek <highegg@gmail.com>
parents:
7017
diff
changeset
|
2 ## Copyright (C) 2008 Jaroslav Hajek |
| 3914 | 3 ## |
| 4 ## This file is part of Octave. | |
| 5 ## | |
| 6 ## Octave is free software; you can redistribute it and/or modify it | |
| 7 ## under the terms of the GNU General Public License as published by | |
| 7016 | 8 ## the Free Software Foundation; either version 3 of the License, or (at |
| 9 ## your option) any later version. | |
| 3914 | 10 ## |
| 11 ## Octave is distributed in the hope that it will be useful, but | |
| 12 ## WITHOUT ANY WARRANTY; without even the implied warranty of | |
| 13 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
| 14 ## General Public License for more details. | |
| 15 ## | |
| 16 ## You should have received a copy of the GNU General Public License | |
| 7016 | 17 ## along with Octave; see the file COPYING. If not, see |
| 18 ## <http://www.gnu.org/licenses/>. | |
| 3914 | 19 |
| 20 ## -*- texinfo -*- | |
|
10793
be55736a0783
Grammarcheck the documentation from m-files.
Rik <octave@nomad.inbox5.com>
parents:
9388
diff
changeset
|
21 ## @deftypefn {Function File} {} repmat (@var{A}, @var{m}, @var{n}) |
| 3914 | 22 ## @deftypefnx {Function File} {} repmat (@var{A}, [@var{m} @var{n}]) |
|
9041
853f96e8008f
Cleanup documentation file matrix.texi
Rik <rdrider0-list@yahoo.com>
parents:
8970
diff
changeset
|
23 ## @deftypefnx {Function File} {} repmat (@var{A}, [@var{m} @var{n} @var{p} @dots{}]) |
| 3914 | 24 ## Form a block matrix of size @var{m} by @var{n}, with a copy of matrix |
| 25 ## @var{A} as each element. If @var{n} is not specified, form an | |
| 26 ## @var{m} by @var{m} block matrix. | |
|
10801
a40e32927b3a
Improve documentation for new repelems function.
Rik <octave@nomad.inbox5.com>
parents:
10793
diff
changeset
|
27 ## @seealso{repelems} |
| 3914 | 28 ## @end deftypefn |
| 29 | |
| 30 ## Author: Paul Kienzle <pkienzle@kienzle.powernet.co.uk> | |
| 31 ## Created: July 2000 | |
| 32 | |
| 3915 | 33 function x = repmat (a, m, n) |
| 34 | |
| 3914 | 35 if (nargin < 2 || nargin > 3) |
| 6046 | 36 print_usage (); |
| 3914 | 37 endif |
| 38 | |
| 4844 | 39 if (nargin == 3) |
| 5195 | 40 if (! (isscalar (m) && isscalar (n))) |
| 4844 | 41 error ("repmat: with 3 arguments m and n must be scalar"); |
| 42 endif | |
| 43 idx = [m, n]; | |
| 44 else | |
| 4030 | 45 if (isscalar (m)) |
| 4844 | 46 idx = [m, m]; |
| 3914 | 47 n = m; |
| 4844 | 48 elseif (isvector (m) && length (m) > 1) |
|
8390
49901b624316
optimize repmat for scalar & matrix case
Jaroslav Hajek <highegg@gmail.com>
parents:
7017
diff
changeset
|
49 ## Ensure that we have a row vector |
| 4844 | 50 idx = m(:).'; |
| 3914 | 51 else |
| 4844 | 52 error ("repmat: invalid dimensional argument"); |
| 3914 | 53 endif |
| 54 endif | |
|
8508
dee629f14bfa
repmat.m: handle negative dimensions properly
John W. Eaton <jwe@octave.org>
parents:
8507
diff
changeset
|
55 |
|
dee629f14bfa
repmat.m: handle negative dimensions properly
John W. Eaton <jwe@octave.org>
parents:
8507
diff
changeset
|
56 if (all (idx < 0)) |
|
dee629f14bfa
repmat.m: handle negative dimensions properly
John W. Eaton <jwe@octave.org>
parents:
8507
diff
changeset
|
57 error ("repmat: invalid dimensions"); |
|
dee629f14bfa
repmat.m: handle negative dimensions properly
John W. Eaton <jwe@octave.org>
parents:
8507
diff
changeset
|
58 else |
|
dee629f14bfa
repmat.m: handle negative dimensions properly
John W. Eaton <jwe@octave.org>
parents:
8507
diff
changeset
|
59 idx = max (idx, 0); |
|
dee629f14bfa
repmat.m: handle negative dimensions properly
John W. Eaton <jwe@octave.org>
parents:
8507
diff
changeset
|
60 endif |
| 3914 | 61 |
| 4844 | 62 if (numel (a) == 1) |
|
8390
49901b624316
optimize repmat for scalar & matrix case
Jaroslav Hajek <highegg@gmail.com>
parents:
7017
diff
changeset
|
63 ## optimize the scalar fill case. |
|
49901b624316
optimize repmat for scalar & matrix case
Jaroslav Hajek <highegg@gmail.com>
parents:
7017
diff
changeset
|
64 x(1:prod (idx)) = a; |
|
49901b624316
optimize repmat for scalar & matrix case
Jaroslav Hajek <highegg@gmail.com>
parents:
7017
diff
changeset
|
65 x = reshape (x, idx); |
|
49901b624316
optimize repmat for scalar & matrix case
Jaroslav Hajek <highegg@gmail.com>
parents:
7017
diff
changeset
|
66 elseif (ndims (a) == 2 && length (idx) < 3) |
|
49901b624316
optimize repmat for scalar & matrix case
Jaroslav Hajek <highegg@gmail.com>
parents:
7017
diff
changeset
|
67 if (issparse (a)) |
|
9388
5556563c6551
repmat.m: call cron, not spkron
Marco Caliari <marco.caliari@univr.it>
parents:
9041
diff
changeset
|
68 x = kron (ones (idx), a); |
| 4844 | 69 else |
|
8390
49901b624316
optimize repmat for scalar & matrix case
Jaroslav Hajek <highegg@gmail.com>
parents:
7017
diff
changeset
|
70 ## indexing is now faster, so we use it rather than kron. |
|
49901b624316
optimize repmat for scalar & matrix case
Jaroslav Hajek <highegg@gmail.com>
parents:
7017
diff
changeset
|
71 m = rows (a); n = columns (a); |
|
49901b624316
optimize repmat for scalar & matrix case
Jaroslav Hajek <highegg@gmail.com>
parents:
7017
diff
changeset
|
72 p = idx(1); q = idx(2); |
|
49901b624316
optimize repmat for scalar & matrix case
Jaroslav Hajek <highegg@gmail.com>
parents:
7017
diff
changeset
|
73 x = reshape (a, m, 1, n, 1); |
|
49901b624316
optimize repmat for scalar & matrix case
Jaroslav Hajek <highegg@gmail.com>
parents:
7017
diff
changeset
|
74 x = x(:, ones (1, p), :, ones (1, q)); |
|
49901b624316
optimize repmat for scalar & matrix case
Jaroslav Hajek <highegg@gmail.com>
parents:
7017
diff
changeset
|
75 x = reshape (x, m*p, n*q); |
| 4844 | 76 endif |
| 3914 | 77 else |
| 8970 | 78 aidx = size (a); |
| 79 ## ensure matching size | |
| 80 idx(end+1:length (aidx)) = 1; | |
| 81 aidx(end+1:length (idx)) = 1; | |
| 82 ## create subscript array | |
| 83 cidx = cell (2, length (aidx)); | |
| 8507 | 84 for i = 1:length (aidx) |
| 8970 | 85 cidx{1,i} = ':'; |
| 86 cidx{2,i} = ones (1, idx (i)); | |
| 4844 | 87 endfor |
| 8970 | 88 aaidx = aidx; |
| 89 # add singleton dims | |
| 90 aaidx(2,:) = 1; | |
| 91 a = reshape (a, aaidx(:)); | |
| 92 x = reshape (a (cidx{:}), idx .* aidx); | |
| 3914 | 93 endif |
| 94 | |
| 95 endfunction | |
| 6987 | 96 |
| 97 # Test various methods of providing size parameters | |
| 98 %!shared x | |
| 99 %! x = [1 2;3 4]; | |
| 100 %!assert(repmat(x, [1 1]), repmat(x, 1)); | |
| 101 %!assert(repmat(x, [3 3]), repmat(x, 3)); | |
| 102 %!assert(repmat(x, [1 1]), repmat(x, 1, 1)); | |
| 103 %!assert(repmat(x, [1 3]), repmat(x, 1, 3)); | |
| 104 %!assert(repmat(x, [3 1]), repmat(x, 3, 1)); | |
| 105 %!assert(repmat(x, [3 3]), repmat(x, 3, 3)); | |
| 106 | |
| 107 # Tests for numel==1 case: | |
| 108 %!shared x, r | |
| 109 %! x = [ 65 ]; | |
| 110 %! r = kron(ones(2,2), x); | |
| 111 %!assert(r, repmat(x, [2 2])); | |
| 112 %!assert(char(r), repmat(char(x), [2 2])); | |
| 113 %!assert(int8(r), repmat(int8(x), [2 2])); | |
| 114 | |
| 115 # Tests for ndims==2 case: | |
| 116 %!shared x, r | |
| 117 %! x = [ 65 66 67 ]; | |
| 118 %! r = kron(ones(2,2), x); | |
| 119 %!assert(r, repmat(x, [2 2])); | |
| 120 %!assert(char(r), repmat(char(x), [2 2])); | |
| 121 %!assert(int8(r), repmat(int8(x), [2 2])); | |
| 122 | |
| 123 # Tests for dim>2 case: | |
| 124 %!shared x, r | |
| 125 %! x = [ 65 66 67 ]; | |
| 126 %! r = kron(ones(2,2), x); | |
| 127 %! r(:,:,2) = r(:,:,1); | |
| 128 %!assert(r, repmat(x, [2 2 2])); | |
| 129 %!assert(char(r), repmat(char(x), [2 2 2])); | |
| 130 %!assert(int8(r), repmat(int8(x), [2 2 2])); | |
| 131 | |
| 132 # Test that sparsity is kept | |
| 133 %!assert(sparse(4,4), repmat(sparse(2,2),[2 2])); | |
| 134 | |
|
8508
dee629f14bfa
repmat.m: handle negative dimensions properly
John W. Eaton <jwe@octave.org>
parents:
8507
diff
changeset
|
135 |
|
dee629f14bfa
repmat.m: handle negative dimensions properly
John W. Eaton <jwe@octave.org>
parents:
8507
diff
changeset
|
136 %!assert (size (repmat (".", -1, 1)), [0, 1]); |
|
dee629f14bfa
repmat.m: handle negative dimensions properly
John W. Eaton <jwe@octave.org>
parents:
8507
diff
changeset
|
137 %!assert (size (repmat (".", 1, -1)), [1, 0]); |
|
dee629f14bfa
repmat.m: handle negative dimensions properly
John W. Eaton <jwe@octave.org>
parents:
8507
diff
changeset
|
138 %!error (size (repmat (".", -1, -1))); |