7017
|
1 ## Copyright (C) 1995, 1996, 1999, 2000, 2002, 2005, 2006, 2007 |
|
2 ## Kurt Hornik |
3426
|
3 ## |
3922
|
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. |
3426
|
10 ## |
3922
|
11 ## Octave is distributed in the hope that it will be useful, but |
2539
|
12 ## WITHOUT ANY WARRANTY; without even the implied warranty of |
|
13 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
3426
|
14 ## General Public License for more details. |
|
15 ## |
2539
|
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/>. |
2539
|
19 |
3381
|
20 ## -*- texinfo -*- |
3367
|
21 ## @deftypefn {Function File} {} detrend (@var{x}, @var{p}) |
|
22 ## If @var{x} is a vector, @code{detrend (@var{x}, @var{p})} removes the |
|
23 ## best fit of a polynomial of order @var{p} from the data @var{x}. |
3426
|
24 ## |
3367
|
25 ## If @var{x} is a matrix, @code{detrend (@var{x}, @var{p})} does the same |
|
26 ## for each column in @var{x}. |
3426
|
27 ## |
3367
|
28 ## The second argument is optional. If it is not specified, a value of 1 |
|
29 ## is assumed. This corresponds to removing a linear trend. |
|
30 ## @end deftypefn |
2539
|
31 |
5428
|
32 ## Author: KH <Kurt.Hornik@wu-wien.ac.at> |
2539
|
33 ## Created: 11 October 1994 |
|
34 ## Adapted-By: jwe |
3426
|
35 |
2539
|
36 function y = detrend (x, p) |
3426
|
37 |
2539
|
38 if (nargin == 1) |
|
39 p = 1; |
|
40 elseif (nargin == 2) |
4030
|
41 if (! (isscalar (p) && p == round (p) && p >= 0)) |
3457
|
42 error ("detrend: p must be a nonnegative integer"); |
2539
|
43 endif |
|
44 else |
6046
|
45 print_usage (); |
2539
|
46 endif |
3426
|
47 |
2539
|
48 [m, n] = size (x); |
|
49 if (m == 1) |
|
50 x = x'; |
|
51 endif |
3426
|
52 |
2539
|
53 r = rows (x); |
|
54 b = ((1 : r)' * ones (1, p + 1)) .^ (ones (r, 1) * (0 : p)); |
|
55 y = x - b * (b \ x); |
3426
|
56 |
2539
|
57 if (m == 1) |
|
58 y = y'; |
|
59 endif |
3426
|
60 |
2539
|
61 endfunction |
7279
|
62 |
|
63 %!test |
|
64 %! N=32; |
|
65 %! x = (0:1:N-1)/N + 2; |
|
66 %! y = detrend(x); |
|
67 %! assert(all (all (abs (y) < 20*eps))); |
|
68 |
|
69 %!test |
|
70 %! N=32; |
|
71 %! t = (0:1:N-1)/N; |
|
72 %! x = t .* t + 2; |
|
73 %! y = detrend(x,2); |
|
74 %! assert(all (all (abs (y) < 30*eps))); |
|
75 |
|
76 %!test |
|
77 %! N=32; |
|
78 %! t = (0:1:N-1)/N; |
|
79 %! x = [t;4*t-3]'; |
|
80 %! y = detrend(x); |
|
81 %! assert(all (all (abs (y) < 20*eps))); |
|
82 |