3 * Copyright (C) 2000-2002 Michel Lespinasse <walken@zoy.org>
4 * Copyright (C) 1999-2000 Aaron Holtzman <aholtzma@ess.engr.uvic.ca>
6 * This file is part of a52dec, a free ATSC A-52 stream decoder.
7 * See http://liba52.sourceforge.net/ for updates.
9 * a52dec is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation; either version 2 of the License, or
12 * (at your option) any later version.
14 * a52dec is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU General Public License for more details.
19 * You should have received a copy of the GNU General Public License
20 * along with this program; if not, write to the Free Software
21 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
30 #include "a52_internal.h"
32 #define CONVERT(acmod,output) (((output) << 3) + (acmod))
34 int a52_downmix_init (int input, int flags, sample_t * level,
35 sample_t clev, sample_t slev)
37 static uint8_t table[11][8] = {
38 {A52_CHANNEL, A52_DOLBY, A52_STEREO, A52_STEREO,
39 A52_STEREO, A52_STEREO, A52_STEREO, A52_STEREO},
40 {A52_MONO, A52_MONO, A52_MONO, A52_MONO,
41 A52_MONO, A52_MONO, A52_MONO, A52_MONO},
42 {A52_CHANNEL, A52_DOLBY, A52_STEREO, A52_STEREO,
43 A52_STEREO, A52_STEREO, A52_STEREO, A52_STEREO},
44 {A52_CHANNEL, A52_DOLBY, A52_STEREO, A52_3F,
45 A52_STEREO, A52_3F, A52_STEREO, A52_3F},
46 {A52_CHANNEL, A52_DOLBY, A52_STEREO, A52_STEREO,
47 A52_2F1R, A52_2F1R, A52_2F1R, A52_2F1R},
48 {A52_CHANNEL, A52_DOLBY, A52_STEREO, A52_STEREO,
49 A52_2F1R, A52_3F1R, A52_2F1R, A52_3F1R},
50 {A52_CHANNEL, A52_DOLBY, A52_STEREO, A52_3F,
51 A52_2F2R, A52_2F2R, A52_2F2R, A52_2F2R},
52 {A52_CHANNEL, A52_DOLBY, A52_STEREO, A52_3F,
53 A52_2F2R, A52_3F2R, A52_2F2R, A52_3F2R},
54 {A52_CHANNEL1, A52_MONO, A52_MONO, A52_MONO,
55 A52_MONO, A52_MONO, A52_MONO, A52_MONO},
56 {A52_CHANNEL2, A52_MONO, A52_MONO, A52_MONO,
57 A52_MONO, A52_MONO, A52_MONO, A52_MONO},
58 {A52_CHANNEL, A52_DOLBY, A52_STEREO, A52_DOLBY,
59 A52_DOLBY, A52_DOLBY, A52_DOLBY, A52_DOLBY}
63 output = flags & A52_CHANNEL_MASK;
64 if (output > A52_DOLBY)
67 output = table[output][input & 7];
69 if ((output == A52_STEREO) &&
70 ((input == A52_DOLBY) || ((input == A52_3F) && (clev == LEVEL_3DB))))
73 if (flags & A52_ADJUST_LEVEL)
74 switch (CONVERT (input & 7, output)) {
76 case CONVERT (A52_3F, A52_MONO):
77 *level *= LEVEL_3DB / (1 + clev);
80 case CONVERT (A52_STEREO, A52_MONO):
81 case CONVERT (A52_2F2R, A52_2F1R):
82 case CONVERT (A52_3F2R, A52_3F1R):
87 case CONVERT (A52_3F2R, A52_2F1R):
88 if (clev < LEVEL_PLUS3DB - 1)
91 case CONVERT (A52_3F, A52_STEREO):
92 case CONVERT (A52_3F1R, A52_2F1R):
93 case CONVERT (A52_3F1R, A52_2F2R):
94 case CONVERT (A52_3F2R, A52_2F2R):
98 case CONVERT (A52_2F1R, A52_MONO):
99 *level *= LEVEL_PLUS3DB / (2 + slev);
102 case CONVERT (A52_2F1R, A52_STEREO):
103 case CONVERT (A52_3F1R, A52_3F):
104 *level /= 1 + slev * LEVEL_3DB;
107 case CONVERT (A52_3F1R, A52_MONO):
108 *level *= LEVEL_3DB / (1 + clev + 0.5 * slev);
111 case CONVERT (A52_3F1R, A52_STEREO):
112 *level /= 1 + clev + slev * LEVEL_3DB;
115 case CONVERT (A52_2F2R, A52_MONO):
116 *level *= LEVEL_3DB / (1 + slev);
119 case CONVERT (A52_2F2R, A52_STEREO):
120 case CONVERT (A52_3F2R, A52_3F):
124 case CONVERT (A52_3F2R, A52_MONO):
125 *level *= LEVEL_3DB / (1 + clev + slev);
128 case CONVERT (A52_3F2R, A52_STEREO):
129 *level /= 1 + clev + slev;
132 case CONVERT (A52_MONO, A52_DOLBY):
133 *level *= LEVEL_PLUS3DB;
136 case CONVERT (A52_3F, A52_DOLBY):
137 case CONVERT (A52_2F1R, A52_DOLBY):
138 *level *= 1 / (1 + LEVEL_3DB);
141 case CONVERT (A52_3F1R, A52_DOLBY):
142 case CONVERT (A52_2F2R, A52_DOLBY):
143 *level *= 1 / (1 + 2 * LEVEL_3DB);
146 case CONVERT (A52_3F2R, A52_DOLBY):
147 *level *= 1 / (1 + 3 * LEVEL_3DB);
154 int a52_downmix_coeff (sample_t * coeff, int acmod, int output, sample_t level,
155 sample_t clev, sample_t slev)
157 switch (CONVERT (acmod, output & A52_CHANNEL_MASK)) {
159 case CONVERT (A52_CHANNEL, A52_CHANNEL):
160 case CONVERT (A52_MONO, A52_MONO):
161 case CONVERT (A52_STEREO, A52_STEREO):
162 case CONVERT (A52_3F, A52_3F):
163 case CONVERT (A52_2F1R, A52_2F1R):
164 case CONVERT (A52_3F1R, A52_3F1R):
165 case CONVERT (A52_2F2R, A52_2F2R):
166 case CONVERT (A52_3F2R, A52_3F2R):
167 case CONVERT (A52_STEREO, A52_DOLBY):
168 coeff[0] = coeff[1] = coeff[2] = coeff[3] = coeff[4] = level;
171 case CONVERT (A52_CHANNEL, A52_MONO):
172 coeff[0] = coeff[1] = level * LEVEL_6DB;
175 case CONVERT (A52_STEREO, A52_MONO):
176 coeff[0] = coeff[1] = level * LEVEL_3DB;
179 case CONVERT (A52_3F, A52_MONO):
180 coeff[0] = coeff[2] = level * LEVEL_3DB;
181 coeff[1] = level * clev * LEVEL_PLUS3DB;
184 case CONVERT (A52_2F1R, A52_MONO):
185 coeff[0] = coeff[1] = level * LEVEL_3DB;
186 coeff[2] = level * slev * LEVEL_3DB;
189 case CONVERT (A52_2F2R, A52_MONO):
190 coeff[0] = coeff[1] = level * LEVEL_3DB;
191 coeff[2] = coeff[3] = level * slev * LEVEL_3DB;
194 case CONVERT (A52_3F1R, A52_MONO):
195 coeff[0] = coeff[2] = level * LEVEL_3DB;
196 coeff[1] = level * clev * LEVEL_PLUS3DB;
197 coeff[3] = level * slev * LEVEL_3DB;
200 case CONVERT (A52_3F2R, A52_MONO):
201 coeff[0] = coeff[2] = level * LEVEL_3DB;
202 coeff[1] = level * clev * LEVEL_PLUS3DB;
203 coeff[3] = coeff[4] = level * slev * LEVEL_3DB;
206 case CONVERT (A52_MONO, A52_DOLBY):
207 coeff[0] = level * LEVEL_3DB;
210 case CONVERT (A52_3F, A52_DOLBY):
212 case CONVERT (A52_3F, A52_STEREO):
213 case CONVERT (A52_3F1R, A52_2F1R):
214 case CONVERT (A52_3F2R, A52_2F2R):
215 coeff[0] = coeff[2] = coeff[3] = coeff[4] = level;
216 coeff[1] = level * clev;
219 case CONVERT (A52_2F1R, A52_DOLBY):
221 case CONVERT (A52_2F1R, A52_STEREO):
222 coeff[0] = coeff[1] = level;
223 coeff[2] = level * slev * LEVEL_3DB;
226 case CONVERT (A52_3F1R, A52_DOLBY):
229 case CONVERT (A52_3F1R, A52_STEREO):
230 coeff[0] = coeff[2] = level;
231 coeff[1] = level * clev;
232 coeff[3] = level * slev * LEVEL_3DB;
235 case CONVERT (A52_2F2R, A52_DOLBY):
237 case CONVERT (A52_2F2R, A52_STEREO):
238 coeff[0] = coeff[1] = level;
239 coeff[2] = coeff[3] = level * slev;
242 case CONVERT (A52_3F2R, A52_DOLBY):
244 case CONVERT (A52_3F2R, A52_2F1R):
246 case CONVERT (A52_3F2R, A52_STEREO):
247 coeff[0] = coeff[2] = level;
248 coeff[1] = level * clev;
249 coeff[3] = coeff[4] = level * slev;
252 case CONVERT (A52_3F1R, A52_3F):
253 coeff[0] = coeff[1] = coeff[2] = level;
254 coeff[3] = level * slev * LEVEL_3DB;
257 case CONVERT (A52_3F2R, A52_3F):
258 coeff[0] = coeff[1] = coeff[2] = level;
259 coeff[3] = coeff[4] = level * slev;
262 case CONVERT (A52_2F2R, A52_2F1R):
263 coeff[0] = coeff[1] = level;
264 coeff[2] = coeff[3] = level * LEVEL_3DB;
267 case CONVERT (A52_3F2R, A52_3F1R):
268 coeff[0] = coeff[1] = coeff[2] = level;
269 coeff[3] = coeff[4] = level * LEVEL_3DB;
272 case CONVERT (A52_2F1R, A52_2F2R):
273 coeff[0] = coeff[1] = level;
274 coeff[2] = level * LEVEL_3DB;
277 case CONVERT (A52_3F1R, A52_2F2R):
278 coeff[0] = coeff[2] = level;
279 coeff[1] = level * clev;
280 coeff[3] = level * LEVEL_3DB;
283 case CONVERT (A52_3F1R, A52_3F2R):
284 coeff[0] = coeff[1] = coeff[2] = level;
285 coeff[3] = level * LEVEL_3DB;
288 case CONVERT (A52_CHANNEL, A52_CHANNEL1):
293 case CONVERT (A52_CHANNEL, A52_CHANNEL2):
299 return -1; /* NOTREACHED */
302 static void mix2to1 (sample_t * dest, sample_t * src, sample_t bias)
306 for (i = 0; i < 256; i++)
307 dest[i] += src[i] + bias;
310 static void mix3to1 (sample_t * samples, sample_t bias)
314 for (i = 0; i < 256; i++)
315 samples[i] += samples[i + 256] + samples[i + 512] + bias;
318 static void mix4to1 (sample_t * samples, sample_t bias)
322 for (i = 0; i < 256; i++)
323 samples[i] += (samples[i + 256] + samples[i + 512] +
324 samples[i + 768] + bias);
327 static void mix5to1 (sample_t * samples, sample_t bias)
331 for (i = 0; i < 256; i++)
332 samples[i] += (samples[i + 256] + samples[i + 512] +
333 samples[i + 768] + samples[i + 1024] + bias);
336 static void mix3to2 (sample_t * samples, sample_t bias)
341 for (i = 0; i < 256; i++) {
342 common = samples[i + 256] + bias;
343 samples[i] += common;
344 samples[i + 256] = samples[i + 512] + common;
348 static void mix21to2 (sample_t * left, sample_t * right, sample_t bias)
353 for (i = 0; i < 256; i++) {
354 common = right[i + 256] + bias;
360 static void mix21toS (sample_t * samples, sample_t bias)
365 for (i = 0; i < 256; i++) {
366 surround = samples[i + 512];
367 samples[i] += bias - surround;
368 samples[i + 256] += bias + surround;
372 static void mix31to2 (sample_t * samples, sample_t bias)
377 for (i = 0; i < 256; i++) {
378 common = samples[i + 256] + samples[i + 768] + bias;
379 samples[i] += common;
380 samples[i + 256] = samples[i + 512] + common;
384 static void mix31toS (sample_t * samples, sample_t bias)
387 sample_t common, surround;
389 for (i = 0; i < 256; i++) {
390 common = samples[i + 256] + bias;
391 surround = samples[i + 768];
392 samples[i] += common - surround;
393 samples[i + 256] = samples[i + 512] + common + surround;
397 static void mix22toS (sample_t * samples, sample_t bias)
402 for (i = 0; i < 256; i++) {
403 surround = samples[i + 512] + samples[i + 768];
404 samples[i] += bias - surround;
405 samples[i + 256] += bias + surround;
409 static void mix32to2 (sample_t * samples, sample_t bias)
414 for (i = 0; i < 256; i++) {
415 common = samples[i + 256] + bias;
416 samples[i] += common + samples[i + 768];
417 samples[i + 256] = common + samples[i + 512] + samples[i + 1024];
421 static void mix32toS (sample_t * samples, sample_t bias)
424 sample_t common, surround;
426 for (i = 0; i < 256; i++) {
427 common = samples[i + 256] + bias;
428 surround = samples[i + 768] + samples[i + 1024];
429 samples[i] += common - surround;
430 samples[i + 256] = samples[i + 512] + common + surround;
434 static void move2to1 (sample_t * src, sample_t * dest, sample_t bias)
438 for (i = 0; i < 256; i++)
439 dest[i] = src[i] + src[i + 256] + bias;
442 static void zero (sample_t * samples)
446 for (i = 0; i < 256; i++)
450 void a52_downmix (sample_t * samples, int acmod, int output, sample_t bias,
451 sample_t clev, sample_t slev)
453 switch (CONVERT (acmod, output & A52_CHANNEL_MASK)) {
455 case CONVERT (A52_CHANNEL, A52_CHANNEL2):
456 memcpy (samples, samples + 256, 256 * sizeof (sample_t));
459 case CONVERT (A52_CHANNEL, A52_MONO):
460 case CONVERT (A52_STEREO, A52_MONO):
462 mix2to1 (samples, samples + 256, bias);
465 case CONVERT (A52_2F1R, A52_MONO):
468 case CONVERT (A52_3F, A52_MONO):
470 mix3to1 (samples, bias);
473 case CONVERT (A52_3F1R, A52_MONO):
476 case CONVERT (A52_2F2R, A52_MONO):
479 mix4to1 (samples, bias);
482 case CONVERT (A52_3F2R, A52_MONO):
485 mix5to1 (samples, bias);
488 case CONVERT (A52_MONO, A52_DOLBY):
489 memcpy (samples + 256, samples, 256 * sizeof (sample_t));
492 case CONVERT (A52_3F, A52_STEREO):
493 case CONVERT (A52_3F, A52_DOLBY):
495 mix3to2 (samples, bias);
498 case CONVERT (A52_2F1R, A52_STEREO):
501 mix21to2 (samples, samples + 256, bias);
504 case CONVERT (A52_2F1R, A52_DOLBY):
505 mix21toS (samples, bias);
508 case CONVERT (A52_3F1R, A52_STEREO):
511 mix31to2 (samples, bias);
514 case CONVERT (A52_3F1R, A52_DOLBY):
515 mix31toS (samples, bias);
518 case CONVERT (A52_2F2R, A52_STEREO):
521 mix2to1 (samples, samples + 512, bias);
522 mix2to1 (samples + 256, samples + 768, bias);
525 case CONVERT (A52_2F2R, A52_DOLBY):
526 mix22toS (samples, bias);
529 case CONVERT (A52_3F2R, A52_STEREO):
532 mix32to2 (samples, bias);
535 case CONVERT (A52_3F2R, A52_DOLBY):
536 mix32toS (samples, bias);
539 case CONVERT (A52_3F1R, A52_3F):
542 mix21to2 (samples, samples + 512, bias);
545 case CONVERT (A52_3F2R, A52_3F):
548 mix2to1 (samples, samples + 768, bias);
549 mix2to1 (samples + 512, samples + 1024, bias);
552 case CONVERT (A52_3F1R, A52_2F1R):
553 mix3to2 (samples, bias);
554 memcpy (samples + 512, samples + 768, 256 * sizeof (sample_t));
557 case CONVERT (A52_2F2R, A52_2F1R):
558 mix2to1 (samples + 512, samples + 768, bias);
561 case CONVERT (A52_3F2R, A52_2F1R):
562 mix3to2 (samples, bias);
563 move2to1 (samples + 768, samples + 512, bias);
566 case CONVERT (A52_3F2R, A52_3F1R):
567 mix2to1 (samples + 768, samples + 1024, bias);
570 case CONVERT (A52_2F1R, A52_2F2R):
571 memcpy (samples + 768, samples + 512, 256 * sizeof (sample_t));
574 case CONVERT (A52_3F1R, A52_2F2R):
575 mix3to2 (samples, bias);
576 memcpy (samples + 512, samples + 768, 256 * sizeof (sample_t));
579 case CONVERT (A52_3F2R, A52_2F2R):
580 mix3to2 (samples, bias);
581 memcpy (samples + 512, samples + 768, 256 * sizeof (sample_t));
582 memcpy (samples + 768, samples + 1024, 256 * sizeof (sample_t));
585 case CONVERT (A52_3F1R, A52_3F2R):
586 memcpy (samples + 1027, samples + 768, 256 * sizeof (sample_t));
591 void a52_upmix (sample_t * samples, int acmod, int output)
593 switch (CONVERT (acmod, output & A52_CHANNEL_MASK)) {
595 case CONVERT (A52_CHANNEL, A52_CHANNEL2):
596 memcpy (samples + 256, samples, 256 * sizeof (sample_t));
599 case CONVERT (A52_3F2R, A52_MONO):
600 zero (samples + 1024);
601 case CONVERT (A52_3F1R, A52_MONO):
602 case CONVERT (A52_2F2R, A52_MONO):
603 zero (samples + 768);
604 case CONVERT (A52_3F, A52_MONO):
605 case CONVERT (A52_2F1R, A52_MONO):
606 zero (samples + 512);
607 case CONVERT (A52_CHANNEL, A52_MONO):
608 case CONVERT (A52_STEREO, A52_MONO):
609 zero (samples + 256);
612 case CONVERT (A52_3F2R, A52_STEREO):
613 case CONVERT (A52_3F2R, A52_DOLBY):
614 zero (samples + 1024);
615 case CONVERT (A52_3F1R, A52_STEREO):
616 case CONVERT (A52_3F1R, A52_DOLBY):
617 zero (samples + 768);
618 case CONVERT (A52_3F, A52_STEREO):
619 case CONVERT (A52_3F, A52_DOLBY):
621 memcpy (samples + 512, samples + 256, 256 * sizeof (sample_t));
622 zero (samples + 256);
625 case CONVERT (A52_2F2R, A52_STEREO):
626 case CONVERT (A52_2F2R, A52_DOLBY):
627 zero (samples + 768);
628 case CONVERT (A52_2F1R, A52_STEREO):
629 case CONVERT (A52_2F1R, A52_DOLBY):
630 zero (samples + 512);
633 case CONVERT (A52_3F2R, A52_3F):
634 zero (samples + 1024);
635 case CONVERT (A52_3F1R, A52_3F):
636 case CONVERT (A52_2F2R, A52_2F1R):
637 zero (samples + 768);
640 case CONVERT (A52_3F2R, A52_3F1R):
641 zero (samples + 1024);
644 case CONVERT (A52_3F2R, A52_2F1R):
645 zero (samples + 1024);
646 case CONVERT (A52_3F1R, A52_2F1R):
648 memcpy (samples + 768, samples + 512, 256 * sizeof (sample_t));
651 case CONVERT (A52_3F2R, A52_2F2R):
652 memcpy (samples + 1024, samples + 768, 256 * sizeof (sample_t));