242 lines
9.8 KiB
C
242 lines
9.8 KiB
C
|
/*
|
||
|
|
* ATRAC3+ compatible decoder
|
||
|
|
*
|
||
|
|
* Copyright (c) 2010-2013 Maxim Poliakovski
|
||
|
|
*
|
||
|
|
* This file is part of FFmpeg.
|
||
|
|
*
|
||
|
|
* FFmpeg is free software; you can redistribute it and/or
|
||
|
|
* modify it under the terms of the GNU Lesser General Public
|
||
|
|
* License as published by the Free Software Foundation; either
|
||
|
|
* version 2.1 of the License, or (at your option) any later version.
|
||
|
|
*
|
||
|
|
* FFmpeg is distributed in the hope that it will be useful,
|
||
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||
|
|
* Lesser General Public License for more details.
|
||
|
|
*
|
||
|
|
* You should have received a copy of the GNU Lesser General Public
|
||
|
|
* License along with FFmpeg; if not, write to the Free Software
|
||
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
||
|
|
*/
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @file
|
||
|
|
* Global structures, constants and data for ATRAC3+ decoder.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#ifndef AVCODEC_ATRAC3PLUS_H
|
||
|
|
#define AVCODEC_ATRAC3PLUS_H
|
||
|
|
|
||
|
|
#include <stdint.h>
|
||
|
|
|
||
|
|
#include "libavutil/float_dsp.h"
|
||
|
|
#include "atrac.h"
|
||
|
|
#include "avcodec.h"
|
||
|
|
#include "fft.h"
|
||
|
|
#include "get_bits.h"
|
||
|
|
|
||
|
|
/** Global unit sizes */
|
||
|
|
#define ATRAC3P_SUBBANDS 16 ///< number of PQF subbands
|
||
|
|
#define ATRAC3P_SUBBAND_SAMPLES 128 ///< number of samples per subband
|
||
|
|
#define ATRAC3P_FRAME_SAMPLES (ATRAC3P_SUBBAND_SAMPLES * ATRAC3P_SUBBANDS)
|
||
|
|
|
||
|
|
#define ATRAC3P_PQF_FIR_LEN 12 ///< length of the prototype FIR of the PQF
|
||
|
|
|
||
|
|
/** Global constants */
|
||
|
|
#define ATRAC3P_POWER_COMP_OFF 15 ///< disable power compensation
|
||
|
|
|
||
|
|
/** ATRAC3+ channel unit types */
|
||
|
|
enum Atrac3pChannelUnitTypes {
|
||
|
|
CH_UNIT_MONO = 0, ///< unit containing one coded channel
|
||
|
|
CH_UNIT_STEREO = 1, ///< unit containing two jointly-coded channels
|
||
|
|
CH_UNIT_EXTENSION = 2, ///< unit containing extension information
|
||
|
|
CH_UNIT_TERMINATOR = 3 ///< unit sequence terminator
|
||
|
|
};
|
||
|
|
|
||
|
|
/** Per-channel IPQF history */
|
||
|
|
typedef struct Atrac3pIPQFChannelCtx {
|
||
|
|
DECLARE_ALIGNED(32, float, buf1)[ATRAC3P_PQF_FIR_LEN * 2][8];
|
||
|
|
DECLARE_ALIGNED(32, float, buf2)[ATRAC3P_PQF_FIR_LEN * 2][8];
|
||
|
|
int pos;
|
||
|
|
} Atrac3pIPQFChannelCtx;
|
||
|
|
|
||
|
|
/** Amplitude envelope of a group of sine waves */
|
||
|
|
typedef struct Atrac3pWaveEnvelope {
|
||
|
|
int has_start_point; ///< indicates start point within the GHA window
|
||
|
|
int has_stop_point; ///< indicates stop point within the GHA window
|
||
|
|
int start_pos; ///< start position expressed in n*4 samples
|
||
|
|
int stop_pos; ///< stop position expressed in n*4 samples
|
||
|
|
} Atrac3pWaveEnvelope;
|
||
|
|
|
||
|
|
/** Parameters of a group of sine waves */
|
||
|
|
typedef struct Atrac3pWavesData {
|
||
|
|
Atrac3pWaveEnvelope pend_env; ///< pending envelope from the previous frame
|
||
|
|
Atrac3pWaveEnvelope curr_env; ///< group envelope from the current frame
|
||
|
|
int num_wavs; ///< number of sine waves in the group
|
||
|
|
int start_index; ///< start index into global tones table for that subband
|
||
|
|
} Atrac3pWavesData;
|
||
|
|
|
||
|
|
/** Parameters of a single sine wave */
|
||
|
|
typedef struct Atrac3pWaveParam {
|
||
|
|
int freq_index; ///< wave frequency index
|
||
|
|
int amp_sf; ///< quantized amplitude scale factor
|
||
|
|
int amp_index; ///< quantized amplitude index
|
||
|
|
int phase_index; ///< quantized phase index
|
||
|
|
} Atrac3pWaveParam;
|
||
|
|
|
||
|
|
/** Sound channel parameters */
|
||
|
|
typedef struct Atrac3pChanParams {
|
||
|
|
int ch_num;
|
||
|
|
int num_coded_vals; ///< number of transmitted quant unit values
|
||
|
|
int fill_mode;
|
||
|
|
int split_point;
|
||
|
|
int table_type; ///< table type: 0 - tone?, 1- noise?
|
||
|
|
int qu_wordlen[32]; ///< array of word lengths for each quant unit
|
||
|
|
int qu_sf_idx[32]; ///< array of scale factor indexes for each quant unit
|
||
|
|
int qu_tab_idx[32]; ///< array of code table indexes for each quant unit
|
||
|
|
int16_t spectrum[2048]; ///< decoded IMDCT spectrum
|
||
|
|
uint8_t power_levs[5]; ///< power compensation levels
|
||
|
|
|
||
|
|
/* imdct window shape history (2 frames) for overlapping. */
|
||
|
|
uint8_t wnd_shape_hist[2][ATRAC3P_SUBBANDS]; ///< IMDCT window shape, 0=sine/1=steep
|
||
|
|
uint8_t *wnd_shape; ///< IMDCT window shape for current frame
|
||
|
|
uint8_t *wnd_shape_prev; ///< IMDCT window shape for previous frame
|
||
|
|
|
||
|
|
/* gain control data history (2 frames) for overlapping. */
|
||
|
|
AtracGainInfo gain_data_hist[2][ATRAC3P_SUBBANDS]; ///< gain control data for all subbands
|
||
|
|
AtracGainInfo *gain_data; ///< gain control data for next frame
|
||
|
|
AtracGainInfo *gain_data_prev; ///< gain control data for previous frame
|
||
|
|
int num_gain_subbands; ///< number of subbands with gain control data
|
||
|
|
|
||
|
|
/* tones data history (2 frames) for overlapping. */
|
||
|
|
Atrac3pWavesData tones_info_hist[2][ATRAC3P_SUBBANDS];
|
||
|
|
Atrac3pWavesData *tones_info;
|
||
|
|
Atrac3pWavesData *tones_info_prev;
|
||
|
|
} Atrac3pChanParams;
|
||
|
|
|
||
|
|
/* Per-unit sine wave parameters */
|
||
|
|
typedef struct Atrac3pWaveSynthParams {
|
||
|
|
int tones_present; ///< 1 - tones info present
|
||
|
|
int amplitude_mode; ///< 1 - low range, 0 - high range
|
||
|
|
int num_tone_bands; ///< number of PQF bands with tones
|
||
|
|
uint8_t tone_sharing[ATRAC3P_SUBBANDS]; ///< 1 - subband-wise tone sharing flags
|
||
|
|
uint8_t tone_master[ATRAC3P_SUBBANDS]; ///< 1 - subband-wise tone channel swapping
|
||
|
|
uint8_t invert_phase[ATRAC3P_SUBBANDS]; ///< 1 - subband-wise phase inversion
|
||
|
|
int tones_index; ///< total sum of tones in this unit
|
||
|
|
Atrac3pWaveParam waves[48];
|
||
|
|
} Atrac3pWaveSynthParams;
|
||
|
|
|
||
|
|
/** Channel unit parameters */
|
||
|
|
typedef struct Atrac3pChanUnitCtx {
|
||
|
|
/* channel unit variables */
|
||
|
|
int unit_type; ///< unit type (mono/stereo)
|
||
|
|
int num_quant_units;
|
||
|
|
int num_subbands;
|
||
|
|
int used_quant_units; ///< number of quant units with coded spectrum
|
||
|
|
int num_coded_subbands; ///< number of subbands with coded spectrum
|
||
|
|
int mute_flag; ///< mute flag
|
||
|
|
int use_full_table; ///< 1 - full table list, 0 - restricted one
|
||
|
|
int noise_present; ///< 1 - global noise info present
|
||
|
|
int noise_level_index; ///< global noise level index
|
||
|
|
int noise_table_index; ///< global noise RNG table index
|
||
|
|
uint8_t swap_channels[ATRAC3P_SUBBANDS]; ///< 1 - perform subband-wise channel swapping
|
||
|
|
uint8_t negate_coeffs[ATRAC3P_SUBBANDS]; ///< 1 - subband-wise IMDCT coefficients negation
|
||
|
|
Atrac3pChanParams channels[2];
|
||
|
|
|
||
|
|
/* Variables related to GHA tones */
|
||
|
|
Atrac3pWaveSynthParams wave_synth_hist[2]; ///< waves synth history for two frames
|
||
|
|
Atrac3pWaveSynthParams *waves_info;
|
||
|
|
Atrac3pWaveSynthParams *waves_info_prev;
|
||
|
|
|
||
|
|
Atrac3pIPQFChannelCtx ipqf_ctx[2];
|
||
|
|
DECLARE_ALIGNED(32, float, prev_buf)[2][ATRAC3P_FRAME_SAMPLES]; ///< overlapping buffer
|
||
|
|
} Atrac3pChanUnitCtx;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Initialize VLC tables for bitstream parsing.
|
||
|
|
*/
|
||
|
|
void ff_atrac3p_init_vlcs(void);
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Decode bitstream data of a channel unit.
|
||
|
|
*
|
||
|
|
* @param[in] gb the GetBit context
|
||
|
|
* @param[in,out] ctx ptr to the channel unit context
|
||
|
|
* @param[in] num_channels number of channels to process
|
||
|
|
* @param[in] avctx ptr to the AVCodecContext
|
||
|
|
* @return result code: 0 = OK, otherwise - error code
|
||
|
|
*/
|
||
|
|
int ff_atrac3p_decode_channel_unit(GetBitContext *gb, Atrac3pChanUnitCtx *ctx,
|
||
|
|
int num_channels, AVCodecContext *avctx);
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Initialize IMDCT transform.
|
||
|
|
*
|
||
|
|
* @param[in] avctx ptr to the AVCodecContext
|
||
|
|
* @param[in] mdct_ctx pointer to MDCT transform context
|
||
|
|
*/
|
||
|
|
void ff_atrac3p_init_imdct(AVCodecContext *avctx, FFTContext *mdct_ctx);
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Initialize sine waves synthesizer.
|
||
|
|
*/
|
||
|
|
void ff_atrac3p_init_wave_synth(void);
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Synthesize sine waves for a particular subband.
|
||
|
|
*
|
||
|
|
* @param[in] ch_unit pointer to the channel unit context
|
||
|
|
* @param[in] fdsp pointer to float DSP context
|
||
|
|
* @param[in] ch_num which channel to process
|
||
|
|
* @param[in] sb which subband to process
|
||
|
|
* @param[out] out receives processed data
|
||
|
|
*/
|
||
|
|
void ff_atrac3p_generate_tones(Atrac3pChanUnitCtx *ch_unit, AVFloatDSPContext *fdsp,
|
||
|
|
int ch_num, int sb, float *out);
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Perform power compensation aka noise dithering.
|
||
|
|
*
|
||
|
|
* @param[in] ctx ptr to the channel context
|
||
|
|
* @param[in] fdsp pointer to float DSP context
|
||
|
|
* @param[in] ch_index which channel to process
|
||
|
|
* @param[in,out] sp ptr to channel spectrum to process
|
||
|
|
* @param[in] rng_index indicates which RNG table to use
|
||
|
|
* @param[in] sb_num which subband to process
|
||
|
|
*/
|
||
|
|
void ff_atrac3p_power_compensation(Atrac3pChanUnitCtx *ctx, AVFloatDSPContext *fdsp,
|
||
|
|
int ch_index, float *sp, int rng_index, int sb_num);
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Regular IMDCT and windowing without overlapping,
|
||
|
|
* with spectrum reversal in the odd subbands.
|
||
|
|
*
|
||
|
|
* @param[in] fdsp pointer to float DSP context
|
||
|
|
* @param[in] mdct_ctx pointer to MDCT transform context
|
||
|
|
* @param[in] pIn float input
|
||
|
|
* @param[out] pOut float output
|
||
|
|
* @param[in] wind_id which MDCT window to apply
|
||
|
|
* @param[in] sb subband number
|
||
|
|
*/
|
||
|
|
void ff_atrac3p_imdct(AVFloatDSPContext *fdsp, FFTContext *mdct_ctx, float *pIn,
|
||
|
|
float *pOut, int wind_id, int sb);
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Subband synthesis filter based on the polyphase quadrature (pseudo-QMF)
|
||
|
|
* filter bank.
|
||
|
|
*
|
||
|
|
* @param[in] dct_ctx ptr to the pre-initialized IDCT context
|
||
|
|
* @param[in,out] hist ptr to the filter history
|
||
|
|
* @param[in] in input data to process
|
||
|
|
* @param[out] out receives processed data
|
||
|
|
*/
|
||
|
|
void ff_atrac3p_ipqf(FFTContext *dct_ctx, Atrac3pIPQFChannelCtx *hist,
|
||
|
|
const float *in, float *out);
|
||
|
|
|
||
|
|
extern const uint16_t ff_atrac3p_qu_to_spec_pos[33];
|
||
|
|
extern const float ff_atrac3p_sf_tab[64];
|
||
|
|
extern const float ff_atrac3p_mant_tab[8];
|
||
|
|
|
||
|
|
#endif /* AVCODEC_ATRAC3PLUS_H */
|