yading@10
|
1 /*
|
yading@10
|
2 * MLP decoder
|
yading@10
|
3 * Copyright (c) 2007-2008 Ian Caulfield
|
yading@10
|
4 *
|
yading@10
|
5 * This file is part of FFmpeg.
|
yading@10
|
6 *
|
yading@10
|
7 * FFmpeg is free software; you can redistribute it and/or
|
yading@10
|
8 * modify it under the terms of the GNU Lesser General Public
|
yading@10
|
9 * License as published by the Free Software Foundation; either
|
yading@10
|
10 * version 2.1 of the License, or (at your option) any later version.
|
yading@10
|
11 *
|
yading@10
|
12 * FFmpeg is distributed in the hope that it will be useful,
|
yading@10
|
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
yading@10
|
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
yading@10
|
15 * Lesser General Public License for more details.
|
yading@10
|
16 *
|
yading@10
|
17 * You should have received a copy of the GNU Lesser General Public
|
yading@10
|
18 * License along with FFmpeg; if not, write to the Free Software
|
yading@10
|
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
yading@10
|
20 */
|
yading@10
|
21
|
yading@10
|
22 /**
|
yading@10
|
23 * @file
|
yading@10
|
24 * MLP decoder
|
yading@10
|
25 */
|
yading@10
|
26
|
yading@10
|
27 #include <stdint.h>
|
yading@10
|
28
|
yading@10
|
29 #include "avcodec.h"
|
yading@10
|
30 #include "libavutil/intreadwrite.h"
|
yading@10
|
31 #include "libavutil/channel_layout.h"
|
yading@10
|
32 #include "get_bits.h"
|
yading@10
|
33 #include "internal.h"
|
yading@10
|
34 #include "libavutil/crc.h"
|
yading@10
|
35 #include "parser.h"
|
yading@10
|
36 #include "mlp_parser.h"
|
yading@10
|
37 #include "mlpdsp.h"
|
yading@10
|
38 #include "mlp.h"
|
yading@10
|
39
|
yading@10
|
40 /** number of bits used for VLC lookup - longest Huffman code is 9 */
|
yading@10
|
41 #define VLC_BITS 9
|
yading@10
|
42
|
yading@10
|
43 typedef struct SubStream {
|
yading@10
|
44 /// Set if a valid restart header has been read. Otherwise the substream cannot be decoded.
|
yading@10
|
45 uint8_t restart_seen;
|
yading@10
|
46
|
yading@10
|
47 //@{
|
yading@10
|
48 /** restart header data */
|
yading@10
|
49 /// The type of noise to be used in the rematrix stage.
|
yading@10
|
50 uint16_t noise_type;
|
yading@10
|
51
|
yading@10
|
52 /// The index of the first channel coded in this substream.
|
yading@10
|
53 uint8_t min_channel;
|
yading@10
|
54 /// The index of the last channel coded in this substream.
|
yading@10
|
55 uint8_t max_channel;
|
yading@10
|
56 /// The number of channels input into the rematrix stage.
|
yading@10
|
57 uint8_t max_matrix_channel;
|
yading@10
|
58 /// For each channel output by the matrix, the output channel to map it to
|
yading@10
|
59 uint8_t ch_assign[MAX_CHANNELS];
|
yading@10
|
60 /// The channel layout for this substream
|
yading@10
|
61 uint64_t ch_layout;
|
yading@10
|
62
|
yading@10
|
63 /// Channel coding parameters for channels in the substream
|
yading@10
|
64 ChannelParams channel_params[MAX_CHANNELS];
|
yading@10
|
65
|
yading@10
|
66 /// The left shift applied to random noise in 0x31ea substreams.
|
yading@10
|
67 uint8_t noise_shift;
|
yading@10
|
68 /// The current seed value for the pseudorandom noise generator(s).
|
yading@10
|
69 uint32_t noisegen_seed;
|
yading@10
|
70
|
yading@10
|
71 /// Set if the substream contains extra info to check the size of VLC blocks.
|
yading@10
|
72 uint8_t data_check_present;
|
yading@10
|
73
|
yading@10
|
74 /// Bitmask of which parameter sets are conveyed in a decoding parameter block.
|
yading@10
|
75 uint8_t param_presence_flags;
|
yading@10
|
76 #define PARAM_BLOCKSIZE (1 << 7)
|
yading@10
|
77 #define PARAM_MATRIX (1 << 6)
|
yading@10
|
78 #define PARAM_OUTSHIFT (1 << 5)
|
yading@10
|
79 #define PARAM_QUANTSTEP (1 << 4)
|
yading@10
|
80 #define PARAM_FIR (1 << 3)
|
yading@10
|
81 #define PARAM_IIR (1 << 2)
|
yading@10
|
82 #define PARAM_HUFFOFFSET (1 << 1)
|
yading@10
|
83 #define PARAM_PRESENCE (1 << 0)
|
yading@10
|
84 //@}
|
yading@10
|
85
|
yading@10
|
86 //@{
|
yading@10
|
87 /** matrix data */
|
yading@10
|
88
|
yading@10
|
89 /// Number of matrices to be applied.
|
yading@10
|
90 uint8_t num_primitive_matrices;
|
yading@10
|
91
|
yading@10
|
92 /// matrix output channel
|
yading@10
|
93 uint8_t matrix_out_ch[MAX_MATRICES];
|
yading@10
|
94
|
yading@10
|
95 /// Whether the LSBs of the matrix output are encoded in the bitstream.
|
yading@10
|
96 uint8_t lsb_bypass[MAX_MATRICES];
|
yading@10
|
97 /// Matrix coefficients, stored as 2.14 fixed point.
|
yading@10
|
98 int32_t matrix_coeff[MAX_MATRICES][MAX_CHANNELS];
|
yading@10
|
99 /// Left shift to apply to noise values in 0x31eb substreams.
|
yading@10
|
100 uint8_t matrix_noise_shift[MAX_MATRICES];
|
yading@10
|
101 //@}
|
yading@10
|
102
|
yading@10
|
103 /// Left shift to apply to Huffman-decoded residuals.
|
yading@10
|
104 uint8_t quant_step_size[MAX_CHANNELS];
|
yading@10
|
105
|
yading@10
|
106 /// number of PCM samples in current audio block
|
yading@10
|
107 uint16_t blocksize;
|
yading@10
|
108 /// Number of PCM samples decoded so far in this frame.
|
yading@10
|
109 uint16_t blockpos;
|
yading@10
|
110
|
yading@10
|
111 /// Left shift to apply to decoded PCM values to get final 24-bit output.
|
yading@10
|
112 int8_t output_shift[MAX_CHANNELS];
|
yading@10
|
113
|
yading@10
|
114 /// Running XOR of all output samples.
|
yading@10
|
115 int32_t lossless_check_data;
|
yading@10
|
116
|
yading@10
|
117 } SubStream;
|
yading@10
|
118
|
yading@10
|
119 typedef struct MLPDecodeContext {
|
yading@10
|
120 AVCodecContext *avctx;
|
yading@10
|
121
|
yading@10
|
122 /// Current access unit being read has a major sync.
|
yading@10
|
123 int is_major_sync_unit;
|
yading@10
|
124
|
yading@10
|
125 /// Set if a valid major sync block has been read. Otherwise no decoding is possible.
|
yading@10
|
126 uint8_t params_valid;
|
yading@10
|
127
|
yading@10
|
128 /// Number of substreams contained within this stream.
|
yading@10
|
129 uint8_t num_substreams;
|
yading@10
|
130
|
yading@10
|
131 /// Index of the last substream to decode - further substreams are skipped.
|
yading@10
|
132 uint8_t max_decoded_substream;
|
yading@10
|
133
|
yading@10
|
134 /// Stream needs channel reordering to comply with FFmpeg's channel order
|
yading@10
|
135 uint8_t needs_reordering;
|
yading@10
|
136
|
yading@10
|
137 /// number of PCM samples contained in each frame
|
yading@10
|
138 int access_unit_size;
|
yading@10
|
139 /// next power of two above the number of samples in each frame
|
yading@10
|
140 int access_unit_size_pow2;
|
yading@10
|
141
|
yading@10
|
142 SubStream substream[MAX_SUBSTREAMS];
|
yading@10
|
143
|
yading@10
|
144 int matrix_changed;
|
yading@10
|
145 int filter_changed[MAX_CHANNELS][NUM_FILTERS];
|
yading@10
|
146
|
yading@10
|
147 int8_t noise_buffer[MAX_BLOCKSIZE_POW2];
|
yading@10
|
148 int8_t bypassed_lsbs[MAX_BLOCKSIZE][MAX_CHANNELS];
|
yading@10
|
149 int32_t sample_buffer[MAX_BLOCKSIZE][MAX_CHANNELS];
|
yading@10
|
150
|
yading@10
|
151 MLPDSPContext dsp;
|
yading@10
|
152 } MLPDecodeContext;
|
yading@10
|
153
|
yading@10
|
154 static const uint64_t thd_channel_order[] = {
|
yading@10
|
155 AV_CH_FRONT_LEFT, AV_CH_FRONT_RIGHT, // LR
|
yading@10
|
156 AV_CH_FRONT_CENTER, // C
|
yading@10
|
157 AV_CH_LOW_FREQUENCY, // LFE
|
yading@10
|
158 AV_CH_SIDE_LEFT, AV_CH_SIDE_RIGHT, // LRs
|
yading@10
|
159 AV_CH_TOP_FRONT_LEFT, AV_CH_TOP_FRONT_RIGHT, // LRvh
|
yading@10
|
160 AV_CH_FRONT_LEFT_OF_CENTER, AV_CH_FRONT_RIGHT_OF_CENTER, // LRc
|
yading@10
|
161 AV_CH_BACK_LEFT, AV_CH_BACK_RIGHT, // LRrs
|
yading@10
|
162 AV_CH_BACK_CENTER, // Cs
|
yading@10
|
163 AV_CH_TOP_CENTER, // Ts
|
yading@10
|
164 AV_CH_SURROUND_DIRECT_LEFT, AV_CH_SURROUND_DIRECT_RIGHT, // LRsd
|
yading@10
|
165 AV_CH_WIDE_LEFT, AV_CH_WIDE_RIGHT, // LRw
|
yading@10
|
166 AV_CH_TOP_FRONT_CENTER, // Cvh
|
yading@10
|
167 AV_CH_LOW_FREQUENCY_2, // LFE2
|
yading@10
|
168 };
|
yading@10
|
169
|
yading@10
|
170 static uint64_t thd_channel_layout_extract_channel(uint64_t channel_layout,
|
yading@10
|
171 int index)
|
yading@10
|
172 {
|
yading@10
|
173 int i;
|
yading@10
|
174
|
yading@10
|
175 if (av_get_channel_layout_nb_channels(channel_layout) <= index)
|
yading@10
|
176 return 0;
|
yading@10
|
177
|
yading@10
|
178 for (i = 0; i < FF_ARRAY_ELEMS(thd_channel_order); i++)
|
yading@10
|
179 if (channel_layout & thd_channel_order[i] && !index--)
|
yading@10
|
180 return thd_channel_order[i];
|
yading@10
|
181 return 0;
|
yading@10
|
182 }
|
yading@10
|
183
|
yading@10
|
184 static VLC huff_vlc[3];
|
yading@10
|
185
|
yading@10
|
186 /** Initialize static data, constant between all invocations of the codec. */
|
yading@10
|
187
|
yading@10
|
188 static av_cold void init_static(void)
|
yading@10
|
189 {
|
yading@10
|
190 if (!huff_vlc[0].bits) {
|
yading@10
|
191 INIT_VLC_STATIC(&huff_vlc[0], VLC_BITS, 18,
|
yading@10
|
192 &ff_mlp_huffman_tables[0][0][1], 2, 1,
|
yading@10
|
193 &ff_mlp_huffman_tables[0][0][0], 2, 1, 512);
|
yading@10
|
194 INIT_VLC_STATIC(&huff_vlc[1], VLC_BITS, 16,
|
yading@10
|
195 &ff_mlp_huffman_tables[1][0][1], 2, 1,
|
yading@10
|
196 &ff_mlp_huffman_tables[1][0][0], 2, 1, 512);
|
yading@10
|
197 INIT_VLC_STATIC(&huff_vlc[2], VLC_BITS, 15,
|
yading@10
|
198 &ff_mlp_huffman_tables[2][0][1], 2, 1,
|
yading@10
|
199 &ff_mlp_huffman_tables[2][0][0], 2, 1, 512);
|
yading@10
|
200 }
|
yading@10
|
201
|
yading@10
|
202 ff_mlp_init_crc();
|
yading@10
|
203 }
|
yading@10
|
204
|
yading@10
|
205 static inline int32_t calculate_sign_huff(MLPDecodeContext *m,
|
yading@10
|
206 unsigned int substr, unsigned int ch)
|
yading@10
|
207 {
|
yading@10
|
208 SubStream *s = &m->substream[substr];
|
yading@10
|
209 ChannelParams *cp = &s->channel_params[ch];
|
yading@10
|
210 int lsb_bits = cp->huff_lsbs - s->quant_step_size[ch];
|
yading@10
|
211 int sign_shift = lsb_bits + (cp->codebook ? 2 - cp->codebook : -1);
|
yading@10
|
212 int32_t sign_huff_offset = cp->huff_offset;
|
yading@10
|
213
|
yading@10
|
214 if (cp->codebook > 0)
|
yading@10
|
215 sign_huff_offset -= 7 << lsb_bits;
|
yading@10
|
216
|
yading@10
|
217 if (sign_shift >= 0)
|
yading@10
|
218 sign_huff_offset -= 1 << sign_shift;
|
yading@10
|
219
|
yading@10
|
220 return sign_huff_offset;
|
yading@10
|
221 }
|
yading@10
|
222
|
yading@10
|
223 /** Read a sample, consisting of either, both or neither of entropy-coded MSBs
|
yading@10
|
224 * and plain LSBs. */
|
yading@10
|
225
|
yading@10
|
226 static inline int read_huff_channels(MLPDecodeContext *m, GetBitContext *gbp,
|
yading@10
|
227 unsigned int substr, unsigned int pos)
|
yading@10
|
228 {
|
yading@10
|
229 SubStream *s = &m->substream[substr];
|
yading@10
|
230 unsigned int mat, channel;
|
yading@10
|
231
|
yading@10
|
232 for (mat = 0; mat < s->num_primitive_matrices; mat++)
|
yading@10
|
233 if (s->lsb_bypass[mat])
|
yading@10
|
234 m->bypassed_lsbs[pos + s->blockpos][mat] = get_bits1(gbp);
|
yading@10
|
235
|
yading@10
|
236 for (channel = s->min_channel; channel <= s->max_channel; channel++) {
|
yading@10
|
237 ChannelParams *cp = &s->channel_params[channel];
|
yading@10
|
238 int codebook = cp->codebook;
|
yading@10
|
239 int quant_step_size = s->quant_step_size[channel];
|
yading@10
|
240 int lsb_bits = cp->huff_lsbs - quant_step_size;
|
yading@10
|
241 int result = 0;
|
yading@10
|
242
|
yading@10
|
243 if (codebook > 0)
|
yading@10
|
244 result = get_vlc2(gbp, huff_vlc[codebook-1].table,
|
yading@10
|
245 VLC_BITS, (9 + VLC_BITS - 1) / VLC_BITS);
|
yading@10
|
246
|
yading@10
|
247 if (result < 0)
|
yading@10
|
248 return AVERROR_INVALIDDATA;
|
yading@10
|
249
|
yading@10
|
250 if (lsb_bits > 0)
|
yading@10
|
251 result = (result << lsb_bits) + get_bits(gbp, lsb_bits);
|
yading@10
|
252
|
yading@10
|
253 result += cp->sign_huff_offset;
|
yading@10
|
254 result <<= quant_step_size;
|
yading@10
|
255
|
yading@10
|
256 m->sample_buffer[pos + s->blockpos][channel] = result;
|
yading@10
|
257 }
|
yading@10
|
258
|
yading@10
|
259 return 0;
|
yading@10
|
260 }
|
yading@10
|
261
|
yading@10
|
262 static av_cold int mlp_decode_init(AVCodecContext *avctx)
|
yading@10
|
263 {
|
yading@10
|
264 MLPDecodeContext *m = avctx->priv_data;
|
yading@10
|
265 int substr;
|
yading@10
|
266
|
yading@10
|
267 init_static();
|
yading@10
|
268 m->avctx = avctx;
|
yading@10
|
269 for (substr = 0; substr < MAX_SUBSTREAMS; substr++)
|
yading@10
|
270 m->substream[substr].lossless_check_data = 0xffffffff;
|
yading@10
|
271 ff_mlpdsp_init(&m->dsp);
|
yading@10
|
272
|
yading@10
|
273 return 0;
|
yading@10
|
274 }
|
yading@10
|
275
|
yading@10
|
276 /** Read a major sync info header - contains high level information about
|
yading@10
|
277 * the stream - sample rate, channel arrangement etc. Most of this
|
yading@10
|
278 * information is not actually necessary for decoding, only for playback.
|
yading@10
|
279 */
|
yading@10
|
280
|
yading@10
|
281 static int read_major_sync(MLPDecodeContext *m, GetBitContext *gb)
|
yading@10
|
282 {
|
yading@10
|
283 MLPHeaderInfo mh;
|
yading@10
|
284 int substr, ret;
|
yading@10
|
285
|
yading@10
|
286 if ((ret = ff_mlp_read_major_sync(m->avctx, &mh, gb)) != 0)
|
yading@10
|
287 return ret;
|
yading@10
|
288
|
yading@10
|
289 if (mh.group1_bits == 0) {
|
yading@10
|
290 av_log(m->avctx, AV_LOG_ERROR, "invalid/unknown bits per sample\n");
|
yading@10
|
291 return AVERROR_INVALIDDATA;
|
yading@10
|
292 }
|
yading@10
|
293 if (mh.group2_bits > mh.group1_bits) {
|
yading@10
|
294 av_log(m->avctx, AV_LOG_ERROR,
|
yading@10
|
295 "Channel group 2 cannot have more bits per sample than group 1.\n");
|
yading@10
|
296 return AVERROR_INVALIDDATA;
|
yading@10
|
297 }
|
yading@10
|
298
|
yading@10
|
299 if (mh.group2_samplerate && mh.group2_samplerate != mh.group1_samplerate) {
|
yading@10
|
300 av_log(m->avctx, AV_LOG_ERROR,
|
yading@10
|
301 "Channel groups with differing sample rates are not currently supported.\n");
|
yading@10
|
302 return AVERROR_INVALIDDATA;
|
yading@10
|
303 }
|
yading@10
|
304
|
yading@10
|
305 if (mh.group1_samplerate == 0) {
|
yading@10
|
306 av_log(m->avctx, AV_LOG_ERROR, "invalid/unknown sampling rate\n");
|
yading@10
|
307 return AVERROR_INVALIDDATA;
|
yading@10
|
308 }
|
yading@10
|
309 if (mh.group1_samplerate > MAX_SAMPLERATE) {
|
yading@10
|
310 av_log(m->avctx, AV_LOG_ERROR,
|
yading@10
|
311 "Sampling rate %d is greater than the supported maximum (%d).\n",
|
yading@10
|
312 mh.group1_samplerate, MAX_SAMPLERATE);
|
yading@10
|
313 return AVERROR_INVALIDDATA;
|
yading@10
|
314 }
|
yading@10
|
315 if (mh.access_unit_size > MAX_BLOCKSIZE) {
|
yading@10
|
316 av_log(m->avctx, AV_LOG_ERROR,
|
yading@10
|
317 "Block size %d is greater than the supported maximum (%d).\n",
|
yading@10
|
318 mh.access_unit_size, MAX_BLOCKSIZE);
|
yading@10
|
319 return AVERROR_INVALIDDATA;
|
yading@10
|
320 }
|
yading@10
|
321 if (mh.access_unit_size_pow2 > MAX_BLOCKSIZE_POW2) {
|
yading@10
|
322 av_log(m->avctx, AV_LOG_ERROR,
|
yading@10
|
323 "Block size pow2 %d is greater than the supported maximum (%d).\n",
|
yading@10
|
324 mh.access_unit_size_pow2, MAX_BLOCKSIZE_POW2);
|
yading@10
|
325 return AVERROR_INVALIDDATA;
|
yading@10
|
326 }
|
yading@10
|
327
|
yading@10
|
328 if (mh.num_substreams == 0)
|
yading@10
|
329 return AVERROR_INVALIDDATA;
|
yading@10
|
330 if (m->avctx->codec_id == AV_CODEC_ID_MLP && mh.num_substreams > 2) {
|
yading@10
|
331 av_log(m->avctx, AV_LOG_ERROR, "MLP only supports up to 2 substreams.\n");
|
yading@10
|
332 return AVERROR_INVALIDDATA;
|
yading@10
|
333 }
|
yading@10
|
334 if (mh.num_substreams > MAX_SUBSTREAMS) {
|
yading@10
|
335 avpriv_request_sample(m->avctx,
|
yading@10
|
336 "%d substreams (more than the "
|
yading@10
|
337 "maximum supported by the decoder)",
|
yading@10
|
338 mh.num_substreams);
|
yading@10
|
339 return AVERROR_PATCHWELCOME;
|
yading@10
|
340 }
|
yading@10
|
341
|
yading@10
|
342 m->access_unit_size = mh.access_unit_size;
|
yading@10
|
343 m->access_unit_size_pow2 = mh.access_unit_size_pow2;
|
yading@10
|
344
|
yading@10
|
345 m->num_substreams = mh.num_substreams;
|
yading@10
|
346 m->max_decoded_substream = m->num_substreams - 1;
|
yading@10
|
347
|
yading@10
|
348 m->avctx->sample_rate = mh.group1_samplerate;
|
yading@10
|
349 m->avctx->frame_size = mh.access_unit_size;
|
yading@10
|
350
|
yading@10
|
351 m->avctx->bits_per_raw_sample = mh.group1_bits;
|
yading@10
|
352 if (mh.group1_bits > 16)
|
yading@10
|
353 m->avctx->sample_fmt = AV_SAMPLE_FMT_S32;
|
yading@10
|
354 else
|
yading@10
|
355 m->avctx->sample_fmt = AV_SAMPLE_FMT_S16;
|
yading@10
|
356
|
yading@10
|
357 m->params_valid = 1;
|
yading@10
|
358 for (substr = 0; substr < MAX_SUBSTREAMS; substr++)
|
yading@10
|
359 m->substream[substr].restart_seen = 0;
|
yading@10
|
360
|
yading@10
|
361 /* Set the layout for each substream. When there's more than one, the first
|
yading@10
|
362 * substream is Stereo. Subsequent substreams' layouts are indicated in the
|
yading@10
|
363 * major sync. */
|
yading@10
|
364 if (m->avctx->codec_id == AV_CODEC_ID_MLP) {
|
yading@10
|
365 if ((substr = (mh.num_substreams > 1)))
|
yading@10
|
366 m->substream[0].ch_layout = AV_CH_LAYOUT_STEREO;
|
yading@10
|
367 m->substream[substr].ch_layout = mh.channel_layout_mlp;
|
yading@10
|
368 } else {
|
yading@10
|
369 if ((substr = (mh.num_substreams > 1)))
|
yading@10
|
370 m->substream[0].ch_layout = AV_CH_LAYOUT_STEREO;
|
yading@10
|
371 if (mh.num_substreams > 2)
|
yading@10
|
372 if (mh.channel_layout_thd_stream2)
|
yading@10
|
373 m->substream[2].ch_layout = mh.channel_layout_thd_stream2;
|
yading@10
|
374 else
|
yading@10
|
375 m->substream[2].ch_layout = mh.channel_layout_thd_stream1;
|
yading@10
|
376 m->substream[substr].ch_layout = mh.channel_layout_thd_stream1;
|
yading@10
|
377
|
yading@10
|
378 if (m->avctx->channels<=2 && m->substream[substr].ch_layout == AV_CH_LAYOUT_MONO && m->max_decoded_substream == 1) {
|
yading@10
|
379 av_log(m->avctx, AV_LOG_DEBUG, "Mono stream with 2 substreams, ignoring 2nd\n");
|
yading@10
|
380 m->max_decoded_substream = 0;
|
yading@10
|
381 if (m->avctx->channels==2)
|
yading@10
|
382 m->avctx->channel_layout = AV_CH_LAYOUT_STEREO;
|
yading@10
|
383 }
|
yading@10
|
384 }
|
yading@10
|
385
|
yading@10
|
386 m->needs_reordering = mh.channel_arrangement >= 18 && mh.channel_arrangement <= 20;
|
yading@10
|
387
|
yading@10
|
388 return 0;
|
yading@10
|
389 }
|
yading@10
|
390
|
yading@10
|
391 /** Read a restart header from a block in a substream. This contains parameters
|
yading@10
|
392 * required to decode the audio that do not change very often. Generally
|
yading@10
|
393 * (always) present only in blocks following a major sync. */
|
yading@10
|
394
|
yading@10
|
395 static int read_restart_header(MLPDecodeContext *m, GetBitContext *gbp,
|
yading@10
|
396 const uint8_t *buf, unsigned int substr)
|
yading@10
|
397 {
|
yading@10
|
398 SubStream *s = &m->substream[substr];
|
yading@10
|
399 unsigned int ch;
|
yading@10
|
400 int sync_word, tmp;
|
yading@10
|
401 uint8_t checksum;
|
yading@10
|
402 uint8_t lossless_check;
|
yading@10
|
403 int start_count = get_bits_count(gbp);
|
yading@10
|
404 const int max_matrix_channel = m->avctx->codec_id == AV_CODEC_ID_MLP
|
yading@10
|
405 ? MAX_MATRIX_CHANNEL_MLP
|
yading@10
|
406 : MAX_MATRIX_CHANNEL_TRUEHD;
|
yading@10
|
407 int max_channel, min_channel, matrix_channel;
|
yading@10
|
408
|
yading@10
|
409 sync_word = get_bits(gbp, 13);
|
yading@10
|
410
|
yading@10
|
411 if (sync_word != 0x31ea >> 1) {
|
yading@10
|
412 av_log(m->avctx, AV_LOG_ERROR,
|
yading@10
|
413 "restart header sync incorrect (got 0x%04x)\n", sync_word);
|
yading@10
|
414 return AVERROR_INVALIDDATA;
|
yading@10
|
415 }
|
yading@10
|
416
|
yading@10
|
417 s->noise_type = get_bits1(gbp);
|
yading@10
|
418
|
yading@10
|
419 if (m->avctx->codec_id == AV_CODEC_ID_MLP && s->noise_type) {
|
yading@10
|
420 av_log(m->avctx, AV_LOG_ERROR, "MLP must have 0x31ea sync word.\n");
|
yading@10
|
421 return AVERROR_INVALIDDATA;
|
yading@10
|
422 }
|
yading@10
|
423
|
yading@10
|
424 skip_bits(gbp, 16); /* Output timestamp */
|
yading@10
|
425
|
yading@10
|
426 min_channel = get_bits(gbp, 4);
|
yading@10
|
427 max_channel = get_bits(gbp, 4);
|
yading@10
|
428 matrix_channel = get_bits(gbp, 4);
|
yading@10
|
429
|
yading@10
|
430 if (matrix_channel > max_matrix_channel) {
|
yading@10
|
431 av_log(m->avctx, AV_LOG_ERROR,
|
yading@10
|
432 "Max matrix channel cannot be greater than %d.\n",
|
yading@10
|
433 max_matrix_channel);
|
yading@10
|
434 return AVERROR_INVALIDDATA;
|
yading@10
|
435 }
|
yading@10
|
436
|
yading@10
|
437 if (max_channel != matrix_channel) {
|
yading@10
|
438 av_log(m->avctx, AV_LOG_ERROR,
|
yading@10
|
439 "Max channel must be equal max matrix channel.\n");
|
yading@10
|
440 return AVERROR_INVALIDDATA;
|
yading@10
|
441 }
|
yading@10
|
442
|
yading@10
|
443 /* This should happen for TrueHD streams with >6 channels and MLP's noise
|
yading@10
|
444 * type. It is not yet known if this is allowed. */
|
yading@10
|
445 if (max_channel > MAX_MATRIX_CHANNEL_MLP && !s->noise_type) {
|
yading@10
|
446 avpriv_request_sample(m->avctx,
|
yading@10
|
447 "%d channels (more than the "
|
yading@10
|
448 "maximum supported by the decoder)",
|
yading@10
|
449 max_channel + 2);
|
yading@10
|
450 return AVERROR_PATCHWELCOME;
|
yading@10
|
451 }
|
yading@10
|
452
|
yading@10
|
453 if (min_channel > max_channel) {
|
yading@10
|
454 av_log(m->avctx, AV_LOG_ERROR,
|
yading@10
|
455 "Substream min channel cannot be greater than max channel.\n");
|
yading@10
|
456 return AVERROR_INVALIDDATA;
|
yading@10
|
457 }
|
yading@10
|
458
|
yading@10
|
459 s->min_channel = min_channel;
|
yading@10
|
460 s->max_channel = max_channel;
|
yading@10
|
461 s->max_matrix_channel = matrix_channel;
|
yading@10
|
462
|
yading@10
|
463 #if FF_API_REQUEST_CHANNELS
|
yading@10
|
464 if (m->avctx->request_channels > 0 &&
|
yading@10
|
465 m->avctx->request_channels <= s->max_channel + 1 &&
|
yading@10
|
466 m->max_decoded_substream > substr) {
|
yading@10
|
467 av_log(m->avctx, AV_LOG_DEBUG,
|
yading@10
|
468 "Extracting %d-channel downmix from substream %d. "
|
yading@10
|
469 "Further substreams will be skipped.\n",
|
yading@10
|
470 s->max_channel + 1, substr);
|
yading@10
|
471 m->max_decoded_substream = substr;
|
yading@10
|
472 } else
|
yading@10
|
473 #endif
|
yading@10
|
474 if (m->avctx->request_channel_layout == s->ch_layout &&
|
yading@10
|
475 m->max_decoded_substream > substr) {
|
yading@10
|
476 av_log(m->avctx, AV_LOG_DEBUG,
|
yading@10
|
477 "Extracting %d-channel downmix (0x%"PRIx64") from substream %d. "
|
yading@10
|
478 "Further substreams will be skipped.\n",
|
yading@10
|
479 s->max_channel + 1, s->ch_layout, substr);
|
yading@10
|
480 m->max_decoded_substream = substr;
|
yading@10
|
481 }
|
yading@10
|
482
|
yading@10
|
483 s->noise_shift = get_bits(gbp, 4);
|
yading@10
|
484 s->noisegen_seed = get_bits(gbp, 23);
|
yading@10
|
485
|
yading@10
|
486 skip_bits(gbp, 19);
|
yading@10
|
487
|
yading@10
|
488 s->data_check_present = get_bits1(gbp);
|
yading@10
|
489 lossless_check = get_bits(gbp, 8);
|
yading@10
|
490 if (substr == m->max_decoded_substream
|
yading@10
|
491 && s->lossless_check_data != 0xffffffff) {
|
yading@10
|
492 tmp = xor_32_to_8(s->lossless_check_data);
|
yading@10
|
493 if (tmp != lossless_check)
|
yading@10
|
494 av_log(m->avctx, AV_LOG_WARNING,
|
yading@10
|
495 "Lossless check failed - expected %02x, calculated %02x.\n",
|
yading@10
|
496 lossless_check, tmp);
|
yading@10
|
497 }
|
yading@10
|
498
|
yading@10
|
499 skip_bits(gbp, 16);
|
yading@10
|
500
|
yading@10
|
501 memset(s->ch_assign, 0, sizeof(s->ch_assign));
|
yading@10
|
502
|
yading@10
|
503 for (ch = 0; ch <= s->max_matrix_channel; ch++) {
|
yading@10
|
504 int ch_assign = get_bits(gbp, 6);
|
yading@10
|
505 if (m->avctx->codec_id == AV_CODEC_ID_TRUEHD) {
|
yading@10
|
506 uint64_t channel = thd_channel_layout_extract_channel(s->ch_layout,
|
yading@10
|
507 ch_assign);
|
yading@10
|
508 ch_assign = av_get_channel_layout_channel_index(s->ch_layout,
|
yading@10
|
509 channel);
|
yading@10
|
510 }
|
yading@10
|
511 if ((unsigned)ch_assign > s->max_matrix_channel) {
|
yading@10
|
512 avpriv_request_sample(m->avctx,
|
yading@10
|
513 "Assignment of matrix channel %d to invalid output channel %d",
|
yading@10
|
514 ch, ch_assign);
|
yading@10
|
515 return AVERROR_PATCHWELCOME;
|
yading@10
|
516 }
|
yading@10
|
517 s->ch_assign[ch_assign] = ch;
|
yading@10
|
518 }
|
yading@10
|
519
|
yading@10
|
520 checksum = ff_mlp_restart_checksum(buf, get_bits_count(gbp) - start_count);
|
yading@10
|
521
|
yading@10
|
522 if (checksum != get_bits(gbp, 8))
|
yading@10
|
523 av_log(m->avctx, AV_LOG_ERROR, "restart header checksum error\n");
|
yading@10
|
524
|
yading@10
|
525 /* Set default decoding parameters. */
|
yading@10
|
526 s->param_presence_flags = 0xff;
|
yading@10
|
527 s->num_primitive_matrices = 0;
|
yading@10
|
528 s->blocksize = 8;
|
yading@10
|
529 s->lossless_check_data = 0;
|
yading@10
|
530
|
yading@10
|
531 memset(s->output_shift , 0, sizeof(s->output_shift ));
|
yading@10
|
532 memset(s->quant_step_size, 0, sizeof(s->quant_step_size));
|
yading@10
|
533
|
yading@10
|
534 for (ch = s->min_channel; ch <= s->max_channel; ch++) {
|
yading@10
|
535 ChannelParams *cp = &s->channel_params[ch];
|
yading@10
|
536 cp->filter_params[FIR].order = 0;
|
yading@10
|
537 cp->filter_params[IIR].order = 0;
|
yading@10
|
538 cp->filter_params[FIR].shift = 0;
|
yading@10
|
539 cp->filter_params[IIR].shift = 0;
|
yading@10
|
540
|
yading@10
|
541 /* Default audio coding is 24-bit raw PCM. */
|
yading@10
|
542 cp->huff_offset = 0;
|
yading@10
|
543 cp->sign_huff_offset = (-1) << 23;
|
yading@10
|
544 cp->codebook = 0;
|
yading@10
|
545 cp->huff_lsbs = 24;
|
yading@10
|
546 }
|
yading@10
|
547
|
yading@10
|
548 if (substr == m->max_decoded_substream) {
|
yading@10
|
549 m->avctx->channels = s->max_matrix_channel + 1;
|
yading@10
|
550 m->avctx->channel_layout = s->ch_layout;
|
yading@10
|
551
|
yading@10
|
552 if (m->avctx->codec_id == AV_CODEC_ID_MLP && m->needs_reordering) {
|
yading@10
|
553 if (m->avctx->channel_layout == (AV_CH_LAYOUT_QUAD|AV_CH_LOW_FREQUENCY) ||
|
yading@10
|
554 m->avctx->channel_layout == AV_CH_LAYOUT_5POINT0_BACK) {
|
yading@10
|
555 int i = s->ch_assign[4];
|
yading@10
|
556 s->ch_assign[4] = s->ch_assign[3];
|
yading@10
|
557 s->ch_assign[3] = s->ch_assign[2];
|
yading@10
|
558 s->ch_assign[2] = i;
|
yading@10
|
559 } else if (m->avctx->channel_layout == AV_CH_LAYOUT_5POINT1_BACK) {
|
yading@10
|
560 FFSWAP(int, s->ch_assign[2], s->ch_assign[4]);
|
yading@10
|
561 FFSWAP(int, s->ch_assign[3], s->ch_assign[5]);
|
yading@10
|
562 }
|
yading@10
|
563 }
|
yading@10
|
564
|
yading@10
|
565 }
|
yading@10
|
566
|
yading@10
|
567 return 0;
|
yading@10
|
568 }
|
yading@10
|
569
|
yading@10
|
570 /** Read parameters for one of the prediction filters. */
|
yading@10
|
571
|
yading@10
|
572 static int read_filter_params(MLPDecodeContext *m, GetBitContext *gbp,
|
yading@10
|
573 unsigned int substr, unsigned int channel,
|
yading@10
|
574 unsigned int filter)
|
yading@10
|
575 {
|
yading@10
|
576 SubStream *s = &m->substream[substr];
|
yading@10
|
577 FilterParams *fp = &s->channel_params[channel].filter_params[filter];
|
yading@10
|
578 const int max_order = filter ? MAX_IIR_ORDER : MAX_FIR_ORDER;
|
yading@10
|
579 const char fchar = filter ? 'I' : 'F';
|
yading@10
|
580 int i, order;
|
yading@10
|
581
|
yading@10
|
582 // Filter is 0 for FIR, 1 for IIR.
|
yading@10
|
583 av_assert0(filter < 2);
|
yading@10
|
584
|
yading@10
|
585 if (m->filter_changed[channel][filter]++ > 1) {
|
yading@10
|
586 av_log(m->avctx, AV_LOG_ERROR, "Filters may change only once per access unit.\n");
|
yading@10
|
587 return AVERROR_INVALIDDATA;
|
yading@10
|
588 }
|
yading@10
|
589
|
yading@10
|
590 order = get_bits(gbp, 4);
|
yading@10
|
591 if (order > max_order) {
|
yading@10
|
592 av_log(m->avctx, AV_LOG_ERROR,
|
yading@10
|
593 "%cIR filter order %d is greater than maximum %d.\n",
|
yading@10
|
594 fchar, order, max_order);
|
yading@10
|
595 return AVERROR_INVALIDDATA;
|
yading@10
|
596 }
|
yading@10
|
597 fp->order = order;
|
yading@10
|
598
|
yading@10
|
599 if (order > 0) {
|
yading@10
|
600 int32_t *fcoeff = s->channel_params[channel].coeff[filter];
|
yading@10
|
601 int coeff_bits, coeff_shift;
|
yading@10
|
602
|
yading@10
|
603 fp->shift = get_bits(gbp, 4);
|
yading@10
|
604
|
yading@10
|
605 coeff_bits = get_bits(gbp, 5);
|
yading@10
|
606 coeff_shift = get_bits(gbp, 3);
|
yading@10
|
607 if (coeff_bits < 1 || coeff_bits > 16) {
|
yading@10
|
608 av_log(m->avctx, AV_LOG_ERROR,
|
yading@10
|
609 "%cIR filter coeff_bits must be between 1 and 16.\n",
|
yading@10
|
610 fchar);
|
yading@10
|
611 return AVERROR_INVALIDDATA;
|
yading@10
|
612 }
|
yading@10
|
613 if (coeff_bits + coeff_shift > 16) {
|
yading@10
|
614 av_log(m->avctx, AV_LOG_ERROR,
|
yading@10
|
615 "Sum of coeff_bits and coeff_shift for %cIR filter must be 16 or less.\n",
|
yading@10
|
616 fchar);
|
yading@10
|
617 return AVERROR_INVALIDDATA;
|
yading@10
|
618 }
|
yading@10
|
619
|
yading@10
|
620 for (i = 0; i < order; i++)
|
yading@10
|
621 fcoeff[i] = get_sbits(gbp, coeff_bits) << coeff_shift;
|
yading@10
|
622
|
yading@10
|
623 if (get_bits1(gbp)) {
|
yading@10
|
624 int state_bits, state_shift;
|
yading@10
|
625
|
yading@10
|
626 if (filter == FIR) {
|
yading@10
|
627 av_log(m->avctx, AV_LOG_ERROR,
|
yading@10
|
628 "FIR filter has state data specified.\n");
|
yading@10
|
629 return AVERROR_INVALIDDATA;
|
yading@10
|
630 }
|
yading@10
|
631
|
yading@10
|
632 state_bits = get_bits(gbp, 4);
|
yading@10
|
633 state_shift = get_bits(gbp, 4);
|
yading@10
|
634
|
yading@10
|
635 /* TODO: Check validity of state data. */
|
yading@10
|
636
|
yading@10
|
637 for (i = 0; i < order; i++)
|
yading@10
|
638 fp->state[i] = state_bits ? get_sbits(gbp, state_bits) << state_shift : 0;
|
yading@10
|
639 }
|
yading@10
|
640 }
|
yading@10
|
641
|
yading@10
|
642 return 0;
|
yading@10
|
643 }
|
yading@10
|
644
|
yading@10
|
645 /** Read parameters for primitive matrices. */
|
yading@10
|
646
|
yading@10
|
647 static int read_matrix_params(MLPDecodeContext *m, unsigned int substr, GetBitContext *gbp)
|
yading@10
|
648 {
|
yading@10
|
649 SubStream *s = &m->substream[substr];
|
yading@10
|
650 unsigned int mat, ch;
|
yading@10
|
651 const int max_primitive_matrices = m->avctx->codec_id == AV_CODEC_ID_MLP
|
yading@10
|
652 ? MAX_MATRICES_MLP
|
yading@10
|
653 : MAX_MATRICES_TRUEHD;
|
yading@10
|
654
|
yading@10
|
655 if (m->matrix_changed++ > 1) {
|
yading@10
|
656 av_log(m->avctx, AV_LOG_ERROR, "Matrices may change only once per access unit.\n");
|
yading@10
|
657 return AVERROR_INVALIDDATA;
|
yading@10
|
658 }
|
yading@10
|
659
|
yading@10
|
660 s->num_primitive_matrices = get_bits(gbp, 4);
|
yading@10
|
661
|
yading@10
|
662 if (s->num_primitive_matrices > max_primitive_matrices) {
|
yading@10
|
663 av_log(m->avctx, AV_LOG_ERROR,
|
yading@10
|
664 "Number of primitive matrices cannot be greater than %d.\n",
|
yading@10
|
665 max_primitive_matrices);
|
yading@10
|
666 return AVERROR_INVALIDDATA;
|
yading@10
|
667 }
|
yading@10
|
668
|
yading@10
|
669 for (mat = 0; mat < s->num_primitive_matrices; mat++) {
|
yading@10
|
670 int frac_bits, max_chan;
|
yading@10
|
671 s->matrix_out_ch[mat] = get_bits(gbp, 4);
|
yading@10
|
672 frac_bits = get_bits(gbp, 4);
|
yading@10
|
673 s->lsb_bypass [mat] = get_bits1(gbp);
|
yading@10
|
674
|
yading@10
|
675 if (s->matrix_out_ch[mat] > s->max_matrix_channel) {
|
yading@10
|
676 av_log(m->avctx, AV_LOG_ERROR,
|
yading@10
|
677 "Invalid channel %d specified as output from matrix.\n",
|
yading@10
|
678 s->matrix_out_ch[mat]);
|
yading@10
|
679 return AVERROR_INVALIDDATA;
|
yading@10
|
680 }
|
yading@10
|
681 if (frac_bits > 14) {
|
yading@10
|
682 av_log(m->avctx, AV_LOG_ERROR,
|
yading@10
|
683 "Too many fractional bits specified.\n");
|
yading@10
|
684 return AVERROR_INVALIDDATA;
|
yading@10
|
685 }
|
yading@10
|
686
|
yading@10
|
687 max_chan = s->max_matrix_channel;
|
yading@10
|
688 if (!s->noise_type)
|
yading@10
|
689 max_chan+=2;
|
yading@10
|
690
|
yading@10
|
691 for (ch = 0; ch <= max_chan; ch++) {
|
yading@10
|
692 int coeff_val = 0;
|
yading@10
|
693 if (get_bits1(gbp))
|
yading@10
|
694 coeff_val = get_sbits(gbp, frac_bits + 2);
|
yading@10
|
695
|
yading@10
|
696 s->matrix_coeff[mat][ch] = coeff_val << (14 - frac_bits);
|
yading@10
|
697 }
|
yading@10
|
698
|
yading@10
|
699 if (s->noise_type)
|
yading@10
|
700 s->matrix_noise_shift[mat] = get_bits(gbp, 4);
|
yading@10
|
701 else
|
yading@10
|
702 s->matrix_noise_shift[mat] = 0;
|
yading@10
|
703 }
|
yading@10
|
704
|
yading@10
|
705 return 0;
|
yading@10
|
706 }
|
yading@10
|
707
|
yading@10
|
708 /** Read channel parameters. */
|
yading@10
|
709
|
yading@10
|
710 static int read_channel_params(MLPDecodeContext *m, unsigned int substr,
|
yading@10
|
711 GetBitContext *gbp, unsigned int ch)
|
yading@10
|
712 {
|
yading@10
|
713 SubStream *s = &m->substream[substr];
|
yading@10
|
714 ChannelParams *cp = &s->channel_params[ch];
|
yading@10
|
715 FilterParams *fir = &cp->filter_params[FIR];
|
yading@10
|
716 FilterParams *iir = &cp->filter_params[IIR];
|
yading@10
|
717 int ret;
|
yading@10
|
718
|
yading@10
|
719 if (s->param_presence_flags & PARAM_FIR)
|
yading@10
|
720 if (get_bits1(gbp))
|
yading@10
|
721 if ((ret = read_filter_params(m, gbp, substr, ch, FIR)) < 0)
|
yading@10
|
722 return ret;
|
yading@10
|
723
|
yading@10
|
724 if (s->param_presence_flags & PARAM_IIR)
|
yading@10
|
725 if (get_bits1(gbp))
|
yading@10
|
726 if ((ret = read_filter_params(m, gbp, substr, ch, IIR)) < 0)
|
yading@10
|
727 return ret;
|
yading@10
|
728
|
yading@10
|
729 if (fir->order + iir->order > 8) {
|
yading@10
|
730 av_log(m->avctx, AV_LOG_ERROR, "Total filter orders too high.\n");
|
yading@10
|
731 return AVERROR_INVALIDDATA;
|
yading@10
|
732 }
|
yading@10
|
733
|
yading@10
|
734 if (fir->order && iir->order &&
|
yading@10
|
735 fir->shift != iir->shift) {
|
yading@10
|
736 av_log(m->avctx, AV_LOG_ERROR,
|
yading@10
|
737 "FIR and IIR filters must use the same precision.\n");
|
yading@10
|
738 return AVERROR_INVALIDDATA;
|
yading@10
|
739 }
|
yading@10
|
740 /* The FIR and IIR filters must have the same precision.
|
yading@10
|
741 * To simplify the filtering code, only the precision of the
|
yading@10
|
742 * FIR filter is considered. If only the IIR filter is employed,
|
yading@10
|
743 * the FIR filter precision is set to that of the IIR filter, so
|
yading@10
|
744 * that the filtering code can use it. */
|
yading@10
|
745 if (!fir->order && iir->order)
|
yading@10
|
746 fir->shift = iir->shift;
|
yading@10
|
747
|
yading@10
|
748 if (s->param_presence_flags & PARAM_HUFFOFFSET)
|
yading@10
|
749 if (get_bits1(gbp))
|
yading@10
|
750 cp->huff_offset = get_sbits(gbp, 15);
|
yading@10
|
751
|
yading@10
|
752 cp->codebook = get_bits(gbp, 2);
|
yading@10
|
753 cp->huff_lsbs = get_bits(gbp, 5);
|
yading@10
|
754
|
yading@10
|
755 if (cp->huff_lsbs > 24) {
|
yading@10
|
756 av_log(m->avctx, AV_LOG_ERROR, "Invalid huff_lsbs.\n");
|
yading@10
|
757 cp->huff_lsbs = 0;
|
yading@10
|
758 return AVERROR_INVALIDDATA;
|
yading@10
|
759 }
|
yading@10
|
760
|
yading@10
|
761 cp->sign_huff_offset = calculate_sign_huff(m, substr, ch);
|
yading@10
|
762
|
yading@10
|
763 return 0;
|
yading@10
|
764 }
|
yading@10
|
765
|
yading@10
|
766 /** Read decoding parameters that change more often than those in the restart
|
yading@10
|
767 * header. */
|
yading@10
|
768
|
yading@10
|
769 static int read_decoding_params(MLPDecodeContext *m, GetBitContext *gbp,
|
yading@10
|
770 unsigned int substr)
|
yading@10
|
771 {
|
yading@10
|
772 SubStream *s = &m->substream[substr];
|
yading@10
|
773 unsigned int ch;
|
yading@10
|
774 int ret;
|
yading@10
|
775
|
yading@10
|
776 if (s->param_presence_flags & PARAM_PRESENCE)
|
yading@10
|
777 if (get_bits1(gbp))
|
yading@10
|
778 s->param_presence_flags = get_bits(gbp, 8);
|
yading@10
|
779
|
yading@10
|
780 if (s->param_presence_flags & PARAM_BLOCKSIZE)
|
yading@10
|
781 if (get_bits1(gbp)) {
|
yading@10
|
782 s->blocksize = get_bits(gbp, 9);
|
yading@10
|
783 if (s->blocksize < 8 || s->blocksize > m->access_unit_size) {
|
yading@10
|
784 av_log(m->avctx, AV_LOG_ERROR, "Invalid blocksize.\n");
|
yading@10
|
785 s->blocksize = 0;
|
yading@10
|
786 return AVERROR_INVALIDDATA;
|
yading@10
|
787 }
|
yading@10
|
788 }
|
yading@10
|
789
|
yading@10
|
790 if (s->param_presence_flags & PARAM_MATRIX)
|
yading@10
|
791 if (get_bits1(gbp))
|
yading@10
|
792 if ((ret = read_matrix_params(m, substr, gbp)) < 0)
|
yading@10
|
793 return ret;
|
yading@10
|
794
|
yading@10
|
795 if (s->param_presence_flags & PARAM_OUTSHIFT)
|
yading@10
|
796 if (get_bits1(gbp))
|
yading@10
|
797 for (ch = 0; ch <= s->max_matrix_channel; ch++)
|
yading@10
|
798 s->output_shift[ch] = get_sbits(gbp, 4);
|
yading@10
|
799
|
yading@10
|
800 if (s->param_presence_flags & PARAM_QUANTSTEP)
|
yading@10
|
801 if (get_bits1(gbp))
|
yading@10
|
802 for (ch = 0; ch <= s->max_channel; ch++) {
|
yading@10
|
803 ChannelParams *cp = &s->channel_params[ch];
|
yading@10
|
804
|
yading@10
|
805 s->quant_step_size[ch] = get_bits(gbp, 4);
|
yading@10
|
806
|
yading@10
|
807 cp->sign_huff_offset = calculate_sign_huff(m, substr, ch);
|
yading@10
|
808 }
|
yading@10
|
809
|
yading@10
|
810 for (ch = s->min_channel; ch <= s->max_channel; ch++)
|
yading@10
|
811 if (get_bits1(gbp))
|
yading@10
|
812 if ((ret = read_channel_params(m, substr, gbp, ch)) < 0)
|
yading@10
|
813 return ret;
|
yading@10
|
814
|
yading@10
|
815 return 0;
|
yading@10
|
816 }
|
yading@10
|
817
|
yading@10
|
818 #define MSB_MASK(bits) (-1u << bits)
|
yading@10
|
819
|
yading@10
|
820 /** Generate PCM samples using the prediction filters and residual values
|
yading@10
|
821 * read from the data stream, and update the filter state. */
|
yading@10
|
822
|
yading@10
|
823 static void filter_channel(MLPDecodeContext *m, unsigned int substr,
|
yading@10
|
824 unsigned int channel)
|
yading@10
|
825 {
|
yading@10
|
826 SubStream *s = &m->substream[substr];
|
yading@10
|
827 const int32_t *fircoeff = s->channel_params[channel].coeff[FIR];
|
yading@10
|
828 int32_t state_buffer[NUM_FILTERS][MAX_BLOCKSIZE + MAX_FIR_ORDER];
|
yading@10
|
829 int32_t *firbuf = state_buffer[FIR] + MAX_BLOCKSIZE;
|
yading@10
|
830 int32_t *iirbuf = state_buffer[IIR] + MAX_BLOCKSIZE;
|
yading@10
|
831 FilterParams *fir = &s->channel_params[channel].filter_params[FIR];
|
yading@10
|
832 FilterParams *iir = &s->channel_params[channel].filter_params[IIR];
|
yading@10
|
833 unsigned int filter_shift = fir->shift;
|
yading@10
|
834 int32_t mask = MSB_MASK(s->quant_step_size[channel]);
|
yading@10
|
835
|
yading@10
|
836 memcpy(firbuf, fir->state, MAX_FIR_ORDER * sizeof(int32_t));
|
yading@10
|
837 memcpy(iirbuf, iir->state, MAX_IIR_ORDER * sizeof(int32_t));
|
yading@10
|
838
|
yading@10
|
839 m->dsp.mlp_filter_channel(firbuf, fircoeff,
|
yading@10
|
840 fir->order, iir->order,
|
yading@10
|
841 filter_shift, mask, s->blocksize,
|
yading@10
|
842 &m->sample_buffer[s->blockpos][channel]);
|
yading@10
|
843
|
yading@10
|
844 memcpy(fir->state, firbuf - s->blocksize, MAX_FIR_ORDER * sizeof(int32_t));
|
yading@10
|
845 memcpy(iir->state, iirbuf - s->blocksize, MAX_IIR_ORDER * sizeof(int32_t));
|
yading@10
|
846 }
|
yading@10
|
847
|
yading@10
|
848 /** Read a block of PCM residual data (or actual if no filtering active). */
|
yading@10
|
849
|
yading@10
|
850 static int read_block_data(MLPDecodeContext *m, GetBitContext *gbp,
|
yading@10
|
851 unsigned int substr)
|
yading@10
|
852 {
|
yading@10
|
853 SubStream *s = &m->substream[substr];
|
yading@10
|
854 unsigned int i, ch, expected_stream_pos = 0;
|
yading@10
|
855 int ret;
|
yading@10
|
856
|
yading@10
|
857 if (s->data_check_present) {
|
yading@10
|
858 expected_stream_pos = get_bits_count(gbp);
|
yading@10
|
859 expected_stream_pos += get_bits(gbp, 16);
|
yading@10
|
860 avpriv_request_sample(m->avctx,
|
yading@10
|
861 "Substreams with VLC block size check info");
|
yading@10
|
862 }
|
yading@10
|
863
|
yading@10
|
864 if (s->blockpos + s->blocksize > m->access_unit_size) {
|
yading@10
|
865 av_log(m->avctx, AV_LOG_ERROR, "too many audio samples in frame\n");
|
yading@10
|
866 return AVERROR_INVALIDDATA;
|
yading@10
|
867 }
|
yading@10
|
868
|
yading@10
|
869 memset(&m->bypassed_lsbs[s->blockpos][0], 0,
|
yading@10
|
870 s->blocksize * sizeof(m->bypassed_lsbs[0]));
|
yading@10
|
871
|
yading@10
|
872 for (i = 0; i < s->blocksize; i++)
|
yading@10
|
873 if ((ret = read_huff_channels(m, gbp, substr, i)) < 0)
|
yading@10
|
874 return ret;
|
yading@10
|
875
|
yading@10
|
876 for (ch = s->min_channel; ch <= s->max_channel; ch++)
|
yading@10
|
877 filter_channel(m, substr, ch);
|
yading@10
|
878
|
yading@10
|
879 s->blockpos += s->blocksize;
|
yading@10
|
880
|
yading@10
|
881 if (s->data_check_present) {
|
yading@10
|
882 if (get_bits_count(gbp) != expected_stream_pos)
|
yading@10
|
883 av_log(m->avctx, AV_LOG_ERROR, "block data length mismatch\n");
|
yading@10
|
884 skip_bits(gbp, 8);
|
yading@10
|
885 }
|
yading@10
|
886
|
yading@10
|
887 return 0;
|
yading@10
|
888 }
|
yading@10
|
889
|
yading@10
|
890 /** Data table used for TrueHD noise generation function. */
|
yading@10
|
891
|
yading@10
|
892 static const int8_t noise_table[256] = {
|
yading@10
|
893 30, 51, 22, 54, 3, 7, -4, 38, 14, 55, 46, 81, 22, 58, -3, 2,
|
yading@10
|
894 52, 31, -7, 51, 15, 44, 74, 30, 85, -17, 10, 33, 18, 80, 28, 62,
|
yading@10
|
895 10, 32, 23, 69, 72, 26, 35, 17, 73, 60, 8, 56, 2, 6, -2, -5,
|
yading@10
|
896 51, 4, 11, 50, 66, 76, 21, 44, 33, 47, 1, 26, 64, 48, 57, 40,
|
yading@10
|
897 38, 16, -10, -28, 92, 22, -18, 29, -10, 5, -13, 49, 19, 24, 70, 34,
|
yading@10
|
898 61, 48, 30, 14, -6, 25, 58, 33, 42, 60, 67, 17, 54, 17, 22, 30,
|
yading@10
|
899 67, 44, -9, 50, -11, 43, 40, 32, 59, 82, 13, 49, -14, 55, 60, 36,
|
yading@10
|
900 48, 49, 31, 47, 15, 12, 4, 65, 1, 23, 29, 39, 45, -2, 84, 69,
|
yading@10
|
901 0, 72, 37, 57, 27, 41, -15, -16, 35, 31, 14, 61, 24, 0, 27, 24,
|
yading@10
|
902 16, 41, 55, 34, 53, 9, 56, 12, 25, 29, 53, 5, 20, -20, -8, 20,
|
yading@10
|
903 13, 28, -3, 78, 38, 16, 11, 62, 46, 29, 21, 24, 46, 65, 43, -23,
|
yading@10
|
904 89, 18, 74, 21, 38, -12, 19, 12, -19, 8, 15, 33, 4, 57, 9, -8,
|
yading@10
|
905 36, 35, 26, 28, 7, 83, 63, 79, 75, 11, 3, 87, 37, 47, 34, 40,
|
yading@10
|
906 39, 19, 20, 42, 27, 34, 39, 77, 13, 42, 59, 64, 45, -1, 32, 37,
|
yading@10
|
907 45, -5, 53, -6, 7, 36, 50, 23, 6, 32, 9, -21, 18, 71, 27, 52,
|
yading@10
|
908 -25, 31, 35, 42, -1, 68, 63, 52, 26, 43, 66, 37, 41, 25, 40, 70,
|
yading@10
|
909 };
|
yading@10
|
910
|
yading@10
|
911 /** Noise generation functions.
|
yading@10
|
912 * I'm not sure what these are for - they seem to be some kind of pseudorandom
|
yading@10
|
913 * sequence generators, used to generate noise data which is used when the
|
yading@10
|
914 * channels are rematrixed. I'm not sure if they provide a practical benefit
|
yading@10
|
915 * to compression, or just obfuscate the decoder. Are they for some kind of
|
yading@10
|
916 * dithering? */
|
yading@10
|
917
|
yading@10
|
918 /** Generate two channels of noise, used in the matrix when
|
yading@10
|
919 * restart sync word == 0x31ea. */
|
yading@10
|
920
|
yading@10
|
921 static void generate_2_noise_channels(MLPDecodeContext *m, unsigned int substr)
|
yading@10
|
922 {
|
yading@10
|
923 SubStream *s = &m->substream[substr];
|
yading@10
|
924 unsigned int i;
|
yading@10
|
925 uint32_t seed = s->noisegen_seed;
|
yading@10
|
926 unsigned int maxchan = s->max_matrix_channel;
|
yading@10
|
927
|
yading@10
|
928 for (i = 0; i < s->blockpos; i++) {
|
yading@10
|
929 uint16_t seed_shr7 = seed >> 7;
|
yading@10
|
930 m->sample_buffer[i][maxchan+1] = ((int8_t)(seed >> 15)) << s->noise_shift;
|
yading@10
|
931 m->sample_buffer[i][maxchan+2] = ((int8_t) seed_shr7) << s->noise_shift;
|
yading@10
|
932
|
yading@10
|
933 seed = (seed << 16) ^ seed_shr7 ^ (seed_shr7 << 5);
|
yading@10
|
934 }
|
yading@10
|
935
|
yading@10
|
936 s->noisegen_seed = seed;
|
yading@10
|
937 }
|
yading@10
|
938
|
yading@10
|
939 /** Generate a block of noise, used when restart sync word == 0x31eb. */
|
yading@10
|
940
|
yading@10
|
941 static void fill_noise_buffer(MLPDecodeContext *m, unsigned int substr)
|
yading@10
|
942 {
|
yading@10
|
943 SubStream *s = &m->substream[substr];
|
yading@10
|
944 unsigned int i;
|
yading@10
|
945 uint32_t seed = s->noisegen_seed;
|
yading@10
|
946
|
yading@10
|
947 for (i = 0; i < m->access_unit_size_pow2; i++) {
|
yading@10
|
948 uint8_t seed_shr15 = seed >> 15;
|
yading@10
|
949 m->noise_buffer[i] = noise_table[seed_shr15];
|
yading@10
|
950 seed = (seed << 8) ^ seed_shr15 ^ (seed_shr15 << 5);
|
yading@10
|
951 }
|
yading@10
|
952
|
yading@10
|
953 s->noisegen_seed = seed;
|
yading@10
|
954 }
|
yading@10
|
955
|
yading@10
|
956
|
yading@10
|
957 /** Apply the channel matrices in turn to reconstruct the original audio
|
yading@10
|
958 * samples. */
|
yading@10
|
959
|
yading@10
|
960 static void rematrix_channels(MLPDecodeContext *m, unsigned int substr)
|
yading@10
|
961 {
|
yading@10
|
962 SubStream *s = &m->substream[substr];
|
yading@10
|
963 unsigned int mat, src_ch, i;
|
yading@10
|
964 unsigned int maxchan;
|
yading@10
|
965
|
yading@10
|
966 maxchan = s->max_matrix_channel;
|
yading@10
|
967 if (!s->noise_type) {
|
yading@10
|
968 generate_2_noise_channels(m, substr);
|
yading@10
|
969 maxchan += 2;
|
yading@10
|
970 } else {
|
yading@10
|
971 fill_noise_buffer(m, substr);
|
yading@10
|
972 }
|
yading@10
|
973
|
yading@10
|
974 for (mat = 0; mat < s->num_primitive_matrices; mat++) {
|
yading@10
|
975 int matrix_noise_shift = s->matrix_noise_shift[mat];
|
yading@10
|
976 unsigned int dest_ch = s->matrix_out_ch[mat];
|
yading@10
|
977 int32_t mask = MSB_MASK(s->quant_step_size[dest_ch]);
|
yading@10
|
978 int32_t *coeffs = s->matrix_coeff[mat];
|
yading@10
|
979 int index = s->num_primitive_matrices - mat;
|
yading@10
|
980 int index2 = 2 * index + 1;
|
yading@10
|
981
|
yading@10
|
982 /* TODO: DSPContext? */
|
yading@10
|
983
|
yading@10
|
984 for (i = 0; i < s->blockpos; i++) {
|
yading@10
|
985 int32_t bypassed_lsb = m->bypassed_lsbs[i][mat];
|
yading@10
|
986 int32_t *samples = m->sample_buffer[i];
|
yading@10
|
987 int64_t accum = 0;
|
yading@10
|
988
|
yading@10
|
989 for (src_ch = 0; src_ch <= maxchan; src_ch++)
|
yading@10
|
990 accum += (int64_t) samples[src_ch] * coeffs[src_ch];
|
yading@10
|
991
|
yading@10
|
992 if (matrix_noise_shift) {
|
yading@10
|
993 index &= m->access_unit_size_pow2 - 1;
|
yading@10
|
994 accum += m->noise_buffer[index] << (matrix_noise_shift + 7);
|
yading@10
|
995 index += index2;
|
yading@10
|
996 }
|
yading@10
|
997
|
yading@10
|
998 samples[dest_ch] = ((accum >> 14) & mask) + bypassed_lsb;
|
yading@10
|
999 }
|
yading@10
|
1000 }
|
yading@10
|
1001 }
|
yading@10
|
1002
|
yading@10
|
1003 /** Write the audio data into the output buffer. */
|
yading@10
|
1004
|
yading@10
|
1005 static int output_data(MLPDecodeContext *m, unsigned int substr,
|
yading@10
|
1006 AVFrame *frame, int *got_frame_ptr)
|
yading@10
|
1007 {
|
yading@10
|
1008 AVCodecContext *avctx = m->avctx;
|
yading@10
|
1009 SubStream *s = &m->substream[substr];
|
yading@10
|
1010 unsigned int i, out_ch = 0;
|
yading@10
|
1011 int32_t *data_32;
|
yading@10
|
1012 int16_t *data_16;
|
yading@10
|
1013 int ret;
|
yading@10
|
1014 int is32 = (m->avctx->sample_fmt == AV_SAMPLE_FMT_S32);
|
yading@10
|
1015
|
yading@10
|
1016 if (m->avctx->channels != s->max_matrix_channel + 1) {
|
yading@10
|
1017 av_log(m->avctx, AV_LOG_ERROR, "channel count mismatch\n");
|
yading@10
|
1018 return AVERROR_INVALIDDATA;
|
yading@10
|
1019 }
|
yading@10
|
1020
|
yading@10
|
1021 if (!s->blockpos) {
|
yading@10
|
1022 av_log(avctx, AV_LOG_ERROR, "No samples to output.\n");
|
yading@10
|
1023 return AVERROR_INVALIDDATA;
|
yading@10
|
1024 }
|
yading@10
|
1025
|
yading@10
|
1026 /* get output buffer */
|
yading@10
|
1027 frame->nb_samples = s->blockpos;
|
yading@10
|
1028 if ((ret = ff_get_buffer(avctx, frame, 0)) < 0)
|
yading@10
|
1029 return ret;
|
yading@10
|
1030 data_32 = (int32_t *)frame->data[0];
|
yading@10
|
1031 data_16 = (int16_t *)frame->data[0];
|
yading@10
|
1032
|
yading@10
|
1033 for (i = 0; i < s->blockpos; i++) {
|
yading@10
|
1034 for (out_ch = 0; out_ch <= s->max_matrix_channel; out_ch++) {
|
yading@10
|
1035 int mat_ch = s->ch_assign[out_ch];
|
yading@10
|
1036 int32_t sample = m->sample_buffer[i][mat_ch]
|
yading@10
|
1037 << s->output_shift[mat_ch];
|
yading@10
|
1038 s->lossless_check_data ^= (sample & 0xffffff) << mat_ch;
|
yading@10
|
1039 if (is32) *data_32++ = sample << 8;
|
yading@10
|
1040 else *data_16++ = sample >> 8;
|
yading@10
|
1041 }
|
yading@10
|
1042 }
|
yading@10
|
1043
|
yading@10
|
1044 *got_frame_ptr = 1;
|
yading@10
|
1045
|
yading@10
|
1046 return 0;
|
yading@10
|
1047 }
|
yading@10
|
1048
|
yading@10
|
1049 /** Read an access unit from the stream.
|
yading@10
|
1050 * @return negative on error, 0 if not enough data is present in the input stream,
|
yading@10
|
1051 * otherwise the number of bytes consumed. */
|
yading@10
|
1052
|
yading@10
|
1053 static int read_access_unit(AVCodecContext *avctx, void* data,
|
yading@10
|
1054 int *got_frame_ptr, AVPacket *avpkt)
|
yading@10
|
1055 {
|
yading@10
|
1056 const uint8_t *buf = avpkt->data;
|
yading@10
|
1057 int buf_size = avpkt->size;
|
yading@10
|
1058 MLPDecodeContext *m = avctx->priv_data;
|
yading@10
|
1059 GetBitContext gb;
|
yading@10
|
1060 unsigned int length, substr;
|
yading@10
|
1061 unsigned int substream_start;
|
yading@10
|
1062 unsigned int header_size = 4;
|
yading@10
|
1063 unsigned int substr_header_size = 0;
|
yading@10
|
1064 uint8_t substream_parity_present[MAX_SUBSTREAMS];
|
yading@10
|
1065 uint16_t substream_data_len[MAX_SUBSTREAMS];
|
yading@10
|
1066 uint8_t parity_bits;
|
yading@10
|
1067 int ret;
|
yading@10
|
1068
|
yading@10
|
1069 if (buf_size < 4)
|
yading@10
|
1070 return 0;
|
yading@10
|
1071
|
yading@10
|
1072 length = (AV_RB16(buf) & 0xfff) * 2;
|
yading@10
|
1073
|
yading@10
|
1074 if (length < 4 || length > buf_size)
|
yading@10
|
1075 return AVERROR_INVALIDDATA;
|
yading@10
|
1076
|
yading@10
|
1077 init_get_bits(&gb, (buf + 4), (length - 4) * 8);
|
yading@10
|
1078
|
yading@10
|
1079 m->is_major_sync_unit = 0;
|
yading@10
|
1080 if (show_bits_long(&gb, 31) == (0xf8726fba >> 1)) {
|
yading@10
|
1081 if (read_major_sync(m, &gb) < 0)
|
yading@10
|
1082 goto error;
|
yading@10
|
1083 m->is_major_sync_unit = 1;
|
yading@10
|
1084 header_size += 28;
|
yading@10
|
1085 }
|
yading@10
|
1086
|
yading@10
|
1087 if (!m->params_valid) {
|
yading@10
|
1088 av_log(m->avctx, AV_LOG_WARNING,
|
yading@10
|
1089 "Stream parameters not seen; skipping frame.\n");
|
yading@10
|
1090 *got_frame_ptr = 0;
|
yading@10
|
1091 return length;
|
yading@10
|
1092 }
|
yading@10
|
1093
|
yading@10
|
1094 substream_start = 0;
|
yading@10
|
1095
|
yading@10
|
1096 for (substr = 0; substr < m->num_substreams; substr++) {
|
yading@10
|
1097 int extraword_present, checkdata_present, end, nonrestart_substr;
|
yading@10
|
1098
|
yading@10
|
1099 extraword_present = get_bits1(&gb);
|
yading@10
|
1100 nonrestart_substr = get_bits1(&gb);
|
yading@10
|
1101 checkdata_present = get_bits1(&gb);
|
yading@10
|
1102 skip_bits1(&gb);
|
yading@10
|
1103
|
yading@10
|
1104 end = get_bits(&gb, 12) * 2;
|
yading@10
|
1105
|
yading@10
|
1106 substr_header_size += 2;
|
yading@10
|
1107
|
yading@10
|
1108 if (extraword_present) {
|
yading@10
|
1109 if (m->avctx->codec_id == AV_CODEC_ID_MLP) {
|
yading@10
|
1110 av_log(m->avctx, AV_LOG_ERROR, "There must be no extraword for MLP.\n");
|
yading@10
|
1111 goto error;
|
yading@10
|
1112 }
|
yading@10
|
1113 skip_bits(&gb, 16);
|
yading@10
|
1114 substr_header_size += 2;
|
yading@10
|
1115 }
|
yading@10
|
1116
|
yading@10
|
1117 if (!(nonrestart_substr ^ m->is_major_sync_unit)) {
|
yading@10
|
1118 av_log(m->avctx, AV_LOG_ERROR, "Invalid nonrestart_substr.\n");
|
yading@10
|
1119 goto error;
|
yading@10
|
1120 }
|
yading@10
|
1121
|
yading@10
|
1122 if (end + header_size + substr_header_size > length) {
|
yading@10
|
1123 av_log(m->avctx, AV_LOG_ERROR,
|
yading@10
|
1124 "Indicated length of substream %d data goes off end of "
|
yading@10
|
1125 "packet.\n", substr);
|
yading@10
|
1126 end = length - header_size - substr_header_size;
|
yading@10
|
1127 }
|
yading@10
|
1128
|
yading@10
|
1129 if (end < substream_start) {
|
yading@10
|
1130 av_log(avctx, AV_LOG_ERROR,
|
yading@10
|
1131 "Indicated end offset of substream %d data "
|
yading@10
|
1132 "is smaller than calculated start offset.\n",
|
yading@10
|
1133 substr);
|
yading@10
|
1134 goto error;
|
yading@10
|
1135 }
|
yading@10
|
1136
|
yading@10
|
1137 if (substr > m->max_decoded_substream)
|
yading@10
|
1138 continue;
|
yading@10
|
1139
|
yading@10
|
1140 substream_parity_present[substr] = checkdata_present;
|
yading@10
|
1141 substream_data_len[substr] = end - substream_start;
|
yading@10
|
1142 substream_start = end;
|
yading@10
|
1143 }
|
yading@10
|
1144
|
yading@10
|
1145 parity_bits = ff_mlp_calculate_parity(buf, 4);
|
yading@10
|
1146 parity_bits ^= ff_mlp_calculate_parity(buf + header_size, substr_header_size);
|
yading@10
|
1147
|
yading@10
|
1148 if ((((parity_bits >> 4) ^ parity_bits) & 0xF) != 0xF) {
|
yading@10
|
1149 av_log(avctx, AV_LOG_ERROR, "Parity check failed.\n");
|
yading@10
|
1150 goto error;
|
yading@10
|
1151 }
|
yading@10
|
1152
|
yading@10
|
1153 buf += header_size + substr_header_size;
|
yading@10
|
1154
|
yading@10
|
1155 for (substr = 0; substr <= m->max_decoded_substream; substr++) {
|
yading@10
|
1156 SubStream *s = &m->substream[substr];
|
yading@10
|
1157 init_get_bits(&gb, buf, substream_data_len[substr] * 8);
|
yading@10
|
1158
|
yading@10
|
1159 m->matrix_changed = 0;
|
yading@10
|
1160 memset(m->filter_changed, 0, sizeof(m->filter_changed));
|
yading@10
|
1161
|
yading@10
|
1162 s->blockpos = 0;
|
yading@10
|
1163 do {
|
yading@10
|
1164 if (get_bits1(&gb)) {
|
yading@10
|
1165 if (get_bits1(&gb)) {
|
yading@10
|
1166 /* A restart header should be present. */
|
yading@10
|
1167 if (read_restart_header(m, &gb, buf, substr) < 0)
|
yading@10
|
1168 goto next_substr;
|
yading@10
|
1169 s->restart_seen = 1;
|
yading@10
|
1170 }
|
yading@10
|
1171
|
yading@10
|
1172 if (!s->restart_seen)
|
yading@10
|
1173 goto next_substr;
|
yading@10
|
1174 if (read_decoding_params(m, &gb, substr) < 0)
|
yading@10
|
1175 goto next_substr;
|
yading@10
|
1176 }
|
yading@10
|
1177
|
yading@10
|
1178 if (!s->restart_seen)
|
yading@10
|
1179 goto next_substr;
|
yading@10
|
1180
|
yading@10
|
1181 if ((ret = read_block_data(m, &gb, substr)) < 0)
|
yading@10
|
1182 return ret;
|
yading@10
|
1183
|
yading@10
|
1184 if (get_bits_count(&gb) >= substream_data_len[substr] * 8)
|
yading@10
|
1185 goto substream_length_mismatch;
|
yading@10
|
1186
|
yading@10
|
1187 } while (!get_bits1(&gb));
|
yading@10
|
1188
|
yading@10
|
1189 skip_bits(&gb, (-get_bits_count(&gb)) & 15);
|
yading@10
|
1190
|
yading@10
|
1191 if (substream_data_len[substr] * 8 - get_bits_count(&gb) >= 32) {
|
yading@10
|
1192 int shorten_by;
|
yading@10
|
1193
|
yading@10
|
1194 if (get_bits(&gb, 16) != 0xD234)
|
yading@10
|
1195 return AVERROR_INVALIDDATA;
|
yading@10
|
1196
|
yading@10
|
1197 shorten_by = get_bits(&gb, 16);
|
yading@10
|
1198 if (m->avctx->codec_id == AV_CODEC_ID_TRUEHD && shorten_by & 0x2000)
|
yading@10
|
1199 s->blockpos -= FFMIN(shorten_by & 0x1FFF, s->blockpos);
|
yading@10
|
1200 else if (m->avctx->codec_id == AV_CODEC_ID_MLP && shorten_by != 0xD234)
|
yading@10
|
1201 return AVERROR_INVALIDDATA;
|
yading@10
|
1202
|
yading@10
|
1203 if (substr == m->max_decoded_substream)
|
yading@10
|
1204 av_log(m->avctx, AV_LOG_INFO, "End of stream indicated.\n");
|
yading@10
|
1205 }
|
yading@10
|
1206
|
yading@10
|
1207 if (substream_parity_present[substr]) {
|
yading@10
|
1208 uint8_t parity, checksum;
|
yading@10
|
1209
|
yading@10
|
1210 if (substream_data_len[substr] * 8 - get_bits_count(&gb) != 16)
|
yading@10
|
1211 goto substream_length_mismatch;
|
yading@10
|
1212
|
yading@10
|
1213 parity = ff_mlp_calculate_parity(buf, substream_data_len[substr] - 2);
|
yading@10
|
1214 checksum = ff_mlp_checksum8 (buf, substream_data_len[substr] - 2);
|
yading@10
|
1215
|
yading@10
|
1216 if ((get_bits(&gb, 8) ^ parity) != 0xa9 )
|
yading@10
|
1217 av_log(m->avctx, AV_LOG_ERROR, "Substream %d parity check failed.\n", substr);
|
yading@10
|
1218 if ( get_bits(&gb, 8) != checksum)
|
yading@10
|
1219 av_log(m->avctx, AV_LOG_ERROR, "Substream %d checksum failed.\n" , substr);
|
yading@10
|
1220 }
|
yading@10
|
1221
|
yading@10
|
1222 if (substream_data_len[substr] * 8 != get_bits_count(&gb))
|
yading@10
|
1223 goto substream_length_mismatch;
|
yading@10
|
1224
|
yading@10
|
1225 next_substr:
|
yading@10
|
1226 if (!s->restart_seen)
|
yading@10
|
1227 av_log(m->avctx, AV_LOG_ERROR,
|
yading@10
|
1228 "No restart header present in substream %d.\n", substr);
|
yading@10
|
1229
|
yading@10
|
1230 buf += substream_data_len[substr];
|
yading@10
|
1231 }
|
yading@10
|
1232
|
yading@10
|
1233 rematrix_channels(m, m->max_decoded_substream);
|
yading@10
|
1234
|
yading@10
|
1235 if ((ret = output_data(m, m->max_decoded_substream, data, got_frame_ptr)) < 0)
|
yading@10
|
1236 return ret;
|
yading@10
|
1237
|
yading@10
|
1238 return length;
|
yading@10
|
1239
|
yading@10
|
1240 substream_length_mismatch:
|
yading@10
|
1241 av_log(m->avctx, AV_LOG_ERROR, "substream %d length mismatch\n", substr);
|
yading@10
|
1242 return AVERROR_INVALIDDATA;
|
yading@10
|
1243
|
yading@10
|
1244 error:
|
yading@10
|
1245 m->params_valid = 0;
|
yading@10
|
1246 return AVERROR_INVALIDDATA;
|
yading@10
|
1247 }
|
yading@10
|
1248
|
yading@10
|
1249 #if CONFIG_MLP_DECODER
|
yading@10
|
1250 AVCodec ff_mlp_decoder = {
|
yading@10
|
1251 .name = "mlp",
|
yading@10
|
1252 .type = AVMEDIA_TYPE_AUDIO,
|
yading@10
|
1253 .id = AV_CODEC_ID_MLP,
|
yading@10
|
1254 .priv_data_size = sizeof(MLPDecodeContext),
|
yading@10
|
1255 .init = mlp_decode_init,
|
yading@10
|
1256 .decode = read_access_unit,
|
yading@10
|
1257 .capabilities = CODEC_CAP_DR1,
|
yading@10
|
1258 .long_name = NULL_IF_CONFIG_SMALL("MLP (Meridian Lossless Packing)"),
|
yading@10
|
1259 };
|
yading@10
|
1260 #endif
|
yading@10
|
1261 #if CONFIG_TRUEHD_DECODER
|
yading@10
|
1262 AVCodec ff_truehd_decoder = {
|
yading@10
|
1263 .name = "truehd",
|
yading@10
|
1264 .type = AVMEDIA_TYPE_AUDIO,
|
yading@10
|
1265 .id = AV_CODEC_ID_TRUEHD,
|
yading@10
|
1266 .priv_data_size = sizeof(MLPDecodeContext),
|
yading@10
|
1267 .init = mlp_decode_init,
|
yading@10
|
1268 .decode = read_access_unit,
|
yading@10
|
1269 .capabilities = CODEC_CAP_DR1,
|
yading@10
|
1270 .long_name = NULL_IF_CONFIG_SMALL("TrueHD"),
|
yading@10
|
1271 };
|
yading@10
|
1272 #endif /* CONFIG_TRUEHD_DECODER */
|