cook: use planar sample format
[libav.git] / libavcodec / dcadec.c
CommitLineData
01ca9ac3
KS
1/*
2 * DCA compatible decoder
3 * Copyright (C) 2004 Gildas Bazin
4 * Copyright (C) 2004 Benjamin Zores
5 * Copyright (C) 2006 Benjamin Larsson
6 * Copyright (C) 2007 Konstantin Shishkov
7 *
2912e87a 8 * This file is part of Libav.
01ca9ac3 9 *
2912e87a 10 * Libav is free software; you can redistribute it and/or
01ca9ac3
KS
11 * modify it under the terms of the GNU Lesser General Public
12 * License as published by the Free Software Foundation; either
13 * version 2.1 of the License, or (at your option) any later version.
14 *
2912e87a 15 * Libav is distributed in the hope that it will be useful,
01ca9ac3
KS
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18 * Lesser General Public License for more details.
19 *
20 * You should have received a copy of the GNU Lesser General Public
2912e87a 21 * License along with Libav; if not, write to the Free Software
01ca9ac3
KS
22 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
23 */
24
01ca9ac3
KS
25#include <math.h>
26#include <stddef.h>
27#include <stdio.h>
28
f5a2d285 29#include "libavutil/common.h"
cb5042d0 30#include "libavutil/float_dsp.h"
843c7aa8 31#include "libavutil/intmath.h"
0dc7df28 32#include "libavutil/intreadwrite.h"
681e7268 33#include "libavutil/mathematics.h"
737eb597 34#include "libavutil/audioconvert.h"
01ca9ac3
KS
35#include "avcodec.h"
36#include "dsputil.h"
1429224b 37#include "fft.h"
9106a698 38#include "get_bits.h"
b2755007 39#include "put_bits.h"
01ca9ac3
KS
40#include "dcadata.h"
41#include "dcahuff.h"
98c98e04 42#include "dca.h"
11ac796f 43#include "dca_parser.h"
4f99c31c 44#include "synth_filter.h"
309d16a4 45#include "dcadsp.h"
c73d99e6 46#include "fmtconvert.h"
01ca9ac3 47
bf5d46d8
MR
48#if ARCH_ARM
49# include "arm/dca.h"
50#endif
51
01ca9ac3
KS
52//#define TRACE
53
f37b4efe
SG
54#define DCA_PRIM_CHANNELS_MAX (7)
55#define DCA_SUBBANDS (32)
56#define DCA_ABITS_MAX (32) /* Should be 28 */
57#define DCA_SUBSUBFRAMES_MAX (4)
58#define DCA_SUBFRAMES_MAX (16)
59#define DCA_BLOCKS_MAX (16)
60#define DCA_LFE_MAX (3)
01ca9ac3
KS
61
62enum DCAMode {
63 DCA_MONO = 0,
64 DCA_CHANNEL,
65 DCA_STEREO,
66 DCA_STEREO_SUMDIFF,
67 DCA_STEREO_TOTAL,
68 DCA_3F,
69 DCA_2F1R,
70 DCA_3F1R,
71 DCA_2F2R,
72 DCA_3F2R,
73 DCA_4F2R
74};
75
f5a2d285
AH
76/* these are unconfirmed but should be mostly correct */
77enum DCAExSSSpeakerMask {
78 DCA_EXSS_FRONT_CENTER = 0x0001,
79 DCA_EXSS_FRONT_LEFT_RIGHT = 0x0002,
80 DCA_EXSS_SIDE_REAR_LEFT_RIGHT = 0x0004,
81 DCA_EXSS_LFE = 0x0008,
82 DCA_EXSS_REAR_CENTER = 0x0010,
83 DCA_EXSS_FRONT_HIGH_LEFT_RIGHT = 0x0020,
84 DCA_EXSS_REAR_LEFT_RIGHT = 0x0040,
85 DCA_EXSS_FRONT_HIGH_CENTER = 0x0080,
86 DCA_EXSS_OVERHEAD = 0x0100,
87 DCA_EXSS_CENTER_LEFT_RIGHT = 0x0200,
88 DCA_EXSS_WIDE_LEFT_RIGHT = 0x0400,
89 DCA_EXSS_SIDE_LEFT_RIGHT = 0x0800,
90 DCA_EXSS_LFE2 = 0x1000,
91 DCA_EXSS_SIDE_HIGH_LEFT_RIGHT = 0x2000,
92 DCA_EXSS_REAR_HIGH_CENTER = 0x4000,
93 DCA_EXSS_REAR_HIGH_LEFT_RIGHT = 0x8000,
94};
95
96enum DCAExtensionMask {
97 DCA_EXT_CORE = 0x001, ///< core in core substream
98 DCA_EXT_XXCH = 0x002, ///< XXCh channels extension in core substream
99 DCA_EXT_X96 = 0x004, ///< 96/24 extension in core substream
100 DCA_EXT_XCH = 0x008, ///< XCh channel extension in core substream
101 DCA_EXT_EXSS_CORE = 0x010, ///< core in ExSS (extension substream)
102 DCA_EXT_EXSS_XBR = 0x020, ///< extended bitrate extension in ExSS
103 DCA_EXT_EXSS_XXCH = 0x040, ///< XXCh channels extension in ExSS
104 DCA_EXT_EXSS_X96 = 0x080, ///< 96/24 extension in ExSS
105 DCA_EXT_EXSS_LBR = 0x100, ///< low bitrate component in ExSS
106 DCA_EXT_EXSS_XLL = 0x200, ///< lossless extension in ExSS
107};
108
7e06e0ed
AH
109/* -1 are reserved or unknown */
110static const int dca_ext_audio_descr_mask[] = {
111 DCA_EXT_XCH,
112 -1,
113 DCA_EXT_X96,
114 DCA_EXT_XCH | DCA_EXT_X96,
115 -1,
116 -1,
117 DCA_EXT_XXCH,
118 -1,
119};
120
121/* extensions that reside in core substream */
122#define DCA_CORE_EXTS (DCA_EXT_XCH | DCA_EXT_XXCH | DCA_EXT_X96)
123
87c3b9bc
BL
124/* Tables for mapping dts channel configurations to libavcodec multichannel api.
125 * Some compromises have been made for special configurations. Most configurations
126 * are never used so complete accuracy is not needed.
127 *
128 * L = left, R = right, C = center, S = surround, F = front, R = rear, T = total, OV = overhead.
e22192ec 129 * S -> side, when both rear and back are configured move one of them to the side channel
87c3b9bc 130 * OV -> center back
c2fcd0a7 131 * All 2 channel configurations -> AV_CH_LAYOUT_STEREO
87c3b9bc 132 */
cc276c85 133static const uint64_t dca_core_channel_layout[] = {
f37b4efe
SG
134 AV_CH_FRONT_CENTER, ///< 1, A
135 AV_CH_LAYOUT_STEREO, ///< 2, A + B (dual mono)
136 AV_CH_LAYOUT_STEREO, ///< 2, L + R (stereo)
137 AV_CH_LAYOUT_STEREO, ///< 2, (L + R) + (L - R) (sum-difference)
138 AV_CH_LAYOUT_STEREO, ///< 2, LT + RT (left and right total)
139 AV_CH_LAYOUT_STEREO | AV_CH_FRONT_CENTER, ///< 3, C + L + R
140 AV_CH_LAYOUT_STEREO | AV_CH_BACK_CENTER, ///< 3, L + R + S
141 AV_CH_LAYOUT_STEREO | AV_CH_FRONT_CENTER | AV_CH_BACK_CENTER, ///< 4, C + L + R + S
142 AV_CH_LAYOUT_STEREO | AV_CH_SIDE_LEFT | AV_CH_SIDE_RIGHT, ///< 4, L + R + SL + SR
143
144 AV_CH_LAYOUT_STEREO | AV_CH_FRONT_CENTER | AV_CH_SIDE_LEFT |
145 AV_CH_SIDE_RIGHT, ///< 5, C + L + R + SL + SR
146
147 AV_CH_LAYOUT_STEREO | AV_CH_SIDE_LEFT | AV_CH_SIDE_RIGHT |
148 AV_CH_FRONT_LEFT_OF_CENTER | AV_CH_FRONT_RIGHT_OF_CENTER, ///< 6, CL + CR + L + R + SL + SR
149
150 AV_CH_LAYOUT_STEREO | AV_CH_BACK_LEFT | AV_CH_BACK_RIGHT |
151 AV_CH_FRONT_CENTER | AV_CH_BACK_CENTER, ///< 6, C + L + R + LR + RR + OV
152
153 AV_CH_FRONT_CENTER | AV_CH_FRONT_RIGHT_OF_CENTER |
154 AV_CH_FRONT_LEFT_OF_CENTER | AV_CH_BACK_CENTER |
155 AV_CH_BACK_LEFT | AV_CH_BACK_RIGHT, ///< 6, CF + CR + LF + RF + LR + RR
156
157 AV_CH_FRONT_LEFT_OF_CENTER | AV_CH_FRONT_CENTER |
158 AV_CH_FRONT_RIGHT_OF_CENTER | AV_CH_LAYOUT_STEREO |
159 AV_CH_SIDE_LEFT | AV_CH_SIDE_RIGHT, ///< 7, CL + C + CR + L + R + SL + SR
160
161 AV_CH_FRONT_LEFT_OF_CENTER | AV_CH_FRONT_RIGHT_OF_CENTER |
162 AV_CH_LAYOUT_STEREO | AV_CH_SIDE_LEFT | AV_CH_SIDE_RIGHT |
163 AV_CH_BACK_LEFT | AV_CH_BACK_RIGHT, ///< 8, CL + CR + L + R + SL1 + SL2 + SR1 + SR2
164
165 AV_CH_FRONT_LEFT_OF_CENTER | AV_CH_FRONT_CENTER |
166 AV_CH_FRONT_RIGHT_OF_CENTER | AV_CH_LAYOUT_STEREO |
167 AV_CH_SIDE_LEFT | AV_CH_BACK_CENTER | AV_CH_SIDE_RIGHT, ///< 8, CL + C + CR + L + R + SL + S + SR
87c3b9bc
BL
168};
169
92765276 170static const int8_t dca_lfe_index[] = {
f37b4efe 171 1, 2, 2, 2, 2, 3, 2, 3, 2, 3, 2, 3, 1, 3, 2, 3
92765276
BL
172};
173
3d5a9ba1
NB
174static const int8_t dca_channel_reorder_lfe[][9] = {
175 { 0, -1, -1, -1, -1, -1, -1, -1, -1},
176 { 0, 1, -1, -1, -1, -1, -1, -1, -1},
177 { 0, 1, -1, -1, -1, -1, -1, -1, -1},
178 { 0, 1, -1, -1, -1, -1, -1, -1, -1},
179 { 0, 1, -1, -1, -1, -1, -1, -1, -1},
180 { 2, 0, 1, -1, -1, -1, -1, -1, -1},
181 { 0, 1, 3, -1, -1, -1, -1, -1, -1},
182 { 2, 0, 1, 4, -1, -1, -1, -1, -1},
183 { 0, 1, 3, 4, -1, -1, -1, -1, -1},
184 { 2, 0, 1, 4, 5, -1, -1, -1, -1},
185 { 3, 4, 0, 1, 5, 6, -1, -1, -1},
186 { 2, 0, 1, 4, 5, 6, -1, -1, -1},
187 { 0, 6, 4, 5, 2, 3, -1, -1, -1},
188 { 4, 2, 5, 0, 1, 6, 7, -1, -1},
189 { 5, 6, 0, 1, 7, 3, 8, 4, -1},
190 { 4, 2, 5, 0, 1, 6, 8, 7, -1},
92765276
BL
191};
192
3d5a9ba1
NB
193static const int8_t dca_channel_reorder_lfe_xch[][9] = {
194 { 0, 2, -1, -1, -1, -1, -1, -1, -1},
195 { 0, 1, 3, -1, -1, -1, -1, -1, -1},
196 { 0, 1, 3, -1, -1, -1, -1, -1, -1},
197 { 0, 1, 3, -1, -1, -1, -1, -1, -1},
198 { 0, 1, 3, -1, -1, -1, -1, -1, -1},
199 { 2, 0, 1, 4, -1, -1, -1, -1, -1},
200 { 0, 1, 3, 4, -1, -1, -1, -1, -1},
201 { 2, 0, 1, 4, 5, -1, -1, -1, -1},
202 { 0, 1, 4, 5, 3, -1, -1, -1, -1},
203 { 2, 0, 1, 5, 6, 4, -1, -1, -1},
204 { 3, 4, 0, 1, 6, 7, 5, -1, -1},
205 { 2, 0, 1, 4, 5, 6, 7, -1, -1},
206 { 0, 6, 4, 5, 2, 3, 7, -1, -1},
207 { 4, 2, 5, 0, 1, 7, 8, 6, -1},
208 { 5, 6, 0, 1, 8, 3, 9, 4, 7},
209 { 4, 2, 5, 0, 1, 6, 9, 8, 7},
92765276
BL
210};
211
3d5a9ba1
NB
212static const int8_t dca_channel_reorder_nolfe[][9] = {
213 { 0, -1, -1, -1, -1, -1, -1, -1, -1},
214 { 0, 1, -1, -1, -1, -1, -1, -1, -1},
215 { 0, 1, -1, -1, -1, -1, -1, -1, -1},
216 { 0, 1, -1, -1, -1, -1, -1, -1, -1},
217 { 0, 1, -1, -1, -1, -1, -1, -1, -1},
218 { 2, 0, 1, -1, -1, -1, -1, -1, -1},
219 { 0, 1, 2, -1, -1, -1, -1, -1, -1},
220 { 2, 0, 1, 3, -1, -1, -1, -1, -1},
221 { 0, 1, 2, 3, -1, -1, -1, -1, -1},
222 { 2, 0, 1, 3, 4, -1, -1, -1, -1},
223 { 2, 3, 0, 1, 4, 5, -1, -1, -1},
224 { 2, 0, 1, 3, 4, 5, -1, -1, -1},
225 { 0, 5, 3, 4, 1, 2, -1, -1, -1},
226 { 3, 2, 4, 0, 1, 5, 6, -1, -1},
227 { 4, 5, 0, 1, 6, 2, 7, 3, -1},
228 { 3, 2, 4, 0, 1, 5, 7, 6, -1},
229};
230
231static const int8_t dca_channel_reorder_nolfe_xch[][9] = {
232 { 0, 1, -1, -1, -1, -1, -1, -1, -1},
233 { 0, 1, 2, -1, -1, -1, -1, -1, -1},
234 { 0, 1, 2, -1, -1, -1, -1, -1, -1},
235 { 0, 1, 2, -1, -1, -1, -1, -1, -1},
236 { 0, 1, 2, -1, -1, -1, -1, -1, -1},
237 { 2, 0, 1, 3, -1, -1, -1, -1, -1},
238 { 0, 1, 2, 3, -1, -1, -1, -1, -1},
239 { 2, 0, 1, 3, 4, -1, -1, -1, -1},
240 { 0, 1, 3, 4, 2, -1, -1, -1, -1},
241 { 2, 0, 1, 4, 5, 3, -1, -1, -1},
242 { 2, 3, 0, 1, 5, 6, 4, -1, -1},
243 { 2, 0, 1, 3, 4, 5, 6, -1, -1},
244 { 0, 5, 3, 4, 1, 2, 6, -1, -1},
245 { 3, 2, 4, 0, 1, 6, 7, 5, -1},
246 { 4, 5, 0, 1, 7, 2, 8, 3, 6},
247 { 3, 2, 4, 0, 1, 5, 8, 7, 6},
248};
87c3b9bc 249
f37b4efe 250#define DCA_DOLBY 101 /* FIXME */
01ca9ac3 251
f37b4efe
SG
252#define DCA_CHANNEL_BITS 6
253#define DCA_CHANNEL_MASK 0x3F
01ca9ac3 254
f37b4efe 255#define DCA_LFE 0x80
01ca9ac3 256
f37b4efe 257#define HEADER_SIZE 14
01ca9ac3 258
f37b4efe
SG
259#define DCA_MAX_FRAME_SIZE 16384
260#define DCA_MAX_EXSS_HEADER_SIZE 4096
01ca9ac3 261
f37b4efe 262#define DCA_BUFFER_PADDING_SIZE 1024
39f4d329 263
01ca9ac3
KS
264/** Bit allocation */
265typedef struct {
266 int offset; ///< code values offset
267 int maxbits[8]; ///< max bits in VLC
268 int wrap; ///< wrap for get_vlc2()
269 VLC vlc[8]; ///< actual codes
270} BitAlloc;
271
272static BitAlloc dca_bitalloc_index; ///< indexes for samples VLC select
273static BitAlloc dca_tmode; ///< transition mode VLCs
274static BitAlloc dca_scalefactor; ///< scalefactor VLCs
275static BitAlloc dca_smpl_bitalloc[11]; ///< samples VLCs
276
f37b4efe
SG
277static av_always_inline int get_bitalloc(GetBitContext *gb, BitAlloc *ba,
278 int idx)
01ca9ac3 279{
f37b4efe
SG
280 return get_vlc2(gb, ba->vlc[idx].table, ba->vlc[idx].bits, ba->wrap) +
281 ba->offset;
01ca9ac3
KS
282}
283
284typedef struct {
285 AVCodecContext *avctx;
0eea2129 286 AVFrame frame;
01ca9ac3
KS
287 /* Frame header */
288 int frame_type; ///< type of the current frame
289 int samples_deficit; ///< deficit sample count
290 int crc_present; ///< crc is present in the bitstream
291 int sample_blocks; ///< number of PCM sample blocks
292 int frame_size; ///< primary frame byte size
293 int amode; ///< audio channels arrangement
294 int sample_rate; ///< audio sampling rate
295 int bit_rate; ///< transmission bit rate
9ed73b48 296 int bit_rate_index; ///< transmission bit rate index
01ca9ac3
KS
297
298 int downmix; ///< embedded downmix enabled
299 int dynrange; ///< embedded dynamic range flag
300 int timestamp; ///< embedded time stamp flag
301 int aux_data; ///< auxiliary data flag
302 int hdcd; ///< source material is mastered in HDCD
303 int ext_descr; ///< extension audio descriptor flag
304 int ext_coding; ///< extended coding flag
305 int aspf; ///< audio sync word insertion flag
306 int lfe; ///< low frequency effects flag
307 int predictor_history; ///< predictor history flag
308 int header_crc; ///< header crc check bytes
309 int multirate_inter; ///< multirate interpolator switch
310 int version; ///< encoder software revision
311 int copy_history; ///< copy history
312 int source_pcm_res; ///< source pcm resolution
313 int front_sum; ///< front sum/difference flag
314 int surround_sum; ///< surround sum/difference flag
315 int dialog_norm; ///< dialog normalisation parameter
316
317 /* Primary audio coding header */
318 int subframes; ///< number of subframes
1360f07e 319 int is_channels_set; ///< check for if the channel number is already set
ebf71dbd 320 int total_channels; ///< number of channels including extensions
01ca9ac3
KS
321 int prim_channels; ///< number of primary audio channels
322 int subband_activity[DCA_PRIM_CHANNELS_MAX]; ///< subband activity count
323 int vq_start_subband[DCA_PRIM_CHANNELS_MAX]; ///< high frequency vq start subband
324 int joint_intensity[DCA_PRIM_CHANNELS_MAX]; ///< joint intensity coding index
325 int transient_huffman[DCA_PRIM_CHANNELS_MAX]; ///< transient mode code book
326 int scalefactor_huffman[DCA_PRIM_CHANNELS_MAX]; ///< scale factor code book
327 int bitalloc_huffman[DCA_PRIM_CHANNELS_MAX]; ///< bit allocation quantizer select
328 int quant_index_huffman[DCA_PRIM_CHANNELS_MAX][DCA_ABITS_MAX]; ///< quantization index codebook select
329 float scalefactor_adj[DCA_PRIM_CHANNELS_MAX][DCA_ABITS_MAX]; ///< scale factor adjustment
330
331 /* Primary audio coding side information */
f37b4efe
SG
332 int subsubframes[DCA_SUBFRAMES_MAX]; ///< number of subsubframes
333 int partial_samples[DCA_SUBFRAMES_MAX]; ///< partial subsubframe samples count
01ca9ac3
KS
334 int prediction_mode[DCA_PRIM_CHANNELS_MAX][DCA_SUBBANDS]; ///< prediction mode (ADPCM used or not)
335 int prediction_vq[DCA_PRIM_CHANNELS_MAX][DCA_SUBBANDS]; ///< prediction VQ coefs
336 int bitalloc[DCA_PRIM_CHANNELS_MAX][DCA_SUBBANDS]; ///< bit allocation index
337 int transition_mode[DCA_PRIM_CHANNELS_MAX][DCA_SUBBANDS]; ///< transition mode (transients)
338 int scale_factor[DCA_PRIM_CHANNELS_MAX][DCA_SUBBANDS][2]; ///< scale factors (2 if transient)
339 int joint_huff[DCA_PRIM_CHANNELS_MAX]; ///< joint subband scale factors codebook
340 int joint_scale_factor[DCA_PRIM_CHANNELS_MAX][DCA_SUBBANDS]; ///< joint subband scale factors
341 int downmix_coef[DCA_PRIM_CHANNELS_MAX][2]; ///< stereo downmix coefficients
342 int dynrange_coef; ///< dynamic range coefficient
343
344 int high_freq_vq[DCA_PRIM_CHANNELS_MAX][DCA_SUBBANDS]; ///< VQ encoded high frequency subbands
345
6baef06e 346 float lfe_data[2 * DCA_LFE_MAX * (DCA_BLOCKS_MAX + 4)]; ///< Low frequency effect data
01ca9ac3
KS
347 int lfe_scale_factor;
348
349 /* Subband samples history (for ADPCM) */
bf5d46d8 350 DECLARE_ALIGNED(16, float, subband_samples_hist)[DCA_PRIM_CHANNELS_MAX][DCA_SUBBANDS][4];
9d35fa52
VS
351 DECLARE_ALIGNED(32, float, subband_fir_hist)[DCA_PRIM_CHANNELS_MAX][512];
352 DECLARE_ALIGNED(32, float, subband_fir_noidea)[DCA_PRIM_CHANNELS_MAX][32];
47f0e052 353 int hist_index[DCA_PRIM_CHANNELS_MAX];
9d35fa52 354 DECLARE_ALIGNED(32, float, raXin)[32];
01ca9ac3
KS
355
356 int output; ///< type of output
1402ee72 357 float scale_bias; ///< output scale
01ca9ac3 358
9d35fa52 359 DECLARE_ALIGNED(32, float, subband_samples)[DCA_BLOCKS_MAX][DCA_PRIM_CHANNELS_MAX][DCA_SUBBANDS][8];
f37b4efe
SG
360 DECLARE_ALIGNED(32, float, samples)[(DCA_PRIM_CHANNELS_MAX + 1) * 256];
361 const float *samples_chanptr[DCA_PRIM_CHANNELS_MAX + 1];
01ca9ac3 362
39f4d329 363 uint8_t dca_buffer[DCA_MAX_FRAME_SIZE + DCA_MAX_EXSS_HEADER_SIZE + DCA_BUFFER_PADDING_SIZE];
01ca9ac3
KS
364 int dca_buffer_size; ///< how much data is in the dca_buffer
365
f37b4efe 366 const int8_t *channel_order_tab; ///< channel reordering table, lfe and non lfe
01ca9ac3
KS
367 GetBitContext gb;
368 /* Current position in DCA frame */
369 int current_subframe;
370 int current_subsubframe;
371
7e06e0ed
AH
372 int core_ext_mask; ///< present extensions in the core substream
373
d0a18850 374 /* XCh extension information */
7e06e0ed 375 int xch_present; ///< XCh extension present and valid
d0a18850
NB
376 int xch_base_channel; ///< index of first (only) channel containing XCH data
377
f5a2d285
AH
378 /* ExSS header parser */
379 int static_fields; ///< static fields present
380 int mix_metadata; ///< mixing metadata present
381 int num_mix_configs; ///< number of mix out configurations
382 int mix_config_num_ch[4]; ///< number of channels in each mix out configuration
383
384 int profile;
385
01ca9ac3 386 int debug_flag; ///< used for suppressing repeated error messages output
cb5042d0 387 AVFloatDSPContext fdsp;
01b22147 388 FFTContext imdct;
f462ed1f 389 SynthFilterContext synth;
309d16a4 390 DCADSPContext dcadsp;
c73d99e6 391 FmtConvertContext fmt_conv;
01ca9ac3
KS
392} DCAContext;
393
0cfa85dd
KS
394static const uint16_t dca_vlc_offs[] = {
395 0, 512, 640, 768, 1282, 1794, 2436, 3080, 3770, 4454, 5364,
396 5372, 5380, 5388, 5392, 5396, 5412, 5420, 5428, 5460, 5492, 5508,
397 5572, 5604, 5668, 5796, 5860, 5892, 6412, 6668, 6796, 7308, 7564,
398 7820, 8076, 8620, 9132, 9388, 9910, 10166, 10680, 11196, 11726, 12240,
399 12752, 13298, 13810, 14326, 14840, 15500, 16022, 16540, 17158, 17678, 18264,
400 18796, 19352, 19926, 20468, 21472, 22398, 23014, 23622,
401};
402
98a6fff9 403static av_cold void dca_init_vlcs(void)
01ca9ac3 404{
5e534865 405 static int vlcs_initialized = 0;
0cfa85dd
KS
406 int i, j, c = 14;
407 static VLC_TYPE dca_table[23622][2];
01ca9ac3 408
5e534865 409 if (vlcs_initialized)
01ca9ac3
KS
410 return;
411
412 dca_bitalloc_index.offset = 1;
32dd6a9c 413 dca_bitalloc_index.wrap = 2;
0cfa85dd
KS
414 for (i = 0; i < 5; i++) {
415 dca_bitalloc_index.vlc[i].table = &dca_table[dca_vlc_offs[i]];
416 dca_bitalloc_index.vlc[i].table_allocated = dca_vlc_offs[i + 1] - dca_vlc_offs[i];
01ca9ac3
KS
417 init_vlc(&dca_bitalloc_index.vlc[i], bitalloc_12_vlc_bits[i], 12,
418 bitalloc_12_bits[i], 1, 1,
0cfa85dd
KS
419 bitalloc_12_codes[i], 2, 2, INIT_VLC_USE_NEW_STATIC);
420 }
01ca9ac3
KS
421 dca_scalefactor.offset = -64;
422 dca_scalefactor.wrap = 2;
0cfa85dd
KS
423 for (i = 0; i < 5; i++) {
424 dca_scalefactor.vlc[i].table = &dca_table[dca_vlc_offs[i + 5]];
425 dca_scalefactor.vlc[i].table_allocated = dca_vlc_offs[i + 6] - dca_vlc_offs[i + 5];
01ca9ac3
KS
426 init_vlc(&dca_scalefactor.vlc[i], SCALES_VLC_BITS, 129,
427 scales_bits[i], 1, 1,
0cfa85dd
KS
428 scales_codes[i], 2, 2, INIT_VLC_USE_NEW_STATIC);
429 }
01ca9ac3
KS
430 dca_tmode.offset = 0;
431 dca_tmode.wrap = 1;
0cfa85dd
KS
432 for (i = 0; i < 4; i++) {
433 dca_tmode.vlc[i].table = &dca_table[dca_vlc_offs[i + 10]];
434 dca_tmode.vlc[i].table_allocated = dca_vlc_offs[i + 11] - dca_vlc_offs[i + 10];
01ca9ac3
KS
435 init_vlc(&dca_tmode.vlc[i], tmode_vlc_bits[i], 4,
436 tmode_bits[i], 1, 1,
0cfa85dd
KS
437 tmode_codes[i], 2, 2, INIT_VLC_USE_NEW_STATIC);
438 }
01ca9ac3 439
d1177cb5 440 for (i = 0; i < 10; i++)
f37b4efe
SG
441 for (j = 0; j < 7; j++) {
442 if (!bitalloc_codes[i][j])
443 break;
444 dca_smpl_bitalloc[i + 1].offset = bitalloc_offsets[i];
445 dca_smpl_bitalloc[i + 1].wrap = 1 + (j > 4);
446 dca_smpl_bitalloc[i + 1].vlc[j].table = &dca_table[dca_vlc_offs[c]];
447 dca_smpl_bitalloc[i + 1].vlc[j].table_allocated = dca_vlc_offs[c + 1] - dca_vlc_offs[c];
448
449 init_vlc(&dca_smpl_bitalloc[i + 1].vlc[j], bitalloc_maxbits[i][j],
01ca9ac3
KS
450 bitalloc_sizes[i],
451 bitalloc_bits[i][j], 1, 1,
0cfa85dd
KS
452 bitalloc_codes[i][j], 2, 2, INIT_VLC_USE_NEW_STATIC);
453 c++;
01ca9ac3 454 }
5e534865 455 vlcs_initialized = 1;
01ca9ac3
KS
456}
457
458static inline void get_array(GetBitContext *gb, int *dst, int len, int bits)
459{
f37b4efe 460 while (len--)
01ca9ac3
KS
461 *dst++ = get_bits(gb, bits);
462}
463
f37b4efe 464static int dca_parse_audio_coding_header(DCAContext *s, int base_channel)
01ca9ac3
KS
465{
466 int i, j;
467 static const float adj_table[4] = { 1.0, 1.1250, 1.2500, 1.4375 };
468 static const int bitlen[11] = { 0, 1, 2, 2, 2, 2, 3, 3, 3, 3, 3 };
f37b4efe 469 static const int thr[11] = { 0, 1, 3, 3, 3, 3, 7, 7, 7, 7, 7 };
01ca9ac3 470
f37b4efe
SG
471 s->total_channels = get_bits(&s->gb, 3) + 1 + base_channel;
472 s->prim_channels = s->total_channels;
774e9acf 473
6baef06e 474 if (s->prim_channels > DCA_PRIM_CHANNELS_MAX)
774e9acf 475 s->prim_channels = DCA_PRIM_CHANNELS_MAX;
6baef06e
NB
476
477
774e9acf 478 for (i = base_channel; i < s->prim_channels; i++) {
6baef06e
NB
479 s->subband_activity[i] = get_bits(&s->gb, 5) + 2;
480 if (s->subband_activity[i] > DCA_SUBBANDS)
481 s->subband_activity[i] = DCA_SUBBANDS;
482 }
774e9acf 483 for (i = base_channel; i < s->prim_channels; i++) {
6baef06e
NB
484 s->vq_start_subband[i] = get_bits(&s->gb, 5) + 1;
485 if (s->vq_start_subband[i] > DCA_SUBBANDS)
486 s->vq_start_subband[i] = DCA_SUBBANDS;
487 }
774e9acf
NB
488 get_array(&s->gb, s->joint_intensity + base_channel, s->prim_channels - base_channel, 3);
489 get_array(&s->gb, s->transient_huffman + base_channel, s->prim_channels - base_channel, 2);
490 get_array(&s->gb, s->scalefactor_huffman + base_channel, s->prim_channels - base_channel, 3);
491 get_array(&s->gb, s->bitalloc_huffman + base_channel, s->prim_channels - base_channel, 3);
6baef06e
NB
492
493 /* Get codebooks quantization indexes */
774e9acf
NB
494 if (!base_channel)
495 memset(s->quant_index_huffman, 0, sizeof(s->quant_index_huffman));
6baef06e 496 for (j = 1; j < 11; j++)
774e9acf 497 for (i = base_channel; i < s->prim_channels; i++)
6baef06e
NB
498 s->quant_index_huffman[i][j] = get_bits(&s->gb, bitlen[j]);
499
500 /* Get scale factor adjustment */
501 for (j = 0; j < 11; j++)
774e9acf 502 for (i = base_channel; i < s->prim_channels; i++)
6baef06e
NB
503 s->scalefactor_adj[i][j] = 1;
504
505 for (j = 1; j < 11; j++)
774e9acf 506 for (i = base_channel; i < s->prim_channels; i++)
6baef06e
NB
507 if (s->quant_index_huffman[i][j] < thr[j])
508 s->scalefactor_adj[i][j] = adj_table[get_bits(&s->gb, 2)];
509
510 if (s->crc_present) {
511 /* Audio header CRC check */
512 get_bits(&s->gb, 16);
513 }
514
f37b4efe 515 s->current_subframe = 0;
6baef06e
NB
516 s->current_subsubframe = 0;
517
518#ifdef TRACE
519 av_log(s->avctx, AV_LOG_DEBUG, "subframes: %i\n", s->subframes);
520 av_log(s->avctx, AV_LOG_DEBUG, "prim channels: %i\n", s->prim_channels);
f37b4efe
SG
521 for (i = base_channel; i < s->prim_channels; i++) {
522 av_log(s->avctx, AV_LOG_DEBUG, "subband activity: %i\n",
523 s->subband_activity[i]);
524 av_log(s->avctx, AV_LOG_DEBUG, "vq start subband: %i\n",
525 s->vq_start_subband[i]);
526 av_log(s->avctx, AV_LOG_DEBUG, "joint intensity: %i\n",
527 s->joint_intensity[i]);
528 av_log(s->avctx, AV_LOG_DEBUG, "transient mode codebook: %i\n",
529 s->transient_huffman[i]);
530 av_log(s->avctx, AV_LOG_DEBUG, "scale factor codebook: %i\n",
531 s->scalefactor_huffman[i]);
532 av_log(s->avctx, AV_LOG_DEBUG, "bit allocation quantizer: %i\n",
533 s->bitalloc_huffman[i]);
6baef06e
NB
534 av_log(s->avctx, AV_LOG_DEBUG, "quant index huff:");
535 for (j = 0; j < 11; j++)
f37b4efe 536 av_log(s->avctx, AV_LOG_DEBUG, " %i", s->quant_index_huffman[i][j]);
6baef06e
NB
537 av_log(s->avctx, AV_LOG_DEBUG, "\n");
538 av_log(s->avctx, AV_LOG_DEBUG, "scalefac adj:");
539 for (j = 0; j < 11; j++)
540 av_log(s->avctx, AV_LOG_DEBUG, " %1.3f", s->scalefactor_adj[i][j]);
541 av_log(s->avctx, AV_LOG_DEBUG, "\n");
542 }
543#endif
544
f37b4efe 545 return 0;
6baef06e
NB
546}
547
f37b4efe 548static int dca_parse_frame_header(DCAContext *s)
6baef06e 549{
01ca9ac3
KS
550 init_get_bits(&s->gb, s->dca_buffer, s->dca_buffer_size * 8);
551
552 /* Sync code */
999e7ebd 553 skip_bits_long(&s->gb, 32);
01ca9ac3
KS
554
555 /* Frame header */
556 s->frame_type = get_bits(&s->gb, 1);
557 s->samples_deficit = get_bits(&s->gb, 5) + 1;
558 s->crc_present = get_bits(&s->gb, 1);
559 s->sample_blocks = get_bits(&s->gb, 7) + 1;
560 s->frame_size = get_bits(&s->gb, 14) + 1;
561 if (s->frame_size < 95)
f44059d2 562 return AVERROR_INVALIDDATA;
01ca9ac3 563 s->amode = get_bits(&s->gb, 6);
19cf7163 564 s->sample_rate = avpriv_dca_sample_rates[get_bits(&s->gb, 4)];
01ca9ac3 565 if (!s->sample_rate)
f44059d2 566 return AVERROR_INVALIDDATA;
d5b3a863 567 s->bit_rate_index = get_bits(&s->gb, 5);
9ed73b48 568 s->bit_rate = dca_bit_rates[s->bit_rate_index];
01ca9ac3 569 if (!s->bit_rate)
f44059d2 570 return AVERROR_INVALIDDATA;
01ca9ac3
KS
571
572 s->downmix = get_bits(&s->gb, 1);
573 s->dynrange = get_bits(&s->gb, 1);
574 s->timestamp = get_bits(&s->gb, 1);
575 s->aux_data = get_bits(&s->gb, 1);
576 s->hdcd = get_bits(&s->gb, 1);
577 s->ext_descr = get_bits(&s->gb, 3);
578 s->ext_coding = get_bits(&s->gb, 1);
579 s->aspf = get_bits(&s->gb, 1);
580 s->lfe = get_bits(&s->gb, 2);
581 s->predictor_history = get_bits(&s->gb, 1);
582
583 /* TODO: check CRC */
584 if (s->crc_present)
585 s->header_crc = get_bits(&s->gb, 16);
586
587 s->multirate_inter = get_bits(&s->gb, 1);
588 s->version = get_bits(&s->gb, 4);
589 s->copy_history = get_bits(&s->gb, 2);
590 s->source_pcm_res = get_bits(&s->gb, 3);
591 s->front_sum = get_bits(&s->gb, 1);
592 s->surround_sum = get_bits(&s->gb, 1);
593 s->dialog_norm = get_bits(&s->gb, 4);
594
595 /* FIXME: channels mixing levels */
cc826626 596 s->output = s->amode;
f37b4efe
SG
597 if (s->lfe)
598 s->output |= DCA_LFE;
01ca9ac3
KS
599
600#ifdef TRACE
601 av_log(s->avctx, AV_LOG_DEBUG, "frame type: %i\n", s->frame_type);
602 av_log(s->avctx, AV_LOG_DEBUG, "samples deficit: %i\n", s->samples_deficit);
603 av_log(s->avctx, AV_LOG_DEBUG, "crc present: %i\n", s->crc_present);
604 av_log(s->avctx, AV_LOG_DEBUG, "sample blocks: %i (%i samples)\n",
605 s->sample_blocks, s->sample_blocks * 32);
606 av_log(s->avctx, AV_LOG_DEBUG, "frame size: %i bytes\n", s->frame_size);
607 av_log(s->avctx, AV_LOG_DEBUG, "amode: %i (%i channels)\n",
608 s->amode, dca_channels[s->amode]);
49c91c34
BL
609 av_log(s->avctx, AV_LOG_DEBUG, "sample rate: %i Hz\n",
610 s->sample_rate);
611 av_log(s->avctx, AV_LOG_DEBUG, "bit rate: %i bits/s\n",
612 s->bit_rate);
01ca9ac3
KS
613 av_log(s->avctx, AV_LOG_DEBUG, "downmix: %i\n", s->downmix);
614 av_log(s->avctx, AV_LOG_DEBUG, "dynrange: %i\n", s->dynrange);
615 av_log(s->avctx, AV_LOG_DEBUG, "timestamp: %i\n", s->timestamp);
616 av_log(s->avctx, AV_LOG_DEBUG, "aux_data: %i\n", s->aux_data);
617 av_log(s->avctx, AV_LOG_DEBUG, "hdcd: %i\n", s->hdcd);
618 av_log(s->avctx, AV_LOG_DEBUG, "ext descr: %i\n", s->ext_descr);
619 av_log(s->avctx, AV_LOG_DEBUG, "ext coding: %i\n", s->ext_coding);
620 av_log(s->avctx, AV_LOG_DEBUG, "aspf: %i\n", s->aspf);
621 av_log(s->avctx, AV_LOG_DEBUG, "lfe: %i\n", s->lfe);
622 av_log(s->avctx, AV_LOG_DEBUG, "predictor history: %i\n",
623 s->predictor_history);
624 av_log(s->avctx, AV_LOG_DEBUG, "header crc: %i\n", s->header_crc);
625 av_log(s->avctx, AV_LOG_DEBUG, "multirate inter: %i\n",
626 s->multirate_inter);
627 av_log(s->avctx, AV_LOG_DEBUG, "version number: %i\n", s->version);
628 av_log(s->avctx, AV_LOG_DEBUG, "copy history: %i\n", s->copy_history);
629 av_log(s->avctx, AV_LOG_DEBUG,
630 "source pcm resolution: %i (%i bits/sample)\n",
631 s->source_pcm_res, dca_bits_per_sample[s->source_pcm_res]);
632 av_log(s->avctx, AV_LOG_DEBUG, "front sum: %i\n", s->front_sum);
633 av_log(s->avctx, AV_LOG_DEBUG, "surround sum: %i\n", s->surround_sum);
634 av_log(s->avctx, AV_LOG_DEBUG, "dialog norm: %i\n", s->dialog_norm);
635 av_log(s->avctx, AV_LOG_DEBUG, "\n");
636#endif
637
638 /* Primary audio coding header */
639 s->subframes = get_bits(&s->gb, 4) + 1;
01ca9ac3 640
774e9acf 641 return dca_parse_audio_coding_header(s, 0);
01ca9ac3
KS
642}
643
644
e6ffd997 645static inline int get_scale(GetBitContext *gb, int level, int value, int log2range)
01ca9ac3 646{
f37b4efe
SG
647 if (level < 5) {
648 /* huffman encoded */
649 value += get_bitalloc(gb, &dca_scalefactor, level);
93b53ffb 650 value = av_clip(value, 0, (1 << log2range) - 1);
e6ffd997
RB
651 } else if (level < 8) {
652 if (level + 1 > log2range) {
653 skip_bits(gb, level + 1 - log2range);
654 value = get_bits(gb, log2range);
655 } else {
656 value = get_bits(gb, level + 1);
657 }
658 }
f37b4efe 659 return value;
01ca9ac3
KS
660}
661
f37b4efe 662static int dca_subframe_header(DCAContext *s, int base_channel, int block_index)
01ca9ac3
KS
663{
664 /* Primary audio coding side information */
665 int j, k;
666
39f4d329 667 if (get_bits_left(&s->gb) < 0)
f44059d2 668 return AVERROR_INVALIDDATA;
39f4d329 669
774e9acf 670 if (!base_channel) {
f37b4efe 671 s->subsubframes[s->current_subframe] = get_bits(&s->gb, 2) + 1;
774e9acf
NB
672 s->partial_samples[s->current_subframe] = get_bits(&s->gb, 3);
673 }
674
675 for (j = base_channel; j < s->prim_channels; j++) {
01ca9ac3
KS
676 for (k = 0; k < s->subband_activity[j]; k++)
677 s->prediction_mode[j][k] = get_bits(&s->gb, 1);
678 }
679
680 /* Get prediction codebook */
774e9acf 681 for (j = base_channel; j < s->prim_channels; j++) {
01ca9ac3
KS
682 for (k = 0; k < s->subband_activity[j]; k++) {
683 if (s->prediction_mode[j][k] > 0) {
684 /* (Prediction coefficient VQ address) */
685 s->prediction_vq[j][k] = get_bits(&s->gb, 12);
686 }
687 }
688 }
689
690 /* Bit allocation index */
774e9acf 691 for (j = base_channel; j < s->prim_channels; j++) {
01ca9ac3
KS
692 for (k = 0; k < s->vq_start_subband[j]; k++) {
693 if (s->bitalloc_huffman[j] == 6)
694 s->bitalloc[j][k] = get_bits(&s->gb, 5);
695 else if (s->bitalloc_huffman[j] == 5)
696 s->bitalloc[j][k] = get_bits(&s->gb, 4);
ebf71dbd
BL
697 else if (s->bitalloc_huffman[j] == 7) {
698 av_log(s->avctx, AV_LOG_ERROR,
699 "Invalid bit allocation index\n");
f44059d2 700 return AVERROR_INVALIDDATA;
ebf71dbd 701 } else {
01ca9ac3 702 s->bitalloc[j][k] =
c5d13492 703 get_bitalloc(&s->gb, &dca_bitalloc_index, s->bitalloc_huffman[j]);
01ca9ac3
KS
704 }
705
706 if (s->bitalloc[j][k] > 26) {
1218777f
DB
707 av_dlog(s->avctx, "bitalloc index [%i][%i] too big (%i)\n",
708 j, k, s->bitalloc[j][k]);
f44059d2 709 return AVERROR_INVALIDDATA;
01ca9ac3
KS
710 }
711 }
712 }
713
714 /* Transition mode */
774e9acf 715 for (j = base_channel; j < s->prim_channels; j++) {
01ca9ac3
KS
716 for (k = 0; k < s->subband_activity[j]; k++) {
717 s->transition_mode[j][k] = 0;
774e9acf 718 if (s->subsubframes[s->current_subframe] > 1 &&
01ca9ac3
KS
719 k < s->vq_start_subband[j] && s->bitalloc[j][k] > 0) {
720 s->transition_mode[j][k] =
721 get_bitalloc(&s->gb, &dca_tmode, s->transient_huffman[j]);
722 }
723 }
724 }
725
39f4d329 726 if (get_bits_left(&s->gb) < 0)
f44059d2 727 return AVERROR_INVALIDDATA;
39f4d329 728
774e9acf 729 for (j = base_channel; j < s->prim_channels; j++) {
a9f87158 730 const uint32_t *scale_table;
e6ffd997 731 int scale_sum, log_size;
01ca9ac3 732
f37b4efe
SG
733 memset(s->scale_factor[j], 0,
734 s->subband_activity[j] * sizeof(s->scale_factor[0][0][0]) * 2);
01ca9ac3 735
e6ffd997 736 if (s->scalefactor_huffman[j] == 6) {
a9f87158 737 scale_table = scale_factor_quant7;
e6ffd997
RB
738 log_size = 7;
739 } else {
a9f87158 740 scale_table = scale_factor_quant6;
e6ffd997
RB
741 log_size = 6;
742 }
01ca9ac3
KS
743
744 /* When huffman coded, only the difference is encoded */
745 scale_sum = 0;
746
747 for (k = 0; k < s->subband_activity[j]; k++) {
748 if (k >= s->vq_start_subband[j] || s->bitalloc[j][k] > 0) {
e6ffd997 749 scale_sum = get_scale(&s->gb, s->scalefactor_huffman[j], scale_sum, log_size);
01ca9ac3
KS
750 s->scale_factor[j][k][0] = scale_table[scale_sum];
751 }
752
753 if (k < s->vq_start_subband[j] && s->transition_mode[j][k]) {
754 /* Get second scale factor */
e6ffd997 755 scale_sum = get_scale(&s->gb, s->scalefactor_huffman[j], scale_sum, log_size);
01ca9ac3
KS
756 s->scale_factor[j][k][1] = scale_table[scale_sum];
757 }
758 }
759 }
760
761 /* Joint subband scale factor codebook select */
774e9acf 762 for (j = base_channel; j < s->prim_channels; j++) {
01ca9ac3
KS
763 /* Transmitted only if joint subband coding enabled */
764 if (s->joint_intensity[j] > 0)
765 s->joint_huff[j] = get_bits(&s->gb, 3);
766 }
767
39f4d329 768 if (get_bits_left(&s->gb) < 0)
f44059d2 769 return AVERROR_INVALIDDATA;
39f4d329 770
01ca9ac3 771 /* Scale factors for joint subband coding */
774e9acf 772 for (j = base_channel; j < s->prim_channels; j++) {
01ca9ac3
KS
773 int source_channel;
774
775 /* Transmitted only if joint subband coding enabled */
776 if (s->joint_intensity[j] > 0) {
777 int scale = 0;
778 source_channel = s->joint_intensity[j] - 1;
779
780 /* When huffman coded, only the difference is encoded
781 * (is this valid as well for joint scales ???) */
782
783 for (k = s->subband_activity[j]; k < s->subband_activity[source_channel]; k++) {
e6ffd997 784 scale = get_scale(&s->gb, s->joint_huff[j], 64 /* bias */, 7);
01ca9ac3
KS
785 s->joint_scale_factor[j][k] = scale; /*joint_scale_table[scale]; */
786 }
787
268fcbe2 788 if (!(s->debug_flag & 0x02)) {
01ca9ac3
KS
789 av_log(s->avctx, AV_LOG_DEBUG,
790 "Joint stereo coding not supported\n");
791 s->debug_flag |= 0x02;
792 }
793 }
794 }
795
796 /* Stereo downmix coefficients */
774e9acf 797 if (!base_channel && s->prim_channels > 2) {
d1177cb5 798 if (s->downmix) {
774e9acf 799 for (j = base_channel; j < s->prim_channels; j++) {
6369e6eb
JR
800 s->downmix_coef[j][0] = get_bits(&s->gb, 7);
801 s->downmix_coef[j][1] = get_bits(&s->gb, 7);
802 }
c31a76e4
JR
803 } else {
804 int am = s->amode & DCA_CHANNEL_MASK;
e6ffd997
RB
805 if (am >= FF_ARRAY_ELEMS(dca_default_coeffs)) {
806 av_log(s->avctx, AV_LOG_ERROR,
807 "Invalid channel mode %d\n", am);
808 return AVERROR_INVALIDDATA;
809 }
774e9acf 810 for (j = base_channel; j < s->prim_channels; j++) {
c31a76e4
JR
811 s->downmix_coef[j][0] = dca_default_coeffs[am][j][0];
812 s->downmix_coef[j][1] = dca_default_coeffs[am][j][1];
813 }
814 }
01ca9ac3
KS
815 }
816
817 /* Dynamic range coefficient */
ace7f813 818 if (!base_channel && s->dynrange)
01ca9ac3
KS
819 s->dynrange_coef = get_bits(&s->gb, 8);
820
821 /* Side information CRC check word */
822 if (s->crc_present) {
823 get_bits(&s->gb, 16);
824 }
825
826 /*
827 * Primary audio data arrays
828 */
829
830 /* VQ encoded high frequency subbands */
774e9acf 831 for (j = base_channel; j < s->prim_channels; j++)
01ca9ac3
KS
832 for (k = s->vq_start_subband[j]; k < s->subband_activity[j]; k++)
833 /* 1 vector -> 32 samples */
834 s->high_freq_vq[j][k] = get_bits(&s->gb, 10);
835
836 /* Low frequency effect data */
774e9acf 837 if (!base_channel && s->lfe) {
01ca9ac3 838 /* LFE samples */
6baef06e 839 int lfe_samples = 2 * s->lfe * (4 + block_index);
774e9acf 840 int lfe_end_sample = 2 * s->lfe * (4 + block_index + s->subsubframes[s->current_subframe]);
01ca9ac3
KS
841 float lfe_scale;
842
6baef06e 843 for (j = lfe_samples; j < lfe_end_sample; j++) {
01ca9ac3
KS
844 /* Signed 8 bits int */
845 s->lfe_data[j] = get_sbits(&s->gb, 8);
846 }
847
848 /* Scale factor index */
e6ffd997
RB
849 skip_bits(&s->gb, 1);
850 s->lfe_scale_factor = scale_factor_quant7[get_bits(&s->gb, 7)];
01ca9ac3
KS
851
852 /* Quantization step size * scale factor */
853 lfe_scale = 0.035 * s->lfe_scale_factor;
854
6baef06e 855 for (j = lfe_samples; j < lfe_end_sample; j++)
01ca9ac3
KS
856 s->lfe_data[j] *= lfe_scale;
857 }
858
859#ifdef TRACE
f37b4efe
SG
860 av_log(s->avctx, AV_LOG_DEBUG, "subsubframes: %i\n",
861 s->subsubframes[s->current_subframe]);
01ca9ac3 862 av_log(s->avctx, AV_LOG_DEBUG, "partial samples: %i\n",
774e9acf 863 s->partial_samples[s->current_subframe]);
f37b4efe 864
774e9acf 865 for (j = base_channel; j < s->prim_channels; j++) {
01ca9ac3
KS
866 av_log(s->avctx, AV_LOG_DEBUG, "prediction mode:");
867 for (k = 0; k < s->subband_activity[j]; k++)
868 av_log(s->avctx, AV_LOG_DEBUG, " %i", s->prediction_mode[j][k]);
869 av_log(s->avctx, AV_LOG_DEBUG, "\n");
870 }
774e9acf 871 for (j = base_channel; j < s->prim_channels; j++) {
01ca9ac3 872 for (k = 0; k < s->subband_activity[j]; k++)
f37b4efe
SG
873 av_log(s->avctx, AV_LOG_DEBUG,
874 "prediction coefs: %f, %f, %f, %f\n",
875 (float) adpcm_vb[s->prediction_vq[j][k]][0] / 8192,
876 (float) adpcm_vb[s->prediction_vq[j][k]][1] / 8192,
877 (float) adpcm_vb[s->prediction_vq[j][k]][2] / 8192,
878 (float) adpcm_vb[s->prediction_vq[j][k]][3] / 8192);
01ca9ac3 879 }
774e9acf 880 for (j = base_channel; j < s->prim_channels; j++) {
01ca9ac3
KS
881 av_log(s->avctx, AV_LOG_DEBUG, "bitalloc index: ");
882 for (k = 0; k < s->vq_start_subband[j]; k++)
883 av_log(s->avctx, AV_LOG_DEBUG, "%2.2i ", s->bitalloc[j][k]);
884 av_log(s->avctx, AV_LOG_DEBUG, "\n");
885 }
774e9acf 886 for (j = base_channel; j < s->prim_channels; j++) {
01ca9ac3
KS
887 av_log(s->avctx, AV_LOG_DEBUG, "Transition mode:");
888 for (k = 0; k < s->subband_activity[j]; k++)
889 av_log(s->avctx, AV_LOG_DEBUG, " %i", s->transition_mode[j][k]);
890 av_log(s->avctx, AV_LOG_DEBUG, "\n");
891 }
774e9acf 892 for (j = base_channel; j < s->prim_channels; j++) {
01ca9ac3
KS
893 av_log(s->avctx, AV_LOG_DEBUG, "Scale factor:");
894 for (k = 0; k < s->subband_activity[j]; k++) {
895 if (k >= s->vq_start_subband[j] || s->bitalloc[j][k] > 0)
896 av_log(s->avctx, AV_LOG_DEBUG, " %i", s->scale_factor[j][k][0]);
897 if (k < s->vq_start_subband[j] && s->transition_mode[j][k])
898 av_log(s->avctx, AV_LOG_DEBUG, " %i(t)", s->scale_factor[j][k][1]);
899 }
900 av_log(s->avctx, AV_LOG_DEBUG, "\n");
901 }
774e9acf 902 for (j = base_channel; j < s->prim_channels; j++) {
01ca9ac3 903 if (s->joint_intensity[j] > 0) {
56e4603e 904 int source_channel = s->joint_intensity[j] - 1;
01ca9ac3
KS
905 av_log(s->avctx, AV_LOG_DEBUG, "Joint scale factor index:\n");
906 for (k = s->subband_activity[j]; k < s->subband_activity[source_channel]; k++)
907 av_log(s->avctx, AV_LOG_DEBUG, " %i", s->joint_scale_factor[j][k]);
908 av_log(s->avctx, AV_LOG_DEBUG, "\n");
909 }
910 }
774e9acf 911 if (!base_channel && s->prim_channels > 2 && s->downmix) {
01ca9ac3
KS
912 av_log(s->avctx, AV_LOG_DEBUG, "Downmix coeffs:\n");
913 for (j = 0; j < s->prim_channels; j++) {
f37b4efe
SG
914 av_log(s->avctx, AV_LOG_DEBUG, "Channel 0, %d = %f\n", j,
915 dca_downmix_coeffs[s->downmix_coef[j][0]]);
916 av_log(s->avctx, AV_LOG_DEBUG, "Channel 1, %d = %f\n", j,
917 dca_downmix_coeffs[s->downmix_coef[j][1]]);
01ca9ac3
KS
918 }
919 av_log(s->avctx, AV_LOG_DEBUG, "\n");
920 }
774e9acf 921 for (j = base_channel; j < s->prim_channels; j++)
01ca9ac3
KS
922 for (k = s->vq_start_subband[j]; k < s->subband_activity[j]; k++)
923 av_log(s->avctx, AV_LOG_DEBUG, "VQ index: %i\n", s->high_freq_vq[j][k]);
774e9acf 924 if (!base_channel && s->lfe) {
6baef06e
NB
925 int lfe_samples = 2 * s->lfe * (4 + block_index);
926 int lfe_end_sample = 2 * s->lfe * (4 + block_index + s->subsubframes[s->current_subframe]);
927
01ca9ac3 928 av_log(s->avctx, AV_LOG_DEBUG, "LFE samples:\n");
6baef06e 929 for (j = lfe_samples; j < lfe_end_sample; j++)
01ca9ac3
KS
930 av_log(s->avctx, AV_LOG_DEBUG, " %f", s->lfe_data[j]);
931 av_log(s->avctx, AV_LOG_DEBUG, "\n");
932 }
933#endif
934
935 return 0;
936}
937
f37b4efe 938static void qmf_32_subbands(DCAContext *s, int chans,
01ca9ac3 939 float samples_in[32][8], float *samples_out,
9d06d7bc 940 float scale)
01ca9ac3 941{
b6398969 942 const float *prCoeff;
d49dded5 943 int i;
01ca9ac3 944
b92d483b 945 int sb_act = s->subband_activity[chans];
89df5e95 946 int subindex;
01ca9ac3 947
f37b4efe 948 scale *= sqrt(1 / 8.0);
01ca9ac3
KS
949
950 /* Select filter */
951 if (!s->multirate_inter) /* Non-perfect reconstruction */
b6398969 952 prCoeff = fir_32bands_nonperfect;
01ca9ac3 953 else /* Perfect reconstruction */
b6398969 954 prCoeff = fir_32bands_perfect;
01ca9ac3 955
bf00a73a
MR
956 for (i = sb_act; i < 32; i++)
957 s->raXin[i] = 0.0;
958
01ca9ac3
KS
959 /* Reconstructed channel sample index */
960 for (subindex = 0; subindex < 8; subindex++) {
01ca9ac3 961 /* Load in one sample from each subband and clear inactive subbands */
f37b4efe 962 for (i = 0; i < sb_act; i++) {
559c244d 963 unsigned sign = (i - 1) & 2;
f37b4efe 964 uint32_t v = AV_RN32A(&samples_in[i][subindex]) ^ sign << 30;
0dc7df28 965 AV_WN32A(&s->raXin[i], v);
89df5e95 966 }
01ca9ac3 967
f462ed1f 968 s->synth.synth_filter_float(&s->imdct,
f37b4efe
SG
969 s->subband_fir_hist[chans],
970 &s->hist_index[chans],
971 s->subband_fir_noidea[chans], prCoeff,
972 samples_out, s->raXin, scale);
973 samples_out += 32;
01ca9ac3
KS
974 }
975}
976
309d16a4 977static void lfe_interpolation_fir(DCAContext *s, int decimation_select,
01ca9ac3 978 int num_deci_sample, float *samples_in,
9d06d7bc 979 float *samples_out, float scale)
01ca9ac3
KS
980{
981 /* samples_in: An array holding decimated samples.
982 * Samples in current subframe starts from samples_in[0],
983 * while samples_in[-1], samples_in[-2], ..., stores samples
984 * from last subframe as history.
985 *
986 * samples_out: An array holding interpolated samples
987 */
988
309d16a4 989 int decifactor;
01ca9ac3 990 const float *prCoeff;
01ca9ac3
KS
991 int deciindex;
992
993 /* Select decimation filter */
994 if (decimation_select == 1) {
766fefe8 995 decifactor = 64;
01ca9ac3
KS
996 prCoeff = lfe_fir_128;
997 } else {
766fefe8 998 decifactor = 32;
01ca9ac3
KS
999 prCoeff = lfe_fir_64;
1000 }
1001 /* Interpolation */
1002 for (deciindex = 0; deciindex < num_deci_sample; deciindex++) {
f37b4efe 1003 s->dcadsp.lfe_fir(samples_out, samples_in, prCoeff, decifactor, scale);
766fefe8 1004 samples_in++;
309d16a4 1005 samples_out += 2 * decifactor;
01ca9ac3
KS
1006 }
1007}
1008
1009/* downmixing routines */
f37b4efe
SG
1010#define MIX_REAR1(samples, si1, rs, coef) \
1011 samples[i] += samples[si1] * coef[rs][0]; \
1012 samples[i+256] += samples[si1] * coef[rs][1];
1013
1014#define MIX_REAR2(samples, si1, si2, rs, coef) \
1015 samples[i] += samples[si1] * coef[rs][0] + samples[si2] * coef[rs + 1][0]; \
1016 samples[i+256] += samples[si1] * coef[rs][1] + samples[si2] * coef[rs + 1][1];
1017
1018#define MIX_FRONT3(samples, coef) \
1019 t = samples[i + c]; \
1020 u = samples[i + l]; \
1021 v = samples[i + r]; \
9d06d7bc
JR
1022 samples[i] = t * coef[0][0] + u * coef[1][0] + v * coef[2][0]; \
1023 samples[i+256] = t * coef[0][1] + u * coef[1][1] + v * coef[2][1];
01ca9ac3 1024
f37b4efe
SG
1025#define DOWNMIX_TO_STEREO(op1, op2) \
1026 for (i = 0; i < 256; i++) { \
1027 op1 \
1028 op2 \
01ca9ac3
KS
1029 }
1030
c31a76e4 1031static void dca_downmix(float *samples, int srcfmt,
df984493 1032 int downmix_coef[DCA_PRIM_CHANNELS_MAX][2],
9d06d7bc 1033 const int8_t *channel_mapping)
01ca9ac3 1034{
f37b4efe 1035 int c, l, r, sl, sr, s;
01ca9ac3 1036 int i;
df984493 1037 float t, u, v;
c31a76e4
JR
1038 float coef[DCA_PRIM_CHANNELS_MAX][2];
1039
f37b4efe 1040 for (i = 0; i < DCA_PRIM_CHANNELS_MAX; i++) {
c31a76e4
JR
1041 coef[i][0] = dca_downmix_coeffs[downmix_coef[i][0]];
1042 coef[i][1] = dca_downmix_coeffs[downmix_coef[i][1]];
1043 }
01ca9ac3
KS
1044
1045 switch (srcfmt) {
1046 case DCA_MONO:
1047 case DCA_CHANNEL:
1048 case DCA_STEREO_TOTAL:
1049 case DCA_STEREO_SUMDIFF:
1050 case DCA_4F2R:
1051 av_log(NULL, 0, "Not implemented!\n");
1052 break;
1053 case DCA_STEREO:
1054 break;
1055 case DCA_3F:
df984493
NB
1056 c = channel_mapping[0] * 256;
1057 l = channel_mapping[1] * 256;
1058 r = channel_mapping[2] * 256;
f37b4efe 1059 DOWNMIX_TO_STEREO(MIX_FRONT3(samples, coef), );
01ca9ac3
KS
1060 break;
1061 case DCA_2F1R:
df984493 1062 s = channel_mapping[2] * 256;
f37b4efe 1063 DOWNMIX_TO_STEREO(MIX_REAR1(samples, i + s, 2, coef), );
01ca9ac3
KS
1064 break;
1065 case DCA_3F1R:
df984493
NB
1066 c = channel_mapping[0] * 256;
1067 l = channel_mapping[1] * 256;
1068 r = channel_mapping[2] * 256;
1069 s = channel_mapping[3] * 256;
c31a76e4 1070 DOWNMIX_TO_STEREO(MIX_FRONT3(samples, coef),
df984493 1071 MIX_REAR1(samples, i + s, 3, coef));
01ca9ac3
KS
1072 break;
1073 case DCA_2F2R:
df984493
NB
1074 sl = channel_mapping[2] * 256;
1075 sr = channel_mapping[3] * 256;
f37b4efe 1076 DOWNMIX_TO_STEREO(MIX_REAR2(samples, i + sl, i + sr, 2, coef), );
01ca9ac3
KS
1077 break;
1078 case DCA_3F2R:
f37b4efe
SG
1079 c = channel_mapping[0] * 256;
1080 l = channel_mapping[1] * 256;
1081 r = channel_mapping[2] * 256;
df984493
NB
1082 sl = channel_mapping[3] * 256;
1083 sr = channel_mapping[4] * 256;
c31a76e4 1084 DOWNMIX_TO_STEREO(MIX_FRONT3(samples, coef),
df984493 1085 MIX_REAR2(samples, i + sl, i + sr, 3, coef));
01ca9ac3
KS
1086 break;
1087 }
1088}
1089
1090
00a856e3 1091#ifndef decode_blockcodes
01ca9ac3
KS
1092/* Very compact version of the block code decoder that does not use table
1093 * look-up but is slightly slower */
1094static int decode_blockcode(int code, int levels, int *values)
1095{
1096 int i;
1097 int offset = (levels - 1) >> 1;
1098
1099 for (i = 0; i < 4; i++) {
843c7aa8 1100 int div = FASTDIV(code, levels);
f37b4efe 1101 values[i] = code - offset - div * levels;
843c7aa8 1102 code = div;
01ca9ac3
KS
1103 }
1104
00a856e3
MR
1105 return code;
1106}
1107
1108static int decode_blockcodes(int code1, int code2, int levels, int *values)
1109{
1110 return decode_blockcode(code1, levels, values) |
1111 decode_blockcode(code2, levels, values + 4);
01ca9ac3 1112}
00a856e3 1113#endif
01ca9ac3 1114
f37b4efe
SG
1115static const uint8_t abits_sizes[7] = { 7, 10, 12, 13, 15, 17, 19 };
1116static const uint8_t abits_levels[7] = { 3, 5, 7, 9, 13, 17, 25 };
01ca9ac3 1117
bf5d46d8
MR
1118#ifndef int8x8_fmul_int32
1119static inline void int8x8_fmul_int32(float *dst, const int8_t *src, int scale)
1120{
1121 float fscale = scale / 16.0;
1122 int i;
1123 for (i = 0; i < 8; i++)
1124 dst[i] = src[i] * fscale;
1125}
1126#endif
1127
f37b4efe 1128static int dca_subsubframe(DCAContext *s, int base_channel, int block_index)
01ca9ac3
KS
1129{
1130 int k, l;
1131 int subsubframe = s->current_subsubframe;
1132
a9f87158 1133 const float *quant_step_table;
01ca9ac3
KS
1134
1135 /* FIXME */
77b4b7c3 1136 float (*subband_samples)[DCA_SUBBANDS][8] = s->subband_samples[block_index];
69e17136 1137 LOCAL_ALIGNED_16(int, block, [8]);
01ca9ac3
KS
1138
1139 /*
1140 * Audio data
1141 */
1142
1143 /* Select quantization step size table */
9ed73b48 1144 if (s->bit_rate_index == 0x1f)
a9f87158 1145 quant_step_table = lossless_quant_d;
01ca9ac3 1146 else
a9f87158 1147 quant_step_table = lossy_quant_d;
01ca9ac3 1148
774e9acf 1149 for (k = base_channel; k < s->prim_channels; k++) {
39f4d329 1150 if (get_bits_left(&s->gb) < 0)
f44059d2 1151 return AVERROR_INVALIDDATA;
39f4d329 1152
01ca9ac3
KS
1153 for (l = 0; l < s->vq_start_subband[k]; l++) {
1154 int m;
1155
1156 /* Select the mid-tread linear quantizer */
1157 int abits = s->bitalloc[k][l];
1158
1159 float quant_step_size = quant_step_table[abits];
01ca9ac3
KS
1160
1161 /*
1162 * Determine quantization index code book and its type
1163 */
1164
1165 /* Select quantization index code book */
1166 int sel = s->quant_index_huffman[k][abits];
1167
1168 /*
1169 * Extract bits from the bit stream
1170 */
f37b4efe 1171 if (!abits) {
01ca9ac3 1172 memset(subband_samples[k][l], 0, 8 * sizeof(subband_samples[0][0][0]));
69e17136
MR
1173 } else {
1174 /* Deal with transients */
1175 int sfi = s->transition_mode[k][l] && subsubframe >= s->transition_mode[k][l];
f37b4efe
SG
1176 float rscale = quant_step_size * s->scale_factor[k][l][sfi] *
1177 s->scalefactor_adj[k][sel];
69e17136 1178
f37b4efe
SG
1179 if (abits >= 11 || !dca_smpl_bitalloc[abits].vlc[sel].table) {
1180 if (abits <= 7) {
2bb29da6 1181 /* Block code */
00a856e3 1182 int block_code1, block_code2, size, levels, err;
2bb29da6 1183
f37b4efe
SG
1184 size = abits_sizes[abits - 1];
1185 levels = abits_levels[abits - 1];
2bb29da6
MR
1186
1187 block_code1 = get_bits(&s->gb, size);
2bb29da6 1188 block_code2 = get_bits(&s->gb, size);
00a856e3
MR
1189 err = decode_blockcodes(block_code1, block_code2,
1190 levels, block);
1191 if (err) {
1192 av_log(s->avctx, AV_LOG_ERROR,
1193 "ERROR: block code look-up failed\n");
1194 return AVERROR_INVALIDDATA;
1195 }
f37b4efe 1196 } else {
2bb29da6
MR
1197 /* no coding */
1198 for (m = 0; m < 8; m++)
1199 block[m] = get_sbits(&s->gb, abits - 3);
1200 }
f37b4efe 1201 } else {
2bb29da6 1202 /* Huffman coded */
01ca9ac3 1203 for (m = 0; m < 8; m++)
f37b4efe
SG
1204 block[m] = get_bitalloc(&s->gb,
1205 &dca_smpl_bitalloc[abits], sel);
01ca9ac3 1206 }
01ca9ac3 1207
c73d99e6 1208 s->fmt_conv.int32_to_float_fmul_scalar(subband_samples[k][l],
f37b4efe 1209 block, rscale, 8);
69e17136 1210 }
01ca9ac3
KS
1211
1212 /*
1213 * Inverse ADPCM if in prediction mode
1214 */
1215 if (s->prediction_mode[k][l]) {
1216 int n;
1217 for (m = 0; m < 8; m++) {
1218 for (n = 1; n <= 4; n++)
1219 if (m >= n)
1220 subband_samples[k][l][m] +=
1221 (adpcm_vb[s->prediction_vq[k][l]][n - 1] *
1222 subband_samples[k][l][m - n] / 8192);
1223 else if (s->predictor_history)
1224 subband_samples[k][l][m] +=
1225 (adpcm_vb[s->prediction_vq[k][l]][n - 1] *
f37b4efe 1226 s->subband_samples_hist[k][l][m - n + 4] / 8192);
01ca9ac3
KS
1227 }
1228 }
1229 }
1230
1231 /*
1232 * Decode VQ encoded high frequencies
1233 */
1234 for (l = s->vq_start_subband[k]; l < s->subband_activity[k]; l++) {
1235 /* 1 vector -> 32 samples but we only need the 8 samples
1236 * for this subsubframe. */
bf5d46d8 1237 int hfvq = s->high_freq_vq[k][l];
01ca9ac3
KS
1238
1239 if (!s->debug_flag & 0x01) {
f37b4efe
SG
1240 av_log(s->avctx, AV_LOG_DEBUG,
1241 "Stream with high frequencies VQ coding\n");
01ca9ac3
KS
1242 s->debug_flag |= 0x01;
1243 }
1244
bf5d46d8
MR
1245 int8x8_fmul_int32(subband_samples[k][l],
1246 &high_freq_vq[hfvq][subsubframe * 8],
1247 s->scale_factor[k][l][0]);
01ca9ac3
KS
1248 }
1249 }
1250
1251 /* Check for DSYNC after subsubframe */
774e9acf 1252 if (s->aspf || subsubframe == s->subsubframes[s->current_subframe] - 1) {
01ca9ac3
KS
1253 if (0xFFFF == get_bits(&s->gb, 16)) { /* 0xFFFF */
1254#ifdef TRACE
1255 av_log(s->avctx, AV_LOG_DEBUG, "Got subframe DSYNC\n");
1256#endif
1257 } else {
1258 av_log(s->avctx, AV_LOG_ERROR, "Didn't get subframe DSYNC\n");
1259 }
1260 }
1261
1262 /* Backup predictor history for adpcm */
774e9acf 1263 for (k = base_channel; k < s->prim_channels; k++)
01ca9ac3 1264 for (l = 0; l < s->vq_start_subband[k]; l++)
f37b4efe
SG
1265 memcpy(s->subband_samples_hist[k][l],
1266 &subband_samples[k][l][4],
1267 4 * sizeof(subband_samples[0][0][0]));
01ca9ac3 1268
6baef06e
NB
1269 return 0;
1270}
1271
f37b4efe 1272static int dca_filter_channels(DCAContext *s, int block_index)
6baef06e
NB
1273{
1274 float (*subband_samples)[DCA_SUBBANDS][8] = s->subband_samples[block_index];
1275 int k;
1276
01ca9ac3
KS
1277 /* 32 subbands QMF */
1278 for (k = 0; k < s->prim_channels; k++) {
f37b4efe
SG
1279/* static float pcm_to_double[8] = { 32768.0, 32768.0, 524288.0, 524288.0,
1280 0, 8388608.0, 8388608.0 };*/
1281 qmf_32_subbands(s, k, subband_samples[k],
1282 &s->samples[256 * s->channel_order_tab[k]],
1283 M_SQRT1_2 * s->scale_bias /* pcm_to_double[s->source_pcm_res] */);
01ca9ac3
KS
1284 }
1285
1286 /* Down mixing */
6baef06e 1287 if (s->avctx->request_channels == 2 && s->prim_channels > 2) {
9d06d7bc 1288 dca_downmix(s->samples, s->amode, s->downmix_coef, s->channel_order_tab);
01ca9ac3
KS
1289 }
1290
1291 /* Generate LFE samples for this subsubframe FIXME!!! */
1292 if (s->output & DCA_LFE) {
309d16a4 1293 lfe_interpolation_fir(s, s->lfe, 2 * s->lfe,
6baef06e 1294 s->lfe_data + 2 * s->lfe * (block_index + 4),
92765276 1295 &s->samples[256 * dca_lfe_index[s->amode]],
f37b4efe 1296 (1.0 / 256.0) * s->scale_bias);
01ca9ac3
KS
1297 /* Outputs 20bits pcm samples */
1298 }
1299
1300 return 0;
1301}
1302
1303
f37b4efe 1304static int dca_subframe_footer(DCAContext *s, int base_channel)
01ca9ac3
KS
1305{
1306 int aux_data_count = 0, i;
01ca9ac3
KS
1307
1308 /*
1309 * Unpack optional information
1310 */
1311
774e9acf
NB
1312 /* presumably optional information only appears in the core? */
1313 if (!base_channel) {
d1177cb5 1314 if (s->timestamp)
999e7ebd 1315 skip_bits_long(&s->gb, 32);
01ca9ac3 1316
d1177cb5
NB
1317 if (s->aux_data)
1318 aux_data_count = get_bits(&s->gb, 6);
01ca9ac3 1319
d1177cb5
NB
1320 for (i = 0; i < aux_data_count; i++)
1321 get_bits(&s->gb, 8);
01ca9ac3 1322
d1177cb5
NB
1323 if (s->crc_present && (s->downmix || s->dynrange))
1324 get_bits(&s->gb, 16);
774e9acf 1325 }
01ca9ac3 1326
01ca9ac3
KS
1327 return 0;
1328}
1329
1330/**
1331 * Decode a dca frame block
1332 *
1333 * @param s pointer to the DCAContext
1334 */
1335
f37b4efe 1336static int dca_decode_block(DCAContext *s, int base_channel, int block_index)
01ca9ac3 1337{
f44059d2 1338 int ret;
01ca9ac3
KS
1339
1340 /* Sanity check */
1341 if (s->current_subframe >= s->subframes) {
1342 av_log(s->avctx, AV_LOG_DEBUG, "check failed: %i>%i",
1343 s->current_subframe, s->subframes);
f44059d2 1344 return AVERROR_INVALIDDATA;
01ca9ac3
KS
1345 }
1346
1347 if (!s->current_subsubframe) {
1348#ifdef TRACE
1349 av_log(s->avctx, AV_LOG_DEBUG, "DSYNC dca_subframe_header\n");
1350#endif
1351 /* Read subframe header */
f44059d2
JR
1352 if ((ret = dca_subframe_header(s, base_channel, block_index)))
1353 return ret;
01ca9ac3
KS
1354 }
1355
1356 /* Read subsubframe */
1357#ifdef TRACE
1358 av_log(s->avctx, AV_LOG_DEBUG, "DSYNC dca_subsubframe\n");
1359#endif
f44059d2
JR
1360 if ((ret = dca_subsubframe(s, base_channel, block_index)))
1361 return ret;
01ca9ac3
KS
1362
1363 /* Update state */
1364 s->current_subsubframe++;
774e9acf 1365 if (s->current_subsubframe >= s->subsubframes[s->current_subframe]) {
01ca9ac3
KS
1366 s->current_subsubframe = 0;
1367 s->current_subframe++;
1368 }
1369 if (s->current_subframe >= s->subframes) {
1370#ifdef TRACE
1371 av_log(s->avctx, AV_LOG_DEBUG, "DSYNC dca_subframe_footer\n");
1372#endif
1373 /* Read subframe footer */
f44059d2
JR
1374 if ((ret = dca_subframe_footer(s, base_channel)))
1375 return ret;
01ca9ac3
KS
1376 }
1377
1378 return 0;
1379}
1380
1381/**
f5a2d285
AH
1382 * Return the number of channels in an ExSS speaker mask (HD)
1383 */
1384static int dca_exss_mask2count(int mask)
1385{
1386 /* count bits that mean speaker pairs twice */
f37b4efe
SG
1387 return av_popcount(mask) +
1388 av_popcount(mask & (DCA_EXSS_CENTER_LEFT_RIGHT |
1389 DCA_EXSS_FRONT_LEFT_RIGHT |
1390 DCA_EXSS_FRONT_HIGH_LEFT_RIGHT |
1391 DCA_EXSS_WIDE_LEFT_RIGHT |
1392 DCA_EXSS_SIDE_LEFT_RIGHT |
1393 DCA_EXSS_SIDE_HIGH_LEFT_RIGHT |
1394 DCA_EXSS_SIDE_REAR_LEFT_RIGHT |
1395 DCA_EXSS_REAR_LEFT_RIGHT |
1396 DCA_EXSS_REAR_HIGH_LEFT_RIGHT));
f5a2d285
AH
1397}
1398
1399/**
1400 * Skip mixing coefficients of a single mix out configuration (HD)
1401 */
1402static void dca_exss_skip_mix_coeffs(GetBitContext *gb, int channels, int out_ch)
1403{
e86e8581
GT
1404 int i;
1405
1406 for (i = 0; i < channels; i++) {
f5a2d285
AH
1407 int mix_map_mask = get_bits(gb, out_ch);
1408 int num_coeffs = av_popcount(mix_map_mask);
1409 skip_bits_long(gb, num_coeffs * 6);
1410 }
1411}
1412
1413/**
1414 * Parse extension substream asset header (HD)
1415 */
1416static int dca_exss_parse_asset_header(DCAContext *s)
1417{
1418 int header_pos = get_bits_count(&s->gb);
1419 int header_size;
1420 int channels;
1421 int embedded_stereo = 0;
f37b4efe 1422 int embedded_6ch = 0;
f5a2d285
AH
1423 int drc_code_present;
1424 int extensions_mask;
1425 int i, j;
1426
1427 if (get_bits_left(&s->gb) < 16)
1428 return -1;
1429
1430 /* We will parse just enough to get to the extensions bitmask with which
1431 * we can set the profile value. */
1432
1433 header_size = get_bits(&s->gb, 9) + 1;
1434 skip_bits(&s->gb, 3); // asset index
1435
1436 if (s->static_fields) {
1437 if (get_bits1(&s->gb))
1438 skip_bits(&s->gb, 4); // asset type descriptor
1439 if (get_bits1(&s->gb))
1440 skip_bits_long(&s->gb, 24); // language descriptor
1441
1442 if (get_bits1(&s->gb)) {
1443 /* How can one fit 1024 bytes of text here if the maximum value
1444 * for the asset header size field above was 512 bytes? */
1445 int text_length = get_bits(&s->gb, 10) + 1;
1446 if (get_bits_left(&s->gb) < text_length * 8)
1447 return -1;
1448 skip_bits_long(&s->gb, text_length * 8); // info text
1449 }
1450
1451 skip_bits(&s->gb, 5); // bit resolution - 1
1452 skip_bits(&s->gb, 4); // max sample rate code
1453 channels = get_bits(&s->gb, 8) + 1;
1454
1455 if (get_bits1(&s->gb)) { // 1-to-1 channels to speakers
1456 int spkr_remap_sets;
1457 int spkr_mask_size = 16;
1458 int num_spkrs[7];
1459
1460 if (channels > 2)
1461 embedded_stereo = get_bits1(&s->gb);
1462 if (channels > 6)
1463 embedded_6ch = get_bits1(&s->gb);
1464
1465 if (get_bits1(&s->gb)) {
1466 spkr_mask_size = (get_bits(&s->gb, 2) + 1) << 2;
1467 skip_bits(&s->gb, spkr_mask_size); // spkr activity mask
1468 }
1469
1470 spkr_remap_sets = get_bits(&s->gb, 3);
1471
1472 for (i = 0; i < spkr_remap_sets; i++) {
1473 /* std layout mask for each remap set */
1474 num_spkrs[i] = dca_exss_mask2count(get_bits(&s->gb, spkr_mask_size));
1475 }
1476
1477 for (i = 0; i < spkr_remap_sets; i++) {
1478 int num_dec_ch_remaps = get_bits(&s->gb, 5) + 1;
1479 if (get_bits_left(&s->gb) < 0)
1480 return -1;
1481
1482 for (j = 0; j < num_spkrs[i]; j++) {
1483 int remap_dec_ch_mask = get_bits_long(&s->gb, num_dec_ch_remaps);
1484 int num_dec_ch = av_popcount(remap_dec_ch_mask);
1485 skip_bits_long(&s->gb, num_dec_ch * 5); // remap codes
1486 }
1487 }
1488
1489 } else {
1490 skip_bits(&s->gb, 3); // representation type
1491 }
1492 }
1493
1494 drc_code_present = get_bits1(&s->gb);
1495 if (drc_code_present)
1496 get_bits(&s->gb, 8); // drc code
1497
1498 if (get_bits1(&s->gb))
1499 skip_bits(&s->gb, 5); // dialog normalization code
1500
1501 if (drc_code_present && embedded_stereo)
1502 get_bits(&s->gb, 8); // drc stereo code
1503
1504 if (s->mix_metadata && get_bits1(&s->gb)) {
1505 skip_bits(&s->gb, 1); // external mix
1506 skip_bits(&s->gb, 6); // post mix gain code
1507
1508 if (get_bits(&s->gb, 2) != 3) // mixer drc code
1509 skip_bits(&s->gb, 3); // drc limit
1510 else
1511 skip_bits(&s->gb, 8); // custom drc code
1512
1513 if (get_bits1(&s->gb)) // channel specific scaling
1514 for (i = 0; i < s->num_mix_configs; i++)
1515 skip_bits_long(&s->gb, s->mix_config_num_ch[i] * 6); // scale codes
1516 else
1517 skip_bits_long(&s->gb, s->num_mix_configs * 6); // scale codes
1518
1519 for (i = 0; i < s->num_mix_configs; i++) {
1520 if (get_bits_left(&s->gb) < 0)
1521 return -1;
1522 dca_exss_skip_mix_coeffs(&s->gb, channels, s->mix_config_num_ch[i]);
1523 if (embedded_6ch)
1524 dca_exss_skip_mix_coeffs(&s->gb, 6, s->mix_config_num_ch[i]);
1525 if (embedded_stereo)
1526 dca_exss_skip_mix_coeffs(&s->gb, 2, s->mix_config_num_ch[i]);
1527 }
1528 }
1529
1530 switch (get_bits(&s->gb, 2)) {
1531 case 0: extensions_mask = get_bits(&s->gb, 12); break;
1532 case 1: extensions_mask = DCA_EXT_EXSS_XLL; break;
1533 case 2: extensions_mask = DCA_EXT_EXSS_LBR; break;
1534 case 3: extensions_mask = 0; /* aux coding */ break;
1535 }
1536
1537 /* not parsed further, we were only interested in the extensions mask */
1538
1539 if (get_bits_left(&s->gb) < 0)
1540 return -1;
1541
1542 if (get_bits_count(&s->gb) - header_pos > header_size * 8) {
1543 av_log(s->avctx, AV_LOG_WARNING, "Asset header size mismatch.\n");
1544 return -1;
1545 }
1546 skip_bits_long(&s->gb, header_pos + header_size * 8 - get_bits_count(&s->gb));
1547
1548 if (extensions_mask & DCA_EXT_EXSS_XLL)
1549 s->profile = FF_PROFILE_DTS_HD_MA;
8f4a5d22
AH
1550 else if (extensions_mask & (DCA_EXT_EXSS_XBR | DCA_EXT_EXSS_X96 |
1551 DCA_EXT_EXSS_XXCH))
f5a2d285 1552 s->profile = FF_PROFILE_DTS_HD_HRA;
f5a2d285
AH
1553
1554 if (!(extensions_mask & DCA_EXT_CORE))
1555 av_log(s->avctx, AV_LOG_WARNING, "DTS core detection mismatch.\n");
7e06e0ed 1556 if ((extensions_mask & DCA_CORE_EXTS) != s->core_ext_mask)
f37b4efe
SG
1557 av_log(s->avctx, AV_LOG_WARNING,
1558 "DTS extensions detection mismatch (%d, %d)\n",
7e06e0ed 1559 extensions_mask & DCA_CORE_EXTS, s->core_ext_mask);
f5a2d285
AH
1560
1561 return 0;
1562}
1563
1564/**
1565 * Parse extension substream header (HD)
1566 */
1567static void dca_exss_parse_header(DCAContext *s)
1568{
1569 int ss_index;
1570 int blownup;
f5a2d285
AH
1571 int num_audiop = 1;
1572 int num_assets = 1;
1573 int active_ss_mask[8];
1574 int i, j;
1575
1576 if (get_bits_left(&s->gb) < 52)
1577 return;
1578
1579 skip_bits(&s->gb, 8); // user data
1580 ss_index = get_bits(&s->gb, 2);
1581
1582 blownup = get_bits1(&s->gb);
f37b4efe 1583 skip_bits(&s->gb, 8 + 4 * blownup); // header_size
e65ab9d9 1584 skip_bits(&s->gb, 16 + 4 * blownup); // hd_size
f5a2d285
AH
1585
1586 s->static_fields = get_bits1(&s->gb);
1587 if (s->static_fields) {
1588 skip_bits(&s->gb, 2); // reference clock code
1589 skip_bits(&s->gb, 3); // frame duration code
1590
1591 if (get_bits1(&s->gb))
1592 skip_bits_long(&s->gb, 36); // timestamp
1593
1594 /* a single stream can contain multiple audio assets that can be
1595 * combined to form multiple audio presentations */
1596
1597 num_audiop = get_bits(&s->gb, 3) + 1;
1598 if (num_audiop > 1) {
1599 av_log_ask_for_sample(s->avctx, "Multiple DTS-HD audio presentations.");
1600 /* ignore such streams for now */
1601 return;
1602 }
1603
1604 num_assets = get_bits(&s->gb, 3) + 1;
1605 if (num_assets > 1) {
1606 av_log_ask_for_sample(s->avctx, "Multiple DTS-HD audio assets.");
1607 /* ignore such streams for now */
1608 return;
1609 }
1610
1611 for (i = 0; i < num_audiop; i++)
1612 active_ss_mask[i] = get_bits(&s->gb, ss_index + 1);
1613
1614 for (i = 0; i < num_audiop; i++)
1615 for (j = 0; j <= ss_index; j++)
1616 if (active_ss_mask[i] & (1 << j))
1617 skip_bits(&s->gb, 8); // active asset mask
1618
1619 s->mix_metadata = get_bits1(&s->gb);
1620 if (s->mix_metadata) {
1621 int mix_out_mask_size;
1622
1623 skip_bits(&s->gb, 2); // adjustment level
f37b4efe
SG
1624 mix_out_mask_size = (get_bits(&s->gb, 2) + 1) << 2;
1625 s->num_mix_configs = get_bits(&s->gb, 2) + 1;
f5a2d285
AH
1626
1627 for (i = 0; i < s->num_mix_configs; i++) {
f37b4efe 1628 int mix_out_mask = get_bits(&s->gb, mix_out_mask_size);
f5a2d285
AH
1629 s->mix_config_num_ch[i] = dca_exss_mask2count(mix_out_mask);
1630 }
1631 }
1632 }
1633
1634 for (i = 0; i < num_assets; i++)
f37b4efe 1635 skip_bits_long(&s->gb, 16 + 4 * blownup); // asset size
f5a2d285
AH
1636
1637 for (i = 0; i < num_assets; i++) {
1638 if (dca_exss_parse_asset_header(s))
1639 return;
1640 }
1641
1642 /* not parsed further, we were only interested in the extensions mask
1643 * from the asset header */
1644}
1645
1646/**
01ca9ac3
KS
1647 * Main frame decoding function
1648 * FIXME add arguments
1649 */
0eea2129
JR
1650static int dca_decode_frame(AVCodecContext *avctx, void *data,
1651 int *got_frame_ptr, AVPacket *avpkt)
01ca9ac3 1652{
7a00bbad
TB
1653 const uint8_t *buf = avpkt->data;
1654 int buf_size = avpkt->size;
01ca9ac3 1655
6baef06e 1656 int lfe_samples;
774e9acf 1657 int num_core_channels = 0;
aae6eead 1658 int i, ret;
0eea2129
JR
1659 float *samples_flt;
1660 int16_t *samples_s16;
01ca9ac3
KS
1661 DCAContext *s = avctx->priv_data;
1662 int channels;
f5a2d285 1663 int core_ss_end;
01ca9ac3
KS
1664
1665
d0a18850 1666 s->xch_present = 0;
f5a2d285 1667
11ac796f
JR
1668 s->dca_buffer_size = ff_dca_convert_bitstream(buf, buf_size, s->dca_buffer,
1669 DCA_MAX_FRAME_SIZE + DCA_MAX_EXSS_HEADER_SIZE);
f44059d2 1670 if (s->dca_buffer_size == AVERROR_INVALIDDATA) {
56fd7cc5 1671 av_log(avctx, AV_LOG_ERROR, "Not a valid DCA frame\n");
f44059d2 1672 return AVERROR_INVALIDDATA;
01ca9ac3
KS
1673 }
1674
1675 init_get_bits(&s->gb, s->dca_buffer, s->dca_buffer_size * 8);
aae6eead 1676 if ((ret = dca_parse_frame_header(s)) < 0) {
01ca9ac3 1677 //seems like the frame is corrupt, try with the next one
aae6eead 1678 return ret;
01ca9ac3
KS
1679 }
1680 //set AVCodec values with parsed data
1681 avctx->sample_rate = s->sample_rate;
f37b4efe 1682 avctx->bit_rate = s->bit_rate;
01ca9ac3 1683
f5a2d285
AH
1684 s->profile = FF_PROFILE_DTS;
1685
6baef06e 1686 for (i = 0; i < (s->sample_blocks / 8); i++) {
272fcc32
JR
1687 if ((ret = dca_decode_block(s, 0, i))) {
1688 av_log(avctx, AV_LOG_ERROR, "error decoding block\n");
1689 return ret;
1690 }
774e9acf
NB
1691 }
1692
1693 /* record number of core channels incase less than max channels are requested */
1694 num_core_channels = s->prim_channels;
1695
7e06e0ed
AH
1696 if (s->ext_coding)
1697 s->core_ext_mask = dca_ext_audio_descr_mask[s->ext_descr];
1698 else
1699 s->core_ext_mask = 0;
774e9acf 1700
f5a2d285
AH
1701 core_ss_end = FFMIN(s->frame_size, s->dca_buffer_size) * 8;
1702
7e06e0ed
AH
1703 /* only scan for extensions if ext_descr was unknown or indicated a
1704 * supported XCh extension */
1705 if (s->core_ext_mask < 0 || s->core_ext_mask & DCA_EXT_XCH) {
1706
1707 /* if ext_descr was unknown, clear s->core_ext_mask so that the
1708 * extensions scan can fill it up */
1709 s->core_ext_mask = FFMAX(s->core_ext_mask, 0);
1710
1711 /* extensions start at 32-bit boundaries into bitstream */
1712 skip_bits_long(&s->gb, (-get_bits_count(&s->gb)) & 31);
1713
f37b4efe
SG
1714 while (core_ss_end - get_bits_count(&s->gb) >= 32) {
1715 uint32_t bits = get_bits_long(&s->gb, 32);
774e9acf 1716
f37b4efe
SG
1717 switch (bits) {
1718 case 0x5a5a5a5a: {
1719 int ext_amode, xch_fsize;
0712c230 1720
f37b4efe 1721 s->xch_base_channel = s->prim_channels;
d0a18850 1722
f37b4efe
SG
1723 /* validate sync word using XCHFSIZE field */
1724 xch_fsize = show_bits(&s->gb, 10);
1725 if ((s->frame_size != (get_bits_count(&s->gb) >> 3) - 4 + xch_fsize) &&
1726 (s->frame_size != (get_bits_count(&s->gb) >> 3) - 4 + xch_fsize + 1))
1727 continue;
f5a2d285 1728
f37b4efe
SG
1729 /* skip length-to-end-of-frame field for the moment */
1730 skip_bits(&s->gb, 10);
774e9acf 1731
f37b4efe 1732 s->core_ext_mask |= DCA_EXT_XCH;
774e9acf 1733
f37b4efe
SG
1734 /* extension amode(number of channels in extension) should be 1 */
1735 /* AFAIK XCh is not used for more channels */
1736 if ((ext_amode = get_bits(&s->gb, 4)) != 1) {
1737 av_log(avctx, AV_LOG_ERROR, "XCh extension amode %d not"
1738 " supported!\n", ext_amode);
272fcc32
JR
1739 continue;
1740 }
f37b4efe
SG
1741
1742 /* much like core primary audio coding header */
1743 dca_parse_audio_coding_header(s, s->xch_base_channel);
1744
1745 for (i = 0; i < (s->sample_blocks / 8); i++)
1746 if ((ret = dca_decode_block(s, s->xch_base_channel, i))) {
1747 av_log(avctx, AV_LOG_ERROR, "error decoding XCh extension\n");
1748 continue;
1749 }
1750
1751 s->xch_present = 1;
1752 break;
774e9acf 1753 }
f37b4efe
SG
1754 case 0x47004a03:
1755 /* XXCh: extended channels */
1756 /* usually found either in core or HD part in DTS-HD HRA streams,
1757 * but not in DTS-ES which contains XCh extensions instead */
1758 s->core_ext_mask |= DCA_EXT_XXCH;
1759 break;
1760
1761 case 0x1d95f262: {
1762 int fsize96 = show_bits(&s->gb, 12) + 1;
1763 if (s->frame_size != (get_bits_count(&s->gb) >> 3) - 4 + fsize96)
1764 continue;
774e9acf 1765
f37b4efe
SG
1766 av_log(avctx, AV_LOG_DEBUG, "X96 extension found at %d bits\n",
1767 get_bits_count(&s->gb));
1768 skip_bits(&s->gb, 12);
1769 av_log(avctx, AV_LOG_DEBUG, "FSIZE96 = %d bytes\n", fsize96);
1770 av_log(avctx, AV_LOG_DEBUG, "REVNO = %d\n", get_bits(&s->gb, 4));
774e9acf 1771
f37b4efe
SG
1772 s->core_ext_mask |= DCA_EXT_X96;
1773 break;
1774 }
1775 }
6baef06e 1776
f37b4efe
SG
1777 skip_bits_long(&s->gb, (-get_bits_count(&s->gb)) & 31);
1778 }
7e06e0ed
AH
1779 } else {
1780 /* no supported extensions, skip the rest of the core substream */
1781 skip_bits_long(&s->gb, core_ss_end - get_bits_count(&s->gb));
1782 }
1783
1784 if (s->core_ext_mask & DCA_EXT_X96)
1785 s->profile = FF_PROFILE_DTS_96_24;
1786 else if (s->core_ext_mask & (DCA_EXT_XCH | DCA_EXT_XXCH))
1787 s->profile = FF_PROFILE_DTS_ES;
1788
f5a2d285 1789 /* check for ExSS (HD part) */
f37b4efe
SG
1790 if (s->dca_buffer_size - s->frame_size > 32 &&
1791 get_bits_long(&s->gb, 32) == DCA_HD_MARKER)
f5a2d285
AH
1792 dca_exss_parse_header(s);
1793
1794 avctx->profile = s->profile;
1795
cc826626 1796 channels = s->prim_channels + !!s->lfe;
92765276 1797
f37b4efe 1798 if (s->amode < 16) {
87c3b9bc 1799 avctx->channel_layout = dca_core_channel_layout[s->amode];
87c3b9bc 1800
d0a18850 1801 if (s->xch_present && (!avctx->request_channels ||
08634e7b 1802 avctx->request_channels > num_core_channels + !!s->lfe)) {
63e8d976 1803 avctx->channel_layout |= AV_CH_BACK_CENTER;
774e9acf 1804 if (s->lfe) {
63e8d976 1805 avctx->channel_layout |= AV_CH_LOW_FREQUENCY;
774e9acf
NB
1806 s->channel_order_tab = dca_channel_reorder_lfe_xch[s->amode];
1807 } else {
1808 s->channel_order_tab = dca_channel_reorder_nolfe_xch[s->amode];
1809 }
1810 } else {
08634e7b
NB
1811 channels = num_core_channels + !!s->lfe;
1812 s->xch_present = 0; /* disable further xch processing */
774e9acf 1813 if (s->lfe) {
63e8d976 1814 avctx->channel_layout |= AV_CH_LOW_FREQUENCY;
774e9acf
NB
1815 s->channel_order_tab = dca_channel_reorder_lfe[s->amode];
1816 } else
1817 s->channel_order_tab = dca_channel_reorder_nolfe[s->amode];
1818 }
92765276 1819
08634e7b
NB
1820 if (channels > !!s->lfe &&
1821 s->channel_order_tab[channels - 1 - !!s->lfe] < 0)
f44059d2 1822 return AVERROR_INVALIDDATA;
4e06acbd 1823
d1177cb5 1824 if (avctx->request_channels == 2 && s->prim_channels > 2) {
92765276
BL
1825 channels = 2;
1826 s->output = DCA_STEREO;
63e8d976 1827 avctx->channel_layout = AV_CH_LAYOUT_STEREO;
92765276
BL
1828 }
1829 } else {
f37b4efe 1830 av_log(avctx, AV_LOG_ERROR, "Non standard configuration %d !\n", s->amode);
f44059d2 1831 return AVERROR_INVALIDDATA;
92765276
BL
1832 }
1833
cc826626 1834
4a24837e 1835 /* There is nothing that prevents a dts frame to change channel configuration
43fb279f 1836 but Libav doesn't support that so only set the channels if it is previously
4a24837e
BL
1837 unset. Ideally during the first probe for channels the crc should be checked
1838 and only set avctx->channels when the crc is ok. Right now the decoder could
1839 set the channels based on a broken first frame.*/
1360f07e
DK
1840 if (s->is_channels_set == 0) {
1841 s->is_channels_set = 1;
d425a03b 1842 avctx->channels = channels;
1360f07e
DK
1843 }
1844 if (avctx->channels != channels) {
1845 av_log(avctx, AV_LOG_ERROR, "DCA decoder does not support number of "
1846 "channels changing in stream. Skipping frame.\n");
f44059d2 1847 return AVERROR_PATCHWELCOME;
1360f07e 1848 }
4a24837e 1849
0eea2129
JR
1850 /* get output buffer */
1851 s->frame.nb_samples = 256 * (s->sample_blocks / 8);
1852 if ((ret = avctx->get_buffer(avctx, &s->frame)) < 0) {
1853 av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
1854 return ret;
1855 }
f37b4efe
SG
1856 samples_flt = (float *) s->frame.data[0];
1857 samples_s16 = (int16_t *) s->frame.data[0];
6baef06e
NB
1858
1859 /* filter to get final output */
01ca9ac3 1860 for (i = 0; i < (s->sample_blocks / 8); i++) {
6baef06e 1861 dca_filter_channels(s, i);
b12b16c5
NB
1862
1863 /* If this was marked as a DTS-ES stream we need to subtract back- */
1864 /* channel from SL & SR to remove matrixed back-channel signal */
f37b4efe
SG
1865 if ((s->source_pcm_res & 1) && s->xch_present) {
1866 float *back_chan = s->samples + s->channel_order_tab[s->xch_base_channel] * 256;
1867 float *lt_chan = s->samples + s->channel_order_tab[s->xch_base_channel - 2] * 256;
1868 float *rt_chan = s->samples + s->channel_order_tab[s->xch_base_channel - 1] * 256;
cb5042d0
JR
1869 s->fdsp.vector_fmac_scalar(lt_chan, back_chan, -M_SQRT1_2, 256);
1870 s->fdsp.vector_fmac_scalar(rt_chan, back_chan, -M_SQRT1_2, 256);
b12b16c5
NB
1871 }
1872
9aa8193a
JR
1873 if (avctx->sample_fmt == AV_SAMPLE_FMT_FLT) {
1874 s->fmt_conv.float_interleave(samples_flt, s->samples_chanptr, 256,
1875 channels);
1876 samples_flt += 256 * channels;
1877 } else {
1878 s->fmt_conv.float_to_int16_interleave(samples_s16,
1879 s->samples_chanptr, 256,
1880 channels);
1881 samples_s16 += 256 * channels;
1882 }
01ca9ac3
KS
1883 }
1884
6baef06e
NB
1885 /* update lfe history */
1886 lfe_samples = 2 * s->lfe * (s->sample_blocks / 8);
f37b4efe 1887 for (i = 0; i < 2 * s->lfe * 4; i++)
6baef06e 1888 s->lfe_data[i] = s->lfe_data[i + lfe_samples];
6baef06e 1889
f37b4efe
SG
1890 *got_frame_ptr = 1;
1891 *(AVFrame *) data = s->frame;
0eea2129 1892
01ca9ac3
KS
1893 return buf_size;
1894}
1895
1896
1897
1898/**
01ca9ac3
KS
1899 * DCA initialization
1900 *
1901 * @param avctx pointer to the AVCodecContext
1902 */
1903
f37b4efe 1904static av_cold int dca_decode_init(AVCodecContext *avctx)
01ca9ac3
KS
1905{
1906 DCAContext *s = avctx->priv_data;
8ed96a5e 1907 int i;
01ca9ac3
KS
1908
1909 s->avctx = avctx;
1910 dca_init_vlcs();
01ca9ac3 1911
cb5042d0 1912 avpriv_float_dsp_init(&s->fdsp, avctx->flags & CODEC_FLAG_BITEXACT);
7d485f16 1913 ff_mdct_init(&s->imdct, 6, 1, 1.0);
f462ed1f 1914 ff_synth_filter_init(&s->synth);
309d16a4 1915 ff_dcadsp_init(&s->dcadsp);
c73d99e6 1916 ff_fmt_convert_init(&s->fmt_conv, avctx);
cfec09e9 1917
f37b4efe 1918 for (i = 0; i < DCA_PRIM_CHANNELS_MAX + 1; i++)
8ed96a5e 1919 s->samples_chanptr[i] = s->samples + i * 256;
1402ee72 1920
9aa8193a
JR
1921 if (avctx->request_sample_fmt == AV_SAMPLE_FMT_FLT) {
1922 avctx->sample_fmt = AV_SAMPLE_FMT_FLT;
f37b4efe 1923 s->scale_bias = 1.0 / 32768.0;
9aa8193a
JR
1924 } else {
1925 avctx->sample_fmt = AV_SAMPLE_FMT_S16;
f37b4efe 1926 s->scale_bias = 1.0;
9aa8193a 1927 }
1402ee72 1928
b5ec6383
JR
1929 /* allow downmixing to stereo */
1930 if (avctx->channels > 0 && avctx->request_channels < avctx->channels &&
1931 avctx->request_channels == 2) {
1932 avctx->channels = avctx->request_channels;
1933 }
1402ee72 1934
0eea2129
JR
1935 avcodec_get_frame_defaults(&s->frame);
1936 avctx->coded_frame = &s->frame;
1937
01ca9ac3
KS
1938 return 0;
1939}
1940
f37b4efe 1941static av_cold int dca_decode_end(AVCodecContext *avctx)
89df5e95
MN
1942{
1943 DCAContext *s = avctx->priv_data;
1944 ff_mdct_end(&s->imdct);
1945 return 0;
1946}
01ca9ac3 1947
f4096bf6
AH
1948static const AVProfile profiles[] = {
1949 { FF_PROFILE_DTS, "DTS" },
1950 { FF_PROFILE_DTS_ES, "DTS-ES" },
1951 { FF_PROFILE_DTS_96_24, "DTS 96/24" },
1952 { FF_PROFILE_DTS_HD_HRA, "DTS-HD HRA" },
1953 { FF_PROFILE_DTS_HD_MA, "DTS-HD MA" },
1954 { FF_PROFILE_UNKNOWN },
1955};
1956
d36beb3f 1957AVCodec ff_dca_decoder = {
f37b4efe
SG
1958 .name = "dca",
1959 .type = AVMEDIA_TYPE_AUDIO,
36ef5369 1960 .id = AV_CODEC_ID_DTS,
f37b4efe
SG
1961 .priv_data_size = sizeof(DCAContext),
1962 .init = dca_decode_init,
1963 .decode = dca_decode_frame,
1964 .close = dca_decode_end,
1965 .long_name = NULL_IF_CONFIG_SMALL("DCA (DTS Coherent Acoustics)"),
1966 .capabilities = CODEC_CAP_CHANNEL_CONF | CODEC_CAP_DR1,
1967 .sample_fmts = (const enum AVSampleFormat[]) { AV_SAMPLE_FMT_FLT,
1968 AV_SAMPLE_FMT_S16,
1969 AV_SAMPLE_FMT_NONE },
1970 .profiles = NULL_IF_CONFIG_SMALL(profiles),
01ca9ac3 1971};