diff --git a/libavcodec/imc.c b/libavcodec/imc.c index e3f5e15324..b803176761 100644 --- a/libavcodec/imc.c +++ b/libavcodec/imc.c @@ -100,6 +100,8 @@ typedef struct { DECLARE_ALIGNED(32, FFTComplex, samples)[COEFFS / 2]; float *out_samples; + int coef0_pos; + int8_t cyclTab[32], cyclTab2[32]; float weights1[31], weights2[31]; } IMCContext; @@ -336,6 +338,17 @@ static void imc_read_level_coeffs(IMCContext *q, int stream_format_code, } } +static void imc_read_level_coeffs_raw(IMCContext *q, int stream_format_code, + int *levlCoeffs) +{ + int i; + + q->coef0_pos = get_bits(&q->gb, 5); + levlCoeffs[0] = get_bits(&q->gb, 7); + for (i = 1; i < BANDS; i++) + levlCoeffs[i] = get_bits(&q->gb, 4); +} + static void imc_decode_level_coefficients(IMCContext *q, int *levlCoeffBuf, float *flcoeffs1, float *flcoeffs2) { @@ -390,6 +403,28 @@ static void imc_decode_level_coefficients2(IMCContext *q, int *levlCoeffBuf, } } +static void imc_decode_level_coefficients_raw(IMCContext *q, int *levlCoeffBuf, + float *flcoeffs1, float *flcoeffs2) +{ + int i, level, pos; + float tmp, tmp2; + + pos = q->coef0_pos; + flcoeffs1[pos] = 20000.0 / pow (2, levlCoeffBuf[0] * 0.18945); // 0.18945 = log2(10) * 0.05703125 + flcoeffs2[pos] = log2f(flcoeffs1[0]); + tmp = flcoeffs1[pos]; + tmp2 = flcoeffs2[pos]; + + levlCoeffBuf++; + for (i = 0; i < BANDS; i++) { + if (i == pos) + continue; + level = *levlCoeffBuf++; + flcoeffs1[i] = tmp * powf(10.0, -level * 0.4375); //todo tab + flcoeffs2[i] = tmp2 - 1.4533435415 * level; // 1.4533435415 = log2(10) * 0.4375 + } +} + /** * Perform bit allocation depending on bits available */ @@ -762,100 +797,10 @@ static int imc_get_coeffs(IMCContext *q, IMCChannel *chctx) return 0; } -static int imc_decode_block(AVCodecContext *avctx, IMCContext *q, int ch) +static void imc_refine_bit_allocation(IMCContext *q, IMCChannel *chctx) { - int stream_format_code; - int imc_hdr, i, j, ret; - int flag; + int i, j; int bits, summer; - int counter, bitscount; - IMCChannel *chctx = q->chctx + ch; - - - /* Check the frame header */ - imc_hdr = get_bits(&q->gb, 9); - if (imc_hdr & 0x18) { - av_log(avctx, AV_LOG_ERROR, "frame header check failed!\n"); - av_log(avctx, AV_LOG_ERROR, "got %X.\n", imc_hdr); - return AVERROR_INVALIDDATA; - } - stream_format_code = get_bits(&q->gb, 3); - - if (stream_format_code & 1) { - avpriv_request_sample(avctx, "Stream format %X", stream_format_code); - return AVERROR_PATCHWELCOME; - } - - if (stream_format_code & 0x04) - chctx->decoder_reset = 1; - - if (chctx->decoder_reset) { - for (i = 0; i < BANDS; i++) - chctx->old_floor[i] = 1.0; - for (i = 0; i < COEFFS; i++) - chctx->CWdecoded[i] = 0; - chctx->decoder_reset = 0; - } - - flag = get_bits1(&q->gb); - imc_read_level_coeffs(q, stream_format_code, chctx->levlCoeffBuf); - - if (stream_format_code & 0x4) - imc_decode_level_coefficients(q, chctx->levlCoeffBuf, - chctx->flcoeffs1, chctx->flcoeffs2); - else - imc_decode_level_coefficients2(q, chctx->levlCoeffBuf, chctx->old_floor, - chctx->flcoeffs1, chctx->flcoeffs2); - - memcpy(chctx->old_floor, chctx->flcoeffs1, 32 * sizeof(float)); - - counter = 0; - for (i = 0; i < BANDS; i++) { - if (chctx->levlCoeffBuf[i] == 16) { - chctx->bandWidthT[i] = 0; - counter++; - } else - chctx->bandWidthT[i] = band_tab[i + 1] - band_tab[i]; - } - memset(chctx->bandFlagsBuf, 0, BANDS * sizeof(int)); - for (i = 0; i < BANDS - 1; i++) { - if (chctx->bandWidthT[i]) - chctx->bandFlagsBuf[i] = get_bits1(&q->gb); - } - - imc_calculate_coeffs(q, chctx->flcoeffs1, chctx->flcoeffs2, chctx->bandWidthT, chctx->flcoeffs3, chctx->flcoeffs5); - - bitscount = 0; - /* first 4 bands will be assigned 5 bits per coefficient */ - if (stream_format_code & 0x2) { - bitscount += 15; - - chctx->bitsBandT[0] = 5; - chctx->CWlengthT[0] = 5; - chctx->CWlengthT[1] = 5; - chctx->CWlengthT[2] = 5; - for (i = 1; i < 4; i++) { - bits = (chctx->levlCoeffBuf[i] == 16) ? 0 : 5; - chctx->bitsBandT[i] = bits; - for (j = band_tab[i]; j < band_tab[i + 1]; j++) { - chctx->CWlengthT[j] = bits; - bitscount += bits; - } - } - } - if (avctx->codec_id == AV_CODEC_ID_IAC) { - bitscount += !!chctx->bandWidthT[BANDS - 1]; - if (!(stream_format_code & 0x2)) - bitscount += 16; - } - - if ((ret = bit_allocation(q, chctx, stream_format_code, - 512 - bitscount - get_bits_count(&q->gb), - flag)) < 0) { - av_log(avctx, AV_LOG_ERROR, "Bit allocations failed\n"); - chctx->decoder_reset = 1; - return ret; - } for (i = 0; i < BANDS; i++) { chctx->sumLenArr[i] = 0; @@ -894,6 +839,124 @@ static int imc_decode_block(AVCodecContext *avctx, IMCContext *q, int ch) } } imc_adjust_bit_allocation(q, chctx, summer); +} + +static int imc_decode_block(AVCodecContext *avctx, IMCContext *q, int ch) +{ + int stream_format_code; + int imc_hdr, i, j, ret; + int flag; + int bits; + int counter, bitscount; + IMCChannel *chctx = q->chctx + ch; + + + /* Check the frame header */ + imc_hdr = get_bits(&q->gb, 9); + if (imc_hdr & 0x18) { + av_log(avctx, AV_LOG_ERROR, "frame header check failed!\n"); + av_log(avctx, AV_LOG_ERROR, "got %X.\n", imc_hdr); + return AVERROR_INVALIDDATA; + } + stream_format_code = get_bits(&q->gb, 3); + + if (stream_format_code & 0x04) + chctx->decoder_reset = 1; + + if (chctx->decoder_reset) { + for (i = 0; i < BANDS; i++) + chctx->old_floor[i] = 1.0; + for (i = 0; i < COEFFS; i++) + chctx->CWdecoded[i] = 0; + chctx->decoder_reset = 0; + } + + flag = get_bits1(&q->gb); + if (stream_format_code & 0x1) + imc_decode_level_coefficients_raw(q, chctx->levlCoeffBuf, + chctx->flcoeffs1, chctx->flcoeffs2); + else if (stream_format_code & 0x1) + imc_read_level_coeffs_raw(q, stream_format_code, chctx->levlCoeffBuf); + else + imc_read_level_coeffs(q, stream_format_code, chctx->levlCoeffBuf); + + if (stream_format_code & 0x4) + imc_decode_level_coefficients(q, chctx->levlCoeffBuf, + chctx->flcoeffs1, chctx->flcoeffs2); + else + imc_decode_level_coefficients2(q, chctx->levlCoeffBuf, chctx->old_floor, + chctx->flcoeffs1, chctx->flcoeffs2); + + memcpy(chctx->old_floor, chctx->flcoeffs1, 32 * sizeof(float)); + + counter = 0; + if (stream_format_code & 0x1) { + for (i = 0; i < BANDS; i++) { + chctx->bandWidthT[i] = band_tab[i + 1] - band_tab[i]; + chctx->bandFlagsBuf[i] = 0; + chctx->flcoeffs3[i] = chctx->flcoeffs2[i] * 2; + chctx->flcoeffs5[i] = 1.0; + } + } else { + for (i = 0; i < BANDS; i++) { + if (chctx->levlCoeffBuf[i] == 16) { + chctx->bandWidthT[i] = 0; + counter++; + } else + chctx->bandWidthT[i] = band_tab[i + 1] - band_tab[i]; + } + + memset(chctx->bandFlagsBuf, 0, BANDS * sizeof(int)); + for (i = 0; i < BANDS - 1; i++) + if (chctx->bandWidthT[i]) + chctx->bandFlagsBuf[i] = get_bits1(&q->gb); + + imc_calculate_coeffs(q, chctx->flcoeffs1, chctx->flcoeffs2, + chctx->bandWidthT, chctx->flcoeffs3, + chctx->flcoeffs5); + } + + bitscount = 0; + /* first 4 bands will be assigned 5 bits per coefficient */ + if (stream_format_code & 0x2) { + bitscount += 15; + + chctx->bitsBandT[0] = 5; + chctx->CWlengthT[0] = 5; + chctx->CWlengthT[1] = 5; + chctx->CWlengthT[2] = 5; + for (i = 1; i < 4; i++) { + if (stream_format_code & 0x1) + bits = 5; + else + bits = (chctx->levlCoeffBuf[i] == 16) ? 0 : 5; + chctx->bitsBandT[i] = bits; + for (j = band_tab[i]; j < band_tab[i + 1]; j++) { + chctx->CWlengthT[j] = bits; + bitscount += bits; + } + } + } + if (avctx->codec_id == AV_CODEC_ID_IAC) { + bitscount += !!chctx->bandWidthT[BANDS - 1]; + if (!(stream_format_code & 0x2)) + bitscount += 16; + } + + if ((ret = bit_allocation(q, chctx, stream_format_code, + 512 - bitscount - get_bits_count(&q->gb), + flag)) < 0) { + av_log(avctx, AV_LOG_ERROR, "Bit allocations failed\n"); + chctx->decoder_reset = 1; + return ret; + } + + if (stream_format_code & 0x1) { + for (i = 0; i < BANDS; i++) + chctx->skipFlags[i] = 0; + } else { + imc_refine_bit_allocation(q, chctx); + } for (i = 0; i < BANDS; i++) { chctx->sumLenArr[i] = 0;