Mercurial > hg > octave-lyh
annotate scripts/polynomial/conv.m @ 8158:15e4a450bf84
conv.m: Correct row/col orientation of output
author | Ben Abbott <bpabbott@mac.com> |
---|---|
date | Mon, 29 Sep 2008 09:33:06 -0400 |
parents | 83a8781b529d |
children | eb63fbe60fab |
rev | line source |
---|---|
7017 | 1 ## Copyright (C) 1994, 1995, 1996, 1997, 1998, 1999, 2000, 2002, 2004, |
2 ## 2005, 2006, 2007 John W. Eaton | |
2313 | 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. | |
2313 | 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/>. | |
1025 | 19 |
3368 | 20 ## -*- texinfo -*- |
21 ## @deftypefn {Function File} {} conv (@var{a}, @var{b}) | |
2311 | 22 ## Convolve two vectors. |
3426 | 23 ## |
3368 | 24 ## @code{y = conv (a, b)} returns a vector of length equal to |
25 ## @code{length (a) + length (b) - 1}. | |
26 ## If @var{a} and @var{b} are polynomial coefficient vectors, @code{conv} | |
27 ## returns the coefficients of the product polynomial. | |
5642 | 28 ## @seealso{deconv, poly, roots, residue, polyval, polyderiv, polyinteg} |
3368 | 29 ## @end deftypefn |
2311 | 30 |
3202 | 31 ## Author: Tony Richardson <arichard@stark.cc.oh.us> |
2312 | 32 ## Created: June 1994 |
33 ## Adapted-By: jwe | |
34 | |
787 | 35 function y = conv (a, b) |
2325 | 36 |
787 | 37 if (nargin != 2) |
6046 | 38 print_usage (); |
787 | 39 endif |
40 | |
4030 | 41 if (! (isvector (a) && isvector (b))) |
787 | 42 error("conv: both arguments must be vectors"); |
43 endif | |
44 | |
45 la = length (a); | |
46 lb = length (b); | |
47 | |
48 ly = la + lb - 1; | |
49 | |
2303 | 50 ## Use the shortest vector as the coefficent vector to filter. |
8158
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
51 ## Preserve the row/column orientation of the longer input. |
787 | 52 if (la < lb) |
53 if (ly > lb) | |
8158
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
54 if (size (b, 1) <= size (b, 2)) |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
55 x = [b, (zeros (1, ly - lb))]; |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
56 else |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
57 x = [b; (zeros (ly - lb, 1))]; |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
58 endif |
787 | 59 else |
60 x = b; | |
61 endif | |
62 y = filter (a, 1, x); | |
63 else | |
64 if(ly > la) | |
8158
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
65 if (size (a, 1) <= size (a, 2)) |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
66 x = [a, (zeros (1, ly - la))]; |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
67 else |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
68 x = [a; (zeros (ly - la, 1))]; |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
69 endif |
787 | 70 else |
71 x = a; | |
72 endif | |
73 y = filter (b, 1, x); | |
74 endif | |
75 | |
76 endfunction | |
7411 | 77 |
8158
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
78 %!assert(all (all (conv (ones (3, 1), ones (3, 1)) == [1; 2; 3; 2; 1]))); |
7411 | 79 |
80 %!assert(all (all (conv (ones (1, 3), ones (3, 1)) == [1, 2, 3, 2, 1]))); | |
81 | |
82 %!assert(all (all (conv (3, [1, 2, 3]) == [3, 6, 9]))); | |
83 | |
84 %!error conv ([1, 2; 3, 4], 3); | |
85 | |
86 %!assert(conv (2, 3),6); | |
87 | |
88 %!error conv (2, []); | |
89 | |
8158
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
90 %!test |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
91 %! a = 1:10; |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
92 %! b = 1:3; |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
93 %! c = conv (a, b); |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
94 %! assert (size(c), [1, numel(a)+numel(b)-1]) |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
95 %!test |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
96 %! a = (1:10).'; |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
97 %! b = 1:3; |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
98 %! c = conv (a, b); |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
99 %! assert (size(c), [numel(a)+numel(b)-1, 1]) |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
100 %!test |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
101 %! a = 1:10; |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
102 %! b = (1:3).'; |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
103 %! c = conv (a, b); |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
104 %! assert (size(c), [1, numel(a)+numel(b)-1]) |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
105 |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
106 %!test |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
107 %! b = 1:10; |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
108 %! a = 1:3; |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
109 %! c = conv (a, b); |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
110 %! assert (size(c), [1, numel(a)+numel(b)-1]) |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
111 %!test |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
112 %! b = (1:10).'; |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
113 %! a = 1:3; |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
114 %! c = conv (a, b); |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
115 %! assert (size(c), [numel(a)+numel(b)-1, 1]) |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
116 %!test |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
117 %! b = 1:10; |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
118 %! a = (1:3).'; |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
119 %! c = conv (a, b); |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
120 %! assert (size(c), [1, numel(a)+numel(b)-1]) |
15e4a450bf84
conv.m: Correct row/col orientation of output
Ben Abbott <bpabbott@mac.com>
parents:
7411
diff
changeset
|
121 |