mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2024-12-23 12:43:46 +02:00
Split applying of MDCT into several separate functions.
Originally committed as revision 25990 to svn://svn.ffmpeg.org/ffmpeg/trunk
This commit is contained in:
parent
4acc790f3c
commit
0a0a8500fb
@ -312,6 +312,22 @@ static void mdct512(int32_t *out, int16_t *in)
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Apply KBD window to input samples prior to MDCT.
|
||||||
|
*/
|
||||||
|
static void apply_window(int16_t *output, const int16_t *input,
|
||||||
|
const int16_t *window, int n)
|
||||||
|
{
|
||||||
|
int i;
|
||||||
|
int n2 = n >> 1;
|
||||||
|
|
||||||
|
for (i = 0; i < n2; i++) {
|
||||||
|
output[i] = MUL16(input[i], window[i]) >> 15;
|
||||||
|
output[n-i-1] = MUL16(input[n-i-1], window[i]) >> 15;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Calculate the log2() of the maximum absolute value in an array.
|
* Calculate the log2() of the maximum absolute value in an array.
|
||||||
* @param tab input array
|
* @param tab input array
|
||||||
@ -351,6 +367,50 @@ static void lshift_tab(int16_t *tab, int n, int lshift)
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Normalize the input samples to use the maximum available precision.
|
||||||
|
* This assumes signed 16-bit input samples. Exponents are reduced by 9 to
|
||||||
|
* match the 24-bit internal precision for MDCT coefficients.
|
||||||
|
*
|
||||||
|
* @return exponent shift
|
||||||
|
*/
|
||||||
|
static int normalize_samples(AC3EncodeContext *s,
|
||||||
|
int16_t windowed_samples[AC3_WINDOW_SIZE])
|
||||||
|
{
|
||||||
|
int v = 14 - log2_tab(windowed_samples, AC3_WINDOW_SIZE);
|
||||||
|
v = FFMAX(0, v);
|
||||||
|
lshift_tab(windowed_samples, AC3_WINDOW_SIZE, v);
|
||||||
|
return v - 9;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Apply the MDCT to input samples to generate frequency coefficients.
|
||||||
|
* This applies the KBD window and normalizes the input to reduce precision
|
||||||
|
* loss due to fixed-point calculations.
|
||||||
|
*/
|
||||||
|
static void apply_mdct(AC3EncodeContext *s,
|
||||||
|
int16_t planar_samples[AC3_MAX_CHANNELS][AC3_BLOCK_SIZE+AC3_FRAME_SIZE],
|
||||||
|
int8_t exp_shift[AC3_MAX_BLOCKS][AC3_MAX_CHANNELS],
|
||||||
|
int32_t mdct_coef[AC3_MAX_BLOCKS][AC3_MAX_CHANNELS][AC3_MAX_COEFS])
|
||||||
|
{
|
||||||
|
int blk, ch;
|
||||||
|
int16_t windowed_samples[AC3_WINDOW_SIZE];
|
||||||
|
|
||||||
|
for (ch = 0; ch < s->channels; ch++) {
|
||||||
|
for (blk = 0; blk < AC3_MAX_BLOCKS; blk++) {
|
||||||
|
const int16_t *input_samples = &planar_samples[ch][blk * AC3_BLOCK_SIZE];
|
||||||
|
|
||||||
|
apply_window(windowed_samples, input_samples, ff_ac3_window, AC3_WINDOW_SIZE);
|
||||||
|
|
||||||
|
exp_shift[blk][ch] = normalize_samples(s, windowed_samples);
|
||||||
|
|
||||||
|
mdct512(mdct_coef[blk][ch], windowed_samples);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Calculate the sum of absolute differences (SAD) between 2 sets of exponents.
|
* Calculate the sum of absolute differences (SAD) between 2 sets of exponents.
|
||||||
*/
|
*/
|
||||||
@ -1117,7 +1177,6 @@ static int ac3_encode_frame(AVCodecContext *avctx,
|
|||||||
int v;
|
int v;
|
||||||
int blk, blk1, blk2, ch, i;
|
int blk, blk1, blk2, ch, i;
|
||||||
int16_t planar_samples[AC3_MAX_CHANNELS][AC3_BLOCK_SIZE+AC3_FRAME_SIZE];
|
int16_t planar_samples[AC3_MAX_CHANNELS][AC3_BLOCK_SIZE+AC3_FRAME_SIZE];
|
||||||
int16_t windowed_samples[AC3_WINDOW_SIZE];
|
|
||||||
int32_t mdct_coef[AC3_MAX_BLOCKS][AC3_MAX_CHANNELS][AC3_MAX_COEFS];
|
int32_t mdct_coef[AC3_MAX_BLOCKS][AC3_MAX_CHANNELS][AC3_MAX_COEFS];
|
||||||
uint8_t exp[AC3_MAX_BLOCKS][AC3_MAX_CHANNELS][AC3_MAX_COEFS];
|
uint8_t exp[AC3_MAX_BLOCKS][AC3_MAX_CHANNELS][AC3_MAX_COEFS];
|
||||||
uint8_t exp_strategy[AC3_MAX_BLOCKS][AC3_MAX_CHANNELS];
|
uint8_t exp_strategy[AC3_MAX_BLOCKS][AC3_MAX_CHANNELS];
|
||||||
@ -1128,30 +1187,7 @@ static int ac3_encode_frame(AVCodecContext *avctx,
|
|||||||
|
|
||||||
deinterleave_input_samples(s, samples, planar_samples);
|
deinterleave_input_samples(s, samples, planar_samples);
|
||||||
|
|
||||||
/* apply MDCT */
|
apply_mdct(s, planar_samples, exp_shift, mdct_coef);
|
||||||
for (ch = 0; ch < s->channels; ch++) {
|
|
||||||
for (blk = 0; blk < AC3_MAX_BLOCKS; blk++) {
|
|
||||||
int16_t *input_samples = &planar_samples[ch][blk * AC3_BLOCK_SIZE];
|
|
||||||
|
|
||||||
/* apply the MDCT window */
|
|
||||||
for (i = 0; i < AC3_BLOCK_SIZE; i++) {
|
|
||||||
windowed_samples[i] = MUL16(input_samples[i],
|
|
||||||
ff_ac3_window[i]) >> 15;
|
|
||||||
windowed_samples[AC3_WINDOW_SIZE-i-1] = MUL16(input_samples[AC3_WINDOW_SIZE-i-1],
|
|
||||||
ff_ac3_window[i]) >> 15;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Normalize the samples to use the maximum available precision */
|
|
||||||
v = 14 - log2_tab(windowed_samples, AC3_WINDOW_SIZE);
|
|
||||||
if (v < 0)
|
|
||||||
v = 0;
|
|
||||||
exp_shift[blk][ch] = v - 9;
|
|
||||||
lshift_tab(windowed_samples, AC3_WINDOW_SIZE, v);
|
|
||||||
|
|
||||||
/* do the MDCT */
|
|
||||||
mdct512(mdct_coef[blk][ch], windowed_samples);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/* extract exponents */
|
/* extract exponents */
|
||||||
for (ch = 0; ch < s->channels; ch++) {
|
for (ch = 0; ch < s->channels; ch++) {
|
||||||
|
Loading…
Reference in New Issue
Block a user