FFmpeg
mlpdec.c
Go to the documentation of this file.
1 /*
2  * MLP decoder
3  * Copyright (c) 2007-2008 Ian Caulfield
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21 
22 /**
23  * @file
24  * MLP decoder
25  */
26 
27 #include "config_components.h"
28 
29 #include <stdint.h>
30 
31 #include "avcodec.h"
32 #include "libavutil/internal.h"
33 #include "libavutil/intreadwrite.h"
35 #include "libavutil/mem_internal.h"
36 #include "libavutil/thread.h"
37 #include "libavutil/opt.h"
38 #include "codec_internal.h"
39 #include "decode.h"
40 #include "get_bits.h"
41 #include "mlp_parse.h"
42 #include "mlpdsp.h"
43 #include "mlp.h"
44 #include "config.h"
45 #include "profiles.h"
46 
47 /** number of bits used for VLC lookup - longest Huffman code is 9 */
48 #if ARCH_ARM
49 #define VLC_BITS 5
50 #define VLC_STATIC_SIZE 64
51 #else
52 #define VLC_BITS 9
53 #define VLC_STATIC_SIZE 512
54 #endif
55 
56 typedef struct SubStream {
57  /// Set if a valid restart header has been read. Otherwise the substream cannot be decoded.
58  uint8_t restart_seen;
59  /// Set if end of stream is encountered
60  uint8_t end_of_stream;
61 
62  //@{
63  /** restart header data */
64  /// The type of noise to be used in the rematrix stage.
65  uint16_t noise_type;
66 
67  /// The index of the first channel coded in this substream.
68  uint8_t min_channel;
69  /// The index of the last channel coded in this substream.
70  uint8_t max_channel;
71  /// The coded channels mask in this substream.
72  uint64_t coded_channels;
73  /// The number of channels input into the rematrix stage.
75  /// For each channel output by the matrix, the output channel to map it to
77  /// The channel layout for this substream
78  uint64_t mask;
79  /// The matrix encoding mode for this substream
82 
83  /// Channel coding parameters for channels in the substream
85 
86  /// The left shift applied to random noise in 0x31ea substreams.
87  uint8_t noise_shift;
88  /// The current seed value for the pseudorandom noise generator(s).
89  uint32_t noisegen_seed;
90 
91  /// Set if the substream contains extra info to check the size of VLC blocks.
93 
94  /// Bitmask of which parameter sets are conveyed in a decoding parameter block.
96  //@}
97 
98  //@{
99  /** matrix data */
100 
101  /// Number of matrices to be applied.
103 
104  /// matrix output channel
106 
107  /// Whether the LSBs of the matrix output are encoded in the bitstream.
109  /// Matrix coefficients, stored as 2.14 fixed point.
111  /// Left shift to apply to noise values in 0x31eb substreams.
113  //@}
114 
115  /// Left shift to apply to Huffman-decoded residuals.
117 
118  /// number of PCM samples in current audio block
119  uint16_t blocksize;
120  /// Number of PCM samples decoded so far in this frame.
121  uint16_t blockpos;
122 
123  /// Left shift to apply to decoded PCM values to get final 24-bit output.
125 
126  /// Running XOR of all output samples.
128 
129 } SubStream;
130 
131 typedef struct MLPDecodeContext {
132  const AVClass *class;
134 
136 
137  /// Current access unit being read has a major sync.
139 
140  /// Size of the major sync unit, in bytes
142 
143  /// Set if a valid major sync block has been read. Otherwise no decoding is possible.
144  uint8_t params_valid;
145 
146  /// Number of substreams contained within this stream.
147  uint8_t num_substreams;
148 
149  /// Which substream of substreams carry 16-channel presentation
151 
152  /// Which substream of substreams carry 2/6/8-channel presentation
153  uint8_t substream_info;
154 
155  /// Index of the last substream to decode - further substreams are skipped.
157 
158  /// Stream needs channel reordering to comply with FFmpeg's channel order
160 
161  /// number of PCM samples contained in each frame
163  /// next power of two above the number of samples in each frame
165 
167 
170 
174 
176  int32_t (*pack_output)(int32_t lossless_check_data,
177  uint16_t blockpos,
179  void *data,
180  uint8_t *ch_assign,
181  int8_t *output_shift,
182  uint8_t max_matrix_channel,
183  int is32);
185 
186 static const enum AVChannel thd_channel_order[] = {
189  AV_CHAN_LOW_FREQUENCY, // LFE
194  AV_CHAN_BACK_CENTER, // Cs
195  AV_CHAN_TOP_CENTER, // Ts
199  AV_CHAN_LOW_FREQUENCY_2, // LFE2
200 };
201 
203 {
206  av_channel_layout_subset(layout, UINT64_MAX);
207 }
208 
209 static enum AVChannel thd_channel_layout_extract_channel(uint64_t channel_layout,
210  int index)
211 {
212  int i;
213 
214  if (av_popcount64(channel_layout) <= index)
215  return AV_CHAN_NONE;
216 
217  for (i = 0; i < FF_ARRAY_ELEMS(thd_channel_order); i++)
218  if (channel_layout & (1ULL << thd_channel_order[i]) && !index--)
219  return thd_channel_order[i];
220  return AV_CHAN_NONE;
221 }
222 
223 static VLC huff_vlc[3];
224 
225 /** Initialize static data, constant between all invocations of the codec. */
226 
227 static av_cold void init_static(void)
228 {
229  for (int i = 0; i < 3; i++) {
230  static VLCElem vlc_buf[3 * VLC_STATIC_SIZE];
231  huff_vlc[i].table = &vlc_buf[i * VLC_STATIC_SIZE];
233  vlc_init(&huff_vlc[i], VLC_BITS, 18,
234  &ff_mlp_huffman_tables[i][0][1], 2, 1,
236  }
237 
238  ff_mlp_init_crc();
239 }
240 
242  unsigned int substr, unsigned int ch)
243 {
244  SubStream *s = &m->substream[substr];
245  ChannelParams *cp = &s->channel_params[ch];
246  int lsb_bits = cp->huff_lsbs - s->quant_step_size[ch];
247  int sign_shift = lsb_bits + (cp->codebook ? 2 - cp->codebook : -1);
248  int32_t sign_huff_offset = cp->huff_offset;
249 
250  if (cp->codebook > 0)
251  sign_huff_offset -= 7 << lsb_bits;
252 
253  if (sign_shift >= 0)
254  sign_huff_offset -= 1 << sign_shift;
255 
256  return sign_huff_offset;
257 }
258 
259 /** Read a sample, consisting of either, both or neither of entropy-coded MSBs
260  * and plain LSBs. */
261 
263  unsigned int substr, unsigned int pos)
264 {
265  SubStream *s = &m->substream[substr];
266  unsigned int mat, channel;
267 
268  for (mat = 0; mat < s->num_primitive_matrices; mat++)
269  if (s->lsb_bypass[mat])
270  m->bypassed_lsbs[pos + s->blockpos][mat] = get_bits1(gbp);
271 
272  for (channel = s->min_channel; channel <= s->max_channel; channel++) {
273  ChannelParams *cp = &s->channel_params[channel];
274  int codebook = cp->codebook;
275  int quant_step_size = s->quant_step_size[channel];
276  int lsb_bits = cp->huff_lsbs - quant_step_size;
277  int result = 0;
278 
279  if (codebook > 0)
281  VLC_BITS, (9 + VLC_BITS - 1) / VLC_BITS);
282 
283  if (result < 0)
284  return AVERROR_INVALIDDATA;
285 
286  if (lsb_bits > 0)
287  result = (result << lsb_bits) + get_bits_long(gbp, lsb_bits);
288 
289  result += cp->sign_huff_offset;
290  result *= 1 << quant_step_size;
291 
292  m->sample_buffer[pos + s->blockpos][channel] = result;
293  }
294 
295  return 0;
296 }
297 
299 {
300  static AVOnce init_static_once = AV_ONCE_INIT;
301  MLPDecodeContext *m = avctx->priv_data;
302  int substr;
303 
304  m->avctx = avctx;
305  for (substr = 0; substr < MAX_SUBSTREAMS; substr++)
306  m->substream[substr].lossless_check_data = 0xffffffff;
307  ff_mlpdsp_init(&m->dsp);
308 
309  if (m->downmix_layout.nb_channels) {
320  }
321  else
322  av_log(avctx, AV_LOG_WARNING, "Invalid downmix layout\n");
323  }
324 
325  ff_thread_once(&init_static_once, init_static);
326 
327  return 0;
328 }
329 
330 /** Read a major sync info header - contains high level information about
331  * the stream - sample rate, channel arrangement etc. Most of this
332  * information is not actually necessary for decoding, only for playback.
333  */
334 
336 {
338  int substr, ret;
339 
340  if ((ret = ff_mlp_read_major_sync(m->avctx, &mh, gb)) != 0)
341  return ret;
342 
343  if (mh.group1_bits == 0) {
344  av_log(m->avctx, AV_LOG_ERROR, "invalid/unknown bits per sample\n");
345  return AVERROR_INVALIDDATA;
346  }
347  if (mh.group2_bits > mh.group1_bits) {
349  "Channel group 2 cannot have more bits per sample than group 1.\n");
350  return AVERROR_INVALIDDATA;
351  }
352 
353  if (mh.group2_samplerate && mh.group2_samplerate != mh.group1_samplerate) {
355  "Channel groups with differing sample rates are not currently supported.\n");
356  return AVERROR_INVALIDDATA;
357  }
358 
359  if (mh.group1_samplerate == 0) {
360  av_log(m->avctx, AV_LOG_ERROR, "invalid/unknown sampling rate\n");
361  return AVERROR_INVALIDDATA;
362  }
363  if (mh.group1_samplerate > MAX_SAMPLERATE) {
365  "Sampling rate %d is greater than the supported maximum (%d).\n",
366  mh.group1_samplerate, MAX_SAMPLERATE);
367  return AVERROR_INVALIDDATA;
368  }
369  if (mh.access_unit_size > MAX_BLOCKSIZE) {
371  "Block size %d is greater than the supported maximum (%d).\n",
372  mh.access_unit_size, MAX_BLOCKSIZE);
373  return AVERROR_INVALIDDATA;
374  }
375  if (mh.access_unit_size_pow2 > MAX_BLOCKSIZE_POW2) {
377  "Block size pow2 %d is greater than the supported maximum (%d).\n",
378  mh.access_unit_size_pow2, MAX_BLOCKSIZE_POW2);
379  return AVERROR_INVALIDDATA;
380  }
381 
382  if (mh.num_substreams == 0)
383  return AVERROR_INVALIDDATA;
384  if (m->avctx->codec_id == AV_CODEC_ID_MLP && mh.num_substreams > 2) {
385  av_log(m->avctx, AV_LOG_ERROR, "MLP only supports up to 2 substreams.\n");
386  return AVERROR_INVALIDDATA;
387  }
388  if (mh.num_substreams > MAX_SUBSTREAMS) {
390  "%d substreams (more than the "
391  "maximum supported by the decoder)",
392  mh.num_substreams);
393  return AVERROR_PATCHWELCOME;
394  }
395 
396  m->major_sync_header_size = mh.header_size;
397 
398  m->access_unit_size = mh.access_unit_size;
399  m->access_unit_size_pow2 = mh.access_unit_size_pow2;
400 
401  m->num_substreams = mh.num_substreams;
402  m->extended_substream_info = mh.extended_substream_info;
403  m->substream_info = mh.substream_info;
404 
405  /* If there is a 4th substream and the MSB of substream_info is set,
406  * there is a 16-channel spatial presentation (Atmos in TrueHD).
407  */
409  && m->num_substreams == 4 && m->substream_info >> 7 == 1) {
411  }
412 
413  /* limit to decoding 3 substreams, as the 4th is used by Dolby Atmos for non-audio data */
415 
416  m->avctx->sample_rate = mh.group1_samplerate;
417  m->avctx->frame_size = mh.access_unit_size;
418 
419  m->avctx->bits_per_raw_sample = mh.group1_bits;
420  if (mh.group1_bits > 16)
422  else
428 
429  m->params_valid = 1;
430  for (substr = 0; substr < MAX_SUBSTREAMS; substr++)
431  m->substream[substr].restart_seen = 0;
432 
433  /* Set the layout for each substream. When there's more than one, the first
434  * substream is Stereo. Subsequent substreams' layouts are indicated in the
435  * major sync. */
436  if (m->avctx->codec_id == AV_CODEC_ID_MLP) {
437  if (mh.stream_type != SYNC_MLP) {
439  "unexpected stream_type %X in MLP",
440  mh.stream_type);
441  return AVERROR_PATCHWELCOME;
442  }
443  if ((substr = (mh.num_substreams > 1)))
445  m->substream[substr].mask = mh.channel_layout_mlp;
446  } else {
447  if (mh.stream_type != SYNC_TRUEHD) {
449  "unexpected stream_type %X in !MLP",
450  mh.stream_type);
451  return AVERROR_PATCHWELCOME;
452  }
453  m->substream[1].mask = mh.channel_layout_thd_stream1;
454  if (mh.channels_thd_stream1 == 2 &&
455  mh.channels_thd_stream2 == 2 &&
456  m->avctx->ch_layout.nb_channels == 2)
458  if ((substr = (mh.num_substreams > 1)))
460  if (mh.num_substreams == 1 &&
461  mh.channels_thd_stream1 == 1 &&
462  mh.channels_thd_stream2 == 1 &&
463  m->avctx->ch_layout.nb_channels == 1)
465  if (mh.num_substreams > 2)
466  if (mh.channel_layout_thd_stream2)
467  m->substream[2].mask = mh.channel_layout_thd_stream2;
468  else
469  m->substream[2].mask = mh.channel_layout_thd_stream1;
470  if (m->avctx->ch_layout.nb_channels > 2)
471  if (mh.num_substreams > 2)
472  m->substream[1].mask = mh.channel_layout_thd_stream1;
473  else
474  m->substream[mh.num_substreams > 1].mask = mh.channel_layout_thd_stream2;
475  }
476 
477  m->needs_reordering = mh.channel_arrangement >= 18 && mh.channel_arrangement <= 20;
478 
479  /* Parse the TrueHD decoder channel modifiers and set each substream's
480  * AVMatrixEncoding accordingly.
481  *
482  * The meaning of the modifiers depends on the channel layout:
483  *
484  * - THD_CH_MODIFIER_LTRT, THD_CH_MODIFIER_LBINRBIN only apply to 2-channel
485  *
486  * - THD_CH_MODIFIER_MONO applies to 1-channel or 2-channel (dual mono)
487  *
488  * - THD_CH_MODIFIER_SURROUNDEX, THD_CH_MODIFIER_NOTSURROUNDEX only apply to
489  * layouts with an Ls/Rs channel pair
490  */
491  for (substr = 0; substr < MAX_SUBSTREAMS; substr++)
494  if (mh.num_substreams > 2 &&
495  mh.channel_layout_thd_stream2 & AV_CH_SIDE_LEFT &&
496  mh.channel_layout_thd_stream2 & AV_CH_SIDE_RIGHT &&
497  mh.channel_modifier_thd_stream2 == THD_CH_MODIFIER_SURROUNDEX)
499 
500  if (mh.num_substreams > 1 &&
501  mh.channel_layout_thd_stream1 & AV_CH_SIDE_LEFT &&
502  mh.channel_layout_thd_stream1 & AV_CH_SIDE_RIGHT &&
503  mh.channel_modifier_thd_stream1 == THD_CH_MODIFIER_SURROUNDEX)
505 
506  if (mh.num_substreams > 0)
507  switch (mh.channel_modifier_thd_stream0) {
510  break;
513  break;
514  default:
515  break;
516  }
517  }
518 
519  return 0;
520 }
521 
522 /** Read a restart header from a block in a substream. This contains parameters
523  * required to decode the audio that do not change very often. Generally
524  * (always) present only in blocks following a major sync. */
525 
527  const uint8_t *buf, unsigned int substr)
528 {
529  SubStream *s = &m->substream[substr];
530  unsigned int ch;
531  int sync_word, tmp;
532  uint8_t checksum;
533  uint8_t lossless_check;
534  int start_count = get_bits_count(gbp);
535  int min_channel, max_channel, max_matrix_channel, noise_type;
536  const int std_max_matrix_channel = m->avctx->codec_id == AV_CODEC_ID_MLP
539 
540  sync_word = get_bits(gbp, 13);
541 
542  if (sync_word != 0x31ea >> 1) {
544  "restart header sync incorrect (got 0x%04x)\n", sync_word);
545  return AVERROR_INVALIDDATA;
546  }
547 
548  noise_type = get_bits1(gbp);
549 
550  if (m->avctx->codec_id == AV_CODEC_ID_MLP && noise_type) {
551  av_log(m->avctx, AV_LOG_ERROR, "MLP must have 0x31ea sync word.\n");
552  return AVERROR_INVALIDDATA;
553  }
554 
555  skip_bits(gbp, 16); /* Output timestamp */
556 
557  min_channel = get_bits(gbp, 4);
558  max_channel = get_bits(gbp, 4);
559  max_matrix_channel = get_bits(gbp, 4);
560 
561  if (max_matrix_channel > std_max_matrix_channel) {
563  "Max matrix channel cannot be greater than %d.\n",
564  std_max_matrix_channel);
565  return AVERROR_INVALIDDATA;
566  }
567 
568  /* This should happen for TrueHD streams with >6 channels and MLP's noise
569  * type. It is not yet known if this is allowed. */
570  if (max_matrix_channel > MAX_MATRIX_CHANNEL_MLP && !noise_type) {
572  "%d channels (more than the "
573  "maximum supported by the decoder)",
574  max_channel + 2);
575  return AVERROR_PATCHWELCOME;
576  }
577 
578  if (max_channel + 1 > MAX_CHANNELS || max_channel + 1 < min_channel)
579  return AVERROR_INVALIDDATA;
580 
581  s->min_channel = min_channel;
582  s->max_channel = max_channel;
583  s->coded_channels = ((1LL << (max_channel - min_channel + 1)) - 1) << min_channel;
584  s->max_matrix_channel = max_matrix_channel;
585  s->noise_type = noise_type;
586 
587  if (mlp_channel_layout_subset(&m->downmix_layout, s->mask) &&
588  m->max_decoded_substream > substr) {
590  "Extracting %d-channel downmix (0x%"PRIx64") from substream %d. "
591  "Further substreams will be skipped.\n",
592  s->max_channel + 1, s->mask, substr);
593  m->max_decoded_substream = substr;
594  }
595 
596  s->noise_shift = get_bits(gbp, 4);
597  s->noisegen_seed = get_bits(gbp, 23);
598 
599  skip_bits(gbp, 19);
600 
601  s->data_check_present = get_bits1(gbp);
602  lossless_check = get_bits(gbp, 8);
603  if (substr == m->max_decoded_substream
604  && s->lossless_check_data != 0xffffffff) {
605  tmp = xor_32_to_8(s->lossless_check_data);
606  if (tmp != lossless_check)
608  "Lossless check failed - expected %02x, calculated %02x.\n",
609  lossless_check, tmp);
610  }
611 
612  skip_bits(gbp, 16);
613 
614  memset(s->ch_assign, 0, sizeof(s->ch_assign));
615 
616  for (ch = 0; ch <= s->max_matrix_channel; ch++) {
617  int ch_assign = get_bits(gbp, 6);
618  if (m->avctx->codec_id == AV_CODEC_ID_TRUEHD) {
619  AVChannelLayout l;
620  enum AVChannel channel = thd_channel_layout_extract_channel(s->mask, ch_assign);
621 
622  av_channel_layout_from_mask(&l, s->mask);
624  }
625  if (ch_assign < 0 || ch_assign > s->max_matrix_channel) {
627  "Assignment of matrix channel %d to invalid output channel %d",
628  ch, ch_assign);
629  return AVERROR_PATCHWELCOME;
630  }
631  s->ch_assign[ch_assign] = ch;
632  }
633 
634  checksum = ff_mlp_restart_checksum(buf, get_bits_count(gbp) - start_count);
635 
636  if (checksum != get_bits(gbp, 8))
637  av_log(m->avctx, AV_LOG_ERROR, "restart header checksum error\n");
638 
639  /* Set default decoding parameters. */
640  s->param_presence_flags = 0xff;
641  s->num_primitive_matrices = 0;
642  s->blocksize = 8;
643  s->lossless_check_data = 0;
644 
645  memset(s->output_shift , 0, sizeof(s->output_shift ));
646  memset(s->quant_step_size, 0, sizeof(s->quant_step_size));
647 
648  for (ch = s->min_channel; ch <= s->max_channel; ch++) {
649  ChannelParams *cp = &s->channel_params[ch];
650  cp->filter_params[FIR].order = 0;
651  cp->filter_params[IIR].order = 0;
652  cp->filter_params[FIR].shift = 0;
653  cp->filter_params[IIR].shift = 0;
654 
655  /* Default audio coding is 24-bit raw PCM. */
656  cp->huff_offset = 0;
657  cp->sign_huff_offset = -(1 << 23);
658  cp->codebook = 0;
659  cp->huff_lsbs = 24;
660  }
661 
662  if (substr == m->max_decoded_substream) {
665  m->pack_output = m->dsp.mlp_select_pack_output(s->ch_assign,
666  s->output_shift,
667  s->max_matrix_channel,
669 
670  if (m->avctx->codec_id == AV_CODEC_ID_MLP && m->needs_reordering) {
671  if (s->mask == (AV_CH_LAYOUT_QUAD|AV_CH_LOW_FREQUENCY) ||
672  s->mask == AV_CH_LAYOUT_5POINT0_BACK) {
673  int i = s->ch_assign[4];
674  s->ch_assign[4] = s->ch_assign[3];
675  s->ch_assign[3] = s->ch_assign[2];
676  s->ch_assign[2] = i;
677  } else if (s->mask == AV_CH_LAYOUT_5POINT1_BACK) {
678  FFSWAP(int, s->ch_assign[2], s->ch_assign[4]);
679  FFSWAP(int, s->ch_assign[3], s->ch_assign[5]);
680  }
681  }
682 
683  }
684 
685  return 0;
686 }
687 
688 /** Read parameters for one of the prediction filters. */
689 
691  unsigned int substr, unsigned int channel,
692  unsigned int filter)
693 {
694  SubStream *s = &m->substream[substr];
695  FilterParams *fp = &s->channel_params[channel].filter_params[filter];
696  const int max_order = filter ? MAX_IIR_ORDER : MAX_FIR_ORDER;
697  const char fchar = filter ? 'I' : 'F';
698  int i, order;
699 
700  // Filter is 0 for FIR, 1 for IIR.
701  av_assert0(filter < 2);
702 
703  if (m->filter_changed[channel][filter]++ > 1) {
704  av_log(m->avctx, AV_LOG_ERROR, "Filters may change only once per access unit.\n");
705  return AVERROR_INVALIDDATA;
706  }
707 
708  order = get_bits(gbp, 4);
709  if (order > max_order) {
711  "%cIR filter order %d is greater than maximum %d.\n",
712  fchar, order, max_order);
713  return AVERROR_INVALIDDATA;
714  }
715  fp->order = order;
716 
717  if (order > 0) {
718  int32_t *fcoeff = s->channel_params[channel].coeff[filter];
719  int coeff_bits, coeff_shift;
720 
721  fp->shift = get_bits(gbp, 4);
722 
723  coeff_bits = get_bits(gbp, 5);
724  coeff_shift = get_bits(gbp, 3);
725  if (coeff_bits < 1 || coeff_bits > 16) {
727  "%cIR filter coeff_bits must be between 1 and 16.\n",
728  fchar);
729  return AVERROR_INVALIDDATA;
730  }
731  if (coeff_bits + coeff_shift > 16) {
733  "Sum of coeff_bits and coeff_shift for %cIR filter must be 16 or less.\n",
734  fchar);
735  return AVERROR_INVALIDDATA;
736  }
737 
738  for (i = 0; i < order; i++)
739  fcoeff[i] = get_sbits(gbp, coeff_bits) * (1 << coeff_shift);
740 
741  if (get_bits1(gbp)) {
742  int state_bits, state_shift;
743 
744  if (filter == FIR) {
746  "FIR filter has state data specified.\n");
747  return AVERROR_INVALIDDATA;
748  }
749 
750  state_bits = get_bits(gbp, 4);
751  state_shift = get_bits(gbp, 4);
752 
753  /* TODO: Check validity of state data. */
754 
755  for (i = 0; i < order; i++)
756  fp->state[i] = state_bits ? get_sbits(gbp, state_bits) * (1 << state_shift) : 0;
757  }
758  }
759 
760  return 0;
761 }
762 
763 /** Read parameters for primitive matrices. */
764 
765 static int read_matrix_params(MLPDecodeContext *m, unsigned int substr, GetBitContext *gbp)
766 {
767  SubStream *s = &m->substream[substr];
768  unsigned int mat, ch;
769  const int max_primitive_matrices = m->avctx->codec_id == AV_CODEC_ID_MLP
772 
773  if (m->matrix_changed++ > 1) {
774  av_log(m->avctx, AV_LOG_ERROR, "Matrices may change only once per access unit.\n");
775  return AVERROR_INVALIDDATA;
776  }
777 
778  s->num_primitive_matrices = get_bits(gbp, 4);
779 
780  if (s->num_primitive_matrices > max_primitive_matrices) {
782  "Number of primitive matrices cannot be greater than %d.\n",
783  max_primitive_matrices);
784  goto error;
785  }
786 
787  for (mat = 0; mat < s->num_primitive_matrices; mat++) {
788  int frac_bits, max_chan;
789  s->matrix_out_ch[mat] = get_bits(gbp, 4);
790  frac_bits = get_bits(gbp, 4);
791  s->lsb_bypass [mat] = get_bits1(gbp);
792 
793  if (s->matrix_out_ch[mat] > s->max_matrix_channel) {
795  "Invalid channel %d specified as output from matrix.\n",
796  s->matrix_out_ch[mat]);
797  goto error;
798  }
799  if (frac_bits > 14) {
801  "Too many fractional bits specified.\n");
802  goto error;
803  }
804 
805  max_chan = s->max_matrix_channel;
806  if (!s->noise_type)
807  max_chan+=2;
808 
809  for (ch = 0; ch <= max_chan; ch++) {
810  int coeff_val = 0;
811  if (get_bits1(gbp))
812  coeff_val = get_sbits(gbp, frac_bits + 2);
813 
814  s->matrix_coeff[mat][ch] = coeff_val * (1 << (14 - frac_bits));
815  }
816 
817  if (s->noise_type)
818  s->matrix_noise_shift[mat] = get_bits(gbp, 4);
819  else
820  s->matrix_noise_shift[mat] = 0;
821  }
822 
823  return 0;
824 error:
825  s->num_primitive_matrices = 0;
826  memset(s->matrix_out_ch, 0, sizeof(s->matrix_out_ch));
827 
828  return AVERROR_INVALIDDATA;
829 }
830 
831 /** Read channel parameters. */
832 
833 static int read_channel_params(MLPDecodeContext *m, unsigned int substr,
834  GetBitContext *gbp, unsigned int ch)
835 {
836  SubStream *s = &m->substream[substr];
837  ChannelParams *cp = &s->channel_params[ch];
838  FilterParams *fir = &cp->filter_params[FIR];
839  FilterParams *iir = &cp->filter_params[IIR];
840  int ret;
841 
842  if (s->param_presence_flags & PARAM_FIR)
843  if (get_bits1(gbp))
844  if ((ret = read_filter_params(m, gbp, substr, ch, FIR)) < 0)
845  return ret;
846 
847  if (s->param_presence_flags & PARAM_IIR)
848  if (get_bits1(gbp))
849  if ((ret = read_filter_params(m, gbp, substr, ch, IIR)) < 0)
850  return ret;
851 
852  if (fir->order + iir->order > 8) {
853  av_log(m->avctx, AV_LOG_ERROR, "Total filter orders too high.\n");
854  return AVERROR_INVALIDDATA;
855  }
856 
857  if (fir->order && iir->order &&
858  fir->shift != iir->shift) {
860  "FIR and IIR filters must use the same precision.\n");
861  return AVERROR_INVALIDDATA;
862  }
863  /* The FIR and IIR filters must have the same precision.
864  * To simplify the filtering code, only the precision of the
865  * FIR filter is considered. If only the IIR filter is employed,
866  * the FIR filter precision is set to that of the IIR filter, so
867  * that the filtering code can use it. */
868  if (!fir->order && iir->order)
869  fir->shift = iir->shift;
870 
871  if (s->param_presence_flags & PARAM_HUFFOFFSET)
872  if (get_bits1(gbp))
873  cp->huff_offset = get_sbits(gbp, 15);
874 
875  cp->codebook = get_bits(gbp, 2);
876  cp->huff_lsbs = get_bits(gbp, 5);
877 
878  if (cp->codebook > 0 && cp->huff_lsbs > 24) {
879  av_log(m->avctx, AV_LOG_ERROR, "Invalid huff_lsbs.\n");
880  cp->huff_lsbs = 0;
881  return AVERROR_INVALIDDATA;
882  }
883 
884  return 0;
885 }
886 
887 /** Read decoding parameters that change more often than those in the restart
888  * header. */
889 
891  unsigned int substr)
892 {
893  SubStream *s = &m->substream[substr];
894  unsigned int ch;
895  int ret = 0;
896  unsigned recompute_sho = 0;
897 
898  if (s->param_presence_flags & PARAM_PRESENCE)
899  if (get_bits1(gbp))
900  s->param_presence_flags = get_bits(gbp, 8);
901 
902  if (s->param_presence_flags & PARAM_BLOCKSIZE)
903  if (get_bits1(gbp)) {
904  s->blocksize = get_bits(gbp, 9);
905  if (s->blocksize < 8 || s->blocksize > m->access_unit_size) {
906  av_log(m->avctx, AV_LOG_ERROR, "Invalid blocksize.\n");
907  s->blocksize = 0;
908  return AVERROR_INVALIDDATA;
909  }
910  }
911 
912  if (s->param_presence_flags & PARAM_MATRIX)
913  if (get_bits1(gbp))
914  if ((ret = read_matrix_params(m, substr, gbp)) < 0)
915  return ret;
916 
917  if (s->param_presence_flags & PARAM_OUTSHIFT)
918  if (get_bits1(gbp)) {
919  for (ch = 0; ch <= s->max_matrix_channel; ch++) {
920  s->output_shift[ch] = get_sbits(gbp, 4);
921  if (s->output_shift[ch] < 0) {
922  avpriv_request_sample(m->avctx, "Negative output_shift");
923  s->output_shift[ch] = 0;
924  }
925  }
926  if (substr == m->max_decoded_substream)
927  m->pack_output = m->dsp.mlp_select_pack_output(s->ch_assign,
928  s->output_shift,
929  s->max_matrix_channel,
931  }
932 
933  if (s->param_presence_flags & PARAM_QUANTSTEP)
934  if (get_bits1(gbp))
935  for (ch = 0; ch <= s->max_channel; ch++) {
936  s->quant_step_size[ch] = get_bits(gbp, 4);
937 
938  recompute_sho |= 1<<ch;
939  }
940 
941  for (ch = s->min_channel; ch <= s->max_channel; ch++)
942  if (get_bits1(gbp)) {
943  recompute_sho |= 1<<ch;
944  if ((ret = read_channel_params(m, substr, gbp, ch)) < 0)
945  goto fail;
946  }
947 
948 
949 fail:
950  for (ch = 0; ch <= s->max_channel; ch++) {
951  if (recompute_sho & (1<<ch)) {
952  ChannelParams *cp = &s->channel_params[ch];
953 
954  if (cp->codebook > 0 && cp->huff_lsbs < s->quant_step_size[ch]) {
955  if (ret >= 0) {
956  av_log(m->avctx, AV_LOG_ERROR, "quant_step_size larger than huff_lsbs\n");
958  }
959  s->quant_step_size[ch] = 0;
960  }
961 
962  cp->sign_huff_offset = calculate_sign_huff(m, substr, ch);
963  }
964  }
965  return ret;
966 }
967 
968 #define MSB_MASK(bits) (-(1 << (bits)))
969 
970 /** Generate PCM samples using the prediction filters and residual values
971  * read from the data stream, and update the filter state. */
972 
973 static void filter_channel(MLPDecodeContext *m, unsigned int substr,
974  unsigned int channel)
975 {
976  SubStream *s = &m->substream[substr];
977  const int32_t *fircoeff = s->channel_params[channel].coeff[FIR];
979  int32_t *firbuf = state_buffer[FIR] + MAX_BLOCKSIZE;
980  int32_t *iirbuf = state_buffer[IIR] + MAX_BLOCKSIZE;
981  FilterParams *fir = &s->channel_params[channel].filter_params[FIR];
982  FilterParams *iir = &s->channel_params[channel].filter_params[IIR];
983  unsigned int filter_shift = fir->shift;
984  int32_t mask = MSB_MASK(s->quant_step_size[channel]);
985 
986  memcpy(firbuf, fir->state, MAX_FIR_ORDER * sizeof(int32_t));
987  memcpy(iirbuf, iir->state, MAX_IIR_ORDER * sizeof(int32_t));
988 
989  m->dsp.mlp_filter_channel(firbuf, fircoeff,
990  fir->order, iir->order,
991  filter_shift, mask, s->blocksize,
992  &m->sample_buffer[s->blockpos][channel]);
993 
994  memcpy(fir->state, firbuf - s->blocksize, MAX_FIR_ORDER * sizeof(int32_t));
995  memcpy(iir->state, iirbuf - s->blocksize, MAX_IIR_ORDER * sizeof(int32_t));
996 }
997 
998 /** Read a block of PCM residual data (or actual if no filtering active). */
999 
1001  unsigned int substr)
1002 {
1003  SubStream *s = &m->substream[substr];
1004  unsigned int i, ch, expected_stream_pos = 0;
1005  int ret;
1006 
1007  if (s->data_check_present) {
1008  expected_stream_pos = get_bits_count(gbp);
1009  expected_stream_pos += get_bits(gbp, 16);
1011  "Substreams with VLC block size check info");
1012  }
1013 
1014  if (s->blockpos + s->blocksize > m->access_unit_size) {
1015  av_log(m->avctx, AV_LOG_ERROR, "too many audio samples in frame\n");
1016  return AVERROR_INVALIDDATA;
1017  }
1018 
1019  memset(&m->bypassed_lsbs[s->blockpos][0], 0,
1020  s->blocksize * sizeof(m->bypassed_lsbs[0]));
1021 
1022  for (i = 0; i < s->blocksize; i++)
1023  if ((ret = read_huff_channels(m, gbp, substr, i)) < 0)
1024  return ret;
1025 
1026  for (ch = s->min_channel; ch <= s->max_channel; ch++)
1027  filter_channel(m, substr, ch);
1028 
1029  s->blockpos += s->blocksize;
1030 
1031  if (s->data_check_present) {
1032  if (get_bits_count(gbp) != expected_stream_pos)
1033  av_log(m->avctx, AV_LOG_ERROR, "block data length mismatch\n");
1034  skip_bits(gbp, 8);
1035  }
1036 
1037  return 0;
1038 }
1039 
1040 /** Data table used for TrueHD noise generation function. */
1041 
1042 static const int8_t noise_table[256] = {
1043  30, 51, 22, 54, 3, 7, -4, 38, 14, 55, 46, 81, 22, 58, -3, 2,
1044  52, 31, -7, 51, 15, 44, 74, 30, 85, -17, 10, 33, 18, 80, 28, 62,
1045  10, 32, 23, 69, 72, 26, 35, 17, 73, 60, 8, 56, 2, 6, -2, -5,
1046  51, 4, 11, 50, 66, 76, 21, 44, 33, 47, 1, 26, 64, 48, 57, 40,
1047  38, 16, -10, -28, 92, 22, -18, 29, -10, 5, -13, 49, 19, 24, 70, 34,
1048  61, 48, 30, 14, -6, 25, 58, 33, 42, 60, 67, 17, 54, 17, 22, 30,
1049  67, 44, -9, 50, -11, 43, 40, 32, 59, 82, 13, 49, -14, 55, 60, 36,
1050  48, 49, 31, 47, 15, 12, 4, 65, 1, 23, 29, 39, 45, -2, 84, 69,
1051  0, 72, 37, 57, 27, 41, -15, -16, 35, 31, 14, 61, 24, 0, 27, 24,
1052  16, 41, 55, 34, 53, 9, 56, 12, 25, 29, 53, 5, 20, -20, -8, 20,
1053  13, 28, -3, 78, 38, 16, 11, 62, 46, 29, 21, 24, 46, 65, 43, -23,
1054  89, 18, 74, 21, 38, -12, 19, 12, -19, 8, 15, 33, 4, 57, 9, -8,
1055  36, 35, 26, 28, 7, 83, 63, 79, 75, 11, 3, 87, 37, 47, 34, 40,
1056  39, 19, 20, 42, 27, 34, 39, 77, 13, 42, 59, 64, 45, -1, 32, 37,
1057  45, -5, 53, -6, 7, 36, 50, 23, 6, 32, 9, -21, 18, 71, 27, 52,
1058  -25, 31, 35, 42, -1, 68, 63, 52, 26, 43, 66, 37, 41, 25, 40, 70,
1059 };
1060 
1061 /** Noise generation functions.
1062  * I'm not sure what these are for - they seem to be some kind of pseudorandom
1063  * sequence generators, used to generate noise data which is used when the
1064  * channels are rematrixed. I'm not sure if they provide a practical benefit
1065  * to compression, or just obfuscate the decoder. Are they for some kind of
1066  * dithering? */
1067 
1068 /** Generate two channels of noise, used in the matrix when
1069  * restart sync word == 0x31ea. */
1070 
1071 static void generate_2_noise_channels(MLPDecodeContext *m, unsigned int substr)
1072 {
1073  SubStream *s = &m->substream[substr];
1074  unsigned int i;
1075  uint32_t seed = s->noisegen_seed;
1076  unsigned int maxchan = s->max_matrix_channel;
1077 
1078  for (i = 0; i < s->blockpos; i++) {
1079  uint16_t seed_shr7 = seed >> 7;
1080  m->sample_buffer[i][maxchan+1] = ((int8_t)(seed >> 15)) * (1 << s->noise_shift);
1081  m->sample_buffer[i][maxchan+2] = ((int8_t) seed_shr7) * (1 << s->noise_shift);
1082 
1083  seed = (seed << 16) ^ seed_shr7 ^ (seed_shr7 << 5);
1084  }
1085 
1086  s->noisegen_seed = seed;
1087 }
1088 
1089 /** Generate a block of noise, used when restart sync word == 0x31eb. */
1090 
1091 static void fill_noise_buffer(MLPDecodeContext *m, unsigned int substr)
1092 {
1093  SubStream *s = &m->substream[substr];
1094  unsigned int i;
1095  uint32_t seed = s->noisegen_seed;
1096 
1097  for (i = 0; i < m->access_unit_size_pow2; i++) {
1098  uint8_t seed_shr15 = seed >> 15;
1099  m->noise_buffer[i] = noise_table[seed_shr15];
1100  seed = (seed << 8) ^ seed_shr15 ^ (seed_shr15 << 5);
1101  }
1102 
1103  s->noisegen_seed = seed;
1104 }
1105 
1106 /** Write the audio data into the output buffer. */
1107 
1108 static int output_data(MLPDecodeContext *m, unsigned int substr,
1109  AVFrame *frame, int *got_frame_ptr)
1110 {
1111  AVCodecContext *avctx = m->avctx;
1112  SubStream *s = &m->substream[substr];
1113  unsigned int mat;
1114  unsigned int maxchan;
1115  int ret;
1116  int is32 = (m->avctx->sample_fmt == AV_SAMPLE_FMT_S32);
1117 
1118  if (m->avctx->ch_layout.nb_channels != s->max_matrix_channel + 1) {
1119  av_log(m->avctx, AV_LOG_ERROR, "channel count mismatch\n");
1120  return AVERROR_INVALIDDATA;
1121  }
1122 
1123  if (!s->blockpos) {
1124  av_log(avctx, AV_LOG_ERROR, "No samples to output.\n");
1125  return AVERROR_INVALIDDATA;
1126  }
1127 
1128  maxchan = s->max_matrix_channel;
1129  if (!s->noise_type) {
1130  generate_2_noise_channels(m, substr);
1131  maxchan += 2;
1132  } else {
1133  fill_noise_buffer(m, substr);
1134  }
1135 
1136  /* Apply the channel matrices in turn to reconstruct the original audio
1137  * samples. */
1138  for (mat = 0; mat < s->num_primitive_matrices; mat++) {
1139  unsigned int dest_ch = s->matrix_out_ch[mat];
1140  m->dsp.mlp_rematrix_channel(&m->sample_buffer[0][0],
1141  s->matrix_coeff[mat],
1142  &m->bypassed_lsbs[0][mat],
1143  m->noise_buffer,
1144  s->num_primitive_matrices - mat,
1145  dest_ch,
1146  s->blockpos,
1147  maxchan,
1148  s->matrix_noise_shift[mat],
1150  MSB_MASK(s->quant_step_size[dest_ch]));
1151  }
1152 
1153  /* get output buffer */
1154  frame->nb_samples = s->blockpos;
1155  if ((ret = ff_get_buffer(avctx, frame, 0)) < 0)
1156  return ret;
1157  s->lossless_check_data = m->pack_output(s->lossless_check_data,
1158  s->blockpos,
1159  m->sample_buffer,
1160  frame->data[0],
1161  s->ch_assign,
1162  s->output_shift,
1163  s->max_matrix_channel,
1164  is32);
1165 
1166  /* Update matrix encoding side data */
1167  if (s->matrix_encoding != s->prev_matrix_encoding) {
1168  if ((ret = ff_side_data_update_matrix_encoding(frame, s->matrix_encoding)) < 0)
1169  return ret;
1170 
1171  s->prev_matrix_encoding = s->matrix_encoding;
1172  }
1173 
1174  *got_frame_ptr = 1;
1175 
1176  return 0;
1177 }
1178 
1179 /** Read an access unit from the stream.
1180  * @return negative on error, 0 if not enough data is present in the input stream,
1181  * otherwise the number of bytes consumed. */
1182 
1184  int *got_frame_ptr, AVPacket *avpkt)
1185 {
1186  const uint8_t *buf = avpkt->data;
1187  int buf_size = avpkt->size;
1188  MLPDecodeContext *m = avctx->priv_data;
1189  GetBitContext gb;
1190  unsigned int length, substr;
1191  unsigned int substream_start;
1192  unsigned int header_size = 4;
1193  unsigned int substr_header_size = 0;
1194  uint8_t substream_parity_present[MAX_SUBSTREAMS];
1195  uint16_t substream_data_len[MAX_SUBSTREAMS];
1196  uint8_t parity_bits;
1197  int ret;
1198 
1199  if (buf_size < 4)
1200  return AVERROR_INVALIDDATA;
1201 
1202  length = (AV_RB16(buf) & 0xfff) * 2;
1203 
1204  if (length < 4 || length > buf_size)
1205  return AVERROR_INVALIDDATA;
1206 
1207  init_get_bits(&gb, (buf + 4), (length - 4) * 8);
1208 
1209  m->is_major_sync_unit = 0;
1210  if (show_bits_long(&gb, 31) == (0xf8726fba >> 1)) {
1211  if (read_major_sync(m, &gb) < 0)
1212  goto error;
1213  m->is_major_sync_unit = 1;
1214  header_size += m->major_sync_header_size;
1215  frame->flags |= AV_FRAME_FLAG_KEY;
1216  }
1217 
1218  if (!m->params_valid) {
1220  "Stream parameters not seen; skipping frame.\n");
1221  *got_frame_ptr = 0;
1222  return length;
1223  }
1224 
1225  substream_start = 0;
1226 
1227  for (substr = 0; substr < m->num_substreams; substr++) {
1228  int extraword_present, checkdata_present, end, nonrestart_substr;
1229 
1230  extraword_present = get_bits1(&gb);
1231  nonrestart_substr = get_bits1(&gb);
1232  checkdata_present = get_bits1(&gb);
1233  skip_bits1(&gb);
1234 
1235  end = get_bits(&gb, 12) * 2;
1236 
1237  substr_header_size += 2;
1238 
1239  if (extraword_present) {
1240  if (m->avctx->codec_id == AV_CODEC_ID_MLP) {
1241  av_log(m->avctx, AV_LOG_ERROR, "There must be no extraword for MLP.\n");
1242  goto error;
1243  }
1244  skip_bits(&gb, 16);
1245  substr_header_size += 2;
1246  }
1247 
1248  if (length < header_size + substr_header_size) {
1249  av_log(m->avctx, AV_LOG_ERROR, "Insufficient data for headers\n");
1250  goto error;
1251  }
1252 
1253  if (!(nonrestart_substr ^ m->is_major_sync_unit)) {
1254  av_log(m->avctx, AV_LOG_ERROR, "Invalid nonrestart_substr.\n");
1255  goto error;
1256  }
1257 
1258  if (end + header_size + substr_header_size > length) {
1260  "Indicated length of substream %d data goes off end of "
1261  "packet.\n", substr);
1262  end = length - header_size - substr_header_size;
1263  }
1264 
1265  if (end < substream_start) {
1266  av_log(avctx, AV_LOG_ERROR,
1267  "Indicated end offset of substream %d data "
1268  "is smaller than calculated start offset.\n",
1269  substr);
1270  goto error;
1271  }
1272 
1273  if (substr > m->max_decoded_substream)
1274  continue;
1275 
1276  substream_parity_present[substr] = checkdata_present;
1277  substream_data_len[substr] = end - substream_start;
1278  substream_start = end;
1279  }
1280 
1281  parity_bits = ff_mlp_calculate_parity(buf, 4);
1282  parity_bits ^= ff_mlp_calculate_parity(buf + header_size, substr_header_size);
1283 
1284  if ((((parity_bits >> 4) ^ parity_bits) & 0xF) != 0xF) {
1285  av_log(avctx, AV_LOG_ERROR, "Parity check failed.\n");
1286  goto error;
1287  }
1288 
1289  buf += header_size + substr_header_size;
1290 
1291  for (substr = 0; substr <= m->max_decoded_substream; substr++) {
1292  SubStream *s = &m->substream[substr];
1293 
1294  init_get_bits(&gb, buf, substream_data_len[substr] * 8);
1295 
1296  m->matrix_changed = 0;
1297  memset(m->filter_changed, 0, sizeof(m->filter_changed));
1298 
1299  s->blockpos = 0;
1300  do {
1301  if (get_bits1(&gb)) {
1302  if (get_bits1(&gb)) {
1303  /* A restart header should be present. */
1304  if (read_restart_header(m, &gb, buf, substr) < 0)
1305  goto next_substr;
1306  s->restart_seen = 1;
1307  }
1308 
1309  if (!s->restart_seen)
1310  goto next_substr;
1311  if (read_decoding_params(m, &gb, substr) < 0)
1312  goto next_substr;
1313  }
1314 
1315  if (!s->restart_seen)
1316  goto next_substr;
1317 
1318  if (((avctx->ch_layout.nb_channels == 6 &&
1319  ((m->substream_info >> 2) & 0x3) != 0x3) ||
1320  (avctx->ch_layout.nb_channels == 8 &&
1321  ((m->substream_info >> 4) & 0x7) != 0x7 &&
1322  ((m->substream_info >> 4) & 0x7) != 0x6 &&
1323  ((m->substream_info >> 4) & 0x7) != 0x3)) &&
1324  substr > 0 && substr < m->max_decoded_substream &&
1325  (s->min_channel <= m->substream[substr - 1].max_channel)) {
1326  av_log(avctx, AV_LOG_DEBUG,
1327  "Previous substream(%d) channels overlaps current substream(%d) channels, skipping.\n",
1328  substr - 1, substr);
1329  goto next_substr;
1330  }
1331 
1332  if (substr != m->max_decoded_substream &&
1333  ((s->coded_channels & m->substream[m->max_decoded_substream].coded_channels) != 0)) {
1334  av_log(avctx, AV_LOG_DEBUG,
1335  "Current substream(%d) channels overlaps final substream(%d) channels, skipping.\n",
1336  substr, m->max_decoded_substream);
1337  goto next_substr;
1338  }
1339 
1340  if ((ret = read_block_data(m, &gb, substr)) < 0)
1341  return ret;
1342 
1343  if (get_bits_count(&gb) >= substream_data_len[substr] * 8)
1344  goto substream_length_mismatch;
1345 
1346  } while (!get_bits1(&gb));
1347 
1348  skip_bits(&gb, (-get_bits_count(&gb)) & 15);
1349 
1350  if (substream_data_len[substr] * 8 - get_bits_count(&gb) >= 32) {
1351  int shorten_by;
1352 
1353  if (get_bits(&gb, 16) != 0xD234)
1354  return AVERROR_INVALIDDATA;
1355 
1356  shorten_by = get_bits(&gb, 16);
1357  if (m->avctx->codec_id == AV_CODEC_ID_TRUEHD && shorten_by & 0x2000)
1358  s->blockpos -= FFMIN(shorten_by & 0x1FFF, s->blockpos);
1359  else if (m->avctx->codec_id == AV_CODEC_ID_MLP && shorten_by != 0xD234)
1360  return AVERROR_INVALIDDATA;
1361 
1362  av_log(m->avctx, AV_LOG_DEBUG, "End of stream indicated.\n");
1363  s->end_of_stream = 1;
1364  }
1365 
1366  if (substream_parity_present[substr]) {
1367  uint8_t parity, checksum;
1368 
1369  if (substream_data_len[substr] * 8 - get_bits_count(&gb) != 16)
1370  goto substream_length_mismatch;
1371 
1372  parity = ff_mlp_calculate_parity(buf, substream_data_len[substr] - 2);
1373  checksum = ff_mlp_checksum8 (buf, substream_data_len[substr] - 2);
1374 
1375  if ((get_bits(&gb, 8) ^ parity) != 0xa9 )
1376  av_log(m->avctx, AV_LOG_ERROR, "Substream %d parity check failed.\n", substr);
1377  if ( get_bits(&gb, 8) != checksum)
1378  av_log(m->avctx, AV_LOG_ERROR, "Substream %d checksum failed.\n" , substr);
1379  }
1380 
1381  if (substream_data_len[substr] * 8 != get_bits_count(&gb))
1382  goto substream_length_mismatch;
1383 
1384 next_substr:
1385  if (!s->restart_seen)
1387  "No restart header present in substream %d.\n", substr);
1388 
1389  buf += substream_data_len[substr];
1390  }
1391 
1392  if ((ret = output_data(m, m->max_decoded_substream, frame, got_frame_ptr)) < 0)
1393  return ret;
1394 
1395  for (substr = 0; substr <= m->max_decoded_substream; substr++){
1396  SubStream *s = &m->substream[substr];
1397 
1398  if (s->end_of_stream) {
1399  s->lossless_check_data = 0xffffffff;
1400  s->end_of_stream = 0;
1401  m->params_valid = 0;
1402  }
1403  }
1404 
1405  return length;
1406 
1407 substream_length_mismatch:
1408  av_log(m->avctx, AV_LOG_ERROR, "substream %d length mismatch\n", substr);
1409  return AVERROR_INVALIDDATA;
1410 
1411 error:
1412  m->params_valid = 0;
1413  return AVERROR_INVALIDDATA;
1414 }
1415 
1417 {
1418  MLPDecodeContext *m = avctx->priv_data;
1419 
1420  m->params_valid = 0;
1421  for (int substr = 0; substr <= m->max_decoded_substream; substr++){
1422  SubStream *s = &m->substream[substr];
1423 
1424  s->lossless_check_data = 0xffffffff;
1425  s->prev_matrix_encoding = 0;
1426  }
1427 }
1428 
1429 #define OFFSET(x) offsetof(MLPDecodeContext, x)
1430 #define FLAGS (AV_OPT_FLAG_DECODING_PARAM | AV_OPT_FLAG_AUDIO_PARAM)
1431 static const AVOption options[] = {
1432  { "downmix", "Request a specific channel layout from the decoder", OFFSET(downmix_layout),
1433  AV_OPT_TYPE_CHLAYOUT, {.str = NULL}, .flags = FLAGS },
1434  { NULL },
1435 };
1436 
1437 static const AVClass mlp_decoder_class = {
1438  .class_name = "MLP decoder",
1439  .item_name = av_default_item_name,
1440  .option = options,
1441  .version = LIBAVUTIL_VERSION_INT,
1442 };
1443 
1445  .class_name = "TrueHD decoder",
1446  .item_name = av_default_item_name,
1447  .option = options,
1448  .version = LIBAVUTIL_VERSION_INT,
1449 };
1450 
1451 #if CONFIG_MLP_DECODER
1452 const FFCodec ff_mlp_decoder = {
1453  .p.name = "mlp",
1454  CODEC_LONG_NAME("MLP (Meridian Lossless Packing)"),
1455  .p.type = AVMEDIA_TYPE_AUDIO,
1456  .p.id = AV_CODEC_ID_MLP,
1457  .priv_data_size = sizeof(MLPDecodeContext),
1458  .p.priv_class = &mlp_decoder_class,
1459  .init = mlp_decode_init,
1461  .flush = mlp_decode_flush,
1462  .p.capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_CHANNEL_CONF,
1463 };
1464 #endif
1465 #if CONFIG_TRUEHD_DECODER
1466 const FFCodec ff_truehd_decoder = {
1467  .p.name = "truehd",
1468  CODEC_LONG_NAME("TrueHD"),
1469  .p.type = AVMEDIA_TYPE_AUDIO,
1470  .p.id = AV_CODEC_ID_TRUEHD,
1471  .priv_data_size = sizeof(MLPDecodeContext),
1472  .p.priv_class = &truehd_decoder_class,
1473  .init = mlp_decode_init,
1475  .flush = mlp_decode_flush,
1476  .p.capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_CHANNEL_CONF,
1478 };
1479 #endif /* CONFIG_TRUEHD_DECODER */
MLPDecodeContext::params_valid
uint8_t params_valid
Set if a valid major sync block has been read. Otherwise no decoding is possible.
Definition: mlpdec.c:144
error
static void error(const char *err)
Definition: target_bsf_fuzzer.c:32
AVCodecContext::frame_size
int frame_size
Number of samples per channel in an audio frame.
Definition: avcodec.h:1083
AV_LOG_WARNING
#define AV_LOG_WARNING
Something somehow does not look correct.
Definition: log.h:215
noise_table
static const int8_t noise_table[256]
Data table used for TrueHD noise generation function.
Definition: mlpdec.c:1042
AV_CHANNEL_LAYOUT_STEREO_DOWNMIX
#define AV_CHANNEL_LAYOUT_STEREO_DOWNMIX
Definition: channel_layout.h:428
AV_CH_LAYOUT_5POINT0_BACK
#define AV_CH_LAYOUT_5POINT0_BACK
Definition: channel_layout.h:229
show_bits_long
static unsigned int show_bits_long(GetBitContext *s, int n)
Show 0-32 bits.
Definition: get_bits.h:495
ChannelParams::codebook
uint8_t codebook
Which VLC codebook to use to read residuals.
Definition: mlp.h:103
opt.h
xor_32_to_8
static uint8_t xor_32_to_8(uint32_t value)
XOR four bytes into one.
Definition: mlp.h:172
SubStream::matrix_coeff
int32_t matrix_coeff[MAX_MATRICES][MAX_CHANNELS]
Matrix coefficients, stored as 2.14 fixed point.
Definition: mlpdec.c:110
mem_internal.h
SubStream::prev_matrix_encoding
enum AVMatrixEncoding prev_matrix_encoding
Definition: mlpdec.c:81
AV_CHANNEL_LAYOUT_STEREO
#define AV_CHANNEL_LAYOUT_STEREO
Definition: channel_layout.h:393
AVCodecContext::sample_rate
int sample_rate
samples per second
Definition: avcodec.h:1056
av_popcount64
#define av_popcount64
Definition: common.h:157
FLAGS
#define FLAGS
Definition: mlpdec.c:1430
thread.h
SubStream::end_of_stream
uint8_t end_of_stream
Set if end of stream is encountered.
Definition: mlpdec.c:60
AV_CHAN_WIDE_LEFT
@ AV_CHAN_WIDE_LEFT
Definition: channel_layout.h:72
AV_CH_LAYOUT_MONO
#define AV_CH_LAYOUT_MONO
Definition: channel_layout.h:217
get_bits_long
static unsigned int get_bits_long(GetBitContext *s, int n)
Read 0-32 bits.
Definition: get_bits.h:421
SubStream::output_shift
int8_t output_shift[MAX_CHANNELS]
Left shift to apply to decoded PCM values to get final 24-bit output.
Definition: mlpdec.c:124
THD_CH_MODIFIER_SURROUNDEX
@ THD_CH_MODIFIER_SURROUNDEX
Definition: mlp.h:186
get_bits_count
static int get_bits_count(const GetBitContext *s)
Definition: get_bits.h:266
read_decoding_params
static int read_decoding_params(MLPDecodeContext *m, GetBitContext *gbp, unsigned int substr)
Read decoding parameters that change more often than those in the restart header.
Definition: mlpdec.c:890
PARAM_HUFFOFFSET
#define PARAM_HUFFOFFSET
Definition: mlp.h:79
MAX_SAMPLERATE
#define MAX_SAMPLERATE
maximum sample frequency seen in files
Definition: mlp.h:56
mask
int mask
Definition: mediacodecdec_common.c:154
MLPDecodeContext::pack_output
int32_t(* pack_output)(int32_t lossless_check_data, uint16_t blockpos, int32_t(*sample_buffer)[MAX_CHANNELS], void *data, uint8_t *ch_assign, int8_t *output_shift, uint8_t max_matrix_channel, int is32)
Definition: mlpdec.c:176
init_static
static av_cold void init_static(void)
Initialize static data, constant between all invocations of the codec.
Definition: mlpdec.c:227
AVFrame
This structure describes decoded (raw) audio or video data.
Definition: frame.h:389
tmp
static uint8_t tmp[11]
Definition: aes_ctr.c:28
AVPacket::data
uint8_t * data
Definition: packet.h:539
ff_mlp_calculate_parity
uint8_t ff_mlp_calculate_parity(const uint8_t *buf, unsigned int buf_size)
XOR together all the bytes of a buffer.
Definition: mlp.c:124
AVOption
AVOption.
Definition: opt.h:429
mh
#define mh
Definition: vf_colormatrix.c:105
table
static const uint16_t table[]
Definition: prosumer.c:203
data
const char data[16]
Definition: mxf.c:149
filter_channel
static void filter_channel(MLPDecodeContext *m, unsigned int substr, unsigned int channel)
Generate PCM samples using the prediction filters and residual values read from the data stream,...
Definition: mlpdec.c:973
SubStream::restart_seen
uint8_t restart_seen
Set if a valid restart header has been read. Otherwise the substream cannot be decoded.
Definition: mlpdec.c:58
FFCodec
Definition: codec_internal.h:127
output_data
static int output_data(MLPDecodeContext *m, unsigned int substr, AVFrame *frame, int *got_frame_ptr)
Write the audio data into the output buffer.
Definition: mlpdec.c:1108
SubStream::mask
uint64_t mask
The channel layout for this substream.
Definition: mlpdec.c:78
SubStream::min_channel
uint8_t min_channel
The index of the first channel coded in this substream.
Definition: mlpdec.c:68
filter
void(* filter)(uint8_t *src, int stride, int qscale)
Definition: h263dsp.c:29
MLPDecodeContext::major_sync_header_size
int major_sync_header_size
Size of the major sync unit, in bytes.
Definition: mlpdec.c:141
OFFSET
#define OFFSET(x)
Definition: mlpdec.c:1429
SubStream
Definition: mlpdec.c:56
AVChannelLayout::nb_channels
int nb_channels
Number of channels in this layout.
Definition: channel_layout.h:327
AV_CODEC_ID_TRUEHD
@ AV_CODEC_ID_TRUEHD
Definition: codec_id.h:490
ff_mlpdsp_init
av_cold void ff_mlpdsp_init(MLPDSPContext *c)
Definition: mlpdsp.c:128
init_get_bits
static int init_get_bits(GetBitContext *s, const uint8_t *buffer, int bit_size)
Initialize GetBitContext.
Definition: get_bits.h:514
SubStream::max_channel
uint8_t max_channel
The index of the last channel coded in this substream.
Definition: mlpdec.c:70
SubStream::ch_assign
uint8_t ch_assign[MAX_CHANNELS]
For each channel output by the matrix, the output channel to map it to.
Definition: mlpdec.c:76
PARAM_MATRIX
#define PARAM_MATRIX
Definition: mlp.h:74
skip_bits
static void skip_bits(GetBitContext *s, int n)
Definition: get_bits.h:381
MLPDecodeContext::substream_info
uint8_t substream_info
Which substream of substreams carry 2/6/8-channel presentation.
Definition: mlpdec.c:153
get_bits
static unsigned int get_bits(GetBitContext *s, int n)
Read 1-25 bits.
Definition: get_bits.h:335
AV_CHAN_SURROUND_DIRECT_LEFT
@ AV_CHAN_SURROUND_DIRECT_LEFT
Definition: channel_layout.h:74
FFCodec::p
AVCodec p
The public AVCodec.
Definition: codec_internal.h:131
ChannelParams::filter_params
FilterParams filter_params[NUM_FILTERS]
Definition: mlp.h:98
ff_mlp_checksum8
uint8_t ff_mlp_checksum8(const uint8_t *buf, unsigned int buf_size)
MLP uses checksums that seem to be based on the standard CRC algorithm, but are not (in implementatio...
Definition: mlp.c:98
AVCodecContext::ch_layout
AVChannelLayout ch_layout
Audio channel layout.
Definition: avcodec.h:1071
fail
#define fail()
Definition: checkasm.h:200
MAX_MATRICES
#define MAX_MATRICES
Definition: mlp.h:46
ChannelParams::huff_lsbs
uint8_t huff_lsbs
Size of residual suffix not encoded using VLC.
Definition: mlp.h:104
GetBitContext
Definition: get_bits.h:108
MAX_IIR_ORDER
#define MAX_IIR_ORDER
Definition: mlp.h:68
SYNC_TRUEHD
#define SYNC_TRUEHD
Definition: mlp.h:30
AV_PROFILE_TRUEHD_ATMOS
#define AV_PROFILE_TRUEHD_ATMOS
Definition: defs.h:98
calculate_sign_huff
static int32_t calculate_sign_huff(MLPDecodeContext *m, unsigned int substr, unsigned int ch)
Definition: mlpdec.c:241
AV_CH_LAYOUT_STEREO
#define AV_CH_LAYOUT_STEREO
Definition: channel_layout.h:218
AV_CH_LAYOUT_QUAD
#define AV_CH_LAYOUT_QUAD
Definition: channel_layout.h:226
MAX_MATRICES_MLP
#define MAX_MATRICES_MLP
Maximum number of matrices used in decoding; most streams have one matrix per output channel,...
Definition: mlp.h:44
ff_thread_once
static int ff_thread_once(char *control, void(*routine)(void))
Definition: thread.h:205
AV_LOG_ERROR
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:209
MLPDecodeContext::needs_reordering
uint8_t needs_reordering
Stream needs channel reordering to comply with FFmpeg's channel order.
Definition: mlpdec.c:159
FIR
#define FIR
Definition: mlp.h:82
mlp_decoder_class
static const AVClass mlp_decoder_class
Definition: mlpdec.c:1437
FF_ARRAY_ELEMS
#define FF_ARRAY_ELEMS(a)
Definition: sinewin_tablegen.c:29
av_cold
#define av_cold
Definition: attributes.h:90
ff_side_data_update_matrix_encoding
int ff_side_data_update_matrix_encoding(AVFrame *frame, enum AVMatrixEncoding matrix_encoding)
Add or update AV_FRAME_DATA_MATRIXENCODING side data.
Definition: utils.c:124
AV_MATRIX_ENCODING_DOLBY
@ AV_MATRIX_ENCODING_DOLBY
Definition: channel_layout.h:260
MLPDecodeContext::dsp
MLPDSPContext dsp
Definition: mlpdec.c:175
AV_FRAME_FLAG_KEY
#define AV_FRAME_FLAG_KEY
A flag to mark frames that are keyframes.
Definition: frame.h:640
SubStream::channel_params
ChannelParams channel_params[MAX_CHANNELS]
Channel coding parameters for channels in the substream.
Definition: mlpdec.c:84
AV_CH_LOW_FREQUENCY
#define AV_CH_LOW_FREQUENCY
Definition: channel_layout.h:178
FF_CODEC_DECODE_CB
#define FF_CODEC_DECODE_CB(func)
Definition: codec_internal.h:311
intreadwrite.h
s
#define s(width, name)
Definition: cbs_vp9.c:198
PARAM_IIR
#define PARAM_IIR
Definition: mlp.h:78
MLPDecodeContext::matrix_changed
int matrix_changed
Definition: mlpdec.c:168
AVMEDIA_TYPE_AUDIO
@ AVMEDIA_TYPE_AUDIO
Definition: avutil.h:202
av_channel_layout_from_mask
int av_channel_layout_from_mask(AVChannelLayout *channel_layout, uint64_t mask)
Initialize a native channel layout from a bitmask indicating which channels are present.
Definition: channel_layout.c:250
MLPDSPContext::mlp_filter_channel
void(* mlp_filter_channel)(int32_t *state, const int32_t *coeff, int firorder, int iirorder, unsigned int filter_shift, int32_t mask, int blocksize, int32_t *sample_buffer)
Definition: mlpdsp.h:50
AV_CHAN_SIDE_RIGHT
@ AV_CHAN_SIDE_RIGHT
Definition: channel_layout.h:60
av_assert0
#define av_assert0(cond)
assert() equivalent, that is always enabled.
Definition: avassert.h:40
vlc_init
#define vlc_init(vlc, nb_bits, nb_codes, bits, bits_wrap, bits_size, codes, codes_wrap, codes_size, flags)
Definition: vlc.h:62
AVCodecContext::bits_per_raw_sample
int bits_per_raw_sample
Bits per sample/pixel of internal libavcodec pixel/sample format.
Definition: avcodec.h:1585
get_sbits
static int get_sbits(GetBitContext *s, int n)
Definition: get_bits.h:320
AV_LOG_DEBUG
#define AV_LOG_DEBUG
Stuff which is only useful for libav* developers.
Definition: log.h:230
decode.h
get_bits.h
THD_CH_MODIFIER_LTRT
@ THD_CH_MODIFIER_LTRT
Definition: mlp.h:182
CODEC_LONG_NAME
#define CODEC_LONG_NAME(str)
Definition: codec_internal.h:296
MLPDSPContext::mlp_select_pack_output
int32_t(*(* mlp_select_pack_output)(uint8_t *ch_assign, int8_t *output_shift, uint8_t max_matrix_channel, int is32))(int32_t
Definition: mlpdsp.h:65
AVCodecContext::codec_id
enum AVCodecID codec_id
Definition: avcodec.h:461
if
if(ret)
Definition: filter_design.txt:179
AVMatrixEncoding
AVMatrixEncoding
Definition: channel_layout.h:258
AV_MATRIX_ENCODING_DOLBYHEADPHONE
@ AV_MATRIX_ENCODING_DOLBYHEADPHONE
Definition: channel_layout.h:265
read_access_unit
static int read_access_unit(AVCodecContext *avctx, AVFrame *frame, int *got_frame_ptr, AVPacket *avpkt)
Read an access unit from the stream.
Definition: mlpdec.c:1183
LIBAVUTIL_VERSION_INT
#define LIBAVUTIL_VERSION_INT
Definition: version.h:85
AV_ONCE_INIT
#define AV_ONCE_INIT
Definition: thread.h:203
AVClass
Describe the class of an AVClass context structure.
Definition: log.h:75
result
and forward the result(frame or status change) to the corresponding input. If nothing is possible
NULL
#define NULL
Definition: coverity.c:32
AVERROR_PATCHWELCOME
#define AVERROR_PATCHWELCOME
Not yet implemented in FFmpeg, patches welcome.
Definition: error.h:64
SubStream::lossless_check_data
int32_t lossless_check_data
Running XOR of all output samples.
Definition: mlpdec.c:127
MLPDecodeContext::bypassed_lsbs
int8_t bypassed_lsbs[MAX_BLOCKSIZE][MAX_CHANNELS]
Definition: mlpdec.c:172
SubStream::quant_step_size
uint8_t quant_step_size[MAX_CHANNELS]
Left shift to apply to Huffman-decoded residuals.
Definition: mlpdec.c:116
av_default_item_name
const char * av_default_item_name(void *ptr)
Return the context name.
Definition: log.c:237
get_bits1
static unsigned int get_bits1(GetBitContext *s)
Definition: get_bits.h:388
profiles.h
quant_step_size
static const float quant_step_size[]
Definition: hca_data.h:125
MLPDecodeContext::filter_changed
int filter_changed[MAX_CHANNELS][NUM_FILTERS]
Definition: mlpdec.c:169
options
Definition: swscale.c:42
AV_CHAN_TOP_CENTER
@ AV_CHAN_TOP_CENTER
Definition: channel_layout.h:61
AV_OPT_TYPE_CHLAYOUT
@ AV_OPT_TYPE_CHLAYOUT
Underlying C type is AVChannelLayout.
Definition: opt.h:331
seed
static unsigned int seed
Definition: videogen.c:78
get_vlc2
static av_always_inline int get_vlc2(GetBitContext *s, const VLCElem *table, int bits, int max_depth)
Parse a vlc code.
Definition: get_bits.h:652
AVOnce
#define AVOnce
Definition: thread.h:202
MLPDecodeContext::access_unit_size
int access_unit_size
number of PCM samples contained in each frame
Definition: mlpdec.c:162
index
int index
Definition: gxfenc.c:90
AV_CHAN_FRONT_RIGHT_OF_CENTER
@ AV_CHAN_FRONT_RIGHT_OF_CENTER
Definition: channel_layout.h:57
SubStream::max_matrix_channel
uint8_t max_matrix_channel
The number of channels input into the rematrix stage.
Definition: mlpdec.c:74
MLPDecodeContext::extended_substream_info
uint8_t extended_substream_info
Which substream of substreams carry 16-channel presentation.
Definition: mlpdec.c:150
THD_CH_MODIFIER_LBINRBIN
@ THD_CH_MODIFIER_LBINRBIN
Definition: mlp.h:183
AV_CHAN_FRONT_RIGHT
@ AV_CHAN_FRONT_RIGHT
Definition: channel_layout.h:51
AV_CHAN_FRONT_CENTER
@ AV_CHAN_FRONT_CENTER
Definition: channel_layout.h:52
AV_CODEC_CAP_CHANNEL_CONF
#define AV_CODEC_CAP_CHANNEL_CONF
Codec should fill in channel configuration and samplerate instead of container.
Definition: codec.h:106
FilterParams
filter data
Definition: mlp.h:86
VLC::table_allocated
int table_allocated
Definition: vlc.h:39
MLPDecodeContext::sample_buffer
int32_t sample_buffer[MAX_BLOCKSIZE][MAX_CHANNELS]
Definition: mlpdec.c:173
huff_vlc
static VLC huff_vlc[3]
Definition: mlpdec.c:223
ff_mlp_restart_checksum
uint8_t ff_mlp_restart_checksum(const uint8_t *buf, unsigned int bit_size)
Calculate an 8-bit checksum over a restart header – a non-multiple-of-8 number of bits,...
Definition: mlp.c:105
options
static const AVOption options[]
Definition: mlpdec.c:1431
thd_channel_order
static enum AVChannel thd_channel_order[]
Definition: mlpdec.c:186
ff_get_buffer
int ff_get_buffer(AVCodecContext *avctx, AVFrame *frame, int flags)
Get a buffer for a frame.
Definition: decode.c:1697
AV_CODEC_CAP_DR1
#define AV_CODEC_CAP_DR1
Codec uses get_buffer() or get_encode_buffer() for allocating buffers and supports custom allocators.
Definition: codec.h:52
AVPacket::size
int size
Definition: packet.h:540
MAX_MATRIX_CHANNEL_MLP
#define MAX_MATRIX_CHANNEL_MLP
Last possible matrix channel for each codec.
Definition: mlp.h:33
NULL_IF_CONFIG_SMALL
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
Definition: internal.h:94
SubStream::noise_type
uint16_t noise_type
restart header data
Definition: mlpdec.c:65
AVChannelLayout
An AVChannelLayout holds information about the channel layout of audio data.
Definition: channel_layout.h:317
codec_internal.h
DECLARE_ALIGNED
#define DECLARE_ALIGNED(n, t, v)
Definition: mem_internal.h:102
MLPDecodeContext::num_substreams
uint8_t num_substreams
Number of substreams contained within this stream.
Definition: mlpdec.c:147
MLPDecodeContext::avctx
AVCodecContext * avctx
Definition: mlpdec.c:133
MLPDecodeContext::substream
SubStream substream[MAX_SUBSTREAMS]
Definition: mlpdec.c:166
MSB_MASK
#define MSB_MASK(bits)
Definition: mlpdec.c:968
AV_MATRIX_ENCODING_NONE
@ AV_MATRIX_ENCODING_NONE
Definition: channel_layout.h:259
AVCodecContext::sample_fmt
enum AVSampleFormat sample_fmt
audio sample format
Definition: avcodec.h:1063
mlpdsp.h
AV_CHAN_LOW_FREQUENCY
@ AV_CHAN_LOW_FREQUENCY
Definition: channel_layout.h:53
MAX_SUBSTREAMS
#define MAX_SUBSTREAMS
Maximum number of substreams that can be decoded.
Definition: mlp.h:51
MLPDecodeContext::is_major_sync_unit
int is_major_sync_unit
Current access unit being read has a major sync.
Definition: mlpdec.c:138
MAX_MATRIX_CHANNEL_TRUEHD
#define MAX_MATRIX_CHANNEL_TRUEHD
Definition: mlp.h:34
mlp_decode_init
static av_cold int mlp_decode_init(AVCodecContext *avctx)
Definition: mlpdec.c:298
VLCElem
Definition: vlc.h:32
AV_CHAN_BACK_RIGHT
@ AV_CHAN_BACK_RIGHT
Definition: channel_layout.h:55
parity
mcdeint parity
Definition: vf_mcdeint.c:281
MAX_MATRICES_TRUEHD
#define MAX_MATRICES_TRUEHD
Definition: mlp.h:45
AV_CHAN_SIDE_LEFT
@ AV_CHAN_SIDE_LEFT
Definition: channel_layout.h:59
NUM_FILTERS
#define NUM_FILTERS
number of allowed filters
Definition: mlp.h:64
FilterParams::order
uint8_t order
number of taps in filter
Definition: mlp.h:87
MLPDecodeContext
Definition: mlpdec.c:131
ff_mlp_read_major_sync
int ff_mlp_read_major_sync(void *log, MLPHeaderInfo *mh, GetBitContext *gb)
Read a major sync info header - contains high level information about the stream - sample rate,...
Definition: mlp_parse.c:86
AV_CH_LAYOUT_5POINT1_BACK
#define AV_CH_LAYOUT_5POINT1_BACK
Definition: channel_layout.h:230
AV_CHAN_TOP_FRONT_RIGHT
@ AV_CHAN_TOP_FRONT_RIGHT
Definition: channel_layout.h:64
ChannelParams::huff_offset
int16_t huff_offset
Offset to apply to residual values.
Definition: mlp.h:101
skip_bits1
static void skip_bits1(GetBitContext *s)
Definition: get_bits.h:413
MLPDecodeContext::downmix_layout
AVChannelLayout downmix_layout
Definition: mlpdec.c:135
SubStream::noisegen_seed
uint32_t noisegen_seed
The current seed value for the pseudorandom noise generator(s).
Definition: mlpdec.c:89
AV_CHAN_FRONT_LEFT_OF_CENTER
@ AV_CHAN_FRONT_LEFT_OF_CENTER
Definition: channel_layout.h:56
MLPDecodeContext::max_decoded_substream
uint8_t max_decoded_substream
Index of the last substream to decode - further substreams are skipped.
Definition: mlpdec.c:156
SubStream::lsb_bypass
uint8_t lsb_bypass[MAX_MATRICES]
Whether the LSBs of the matrix output are encoded in the bitstream.
Definition: mlpdec.c:108
ff_mlp_decoder
const FFCodec ff_mlp_decoder
SubStream::matrix_out_ch
uint8_t matrix_out_ch[MAX_MATRICES]
matrix output channel
Definition: mlpdec.c:105
MAX_CHANNELS
#define MAX_CHANNELS
Definition: aac.h:33
MAX_FIR_ORDER
#define MAX_FIR_ORDER
The maximum number of taps in IIR and FIR filters.
Definition: mlp.h:67
av_channel_layout_compare
int av_channel_layout_compare(const AVChannelLayout *chl, const AVChannelLayout *chl1)
Check whether two channel layouts are semantically the same, i.e.
Definition: channel_layout.c:807
MLPDSPContext::mlp_rematrix_channel
void(* mlp_rematrix_channel)(int32_t *samples, const int32_t *coeffs, const uint8_t *bypassed_lsbs, const int8_t *noise_buffer, int index, unsigned int dest_ch, uint16_t blockpos, unsigned int maxchan, int matrix_noise_shift, int access_unit_size_pow2, int32_t mask)
Definition: mlpdsp.h:54
layout
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel layout
Definition: filter_design.txt:18
AVChannel
AVChannel
Definition: channel_layout.h:47
read_restart_header
static int read_restart_header(MLPDecodeContext *m, GetBitContext *gbp, const uint8_t *buf, unsigned int substr)
Read a restart header from a block in a substream.
Definition: mlpdec.c:526
mlp_parse.h
SubStream::coded_channels
uint64_t coded_channels
The coded channels mask in this substream.
Definition: mlpdec.c:72
AV_CHAN_SURROUND_DIRECT_RIGHT
@ AV_CHAN_SURROUND_DIRECT_RIGHT
Definition: channel_layout.h:75
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:256
SubStream::blockpos
uint16_t blockpos
Number of PCM samples decoded so far in this frame.
Definition: mlpdec.c:121
SubStream::noise_shift
uint8_t noise_shift
The left shift applied to random noise in 0x31ea substreams.
Definition: mlpdec.c:87
SubStream::blocksize
uint16_t blocksize
number of PCM samples in current audio block
Definition: mlpdec.c:119
internal.h
PARAM_PRESENCE
#define PARAM_PRESENCE
Definition: mlp.h:80
FFMIN
#define FFMIN(a, b)
Definition: macros.h:49
MLPHeaderInfo
Definition: mlp_parse.h:30
AV_SAMPLE_FMT_S16
@ AV_SAMPLE_FMT_S16
signed 16 bits
Definition: samplefmt.h:58
read_channel_params
static int read_channel_params(MLPDecodeContext *m, unsigned int substr, GetBitContext *gbp, unsigned int ch)
Read channel parameters.
Definition: mlpdec.c:833
AVCodec::name
const char * name
Name of the codec implementation.
Definition: codec.h:194
ChannelParams
sample data coding information
Definition: mlp.h:97
AV_CH_SIDE_RIGHT
#define AV_CH_SIDE_RIGHT
Definition: channel_layout.h:185
MAX_BLOCKSIZE
#define MAX_BLOCKSIZE
Definition: diracdec.c:56
fill_noise_buffer
static void fill_noise_buffer(MLPDecodeContext *m, unsigned int substr)
Generate a block of noise, used when restart sync word == 0x31eb.
Definition: mlpdec.c:1091
ff_mlp_init_crc
av_cold void ff_mlp_init_crc(void)
Definition: mlp.c:83
ChannelParams::sign_huff_offset
int32_t sign_huff_offset
sign/rounding-corrected version of huff_offset
Definition: mlp.h:102
mlp_channel_layout_subset
static int mlp_channel_layout_subset(AVChannelLayout *layout, uint64_t mask)
Definition: mlpdec.c:202
MLPDecodeContext::access_unit_size_pow2
int access_unit_size_pow2
next power of two above the number of samples in each frame
Definition: mlpdec.c:164
thd_channel_layout_extract_channel
static enum AVChannel thd_channel_layout_extract_channel(uint64_t channel_layout, int index)
Definition: mlpdec.c:209
avcodec.h
ret
ret
Definition: filter_design.txt:187
FFSWAP
#define FFSWAP(type, a, b)
Definition: macros.h:52
AVClass::class_name
const char * class_name
The name of the class; usually it is the same name as the context structure type to which the AVClass...
Definition: log.h:80
frame
these buffered frames must be flushed immediately if a new input produces new the filter must not call request_frame to get more It must just process the frame or queue it The task of requesting more frames is left to the filter s request_frame method or the application If a filter has several the filter must be ready for frames arriving randomly on any input any filter with several inputs will most likely require some kind of queuing mechanism It is perfectly acceptable to have a limited queue and to drop frames when the inputs are too unbalanced request_frame For filters that do not use the this method is called when a frame is wanted on an output For a it should directly call filter_frame on the corresponding output For a if there are queued frames already one of these frames should be pushed If the filter should request a frame on one of its repeatedly until at least one frame has been pushed Return or at least make progress towards producing a frame
Definition: filter_design.txt:264
mlp_decode_flush
static void mlp_decode_flush(AVCodecContext *avctx)
Definition: mlpdec.c:1416
ff_truehd_decoder
const FFCodec ff_truehd_decoder
SubStream::num_primitive_matrices
uint8_t num_primitive_matrices
matrix data
Definition: mlpdec.c:102
av_channel_layout_check
int av_channel_layout_check(const AVChannelLayout *channel_layout)
Check whether a channel layout is valid, i.e.
Definition: channel_layout.c:781
pos
unsigned int pos
Definition: spdifenc.c:414
FilterParams::state
int32_t state[MAX_FIR_ORDER]
Definition: mlp.h:90
SYNC_MLP
#define SYNC_MLP
Definition: mlp.h:29
AV_CHAN_BACK_CENTER
@ AV_CHAN_BACK_CENTER
Definition: channel_layout.h:58
SubStream::data_check_present
uint8_t data_check_present
Set if the substream contains extra info to check the size of VLC blocks.
Definition: mlpdec.c:92
AV_CHAN_NONE
@ AV_CHAN_NONE
Invalid channel index.
Definition: channel_layout.h:49
SubStream::matrix_noise_shift
uint8_t matrix_noise_shift[MAX_MATRICES]
Left shift to apply to noise values in 0x31eb substreams.
Definition: mlpdec.c:112
AVCodecContext
main external API structure.
Definition: avcodec.h:451
VLC_BITS
#define VLC_BITS
number of bits used for VLC lookup - longest Huffman code is 9
Definition: mlpdec.c:52
channel_layout.h
AV_CHAN_LOW_FREQUENCY_2
@ AV_CHAN_LOW_FREQUENCY_2
Definition: channel_layout.h:76
VLC_STATIC_SIZE
#define VLC_STATIC_SIZE
Definition: mlpdec.c:53
av_channel_layout_subset
uint64_t av_channel_layout_subset(const AVChannelLayout *channel_layout, uint64_t mask)
Find out what channels from a given set are present in a channel layout, without regard for their pos...
Definition: channel_layout.c:863
AV_MATRIX_ENCODING_DOLBYEX
@ AV_MATRIX_ENCODING_DOLBYEX
Definition: channel_layout.h:264
av_channel_layout_index_from_channel
int av_channel_layout_index_from_channel(const AVChannelLayout *channel_layout, enum AVChannel channel)
Get the index of a given channel in a channel layout.
Definition: channel_layout.c:711
VLC
Definition: vlc.h:36
AVCodecContext::profile
int profile
profile
Definition: avcodec.h:1650
av_channel_layout_uninit
void av_channel_layout_uninit(AVChannelLayout *channel_layout)
Free any allocated data in the channel layout and reset the channel count to 0.
Definition: channel_layout.c:440
PARAM_QUANTSTEP
#define PARAM_QUANTSTEP
Definition: mlp.h:76
AV_CHAN_BACK_LEFT
@ AV_CHAN_BACK_LEFT
Definition: channel_layout.h:54
VLC::table
VLCElem * table
Definition: vlc.h:38
IIR
#define IIR
Definition: mlp.h:83
PARAM_BLOCKSIZE
#define PARAM_BLOCKSIZE
Definition: mlp.h:73
AV_CHAN_TOP_FRONT_CENTER
@ AV_CHAN_TOP_FRONT_CENTER
Definition: channel_layout.h:63
read_filter_params
static int read_filter_params(MLPDecodeContext *m, GetBitContext *gbp, unsigned int substr, unsigned int channel, unsigned int filter)
Read parameters for one of the prediction filters.
Definition: mlpdec.c:690
AV_CHAN_WIDE_RIGHT
@ AV_CHAN_WIDE_RIGHT
Definition: channel_layout.h:73
PARAM_FIR
#define PARAM_FIR
Definition: mlp.h:77
read_huff_channels
static int read_huff_channels(MLPDecodeContext *m, GetBitContext *gbp, unsigned int substr, unsigned int pos)
Read a sample, consisting of either, both or neither of entropy-coded MSBs and plain LSBs.
Definition: mlpdec.c:262
avpriv_request_sample
#define avpriv_request_sample(...)
Definition: tableprint_vlc.h:36
read_major_sync
static int read_major_sync(MLPDecodeContext *m, GetBitContext *gb)
Read a major sync info header - contains high level information about the stream - sample rate,...
Definition: mlpdec.c:335
mlp.h
ff_mlp_huffman_tables
const uint8_t ff_mlp_huffman_tables[3][18][2]
Tables defining the Huffman codes.
Definition: mlp.c:30
SubStream::param_presence_flags
uint8_t param_presence_flags
Bitmask of which parameter sets are conveyed in a decoding parameter block.
Definition: mlpdec.c:95
AVPacket
This structure stores compressed data.
Definition: packet.h:516
AVCodecContext::priv_data
void * priv_data
Definition: avcodec.h:478
PARAM_OUTSHIFT
#define PARAM_OUTSHIFT
Definition: mlp.h:75
AV_CHAN_TOP_FRONT_LEFT
@ AV_CHAN_TOP_FRONT_LEFT
Definition: channel_layout.h:62
int32_t
int32_t
Definition: audioconvert.c:56
truehd_decoder_class
static const AVClass truehd_decoder_class
Definition: mlpdec.c:1444
av_log
#define av_log(a,...)
Definition: tableprint_vlc.h:27
read_matrix_params
static int read_matrix_params(MLPDecodeContext *m, unsigned int substr, GetBitContext *gbp)
Read parameters for primitive matrices.
Definition: mlpdec.c:765
AVERROR_INVALIDDATA
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:61
VLC_INIT_USE_STATIC
#define VLC_INIT_USE_STATIC
Definition: vlc.h:182
AV_CHANNEL_LAYOUT_5POINT0
#define AV_CHANNEL_LAYOUT_5POINT0
Definition: channel_layout.h:402
AV_CHAN_FRONT_LEFT
@ AV_CHAN_FRONT_LEFT
Definition: channel_layout.h:50
MLPDSPContext
Definition: mlpdsp.h:49
ff_truehd_profiles
const AVProfile ff_truehd_profiles[]
Definition: profiles.c:57
AV_CHANNEL_LAYOUT_5POINT1
#define AV_CHANNEL_LAYOUT_5POINT1
Definition: channel_layout.h:403
FilterParams::shift
uint8_t shift
Right shift to apply to output of filter.
Definition: mlp.h:88
AV_SAMPLE_FMT_S32
@ AV_SAMPLE_FMT_S32
signed 32 bits
Definition: samplefmt.h:59
generate_2_noise_channels
static void generate_2_noise_channels(MLPDecodeContext *m, unsigned int substr)
Noise generation functions.
Definition: mlpdec.c:1071
SubStream::matrix_encoding
enum AVMatrixEncoding matrix_encoding
The matrix encoding mode for this substream.
Definition: mlpdec.c:80
channel
channel
Definition: ebur128.h:39
MAX_BLOCKSIZE_POW2
#define MAX_BLOCKSIZE_POW2
next power of two greater than MAX_BLOCKSIZE
Definition: mlp.h:61
MLPDecodeContext::noise_buffer
int8_t noise_buffer[MAX_BLOCKSIZE_POW2]
Definition: mlpdec.c:171
codebook
static const unsigned codebook[256][2]
Definition: cfhdenc.c:41
AV_CODEC_ID_MLP
@ AV_CODEC_ID_MLP
Definition: codec_id.h:475
AV_CH_SIDE_LEFT
#define AV_CH_SIDE_LEFT
Definition: channel_layout.h:184
AV_RB16
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_WL32 unsigned int_TMPL AV_WL24 unsigned int_TMPL AV_WL16 uint64_t_TMPL AV_WB64 unsigned int_TMPL AV_WB32 unsigned int_TMPL AV_WB24 unsigned int_TMPL AV_RB16
Definition: bytestream.h:98
read_block_data
static int read_block_data(MLPDecodeContext *m, GetBitContext *gbp, unsigned int substr)
Read a block of PCM residual data (or actual if no filtering active).
Definition: mlpdec.c:1000