1 /*
2 * TAK decoder
3 * Copyright (c) 2012 Paul B Mahol
4 *
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * FFmpeg 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 GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22 /**
23 * @file
24 * TAK (Tom's lossless Audio Kompressor) decoder
25 * @author Paul B Mahol
26 */
27
36
37 #define MAX_SUBFRAMES 8 ///< max number of subframes per channel
38 #define MAX_PREDICTORS 256
39
41 int8_t
present;
///< decorrelation parameter availability for this channel
42 int8_t
index;
///< index into array of decorrelation types
46
52
54 int nb_samples;
///< number of samples in the current frame
58
65
66 int8_t
dmode;
///< channel decorrelation type in the current frame
67
69
74
76
78 4, 8, 12, 16, 24, 32, 48, 64, 80, 96, 128, 160, 192, 224, 256, 0,
79 };
80
88 { 0x01, 0x0000001, 0x0000001, 0x0000003, 0x0000008 },
89 { 0x02, 0x0000003, 0x0000001, 0x0000007, 0x0000006 },
90 { 0x03, 0x0000005, 0x0000002, 0x000000E, 0x000000D },
91 { 0x03, 0x0000003, 0x0000003, 0x000000D, 0x0000018 },
92 { 0x04, 0x000000B, 0x0000004, 0x000001C, 0x0000019 },
93 { 0x04, 0x0000006, 0x0000006, 0x000001A, 0x0000030 },
94 { 0x05, 0x0000016, 0x0000008, 0x0000038, 0x0000032 },
95 { 0x05, 0x000000C, 0x000000C, 0x0000034, 0x0000060 },
96 { 0x06, 0x000002C, 0x0000010, 0x0000070, 0x0000064 },
97 { 0x06, 0x0000018, 0x0000018, 0x0000068, 0x00000C0 },
98 { 0x07, 0x0000058, 0x0000020, 0x00000E0, 0x00000C8 },
99 { 0x07, 0x0000030, 0x0000030, 0x00000D0, 0x0000180 },
100 { 0x08, 0x00000B0, 0x0000040, 0x00001C0, 0x0000190 },
101 { 0x08, 0x0000060, 0x0000060, 0x00001A0, 0x0000300 },
102 { 0x09, 0x0000160, 0x0000080, 0x0000380, 0x0000320 },
103 { 0x09, 0x00000C0, 0x00000C0, 0x0000340, 0x0000600 },
104 { 0x0A, 0x00002C0, 0x0000100, 0x0000700, 0x0000640 },
105 { 0x0A, 0x0000180, 0x0000180, 0x0000680, 0x0000C00 },
106 { 0x0B, 0x0000580, 0x0000200, 0x0000E00, 0x0000C80 },
107 { 0x0B, 0x0000300, 0x0000300, 0x0000D00, 0x0001800 },
108 { 0x0C, 0x0000B00, 0x0000400, 0x0001C00, 0x0001900 },
109 { 0x0C, 0x0000600, 0x0000600, 0x0001A00, 0x0003000 },
110 { 0x0D, 0x0001600, 0x0000800, 0x0003800, 0x0003200 },
111 { 0x0D, 0x0000C00, 0x0000C00, 0x0003400, 0x0006000 },
112 { 0x0E, 0x0002C00, 0x0001000, 0x0007000, 0x0006400 },
113 { 0x0E, 0x0001800, 0x0001800, 0x0006800, 0x000C000 },
114 { 0x0F, 0x0005800, 0x0002000, 0x000E000, 0x000C800 },
115 { 0x0F, 0x0003000, 0x0003000, 0x000D000, 0x0018000 },
116 { 0x10, 0x000B000, 0x0004000, 0x001C000, 0x0019000 },
117 { 0x10, 0x0006000, 0x0006000, 0x001A000, 0x0030000 },
118 { 0x11, 0x0016000, 0x0008000, 0x0038000, 0x0032000 },
119 { 0x11, 0x000C000, 0x000C000, 0x0034000, 0x0060000 },
120 { 0x12, 0x002C000, 0x0010000, 0x0070000, 0x0064000 },
121 { 0x12, 0x0018000, 0x0018000, 0x0068000, 0x00C0000 },
122 { 0x13, 0x0058000, 0x0020000, 0x00E0000, 0x00C8000 },
123 { 0x13, 0x0030000, 0x0030000, 0x00D0000, 0x0180000 },
124 { 0x14, 0x00B0000, 0x0040000, 0x01C0000, 0x0190000 },
125 { 0x14, 0x0060000, 0x0060000, 0x01A0000, 0x0300000 },
126 { 0x15, 0x0160000, 0x0080000, 0x0380000, 0x0320000 },
127 { 0x15, 0x00C0000, 0x00C0000, 0x0340000, 0x0600000 },
128 { 0x16, 0x02C0000, 0x0100000, 0x0700000, 0x0640000 },
129 { 0x16, 0x0180000, 0x0180000, 0x0680000, 0x0C00000 },
130 { 0x17, 0x0580000, 0x0200000, 0x0E00000, 0x0C80000 },
131 { 0x17, 0x0300000, 0x0300000, 0x0D00000, 0x1800000 },
132 { 0x18, 0x0B00000, 0x0400000, 0x1C00000, 0x1900000 },
133 { 0x18, 0x0600000, 0x0600000, 0x1A00000, 0x3000000 },
134 { 0x19, 0x1600000, 0x0800000, 0x3800000, 0x3200000 },
135 { 0x19, 0x0C00000, 0x0C00000, 0x3400000, 0x6000000 },
136 { 0x1A, 0x2C00000, 0x1000000, 0x7000000, 0x6400000 },
137 { 0x1A, 0x1800000, 0x1800000, 0x6800000, 0xC000000 },
138 };
139
141 {
143 case 8:
145 break;
146 case 16:
148 break;
149 case 24:
151 break;
152 default:
156 }
157
158 return 0;
159 }
160
162 {
165 shift =
FFMAX(0, shift);
168 }
169
171 {
173
175
178
180
182 }
183
185 {
186 int i;
187
188 if (length < 2)
189 return;
190
191 if (mode == 1) {
193 for (i = 0; i < length - 1 >> 1; i++) {
195 coeffs[1] += *coeffs;
196 a1 = coeffs[1];
197 coeffs += 2;
198 }
199 if (length - 1 & 1)
201 } else if (mode == 2) {
203 int a2 = a1 + *coeffs;
205 if (length > 2) {
206 coeffs += 2;
207 for (i = 0; i < length - 2 >> 1; i++) {
208 int a3 = *coeffs +
a1;
214 coeffs += 2;
215 }
216 if (length & 1)
218 }
219 } else if (mode == 3) {
221 int a2 = a1 + *coeffs;
223 if (length > 2) {
227 coeffs += 3;
228 for (i = 0; i < length - 3; i++) {
229 a3 += *coeffs;
233 coeffs++;
234 }
235 }
236 }
237 }
238
240 {
243 int i;
244
245 if (!mode) {
246 memset(decoded, 0, len * sizeof(*decoded));
247 return 0;
248 }
249
253
254 for (i = 0; i <
len; i++) {
260 if (scale == 9) {
262 if (scale_bits > 0) {
263 if (scale_bits == 7) {
265 if (scale_bits > 29)
267 }
270 }
272 } else
274 } else
276 }
277 decoded[i] = (x >> 1) ^ -(x & 1);
278 }
279
280 return 0;
281 }
282
284 {
287
290
292 int wlength, rval;
293
294 wlength = length / s->
uval;
295
296 rval = length - (wlength * s->
uval);
297
298 if (rval < s->uval / 2)
300 else
301 wlength++;
302
303 if (wlength <= 1 || wlength > 128)
305
307
308 for (i = 1; i < wlength; i++) {
310
311 switch (c) {
312 case 6:
314 break;
315 case 5:
316 case 4:
317 case 3: {
318 /* mode += sign ? (1 - c) : (c - 1) */
320 mode += (-sign ^ (c - 1)) + sign;
321 break;
322 }
323 case 2:
324 mode++;
325 break;
326 case 1:
327 mode--;
328 break;
329 }
331 }
332
333 i = 0;
334 while (i < wlength) {
336
338 do {
339 if (i >= wlength - 1)
340 len += rval;
341 else
343 i++;
344
345 if (i == wlength)
346 break;
348
352 }
353 } else {
357 }
358
359 return 0;
360 }
361
363 {
366 else
367 return 0;
368 }
369
371 int subframe_size, int prev_subframe_size)
372 {
374 int x,
y, i, j,
ret = 0;
375 int dshift,
size, filter_quant, filter_order;
377
380
382
383 if (prev_subframe_size > 0 &&
get_bits1(gb)) {
384 if (filter_order > prev_subframe_size)
386
387 decoded -= filter_order;
388 subframe_size += filter_order;
389
390 if (filter_order > subframe_size)
392 } else {
393 int lpc_mode;
394
395 if (filter_order > subframe_size)
397
399 if (lpc_mode > 2)
401
404
405 if (lpc_mode)
407 }
408
411
412 filter_quant = 10;
414 filter_quant -=
get_bits(gb, 3) + 1;
415 if (filter_quant < 3)
417 }
418
423 if (filter_order > 4) {
425
426 for (i = 4; i < filter_order; i++) {
427 if (!(i & 3))
430 }
431 }
432
434 for (i = 1; i < filter_order; i++) {
437
438 for (j = 0; j < (i + 1) / 2; j++) {
439 x = *p1 + (s->
predictors[i] * *p2 + 256 >> 9);
441 *p1++ = x;
442 p2--;
443 }
444
446 }
447
448 x = 1 << (32 - (15 - filter_quant));
449 y = 1 << ((15 - filter_quant) - 1);
450 for (i = 0, j = filter_order - 1; i < filter_order / 2; i++, j--) {
451 s->
filter[j] = x - ((tfilter[i] +
y) >> (15 - filter_quant));
452 s->
filter[i] = x - ((tfilter[j] +
y) >> (15 - filter_quant));
453 }
454
456 subframe_size - filter_order)) < 0)
458
459 for (i = 0; i < filter_order; i++)
460 s->
residues[i] = *decoded++ >> dshift;
461
463 x = subframe_size - filter_order;
464 while (x > 0) {
465 int tmp =
FFMIN(y, x);
466
467 for (i = 0; i < tmp; i++) {
468 int v = 1 << (filter_quant - 1);
469
470 if (filter_order & -16)
472 filter_order & -16);
473 for (j = filter_order & -16; j < filter_order; j += 4) {
478 }
479 v = (av_clip_intp2(v >> filter_quant, 13) << dshift) - *decoded;
481 s->
residues[filter_order + i] = v >> dshift;
482 }
483
484 x -= tmp;
485 if (x > 0)
487 }
488
489 emms_c();
490
491 return 0;
492 }
493
495 {
500 int i = 0,
ret, prev = 0;
501
505
509
513
516
520
523 }
524
525 if (left <= 0)
527 }
529
530 prev = 0;
536 }
537
538 return 0;
539 }
540
542 {
546 int i;
547 int dshift, dfactor;
548
550 case 1: /* left/side */
551 for (i = 0; i <
length; i++) {
555 }
556 break;
557 case 2: /* side/right */
558 for (i = 0; i <
length; i++) {
562 }
563 break;
564 case 3: /* side/mid */
565 for (i = 0; i <
length; i++) {
568 a -= b >> 1;
571 }
572 break;
573 case 4: /* side/left with scale factor */
575 case 5: /* side/right with scale factor */
578 for (i = 0; i <
length; i++) {
581 b = dfactor * (b >> dshift) + 128 >> 8 << dshift;
583 }
584 break;
585 case 6:
587 case 7: {
588 int length2, order_half, filter_order, dval1, dval2;
589 int tmp, x, code_size;
590
591 if (length < 256)
593
598
599 for (i = 0; i < filter_order; i++) {
600 if (!(i & 3))
603 }
604
605 order_half = filter_order / 2;
606 length2 = length - (filter_order - 1);
607
608 /* decorrelate beginning samples */
609 if (dval1) {
610 for (i = 0; i < order_half; i++) {
614 }
615 }
616
617 /* decorrelate ending samples */
618 if (dval2) {
619 for (i = length2 + order_half; i <
length; i++) {
623 }
624 }
625
626
627 for (i = 0; i < filter_order; i++)
629
630 p1 += order_half;
632 for (; length2 > 0; length2 -= tmp) {
633 tmp =
FFMIN(length2, x);
634
635 for (i = 0; i < tmp; i++)
636 s->
residues[filter_order + i] = *p2++ >> dshift;
637
638 for (i = 0; i < tmp; i++) {
640
641 if (filter_order == 16) {
643 filter_order);
644 } else {
653 }
654
655 v = (av_clip_intp2(v >> 10, 13) << dshift) - *p1;
657 }
658
660 }
661
662 emms_c();
663 break;
664 }
665 }
666
667 return 0;
668 }
669
672 {
677 int chan, i,
ret, hsize;
678
681
684
687
694 }
695 }
696
701 }
706 }
709 "invalid number of channels: %d\n", s->
ti.
channels);
711 }
714 "unsupported number of channels: %d\n", s->
ti.
channels);
716 }
717
721 }
722
729 }
733
736
741
746 if (buf_size < 0)
747 return buf_size;
754 if (ret < 0)
756 } else {
757 for (chan = 0; chan < avctx->
channels; chan++)
759 }
760
762 for (chan = 0; chan < avctx->
channels; chan++) {
766 }
767 } else {
769 for (chan = 0; chan < avctx->
channels; chan++)
772
777 }
778
782 }
785 int ch_mask = 0;
786
790
791 for (i = 0; i < chan; i++) {
793
796
797 if (ch_mask & 1 << nbit)
799
808
812 }
813 }
815
816 ch_mask |= 1 << nbit;
817 }
818 } else {
820 for (i = 0; i < chan; i++) {
823 }
824 }
825
826 for (i = 0; i < chan; i++) {
830
833
841 }
842 }
843 }
844
845 for (chan = 0; chan < avctx->
channels; chan++) {
847
850
854 }
855 }
856
863
870 }
871 }
872
873 /* convert to output buffer */
876 for (chan = 0; chan < avctx->
channels; chan++) {
880 samples[i] = decoded[i] + 0x80;
881 }
882 break;
884 for (chan = 0; chan < avctx->
channels; chan++) {
888 samples[i] = decoded[i];
889 }
890 break;
892 for (chan = 0; chan < avctx->
channels; chan++) {
895 samples[i] <<= 8;
896 }
897 break;
898 }
899
900 *got_frame_ptr = 1;
901
903 }
904
906 {
909 return 0;
910 }
911
914 {
917
918 if (dst == src)
919 return 0;
921 return 0;
922 }
923
925 {
927
929
930 return 0;
931 }
932
949 };
static int set_bps_params(AVCodecContext *avctx)
static const uint16_t predictor_sizes[]
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
int ff_tak_decode_frame_header(AVCodecContext *avctx, GetBitContext *gb, TAKStreamInfo *ti, int log_level_offset)
Validate and decode a frame header.
int32_t(* scalarproduct_int16)(const int16_t *v1, const int16_t *v2, int len)
Calculate scalar product of two vectors.
static int shift(int a, int b)
int16_t subframe_len[MAX_SUBFRAMES]
subframe length in samples
This structure describes decoded (raw) audio or video data.
ptrdiff_t const GLvoid * data
static unsigned int get_bits(GetBitContext *s, int n)
Read 1-25 bits.
static av_cold int init(AVCodecContext *avctx)
MCDParam mcdparams[TAK_MAX_CHANNELS]
multichannel decorrelation parameters
int16_t filter[MAX_PREDICTORS]
static int decode_subframe(TAKDecContext *s, int32_t *decoded, int subframe_size, int prev_subframe_size)
#define DECLARE_ALIGNED(n, t, v)
av_cold void ff_audiodsp_init(AudioDSPContext *c)
#define FF_ARRAY_ELEMS(a)
int bits_per_raw_sample
Bits per sample/pixel of internal libavcodec pixel/sample format.
static int decode_segment(TAKDecContext *s, int8_t mode, int32_t *decoded, int len)
static int get_sbits(GetBitContext *s, int n)
int8_t lpc_mode[TAK_MAX_CHANNELS]
unsigned int decode_buffer_size
enum AVSampleFormat sample_fmt
audio sample format
static av_cold int tak_decode_close(AVCodecContext *avctx)
Multithreading support functions.
int ff_tak_check_crc(const uint8_t *buf, unsigned int buf_size)
static int init_thread_copy(AVCodecContext *avctx)
#define CODEC_CAP_DR1
Codec uses get_buffer() for allocating buffers and supports custom allocators.
AVCodecContext * avctx
parent AVCodecContext
static int get_bits_count(const GetBitContext *s)
int bits_per_coded_sample
bits per sample/pixel from the demuxer (needed for huffyuv).
void ff_thread_finish_setup(AVCodecContext *avctx)
If the codec defines update_thread_context(), call this when they are ready for the next thread to st...
static int decode_channel(TAKDecContext *s, int chan)
#define MAX_SUBFRAMES
max number of subframes per channel
int8_t index
index into array of decorrelation types
static int get_bits_left(GetBitContext *gb)
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
#define AV_EF_EXPLODE
abort decoding on minor error detection
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
#define AV_LOG_DEBUG
Stuff which is only useful for libav* developers.
const char * name
Name of the codec implementation.
int8_t present
decorrelation parameter availability for this channel
static int decode_residues(TAKDecContext *s, int32_t *decoded, int length)
Libavcodec external API header.
uint64_t channel_layout
Audio channel layout.
#define ONLY_IF_THREADS_ENABLED(x)
Define a function with only the non-default version specified.
common internal API header
static int update_thread_context(AVCodecContext *dst, const AVCodecContext *src)
static av_cold int tak_decode_init(AVCodecContext *avctx)
int err_recognition
Error recognition; may misdetect some more or less valid parts as errors.
static const struct CParam xcodes[50]
static int get_bits_esc4(GetBitContext *gb)
static int decorrelate(TAKDecContext *s, int c1, int c2, int length)
static void decode_lpc(int32_t *coeffs, int mode, int length)
#define AVERROR_PATCHWELCOME
Not yet implemented in FFmpeg, patches welcome.
AVSampleFormat
Audio sample formats.
#define TAK_MIN_FRAME_HEADER_BYTES
int sample_rate
samples per second
int16_t predictors[MAX_PREDICTORS]
static int init_get_bits8(GetBitContext *s, const uint8_t *buffer, int byte_size)
Initialize GetBitContext.
int ff_thread_get_buffer(AVCodecContext *avctx, ThreadFrame *f, int flags)
Wrapper around get_buffer() for frame-multithreaded codecs.
main external API structure.
int32_t * decoded[TAK_MAX_CHANNELS]
decoded samples for each channel
static unsigned int get_bits1(GetBitContext *s)
static void skip_bits(GetBitContext *s, int n)
int av_samples_get_buffer_size(int *linesize, int nb_channels, int nb_samples, enum AVSampleFormat sample_fmt, int align)
Get the required buffer size for the given audio parameters.
TAK (Tom's lossless Audio Kompressor) decoder/demuxer common functions.
GetBitContext gb
bitstream reader initialized to start at the current frame
int8_t dmode
channel decorrelation type in the current frame
static unsigned int get_bits_long(GetBitContext *s, int n)
Read 0-32 bits.
static int decode(AVCodecContext *avctx, void *data, int *got_sub, AVPacket *avpkt)
int nb_samples
number of samples in the current frame
void av_fast_malloc(void *ptr, unsigned int *size, size_t min_size)
Allocate a buffer, reusing the given one if large enough.
common internal api header.
#define CODEC_CAP_FRAME_THREADS
Codec supports frame-level multithreading.
static int get_unary(GetBitContext *gb, int stop, int len)
Get unary code of limited length.
static const int8_t mc_dmodes[]
#define AV_EF_CRCCHECK
Verify checksums embedded in the bitstream (could be of either encoded or decoded data...
int nb_subframes
number of subframes in the current frame
int av_samples_fill_arrays(uint8_t **audio_data, int *linesize, const uint8_t *buf, int nb_channels, int nb_samples, enum AVSampleFormat sample_fmt, int align)
Fill plane data pointers and linesize for samples with sample format sample_fmt.
static int tak_decode_frame(AVCodecContext *avctx, void *data, int *got_frame_ptr, AVPacket *pkt)
int channels
number of audio channels
#define AV_EF_COMPLIANT
consider all spec non compliances as errors
static const uint8_t * align_get_bits(GetBitContext *s)
static void set_sample_rate_params(AVCodecContext *avctx)
int8_t sample_shift[TAK_MAX_CHANNELS]
shift applied to every sample in the channel
#define FFSWAP(type, a, b)
uint8_t ** extended_data
pointers to the data planes/channels.
This structure stores compressed data.
int nb_samples
number of audio samples (per channel) described by this frame