code cleanup
[speexdsp.git] / libspeex / lpc.c
1 /*
2   Copyright 1992, 1993, 1994 by Jutta Degener and Carsten Bormann,
3   Technische Universitaet Berlin
4
5   Any use of this software is permitted provided that this notice is not
6   removed and that neither the authors nor the Technische Universitaet Berlin
7   are deemed to have made any representations as to the suitability of this
8   software for any purpose nor are held responsible for any defects of
9   this software.  THERE IS ABSOLUTELY NO WARRANTY FOR THIS SOFTWARE.
10
11   As a matter of courtesy, the authors request to be informed about uses
12   this software has found, about bugs in this software, and about any
13   improvements that may be of general interest.
14
15   Berlin, 28.11.1994
16   Jutta Degener
17   Carsten Bormann
18 */
19 /* Speex License:
20
21    This library is free software; you can redistribute it and/or
22    modify it under the terms of the GNU Lesser General Public
23    License as published by the Free Software Foundation; either
24    version 2.1 of the License, or (at your option) any later version.
25    
26    This library is distributed in the hope that it will be useful,
27    but WITHOUT ANY WARRANTY; without even the implied warranty of
28    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
29    Lesser General Public License for more details.
30    
31    You should have received a copy of the GNU Lesser General Public
32    License along with this library; if not, write to the Free Software
33    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
34 */
35
36
37
38 /* LPC- and Reflection Coefficients
39  *
40  * The next two functions calculate linear prediction coefficients
41  * and/or the related reflection coefficients from the first P_MAX+1
42  * values of the autocorrelation function.
43  */
44
45 /* Invented by N. Levinson in 1947, modified by J. Durbin in 1959.
46  */
47
48 #include "lpc.h"
49
50 float                      /* returns minimum mean square error    */
51 wld(
52     float       * lpc, /*      [0...p-1] LPC coefficients      */
53     const float * ac,  /*  in: [0...p] autocorrelation values  */
54     float       * ref, /* out: [0...p-1] reflection coef's     */
55     int p
56     )
57 {
58    int i, j;  float r, error = ac[0];
59
60    if (ac[0] == 0) {
61       for (i = 0; i < p; i++) ref[i] = 0; return 0; }
62
63    for (i = 0; i < p; i++) {
64
65       /* Sum up this iteration's reflection coefficient.
66        */
67       r = -ac[i + 1];
68       for (j = 0; j < i; j++) r -= lpc[j] * ac[i - j];
69       ref[i] = r /= error;
70
71       /*  Update LPC coefficients and total error.
72        */
73       lpc[i] = r;
74       for (j = 0; j < i/2; j++) {
75          float tmp  = lpc[j];
76          lpc[j]     += r * lpc[i-1-j];
77          lpc[i-1-j] += r * tmp;
78       }
79       if (i % 2) lpc[j] += lpc[j] * r;
80
81       error *= 1.0 - r * r;
82    }
83    return error;
84 }
85
86
87 /* Compute the autocorrelation
88  *                      ,--,
89  *              ac(i) = >  x(n) * x(n-i)  for all n
90  *                      `--'
91  * for lags between 0 and lag-1, and x == 0 outside 0...n-1
92  */
93 void autocorr(
94               const float * x,   /*  in: [0...n-1] samples x   */
95               float *ac,   /* out: [0...lag-1] ac values */
96               int lag, int   n)
97 {
98    float d; int i;
99    while (lag--) {
100       for (i = lag, d = 0; i < n; i++) d += x[i] * x[i-lag];
101       ac[lag] = d;
102    }
103 }