Just a small bug fix: Extension matching will be performed by extension's number...
[lcr.git] / ss5_decode.c
1 /*
2  * SS5 signal decoder.
3  *
4  * Copyright            by Andreas Eversberg (jolly@eversberg.eu)
5  *                      based on different decoders such as ISDN4Linux
6  *                      copyright by Karsten Keil
7  *
8  * This software may be used and distributed according to the terms
9  * of the GNU General Public License, incorporated herein by reference.
10  *
11  */
12
13 #include "main.h"
14 #include "ss5_decode.h"
15
16 /* enable level debugging */
17 //#define DEBUG_LEVELS
18
19 #define NCOEFF          8       /* number of frequencies to be analyzed */
20
21 #define TONE_MIN_DB     0.01995262 /* -17 db */
22 #define TONE_DIFF_DB    0.2 // 0.31622777 /*  -5 db */
23 #define NOISE_MIN_DB    (TONE_MIN_DB / 2) /* noise must be higher than the minimum of two tones */
24 #define SNR             1.3     /* noise may not exceed signal by that factor */
25
26 /* For DTMF recognition:
27  * 2 * cos(2 * PI * k / N) precalculated for all k
28  */
29 static signed long long cos2pik[NCOEFF] =
30 {
31         /* k = 2*cos(2*PI*f/8000), k << 15 
32          * 700, 900, 1100, 1300, 1500, 1700, 2400, 2600 */
33         55879, 49834, 42562, 34242, 25080, 15299, -20252, -29753
34 };
35
36 /* detection matrix for two frequencies */
37 static char decode_two[8][8] =
38 {
39         {' ', '1', '2', '4', '7', '*', ' ', ' '}, /* * = code 11 */
40         {'1', ' ', '3', '5', '8', '#', ' ', ' '}, /* # = code 12 */
41         {'2', '3', ' ', '6', '9', 'a', ' ', ' '}, /* a = KP1 */
42         {'4', '5', '6', ' ', '0', 'b', ' ', ' '}, /* b = KP2 */
43         {'7', '8', '9', '0', ' ', 'c', ' ', ' '}, /* c = ST */
44         {'*', '#', 'a', 'b', 'c', ' ', ' ', ' '},
45         {' ', ' ', ' ', ' ', ' ', ' ', ' ', 'C'}, /* C = 2600+2400 */
46         {' ', ' ', ' ', ' ', ' ', ' ', 'C', ' '}
47 };
48
49 static char decode_one[8] =
50         {' ', ' ', ' ', ' ', ' ', ' ', 'A', 'B'}; /* A = 2400, B = 2600 */
51 /*
52  * calculate the coefficients of the given sample and decode
53  */
54
55 char ss5_decode(unsigned char *data, int len)
56 {
57         signed short buf[len];
58         signed long sk, sk1, sk2, low, high;
59         int k, n, i;
60         int f1 = 0, f2 = 0;
61         double result[NCOEFF], power, noise, snr;
62         signed long long cos2pik_;
63         char digit = ' ';
64
65         /* convert samples */
66         for (i = 0; i < len; i++)
67                 buf[i] = audio_law_to_s32[*data++];
68
69         /* now we do noise level calculation */
70         low = 32767;
71         high = -32768;
72         for (n = 0; n < len; n++) {
73                 sk = buf[n];
74                 if (sk < low)
75                         low = sk;
76                 if (sk > high)
77                         high = sk;
78         }
79         noise = ((double)(high-low) / 65536.0);
80
81         /* check for minimum noise, or tone detection will not be necessary */
82         if (noise < NOISE_MIN_DB)
83                 return digit;
84
85         /* now we have a full buffer of signed long samples - we do goertzel */
86         for (k = 0; k < NCOEFF; k++) {
87                 sk = 0;
88                 sk1 = 0;
89                 sk2 = 0;
90                 cos2pik_ = cos2pik[k];
91                 for (n = 0; n < len; n++) {
92                         sk = ((cos2pik_*sk1)>>15) - sk2 + buf[n];
93                         sk2 = sk1;
94                         sk1 = sk;
95                 }
96                 sk >>= 8;
97                 sk2 >>= 8;
98                 if (sk > 32767 || sk < -32767 || sk2 > 32767 || sk2 < -32767)
99                         PERROR("Tone-Detection overflow\n");
100                 /* compute |X(k)|**2 */
101                 result[k] = sqrt (
102                                 (sk * sk) -
103                                 (((cos2pik[k] * sk) >> 15) * sk2) +
104                                 (sk2 * sk2)
105                         ) / len / 62; /* level of 1 is 0 db*/
106         }
107
108         /* find the two loudest frequencies + one less lower frequency to detect noise */
109         power = 0.0;
110         for (i = 0; i < NCOEFF; i++) {
111                 if (result[i] > power) {
112                         power = result[i];
113                         f1 = i;
114                 }
115         }
116         power = 0.0;
117         for (i = 0; i < NCOEFF; i++) {
118                 if (i != f1 && result[i] > power) {
119                         power = result[i];
120                         f2 = i;
121                 }
122         }
123
124         snr = 0;
125         /* check one frequency */
126         if (result[f1] > TONE_MIN_DB /* must be at least -17 db */
127          && result[f1]*SNR > noise) { /*  */
128                 digit = decode_one[f1];
129                 if (digit != ' ')
130                         snr = result[f1] / noise;
131         }
132         /* check two frequencies */
133         if (result[f1] > TONE_MIN_DB && result[f2] > TONE_MIN_DB /* must be at lease -17 db */
134          && result[f1]*TONE_DIFF_DB <= result[f2] /* f2 must be not less than 5 db below f1 */
135          && (result[f1]+result[f2])*SNR > noise) { /* */
136                 digit = decode_two[f1][f2];
137                 if (digit != ' ')
138                         snr = (result[f1]+result[f2]) / noise;
139         }
140
141         /* debug powers */
142 #ifdef DEBUG_LEVELS
143         for (i = 0; i < NCOEFF; i++)
144                 printf("%d:%3d %c ", i, (int)(result[i]*100), (f1==i || f2==i)?'*':' ');
145         printf("N:%3d digit:%c snr=%3d\n", (int)(noise*100), digit, (int)(snr*100));
146 //      if (result[f1]*TONE_DIFF_DB <= result[f2]) /* f2 must be not less than 5 db below f1 */
147 //              printf("jo!");
148 #endif
149
150         return digit;
151 }
152
153 void ss5_test_decode(void)
154 {
155 #ifdef DEBUG_LEVELS
156         double phase;
157         int i, j;
158         signed short sample;
159
160         unsigned char buffer[SS5_DECODER_NPOINTS];
161         for (i = 0; i < 4000; i += 10) {
162                 phase = 2.0 * 3.14159265 * i / 8000.0;
163                 for (j = 0; j < SS5_DECODER_NPOINTS; j++) {
164                         sample = sin(phase * j) * 1000;
165                         buffer[j] = audio_s16_to_law[sample & 0xffff];
166                 }
167                 printf("FRQ:%04d:", i);
168                 ss5_decode(buffer, SS5_DECODER_NPOINTS);
169         }
170 #endif
171 }
172