]> bbs.cooldavid.org Git - net-next-2.6.git/blob - drivers/media/video/cx88/cx88-dsp.c
V4L/DVB: cx88: convert core->tvaudio into an enum
[net-next-2.6.git] / drivers / media / video / cx88 / cx88-dsp.c
1 /*
2  *
3  *  Stereo and SAP detection for cx88
4  *
5  *  Copyright (c) 2009 Marton Balint <cus@fazekas.hu>
6  *
7  *  This program is free software; you can redistribute it and/or modify
8  *  it under the terms of the GNU General Public License as published by
9  *  the Free Software Foundation; either version 2 of the License, or
10  *  (at your option) any later version.
11  *
12  *  This program is distributed in the hope that it will be useful,
13  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
14  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  *  GNU General Public License for more details.
16  *
17  *  You should have received a copy of the GNU General Public License
18  *  along with this program; if not, write to the Free Software
19  *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20  */
21
22 #include <linux/slab.h>
23 #include <linux/kernel.h>
24 #include <linux/module.h>
25 #include <linux/jiffies.h>
26 #include <asm/div64.h>
27
28 #include "cx88.h"
29 #include "cx88-reg.h"
30
31 #define INT_PI                  ((s32)(3.141592653589 * 32768.0))
32
33 #define compat_remainder(a, b) \
34          ((float)(((s32)((a)*100))%((s32)((b)*100)))/100.0)
35
36 #define baseband_freq(carrier, srate, tone) ((s32)( \
37          (compat_remainder(carrier + tone, srate)) / srate * 2 * INT_PI))
38
39 /* We calculate the baseband frequencies of the carrier and the pilot tones
40  * based on the the sampling rate of the audio rds fifo. */
41
42 #define FREQ_A2_CARRIER         baseband_freq(54687.5, 2689.36, 0.0)
43 #define FREQ_A2_DUAL            baseband_freq(54687.5, 2689.36, 274.1)
44 #define FREQ_A2_STEREO          baseband_freq(54687.5, 2689.36, 117.5)
45
46 /* The frequencies below are from the reference driver. They probably need
47  * further adjustments, because they are not tested at all. You may even need
48  * to play a bit with the registers of the chip to select the proper signal
49  * for the input of the audio rds fifo, and measure it's sampling rate to
50  * calculate the proper baseband frequencies... */
51
52 #define FREQ_A2M_CARRIER        ((s32)(2.114516 * 32768.0))
53 #define FREQ_A2M_DUAL           ((s32)(2.754916 * 32768.0))
54 #define FREQ_A2M_STEREO         ((s32)(2.462326 * 32768.0))
55
56 #define FREQ_EIAJ_CARRIER       ((s32)(1.963495 * 32768.0)) /* 5pi/8  */
57 #define FREQ_EIAJ_DUAL          ((s32)(2.562118 * 32768.0))
58 #define FREQ_EIAJ_STEREO        ((s32)(2.601053 * 32768.0))
59
60 #define FREQ_BTSC_DUAL          ((s32)(1.963495 * 32768.0)) /* 5pi/8  */
61 #define FREQ_BTSC_DUAL_REF      ((s32)(1.374446 * 32768.0)) /* 7pi/16 */
62
63 #define FREQ_BTSC_SAP           ((s32)(2.471532 * 32768.0))
64 #define FREQ_BTSC_SAP_REF       ((s32)(1.730072 * 32768.0))
65
66 /* The spectrum of the signal should be empty between these frequencies. */
67 #define FREQ_NOISE_START        ((s32)(0.100000 * 32768.0))
68 #define FREQ_NOISE_END          ((s32)(1.200000 * 32768.0))
69
70 static unsigned int dsp_debug;
71 module_param(dsp_debug, int, 0644);
72 MODULE_PARM_DESC(dsp_debug, "enable audio dsp debug messages");
73
74 #define dprintk(level, fmt, arg...)     if (dsp_debug >= level) \
75         printk(KERN_DEBUG "%s/0: " fmt, core->name , ## arg)
76
77 static s32 int_cos(u32 x)
78 {
79         u32 t2, t4, t6, t8;
80         s32 ret;
81         u16 period = x / INT_PI;
82         if (period % 2)
83                 return -int_cos(x - INT_PI);
84         x = x % INT_PI;
85         if (x > INT_PI/2)
86                 return -int_cos(INT_PI/2 - (x % (INT_PI/2)));
87         /* Now x is between 0 and INT_PI/2.
88          * To calculate cos(x) we use it's Taylor polinom. */
89         t2 = x*x/32768/2;
90         t4 = t2*x/32768*x/32768/3/4;
91         t6 = t4*x/32768*x/32768/5/6;
92         t8 = t6*x/32768*x/32768/7/8;
93         ret = 32768-t2+t4-t6+t8;
94         return ret;
95 }
96
97 static u32 int_goertzel(s16 x[], u32 N, u32 freq)
98 {
99         /* We use the Goertzel algorithm to determine the power of the
100          * given frequency in the signal */
101         s32 s_prev = 0;
102         s32 s_prev2 = 0;
103         s32 coeff = 2*int_cos(freq);
104         u32 i;
105
106         u64 tmp;
107         u32 divisor;
108
109         for (i = 0; i < N; i++) {
110                 s32 s = x[i] + ((s64)coeff*s_prev/32768) - s_prev2;
111                 s_prev2 = s_prev;
112                 s_prev = s;
113         }
114
115         tmp = (s64)s_prev2 * s_prev2 + (s64)s_prev * s_prev -
116                       (s64)coeff * s_prev2 * s_prev / 32768;
117
118         /* XXX: N must be low enough so that N*N fits in s32.
119          * Else we need two divisions. */
120         divisor = N * N;
121         do_div(tmp, divisor);
122
123         return (u32) tmp;
124 }
125
126 static u32 freq_magnitude(s16 x[], u32 N, u32 freq)
127 {
128         u32 sum = int_goertzel(x, N, freq);
129         return (u32)int_sqrt(sum);
130 }
131
132 static u32 noise_magnitude(s16 x[], u32 N, u32 freq_start, u32 freq_end)
133 {
134         int i;
135         u32 sum = 0;
136         u32 freq_step;
137         int samples = 5;
138
139         if (N > 192) {
140                 /* The last 192 samples are enough for noise detection */
141                 x += (N-192);
142                 N = 192;
143         }
144
145         freq_step = (freq_end - freq_start) / (samples - 1);
146
147         for (i = 0; i < samples; i++) {
148                 sum += int_goertzel(x, N, freq_start);
149                 freq_start += freq_step;
150         }
151
152         return (u32)int_sqrt(sum / samples);
153 }
154
155 static s32 detect_a2_a2m_eiaj(struct cx88_core *core, s16 x[], u32 N)
156 {
157         s32 carrier, stereo, dual, noise;
158         s32 carrier_freq, stereo_freq, dual_freq;
159         s32 ret;
160
161         switch (core->tvaudio) {
162         case WW_BG:
163         case WW_DK:
164                 carrier_freq = FREQ_A2_CARRIER;
165                 stereo_freq = FREQ_A2_STEREO;
166                 dual_freq = FREQ_A2_DUAL;
167                 break;
168         case WW_M:
169                 carrier_freq = FREQ_A2M_CARRIER;
170                 stereo_freq = FREQ_A2M_STEREO;
171                 dual_freq = FREQ_A2M_DUAL;
172                 break;
173         case WW_EIAJ:
174                 carrier_freq = FREQ_EIAJ_CARRIER;
175                 stereo_freq = FREQ_EIAJ_STEREO;
176                 dual_freq = FREQ_EIAJ_DUAL;
177                 break;
178         case WW_NONE:
179         case WW_BTSC:
180         case WW_I:
181         case WW_L:
182         case WW_I2SPT:
183         case WW_FM:
184         case WW_I2SADC:
185                 printk(KERN_WARNING "%s/0: unsupported audio mode %d for %s\n",
186                        core->name, core->tvaudio, __func__);
187                 return UNSET;
188         }
189
190         carrier = freq_magnitude(x, N, carrier_freq);
191         stereo  = freq_magnitude(x, N, stereo_freq);
192         dual    = freq_magnitude(x, N, dual_freq);
193         noise   = noise_magnitude(x, N, FREQ_NOISE_START, FREQ_NOISE_END);
194
195         dprintk(1, "detect a2/a2m/eiaj: carrier=%d, stereo=%d, dual=%d, "
196                    "noise=%d\n", carrier, stereo, dual, noise);
197
198         if (stereo > dual)
199                 ret = V4L2_TUNER_SUB_STEREO;
200         else
201                 ret = V4L2_TUNER_SUB_LANG1 | V4L2_TUNER_SUB_LANG2;
202
203         if (core->tvaudio == WW_EIAJ) {
204                 /* EIAJ checks may need adjustments */
205                 if ((carrier > max(stereo, dual)*2) &&
206                     (carrier < max(stereo, dual)*6) &&
207                     (carrier > 20 && carrier < 200) &&
208                     (max(stereo, dual) > min(stereo, dual))) {
209                         /* For EIAJ the carrier is always present,
210                            so we probably don't need noise detection */
211                         return ret;
212                 }
213         } else {
214                 if ((carrier > max(stereo, dual)*2) &&
215                     (carrier < max(stereo, dual)*8) &&
216                     (carrier > 20 && carrier < 200) &&
217                     (noise < 10) &&
218                     (max(stereo, dual) > min(stereo, dual)*2)) {
219                         return ret;
220                 }
221         }
222         return V4L2_TUNER_SUB_MONO;
223 }
224
225 static s32 detect_btsc(struct cx88_core *core, s16 x[], u32 N)
226 {
227         s32 sap_ref = freq_magnitude(x, N, FREQ_BTSC_SAP_REF);
228         s32 sap = freq_magnitude(x, N, FREQ_BTSC_SAP);
229         s32 dual_ref = freq_magnitude(x, N, FREQ_BTSC_DUAL_REF);
230         s32 dual = freq_magnitude(x, N, FREQ_BTSC_DUAL);
231         dprintk(1, "detect btsc: dual_ref=%d, dual=%d, sap_ref=%d, sap=%d"
232                    "\n", dual_ref, dual, sap_ref, sap);
233         /* FIXME: Currently not supported */
234         return UNSET;
235 }
236
237 static s16 *read_rds_samples(struct cx88_core *core, u32 *N)
238 {
239         struct sram_channel *srch = &cx88_sram_channels[SRAM_CH27];
240         s16 *samples;
241
242         unsigned int i;
243         unsigned int bpl = srch->fifo_size/AUD_RDS_LINES;
244         unsigned int spl = bpl/4;
245         unsigned int sample_count = spl*(AUD_RDS_LINES-1);
246
247         u32 current_address = cx_read(srch->ptr1_reg);
248         u32 offset = (current_address - srch->fifo_start + bpl);
249
250         dprintk(1, "read RDS samples: current_address=%08x (offset=%08x), "
251                 "sample_count=%d, aud_intstat=%08x\n", current_address,
252                 current_address - srch->fifo_start, sample_count,
253                 cx_read(MO_AUD_INTSTAT));
254
255         samples = kmalloc(sizeof(s16)*sample_count, GFP_KERNEL);
256         if (!samples)
257                 return NULL;
258
259         *N = sample_count;
260
261         for (i = 0; i < sample_count; i++)  {
262                 offset = offset % (AUD_RDS_LINES*bpl);
263                 samples[i] = cx_read(srch->fifo_start + offset);
264                 offset += 4;
265         }
266
267         if (dsp_debug >= 2) {
268                 dprintk(2, "RDS samples dump: ");
269                 for (i = 0; i < sample_count; i++)
270                         printk("%hd ", samples[i]);
271                 printk(".\n");
272         }
273
274         return samples;
275 }
276
277 s32 cx88_dsp_detect_stereo_sap(struct cx88_core *core)
278 {
279         s16 *samples;
280         u32 N = 0;
281         s32 ret = UNSET;
282
283         /* If audio RDS fifo is disabled, we can't read the samples */
284         if (!(cx_read(MO_AUD_DMACNTRL) & 0x04))
285                 return ret;
286         if (!(cx_read(AUD_CTL) & EN_FMRADIO_EN_RDS))
287                 return ret;
288
289         /* Wait at least 500 ms after an audio standard change */
290         if (time_before(jiffies, core->last_change + msecs_to_jiffies(500)))
291                 return ret;
292
293         samples = read_rds_samples(core, &N);
294
295         if (!samples)
296                 return ret;
297
298         switch (core->tvaudio) {
299         case WW_BG:
300         case WW_DK:
301         case WW_EIAJ:
302         case WW_M:
303                 ret = detect_a2_a2m_eiaj(core, samples, N);
304                 break;
305         case WW_BTSC:
306                 ret = detect_btsc(core, samples, N);
307                 break;
308         case WW_NONE:
309         case WW_I:
310         case WW_L:
311         case WW_I2SPT:
312         case WW_FM:
313         case WW_I2SADC:
314                 break;
315         }
316
317         kfree(samples);
318
319         if (UNSET != ret)
320                 dprintk(1, "stereo/sap detection result:%s%s%s\n",
321                            (ret & V4L2_TUNER_SUB_MONO) ? " mono" : "",
322                            (ret & V4L2_TUNER_SUB_STEREO) ? " stereo" : "",
323                            (ret & V4L2_TUNER_SUB_LANG2) ? " dual" : "");
324
325         return ret;
326 }
327 EXPORT_SYMBOL(cx88_dsp_detect_stereo_sap);
328