1049 lines · 38.0 KB · C
RawDownload
/* * MPEG-4 Parametric Stereo decoding functions * Copyright (c) 2010 Alex Converse <alex.converse@gmail.com> * * 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 * * Note: Rounding-to-nearest used unless otherwise stated * */ #include <stdint.h>#include "libavutil/common.h"#include "libavutil/mathematics.h"#include "avcodec.h"#include "get_bits.h"#include "aacps.h"#if USE_FIXED#include "aacps_fixed_tablegen.h"#else#include "libavutil/internal.h"#include "aacps_tablegen.h"#endif /* USE_FIXED */#include "aacpsdata.c" #define PS_BASELINE 0  ///< Operate in Baseline PS mode                       ///< Baseline implies 10 or 20 stereo bands,                       ///< mixing mode A, and no ipd/opd #define numQMFSlots 32 //numTimeSlots * RATE static const int8_t num_env_tab[2][4] = {    { 0, 1, 2, 4, },    { 1, 2, 3, 4, },}; static const int8_t nr_iidicc_par_tab[] = {    10, 20, 34, 10, 20, 34,}; static const int8_t nr_iidopd_par_tab[] = {     5, 11, 17,  5, 11, 17,}; enum {    huff_iid_df1,    huff_iid_dt1,    huff_iid_df0,    huff_iid_dt0,    huff_icc_df,    huff_icc_dt,    huff_ipd_df,    huff_ipd_dt,    huff_opd_df,    huff_opd_dt,}; static const int huff_iid[] = {    huff_iid_df0,    huff_iid_df1,    huff_iid_dt0,    huff_iid_dt1,}; static VLC vlc_ps[10]; #define READ_PAR_DATA(PAR, OFFSET, MASK, ERR_CONDITION) \/** \ * Read Inter-channel Intensity Difference/Inter-Channel Coherence/ \ * Inter-channel Phase Difference/Overall Phase Difference parameters from the \ * bitstream. \ * \ * @param avctx contains the current codec context \ * @param gb    pointer to the input bitstream \ * @param ps    pointer to the Parametric Stereo context \ * @param PAR   pointer to the parameter to be read \ * @param e     envelope to decode \ * @param dt    1: time delta-coded, 0: frequency delta-coded \ */ \static int read_ ## PAR ## _data(AVCodecContext *avctx, GetBitContext *gb, PSContext *ps, \                        int8_t (*PAR)[PS_MAX_NR_IIDICC], int table_idx, int e, int dt) \{ \    int b, num = ps->nr_ ## PAR ## _par; \    VLC_TYPE (*vlc_table)[2] = vlc_ps[table_idx].table; \    if (dt) { \        int e_prev = e ? e - 1 : ps->num_env_old - 1; \        e_prev = FFMAX(e_prev, 0); \        for (b = 0; b < num; b++) { \            int val = PAR[e_prev][b] + get_vlc2(gb, vlc_table, 9, 3) - OFFSET; \            if (MASK) val &= MASK; \            PAR[e][b] = val; \            if (ERR_CONDITION) \                goto err; \        } \    } else { \        int val = 0; \        for (b = 0; b < num; b++) { \            val += get_vlc2(gb, vlc_table, 9, 3) - OFFSET; \            if (MASK) val &= MASK; \            PAR[e][b] = val; \            if (ERR_CONDITION) \                goto err; \        } \    } \    return 0; \err: \    av_log(avctx, AV_LOG_ERROR, "illegal "#PAR"\n"); \    return -1; \} READ_PAR_DATA(iid,    huff_offset[table_idx],    0, FFABS(ps->iid_par[e][b]) > 7 + 8 * ps->iid_quant)READ_PAR_DATA(icc,    huff_offset[table_idx],    0, ps->icc_par[e][b] > 7U)READ_PAR_DATA(ipdopd,                      0, 0x07, 0) static int ps_read_extension_data(GetBitContext *gb, PSContext *ps, int ps_extension_id){    int e;    int count = get_bits_count(gb);     if (ps_extension_id)        return 0;     ps->enable_ipdopd = get_bits1(gb);    if (ps->enable_ipdopd) {        for (e = 0; e < ps->num_env; e++) {            int dt = get_bits1(gb);            read_ipdopd_data(NULL, gb, ps, ps->ipd_par, dt ? huff_ipd_dt : huff_ipd_df, e, dt);            dt = get_bits1(gb);            read_ipdopd_data(NULL, gb, ps, ps->opd_par, dt ? huff_opd_dt : huff_opd_df, e, dt);        }    }    skip_bits1(gb);      //reserved_ps    return get_bits_count(gb) - count;} static void ipdopd_reset(int8_t *ipd_hist, int8_t *opd_hist){    int i;    for (i = 0; i < PS_MAX_NR_IPDOPD; i++) {        opd_hist[i] = 0;        ipd_hist[i] = 0;    }} int AAC_RENAME(ff_ps_read_data)(AVCodecContext *avctx, GetBitContext *gb_host, PSContext *ps, int bits_left){    int e;    int bit_count_start = get_bits_count(gb_host);    int header;    int bits_consumed;    GetBitContext gbc = *gb_host, *gb = &gbc;     header = get_bits1(gb);    if (header) {     //enable_ps_header        ps->enable_iid = get_bits1(gb);        if (ps->enable_iid) {            int iid_mode = get_bits(gb, 3);            if (iid_mode > 5) {                av_log(avctx, AV_LOG_ERROR, "iid_mode %d is reserved.\n",                       iid_mode);                goto err;            }            ps->nr_iid_par    = nr_iidicc_par_tab[iid_mode];            ps->iid_quant     = iid_mode > 2;            ps->nr_ipdopd_par = nr_iidopd_par_tab[iid_mode];        }        ps->enable_icc = get_bits1(gb);        if (ps->enable_icc) {            ps->icc_mode = get_bits(gb, 3);            if (ps->icc_mode > 5) {                av_log(avctx, AV_LOG_ERROR, "icc_mode %d is reserved.\n",                       ps->icc_mode);                goto err;            }            ps->nr_icc_par = nr_iidicc_par_tab[ps->icc_mode];        }        ps->enable_ext = get_bits1(gb);    }     ps->frame_class = get_bits1(gb);    ps->num_env_old = ps->num_env;    ps->num_env     = num_env_tab[ps->frame_class][get_bits(gb, 2)];     ps->border_position[0] = -1;    if (ps->frame_class) {        for (e = 1; e <= ps->num_env; e++)            ps->border_position[e] = get_bits(gb, 5);    } else        for (e = 1; e <= ps->num_env; e++)            ps->border_position[e] = (e * numQMFSlots >> ff_log2_tab[ps->num_env]) - 1;     if (ps->enable_iid) {        for (e = 0; e < ps->num_env; e++) {            int dt = get_bits1(gb);            if (read_iid_data(avctx, gb, ps, ps->iid_par, huff_iid[2*dt+ps->iid_quant], e, dt))                goto err;        }    } else        memset(ps->iid_par, 0, sizeof(ps->iid_par));     if (ps->enable_icc)        for (e = 0; e < ps->num_env; e++) {            int dt = get_bits1(gb);            if (read_icc_data(avctx, gb, ps, ps->icc_par, dt ? huff_icc_dt : huff_icc_df, e, dt))                goto err;        }    else        memset(ps->icc_par, 0, sizeof(ps->icc_par));     if (ps->enable_ext) {        int cnt = get_bits(gb, 4);        if (cnt == 15) {            cnt += get_bits(gb, 8);        }        cnt *= 8;        while (cnt > 7) {            int ps_extension_id = get_bits(gb, 2);            cnt -= 2 + ps_read_extension_data(gb, ps, ps_extension_id);        }        if (cnt < 0) {            av_log(avctx, AV_LOG_ERROR, "ps extension overflow %d\n", cnt);            goto err;        }        skip_bits(gb, cnt);    }     ps->enable_ipdopd &= !PS_BASELINE;     //Fix up envelopes    if (!ps->num_env || ps->border_position[ps->num_env] < numQMFSlots - 1) {        //Create a fake envelope        int source = ps->num_env ? ps->num_env - 1 : ps->num_env_old - 1;        int b;        if (source >= 0 && source != ps->num_env) {            if (ps->enable_iid) {                memcpy(ps->iid_par+ps->num_env, ps->iid_par+source, sizeof(ps->iid_par[0]));            }            if (ps->enable_icc) {                memcpy(ps->icc_par+ps->num_env, ps->icc_par+source, sizeof(ps->icc_par[0]));            }            if (ps->enable_ipdopd) {                memcpy(ps->ipd_par+ps->num_env, ps->ipd_par+source, sizeof(ps->ipd_par[0]));                memcpy(ps->opd_par+ps->num_env, ps->opd_par+source, sizeof(ps->opd_par[0]));            }        }        if (ps->enable_iid){            for (b = 0; b < ps->nr_iid_par; b++) {                if (FFABS(ps->iid_par[ps->num_env][b]) > 7 + 8 * ps->iid_quant) {                    av_log(avctx, AV_LOG_ERROR, "iid_par invalid\n");                    goto err;                }            }        }        if (ps->enable_icc){            for (b = 0; b < ps->nr_iid_par; b++) {                if (ps->icc_par[ps->num_env][b] > 7U) {                    av_log(avctx, AV_LOG_ERROR, "icc_par invalid\n");                    goto err;                }            }        }        ps->num_env++;        ps->border_position[ps->num_env] = numQMFSlots - 1;    }      ps->is34bands_old = ps->is34bands;    if (!PS_BASELINE && (ps->enable_iid || ps->enable_icc))        ps->is34bands = (ps->enable_iid && ps->nr_iid_par == 34) ||                        (ps->enable_icc && ps->nr_icc_par == 34);     //Baseline    if (!ps->enable_ipdopd) {        memset(ps->ipd_par, 0, sizeof(ps->ipd_par));        memset(ps->opd_par, 0, sizeof(ps->opd_par));    }     if (header)        ps->start = 1;     bits_consumed = get_bits_count(gb) - bit_count_start;    if (bits_consumed <= bits_left) {        skip_bits_long(gb_host, bits_consumed);        return bits_consumed;    }    av_log(avctx, AV_LOG_ERROR, "Expected to read %d PS bits actually read %d.\n", bits_left, bits_consumed);err:    ps->start = 0;    skip_bits_long(gb_host, bits_left);    memset(ps->iid_par, 0, sizeof(ps->iid_par));    memset(ps->icc_par, 0, sizeof(ps->icc_par));    memset(ps->ipd_par, 0, sizeof(ps->ipd_par));    memset(ps->opd_par, 0, sizeof(ps->opd_par));    return bits_left;} /** Split one subband into 2 subsubbands with a symmetric real filter. * The filter must have its non-center even coefficients equal to zero. */static void hybrid2_re(INTFLOAT (*in)[2], INTFLOAT (*out)[32][2], const INTFLOAT filter[8], int len, int reverse){    int i, j;    for (i = 0; i < len; i++, in++) {        INT64FLOAT re_in = AAC_MUL31(filter[6], in[6][0]); //real inphase        INT64FLOAT re_op = 0.0f;                          //real out of phase        INT64FLOAT im_in = AAC_MUL31(filter[6], in[6][1]); //imag inphase        INT64FLOAT im_op = 0.0f;                          //imag out of phase        for (j = 0; j < 6; j += 2) {            re_op += (INT64FLOAT)filter[j+1] * (in[j+1][0] + in[12-j-1][0]);            im_op += (INT64FLOAT)filter[j+1] * (in[j+1][1] + in[12-j-1][1]);        } #if USE_FIXED        re_op = (re_op + 0x40000000) >> 31;        im_op = (im_op + 0x40000000) >> 31;#endif /* USE_FIXED */         out[ reverse][i][0] = (INTFLOAT)(re_in + re_op);        out[ reverse][i][1] = (INTFLOAT)(im_in + im_op);        out[!reverse][i][0] = (INTFLOAT)(re_in - re_op);        out[!reverse][i][1] = (INTFLOAT)(im_in - im_op);    }} /** Split one subband into 6 subsubbands with a complex filter */static void hybrid6_cx(PSDSPContext *dsp, INTFLOAT (*in)[2], INTFLOAT (*out)[32][2],                       TABLE_CONST INTFLOAT (*filter)[8][2], int len){    int i;    int N = 8;    LOCAL_ALIGNED_16(INTFLOAT, temp, [8], [2]);     for (i = 0; i < len; i++, in++) {        dsp->hybrid_analysis(temp, in, (const INTFLOAT (*)[8][2]) filter, 1, N);        out[0][i][0] = temp[6][0];        out[0][i][1] = temp[6][1];        out[1][i][0] = temp[7][0];        out[1][i][1] = temp[7][1];        out[2][i][0] = temp[0][0];        out[2][i][1] = temp[0][1];        out[3][i][0] = temp[1][0];        out[3][i][1] = temp[1][1];        out[4][i][0] = temp[2][0] + temp[5][0];        out[4][i][1] = temp[2][1] + temp[5][1];        out[5][i][0] = temp[3][0] + temp[4][0];        out[5][i][1] = temp[3][1] + temp[4][1];    }} static void hybrid4_8_12_cx(PSDSPContext *dsp,                            INTFLOAT (*in)[2], INTFLOAT (*out)[32][2],                            TABLE_CONST INTFLOAT (*filter)[8][2], int N, int len){    int i;     for (i = 0; i < len; i++, in++) {        dsp->hybrid_analysis(out[0] + i, in, (const INTFLOAT (*)[8][2]) filter, 32, N);    }} static void hybrid_analysis(PSDSPContext *dsp, INTFLOAT out[91][32][2],                            INTFLOAT in[5][44][2], INTFLOAT L[2][38][64],                            int is34, int len){    int i, j;    for (i = 0; i < 5; i++) {        for (j = 0; j < 38; j++) {            in[i][j+6][0] = L[0][j][i];            in[i][j+6][1] = L[1][j][i];        }    }    if (is34) {        hybrid4_8_12_cx(dsp, in[0], out,    f34_0_12, 12, len);        hybrid4_8_12_cx(dsp, in[1], out+12, f34_1_8,   8, len);        hybrid4_8_12_cx(dsp, in[2], out+20, f34_2_4,   4, len);        hybrid4_8_12_cx(dsp, in[3], out+24, f34_2_4,   4, len);        hybrid4_8_12_cx(dsp, in[4], out+28, f34_2_4,   4, len);        dsp->hybrid_analysis_ileave(out + 27, L, 5, len);    } else {        hybrid6_cx(dsp, in[0], out, f20_0_8, len);        hybrid2_re(in[1], out+6, g1_Q2, len, 1);        hybrid2_re(in[2], out+8, g1_Q2, len, 0);        dsp->hybrid_analysis_ileave(out + 7, L, 3, len);    }    //update in_buf    for (i = 0; i < 5; i++) {        memcpy(in[i], in[i]+32, 6 * sizeof(in[i][0]));    }} static void hybrid_synthesis(PSDSPContext *dsp, INTFLOAT out[2][38][64],                             INTFLOAT in[91][32][2], int is34, int len){    int i, n;    if (is34) {        for (n = 0; n < len; n++) {            memset(out[0][n], 0, 5*sizeof(out[0][n][0]));            memset(out[1][n], 0, 5*sizeof(out[1][n][0]));            for (i = 0; i < 12; i++) {                out[0][n][0] += in[   i][n][0];                out[1][n][0] += in[   i][n][1];            }            for (i = 0; i < 8; i++) {                out[0][n][1] += in[12+i][n][0];                out[1][n][1] += in[12+i][n][1];            }            for (i = 0; i < 4; i++) {                out[0][n][2] += in[20+i][n][0];                out[1][n][2] += in[20+i][n][1];                out[0][n][3] += in[24+i][n][0];                out[1][n][3] += in[24+i][n][1];                out[0][n][4] += in[28+i][n][0];                out[1][n][4] += in[28+i][n][1];            }        }        dsp->hybrid_synthesis_deint(out, in + 27, 5, len);    } else {        for (n = 0; n < len; n++) {            out[0][n][0] = in[0][n][0] + in[1][n][0] + in[2][n][0] +                           in[3][n][0] + in[4][n][0] + in[5][n][0];            out[1][n][0] = in[0][n][1] + in[1][n][1] + in[2][n][1] +                           in[3][n][1] + in[4][n][1] + in[5][n][1];            out[0][n][1] = in[6][n][0] + in[7][n][0];            out[1][n][1] = in[6][n][1] + in[7][n][1];            out[0][n][2] = in[8][n][0] + in[9][n][0];            out[1][n][2] = in[8][n][1] + in[9][n][1];        }        dsp->hybrid_synthesis_deint(out, in + 7, 3, len);    }} /// All-pass filter decay slope#define DECAY_SLOPE      Q30(0.05f)/// Number of frequency bands that can be addressed by the parameter index, b(k)static const int   NR_PAR_BANDS[]      = { 20, 34 };static const int   NR_IPDOPD_BANDS[]   = { 11, 17 };/// Number of frequency bands that can be addressed by the sub subband index, kstatic const int   NR_BANDS[]          = { 71, 91 };/// Start frequency band for the all-pass filter decay slopestatic const int   DECAY_CUTOFF[]      = { 10, 32 };/// Number of all-pass filer bandsstatic const int   NR_ALLPASS_BANDS[]  = { 30, 50 };/// First stereo band using the short one sample delaystatic const int   SHORT_DELAY_BAND[]  = { 42, 62 }; /** Table 8.46 */static void map_idx_10_to_20(int8_t *par_mapped, const int8_t *par, int full){    int b;    if (full)        b = 9;    else {        b = 4;        par_mapped[10] = 0;    }    for (; b >= 0; b--) {        par_mapped[2*b+1] = par_mapped[2*b] = par[b];    }} static void map_idx_34_to_20(int8_t *par_mapped, const int8_t *par, int full){    par_mapped[ 0] = (2*par[ 0] +   par[ 1]) / 3;    par_mapped[ 1] = (  par[ 1] + 2*par[ 2]) / 3;    par_mapped[ 2] = (2*par[ 3] +   par[ 4]) / 3;    par_mapped[ 3] = (  par[ 4] + 2*par[ 5]) / 3;    par_mapped[ 4] = (  par[ 6] +   par[ 7]) / 2;    par_mapped[ 5] = (  par[ 8] +   par[ 9]) / 2;    par_mapped[ 6] =    par[10];    par_mapped[ 7] =    par[11];    par_mapped[ 8] = (  par[12] +   par[13]) / 2;    par_mapped[ 9] = (  par[14] +   par[15]) / 2;    par_mapped[10] =    par[16];    if (full) {        par_mapped[11] =    par[17];        par_mapped[12] =    par[18];        par_mapped[13] =    par[19];        par_mapped[14] = (  par[20] +   par[21]) / 2;        par_mapped[15] = (  par[22] +   par[23]) / 2;        par_mapped[16] = (  par[24] +   par[25]) / 2;        par_mapped[17] = (  par[26] +   par[27]) / 2;        par_mapped[18] = (  par[28] +   par[29] +   par[30] +   par[31]) / 4;        par_mapped[19] = (  par[32] +   par[33]) / 2;    }} static void map_val_34_to_20(INTFLOAT par[PS_MAX_NR_IIDICC]){

Lines 1–500 of 1049.

Keyboard shortcuts

General

Show keyboard shortcuts
?
Focus the current search
/

Issues and pull requests

Open filters
f

Issues, pull requests and Inbox

Select next item
j↓
Select previous item
k↑
Open selected item
Enter
Dismiss filters or clear selection
Esc

Repository code

Go to file
t