falkTX
/
FFmpeg
mirror of https://github.com/falkTX/FFmpeg.git
1
0
Fork 0
Code Issues 0 Releases 338 Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
24599 Commits
32 Branches
385MB
Tree: d45b771ca0
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'd45b771ca0'
${ noResults }
FFmpeg/libavcodec/mpegaudiodec.c

2631 lines
79KB
Raw Blame History 

  1. /*

  2.  * MPEG Audio decoder

  3.  * Copyright (c) 2001, 2002 Fabrice Bellard

  4.  *

  5.  * This file is part of FFmpeg.

  6.  *

  7.  * FFmpeg is free software; you can redistribute it and/or

  8.  * modify it under the terms of the GNU Lesser General Public

  9.  * License as published by the Free Software Foundation; either

  10.  * version 2.1 of the License, or (at your option) any later version.

  11.  *

  12.  * FFmpeg is distributed in the hope that it will be useful,

  13.  * but WITHOUT ANY WARRANTY; without even the implied warranty of

  14.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

  15.  * Lesser General Public License for more details.

  16.  *

  17.  * You should have received a copy of the GNU Lesser General Public

  18.  * License along with FFmpeg; if not, write to the Free Software

  19.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

  20.  */

  21. 

  22. /**

  23.  * @file

  24.  * MPEG Audio decoder.

  25.  */

  26. 

  27. #include "avcodec.h"

  28. #include "get_bits.h"

  29. #include "dsputil.h"

  30. 

  31. /*

  32.  * TODO:

  33.  *  - in low precision mode, use more 16 bit multiplies in synth filter

  34.  *  - test lsf / mpeg25 extensively.

  35.  */

  36. 

  37. #include "mpegaudio.h"

  38. #include "mpegaudiodecheader.h"

  39. 

  40. #include "mathops.h"

  41. 

  42. #if CONFIG_FLOAT

  43. #   define SHR(a,b)       ((a)*(1.0f/(1<<(b))))

  44. #   define compute_antialias compute_antialias_float

  45. #   define FIXR_OLD(a)    ((int)((a) * FRAC_ONE + 0.5))

  46. #   define FIXR(x)        ((float)(x))

  47. #   define FIXHR(x)       ((float)(x))

  48. #   define MULH3(x, y, s) ((s)*(y)*(x))

  49. #   define MULLx(x, y, s) ((y)*(x))

  50. #   define RENAME(a) a ## _float

  51. #else

  52. #   define SHR(a,b)       ((a)>>(b))

  53. #   define compute_antialias compute_antialias_integer

  54. /* WARNING: only correct for posititive numbers */

  55. #   define FIXR_OLD(a)    ((int)((a) * FRAC_ONE + 0.5))

  56. #   define FIXR(a)        ((int)((a) * FRAC_ONE + 0.5))

  57. #   define FIXHR(a)       ((int)((a) * (1LL<<32) + 0.5))

  58. #   define MULH3(x, y, s) MULH((s)*(x), y)

  59. #   define MULLx(x, y, s) MULL(x,y,s)

  60. #   define RENAME(a)      a

  61. #endif

  62. 

  63. /****************/

  64. 

  65. #define HEADER_SIZE 4

  66. 

  67. #include "mpegaudiodata.h"

  68. #include "mpegaudiodectab.h"

  69. 

  70. static void compute_antialias_integer(MPADecodeContext *s, GranuleDef *g);

  71. static void compute_antialias_float(MPADecodeContext *s, GranuleDef *g);

  72. static void apply_window_mp3_c(MPA_INT *synth_buf, MPA_INT *window,

  73.                                int *dither_state, OUT_INT *samples, int incr);

  74. 

  75. /* vlc structure for decoding layer 3 huffman tables */

  76. static VLC huff_vlc[16];

  77. static VLC_TYPE huff_vlc_tables[

  78.   0+128+128+128+130+128+154+166+

  79.   142+204+190+170+542+460+662+414

  80.   ][2];

  81. static const int huff_vlc_tables_sizes[16] = {

  82.   0, 128, 128, 128, 130, 128, 154, 166,

  83.   142, 204, 190, 170, 542, 460, 662, 414

  84. };

  85. static VLC huff_quad_vlc[2];

  86. static VLC_TYPE huff_quad_vlc_tables[128+16][2];

  87. static const int huff_quad_vlc_tables_sizes[2] = {

  88.   128, 16

  89. };

  90. /* computed from band_size_long */

  91. static uint16_t band_index_long[9][23];

  92. #include "mpegaudio_tablegen.h"

  93. /* intensity stereo coef table */

  94. static INTFLOAT is_table[2][16];

  95. static INTFLOAT is_table_lsf[2][2][16];

  96. static int32_t csa_table[8][4];

  97. static float csa_table_float[8][4];

  98. static INTFLOAT mdct_win[8][36];

  99. 

  100. /* lower 2 bits: modulo 3, higher bits: shift */

  101. static uint16_t scale_factor_modshift[64];

  102. /* [i][j]:  2^(-j/3) * FRAC_ONE * 2^(i+2) / (2^(i+2) - 1) */

  103. static int32_t scale_factor_mult[15][3];

  104. /* mult table for layer 2 group quantization */

  105. 

  106. #define SCALE_GEN(v) \

  107. { FIXR_OLD(1.0 * (v)), FIXR_OLD(0.7937005259 * (v)), FIXR_OLD(0.6299605249 * (v)) }

  108. 

  109. static const int32_t scale_factor_mult2[3][3] = {

  110.     SCALE_GEN(4.0 / 3.0), /* 3 steps */

  111.     SCALE_GEN(4.0 / 5.0), /* 5 steps */

  112.     SCALE_GEN(4.0 / 9.0), /* 9 steps */

  113. };

  114. 

  115. DECLARE_ALIGNED(16, MPA_INT, RENAME(ff_mpa_synth_window))[512+256];

  116. 

  117. /**

  118.  * Convert region offsets to region sizes and truncate

  119.  * size to big_values.

  120.  */

  121. static void ff_region_offset2size(GranuleDef *g){

  122.     int i, k, j=0;

  123.     g->region_size[2] = (576 / 2);

  124.     for(i=0;i<3;i++) {

  125.         k = FFMIN(g->region_size[i], g->big_values);

  126.         g->region_size[i] = k - j;

  127.         j = k;

  128.     }

  129. }

  130. 

  131. static void ff_init_short_region(MPADecodeContext *s, GranuleDef *g){

  132.     if (g->block_type == 2)

  133.         g->region_size[0] = (36 / 2);

  134.     else {

  135.         if (s->sample_rate_index <= 2)

  136.             g->region_size[0] = (36 / 2);

  137.         else if (s->sample_rate_index != 8)

  138.             g->region_size[0] = (54 / 2);

  139.         else

  140.             g->region_size[0] = (108 / 2);

  141.     }

  142.     g->region_size[1] = (576 / 2);

  143. }

  144. 

  145. static void ff_init_long_region(MPADecodeContext *s, GranuleDef *g, int ra1, int ra2){

  146.     int l;

  147.     g->region_size[0] =

  148.         band_index_long[s->sample_rate_index][ra1 + 1] >> 1;

  149.     /* should not overflow */

  150.     l = FFMIN(ra1 + ra2 + 2, 22);

  151.     g->region_size[1] =

  152.         band_index_long[s->sample_rate_index][l] >> 1;

  153. }

  154. 

  155. static void ff_compute_band_indexes(MPADecodeContext *s, GranuleDef *g){

  156.     if (g->block_type == 2) {

  157.         if (g->switch_point) {

  158.             /* if switched mode, we handle the 36 first samples as

  159.                 long blocks.  For 8000Hz, we handle the 48 first

  160.                 exponents as long blocks (XXX: check this!) */

  161.             if (s->sample_rate_index <= 2)

  162.                 g->long_end = 8;

  163.             else if (s->sample_rate_index != 8)

  164.                 g->long_end = 6;

  165.             else

  166.                 g->long_end = 4; /* 8000 Hz */

  167. 

  168.             g->short_start = 2 + (s->sample_rate_index != 8);

  169.         } else {

  170.             g->long_end = 0;

  171.             g->short_start = 0;

  172.         }

  173.     } else {

  174.         g->short_start = 13;

  175.         g->long_end = 22;

  176.     }

  177. }

  178. 

  179. /* layer 1 unscaling */

  180. /* n = number of bits of the mantissa minus 1 */

  181. static inline int l1_unscale(int n, int mant, int scale_factor)

  182. {

  183.     int shift, mod;

  184.     int64_t val;

  185. 

  186.     shift = scale_factor_modshift[scale_factor];

  187.     mod = shift & 3;

  188.     shift >>= 2;

  189.     val = MUL64(mant + (-1 << n) + 1, scale_factor_mult[n-1][mod]);

  190.     shift += n;

  191.     /* NOTE: at this point, 1 <= shift >= 21 + 15 */

  192.     return (int)((val + (1LL << (shift - 1))) >> shift);

  193. }

  194. 

  195. static inline int l2_unscale_group(int steps, int mant, int scale_factor)

  196. {

  197.     int shift, mod, val;

  198. 

  199.     shift = scale_factor_modshift[scale_factor];

  200.     mod = shift & 3;

  201.     shift >>= 2;

  202. 

  203.     val = (mant - (steps >> 1)) * scale_factor_mult2[steps >> 2][mod];

  204.     /* NOTE: at this point, 0 <= shift <= 21 */

  205.     if (shift > 0)

  206.         val = (val + (1 << (shift - 1))) >> shift;

  207.     return val;

  208. }

  209. 

  210. /* compute value^(4/3) * 2^(exponent/4). It normalized to FRAC_BITS */

  211. static inline int l3_unscale(int value, int exponent)

  212. {

  213.     unsigned int m;

  214.     int e;

  215. 

  216.     e = table_4_3_exp  [4*value + (exponent&3)];

  217.     m = table_4_3_value[4*value + (exponent&3)];

  218.     e -= (exponent >> 2);

  219.     assert(e>=1);

  220.     if (e > 31)

  221.         return 0;

  222.     m = (m + (1 << (e-1))) >> e;

  223. 

  224.     return m;

  225. }

  226. 

  227. /* all integer n^(4/3) computation code */

  228. #define DEV_ORDER 13

  229. 

  230. #define POW_FRAC_BITS 24

  231. #define POW_FRAC_ONE    (1 << POW_FRAC_BITS)

  232. #define POW_FIX(a)   ((int)((a) * POW_FRAC_ONE))

  233. #define POW_MULL(a,b) (((int64_t)(a) * (int64_t)(b)) >> POW_FRAC_BITS)

  234. 

  235. static int dev_4_3_coefs[DEV_ORDER];

  236. 

  237. #if 0 /* unused */

  238. static int pow_mult3[3] = {

  239.     POW_FIX(1.0),

  240.     POW_FIX(1.25992104989487316476),

  241.     POW_FIX(1.58740105196819947474),

  242. };

  243. #endif

  244. 

  245. static av_cold void int_pow_init(void)

  246. {

  247.     int i, a;

  248. 

  249.     a = POW_FIX(1.0);

  250.     for(i=0;i<DEV_ORDER;i++) {

  251.         a = POW_MULL(a, POW_FIX(4.0 / 3.0) - i * POW_FIX(1.0)) / (i + 1);

  252.         dev_4_3_coefs[i] = a;

  253.     }

  254. }

  255. 

  256. #if 0 /* unused, remove? */

  257. /* return the mantissa and the binary exponent */

  258. static int int_pow(int i, int *exp_ptr)

  259. {

  260.     int e, er, eq, j;

  261.     int a, a1;

  262. 

  263.     /* renormalize */

  264.     a = i;

  265.     e = POW_FRAC_BITS;

  266.     while (a < (1 << (POW_FRAC_BITS - 1))) {

  267.         a = a << 1;

  268.         e--;

  269.     }

  270.     a -= (1 << POW_FRAC_BITS);

  271.     a1 = 0;

  272.     for(j = DEV_ORDER - 1; j >= 0; j--)

  273.         a1 = POW_MULL(a, dev_4_3_coefs[j] + a1);

  274.     a = (1 << POW_FRAC_BITS) + a1;

  275.     /* exponent compute (exact) */

  276.     e = e * 4;

  277.     er = e % 3;

  278.     eq = e / 3;

  279.     a = POW_MULL(a, pow_mult3[er]);

  280.     while (a >= 2 * POW_FRAC_ONE) {

  281.         a = a >> 1;

  282.         eq++;

  283.     }

  284.     /* convert to float */

  285.     while (a < POW_FRAC_ONE) {

  286.         a = a << 1;

  287.         eq--;

  288.     }

  289.     /* now POW_FRAC_ONE <= a < 2 * POW_FRAC_ONE */

  290. #if POW_FRAC_BITS > FRAC_BITS

  291.     a = (a + (1 << (POW_FRAC_BITS - FRAC_BITS - 1))) >> (POW_FRAC_BITS - FRAC_BITS);

  292.     /* correct overflow */

  293.     if (a >= 2 * (1 << FRAC_BITS)) {

  294.         a = a >> 1;

  295.         eq++;

  296.     }

  297. #endif

  298.     *exp_ptr = eq;

  299.     return a;

  300. }

  301. #endif

  302. 

  303. static av_cold int decode_init(AVCodecContext * avctx)

  304. {

  305.     MPADecodeContext *s = avctx->priv_data;

  306.     static int init=0;

  307.     int i, j, k;

  308. 

  309.     s->avctx = avctx;

  310.     s->apply_window_mp3 = apply_window_mp3_c;

  311. #if HAVE_MMX

  312.     ff_mpegaudiodec_init_mmx(s);

  313. #endif

  314.     avctx->sample_fmt= OUT_FMT;

  315.     s->error_recognition= avctx->error_recognition;

  316. 

  317.     if (!init && !avctx->parse_only) {

  318.         int offset;

  319. 

  320.         /* scale factors table for layer 1/2 */

  321.         for(i=0;i<64;i++) {

  322.             int shift, mod;

  323.             /* 1.0 (i = 3) is normalized to 2 ^ FRAC_BITS */

  324.             shift = (i / 3);

  325.             mod = i % 3;

  326.             scale_factor_modshift[i] = mod | (shift << 2);

  327.         }

  328. 

  329.         /* scale factor multiply for layer 1 */

  330.         for(i=0;i<15;i++) {

  331.             int n, norm;

  332.             n = i + 2;

  333.             norm = ((INT64_C(1) << n) * FRAC_ONE) / ((1 << n) - 1);

  334.             scale_factor_mult[i][0] = MULLx(norm, FIXR(1.0          * 2.0), FRAC_BITS);

  335.             scale_factor_mult[i][1] = MULLx(norm, FIXR(0.7937005259 * 2.0), FRAC_BITS);

  336.             scale_factor_mult[i][2] = MULLx(norm, FIXR(0.6299605249 * 2.0), FRAC_BITS);

  337.             dprintf(avctx, "%d: norm=%x s=%x %x %x\n",

  338.                     i, norm,

  339.                     scale_factor_mult[i][0],

  340.                     scale_factor_mult[i][1],

  341.                     scale_factor_mult[i][2]);

  342.         }

  343. 

  344.         RENAME(ff_mpa_synth_init)(RENAME(ff_mpa_synth_window));

  345. 

  346.         /* huffman decode tables */

  347.         offset = 0;

  348.         for(i=1;i<16;i++) {

  349.             const HuffTable *h = &mpa_huff_tables[i];

  350.             int xsize, x, y;

  351.             uint8_t  tmp_bits [512];

  352.             uint16_t tmp_codes[512];

  353. 

  354.             memset(tmp_bits , 0, sizeof(tmp_bits ));

  355.             memset(tmp_codes, 0, sizeof(tmp_codes));

  356. 

  357.             xsize = h->xsize;

  358. 

  359.             j = 0;

  360.             for(x=0;x<xsize;x++) {

  361.                 for(y=0;y<xsize;y++){

  362.                     tmp_bits [(x << 5) | y | ((x&&y)<<4)]= h->bits [j  ];

  363.                     tmp_codes[(x << 5) | y | ((x&&y)<<4)]= h->codes[j++];

  364.                 }

  365.             }

  366. 

  367.             /* XXX: fail test */

  368.             huff_vlc[i].table = huff_vlc_tables+offset;

  369.             huff_vlc[i].table_allocated = huff_vlc_tables_sizes[i];

  370.             init_vlc(&huff_vlc[i], 7, 512,

  371.                      tmp_bits, 1, 1, tmp_codes, 2, 2,

  372.                      INIT_VLC_USE_NEW_STATIC);

  373.             offset += huff_vlc_tables_sizes[i];

  374.         }

  375.         assert(offset == FF_ARRAY_ELEMS(huff_vlc_tables));

  376. 

  377.         offset = 0;

  378.         for(i=0;i<2;i++) {

  379.             huff_quad_vlc[i].table = huff_quad_vlc_tables+offset;

  380.             huff_quad_vlc[i].table_allocated = huff_quad_vlc_tables_sizes[i];

  381.             init_vlc(&huff_quad_vlc[i], i == 0 ? 7 : 4, 16,

  382.                      mpa_quad_bits[i], 1, 1, mpa_quad_codes[i], 1, 1,

  383.                      INIT_VLC_USE_NEW_STATIC);

  384.             offset += huff_quad_vlc_tables_sizes[i];

  385.         }

  386.         assert(offset == FF_ARRAY_ELEMS(huff_quad_vlc_tables));

  387. 

  388.         for(i=0;i<9;i++) {

  389.             k = 0;

  390.             for(j=0;j<22;j++) {

  391.                 band_index_long[i][j] = k;

  392.                 k += band_size_long[i][j];

  393.             }

  394.             band_index_long[i][22] = k;

  395.         }

  396. 

  397.         /* compute n ^ (4/3) and store it in mantissa/exp format */

  398. 

  399.         int_pow_init();

  400.         mpegaudio_tableinit();

  401. 

  402.         for(i=0;i<7;i++) {

  403.             float f;

  404.             INTFLOAT v;

  405.             if (i != 6) {

  406.                 f = tan((double)i * M_PI / 12.0);

  407.                 v = FIXR(f / (1.0 + f));

  408.             } else {

  409.                 v = FIXR(1.0);

  410.             }

  411.             is_table[0][i] = v;

  412.             is_table[1][6 - i] = v;

  413.         }

  414.         /* invalid values */

  415.         for(i=7;i<16;i++)

  416.             is_table[0][i] = is_table[1][i] = 0.0;

  417. 

  418.         for(i=0;i<16;i++) {

  419.             double f;

  420.             int e, k;

  421. 

  422.             for(j=0;j<2;j++) {

  423.                 e = -(j + 1) * ((i + 1) >> 1);

  424.                 f = pow(2.0, e / 4.0);

  425.                 k = i & 1;

  426.                 is_table_lsf[j][k ^ 1][i] = FIXR(f);

  427.                 is_table_lsf[j][k][i] = FIXR(1.0);

  428.                 dprintf(avctx, "is_table_lsf %d %d: %x %x\n",

  429.                         i, j, is_table_lsf[j][0][i], is_table_lsf[j][1][i]);

  430.             }

  431.         }

  432. 

  433.         for(i=0;i<8;i++) {

  434.             float ci, cs, ca;

  435.             ci = ci_table[i];

  436.             cs = 1.0 / sqrt(1.0 + ci * ci);

  437.             ca = cs * ci;

  438.             csa_table[i][0] = FIXHR(cs/4);

  439.             csa_table[i][1] = FIXHR(ca/4);

  440.             csa_table[i][2] = FIXHR(ca/4) + FIXHR(cs/4);

  441.             csa_table[i][3] = FIXHR(ca/4) - FIXHR(cs/4);

  442.             csa_table_float[i][0] = cs;

  443.             csa_table_float[i][1] = ca;

  444.             csa_table_float[i][2] = ca + cs;

  445.             csa_table_float[i][3] = ca - cs;

  446.         }

  447. 

  448.         /* compute mdct windows */

  449.         for(i=0;i<36;i++) {

  450.             for(j=0; j<4; j++){

  451.                 double d;

  452. 

  453.                 if(j==2 && i%3 != 1)

  454.                     continue;

  455. 

  456.                 d= sin(M_PI * (i + 0.5) / 36.0);

  457.                 if(j==1){

  458.                     if     (i>=30) d= 0;

  459.                     else if(i>=24) d= sin(M_PI * (i - 18 + 0.5) / 12.0);

  460.                     else if(i>=18) d= 1;

  461.                 }else if(j==3){

  462.                     if     (i<  6) d= 0;

  463.                     else if(i< 12) d= sin(M_PI * (i -  6 + 0.5) / 12.0);

  464.                     else if(i< 18) d= 1;

  465.                 }

  466.                 //merge last stage of imdct into the window coefficients

  467.                 d*= 0.5 / cos(M_PI*(2*i + 19)/72);

  468. 

  469.                 if(j==2)

  470.                     mdct_win[j][i/3] = FIXHR((d / (1<<5)));

  471.                 else

  472.                     mdct_win[j][i  ] = FIXHR((d / (1<<5)));

  473.             }

  474.         }

  475. 

  476.         /* NOTE: we do frequency inversion adter the MDCT by changing

  477.            the sign of the right window coefs */

  478.         for(j=0;j<4;j++) {

  479.             for(i=0;i<36;i+=2) {

  480.                 mdct_win[j + 4][i] = mdct_win[j][i];

  481.                 mdct_win[j + 4][i + 1] = -mdct_win[j][i + 1];

  482.             }

  483.         }

  484. 

  485.         init = 1;

  486.     }

  487. 

  488.     if (avctx->codec_id == CODEC_ID_MP3ADU)

  489.         s->adu_mode = 1;

  490.     return 0;

  491. }

  492. 

  493. /* tab[i][j] = 1.0 / (2.0 * cos(pi*(2*k+1) / 2^(6 - j))) */

  494. 

  495. /* cos(i*pi/64) */

  496. 

  497. #define COS0_0  FIXHR(0.50060299823519630134/2)

  498. #define COS0_1  FIXHR(0.50547095989754365998/2)

  499. #define COS0_2  FIXHR(0.51544730992262454697/2)

  500. #define COS0_3  FIXHR(0.53104259108978417447/2)

  501. #define COS0_4  FIXHR(0.55310389603444452782/2)

  502. #define COS0_5  FIXHR(0.58293496820613387367/2)

  503. #define COS0_6  FIXHR(0.62250412303566481615/2)

  504. #define COS0_7  FIXHR(0.67480834145500574602/2)

  505. #define COS0_8  FIXHR(0.74453627100229844977/2)

  506. #define COS0_9  FIXHR(0.83934964541552703873/2)

  507. #define COS0_10 FIXHR(0.97256823786196069369/2)

  508. #define COS0_11 FIXHR(1.16943993343288495515/4)

  509. #define COS0_12 FIXHR(1.48416461631416627724/4)

  510. #define COS0_13 FIXHR(2.05778100995341155085/8)

  511. #define COS0_14 FIXHR(3.40760841846871878570/8)

  512. #define COS0_15 FIXHR(10.19000812354805681150/32)

  513. 

  514. #define COS1_0 FIXHR(0.50241928618815570551/2)

  515. #define COS1_1 FIXHR(0.52249861493968888062/2)

  516. #define COS1_2 FIXHR(0.56694403481635770368/2)

  517. #define COS1_3 FIXHR(0.64682178335999012954/2)

  518. #define COS1_4 FIXHR(0.78815462345125022473/2)

  519. #define COS1_5 FIXHR(1.06067768599034747134/4)

  520. #define COS1_6 FIXHR(1.72244709823833392782/4)

  521. #define COS1_7 FIXHR(5.10114861868916385802/16)

  522. 

  523. #define COS2_0 FIXHR(0.50979557910415916894/2)

  524. #define COS2_1 FIXHR(0.60134488693504528054/2)

  525. #define COS2_2 FIXHR(0.89997622313641570463/2)

  526. #define COS2_3 FIXHR(2.56291544774150617881/8)

  527. 

  528. #define COS3_0 FIXHR(0.54119610014619698439/2)

  529. #define COS3_1 FIXHR(1.30656296487637652785/4)

  530. 

  531. #define COS4_0 FIXHR(0.70710678118654752439/2)

  532. 

  533. /* butterfly operator */

  534. #define BF(a, b, c, s)\

  535. {\

  536.     tmp0 = val##a + val##b;\

  537.     tmp1 = val##a - val##b;\

  538.     val##a = tmp0;\

  539.     val##b = MULH3(tmp1, c, 1<<(s));\

  540. }

  541. 

  542. #define BF0(a, b, c, s)\

  543. {\

  544.     tmp0 = tab[a] + tab[b];\

  545.     tmp1 = tab[a] - tab[b];\

  546.     val##a = tmp0;\

  547.     val##b = MULH3(tmp1, c, 1<<(s));\

  548. }

  549. 

  550. #define BF1(a, b, c, d)\

  551. {\

  552.     BF(a, b, COS4_0, 1);\

  553.     BF(c, d,-COS4_0, 1);\

  554.     val##c += val##d;\

  555. }

  556. 

  557. #define BF2(a, b, c, d)\

  558. {\

  559.     BF(a, b, COS4_0, 1);\

  560.     BF(c, d,-COS4_0, 1);\

  561.     val##c += val##d;\

  562.     val##a += val##c;\

  563.     val##c += val##b;\

  564.     val##b += val##d;\

  565. }

  566. 

  567. #define ADD(a, b) val##a += val##b

  568. 

  569. /* DCT32 without 1/sqrt(2) coef zero scaling. */

  570. static void dct32(INTFLOAT *out, const INTFLOAT *tab)

  571. {

  572.     INTFLOAT tmp0, tmp1;

  573. 

  574.     INTFLOAT val0 , val1 , val2 , val3 , val4 , val5 , val6 , val7 ,

  575.              val8 , val9 , val10, val11, val12, val13, val14, val15,

  576.              val16, val17, val18, val19, val20, val21, val22, val23,

  577.              val24, val25, val26, val27, val28, val29, val30, val31;

  578. 

  579.     /* pass 1 */

  580.     BF0( 0, 31, COS0_0 , 1);

  581.     BF0(15, 16, COS0_15, 5);

  582.     /* pass 2 */

  583.     BF( 0, 15, COS1_0 , 1);

  584.     BF(16, 31,-COS1_0 , 1);

  585.     /* pass 1 */

  586.     BF0( 7, 24, COS0_7 , 1);

  587.     BF0( 8, 23, COS0_8 , 1);

  588.     /* pass 2 */

  589.     BF( 7,  8, COS1_7 , 4);

  590.     BF(23, 24,-COS1_7 , 4);

  591.     /* pass 3 */

  592.     BF( 0,  7, COS2_0 , 1);

  593.     BF( 8, 15,-COS2_0 , 1);

  594.     BF(16, 23, COS2_0 , 1);

  595.     BF(24, 31,-COS2_0 , 1);

  596.     /* pass 1 */

  597.     BF0( 3, 28, COS0_3 , 1);

  598.     BF0(12, 19, COS0_12, 2);

  599.     /* pass 2 */

  600.     BF( 3, 12, COS1_3 , 1);

  601.     BF(19, 28,-COS1_3 , 1);

  602.     /* pass 1 */

  603.     BF0( 4, 27, COS0_4 , 1);

  604.     BF0(11, 20, COS0_11, 2);

  605.     /* pass 2 */

  606.     BF( 4, 11, COS1_4 , 1);

  607.     BF(20, 27,-COS1_4 , 1);

  608.     /* pass 3 */

  609.     BF( 3,  4, COS2_3 , 3);

  610.     BF(11, 12,-COS2_3 , 3);

  611.     BF(19, 20, COS2_3 , 3);

  612.     BF(27, 28,-COS2_3 , 3);

  613.     /* pass 4 */

  614.     BF( 0,  3, COS3_0 , 1);

  615.     BF( 4,  7,-COS3_0 , 1);

  616.     BF( 8, 11, COS3_0 , 1);

  617.     BF(12, 15,-COS3_0 , 1);

  618.     BF(16, 19, COS3_0 , 1);

  619.     BF(20, 23,-COS3_0 , 1);

  620.     BF(24, 27, COS3_0 , 1);

  621.     BF(28, 31,-COS3_0 , 1);

  622. 

  623. 

  624. 

  625.     /* pass 1 */

  626.     BF0( 1, 30, COS0_1 , 1);

  627.     BF0(14, 17, COS0_14, 3);

  628.     /* pass 2 */

  629.     BF( 1, 14, COS1_1 , 1);

  630.     BF(17, 30,-COS1_1 , 1);

  631.     /* pass 1 */

  632.     BF0( 6, 25, COS0_6 , 1);

  633.     BF0( 9, 22, COS0_9 , 1);

  634.     /* pass 2 */

  635.     BF( 6,  9, COS1_6 , 2);

  636.     BF(22, 25,-COS1_6 , 2);

  637.     /* pass 3 */

  638.     BF( 1,  6, COS2_1 , 1);

  639.     BF( 9, 14,-COS2_1 , 1);

  640.     BF(17, 22, COS2_1 , 1);

  641.     BF(25, 30,-COS2_1 , 1);

  642. 

  643.     /* pass 1 */

  644.     BF0( 2, 29, COS0_2 , 1);

  645.     BF0(13, 18, COS0_13, 3);

  646.     /* pass 2 */

  647.     BF( 2, 13, COS1_2 , 1);

  648.     BF(18, 29,-COS1_2 , 1);

  649.     /* pass 1 */

  650.     BF0( 5, 26, COS0_5 , 1);

  651.     BF0(10, 21, COS0_10, 1);

  652.     /* pass 2 */

  653.     BF( 5, 10, COS1_5 , 2);

  654.     BF(21, 26,-COS1_5 , 2);

  655.     /* pass 3 */

  656.     BF( 2,  5, COS2_2 , 1);

  657.     BF(10, 13,-COS2_2 , 1);

  658.     BF(18, 21, COS2_2 , 1);

  659.     BF(26, 29,-COS2_2 , 1);

  660.     /* pass 4 */

  661.     BF( 1,  2, COS3_1 , 2);

  662.     BF( 5,  6,-COS3_1 , 2);

  663.     BF( 9, 10, COS3_1 , 2);

  664.     BF(13, 14,-COS3_1 , 2);

  665.     BF(17, 18, COS3_1 , 2);

  666.     BF(21, 22,-COS3_1 , 2);

  667.     BF(25, 26, COS3_1 , 2);

  668.     BF(29, 30,-COS3_1 , 2);

  669. 

  670.     /* pass 5 */

  671.     BF1( 0,  1,  2,  3);

  672.     BF2( 4,  5,  6,  7);

  673.     BF1( 8,  9, 10, 11);

  674.     BF2(12, 13, 14, 15);

  675.     BF1(16, 17, 18, 19);

  676.     BF2(20, 21, 22, 23);

  677.     BF1(24, 25, 26, 27);

  678.     BF2(28, 29, 30, 31);

  679. 

  680.     /* pass 6 */

  681. 

  682.     ADD( 8, 12);

  683.     ADD(12, 10);

  684.     ADD(10, 14);

  685.     ADD(14,  9);

  686.     ADD( 9, 13);

  687.     ADD(13, 11);

  688.     ADD(11, 15);

  689. 

  690.     out[ 0] = val0;

  691.     out[16] = val1;

  692.     out[ 8] = val2;

  693.     out[24] = val3;

  694.     out[ 4] = val4;

  695.     out[20] = val5;

  696.     out[12] = val6;

  697.     out[28] = val7;

  698.     out[ 2] = val8;

  699.     out[18] = val9;

  700.     out[10] = val10;

  701.     out[26] = val11;

  702.     out[ 6] = val12;

  703.     out[22] = val13;

  704.     out[14] = val14;

  705.     out[30] = val15;

  706. 

  707.     ADD(24, 28);

  708.     ADD(28, 26);

  709.     ADD(26, 30);

  710.     ADD(30, 25);

  711.     ADD(25, 29);

  712.     ADD(29, 27);

  713.     ADD(27, 31);

  714. 

  715.     out[ 1] = val16 + val24;

  716.     out[17] = val17 + val25;

  717.     out[ 9] = val18 + val26;

  718.     out[25] = val19 + val27;

  719.     out[ 5] = val20 + val28;

  720.     out[21] = val21 + val29;

  721.     out[13] = val22 + val30;

  722.     out[29] = val23 + val31;

  723.     out[ 3] = val24 + val20;

  724.     out[19] = val25 + val21;

  725.     out[11] = val26 + val22;

  726.     out[27] = val27 + val23;

  727.     out[ 7] = val28 + val18;

  728.     out[23] = val29 + val19;

  729.     out[15] = val30 + val17;

  730.     out[31] = val31;

  731. }

  732. 

  733. #if CONFIG_FLOAT

  734. static inline float round_sample(float *sum)

  735. {

  736.     float sum1=*sum;

  737.     *sum = 0;

  738.     return sum1;

  739. }

  740. 

  741. /* signed 16x16 -> 32 multiply add accumulate */

  742. #define MACS(rt, ra, rb) rt+=(ra)*(rb)

  743. 

  744. /* signed 16x16 -> 32 multiply */

  745. #define MULS(ra, rb) ((ra)*(rb))

  746. 

  747. #define MLSS(rt, ra, rb) rt-=(ra)*(rb)

  748. 

  749. #elif FRAC_BITS <= 15

  750. 

  751. static inline int round_sample(int *sum)

  752. {

  753.     int sum1;

  754.     sum1 = (*sum) >> OUT_SHIFT;

  755.     *sum &= (1<<OUT_SHIFT)-1;

  756.     return av_clip(sum1, OUT_MIN, OUT_MAX);

  757. }

  758. 

  759. /* signed 16x16 -> 32 multiply add accumulate */

  760. #define MACS(rt, ra, rb) MAC16(rt, ra, rb)

  761. 

  762. /* signed 16x16 -> 32 multiply */

  763. #define MULS(ra, rb) MUL16(ra, rb)

  764. 

  765. #define MLSS(rt, ra, rb) MLS16(rt, ra, rb)

  766. 

  767. #else

  768. 

  769. static inline int round_sample(int64_t *sum)

  770. {

  771.     int sum1;

  772.     sum1 = (int)((*sum) >> OUT_SHIFT);

  773.     *sum &= (1<<OUT_SHIFT)-1;

  774.     return av_clip(sum1, OUT_MIN, OUT_MAX);

  775. }

  776. 

  777. #   define MULS(ra, rb) MUL64(ra, rb)

  778. #   define MACS(rt, ra, rb) MAC64(rt, ra, rb)

  779. #   define MLSS(rt, ra, rb) MLS64(rt, ra, rb)

  780. #endif

  781. 

  782. #define SUM8(op, sum, w, p)               \

  783. {                                         \

  784.     op(sum, (w)[0 * 64], (p)[0 * 64]);    \

  785.     op(sum, (w)[1 * 64], (p)[1 * 64]);    \

  786.     op(sum, (w)[2 * 64], (p)[2 * 64]);    \

  787.     op(sum, (w)[3 * 64], (p)[3 * 64]);    \

  788.     op(sum, (w)[4 * 64], (p)[4 * 64]);    \

  789.     op(sum, (w)[5 * 64], (p)[5 * 64]);    \

  790.     op(sum, (w)[6 * 64], (p)[6 * 64]);    \

  791.     op(sum, (w)[7 * 64], (p)[7 * 64]);    \

  792. }

  793. 

  794. #define SUM8P2(sum1, op1, sum2, op2, w1, w2, p) \

  795. {                                               \

  796.     INTFLOAT tmp;\

  797.     tmp = p[0 * 64];\

  798.     op1(sum1, (w1)[0 * 64], tmp);\

  799.     op2(sum2, (w2)[0 * 64], tmp);\

  800.     tmp = p[1 * 64];\

  801.     op1(sum1, (w1)[1 * 64], tmp);\

  802.     op2(sum2, (w2)[1 * 64], tmp);\

  803.     tmp = p[2 * 64];\

  804.     op1(sum1, (w1)[2 * 64], tmp);\

  805.     op2(sum2, (w2)[2 * 64], tmp);\

  806.     tmp = p[3 * 64];\

  807.     op1(sum1, (w1)[3 * 64], tmp);\

  808.     op2(sum2, (w2)[3 * 64], tmp);\

  809.     tmp = p[4 * 64];\

  810.     op1(sum1, (w1)[4 * 64], tmp);\

  811.     op2(sum2, (w2)[4 * 64], tmp);\

  812.     tmp = p[5 * 64];\

  813.     op1(sum1, (w1)[5 * 64], tmp);\

  814.     op2(sum2, (w2)[5 * 64], tmp);\

  815.     tmp = p[6 * 64];\

  816.     op1(sum1, (w1)[6 * 64], tmp);\

  817.     op2(sum2, (w2)[6 * 64], tmp);\

  818.     tmp = p[7 * 64];\

  819.     op1(sum1, (w1)[7 * 64], tmp);\

  820.     op2(sum2, (w2)[7 * 64], tmp);\

  821. }

  822. 

  823. void av_cold RENAME(ff_mpa_synth_init)(MPA_INT *window)

  824. {

  825.     int i, j;

  826. 

  827.     /* max = 18760, max sum over all 16 coefs : 44736 */

  828.     for(i=0;i<257;i++) {

  829.         INTFLOAT v;

  830.         v = ff_mpa_enwindow[i];

  831. #if CONFIG_FLOAT

  832.         v *= 1.0 / (1LL<<(16 + FRAC_BITS));

  833. #elif WFRAC_BITS < 16

  834.         v = (v + (1 << (16 - WFRAC_BITS - 1))) >> (16 - WFRAC_BITS);

  835. #endif

  836.         window[i] = v;

  837.         if ((i & 63) != 0)

  838.             v = -v;

  839.         if (i != 0)

  840.             window[512 - i] = v;

  841.     }

  842. 

  843.     // Needed for avoiding shuffles in ASM implementations

  844.     for(i=0; i < 8; i++)

  845.         for(j=0; j < 16; j++)

  846.             window[512+16*i+j] = window[64*i+32-j];

  847. 

  848.     for(i=0; i < 8; i++)

  849.         for(j=0; j < 16; j++)

  850.             window[512+128+16*i+j] = window[64*i+48-j];

  851. }

  852. 

  853. static void apply_window_mp3_c(MPA_INT *synth_buf, MPA_INT *window,

  854.                                int *dither_state, OUT_INT *samples, int incr)

  855. {

  856.     register const MPA_INT *w, *w2, *p;

  857.     int j;

  858.     OUT_INT *samples2;

  859. #if CONFIG_FLOAT

  860.     float sum, sum2;

  861. #elif FRAC_BITS <= 15

  862.     int sum, sum2;

  863. #else

  864.     int64_t sum, sum2;

  865. #endif

  866. 

  867.     /* copy to avoid wrap */

  868.     memcpy(synth_buf + 512, synth_buf, 32 * sizeof(*synth_buf));

  869. 

  870.     samples2 = samples + 31 * incr;

  871.     w = window;

  872.     w2 = window + 31;

  873. 

  874.     sum = *dither_state;

  875.     p = synth_buf + 16;

  876.     SUM8(MACS, sum, w, p);

  877.     p = synth_buf + 48;

  878.     SUM8(MLSS, sum, w + 32, p);

  879.     *samples = round_sample(&sum);

  880.     samples += incr;

  881.     w++;

  882. 

  883.     /* we calculate two samples at the same time to avoid one memory

  884.        access per two sample */

  885.     for(j=1;j<16;j++) {

  886.         sum2 = 0;

  887.         p = synth_buf + 16 + j;

  888.         SUM8P2(sum, MACS, sum2, MLSS, w, w2, p);

  889.         p = synth_buf + 48 - j;

  890.         SUM8P2(sum, MLSS, sum2, MLSS, w + 32, w2 + 32, p);

  891. 

  892.         *samples = round_sample(&sum);

  893.         samples += incr;

  894.         sum += sum2;

  895.         *samples2 = round_sample(&sum);

  896.         samples2 -= incr;

  897.         w++;

  898.         w2--;

  899.     }

  900. 

  901.     p = synth_buf + 32;

  902.     SUM8(MLSS, sum, w + 32, p);

  903.     *samples = round_sample(&sum);

  904.     *dither_state= sum;

  905. }

  906. 

  907. 

  908. /* 32 sub band synthesis filter. Input: 32 sub band samples, Output:

  909.    32 samples. */

  910. /* XXX: optimize by avoiding ring buffer usage */

  911. #if !CONFIG_FLOAT

  912. void ff_mpa_synth_filter(MPA_INT *synth_buf_ptr, int *synth_buf_offset,

  913.                          MPA_INT *window, int *dither_state,

  914.                          OUT_INT *samples, int incr,

  915.                          INTFLOAT sb_samples[SBLIMIT])

  916. {

  917.     register MPA_INT *synth_buf;

  918.     int offset;

  919. #if FRAC_BITS <= 15

  920.     int32_t tmp[32];

  921.     int j;

  922. #endif

  923. 

  924.     offset = *synth_buf_offset;

  925.     synth_buf = synth_buf_ptr + offset;

  926. 

  927. #if FRAC_BITS <= 15

  928.     dct32(tmp, sb_samples);

  929.     for(j=0;j<32;j++) {

  930.         /* NOTE: can cause a loss in precision if very high amplitude

  931.            sound */

  932.         synth_buf[j] = av_clip_int16(tmp[j]);

  933.     }

  934. #else

  935.     dct32(synth_buf, sb_samples);

  936. #endif

  937. 

  938.     apply_window_mp3_c(synth_buf, window, dither_state, samples, incr);

  939. 

  940.     offset = (offset - 32) & 511;

  941.     *synth_buf_offset = offset;

  942. }

  943. #endif

  944. 

  945. #define C3 FIXHR(0.86602540378443864676/2)

  946. 

  947. /* 0.5 / cos(pi*(2*i+1)/36) */

  948. static const INTFLOAT icos36[9] = {

  949.     FIXR(0.50190991877167369479),

  950.     FIXR(0.51763809020504152469), //0

  951.     FIXR(0.55168895948124587824),

  952.     FIXR(0.61038729438072803416),

  953.     FIXR(0.70710678118654752439), //1

  954.     FIXR(0.87172339781054900991),

  955.     FIXR(1.18310079157624925896),

  956.     FIXR(1.93185165257813657349), //2

  957.     FIXR(5.73685662283492756461),

  958. };

  959. 

  960. /* 0.5 / cos(pi*(2*i+1)/36) */

  961. static const INTFLOAT icos36h[9] = {

  962.     FIXHR(0.50190991877167369479/2),

  963.     FIXHR(0.51763809020504152469/2), //0

  964.     FIXHR(0.55168895948124587824/2),

  965.     FIXHR(0.61038729438072803416/2),

  966.     FIXHR(0.70710678118654752439/2), //1

  967.     FIXHR(0.87172339781054900991/2),

  968.     FIXHR(1.18310079157624925896/4),

  969.     FIXHR(1.93185165257813657349/4), //2

  970. //    FIXHR(5.73685662283492756461),

  971. };

  972. 

  973. /* 12 points IMDCT. We compute it "by hand" by factorizing obvious

  974.    cases. */

  975. static void imdct12(INTFLOAT *out, INTFLOAT *in)

  976. {

  977.     INTFLOAT in0, in1, in2, in3, in4, in5, t1, t2;

  978. 

  979.     in0= in[0*3];

  980.     in1= in[1*3] + in[0*3];

  981.     in2= in[2*3] + in[1*3];

  982.     in3= in[3*3] + in[2*3];

  983.     in4= in[4*3] + in[3*3];

  984.     in5= in[5*3] + in[4*3];

  985.     in5 += in3;

  986.     in3 += in1;

  987. 

  988.     in2= MULH3(in2, C3, 2);

  989.     in3= MULH3(in3, C3, 4);

  990. 

  991.     t1 = in0 - in4;

  992.     t2 = MULH3(in1 - in5, icos36h[4], 2);

  993. 

  994.     out[ 7]=

  995.     out[10]= t1 + t2;

  996.     out[ 1]=

  997.     out[ 4]= t1 - t2;

  998. 

  999.     in0 += SHR(in4, 1);

  1000.     in4 = in0 + in2;

  1001.     in5 += 2*in1;

  1002.     in1 = MULH3(in5 + in3, icos36h[1], 1);

  1003.     out[ 8]=

  1004.     out[ 9]= in4 + in1;

  1005.     out[ 2]=

  1006.     out[ 3]= in4 - in1;

  1007. 

  1008.     in0 -= in2;

  1009.     in5 = MULH3(in5 - in3, icos36h[7], 2);

  1010.     out[ 0]=

  1011.     out[ 5]= in0 - in5;

  1012.     out[ 6]=

  1013.     out[11]= in0 + in5;

  1014. }

  1015. 

  1016. /* cos(pi*i/18) */

  1017. #define C1 FIXHR(0.98480775301220805936/2)

  1018. #define C2 FIXHR(0.93969262078590838405/2)

  1019. #define C3 FIXHR(0.86602540378443864676/2)

  1020. #define C4 FIXHR(0.76604444311897803520/2)

  1021. #define C5 FIXHR(0.64278760968653932632/2)

  1022. #define C6 FIXHR(0.5/2)

  1023. #define C7 FIXHR(0.34202014332566873304/2)

  1024. #define C8 FIXHR(0.17364817766693034885/2)

  1025. 

  1026. 

  1027. /* using Lee like decomposition followed by hand coded 9 points DCT */

  1028. static void imdct36(INTFLOAT *out, INTFLOAT *buf, INTFLOAT *in, INTFLOAT *win)

  1029. {

  1030.     int i, j;

  1031.     INTFLOAT t0, t1, t2, t3, s0, s1, s2, s3;

  1032.     INTFLOAT tmp[18], *tmp1, *in1;

  1033. 

  1034.     for(i=17;i>=1;i--)

  1035.         in[i] += in[i-1];

  1036.     for(i=17;i>=3;i-=2)

  1037.         in[i] += in[i-2];

  1038. 

  1039.     for(j=0;j<2;j++) {

  1040.         tmp1 = tmp + j;

  1041.         in1 = in + j;

  1042. 

  1043.         t2 = in1[2*4] + in1[2*8] - in1[2*2];

  1044. 

  1045.         t3 = in1[2*0] + SHR(in1[2*6],1);

  1046.         t1 = in1[2*0] - in1[2*6];

  1047.         tmp1[ 6] = t1 - SHR(t2,1);

  1048.         tmp1[16] = t1 + t2;

  1049. 

  1050.         t0 = MULH3(in1[2*2] + in1[2*4] ,    C2, 2);

  1051.         t1 = MULH3(in1[2*4] - in1[2*8] , -2*C8, 1);

  1052.         t2 = MULH3(in1[2*2] + in1[2*8] ,   -C4, 2);

  1053. 

  1054.         tmp1[10] = t3 - t0 - t2;

  1055.         tmp1[ 2] = t3 + t0 + t1;

  1056.         tmp1[14] = t3 + t2 - t1;

  1057. 

  1058.         tmp1[ 4] = MULH3(in1[2*5] + in1[2*7] - in1[2*1], -C3, 2);

  1059.         t2 = MULH3(in1[2*1] + in1[2*5],    C1, 2);

  1060.         t3 = MULH3(in1[2*5] - in1[2*7], -2*C7, 1);

  1061.         t0 = MULH3(in1[2*3], C3, 2);

  1062. 

  1063.         t1 = MULH3(in1[2*1] + in1[2*7],   -C5, 2);

  1064. 

  1065.         tmp1[ 0] = t2 + t3 + t0;

  1066.         tmp1[12] = t2 + t1 - t0;

  1067.         tmp1[ 8] = t3 - t1 - t0;

  1068.     }

  1069. 

  1070.     i = 0;

  1071.     for(j=0;j<4;j++) {

  1072.         t0 = tmp[i];

  1073.         t1 = tmp[i + 2];

  1074.         s0 = t1 + t0;

  1075.         s2 = t1 - t0;

  1076. 

  1077.         t2 = tmp[i + 1];

  1078.         t3 = tmp[i + 3];

  1079.         s1 = MULH3(t3 + t2, icos36h[j], 2);

  1080.         s3 = MULLx(t3 - t2, icos36[8 - j], FRAC_BITS);

  1081. 

  1082.         t0 = s0 + s1;

  1083.         t1 = s0 - s1;

  1084.         out[(9 + j)*SBLIMIT] =  MULH3(t1, win[9 + j], 1) + buf[9 + j];

  1085.         out[(8 - j)*SBLIMIT] =  MULH3(t1, win[8 - j], 1) + buf[8 - j];

  1086.         buf[9 + j] = MULH3(t0, win[18 + 9 + j], 1);

  1087.         buf[8 - j] = MULH3(t0, win[18 + 8 - j], 1);

  1088. 

  1089.         t0 = s2 + s3;

  1090.         t1 = s2 - s3;

  1091.         out[(9 + 8 - j)*SBLIMIT] =  MULH3(t1, win[9 + 8 - j], 1) + buf[9 + 8 - j];

  1092.         out[(        j)*SBLIMIT] =  MULH3(t1, win[        j], 1) + buf[        j];

  1093.         buf[9 + 8 - j] = MULH3(t0, win[18 + 9 + 8 - j], 1);

  1094.         buf[      + j] = MULH3(t0, win[18         + j], 1);

  1095.         i += 4;

  1096.     }

  1097. 

  1098.     s0 = tmp[16];

  1099.     s1 = MULH3(tmp[17], icos36h[4], 2);

  1100.     t0 = s0 + s1;

  1101.     t1 = s0 - s1;

  1102.     out[(9 + 4)*SBLIMIT] =  MULH3(t1, win[9 + 4], 1) + buf[9 + 4];

  1103.     out[(8 - 4)*SBLIMIT] =  MULH3(t1, win[8 - 4], 1) + buf[8 - 4];

  1104.     buf[9 + 4] = MULH3(t0, win[18 + 9 + 4], 1);

  1105.     buf[8 - 4] = MULH3(t0, win[18 + 8 - 4], 1);

  1106. }

  1107. 

  1108. /* return the number of decoded frames */

  1109. static int mp_decode_layer1(MPADecodeContext *s)

  1110. {

  1111.     int bound, i, v, n, ch, j, mant;

  1112.     uint8_t allocation[MPA_MAX_CHANNELS][SBLIMIT];

  1113.     uint8_t scale_factors[MPA_MAX_CHANNELS][SBLIMIT];

  1114. 

  1115.     if (s->mode == MPA_JSTEREO)

  1116.         bound = (s->mode_ext + 1) * 4;

  1117.     else

  1118.         bound = SBLIMIT;

  1119. 

  1120.     /* allocation bits */

  1121.     for(i=0;i<bound;i++) {

  1122.         for(ch=0;ch<s->nb_channels;ch++) {

  1123.             allocation[ch][i] = get_bits(&s->gb, 4);

  1124.         }

  1125.     }

  1126.     for(i=bound;i<SBLIMIT;i++) {

  1127.         allocation[0][i] = get_bits(&s->gb, 4);

  1128.     }

  1129. 

  1130.     /* scale factors */

  1131.     for(i=0;i<bound;i++) {

  1132.         for(ch=0;ch<s->nb_channels;ch++) {

  1133.             if (allocation[ch][i])

  1134.                 scale_factors[ch][i] = get_bits(&s->gb, 6);

  1135.         }

  1136.     }

  1137.     for(i=bound;i<SBLIMIT;i++) {

  1138.         if (allocation[0][i]) {

  1139.             scale_factors[0][i] = get_bits(&s->gb, 6);

  1140.             scale_factors[1][i] = get_bits(&s->gb, 6);

  1141.         }

  1142.     }

  1143. 

  1144.     /* compute samples */

  1145.     for(j=0;j<12;j++) {

  1146.         for(i=0;i<bound;i++) {

  1147.             for(ch=0;ch<s->nb_channels;ch++) {

  1148.                 n = allocation[ch][i];

  1149.                 if (n) {

  1150.                     mant = get_bits(&s->gb, n + 1);

  1151.                     v = l1_unscale(n, mant, scale_factors[ch][i]);

  1152.                 } else {

  1153.                     v = 0;

  1154.                 }

  1155.                 s->sb_samples[ch][j][i] = v;

  1156.             }

  1157.         }

  1158.         for(i=bound;i<SBLIMIT;i++) {

  1159.             n = allocation[0][i];

  1160.             if (n) {

  1161.                 mant = get_bits(&s->gb, n + 1);

  1162.                 v = l1_unscale(n, mant, scale_factors[0][i]);

  1163.                 s->sb_samples[0][j][i] = v;

  1164.                 v = l1_unscale(n, mant, scale_factors[1][i]);

  1165.                 s->sb_samples[1][j][i] = v;

  1166.             } else {

  1167.                 s->sb_samples[0][j][i] = 0;

  1168.                 s->sb_samples[1][j][i] = 0;

  1169.             }

  1170.         }

  1171.     }

  1172.     return 12;

  1173. }

  1174. 

  1175. static int mp_decode_layer2(MPADecodeContext *s)

  1176. {

  1177.     int sblimit; /* number of used subbands */

  1178.     const unsigned char *alloc_table;

  1179.     int table, bit_alloc_bits, i, j, ch, bound, v;

  1180.     unsigned char bit_alloc[MPA_MAX_CHANNELS][SBLIMIT];

  1181.     unsigned char scale_code[MPA_MAX_CHANNELS][SBLIMIT];

  1182.     unsigned char scale_factors[MPA_MAX_CHANNELS][SBLIMIT][3], *sf;

  1183.     int scale, qindex, bits, steps, k, l, m, b;

  1184. 

  1185.     /* select decoding table */

  1186.     table = ff_mpa_l2_select_table(s->bit_rate / 1000, s->nb_channels,

  1187.                             s->sample_rate, s->lsf);

  1188.     sblimit = ff_mpa_sblimit_table[table];

  1189.     alloc_table = ff_mpa_alloc_tables[table];

  1190. 

  1191.     if (s->mode == MPA_JSTEREO)

  1192.         bound = (s->mode_ext + 1) * 4;

  1193.     else

  1194.         bound = sblimit;

  1195. 

  1196.     dprintf(s->avctx, "bound=%d sblimit=%d\n", bound, sblimit);

  1197. 

  1198.     /* sanity check */

  1199.     if( bound > sblimit ) bound = sblimit;

  1200. 

  1201.     /* parse bit allocation */

  1202.     j = 0;

  1203.     for(i=0;i<bound;i++) {

  1204.         bit_alloc_bits = alloc_table[j];

  1205.         for(ch=0;ch<s->nb_channels;ch++) {

  1206.             bit_alloc[ch][i] = get_bits(&s->gb, bit_alloc_bits);

  1207.         }

  1208.         j += 1 << bit_alloc_bits;

  1209.     }

  1210.     for(i=bound;i<sblimit;i++) {

  1211.         bit_alloc_bits = alloc_table[j];

  1212.         v = get_bits(&s->gb, bit_alloc_bits);

  1213.         bit_alloc[0][i] = v;

  1214.         bit_alloc[1][i] = v;

  1215.         j += 1 << bit_alloc_bits;

  1216.     }

  1217. 

  1218.     /* scale codes */

  1219.     for(i=0;i<sblimit;i++) {

  1220.         for(ch=0;ch<s->nb_channels;ch++) {

  1221.             if (bit_alloc[ch][i])

  1222.                 scale_code[ch][i] = get_bits(&s->gb, 2);

  1223.         }

  1224.     }

  1225. 

  1226.     /* scale factors */

  1227.     for(i=0;i<sblimit;i++) {

  1228.         for(ch=0;ch<s->nb_channels;ch++) {

  1229.             if (bit_alloc[ch][i]) {

  1230.                 sf = scale_factors[ch][i];

  1231.                 switch(scale_code[ch][i]) {

  1232.                 default:

  1233.                 case 0:

  1234.                     sf[0] = get_bits(&s->gb, 6);

  1235.                     sf[1] = get_bits(&s->gb, 6);

  1236.                     sf[2] = get_bits(&s->gb, 6);

  1237.                     break;

  1238.                 case 2:

  1239.                     sf[0] = get_bits(&s->gb, 6);

  1240.                     sf[1] = sf[0];

  1241.                     sf[2] = sf[0];

  1242.                     break;

  1243.                 case 1:

  1244.                     sf[0] = get_bits(&s->gb, 6);

  1245.                     sf[2] = get_bits(&s->gb, 6);

  1246.                     sf[1] = sf[0];

  1247.                     break;

  1248.                 case 3:

  1249.                     sf[0] = get_bits(&s->gb, 6);

  1250.                     sf[2] = get_bits(&s->gb, 6);

  1251.                     sf[1] = sf[2];

  1252.                     break;

  1253.                 }

  1254.             }

  1255.         }

  1256.     }

  1257. 

  1258.     /* samples */

  1259.     for(k=0;k<3;k++) {

  1260.         for(l=0;l<12;l+=3) {

  1261.             j = 0;

  1262.             for(i=0;i<bound;i++) {

  1263.                 bit_alloc_bits = alloc_table[j];

  1264.                 for(ch=0;ch<s->nb_channels;ch++) {

  1265.                     b = bit_alloc[ch][i];

  1266.                     if (b) {

  1267.                         scale = scale_factors[ch][i][k];

  1268.                         qindex = alloc_table[j+b];

  1269.                         bits = ff_mpa_quant_bits[qindex];

  1270.                         if (bits < 0) {

  1271.                             /* 3 values at the same time */

  1272.                             v = get_bits(&s->gb, -bits);

  1273.                             steps = ff_mpa_quant_steps[qindex];

  1274.                             s->sb_samples[ch][k * 12 + l + 0][i] =

  1275.                                 l2_unscale_group(steps, v % steps, scale);

  1276.                             v = v / steps;

  1277.                             s->sb_samples[ch][k * 12 + l + 1][i] =

  1278.                                 l2_unscale_group(steps, v % steps, scale);

  1279.                             v = v / steps;

  1280.                             s->sb_samples[ch][k * 12 + l + 2][i] =

  1281.                                 l2_unscale_group(steps, v, scale);

  1282.                         } else {

  1283.                             for(m=0;m<3;m++) {

  1284.                                 v = get_bits(&s->gb, bits);

  1285.                                 v = l1_unscale(bits - 1, v, scale);

  1286.                                 s->sb_samples[ch][k * 12 + l + m][i] = v;

  1287.                             }

  1288.                         }

  1289.                     } else {

  1290.                         s->sb_samples[ch][k * 12 + l + 0][i] = 0;

  1291.                         s->sb_samples[ch][k * 12 + l + 1][i] = 0;

  1292.                         s->sb_samples[ch][k * 12 + l + 2][i] = 0;

  1293.                     }

  1294.                 }

  1295.                 /* next subband in alloc table */

  1296.                 j += 1 << bit_alloc_bits;

  1297.             }

  1298.             /* XXX: find a way to avoid this duplication of code */

  1299.             for(i=bound;i<sblimit;i++) {

  1300.                 bit_alloc_bits = alloc_table[j];

  1301.                 b = bit_alloc[0][i];

  1302.                 if (b) {

  1303.                     int mant, scale0, scale1;

  1304.                     scale0 = scale_factors[0][i][k];

  1305.                     scale1 = scale_factors[1][i][k];

  1306.                     qindex = alloc_table[j+b];

  1307.                     bits = ff_mpa_quant_bits[qindex];

  1308.                     if (bits < 0) {

  1309.                         /* 3 values at the same time */

  1310.                         v = get_bits(&s->gb, -bits);

  1311.                         steps = ff_mpa_quant_steps[qindex];

  1312.                         mant = v % steps;

  1313.                         v = v / steps;

  1314.                         s->sb_samples[0][k * 12 + l + 0][i] =

  1315.                             l2_unscale_group(steps, mant, scale0);

  1316.                         s->sb_samples[1][k * 12 + l + 0][i] =

  1317.                             l2_unscale_group(steps, mant, scale1);

  1318.                         mant = v % steps;

  1319.                         v = v / steps;

  1320.                         s->sb_samples[0][k * 12 + l + 1][i] =

  1321.                             l2_unscale_group(steps, mant, scale0);

  1322.                         s->sb_samples[1][k * 12 + l + 1][i] =

  1323.                             l2_unscale_group(steps, mant, scale1);

  1324.                         s->sb_samples[0][k * 12 + l + 2][i] =

  1325.                             l2_unscale_group(steps, v, scale0);

  1326.                         s->sb_samples[1][k * 12 + l + 2][i] =

  1327.                             l2_unscale_group(steps, v, scale1);

  1328.                     } else {

  1329.                         for(m=0;m<3;m++) {

  1330.                             mant = get_bits(&s->gb, bits);

  1331.                             s->sb_samples[0][k * 12 + l + m][i] =

  1332.                                 l1_unscale(bits - 1, mant, scale0);

  1333.                             s->sb_samples[1][k * 12 + l + m][i] =

  1334.                                 l1_unscale(bits - 1, mant, scale1);

  1335.                         }

  1336.                     }

  1337.                 } else {

  1338.                     s->sb_samples[0][k * 12 + l + 0][i] = 0;

  1339.                     s->sb_samples[0][k * 12 + l + 1][i] = 0;

  1340.                     s->sb_samples[0][k * 12 + l + 2][i] = 0;

  1341.                     s->sb_samples[1][k * 12 + l + 0][i] = 0;

  1342.                     s->sb_samples[1][k * 12 + l + 1][i] = 0;

  1343.                     s->sb_samples[1][k * 12 + l + 2][i] = 0;

  1344.                 }

  1345.                 /* next subband in alloc table */

  1346.                 j += 1 << bit_alloc_bits;

  1347.             }

  1348.             /* fill remaining samples to zero */

  1349.             for(i=sblimit;i<SBLIMIT;i++) {

  1350.                 for(ch=0;ch<s->nb_channels;ch++) {

  1351.                     s->sb_samples[ch][k * 12 + l + 0][i] = 0;

  1352.                     s->sb_samples[ch][k * 12 + l + 1][i] = 0;

  1353.                     s->sb_samples[ch][k * 12 + l + 2][i] = 0;

  1354.                 }

  1355.             }

  1356.         }

  1357.     }

  1358.     return 3 * 12;

  1359. }

  1360. 

  1361. #define SPLIT(dst,sf,n)\

  1362.     if(n==3){\

  1363.         int m= (sf*171)>>9;\

  1364.         dst= sf - 3*m;\

  1365.         sf=m;\

  1366.     }else if(n==4){\

  1367.         dst= sf&3;\

  1368.         sf>>=2;\

  1369.     }else if(n==5){\

  1370.         int m= (sf*205)>>10;\

  1371.         dst= sf - 5*m;\

  1372.         sf=m;\

  1373.     }else if(n==6){\

  1374.         int m= (sf*171)>>10;\

  1375.         dst= sf - 6*m;\

  1376.         sf=m;\

  1377.     }else{\

  1378.         dst=0;\

  1379.     }

  1380. 

  1381. static av_always_inline void lsf_sf_expand(int *slen,

  1382.                                  int sf, int n1, int n2, int n3)

  1383. {

  1384.     SPLIT(slen[3], sf, n3)

  1385.     SPLIT(slen[2], sf, n2)

  1386.     SPLIT(slen[1], sf, n1)

  1387.     slen[0] = sf;

  1388. }

  1389. 

  1390. static void exponents_from_scale_factors(MPADecodeContext *s,

  1391.                                          GranuleDef *g,

  1392.                                          int16_t *exponents)

  1393. {

  1394.     const uint8_t *bstab, *pretab;

  1395.     int len, i, j, k, l, v0, shift, gain, gains[3];

  1396.     int16_t *exp_ptr;

  1397. 

  1398.     exp_ptr = exponents;

  1399.     gain = g->global_gain - 210;

  1400.     shift = g->scalefac_scale + 1;

  1401. 

  1402.     bstab = band_size_long[s->sample_rate_index];

  1403.     pretab = mpa_pretab[g->preflag];

  1404.     for(i=0;i<g->long_end;i++) {

  1405.         v0 = gain - ((g->scale_factors[i] + pretab[i]) << shift) + 400;

  1406.         len = bstab[i];

  1407.         for(j=len;j>0;j--)

  1408.             *exp_ptr++ = v0;

  1409.     }

  1410. 

  1411.     if (g->short_start < 13) {

  1412.         bstab = band_size_short[s->sample_rate_index];

  1413.         gains[0] = gain - (g->subblock_gain[0] << 3);

  1414.         gains[1] = gain - (g->subblock_gain[1] << 3);

  1415.         gains[2] = gain - (g->subblock_gain[2] << 3);

  1416.         k = g->long_end;

  1417.         for(i=g->short_start;i<13;i++) {

  1418.             len = bstab[i];

  1419.             for(l=0;l<3;l++) {

  1420.                 v0 = gains[l] - (g->scale_factors[k++] << shift) + 400;

  1421.                 for(j=len;j>0;j--)

  1422.                 *exp_ptr++ = v0;

  1423.             }

  1424.         }

  1425.     }

  1426. }

  1427. 

  1428. /* handle n = 0 too */

  1429. static inline int get_bitsz(GetBitContext *s, int n)

  1430. {

  1431.     if (n == 0)

  1432.         return 0;

  1433.     else

  1434.         return get_bits(s, n);

  1435. }

  1436. 

  1437. 

  1438. static void switch_buffer(MPADecodeContext *s, int *pos, int *end_pos, int *end_pos2){

  1439.     if(s->in_gb.buffer && *pos >= s->gb.size_in_bits){

  1440.         s->gb= s->in_gb;

  1441.         s->in_gb.buffer=NULL;

  1442.         assert((get_bits_count(&s->gb) & 7) == 0);

  1443.         skip_bits_long(&s->gb, *pos - *end_pos);

  1444.         *end_pos2=

  1445.         *end_pos= *end_pos2 + get_bits_count(&s->gb) - *pos;

  1446.         *pos= get_bits_count(&s->gb);

  1447.     }

  1448. }

  1449. 

  1450. /* Following is a optimized code for

  1451.             INTFLOAT v = *src

  1452.             if(get_bits1(&s->gb))

  1453.                 v = -v;

  1454.             *dst = v;

  1455. */

  1456. #if CONFIG_FLOAT

  1457. #define READ_FLIP_SIGN(dst,src)\

  1458.             v = AV_RN32A(src) ^ (get_bits1(&s->gb)<<31);\

  1459.             AV_WN32A(dst, v);

  1460. #else

  1461. #define READ_FLIP_SIGN(dst,src)\

  1462.             v= -get_bits1(&s->gb);\

  1463.             *(dst) = (*(src) ^ v) - v;

  1464. #endif

  1465. 

  1466. static int huffman_decode(MPADecodeContext *s, GranuleDef *g,

  1467.                           int16_t *exponents, int end_pos2)

  1468. {

  1469.     int s_index;

  1470.     int i;

  1471.     int last_pos, bits_left;

  1472.     VLC *vlc;

  1473.     int end_pos= FFMIN(end_pos2, s->gb.size_in_bits);

  1474. 

  1475.     /* low frequencies (called big values) */

  1476.     s_index = 0;

  1477.     for(i=0;i<3;i++) {

  1478.         int j, k, l, linbits;

  1479.         j = g->region_size[i];

  1480.         if (j == 0)

  1481.             continue;

  1482.         /* select vlc table */

  1483.         k = g->table_select[i];

  1484.         l = mpa_huff_data[k][0];

  1485.         linbits = mpa_huff_data[k][1];

  1486.         vlc = &huff_vlc[l];

  1487. 

  1488.         if(!l){

  1489.             memset(&g->sb_hybrid[s_index], 0, sizeof(*g->sb_hybrid)*2*j);

  1490.             s_index += 2*j;

  1491.             continue;

  1492.         }

  1493. 

  1494.         /* read huffcode and compute each couple */

  1495.         for(;j>0;j--) {

  1496.             int exponent, x, y;

  1497.             int v;

  1498.             int pos= get_bits_count(&s->gb);

  1499. 

  1500.             if (pos >= end_pos){

  1501. //                av_log(NULL, AV_LOG_ERROR, "pos: %d %d %d %d\n", pos, end_pos, end_pos2, s_index);

  1502.                 switch_buffer(s, &pos, &end_pos, &end_pos2);

  1503. //                av_log(NULL, AV_LOG_ERROR, "new pos: %d %d\n", pos, end_pos);

  1504.                 if(pos >= end_pos)

  1505.                     break;

  1506.             }

  1507.             y = get_vlc2(&s->gb, vlc->table, 7, 3);

  1508. 

  1509.             if(!y){

  1510.                 g->sb_hybrid[s_index  ] =

  1511.                 g->sb_hybrid[s_index+1] = 0;

  1512.                 s_index += 2;

  1513.                 continue;

  1514.             }

  1515. 

  1516.             exponent= exponents[s_index];

  1517. 

  1518.             dprintf(s->avctx, "region=%d n=%d x=%d y=%d exp=%d\n",

  1519.                     i, g->region_size[i] - j, x, y, exponent);

  1520.             if(y&16){

  1521.                 x = y >> 5;

  1522.                 y = y & 0x0f;

  1523.                 if (x < 15){

  1524.                     READ_FLIP_SIGN(g->sb_hybrid+s_index, RENAME(expval_table)[ exponent ]+x)

  1525.                 }else{

  1526.                     x += get_bitsz(&s->gb, linbits);

  1527.                     v = l3_unscale(x, exponent);

  1528.                     if (get_bits1(&s->gb))

  1529.                         v = -v;

  1530.                     g->sb_hybrid[s_index] = v;

  1531.                 }

  1532.                 if (y < 15){

  1533.                     READ_FLIP_SIGN(g->sb_hybrid+s_index+1, RENAME(expval_table)[ exponent ]+y)

  1534.                 }else{

  1535.                     y += get_bitsz(&s->gb, linbits);

  1536.                     v = l3_unscale(y, exponent);

  1537.                     if (get_bits1(&s->gb))

  1538.                         v = -v;

  1539.                     g->sb_hybrid[s_index+1] = v;

  1540.                 }

  1541.             }else{

  1542.                 x = y >> 5;

  1543.                 y = y & 0x0f;

  1544.                 x += y;

  1545.                 if (x < 15){

  1546.                     READ_FLIP_SIGN(g->sb_hybrid+s_index+!!y, RENAME(expval_table)[ exponent ]+x)

  1547.                 }else{

  1548.                     x += get_bitsz(&s->gb, linbits);

  1549.                     v = l3_unscale(x, exponent);

  1550.                     if (get_bits1(&s->gb))

  1551.                         v = -v;

  1552.                     g->sb_hybrid[s_index+!!y] = v;

  1553.                 }

  1554.                 g->sb_hybrid[s_index+ !y] = 0;

  1555.             }

  1556.             s_index+=2;

  1557.         }

  1558.     }

  1559. 

  1560.     /* high frequencies */

  1561.     vlc = &huff_quad_vlc[g->count1table_select];

  1562.     last_pos=0;

  1563.     while (s_index <= 572) {

  1564.         int pos, code;

  1565.         pos = get_bits_count(&s->gb);

  1566.         if (pos >= end_pos) {

  1567.             if (pos > end_pos2 && last_pos){

  1568.                 /* some encoders generate an incorrect size for this

  1569.                    part. We must go back into the data */

  1570.                 s_index -= 4;

  1571.                 skip_bits_long(&s->gb, last_pos - pos);

  1572.                 av_log(s->avctx, AV_LOG_INFO, "overread, skip %d enddists: %d %d\n", last_pos - pos, end_pos-pos, end_pos2-pos);

  1573.                 if(s->error_recognition >= FF_ER_COMPLIANT)

  1574.                     s_index=0;

  1575.                 break;

  1576.             }

  1577. //                av_log(NULL, AV_LOG_ERROR, "pos2: %d %d %d %d\n", pos, end_pos, end_pos2, s_index);

  1578.             switch_buffer(s, &pos, &end_pos, &end_pos2);

  1579. //                av_log(NULL, AV_LOG_ERROR, "new pos2: %d %d %d\n", pos, end_pos, s_index);

  1580.             if(pos >= end_pos)

  1581.                 break;

  1582.         }

  1583.         last_pos= pos;

  1584. 

  1585.         code = get_vlc2(&s->gb, vlc->table, vlc->bits, 1);

  1586.         dprintf(s->avctx, "t=%d code=%d\n", g->count1table_select, code);

  1587.         g->sb_hybrid[s_index+0]=

  1588.         g->sb_hybrid[s_index+1]=

  1589.         g->sb_hybrid[s_index+2]=

  1590.         g->sb_hybrid[s_index+3]= 0;

  1591.         while(code){

  1592.             static const int idxtab[16]={3,3,2,2,1,1,1,1,0,0,0,0,0,0,0,0};

  1593.             int v;

  1594.             int pos= s_index+idxtab[code];

  1595.             code ^= 8>>idxtab[code];

  1596.             READ_FLIP_SIGN(g->sb_hybrid+pos, RENAME(exp_table)+exponents[pos])

  1597.         }

  1598.         s_index+=4;

  1599.     }

  1600.     /* skip extension bits */

  1601.     bits_left = end_pos2 - get_bits_count(&s->gb);

  1602. //av_log(NULL, AV_LOG_ERROR, "left:%d buf:%p\n", bits_left, s->in_gb.buffer);

  1603.     if (bits_left < 0 && s->error_recognition >= FF_ER_COMPLIANT) {

  1604.         av_log(s->avctx, AV_LOG_ERROR, "bits_left=%d\n", bits_left);

  1605.         s_index=0;

  1606.     }else if(bits_left > 0 && s->error_recognition >= FF_ER_AGGRESSIVE){

  1607.         av_log(s->avctx, AV_LOG_ERROR, "bits_left=%d\n", bits_left);

  1608.         s_index=0;

  1609.     }

  1610.     memset(&g->sb_hybrid[s_index], 0, sizeof(*g->sb_hybrid)*(576 - s_index));

  1611.     skip_bits_long(&s->gb, bits_left);

  1612. 

  1613.     i= get_bits_count(&s->gb);

  1614.     switch_buffer(s, &i, &end_pos, &end_pos2);

  1615. 

  1616.     return 0;

  1617. }

  1618. 

  1619. /* Reorder short blocks from bitstream order to interleaved order. It

  1620.    would be faster to do it in parsing, but the code would be far more

  1621.    complicated */

  1622. static void reorder_block(MPADecodeContext *s, GranuleDef *g)

  1623. {

  1624.     int i, j, len;

  1625.     INTFLOAT *ptr, *dst, *ptr1;

  1626.     INTFLOAT tmp[576];

  1627. 

  1628.     if (g->block_type != 2)

  1629.         return;

  1630. 

  1631.     if (g->switch_point) {

  1632.         if (s->sample_rate_index != 8) {

  1633.             ptr = g->sb_hybrid + 36;

  1634.         } else {

  1635.             ptr = g->sb_hybrid + 48;

  1636.         }

  1637.     } else {

  1638.         ptr = g->sb_hybrid;

  1639.     }

  1640. 

  1641.     for(i=g->short_start;i<13;i++) {

  1642.         len = band_size_short[s->sample_rate_index][i];

  1643.         ptr1 = ptr;

  1644.         dst = tmp;

  1645.         for(j=len;j>0;j--) {

  1646.             *dst++ = ptr[0*len];

  1647.             *dst++ = ptr[1*len];

  1648.             *dst++ = ptr[2*len];

  1649.             ptr++;

  1650.         }

  1651.         ptr+=2*len;

  1652.         memcpy(ptr1, tmp, len * 3 * sizeof(*ptr1));

  1653.     }

  1654. }

  1655. 

  1656. #define ISQRT2 FIXR(0.70710678118654752440)

  1657. 

  1658. static void compute_stereo(MPADecodeContext *s,

  1659.                            GranuleDef *g0, GranuleDef *g1)

  1660. {

  1661.     int i, j, k, l;

  1662.     int sf_max, sf, len, non_zero_found;

  1663.     INTFLOAT (*is_tab)[16], *tab0, *tab1, tmp0, tmp1, v1, v2;

  1664.     int non_zero_found_short[3];

  1665. 

  1666.     /* intensity stereo */

  1667.     if (s->mode_ext & MODE_EXT_I_STEREO) {

  1668.         if (!s->lsf) {

  1669.             is_tab = is_table;

  1670.             sf_max = 7;

  1671.         } else {

  1672.             is_tab = is_table_lsf[g1->scalefac_compress & 1];

  1673.             sf_max = 16;

  1674.         }

  1675. 

  1676.         tab0 = g0->sb_hybrid + 576;

  1677.         tab1 = g1->sb_hybrid + 576;

  1678. 

  1679.         non_zero_found_short[0] = 0;

  1680.         non_zero_found_short[1] = 0;

  1681.         non_zero_found_short[2] = 0;

  1682.         k = (13 - g1->short_start) * 3 + g1->long_end - 3;

  1683.         for(i = 12;i >= g1->short_start;i--) {

  1684.             /* for last band, use previous scale factor */

  1685.             if (i != 11)

  1686.                 k -= 3;

  1687.             len = band_size_short[s->sample_rate_index][i];

  1688.             for(l=2;l>=0;l--) {

  1689.                 tab0 -= len;

  1690.                 tab1 -= len;

  1691.                 if (!non_zero_found_short[l]) {

  1692.                     /* test if non zero band. if so, stop doing i-stereo */

  1693.                     for(j=0;j<len;j++) {

  1694.                         if (tab1[j] != 0) {

  1695.                             non_zero_found_short[l] = 1;

  1696.                             goto found1;

  1697.                         }

  1698.                     }

  1699.                     sf = g1->scale_factors[k + l];

  1700.                     if (sf >= sf_max)

  1701.                         goto found1;

  1702. 

  1703.                     v1 = is_tab[0][sf];

  1704.                     v2 = is_tab[1][sf];

  1705.                     for(j=0;j<len;j++) {

  1706.                         tmp0 = tab0[j];

  1707.                         tab0[j] = MULLx(tmp0, v1, FRAC_BITS);

  1708.                         tab1[j] = MULLx(tmp0, v2, FRAC_BITS);

  1709.                     }

  1710.                 } else {

  1711.                 found1:

  1712.                     if (s->mode_ext & MODE_EXT_MS_STEREO) {

  1713.                         /* lower part of the spectrum : do ms stereo

  1714.                            if enabled */

  1715.                         for(j=0;j<len;j++) {

  1716.                             tmp0 = tab0[j];

  1717.                             tmp1 = tab1[j];

  1718.                             tab0[j] = MULLx(tmp0 + tmp1, ISQRT2, FRAC_BITS);

  1719.                             tab1[j] = MULLx(tmp0 - tmp1, ISQRT2, FRAC_BITS);

  1720.                         }

  1721.                     }

  1722.                 }

  1723.             }

  1724.         }

  1725. 

  1726.         non_zero_found = non_zero_found_short[0] |

  1727.             non_zero_found_short[1] |

  1728.             non_zero_found_short[2];

  1729. 

  1730.         for(i = g1->long_end - 1;i >= 0;i--) {

  1731.             len = band_size_long[s->sample_rate_index][i];

  1732.             tab0 -= len;

  1733.             tab1 -= len;

  1734.             /* test if non zero band. if so, stop doing i-stereo */

  1735.             if (!non_zero_found) {

  1736.                 for(j=0;j<len;j++) {

  1737.                     if (tab1[j] != 0) {

  1738.                         non_zero_found = 1;

  1739.                         goto found2;

  1740.                     }

  1741.                 }

  1742.                 /* for last band, use previous scale factor */

  1743.                 k = (i == 21) ? 20 : i;

  1744.                 sf = g1->scale_factors[k];

  1745.                 if (sf >= sf_max)

  1746.                     goto found2;

  1747.                 v1 = is_tab[0][sf];

  1748.                 v2 = is_tab[1][sf];

  1749.                 for(j=0;j<len;j++) {

  1750.                     tmp0 = tab0[j];

  1751.                     tab0[j] = MULLx(tmp0, v1, FRAC_BITS);

  1752.                     tab1[j] = MULLx(tmp0, v2, FRAC_BITS);

  1753.                 }

  1754.             } else {

  1755.             found2:

  1756.                 if (s->mode_ext & MODE_EXT_MS_STEREO) {

  1757.                     /* lower part of the spectrum : do ms stereo

  1758.                        if enabled */

  1759.                     for(j=0;j<len;j++) {

  1760.                         tmp0 = tab0[j];

  1761.                         tmp1 = tab1[j];

  1762.                         tab0[j] = MULLx(tmp0 + tmp1, ISQRT2, FRAC_BITS);

  1763.                         tab1[j] = MULLx(tmp0 - tmp1, ISQRT2, FRAC_BITS);

  1764.                     }

  1765.                 }

  1766.             }

  1767.         }

  1768.     } else if (s->mode_ext & MODE_EXT_MS_STEREO) {

  1769.         /* ms stereo ONLY */

  1770.         /* NOTE: the 1/sqrt(2) normalization factor is included in the

  1771.            global gain */

  1772.         tab0 = g0->sb_hybrid;

  1773.         tab1 = g1->sb_hybrid;

  1774.         for(i=0;i<576;i++) {

  1775.             tmp0 = tab0[i];

  1776.             tmp1 = tab1[i];

  1777.             tab0[i] = tmp0 + tmp1;

  1778.             tab1[i] = tmp0 - tmp1;

  1779.         }

  1780.     }

  1781. }

  1782. 

  1783. static void compute_antialias_integer(MPADecodeContext *s,

  1784.                               GranuleDef *g)

  1785. {

  1786.     int32_t *ptr, *csa;

  1787.     int n, i;

  1788. 

  1789.     /* we antialias only "long" bands */

  1790.     if (g->block_type == 2) {

  1791.         if (!g->switch_point)

  1792.             return;

  1793.         /* XXX: check this for 8000Hz case */

  1794.         n = 1;

  1795.     } else {

  1796.         n = SBLIMIT - 1;

  1797.     }

  1798. 

  1799.     ptr = g->sb_hybrid + 18;

  1800.     for(i = n;i > 0;i--) {

  1801.         int tmp0, tmp1, tmp2;

  1802.         csa = &csa_table[0][0];

  1803. #define INT_AA(j) \

  1804.             tmp0 = ptr[-1-j];\

  1805.             tmp1 = ptr[   j];\

  1806.             tmp2= MULH(tmp0 + tmp1, csa[0+4*j]);\

  1807.             ptr[-1-j] = 4*(tmp2 - MULH(tmp1, csa[2+4*j]));\

  1808.             ptr[   j] = 4*(tmp2 + MULH(tmp0, csa[3+4*j]));

  1809. 

  1810.         INT_AA(0)

  1811.         INT_AA(1)

  1812.         INT_AA(2)

  1813.         INT_AA(3)

  1814.         INT_AA(4)

  1815.         INT_AA(5)

  1816.         INT_AA(6)

  1817.         INT_AA(7)

  1818. 

  1819.         ptr += 18;

  1820.     }

  1821. }

  1822. 

  1823. static void compute_antialias_float(MPADecodeContext *s,

  1824.                               GranuleDef *g)

  1825. {

  1826.     float *ptr;

  1827.     int n, i;

  1828. 

  1829.     /* we antialias only "long" bands */

  1830.     if (g->block_type == 2) {

  1831.         if (!g->switch_point)

  1832.             return;

  1833.         /* XXX: check this for 8000Hz case */

  1834.         n = 1;

  1835.     } else {

  1836.         n = SBLIMIT - 1;

  1837.     }

  1838. 

  1839.     ptr = g->sb_hybrid + 18;

  1840.     for(i = n;i > 0;i--) {

  1841.         float tmp0, tmp1;

  1842.         float *csa = &csa_table_float[0][0];

  1843. #define FLOAT_AA(j)\

  1844.         tmp0= ptr[-1-j];\

  1845.         tmp1= ptr[   j];\

  1846.         ptr[-1-j] = tmp0 * csa[0+4*j] - tmp1 * csa[1+4*j];\

  1847.         ptr[   j] = tmp0 * csa[1+4*j] + tmp1 * csa[0+4*j];

  1848. 

  1849.         FLOAT_AA(0)

  1850.         FLOAT_AA(1)

  1851.         FLOAT_AA(2)

  1852.         FLOAT_AA(3)

  1853.         FLOAT_AA(4)

  1854.         FLOAT_AA(5)

  1855.         FLOAT_AA(6)

  1856.         FLOAT_AA(7)

  1857. 

  1858.         ptr += 18;

  1859.     }

  1860. }

  1861. 

  1862. static void compute_imdct(MPADecodeContext *s,

  1863.                           GranuleDef *g,

  1864.                           INTFLOAT *sb_samples,

  1865.                           INTFLOAT *mdct_buf)

  1866. {

  1867.     INTFLOAT *win, *win1, *out_ptr, *ptr, *buf, *ptr1;

  1868.     INTFLOAT out2[12];

  1869.     int i, j, mdct_long_end, sblimit;

  1870. 

  1871.     /* find last non zero block */

  1872.     ptr = g->sb_hybrid + 576;

  1873.     ptr1 = g->sb_hybrid + 2 * 18;

  1874.     while (ptr >= ptr1) {

  1875.         int32_t *p;

  1876.         ptr -= 6;

  1877.         p= (int32_t*)ptr;

  1878.         if(p[0] | p[1] | p[2] | p[3] | p[4] | p[5])

  1879.             break;

  1880.     }

  1881.     sblimit = ((ptr - g->sb_hybrid) / 18) + 1;

  1882. 

  1883.     if (g->block_type == 2) {

  1884.         /* XXX: check for 8000 Hz */

  1885.         if (g->switch_point)

  1886.             mdct_long_end = 2;

  1887.         else

  1888.             mdct_long_end = 0;

  1889.     } else {

  1890.         mdct_long_end = sblimit;

  1891.     }

  1892. 

  1893.     buf = mdct_buf;

  1894.     ptr = g->sb_hybrid;

  1895.     for(j=0;j<mdct_long_end;j++) {

  1896.         /* apply window & overlap with previous buffer */

  1897.         out_ptr = sb_samples + j;

  1898.         /* select window */

  1899.         if (g->switch_point && j < 2)

  1900.             win1 = mdct_win[0];

  1901.         else

  1902.             win1 = mdct_win[g->block_type];

  1903.         /* select frequency inversion */

  1904.         win = win1 + ((4 * 36) & -(j & 1));

  1905.         imdct36(out_ptr, buf, ptr, win);

  1906.         out_ptr += 18*SBLIMIT;

  1907.         ptr += 18;

  1908.         buf += 18;

  1909.     }

  1910.     for(j=mdct_long_end;j<sblimit;j++) {

  1911.         /* select frequency inversion */

  1912.         win = mdct_win[2] + ((4 * 36) & -(j & 1));

  1913.         out_ptr = sb_samples + j;

  1914. 

  1915.         for(i=0; i<6; i++){

  1916.             *out_ptr = buf[i];

  1917.             out_ptr += SBLIMIT;

  1918.         }

  1919.         imdct12(out2, ptr + 0);

  1920.         for(i=0;i<6;i++) {

  1921.             *out_ptr     = MULH3(out2[i    ], win[i    ], 1) + buf[i + 6*1];

  1922.             buf[i + 6*2] = MULH3(out2[i + 6], win[i + 6], 1);

  1923.             out_ptr += SBLIMIT;

  1924.         }

  1925.         imdct12(out2, ptr + 1);

  1926.         for(i=0;i<6;i++) {

  1927.             *out_ptr     = MULH3(out2[i    ], win[i    ], 1) + buf[i + 6*2];

  1928.             buf[i + 6*0] = MULH3(out2[i + 6], win[i + 6], 1);

  1929.             out_ptr += SBLIMIT;

  1930.         }

  1931.         imdct12(out2, ptr + 2);

  1932.         for(i=0;i<6;i++) {

  1933.             buf[i + 6*0] = MULH3(out2[i    ], win[i    ], 1) + buf[i + 6*0];

  1934.             buf[i + 6*1] = MULH3(out2[i + 6], win[i + 6], 1);

  1935.             buf[i + 6*2] = 0;

  1936.         }

  1937.         ptr += 18;

  1938.         buf += 18;

  1939.     }

  1940.     /* zero bands */

  1941.     for(j=sblimit;j<SBLIMIT;j++) {

  1942.         /* overlap */

  1943.         out_ptr = sb_samples + j;

  1944.         for(i=0;i<18;i++) {

  1945.             *out_ptr = buf[i];

  1946.             buf[i] = 0;

  1947.             out_ptr += SBLIMIT;

  1948.         }

  1949.         buf += 18;

  1950.     }

  1951. }

  1952. 

  1953. /* main layer3 decoding function */

  1954. static int mp_decode_layer3(MPADecodeContext *s)

  1955. {

  1956.     int nb_granules, main_data_begin, private_bits;

  1957.     int gr, ch, blocksplit_flag, i, j, k, n, bits_pos;

  1958.     GranuleDef *g;

  1959.     int16_t exponents[576]; //FIXME try INTFLOAT

  1960. 

  1961.     /* read side info */

  1962.     if (s->lsf) {

  1963.         main_data_begin = get_bits(&s->gb, 8);

  1964.         private_bits = get_bits(&s->gb, s->nb_channels);

  1965.         nb_granules = 1;

  1966.     } else {

  1967.         main_data_begin = get_bits(&s->gb, 9);

  1968.         if (s->nb_channels == 2)

  1969.             private_bits = get_bits(&s->gb, 3);

  1970.         else

  1971.             private_bits = get_bits(&s->gb, 5);

  1972.         nb_granules = 2;

  1973.         for(ch=0;ch<s->nb_channels;ch++) {

  1974.             s->granules[ch][0].scfsi = 0;/* all scale factors are transmitted */

  1975.             s->granules[ch][1].scfsi = get_bits(&s->gb, 4);

  1976.         }

  1977.     }

  1978. 

  1979.     for(gr=0;gr<nb_granules;gr++) {

  1980.         for(ch=0;ch<s->nb_channels;ch++) {

  1981.             dprintf(s->avctx, "gr=%d ch=%d: side_info\n", gr, ch);

  1982.             g = &s->granules[ch][gr];

  1983.             g->part2_3_length = get_bits(&s->gb, 12);

  1984.             g->big_values = get_bits(&s->gb, 9);

  1985.             if(g->big_values > 288){

  1986.                 av_log(s->avctx, AV_LOG_ERROR, "big_values too big\n");

  1987.                 return -1;

  1988.             }

  1989. 

  1990.             g->global_gain = get_bits(&s->gb, 8);

  1991.             /* if MS stereo only is selected, we precompute the

  1992.                1/sqrt(2) renormalization factor */

  1993.             if ((s->mode_ext & (MODE_EXT_MS_STEREO | MODE_EXT_I_STEREO)) ==

  1994.                 MODE_EXT_MS_STEREO)

  1995.                 g->global_gain -= 2;

  1996.             if (s->lsf)

  1997.                 g->scalefac_compress = get_bits(&s->gb, 9);

  1998.             else

  1999.                 g->scalefac_compress = get_bits(&s->gb, 4);

  2000.             blocksplit_flag = get_bits1(&s->gb);

  2001.             if (blocksplit_flag) {

  2002.                 g->block_type = get_bits(&s->gb, 2);

  2003.                 if (g->block_type == 0){

  2004.                     av_log(s->avctx, AV_LOG_ERROR, "invalid block type\n");

  2005.                     return -1;

  2006.                 }

  2007.                 g->switch_point = get_bits1(&s->gb);

  2008.                 for(i=0;i<2;i++)

  2009.                     g->table_select[i] = get_bits(&s->gb, 5);

  2010.                 for(i=0;i<3;i++)

  2011.                     g->subblock_gain[i] = get_bits(&s->gb, 3);

  2012.                 ff_init_short_region(s, g);

  2013.             } else {

  2014.                 int region_address1, region_address2;

  2015.                 g->block_type = 0;

  2016.                 g->switch_point = 0;

  2017.                 for(i=0;i<3;i++)

  2018.                     g->table_select[i] = get_bits(&s->gb, 5);

  2019.                 /* compute huffman coded region sizes */

  2020.                 region_address1 = get_bits(&s->gb, 4);

  2021.                 region_address2 = get_bits(&s->gb, 3);

  2022.                 dprintf(s->avctx, "region1=%d region2=%d\n",

  2023.                         region_address1, region_address2);

  2024.                 ff_init_long_region(s, g, region_address1, region_address2);

  2025.             }

  2026.             ff_region_offset2size(g);

  2027.             ff_compute_band_indexes(s, g);

  2028. 

  2029.             g->preflag = 0;

  2030.             if (!s->lsf)

  2031.                 g->preflag = get_bits1(&s->gb);

  2032.             g->scalefac_scale = get_bits1(&s->gb);

  2033.             g->count1table_select = get_bits1(&s->gb);

  2034.             dprintf(s->avctx, "block_type=%d switch_point=%d\n",

  2035.                     g->block_type, g->switch_point);

  2036.         }

  2037.     }

  2038. 

  2039.   if (!s->adu_mode) {

  2040.     const uint8_t *ptr = s->gb.buffer + (get_bits_count(&s->gb)>>3);

  2041.     assert((get_bits_count(&s->gb) & 7) == 0);

  2042.     /* now we get bits from the main_data_begin offset */

  2043.     dprintf(s->avctx, "seekback: %d\n", main_data_begin);

  2044. //av_log(NULL, AV_LOG_ERROR, "backstep:%d, lastbuf:%d\n", main_data_begin, s->last_buf_size);

  2045. 

  2046.     memcpy(s->last_buf + s->last_buf_size, ptr, EXTRABYTES);

  2047.     s->in_gb= s->gb;

  2048.         init_get_bits(&s->gb, s->last_buf, s->last_buf_size*8);

  2049.         skip_bits_long(&s->gb, 8*(s->last_buf_size - main_data_begin));

  2050.   }

  2051. 

  2052.     for(gr=0;gr<nb_granules;gr++) {

  2053.         for(ch=0;ch<s->nb_channels;ch++) {

  2054.             g = &s->granules[ch][gr];

  2055.             if(get_bits_count(&s->gb)<0){

  2056.                 av_log(s->avctx, AV_LOG_DEBUG, "mdb:%d, lastbuf:%d skipping granule %d\n",

  2057.                                             main_data_begin, s->last_buf_size, gr);

  2058.                 skip_bits_long(&s->gb, g->part2_3_length);

  2059.                 memset(g->sb_hybrid, 0, sizeof(g->sb_hybrid));

  2060.                 if(get_bits_count(&s->gb) >= s->gb.size_in_bits && s->in_gb.buffer){

  2061.                     skip_bits_long(&s->in_gb, get_bits_count(&s->gb) - s->gb.size_in_bits);

  2062.                     s->gb= s->in_gb;

  2063.                     s->in_gb.buffer=NULL;

  2064.                 }

  2065.                 continue;

  2066.             }

  2067. 

  2068.             bits_pos = get_bits_count(&s->gb);

  2069. 

  2070.             if (!s->lsf) {

  2071.                 uint8_t *sc;

  2072.                 int slen, slen1, slen2;

  2073. 

  2074.                 /* MPEG1 scale factors */

  2075.                 slen1 = slen_table[0][g->scalefac_compress];

  2076.                 slen2 = slen_table[1][g->scalefac_compress];

  2077.                 dprintf(s->avctx, "slen1=%d slen2=%d\n", slen1, slen2);

  2078.                 if (g->block_type == 2) {

  2079.                     n = g->switch_point ? 17 : 18;

  2080.                     j = 0;

  2081.                     if(slen1){

  2082.                         for(i=0;i<n;i++)

  2083.                             g->scale_factors[j++] = get_bits(&s->gb, slen1);

  2084.                     }else{

  2085.                         for(i=0;i<n;i++)

  2086.                             g->scale_factors[j++] = 0;

  2087.                     }

  2088.                     if(slen2){

  2089.                         for(i=0;i<18;i++)

  2090.                             g->scale_factors[j++] = get_bits(&s->gb, slen2);

  2091.                         for(i=0;i<3;i++)

  2092.                             g->scale_factors[j++] = 0;

  2093.                     }else{

  2094.                         for(i=0;i<21;i++)

  2095.                             g->scale_factors[j++] = 0;

  2096.                     }

  2097.                 } else {

  2098.                     sc = s->granules[ch][0].scale_factors;

  2099.                     j = 0;

  2100.                     for(k=0;k<4;k++) {

  2101.                         n = (k == 0 ? 6 : 5);

  2102.                         if ((g->scfsi & (0x8 >> k)) == 0) {

  2103.                             slen = (k < 2) ? slen1 : slen2;

  2104.                             if(slen){

  2105.                                 for(i=0;i<n;i++)

  2106.                                     g->scale_factors[j++] = get_bits(&s->gb, slen);

  2107.                             }else{

  2108.                                 for(i=0;i<n;i++)

  2109.                                     g->scale_factors[j++] = 0;

  2110.                             }

  2111.                         } else {

  2112.                             /* simply copy from last granule */

  2113.                             for(i=0;i<n;i++) {

  2114.                                 g->scale_factors[j] = sc[j];

  2115.                                 j++;

  2116.                             }

  2117.                         }

  2118.                     }

  2119.                     g->scale_factors[j++] = 0;

  2120.                 }

  2121.             } else {

  2122.                 int tindex, tindex2, slen[4], sl, sf;

  2123. 

  2124.                 /* LSF scale factors */

  2125.                 if (g->block_type == 2) {

  2126.                     tindex = g->switch_point ? 2 : 1;

  2127.                 } else {

  2128.                     tindex = 0;

  2129.                 }

  2130.                 sf = g->scalefac_compress;

  2131.                 if ((s->mode_ext & MODE_EXT_I_STEREO) && ch == 1) {

  2132.                     /* intensity stereo case */

  2133.                     sf >>= 1;

  2134.                     if (sf < 180) {

  2135.                         lsf_sf_expand(slen, sf, 6, 6, 0);

  2136.                         tindex2 = 3;

  2137.                     } else if (sf < 244) {

  2138.                         lsf_sf_expand(slen, sf - 180, 4, 4, 0);

  2139.                         tindex2 = 4;

  2140.                     } else {

  2141.                         lsf_sf_expand(slen, sf - 244, 3, 0, 0);

  2142.                         tindex2 = 5;

  2143.                     }

  2144.                 } else {

  2145.                     /* normal case */

  2146.                     if (sf < 400) {

  2147.                         lsf_sf_expand(slen, sf, 5, 4, 4);

  2148.                         tindex2 = 0;

  2149.                     } else if (sf < 500) {

  2150.                         lsf_sf_expand(slen, sf - 400, 5, 4, 0);

  2151.                         tindex2 = 1;

  2152.                     } else {

  2153.                         lsf_sf_expand(slen, sf - 500, 3, 0, 0);

  2154.                         tindex2 = 2;

  2155.                         g->preflag = 1;

  2156.                     }

  2157.                 }

  2158. 

  2159.                 j = 0;

  2160.                 for(k=0;k<4;k++) {

  2161.                     n = lsf_nsf_table[tindex2][tindex][k];

  2162.                     sl = slen[k];

  2163.                     if(sl){

  2164.                         for(i=0;i<n;i++)

  2165.                             g->scale_factors[j++] = get_bits(&s->gb, sl);

  2166.                     }else{

  2167.                         for(i=0;i<n;i++)

  2168.                             g->scale_factors[j++] = 0;

  2169.                     }

  2170.                 }

  2171.                 /* XXX: should compute exact size */

  2172.                 for(;j<40;j++)

  2173.                     g->scale_factors[j] = 0;

  2174.             }

  2175. 

  2176.             exponents_from_scale_factors(s, g, exponents);

  2177. 

  2178.             /* read Huffman coded residue */

  2179.             huffman_decode(s, g, exponents, bits_pos + g->part2_3_length);

  2180.         } /* ch */

  2181. 

  2182.         if (s->nb_channels == 2)

  2183.             compute_stereo(s, &s->granules[0][gr], &s->granules[1][gr]);

  2184. 

  2185.         for(ch=0;ch<s->nb_channels;ch++) {

  2186.             g = &s->granules[ch][gr];

  2187. 

  2188.             reorder_block(s, g);

  2189.             compute_antialias(s, g);

  2190.             compute_imdct(s, g, &s->sb_samples[ch][18 * gr][0], s->mdct_buf[ch]);

  2191.         }

  2192.     } /* gr */

  2193.     if(get_bits_count(&s->gb)<0)

  2194.         skip_bits_long(&s->gb, -get_bits_count(&s->gb));

  2195.     return nb_granules * 18;

  2196. }

  2197. 

  2198. static int mp_decode_frame(MPADecodeContext *s,

  2199.                            OUT_INT *samples, const uint8_t *buf, int buf_size)

  2200. {

  2201.     int i, nb_frames, ch;

  2202.     OUT_INT *samples_ptr;

  2203. 

  2204.     init_get_bits(&s->gb, buf + HEADER_SIZE, (buf_size - HEADER_SIZE)*8);

  2205. 

  2206.     /* skip error protection field */

  2207.     if (s->error_protection)

  2208.         skip_bits(&s->gb, 16);

  2209. 

  2210.     dprintf(s->avctx, "frame %d:\n", s->frame_count);

  2211.     switch(s->layer) {

  2212.     case 1:

  2213.         s->avctx->frame_size = 384;

  2214.         nb_frames = mp_decode_layer1(s);

  2215.         break;

  2216.     case 2:

  2217.         s->avctx->frame_size = 1152;

  2218.         nb_frames = mp_decode_layer2(s);

  2219.         break;

  2220.     case 3:

  2221.         s->avctx->frame_size = s->lsf ? 576 : 1152;

  2222.     default:

  2223.         nb_frames = mp_decode_layer3(s);

  2224. 

  2225.         s->last_buf_size=0;

  2226.         if(s->in_gb.buffer){

  2227.             align_get_bits(&s->gb);

  2228.             i= get_bits_left(&s->gb)>>3;

  2229.             if(i >= 0 && i <= BACKSTEP_SIZE){

  2230.                 memmove(s->last_buf, s->gb.buffer + (get_bits_count(&s->gb)>>3), i);

  2231.                 s->last_buf_size=i;

  2232.             }else

  2233.                 av_log(s->avctx, AV_LOG_ERROR, "invalid old backstep %d\n", i);

  2234.             s->gb= s->in_gb;

  2235.             s->in_gb.buffer= NULL;

  2236.         }

  2237. 

  2238.         align_get_bits(&s->gb);

  2239.         assert((get_bits_count(&s->gb) & 7) == 0);

  2240.         i= get_bits_left(&s->gb)>>3;

  2241. 

  2242.         if(i<0 || i > BACKSTEP_SIZE || nb_frames<0){

  2243.             if(i<0)

  2244.                 av_log(s->avctx, AV_LOG_ERROR, "invalid new backstep %d\n", i);

  2245.             i= FFMIN(BACKSTEP_SIZE, buf_size - HEADER_SIZE);

  2246.         }

  2247.         assert(i <= buf_size - HEADER_SIZE && i>= 0);

  2248.         memcpy(s->last_buf + s->last_buf_size, s->gb.buffer + buf_size - HEADER_SIZE - i, i);

  2249.         s->last_buf_size += i;

  2250. 

  2251.         break;

  2252.     }

  2253. 

  2254.     /* apply the synthesis filter */

  2255.     for(ch=0;ch<s->nb_channels;ch++) {

  2256.         samples_ptr = samples + ch;

  2257.         for(i=0;i<nb_frames;i++) {

  2258.             RENAME(ff_mpa_synth_filter)(

  2259. #if CONFIG_FLOAT

  2260.                          s,

  2261. #endif

  2262.                          s->synth_buf[ch], &(s->synth_buf_offset[ch]),

  2263.                          RENAME(ff_mpa_synth_window), &s->dither_state,

  2264.                          samples_ptr, s->nb_channels,

  2265.                          s->sb_samples[ch][i]);

  2266.             samples_ptr += 32 * s->nb_channels;

  2267.         }

  2268.     }

  2269. 

  2270.     return nb_frames * 32 * sizeof(OUT_INT) * s->nb_channels;

  2271. }

  2272. 

  2273. static int decode_frame(AVCodecContext * avctx,

  2274.                         void *data, int *data_size,

  2275.                         AVPacket *avpkt)

  2276. {

  2277.     const uint8_t *buf = avpkt->data;

  2278.     int buf_size = avpkt->size;

  2279.     MPADecodeContext *s = avctx->priv_data;

  2280.     uint32_t header;

  2281.     int out_size;

  2282.     OUT_INT *out_samples = data;

  2283. 

  2284.     if(buf_size < HEADER_SIZE)

  2285.         return -1;

  2286. 

  2287.     header = AV_RB32(buf);

  2288.     if(ff_mpa_check_header(header) < 0){

  2289.         av_log(avctx, AV_LOG_ERROR, "Header missing\n");

  2290.         return -1;

  2291.     }

  2292. 

  2293.     if (ff_mpegaudio_decode_header((MPADecodeHeader *)s, header) == 1) {

  2294.         /* free format: prepare to compute frame size */

  2295.         s->frame_size = -1;

  2296.         return -1;

  2297.     }

  2298.     /* update codec info */

  2299.     avctx->channels = s->nb_channels;

  2300.     avctx->bit_rate = s->bit_rate;

  2301.     avctx->sub_id = s->layer;

  2302. 

  2303.     if(*data_size < 1152*avctx->channels*sizeof(OUT_INT))

  2304.         return -1;

  2305.     *data_size = 0;

  2306. 

  2307.     if(s->frame_size<=0 || s->frame_size > buf_size){

  2308.         av_log(avctx, AV_LOG_ERROR, "incomplete frame\n");

  2309.         return -1;

  2310.     }else if(s->frame_size < buf_size){

  2311.         av_log(avctx, AV_LOG_ERROR, "incorrect frame size\n");

  2312.         buf_size= s->frame_size;

  2313.     }

  2314. 

  2315.     out_size = mp_decode_frame(s, out_samples, buf, buf_size);

  2316.     if(out_size>=0){

  2317.         *data_size = out_size;

  2318.         avctx->sample_rate = s->sample_rate;

  2319.         //FIXME maybe move the other codec info stuff from above here too

  2320.     }else

  2321.         av_log(avctx, AV_LOG_DEBUG, "Error while decoding MPEG audio frame.\n"); //FIXME return -1 / but also return the number of bytes consumed

  2322.     s->frame_size = 0;

  2323.     return buf_size;

  2324. }

  2325. 

  2326. static void flush(AVCodecContext *avctx){

  2327.     MPADecodeContext *s = avctx->priv_data;

  2328.     memset(s->synth_buf, 0, sizeof(s->synth_buf));

  2329.     s->last_buf_size= 0;

  2330. }

  2331. 

  2332. #if CONFIG_MP3ADU_DECODER

  2333. static int decode_frame_adu(AVCodecContext * avctx,

  2334.                         void *data, int *data_size,

  2335.                         AVPacket *avpkt)

  2336. {

  2337.     const uint8_t *buf = avpkt->data;

  2338.     int buf_size = avpkt->size;

  2339.     MPADecodeContext *s = avctx->priv_data;

  2340.     uint32_t header;

  2341.     int len, out_size;

  2342.     OUT_INT *out_samples = data;

  2343. 

  2344.     len = buf_size;

  2345. 

  2346.     // Discard too short frames

  2347.     if (buf_size < HEADER_SIZE) {

  2348.         *data_size = 0;

  2349.         return buf_size;

  2350.     }

  2351. 

  2352. 

  2353.     if (len > MPA_MAX_CODED_FRAME_SIZE)

  2354.         len = MPA_MAX_CODED_FRAME_SIZE;

  2355. 

  2356.     // Get header and restore sync word

  2357.     header = AV_RB32(buf) | 0xffe00000;

  2358. 

  2359.     if (ff_mpa_check_header(header) < 0) { // Bad header, discard frame

  2360.         *data_size = 0;

  2361.         return buf_size;

  2362.     }

  2363. 

  2364.     ff_mpegaudio_decode_header((MPADecodeHeader *)s, header);

  2365.     /* update codec info */

  2366.     avctx->sample_rate = s->sample_rate;

  2367.     avctx->channels = s->nb_channels;

  2368.     avctx->bit_rate = s->bit_rate;

  2369.     avctx->sub_id = s->layer;

  2370. 

  2371.     s->frame_size = len;

  2372. 

  2373.     if (avctx->parse_only) {

  2374.         out_size = buf_size;

  2375.     } else {

  2376.         out_size = mp_decode_frame(s, out_samples, buf, buf_size);

  2377.     }

  2378. 

  2379.     *data_size = out_size;

  2380.     return buf_size;

  2381. }

  2382. #endif /* CONFIG_MP3ADU_DECODER */

  2383. 

  2384. #if CONFIG_MP3ON4_DECODER

  2385. 

  2386. /**

  2387.  * Context for MP3On4 decoder

  2388.  */

  2389. typedef struct MP3On4DecodeContext {

  2390.     int frames;   ///< number of mp3 frames per block (number of mp3 decoder instances)

  2391.     int syncword; ///< syncword patch

  2392.     const uint8_t *coff; ///< channels offsets in output buffer

  2393.     MPADecodeContext *mp3decctx[5]; ///< MPADecodeContext for every decoder instance

  2394. } MP3On4DecodeContext;

  2395. 

  2396. #include "mpeg4audio.h"

  2397. 

  2398. /* Next 3 arrays are indexed by channel config number (passed via codecdata) */

  2399. static const uint8_t mp3Frames[8] = {0,1,1,2,3,3,4,5};   /* number of mp3 decoder instances */

  2400. /* offsets into output buffer, assume output order is FL FR BL BR C LFE */

  2401. static const uint8_t chan_offset[8][5] = {

  2402.     {0},

  2403.     {0},            // C

  2404.     {0},            // FLR

  2405.     {2,0},          // C FLR

  2406.     {2,0,3},        // C FLR BS

  2407.     {4,0,2},        // C FLR BLRS

  2408.     {4,0,2,5},      // C FLR BLRS LFE

  2409.     {4,0,2,6,5},    // C FLR BLRS BLR LFE

  2410. };

  2411. 

  2412. 

  2413. static int decode_init_mp3on4(AVCodecContext * avctx)

  2414. {

  2415.     MP3On4DecodeContext *s = avctx->priv_data;

  2416.     MPEG4AudioConfig cfg;

  2417.     int i;

  2418. 

  2419.     if ((avctx->extradata_size < 2) || (avctx->extradata == NULL)) {

  2420.         av_log(avctx, AV_LOG_ERROR, "Codec extradata missing or too short.\n");

  2421.         return -1;

  2422.     }

  2423. 

  2424.     ff_mpeg4audio_get_config(&cfg, avctx->extradata, avctx->extradata_size);

  2425.     if (!cfg.chan_config || cfg.chan_config > 7) {

  2426.         av_log(avctx, AV_LOG_ERROR, "Invalid channel config number.\n");

  2427.         return -1;

  2428.     }

  2429.     s->frames = mp3Frames[cfg.chan_config];

  2430.     s->coff = chan_offset[cfg.chan_config];

  2431.     avctx->channels = ff_mpeg4audio_channels[cfg.chan_config];

  2432. 

  2433.     if (cfg.sample_rate < 16000)

  2434.         s->syncword = 0xffe00000;

  2435.     else

  2436.         s->syncword = 0xfff00000;

  2437. 

  2438.     /* Init the first mp3 decoder in standard way, so that all tables get builded

  2439.      * We replace avctx->priv_data with the context of the first decoder so that

  2440.      * decode_init() does not have to be changed.

  2441.      * Other decoders will be initialized here copying data from the first context

  2442.      */

  2443.     // Allocate zeroed memory for the first decoder context

  2444.     s->mp3decctx[0] = av_mallocz(sizeof(MPADecodeContext));

  2445.     // Put decoder context in place to make init_decode() happy

  2446.     avctx->priv_data = s->mp3decctx[0];

  2447.     decode_init(avctx);

  2448.     // Restore mp3on4 context pointer

  2449.     avctx->priv_data = s;

  2450.     s->mp3decctx[0]->adu_mode = 1; // Set adu mode

  2451. 

  2452.     /* Create a separate codec/context for each frame (first is already ok).

  2453.      * Each frame is 1 or 2 channels - up to 5 frames allowed

  2454.      */

  2455.     for (i = 1; i < s->frames; i++) {

  2456.         s->mp3decctx[i] = av_mallocz(sizeof(MPADecodeContext));

  2457.         s->mp3decctx[i]->adu_mode = 1;

  2458.         s->mp3decctx[i]->avctx = avctx;

  2459.     }

  2460. 

  2461.     return 0;

  2462. }

  2463. 

  2464. 

  2465. static av_cold int decode_close_mp3on4(AVCodecContext * avctx)

  2466. {

  2467.     MP3On4DecodeContext *s = avctx->priv_data;

  2468.     int i;

  2469. 

  2470.     for (i = 0; i < s->frames; i++)

  2471.         if (s->mp3decctx[i])

  2472.             av_free(s->mp3decctx[i]);

  2473. 

  2474.     return 0;

  2475. }

  2476. 

  2477. 

  2478. static int decode_frame_mp3on4(AVCodecContext * avctx,

  2479.                         void *data, int *data_size,

  2480.                         AVPacket *avpkt)

  2481. {

  2482.     const uint8_t *buf = avpkt->data;

  2483.     int buf_size = avpkt->size;

  2484.     MP3On4DecodeContext *s = avctx->priv_data;

  2485.     MPADecodeContext *m;

  2486.     int fsize, len = buf_size, out_size = 0;

  2487.     uint32_t header;

  2488.     OUT_INT *out_samples = data;

  2489.     OUT_INT decoded_buf[MPA_FRAME_SIZE * MPA_MAX_CHANNELS];

  2490.     OUT_INT *outptr, *bp;

  2491.     int fr, j, n;

  2492. 

  2493.     if(*data_size < MPA_FRAME_SIZE * MPA_MAX_CHANNELS * s->frames * sizeof(OUT_INT))

  2494.         return -1;

  2495. 

  2496.     *data_size = 0;

  2497.     // Discard too short frames

  2498.     if (buf_size < HEADER_SIZE)

  2499.         return -1;

  2500. 

  2501.     // If only one decoder interleave is not needed

  2502.     outptr = s->frames == 1 ? out_samples : decoded_buf;

  2503. 

  2504.     avctx->bit_rate = 0;

  2505. 

  2506.     for (fr = 0; fr < s->frames; fr++) {

  2507.         fsize = AV_RB16(buf) >> 4;

  2508.         fsize = FFMIN3(fsize, len, MPA_MAX_CODED_FRAME_SIZE);

  2509.         m = s->mp3decctx[fr];

  2510.         assert (m != NULL);

  2511. 

  2512.         header = (AV_RB32(buf) & 0x000fffff) | s->syncword; // patch header

  2513. 

  2514.         if (ff_mpa_check_header(header) < 0) // Bad header, discard block

  2515.             break;

  2516. 

  2517.         ff_mpegaudio_decode_header((MPADecodeHeader *)m, header);

  2518.         out_size += mp_decode_frame(m, outptr, buf, fsize);

  2519.         buf += fsize;

  2520.         len -= fsize;

  2521. 

  2522.         if(s->frames > 1) {

  2523.             n = m->avctx->frame_size*m->nb_channels;

  2524.             /* interleave output data */

  2525.             bp = out_samples + s->coff[fr];

  2526.             if(m->nb_channels == 1) {

  2527.                 for(j = 0; j < n; j++) {

  2528.                     *bp = decoded_buf[j];

  2529.                     bp += avctx->channels;

  2530.                 }

  2531.             } else {

  2532.                 for(j = 0; j < n; j++) {

  2533.                     bp[0] = decoded_buf[j++];

  2534.                     bp[1] = decoded_buf[j];

  2535.                     bp += avctx->channels;

  2536.                 }

  2537.             }

  2538.         }

  2539.         avctx->bit_rate += m->bit_rate;

  2540.     }

  2541. 

  2542.     /* update codec info */

  2543.     avctx->sample_rate = s->mp3decctx[0]->sample_rate;

  2544. 

  2545.     *data_size = out_size;

  2546.     return buf_size;

  2547. }

  2548. #endif /* CONFIG_MP3ON4_DECODER */

  2549. 

  2550. #if !CONFIG_FLOAT

  2551. #if CONFIG_MP1_DECODER

  2552. AVCodec mp1_decoder =

  2553. {

  2554.     "mp1",

  2555.     AVMEDIA_TYPE_AUDIO,

  2556.     CODEC_ID_MP1,

  2557.     sizeof(MPADecodeContext),

  2558.     decode_init,

  2559.     NULL,

  2560.     NULL,

  2561.     decode_frame,

  2562.     CODEC_CAP_PARSE_ONLY,

  2563.     .flush= flush,

  2564.     .long_name= NULL_IF_CONFIG_SMALL("MP1 (MPEG audio layer 1)"),

  2565. };

  2566. #endif

  2567. #if CONFIG_MP2_DECODER

  2568. AVCodec mp2_decoder =

  2569. {

  2570.     "mp2",

  2571.     AVMEDIA_TYPE_AUDIO,

  2572.     CODEC_ID_MP2,

  2573.     sizeof(MPADecodeContext),

  2574.     decode_init,

  2575.     NULL,

  2576.     NULL,

  2577.     decode_frame,

  2578.     CODEC_CAP_PARSE_ONLY,

  2579.     .flush= flush,

  2580.     .long_name= NULL_IF_CONFIG_SMALL("MP2 (MPEG audio layer 2)"),

  2581. };

  2582. #endif

  2583. #if CONFIG_MP3_DECODER

  2584. AVCodec mp3_decoder =

  2585. {

  2586.     "mp3",

  2587.     AVMEDIA_TYPE_AUDIO,

  2588.     CODEC_ID_MP3,

  2589.     sizeof(MPADecodeContext),

  2590.     decode_init,

  2591.     NULL,

  2592.     NULL,

  2593.     decode_frame,

  2594.     CODEC_CAP_PARSE_ONLY,

  2595.     .flush= flush,

  2596.     .long_name= NULL_IF_CONFIG_SMALL("MP3 (MPEG audio layer 3)"),

  2597. };

  2598. #endif

  2599. #if CONFIG_MP3ADU_DECODER

  2600. AVCodec mp3adu_decoder =

  2601. {

  2602.     "mp3adu",

  2603.     AVMEDIA_TYPE_AUDIO,

  2604.     CODEC_ID_MP3ADU,

  2605.     sizeof(MPADecodeContext),

  2606.     decode_init,

  2607.     NULL,

  2608.     NULL,

  2609.     decode_frame_adu,

  2610.     CODEC_CAP_PARSE_ONLY,

  2611.     .flush= flush,

  2612.     .long_name= NULL_IF_CONFIG_SMALL("ADU (Application Data Unit) MP3 (MPEG audio layer 3)"),

  2613. };

  2614. #endif

  2615. #if CONFIG_MP3ON4_DECODER

  2616. AVCodec mp3on4_decoder =

  2617. {

  2618.     "mp3on4",

  2619.     AVMEDIA_TYPE_AUDIO,

  2620.     CODEC_ID_MP3ON4,

  2621.     sizeof(MP3On4DecodeContext),

  2622.     decode_init_mp3on4,

  2623.     NULL,

  2624.     decode_close_mp3on4,

  2625.     decode_frame_mp3on4,

  2626.     .flush= flush,

  2627.     .long_name= NULL_IF_CONFIG_SMALL("MP3onMP4"),

  2628. };

  2629. #endif

  2630. #endif