Subversion Repositories Kolibri OS

Rev

Blame | Last modification | View Log | RSS feed

  1. /*
  2.  * DCA compatible decoder
  3.  * Copyright (C) 2004 Gildas Bazin
  4.  * Copyright (C) 2004 Benjamin Zores
  5.  * Copyright (C) 2006 Benjamin Larsson
  6.  * Copyright (C) 2007 Konstantin Shishkov
  7.  *
  8.  * This file is part of FFmpeg.
  9.  *
  10.  * FFmpeg is free software; you can redistribute it and/or
  11.  * modify it under the terms of the GNU Lesser General Public
  12.  * License as published by the Free Software Foundation; either
  13.  * version 2.1 of the License, or (at your option) any later version.
  14.  *
  15.  * FFmpeg is distributed in the hope that it will be useful,
  16.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  18.  * Lesser General Public License for more details.
  19.  *
  20.  * You should have received a copy of the GNU Lesser General Public
  21.  * License along with FFmpeg; if not, write to the Free Software
  22.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  23.  */
  24.  
  25. #ifndef AVCODEC_DCA_H
  26. #define AVCODEC_DCA_H
  27.  
  28. #include <stdint.h>
  29.  
  30. #include "libavutil/float_dsp.h"
  31. #include "libavutil/internal.h"
  32.  
  33. #include "avcodec.h"
  34. #include "dcadsp.h"
  35. #include "fmtconvert.h"
  36. #include "get_bits.h"
  37.  
  38. #define DCA_PRIM_CHANNELS_MAX  (7)
  39. #define DCA_ABITS_MAX         (32)      /* Should be 28 */
  40. #define DCA_SUBSUBFRAMES_MAX   (4)
  41. #define DCA_SUBFRAMES_MAX     (16)
  42. #define DCA_BLOCKS_MAX        (16)
  43. #define DCA_LFE_MAX            (3)
  44. #define DCA_CHSETS_MAX         (4)
  45. #define DCA_CHSET_CHANS_MAX    (8)
  46.  
  47. #define DCA_PRIM_CHANNELS_MAX  (7)
  48. #define DCA_ABITS_MAX         (32)      /* Should be 28 */
  49. #define DCA_SUBSUBFRAMES_MAX   (4)
  50. #define DCA_SUBFRAMES_MAX     (16)
  51. #define DCA_BLOCKS_MAX        (16)
  52. #define DCA_LFE_MAX            (3)
  53. #define DCA_XLL_FBANDS_MAX     (4)
  54. #define DCA_XLL_SEGMENTS_MAX  (16)
  55. #define DCA_XLL_CHSETS_MAX    (16)
  56. #define DCA_XLL_CHANNELS_MAX  (16)
  57. #define DCA_XLL_AORDER_MAX    (15)
  58.  
  59. /* Arbitrary limit; not sure what the maximum really is, but much larger. */
  60. #define DCA_XLL_DMIX_NCOEFFS_MAX (18)
  61.  
  62. #define DCA_MAX_FRAME_SIZE       16384
  63. #define DCA_MAX_EXSS_HEADER_SIZE  4096
  64.  
  65. #define DCA_BUFFER_PADDING_SIZE   1024
  66.  
  67. enum DCAExtensionMask {
  68.     DCA_EXT_CORE       = 0x001, ///< core in core substream
  69.     DCA_EXT_XXCH       = 0x002, ///< XXCh channels extension in core substream
  70.     DCA_EXT_X96        = 0x004, ///< 96/24 extension in core substream
  71.     DCA_EXT_XCH        = 0x008, ///< XCh channel extension in core substream
  72.     DCA_EXT_EXSS_CORE  = 0x010, ///< core in ExSS (extension substream)
  73.     DCA_EXT_EXSS_XBR   = 0x020, ///< extended bitrate extension in ExSS
  74.     DCA_EXT_EXSS_XXCH  = 0x040, ///< XXCh channels extension in ExSS
  75.     DCA_EXT_EXSS_X96   = 0x080, ///< 96/24 extension in ExSS
  76.     DCA_EXT_EXSS_LBR   = 0x100, ///< low bitrate component in ExSS
  77.     DCA_EXT_EXSS_XLL   = 0x200, ///< lossless extension in ExSS
  78. };
  79.  
  80. typedef struct XllChSetSubHeader {
  81.     int channels;               ///< number of channels in channel set, at most 16
  82.     int residual_encode;        ///< residual channel encoding
  83.     int bit_resolution;         ///< input sample bit-width
  84.     int bit_width;              ///< original input sample bit-width
  85.     int sampling_frequency;     ///< sampling frequency
  86.     int samp_freq_interp;       ///< sampling frequency interpolation multiplier
  87.     int replacement_set;        ///< replacement channel set group
  88.     int active_replace_set;     ///< current channel set is active channel set
  89.     int primary_ch_set;
  90.     int downmix_coeff_code_embedded;
  91.     int downmix_embedded;
  92.     int downmix_type;
  93.     int hier_chset;             ///< hierarchical channel set
  94.     int downmix_ncoeffs;
  95.     int downmix_coeffs[DCA_XLL_DMIX_NCOEFFS_MAX];
  96.     int ch_mask_enabled;
  97.     int ch_mask;
  98.     int mapping_coeffs_present;
  99.     int num_freq_bands;
  100.  
  101.     /* m_nOrigChanOrder */
  102.     uint8_t orig_chan_order[DCA_XLL_FBANDS_MAX][DCA_XLL_CHANNELS_MAX];
  103.     uint8_t orig_chan_order_inv[DCA_XLL_FBANDS_MAX][DCA_XLL_CHANNELS_MAX];
  104.     /* Coefficients for channel pairs (at most 8), m_anPWChPairsCoeffs */
  105.     int8_t pw_ch_pairs_coeffs[DCA_XLL_FBANDS_MAX][DCA_XLL_CHANNELS_MAX/2];
  106.     /* m_nCurrHighestLPCOrder */
  107.     uint8_t adapt_order_max[DCA_XLL_FBANDS_MAX];
  108.     /* m_pnAdaptPredOrder */
  109.     uint8_t adapt_order[DCA_XLL_FBANDS_MAX][DCA_XLL_CHANNELS_MAX];
  110.     /* m_pnFixedPredOrder */
  111.     uint8_t fixed_order[DCA_XLL_FBANDS_MAX][DCA_XLL_CHANNELS_MAX];
  112.     /* m_pnLPCReflCoeffsQInd, unsigned version */
  113.     uint8_t lpc_refl_coeffs_q_ind[DCA_XLL_FBANDS_MAX]
  114.                                  [DCA_XLL_CHANNELS_MAX][DCA_XLL_AORDER_MAX];
  115.  
  116.     int lsb_fsize[DCA_XLL_FBANDS_MAX];
  117.     int8_t scalable_lsbs[DCA_XLL_FBANDS_MAX][DCA_XLL_CHANNELS_MAX];
  118.     int8_t bit_width_adj_per_ch[DCA_XLL_FBANDS_MAX][DCA_XLL_CHANNELS_MAX];
  119. } XllChSetSubHeader;
  120.  
  121. typedef struct XllNavi {
  122.     GetBitContext gb;  // Context for parsing the data segments
  123.     unsigned band_size[DCA_XLL_FBANDS_MAX];
  124.     unsigned segment_size[DCA_XLL_FBANDS_MAX][DCA_XLL_SEGMENTS_MAX];
  125.     unsigned chset_size[DCA_XLL_FBANDS_MAX][DCA_XLL_SEGMENTS_MAX][DCA_XLL_CHSETS_MAX];
  126. } XllNavi;
  127.  
  128. typedef struct QMF64_table {
  129.     float dct4_coeff[32][32];
  130.     float dct2_coeff[32][32];
  131.     float rcos[32];
  132.     float rsin[32];
  133. } QMF64_table;
  134.  
  135. typedef struct DCAContext {
  136.     const AVClass *class;       ///< class for AVOptions
  137.     AVCodecContext *avctx;
  138.     /* Frame header */
  139.     int frame_type;             ///< type of the current frame
  140.     int samples_deficit;        ///< deficit sample count
  141.     int crc_present;            ///< crc is present in the bitstream
  142.     int sample_blocks;          ///< number of PCM sample blocks
  143.     int frame_size;             ///< primary frame byte size
  144.     int amode;                  ///< audio channels arrangement
  145.     int sample_rate;            ///< audio sampling rate
  146.     int bit_rate;               ///< transmission bit rate
  147.     int bit_rate_index;         ///< transmission bit rate index
  148.  
  149.     int dynrange;               ///< embedded dynamic range flag
  150.     int timestamp;              ///< embedded time stamp flag
  151.     int aux_data;               ///< auxiliary data flag
  152.     int hdcd;                   ///< source material is mastered in HDCD
  153.     int ext_descr;              ///< extension audio descriptor flag
  154.     int ext_coding;             ///< extended coding flag
  155.     int aspf;                   ///< audio sync word insertion flag
  156.     int lfe;                    ///< low frequency effects flag
  157.     int predictor_history;      ///< predictor history flag
  158.     int header_crc;             ///< header crc check bytes
  159.     int multirate_inter;        ///< multirate interpolator switch
  160.     int version;                ///< encoder software revision
  161.     int copy_history;           ///< copy history
  162.     int source_pcm_res;         ///< source pcm resolution
  163.     int front_sum;              ///< front sum/difference flag
  164.     int surround_sum;           ///< surround sum/difference flag
  165.     int dialog_norm;            ///< dialog normalisation parameter
  166.  
  167.     /* Primary audio coding header */
  168.     int subframes;              ///< number of subframes
  169.     int total_channels;         ///< number of channels including extensions
  170.     int prim_channels;          ///< number of primary audio channels
  171.     int subband_activity[DCA_PRIM_CHANNELS_MAX];    ///< subband activity count
  172.     int vq_start_subband[DCA_PRIM_CHANNELS_MAX];    ///< high frequency vq start subband
  173.     int joint_intensity[DCA_PRIM_CHANNELS_MAX];     ///< joint intensity coding index
  174.     int transient_huffman[DCA_PRIM_CHANNELS_MAX];   ///< transient mode code book
  175.     int scalefactor_huffman[DCA_PRIM_CHANNELS_MAX]; ///< scale factor code book
  176.     int bitalloc_huffman[DCA_PRIM_CHANNELS_MAX];    ///< bit allocation quantizer select
  177.     int quant_index_huffman[DCA_PRIM_CHANNELS_MAX][DCA_ABITS_MAX]; ///< quantization index codebook select
  178.     float scalefactor_adj[DCA_PRIM_CHANNELS_MAX][DCA_ABITS_MAX];   ///< scale factor adjustment
  179.  
  180.     /* Primary audio coding side information */
  181.     int subsubframes[DCA_SUBFRAMES_MAX];                         ///< number of subsubframes
  182.     int partial_samples[DCA_SUBFRAMES_MAX];                      ///< partial subsubframe samples count
  183.     int prediction_mode[DCA_PRIM_CHANNELS_MAX][DCA_SUBBANDS];    ///< prediction mode (ADPCM used or not)
  184.     int prediction_vq[DCA_PRIM_CHANNELS_MAX][DCA_SUBBANDS];      ///< prediction VQ coefs
  185.     int bitalloc[DCA_PRIM_CHANNELS_MAX][DCA_SUBBANDS];           ///< bit allocation index
  186.     int transition_mode[DCA_PRIM_CHANNELS_MAX][DCA_SUBBANDS];    ///< transition mode (transients)
  187.     int32_t scale_factor[DCA_PRIM_CHANNELS_MAX][DCA_SUBBANDS][2];///< scale factors (2 if transient)
  188.     int joint_huff[DCA_PRIM_CHANNELS_MAX];                       ///< joint subband scale factors codebook
  189.     int joint_scale_factor[DCA_PRIM_CHANNELS_MAX][DCA_SUBBANDS]; ///< joint subband scale factors
  190.     float downmix_coef[DCA_PRIM_CHANNELS_MAX + 1][2];            ///< stereo downmix coefficients
  191.     int dynrange_coef;                                           ///< dynamic range coefficient
  192.  
  193.     /* Core substream's embedded downmix coefficients (cf. ETSI TS 102 114 V1.4.1)
  194.      * Input:  primary audio channels (incl. LFE if present)
  195.      * Output: downmix audio channels (up to 4, no LFE) */
  196.     uint8_t  core_downmix;                                       ///< embedded downmix coefficients available
  197.     uint8_t  core_downmix_amode;                                 ///< audio channel arrangement of embedded downmix
  198.     uint16_t core_downmix_codes[DCA_PRIM_CHANNELS_MAX + 1][4];   ///< embedded downmix coefficients (9-bit codes)
  199.  
  200.     int32_t  high_freq_vq[DCA_PRIM_CHANNELS_MAX][DCA_SUBBANDS];  ///< VQ encoded high frequency subbands
  201.  
  202.     float lfe_data[2 * DCA_LFE_MAX * (DCA_BLOCKS_MAX + 4)];      ///< Low frequency effect data
  203.     int lfe_scale_factor;
  204.  
  205.     /* Subband samples history (for ADPCM) */
  206.     DECLARE_ALIGNED(16, float, subband_samples_hist)[DCA_PRIM_CHANNELS_MAX][DCA_SUBBANDS][4];
  207.     /* Half size is sufficient for core decoding, but for 96 kHz data
  208.      * we need QMF with 64 subbands and 1024 samples. */
  209.     DECLARE_ALIGNED(32, float, subband_fir_hist)[DCA_PRIM_CHANNELS_MAX][1024];
  210.     DECLARE_ALIGNED(32, float, subband_fir_noidea)[DCA_PRIM_CHANNELS_MAX][64];
  211.     int hist_index[DCA_PRIM_CHANNELS_MAX];
  212.     DECLARE_ALIGNED(32, float, raXin)[32];
  213.  
  214.     int output;                 ///< type of output
  215.  
  216.     DECLARE_ALIGNED(32, float, subband_samples)[DCA_BLOCKS_MAX][DCA_PRIM_CHANNELS_MAX][DCA_SUBBANDS][8];
  217.     float *samples_chanptr[DCA_PRIM_CHANNELS_MAX + 1];
  218.     float *extra_channels[DCA_PRIM_CHANNELS_MAX + 1];
  219.     uint8_t *extra_channels_buffer;
  220.     unsigned int extra_channels_buffer_size;
  221.  
  222.     uint8_t dca_buffer[DCA_MAX_FRAME_SIZE + DCA_MAX_EXSS_HEADER_SIZE + DCA_BUFFER_PADDING_SIZE];
  223.     int dca_buffer_size;        ///< how much data is in the dca_buffer
  224.  
  225.     const int8_t *channel_order_tab;  ///< channel reordering table, lfe and non lfe
  226.     GetBitContext gb;
  227.     /* Current position in DCA frame */
  228.     int current_subframe;
  229.     int current_subsubframe;
  230.  
  231.     int core_ext_mask;          ///< present extensions in the core substream
  232.     int exss_ext_mask;          ///< Non-core extensions
  233.  
  234.     /* XCh extension information */
  235.     int xch_present;            ///< XCh extension present and valid
  236.     int xch_base_channel;       ///< index of first (only) channel containing XCH data
  237.     int xch_disable;            ///< whether the XCh extension should be decoded or not
  238.  
  239.     /* XXCH extension information */
  240.     int xxch_chset;
  241.     int xxch_nbits_spk_mask;
  242.     uint32_t xxch_core_spkmask;
  243.     uint32_t xxch_spk_masks[4]; /* speaker masks, last element is core mask */
  244.     int xxch_chset_nch[4];
  245.     float xxch_dmix_sf[DCA_CHSETS_MAX];
  246.  
  247.     uint32_t xxch_dmix_embedded;  /* lower layer has mix pre-embedded, per chset */
  248.     float xxch_dmix_coeff[DCA_PRIM_CHANNELS_MAX][32]; /* worst case sizing */
  249.  
  250.     int8_t xxch_order_tab[32];
  251.     int8_t lfe_index;
  252.  
  253.     /* XLL extension information */
  254.     int xll_disable;
  255.     int xll_nch_sets;           ///< number of channel sets per frame
  256.     int xll_channels;           ///< total number of channels (in all channel sets)
  257.     int xll_residual_channels;  ///< number of residual channels
  258.     int xll_segments;           ///< number of segments per frame
  259.     int xll_log_smpl_in_seg;    ///< supposedly this is "nBits4SamplLoci"
  260.     int xll_smpl_in_seg;        ///< samples in segment per one frequency band for the first channel set
  261.     int xll_bits4seg_size;      ///< number of bits used to read segment size
  262.     int xll_banddata_crc;       ///< presence of CRC16 within each frequency band
  263.     int xll_scalable_lsb;
  264.     int xll_bits4ch_mask;       ///< channel position mask
  265.     int xll_fixed_lsb_width;
  266.     XllChSetSubHeader xll_chsets[DCA_XLL_CHSETS_MAX];
  267.     XllNavi xll_navi;
  268.     int *xll_sample_buf;
  269.     unsigned int xll_sample_buf_size;
  270.  
  271.     /* ExSS header parser */
  272.     int static_fields;          ///< static fields present
  273.     int mix_metadata;           ///< mixing metadata present
  274.     int num_mix_configs;        ///< number of mix out configurations
  275.     int mix_config_num_ch[4];   ///< number of channels in each mix out configuration
  276.  
  277.     int profile;
  278.     int one2one_map_chtospkr;
  279.  
  280.     int debug_flag;             ///< used for suppressing repeated error messages output
  281.     AVFloatDSPContext *fdsp;
  282.     FFTContext imdct;
  283.     SynthFilterContext synth;
  284.     DCADSPContext dcadsp;
  285.     QMF64_table *qmf64_table;
  286.     FmtConvertContext fmt_conv;
  287. } DCAContext;
  288.  
  289. extern av_export const uint32_t avpriv_dca_sample_rates[16];
  290.  
  291. /**
  292.  * Convert bitstream to one representation based on sync marker
  293.  */
  294. int avpriv_dca_convert_bitstream(const uint8_t *src, int src_size, uint8_t *dst,
  295.                              int max_size);
  296.  
  297. int ff_dca_xbr_parse_frame(DCAContext *s);
  298. int ff_dca_xxch_decode_frame(DCAContext *s);
  299.  
  300. void ff_dca_exss_parse_header(DCAContext *s);
  301.  
  302. int ff_dca_xll_decode_header(DCAContext *s);
  303. int ff_dca_xll_decode_navi(DCAContext *s, int asset_end);
  304. int ff_dca_xll_decode_audio(DCAContext *s, AVFrame *frame);
  305.  
  306. #endif /* AVCODEC_DCA_H */
  307.