git.sesse.net Git - ffmpeg/blob - libavcodec/aacdec_template.c

   1 /*
   2  * AAC decoder
   3  * Copyright (c) 2005-2006 Oded Shimon ( ods15 ods15 dyndns org )
   4  * Copyright (c) 2006-2007 Maxim Gavrilov ( maxim.gavrilov gmail com )
   5  * Copyright (c) 2008-2013 Alex Converse <alex.converse@gmail.com>
   6  *
   7  * AAC LATM decoder
   8  * Copyright (c) 2008-2010 Paul Kendall <paul@kcbbs.gen.nz>
   9  * Copyright (c) 2010      Janne Grunau <janne-libav@jannau.net>
  10  *
  11  * AAC decoder fixed-point implementation
  12  * Copyright (c) 2013
  13  *      MIPS Technologies, Inc., California.
  14  *
  15  * This file is part of FFmpeg.
  16  *
  17  * FFmpeg is free software; you can redistribute it and/or
  18  * modify it under the terms of the GNU Lesser General Public
  19  * License as published by the Free Software Foundation; either
  20  * version 2.1 of the License, or (at your option) any later version.
  21  *
  22  * FFmpeg is distributed in the hope that it will be useful,
  23  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  24  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  25  * Lesser General Public License for more details.
  26  *
  27  * You should have received a copy of the GNU Lesser General Public
  28  * License along with FFmpeg; if not, write to the Free Software
  29  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  30  */
  31
  32 /**
  33  * @file
  34  * AAC decoder
  35  * @author Oded Shimon  ( ods15 ods15 dyndns org )
  36  * @author Maxim Gavrilov ( maxim.gavrilov gmail com )
  37  *
  38  * AAC decoder fixed-point implementation
  39  * @author Stanislav Ocovaj ( stanislav.ocovaj imgtec com )
  40  * @author Nedeljko Babic ( nedeljko.babic imgtec com )
  41  */
  42
  43 /*
  44  * supported tools
  45  *
  46  * Support?                     Name
  47  * N (code in SoC repo)         gain control
  48  * Y                            block switching
  49  * Y                            window shapes - standard
  50  * N                            window shapes - Low Delay
  51  * Y                            filterbank - standard
  52  * N (code in SoC repo)         filterbank - Scalable Sample Rate
  53  * Y                            Temporal Noise Shaping
  54  * Y                            Long Term Prediction
  55  * Y                            intensity stereo
  56  * Y                            channel coupling
  57  * Y                            frequency domain prediction
  58  * Y                            Perceptual Noise Substitution
  59  * Y                            Mid/Side stereo
  60  * N                            Scalable Inverse AAC Quantization
  61  * N                            Frequency Selective Switch
  62  * N                            upsampling filter
  63  * Y                            quantization & coding - AAC
  64  * N                            quantization & coding - TwinVQ
  65  * N                            quantization & coding - BSAC
  66  * N                            AAC Error Resilience tools
  67  * N                            Error Resilience payload syntax
  68  * N                            Error Protection tool
  69  * N                            CELP
  70  * N                            Silence Compression
  71  * N                            HVXC
  72  * N                            HVXC 4kbits/s VR
  73  * N                            Structured Audio tools
  74  * N                            Structured Audio Sample Bank Format
  75  * N                            MIDI
  76  * N                            Harmonic and Individual Lines plus Noise
  77  * N                            Text-To-Speech Interface
  78  * Y                            Spectral Band Replication
  79  * Y (not in this code)         Layer-1
  80  * Y (not in this code)         Layer-2
  81  * Y (not in this code)         Layer-3
  82  * N                            SinuSoidal Coding (Transient, Sinusoid, Noise)
  83  * Y                            Parametric Stereo
  84  * N                            Direct Stream Transfer
  85  * Y  (not in fixed point code) Enhanced AAC Low Delay (ER AAC ELD)
  86  *
  87  * Note: - HE AAC v1 comprises LC AAC with Spectral Band Replication.
  88  *       - HE AAC v2 comprises LC AAC with Spectral Band Replication and
  89            Parametric Stereo.
  90  */
  91
  92 #include "libavutil/thread.h"
  93
  94 static VLC vlc_scalefactors;
  95 static VLC vlc_spectral[11];
  96
  97 static int output_configure(AACContext *ac,
  98                             uint8_t layout_map[MAX_ELEM_ID*4][3], int tags,
  99                             enum OCStatus oc_type, int get_new_frame);
 100
 101 #define overread_err "Input buffer exhausted before END element found\n"
 102
 103 static int count_channels(uint8_t (*layout)[3], int tags)
 104 {
 105     int i, sum = 0;
 106     for (i = 0; i < tags; i++) {
 107         int syn_ele = layout[i][0];
 108         int pos     = layout[i][2];
 109         sum += (1 + (syn_ele == TYPE_CPE)) *
 110                (pos != AAC_CHANNEL_OFF && pos != AAC_CHANNEL_CC);
 111     }
 112     return sum;
 113 }
 114
 115 /**
 116  * Check for the channel element in the current channel position configuration.
 117  * If it exists, make sure the appropriate element is allocated and map the
 118  * channel order to match the internal FFmpeg channel layout.
 119  *
 120  * @param   che_pos current channel position configuration
 121  * @param   type channel element type
 122  * @param   id channel element id
 123  * @param   channels count of the number of channels in the configuration
 124  *
 125  * @return  Returns error status. 0 - OK, !0 - error
 126  */
 127 static av_cold int che_configure(AACContext *ac,
 128                                  enum ChannelPosition che_pos,
 129                                  int type, int id, int *channels)
 130 {
 131     if (*channels >= MAX_CHANNELS)
 132         return AVERROR_INVALIDDATA;
 133     if (che_pos) {
 134         if (!ac->che[type][id]) {
 135             if (!(ac->che[type][id] = av_mallocz(sizeof(ChannelElement))))
 136                 return AVERROR(ENOMEM);
 137             AAC_RENAME(ff_aac_sbr_ctx_init)(ac, &ac->che[type][id]->sbr, type);
 138         }
 139         if (type != TYPE_CCE) {
 140             if (*channels >= MAX_CHANNELS - (type == TYPE_CPE || (type == TYPE_SCE && ac->oc[1].m4ac.ps == 1))) {
 141                 av_log(ac->avctx, AV_LOG_ERROR, "Too many channels\n");
 142                 return AVERROR_INVALIDDATA;
 143             }
 144             ac->output_element[(*channels)++] = &ac->che[type][id]->ch[0];
 145             if (type == TYPE_CPE ||
 146                 (type == TYPE_SCE && ac->oc[1].m4ac.ps == 1)) {
 147                 ac->output_element[(*channels)++] = &ac->che[type][id]->ch[1];
 148             }
 149         }
 150     } else {
 151         if (ac->che[type][id])
 152             AAC_RENAME(ff_aac_sbr_ctx_close)(&ac->che[type][id]->sbr);
 153         av_freep(&ac->che[type][id]);
 154     }
 155     return 0;
 156 }
 157
 158 static int frame_configure_elements(AVCodecContext *avctx)
 159 {
 160     AACContext *ac = avctx->priv_data;
 161     int type, id, ch, ret;
 162
 163     /* set channel pointers to internal buffers by default */
 164     for (type = 0; type < 4; type++) {
 165         for (id = 0; id < MAX_ELEM_ID; id++) {
 166             ChannelElement *che = ac->che[type][id];
 167             if (che) {
 168                 che->ch[0].ret = che->ch[0].ret_buf;
 169                 che->ch[1].ret = che->ch[1].ret_buf;
 170             }
 171         }
 172     }
 173
 174     /* get output buffer */
 175     av_frame_unref(ac->frame);
 176     if (!avctx->channels)
 177         return 1;
 178
 179     ac->frame->nb_samples = 2048;
 180     if ((ret = ff_get_buffer(avctx, ac->frame, 0)) < 0)
 181         return ret;
 182
 183     /* map output channel pointers to AVFrame data */
 184     for (ch = 0; ch < avctx->channels; ch++) {
 185         if (ac->output_element[ch])
 186             ac->output_element[ch]->ret = (INTFLOAT *)ac->frame->extended_data[ch];
 187     }
 188
 189     return 0;
 190 }
 191
 192 struct elem_to_channel {
 193     uint64_t av_position;
 194     uint8_t syn_ele;
 195     uint8_t elem_id;
 196     uint8_t aac_position;
 197 };
 198
 199 static int assign_pair(struct elem_to_channel e2c_vec[MAX_ELEM_ID],
 200                        uint8_t (*layout_map)[3], int offset, uint64_t left,
 201                        uint64_t right, int pos, uint64_t *layout)
 202 {
 203     if (layout_map[offset][0] == TYPE_CPE) {
 204         e2c_vec[offset] = (struct elem_to_channel) {
 205             .av_position  = left | right,
 206             .syn_ele      = TYPE_CPE,
 207             .elem_id      = layout_map[offset][1],
 208             .aac_position = pos
 209         };
 210         if (e2c_vec[offset].av_position != UINT64_MAX)
 211             *layout |= e2c_vec[offset].av_position;
 212
 213         return 1;
 214     } else {
 215         e2c_vec[offset] = (struct elem_to_channel) {
 216             .av_position  = left,
 217             .syn_ele      = TYPE_SCE,
 218             .elem_id      = layout_map[offset][1],
 219             .aac_position = pos
 220         };
 221         e2c_vec[offset + 1] = (struct elem_to_channel) {
 222             .av_position  = right,
 223             .syn_ele      = TYPE_SCE,
 224             .elem_id      = layout_map[offset + 1][1],
 225             .aac_position = pos
 226         };
 227         if (left != UINT64_MAX)
 228             *layout |= left;
 229
 230         if (right != UINT64_MAX)
 231             *layout |= right;
 232
 233         return 2;
 234     }
 235 }
 236
 237 static int count_paired_channels(uint8_t (*layout_map)[3], int tags, int pos,
 238                                  int *current)
 239 {
 240     int num_pos_channels = 0;
 241     int first_cpe        = 0;
 242     int sce_parity       = 0;
 243     int i;
 244     for (i = *current; i < tags; i++) {
 245         if (layout_map[i][2] != pos)
 246             break;
 247         if (layout_map[i][0] == TYPE_CPE) {
 248             if (sce_parity) {
 249                 if (pos == AAC_CHANNEL_FRONT && !first_cpe) {
 250                     sce_parity = 0;
 251                 } else {
 252                     return -1;
 253                 }
 254             }
 255             num_pos_channels += 2;
 256             first_cpe         = 1;
 257         } else {
 258             num_pos_channels++;
 259             sce_parity ^= 1;
 260         }
 261     }
 262     if (sce_parity &&
 263         ((pos == AAC_CHANNEL_FRONT && first_cpe) || pos == AAC_CHANNEL_SIDE))
 264         return -1;
 265     *current = i;
 266     return num_pos_channels;
 267 }
 268
 269 #define PREFIX_FOR_22POINT2 (AV_CH_LAYOUT_7POINT1_WIDE_BACK|AV_CH_BACK_CENTER|AV_CH_SIDE_LEFT|AV_CH_SIDE_RIGHT|AV_CH_LOW_FREQUENCY_2)
 270 static uint64_t sniff_channel_order(uint8_t (*layout_map)[3], int tags)
 271 {
 272     int i, n, total_non_cc_elements;
 273     struct elem_to_channel e2c_vec[4 * MAX_ELEM_ID] = { { 0 } };
 274     int num_front_channels, num_side_channels, num_back_channels;
 275     uint64_t layout = 0;
 276
 277     if (FF_ARRAY_ELEMS(e2c_vec) < tags)
 278         return 0;
 279
 280     i = 0;
 281     num_front_channels =
 282         count_paired_channels(layout_map, tags, AAC_CHANNEL_FRONT, &i);
 283     if (num_front_channels < 0)
 284         return 0;
 285     num_side_channels =
 286         count_paired_channels(layout_map, tags, AAC_CHANNEL_SIDE, &i);
 287     if (num_side_channels < 0)
 288         return 0;
 289     num_back_channels =
 290         count_paired_channels(layout_map, tags, AAC_CHANNEL_BACK, &i);
 291     if (num_back_channels < 0)
 292         return 0;
 293
 294     if (num_side_channels == 0 && num_back_channels >= 4) {
 295         num_side_channels = 2;
 296         num_back_channels -= 2;
 297     }
 298
 299     i = 0;
 300     if (num_front_channels & 1) {
 301         e2c_vec[i] = (struct elem_to_channel) {
 302             .av_position  = AV_CH_FRONT_CENTER,
 303             .syn_ele      = TYPE_SCE,
 304             .elem_id      = layout_map[i][1],
 305             .aac_position = AAC_CHANNEL_FRONT
 306         };
 307         layout |= e2c_vec[i].av_position;
 308         i++;
 309         num_front_channels--;
 310     }
 311     if (num_front_channels >= 4) {
 312         i += assign_pair(e2c_vec, layout_map, i,
 313                          AV_CH_FRONT_LEFT_OF_CENTER,
 314                          AV_CH_FRONT_RIGHT_OF_CENTER,
 315                          AAC_CHANNEL_FRONT, &layout);
 316         num_front_channels -= 2;
 317     }
 318     if (num_front_channels >= 2) {
 319         i += assign_pair(e2c_vec, layout_map, i,
 320                          AV_CH_FRONT_LEFT,
 321                          AV_CH_FRONT_RIGHT,
 322                          AAC_CHANNEL_FRONT, &layout);
 323         num_front_channels -= 2;
 324     }
 325     while (num_front_channels >= 2) {
 326         i += assign_pair(e2c_vec, layout_map, i,
 327                          UINT64_MAX,
 328                          UINT64_MAX,
 329                          AAC_CHANNEL_FRONT, &layout);
 330         num_front_channels -= 2;
 331     }
 332
 333     if (num_side_channels >= 2) {
 334         i += assign_pair(e2c_vec, layout_map, i,
 335                          AV_CH_SIDE_LEFT,
 336                          AV_CH_SIDE_RIGHT,
 337                          AAC_CHANNEL_FRONT, &layout);
 338         num_side_channels -= 2;
 339     }
 340     while (num_side_channels >= 2) {
 341         i += assign_pair(e2c_vec, layout_map, i,
 342                          UINT64_MAX,
 343                          UINT64_MAX,
 344                          AAC_CHANNEL_SIDE, &layout);
 345         num_side_channels -= 2;
 346     }
 347
 348     while (num_back_channels >= 4) {
 349         i += assign_pair(e2c_vec, layout_map, i,
 350                          UINT64_MAX,
 351                          UINT64_MAX,
 352                          AAC_CHANNEL_BACK, &layout);
 353         num_back_channels -= 2;
 354     }
 355     if (num_back_channels >= 2) {
 356         i += assign_pair(e2c_vec, layout_map, i,
 357                          AV_CH_BACK_LEFT,
 358                          AV_CH_BACK_RIGHT,
 359                          AAC_CHANNEL_BACK, &layout);
 360         num_back_channels -= 2;
 361     }
 362     if (num_back_channels) {
 363         e2c_vec[i] = (struct elem_to_channel) {
 364             .av_position  = AV_CH_BACK_CENTER,
 365             .syn_ele      = TYPE_SCE,
 366             .elem_id      = layout_map[i][1],
 367             .aac_position = AAC_CHANNEL_BACK
 368         };
 369         layout |= e2c_vec[i].av_position;
 370         i++;
 371         num_back_channels--;
 372     }
 373
 374     if (i < tags && layout_map[i][2] == AAC_CHANNEL_LFE) {
 375         e2c_vec[i] = (struct elem_to_channel) {
 376             .av_position  = AV_CH_LOW_FREQUENCY,
 377             .syn_ele      = TYPE_LFE,
 378             .elem_id      = layout_map[i][1],
 379             .aac_position = AAC_CHANNEL_LFE
 380         };
 381         layout |= e2c_vec[i].av_position;
 382         i++;
 383     }
 384     if (i < tags && layout_map[i][2] == AAC_CHANNEL_LFE) {
 385         e2c_vec[i] = (struct elem_to_channel) {
 386             .av_position  = AV_CH_LOW_FREQUENCY_2,
 387             .syn_ele      = TYPE_LFE,
 388             .elem_id      = layout_map[i][1],
 389             .aac_position = AAC_CHANNEL_LFE
 390         };
 391         layout |= e2c_vec[i].av_position;
 392         i++;
 393     }
 394     while (i < tags && layout_map[i][2] == AAC_CHANNEL_LFE) {
 395         e2c_vec[i] = (struct elem_to_channel) {
 396             .av_position  = UINT64_MAX,
 397             .syn_ele      = TYPE_LFE,
 398             .elem_id      = layout_map[i][1],
 399             .aac_position = AAC_CHANNEL_LFE
 400         };
 401         i++;
 402     }
 403
 404     // The previous checks would end up at 8 at this point for 22.2
 405     if (layout == PREFIX_FOR_22POINT2 && tags == 16 && i == 8) {
 406         const uint8_t (*reference_layout_map)[3] = aac_channel_layout_map[12];
 407         for (int j = 0; j < tags; j++) {
 408             if (layout_map[j][0] != reference_layout_map[j][0] ||
 409                 layout_map[j][2] != reference_layout_map[j][2])
 410                 goto end_of_layout_definition;
 411         }
 412
 413         e2c_vec[i] = (struct elem_to_channel) {
 414             .av_position  = AV_CH_TOP_FRONT_CENTER,
 415             .syn_ele      = layout_map[i][0],
 416             .elem_id      = layout_map[i][1],
 417             .aac_position = layout_map[i][2]
 418         }; layout |= e2c_vec[i].av_position; i++;
 419         i += assign_pair(e2c_vec, layout_map, i,
 420                          AV_CH_TOP_FRONT_LEFT,
 421                          AV_CH_TOP_FRONT_RIGHT,
 422                          AAC_CHANNEL_FRONT,
 423                          &layout);
 424         i += assign_pair(e2c_vec, layout_map, i,
 425                          AV_CH_TOP_SIDE_LEFT,
 426                          AV_CH_TOP_SIDE_RIGHT,
 427                          AAC_CHANNEL_SIDE,
 428                          &layout);
 429         e2c_vec[i] = (struct elem_to_channel) {
 430             .av_position  = AV_CH_TOP_CENTER,
 431             .syn_ele      = layout_map[i][0],
 432             .elem_id      = layout_map[i][1],
 433             .aac_position = layout_map[i][2]
 434         }; layout |= e2c_vec[i].av_position; i++;
 435         i += assign_pair(e2c_vec, layout_map, i,
 436                          AV_CH_TOP_BACK_LEFT,
 437                          AV_CH_TOP_BACK_RIGHT,
 438                          AAC_CHANNEL_BACK,
 439                          &layout);
 440         e2c_vec[i] = (struct elem_to_channel) {
 441             .av_position  = AV_CH_TOP_BACK_CENTER,
 442             .syn_ele      = layout_map[i][0],
 443             .elem_id      = layout_map[i][1],
 444             .aac_position = layout_map[i][2]
 445         }; layout |= e2c_vec[i].av_position; i++;
 446         e2c_vec[i] = (struct elem_to_channel) {
 447             .av_position  = AV_CH_BOTTOM_FRONT_CENTER,
 448             .syn_ele      = layout_map[i][0],
 449             .elem_id      = layout_map[i][1],
 450             .aac_position = layout_map[i][2]
 451         }; layout |= e2c_vec[i].av_position; i++;
 452         i += assign_pair(e2c_vec, layout_map, i,
 453                          AV_CH_BOTTOM_FRONT_LEFT,
 454                          AV_CH_BOTTOM_FRONT_RIGHT,
 455                          AAC_CHANNEL_FRONT,
 456                          &layout);
 457     }
 458
 459 end_of_layout_definition:
 460
 461     total_non_cc_elements = n = i;
 462
 463     if (layout == AV_CH_LAYOUT_22POINT2) {
 464         // For 22.2 reorder the result as needed
 465         FFSWAP(struct elem_to_channel, e2c_vec[2], e2c_vec[0]);   // FL & FR first (final), FC third
 466         FFSWAP(struct elem_to_channel, e2c_vec[2], e2c_vec[1]);   // FC second (final), FLc & FRc third
 467         FFSWAP(struct elem_to_channel, e2c_vec[6], e2c_vec[2]);   // LFE1 third (final), FLc & FRc seventh
 468         FFSWAP(struct elem_to_channel, e2c_vec[4], e2c_vec[3]);   // BL & BR fourth (final), SiL & SiR fifth
 469         FFSWAP(struct elem_to_channel, e2c_vec[6], e2c_vec[4]);   // FLc & FRc fifth (final), SiL & SiR seventh
 470         FFSWAP(struct elem_to_channel, e2c_vec[7], e2c_vec[6]);   // LFE2 seventh (final), SiL & SiR eight (final)
 471         FFSWAP(struct elem_to_channel, e2c_vec[9], e2c_vec[8]);   // TpFL & TpFR ninth (final), TFC tenth (final)
 472         FFSWAP(struct elem_to_channel, e2c_vec[11], e2c_vec[10]); // TC eleventh (final), TpSiL & TpSiR twelth
 473         FFSWAP(struct elem_to_channel, e2c_vec[12], e2c_vec[11]); // TpBL & TpBR twelth (final), TpSiL & TpSiR thirteenth (final)
 474     } else {
 475         // For everything else, utilize the AV channel position define as a
 476         // stable sort.
 477         do {
 478             int next_n = 0;
 479             for (i = 1; i < n; i++)
 480                 if (e2c_vec[i - 1].av_position > e2c_vec[i].av_position) {
 481                     FFSWAP(struct elem_to_channel, e2c_vec[i - 1], e2c_vec[i]);
 482                     next_n = i;
 483                 }
 484             n = next_n;
 485         } while (n > 0);
 486
 487     }
 488
 489     for (i = 0; i < total_non_cc_elements; i++) {
 490         layout_map[i][0] = e2c_vec[i].syn_ele;
 491         layout_map[i][1] = e2c_vec[i].elem_id;
 492         layout_map[i][2] = e2c_vec[i].aac_position;
 493     }
 494
 495     return layout;
 496 }
 497
 498 /**
 499  * Save current output configuration if and only if it has been locked.
 500  */
 501 static int push_output_configuration(AACContext *ac) {
 502     int pushed = 0;
 503
 504     if (ac->oc[1].status == OC_LOCKED || ac->oc[0].status == OC_NONE) {
 505         ac->oc[0] = ac->oc[1];
 506         pushed = 1;
 507     }
 508     ac->oc[1].status = OC_NONE;
 509     return pushed;
 510 }
 511
 512 /**
 513  * Restore the previous output configuration if and only if the current
 514  * configuration is unlocked.
 515  */
 516 static void pop_output_configuration(AACContext *ac) {
 517     if (ac->oc[1].status != OC_LOCKED && ac->oc[0].status != OC_NONE) {
 518         ac->oc[1] = ac->oc[0];
 519         ac->avctx->channels = ac->oc[1].channels;
 520         ac->avctx->channel_layout = ac->oc[1].channel_layout;
 521         output_configure(ac, ac->oc[1].layout_map, ac->oc[1].layout_map_tags,
 522                          ac->oc[1].status, 0);
 523     }
 524 }
 525
 526 /**
 527  * Configure output channel order based on the current program
 528  * configuration element.
 529  *
 530  * @return  Returns error status. 0 - OK, !0 - error
 531  */
 532 static int output_configure(AACContext *ac,
 533                             uint8_t layout_map[MAX_ELEM_ID * 4][3], int tags,
 534                             enum OCStatus oc_type, int get_new_frame)
 535 {
 536     AVCodecContext *avctx = ac->avctx;
 537     int i, channels = 0, ret;
 538     uint64_t layout = 0;
 539     uint8_t id_map[TYPE_END][MAX_ELEM_ID] = {{ 0 }};
 540     uint8_t type_counts[TYPE_END] = { 0 };
 541
 542     if (ac->oc[1].layout_map != layout_map) {
 543         memcpy(ac->oc[1].layout_map, layout_map, tags * sizeof(layout_map[0]));
 544         ac->oc[1].layout_map_tags = tags;
 545     }
 546     for (i = 0; i < tags; i++) {
 547         int type =         layout_map[i][0];
 548         int id =           layout_map[i][1];
 549         id_map[type][id] = type_counts[type]++;
 550         if (id_map[type][id] >= MAX_ELEM_ID) {
 551             avpriv_request_sample(ac->avctx, "Too large remapped id");
 552             return AVERROR_PATCHWELCOME;
 553         }
 554     }
 555     // Try to sniff a reasonable channel order, otherwise output the
 556     // channels in the order the PCE declared them.
 557     if (avctx->request_channel_layout != AV_CH_LAYOUT_NATIVE)
 558         layout = sniff_channel_order(layout_map, tags);
 559     for (i = 0; i < tags; i++) {
 560         int type =     layout_map[i][0];
 561         int id =       layout_map[i][1];
 562         int iid =      id_map[type][id];
 563         int position = layout_map[i][2];
 564         // Allocate or free elements depending on if they are in the
 565         // current program configuration.
 566         ret = che_configure(ac, position, type, iid, &channels);
 567         if (ret < 0)
 568             return ret;
 569         ac->tag_che_map[type][id] = ac->che[type][iid];
 570     }
 571     if (ac->oc[1].m4ac.ps == 1 && channels == 2) {
 572         if (layout == AV_CH_FRONT_CENTER) {
 573             layout = AV_CH_FRONT_LEFT|AV_CH_FRONT_RIGHT;
 574         } else {
 575             layout = 0;
 576         }
 577     }
 578
 579     if (layout) avctx->channel_layout = layout;
 580                             ac->oc[1].channel_layout = layout;
 581     avctx->channels       = ac->oc[1].channels       = channels;
 582     ac->oc[1].status = oc_type;
 583
 584     if (get_new_frame) {
 585         if ((ret = frame_configure_elements(ac->avctx)) < 0)
 586             return ret;
 587     }
 588
 589     return 0;
 590 }
 591
 592 static void flush(AVCodecContext *avctx)
 593 {
 594     AACContext *ac= avctx->priv_data;
 595     int type, i, j;
 596
 597     for (type = 3; type >= 0; type--) {
 598         for (i = 0; i < MAX_ELEM_ID; i++) {
 599             ChannelElement *che = ac->che[type][i];
 600             if (che) {
 601                 for (j = 0; j <= 1; j++) {
 602                     memset(che->ch[j].saved, 0, sizeof(che->ch[j].saved));
 603                 }
 604             }
 605         }
 606     }
 607 }
 608
 609 /**
 610  * Set up channel positions based on a default channel configuration
 611  * as specified in table 1.17.
 612  *
 613  * @return  Returns error status. 0 - OK, !0 - error
 614  */
 615 static int set_default_channel_config(AACContext *ac, AVCodecContext *avctx,
 616                                       uint8_t (*layout_map)[3],
 617                                       int *tags,
 618                                       int channel_config)
 619 {
 620     if (channel_config < 1 || (channel_config > 7 && channel_config < 11) ||
 621         channel_config > 13) {
 622         av_log(avctx, AV_LOG_ERROR,
 623                "invalid default channel configuration (%d)\n",
 624                channel_config);
 625         return AVERROR_INVALIDDATA;
 626     }
 627     *tags = tags_per_config[channel_config];
 628     memcpy(layout_map, aac_channel_layout_map[channel_config - 1],
 629            *tags * sizeof(*layout_map));
 630
 631     /*
 632      * AAC specification has 7.1(wide) as a default layout for 8-channel streams.
 633      * However, at least Nero AAC encoder encodes 7.1 streams using the default
 634      * channel config 7, mapping the side channels of the original audio stream
 635      * to the second AAC_CHANNEL_FRONT pair in the AAC stream. Similarly, e.g. FAAD
 636      * decodes the second AAC_CHANNEL_FRONT pair as side channels, therefore decoding
 637      * the incorrect streams as if they were correct (and as the encoder intended).
 638      *
 639      * As actual intended 7.1(wide) streams are very rare, default to assuming a
 640      * 7.1 layout was intended.
 641      */
 642     if (channel_config == 7 && avctx->strict_std_compliance < FF_COMPLIANCE_STRICT && (!ac || !ac->warned_71_wide++)) {
 643         av_log(avctx, AV_LOG_INFO, "Assuming an incorrectly encoded 7.1 channel layout"
 644                " instead of a spec-compliant 7.1(wide) layout, use -strict %d to decode"
 645                " according to the specification instead.\n", FF_COMPLIANCE_STRICT);
 646         layout_map[2][2] = AAC_CHANNEL_SIDE;
 647     }
 648
 649     return 0;
 650 }
 651
 652 static ChannelElement *get_che(AACContext *ac, int type, int elem_id)
 653 {
 654     /* For PCE based channel configurations map the channels solely based
 655      * on tags. */
 656     if (!ac->oc[1].m4ac.chan_config) {
 657         return ac->tag_che_map[type][elem_id];
 658     }
 659     // Allow single CPE stereo files to be signalled with mono configuration.
 660     if (!ac->tags_mapped && type == TYPE_CPE &&
 661         ac->oc[1].m4ac.chan_config == 1) {
 662         uint8_t layout_map[MAX_ELEM_ID*4][3];
 663         int layout_map_tags;
 664         push_output_configuration(ac);
 665
 666         av_log(ac->avctx, AV_LOG_DEBUG, "mono with CPE\n");
 667
 668         if (set_default_channel_config(ac, ac->avctx, layout_map,
 669                                        &layout_map_tags, 2) < 0)
 670             return NULL;
 671         if (output_configure(ac, layout_map, layout_map_tags,
 672                              OC_TRIAL_FRAME, 1) < 0)
 673             return NULL;
 674
 675         ac->oc[1].m4ac.chan_config = 2;
 676         ac->oc[1].m4ac.ps = 0;
 677     }
 678     // And vice-versa
 679     if (!ac->tags_mapped && type == TYPE_SCE &&
 680         ac->oc[1].m4ac.chan_config == 2) {
 681         uint8_t layout_map[MAX_ELEM_ID * 4][3];
 682         int layout_map_tags;
 683         push_output_configuration(ac);
 684
 685         av_log(ac->avctx, AV_LOG_DEBUG, "stereo with SCE\n");
 686
 687         if (set_default_channel_config(ac, ac->avctx, layout_map,
 688                                        &layout_map_tags, 1) < 0)
 689             return NULL;
 690         if (output_configure(ac, layout_map, layout_map_tags,
 691                              OC_TRIAL_FRAME, 1) < 0)
 692             return NULL;
 693
 694         ac->oc[1].m4ac.chan_config = 1;
 695         if (ac->oc[1].m4ac.sbr)
 696             ac->oc[1].m4ac.ps = -1;
 697     }
 698     /* For indexed channel configurations map the channels solely based
 699      * on position. */
 700     switch (ac->oc[1].m4ac.chan_config) {
 701     case 13:
 702         if (ac->tags_mapped > 3 && ((type == TYPE_CPE && elem_id < 8) ||
 703                                     (type == TYPE_SCE && elem_id < 6) ||
 704                                     (type == TYPE_LFE && elem_id < 2))) {
 705             ac->tags_mapped++;
 706             return ac->tag_che_map[type][elem_id] = ac->che[type][elem_id];
 707         }
 708     case 12:
 709     case 7:
 710         if (ac->tags_mapped == 3 && type == TYPE_CPE) {
 711             ac->tags_mapped++;
 712             return ac->tag_che_map[TYPE_CPE][elem_id] = ac->che[TYPE_CPE][2];
 713         }
 714     case 11:
 715         if (ac->tags_mapped == 2 &&
 716             ac->oc[1].m4ac.chan_config == 11 &&
 717             type == TYPE_SCE) {
 718             ac->tags_mapped++;
 719             return ac->tag_che_map[TYPE_SCE][elem_id] = ac->che[TYPE_SCE][1];
 720         }
 721     case 6:
 722         /* Some streams incorrectly code 5.1 audio as
 723          * SCE[0] CPE[0] CPE[1] SCE[1]
 724          * instead of
 725          * SCE[0] CPE[0] CPE[1] LFE[0].
 726          * If we seem to have encountered such a stream, transfer
 727          * the LFE[0] element to the SCE[1]'s mapping */
 728         if (ac->tags_mapped == tags_per_config[ac->oc[1].m4ac.chan_config] - 1 && (type == TYPE_LFE || type == TYPE_SCE)) {
 729             if (!ac->warned_remapping_once && (type != TYPE_LFE || elem_id != 0)) {
 730                 av_log(ac->avctx, AV_LOG_WARNING,
 731                    "This stream seems to incorrectly report its last channel as %s[%d], mapping to LFE[0]\n",
 732                    type == TYPE_SCE ? "SCE" : "LFE", elem_id);
 733                 ac->warned_remapping_once++;
 734             }
 735             ac->tags_mapped++;
 736             return ac->tag_che_map[type][elem_id] = ac->che[TYPE_LFE][0];
 737         }
 738     case 5:
 739         if (ac->tags_mapped == 2 && type == TYPE_CPE) {
 740             ac->tags_mapped++;
 741             return ac->tag_che_map[TYPE_CPE][elem_id] = ac->che[TYPE_CPE][1];
 742         }
 743     case 4:
 744         /* Some streams incorrectly code 4.0 audio as
 745          * SCE[0] CPE[0] LFE[0]
 746          * instead of
 747          * SCE[0] CPE[0] SCE[1].
 748          * If we seem to have encountered such a stream, transfer
 749          * the SCE[1] element to the LFE[0]'s mapping */
 750         if (ac->tags_mapped == tags_per_config[ac->oc[1].m4ac.chan_config] - 1 && (type == TYPE_LFE || type == TYPE_SCE)) {
 751             if (!ac->warned_remapping_once && (type != TYPE_SCE || elem_id != 1)) {
 752                 av_log(ac->avctx, AV_LOG_WARNING,
 753                    "This stream seems to incorrectly report its last channel as %s[%d], mapping to SCE[1]\n",
 754                    type == TYPE_SCE ? "SCE" : "LFE", elem_id);
 755                 ac->warned_remapping_once++;
 756             }
 757             ac->tags_mapped++;
 758             return ac->tag_che_map[type][elem_id] = ac->che[TYPE_SCE][1];
 759         }
 760         if (ac->tags_mapped == 2 &&
 761             ac->oc[1].m4ac.chan_config == 4 &&
 762             type == TYPE_SCE) {
 763             ac->tags_mapped++;
 764             return ac->tag_che_map[TYPE_SCE][elem_id] = ac->che[TYPE_SCE][1];
 765         }
 766     case 3:
 767     case 2:
 768         if (ac->tags_mapped == (ac->oc[1].m4ac.chan_config != 2) &&
 769             type == TYPE_CPE) {
 770             ac->tags_mapped++;
 771             return ac->tag_che_map[TYPE_CPE][elem_id] = ac->che[TYPE_CPE][0];
 772         } else if (ac->oc[1].m4ac.chan_config == 2) {
 773             return NULL;
 774         }
 775     case 1:
 776         if (!ac->tags_mapped && type == TYPE_SCE) {
 777             ac->tags_mapped++;
 778             return ac->tag_che_map[TYPE_SCE][elem_id] = ac->che[TYPE_SCE][0];
 779         }
 780     default:
 781         return NULL;
 782     }
 783 }
 784
 785 /**
 786  * Decode an array of 4 bit element IDs, optionally interleaved with a
 787  * stereo/mono switching bit.
 788  *
 789  * @param type speaker type/position for these channels
 790  */
 791 static void decode_channel_map(uint8_t layout_map[][3],
 792                                enum ChannelPosition type,
 793                                GetBitContext *gb, int n)
 794 {
 795     while (n--) {
 796         enum RawDataBlockType syn_ele;
 797         switch (type) {
 798         case AAC_CHANNEL_FRONT:
 799         case AAC_CHANNEL_BACK:
 800         case AAC_CHANNEL_SIDE:
 801             syn_ele = get_bits1(gb);
 802             break;
 803         case AAC_CHANNEL_CC:
 804             skip_bits1(gb);
 805             syn_ele = TYPE_CCE;
 806             break;
 807         case AAC_CHANNEL_LFE:
 808             syn_ele = TYPE_LFE;
 809             break;
 810         default:
 811             // AAC_CHANNEL_OFF has no channel map
 812             av_assert0(0);
 813         }
 814         layout_map[0][0] = syn_ele;
 815         layout_map[0][1] = get_bits(gb, 4);
 816         layout_map[0][2] = type;
 817         layout_map++;
 818     }
 819 }
 820
 821 static inline void relative_align_get_bits(GetBitContext *gb,
 822                                            int reference_position) {
 823     int n = (reference_position - get_bits_count(gb) & 7);
 824     if (n)
 825         skip_bits(gb, n);
 826 }
 827
 828 /**
 829  * Decode program configuration element; reference: table 4.2.
 830  *
 831  * @return  Returns error status. 0 - OK, !0 - error
 832  */
 833 static int decode_pce(AVCodecContext *avctx, MPEG4AudioConfig *m4ac,
 834                       uint8_t (*layout_map)[3],
 835                       GetBitContext *gb, int byte_align_ref)
 836 {
 837     int num_front, num_side, num_back, num_lfe, num_assoc_data, num_cc;
 838     int sampling_index;
 839     int comment_len;
 840     int tags;
 841
 842     skip_bits(gb, 2);  // object_type
 843
 844     sampling_index = get_bits(gb, 4);
 845     if (m4ac->sampling_index != sampling_index)
 846         av_log(avctx, AV_LOG_WARNING,
 847                "Sample rate index in program config element does not "
 848                "match the sample rate index configured by the container.\n");
 849
 850     num_front       = get_bits(gb, 4);
 851     num_side        = get_bits(gb, 4);
 852     num_back        = get_bits(gb, 4);
 853     num_lfe         = get_bits(gb, 2);
 854     num_assoc_data  = get_bits(gb, 3);
 855     num_cc          = get_bits(gb, 4);
 856
 857     if (get_bits1(gb))
 858         skip_bits(gb, 4); // mono_mixdown_tag
 859     if (get_bits1(gb))
 860         skip_bits(gb, 4); // stereo_mixdown_tag
 861
 862     if (get_bits1(gb))
 863         skip_bits(gb, 3); // mixdown_coeff_index and pseudo_surround
 864
 865     if (get_bits_left(gb) < 5 * (num_front + num_side + num_back + num_cc) + 4 *(num_lfe + num_assoc_data + num_cc)) {
 866         av_log(avctx, AV_LOG_ERROR, "decode_pce: " overread_err);
 867         return -1;
 868     }
 869     decode_channel_map(layout_map       , AAC_CHANNEL_FRONT, gb, num_front);
 870     tags = num_front;
 871     decode_channel_map(layout_map + tags, AAC_CHANNEL_SIDE,  gb, num_side);
 872     tags += num_side;
 873     decode_channel_map(layout_map + tags, AAC_CHANNEL_BACK,  gb, num_back);
 874     tags += num_back;
 875     decode_channel_map(layout_map + tags, AAC_CHANNEL_LFE,   gb, num_lfe);
 876     tags += num_lfe;
 877
 878     skip_bits_long(gb, 4 * num_assoc_data);
 879
 880     decode_channel_map(layout_map + tags, AAC_CHANNEL_CC,    gb, num_cc);
 881     tags += num_cc;
 882
 883     relative_align_get_bits(gb, byte_align_ref);
 884
 885     /* comment field, first byte is length */
 886     comment_len = get_bits(gb, 8) * 8;
 887     if (get_bits_left(gb) < comment_len) {
 888         av_log(avctx, AV_LOG_ERROR, "decode_pce: " overread_err);
 889         return AVERROR_INVALIDDATA;
 890     }
 891     skip_bits_long(gb, comment_len);
 892     return tags;
 893 }
 894
 895 /**
 896  * Decode GA "General Audio" specific configuration; reference: table 4.1.
 897  *
 898  * @param   ac          pointer to AACContext, may be null
 899  * @param   avctx       pointer to AVCCodecContext, used for logging
 900  *
 901  * @return  Returns error status. 0 - OK, !0 - error
 902  */
 903 static int decode_ga_specific_config(AACContext *ac, AVCodecContext *avctx,
 904                                      GetBitContext *gb,
 905                                      int get_bit_alignment,
 906                                      MPEG4AudioConfig *m4ac,
 907                                      int channel_config)
 908 {
 909     int extension_flag, ret, ep_config, res_flags;
 910     uint8_t layout_map[MAX_ELEM_ID*4][3];
 911     int tags = 0;
 912
 913 #if USE_FIXED
 914     if (get_bits1(gb)) { // frameLengthFlag
 915         avpriv_report_missing_feature(avctx, "Fixed point 960/120 MDCT window");
 916         return AVERROR_PATCHWELCOME;
 917     }
 918     m4ac->frame_length_short = 0;
 919 #else
 920     m4ac->frame_length_short = get_bits1(gb);
 921     if (m4ac->frame_length_short && m4ac->sbr == 1) {
 922       avpriv_report_missing_feature(avctx, "SBR with 960 frame length");
 923       if (ac) ac->warned_960_sbr = 1;
 924       m4ac->sbr = 0;
 925       m4ac->ps = 0;
 926     }
 927 #endif
 928
 929     if (get_bits1(gb))       // dependsOnCoreCoder
 930         skip_bits(gb, 14);   // coreCoderDelay
 931     extension_flag = get_bits1(gb);
 932
 933     if (m4ac->object_type == AOT_AAC_SCALABLE ||
 934         m4ac->object_type == AOT_ER_AAC_SCALABLE)
 935         skip_bits(gb, 3);     // layerNr
 936
 937     if (channel_config == 0) {
 938         skip_bits(gb, 4);  // element_instance_tag
 939         tags = decode_pce(avctx, m4ac, layout_map, gb, get_bit_alignment);
 940         if (tags < 0)
 941             return tags;
 942     } else {
 943         if ((ret = set_default_channel_config(ac, avctx, layout_map,
 944                                               &tags, channel_config)))
 945             return ret;
 946     }
 947
 948     if (count_channels(layout_map, tags) > 1) {
 949         m4ac->ps = 0;
 950     } else if (m4ac->sbr == 1 && m4ac->ps == -1)
 951         m4ac->ps = 1;
 952
 953     if (ac && (ret = output_configure(ac, layout_map, tags, OC_GLOBAL_HDR, 0)))
 954         return ret;
 955
 956     if (extension_flag) {
 957         switch (m4ac->object_type) {
 958         case AOT_ER_BSAC:
 959             skip_bits(gb, 5);    // numOfSubFrame
 960             skip_bits(gb, 11);   // layer_length
 961             break;
 962         case AOT_ER_AAC_LC:
 963         case AOT_ER_AAC_LTP:
 964         case AOT_ER_AAC_SCALABLE:
 965         case AOT_ER_AAC_LD:
 966             res_flags = get_bits(gb, 3);
 967             if (res_flags) {
 968                 avpriv_report_missing_feature(avctx,
 969                                               "AAC data resilience (flags %x)",
 970                                               res_flags);
 971                 return AVERROR_PATCHWELCOME;
 972             }
 973             break;
 974         }
 975         skip_bits1(gb);    // extensionFlag3 (TBD in version 3)
 976     }
 977     switch (m4ac->object_type) {
 978     case AOT_ER_AAC_LC:
 979     case AOT_ER_AAC_LTP:
 980     case AOT_ER_AAC_SCALABLE:
 981     case AOT_ER_AAC_LD:
 982         ep_config = get_bits(gb, 2);
 983         if (ep_config) {
 984             avpriv_report_missing_feature(avctx,
 985                                           "epConfig %d", ep_config);
 986             return AVERROR_PATCHWELCOME;
 987         }
 988     }
 989     return 0;
 990 }
 991
 992 static int decode_eld_specific_config(AACContext *ac, AVCodecContext *avctx,
 993                                      GetBitContext *gb,
 994                                      MPEG4AudioConfig *m4ac,
 995                                      int channel_config)
 996 {
 997     int ret, ep_config, res_flags;
 998     uint8_t layout_map[MAX_ELEM_ID*4][3];
 999     int tags = 0;
1000     const int ELDEXT_TERM = 0;
1001
1002     m4ac->ps  = 0;
1003     m4ac->sbr = 0;
1004 #if USE_FIXED
1005     if (get_bits1(gb)) { // frameLengthFlag
1006         avpriv_request_sample(avctx, "960/120 MDCT window");
1007         return AVERROR_PATCHWELCOME;
1008     }
1009 #else
1010     m4ac->frame_length_short = get_bits1(gb);
1011 #endif
1012     res_flags = get_bits(gb, 3);
1013     if (res_flags) {
1014         avpriv_report_missing_feature(avctx,
1015                                       "AAC data resilience (flags %x)",
1016                                       res_flags);
1017         return AVERROR_PATCHWELCOME;
1018     }
1019
1020     if (get_bits1(gb)) { // ldSbrPresentFlag
1021         avpriv_report_missing_feature(avctx,
1022                                       "Low Delay SBR");
1023         return AVERROR_PATCHWELCOME;
1024     }
1025
1026     while (get_bits(gb, 4) != ELDEXT_TERM) {
1027         int len = get_bits(gb, 4);
1028         if (len == 15)
1029             len += get_bits(gb, 8);
1030         if (len == 15 + 255)
1031             len += get_bits(gb, 16);
1032         if (get_bits_left(gb) < len * 8 + 4) {
1033             av_log(avctx, AV_LOG_ERROR, overread_err);
1034             return AVERROR_INVALIDDATA;
1035         }
1036         skip_bits_long(gb, 8 * len);
1037     }
1038
1039     if ((ret = set_default_channel_config(ac, avctx, layout_map,
1040                                           &tags, channel_config)))
1041         return ret;
1042
1043     if (ac && (ret = output_configure(ac, layout_map, tags, OC_GLOBAL_HDR, 0)))
1044         return ret;
1045
1046     ep_config = get_bits(gb, 2);
1047     if (ep_config) {
1048         avpriv_report_missing_feature(avctx,
1049                                       "epConfig %d", ep_config);
1050         return AVERROR_PATCHWELCOME;
1051     }
1052     return 0;
1053 }
1054
1055 /**
1056  * Decode audio specific configuration; reference: table 1.13.
1057  *
1058  * @param   ac          pointer to AACContext, may be null
1059  * @param   avctx       pointer to AVCCodecContext, used for logging
1060  * @param   m4ac        pointer to MPEG4AudioConfig, used for parsing
1061  * @param   gb          buffer holding an audio specific config
1062  * @param   get_bit_alignment relative alignment for byte align operations
1063  * @param   sync_extension look for an appended sync extension
1064  *
1065  * @return  Returns error status or number of consumed bits. <0 - error
1066  */
1067 static int decode_audio_specific_config_gb(AACContext *ac,
1068                                            AVCodecContext *avctx,
1069                                            MPEG4AudioConfig *m4ac,
1070                                            GetBitContext *gb,
1071                                            int get_bit_alignment,
1072                                            int sync_extension)
1073 {
1074     int i, ret;
1075     GetBitContext gbc = *gb;
1076
1077     if ((i = ff_mpeg4audio_get_config_gb(m4ac, &gbc, sync_extension, avctx)) < 0)
1078         return AVERROR_INVALIDDATA;
1079
1080     if (m4ac->sampling_index > 12) {
1081         av_log(avctx, AV_LOG_ERROR,
1082                "invalid sampling rate index %d\n",
1083                m4ac->sampling_index);
1084         return AVERROR_INVALIDDATA;
1085     }
1086     if (m4ac->object_type == AOT_ER_AAC_LD &&
1087         (m4ac->sampling_index < 3 || m4ac->sampling_index > 7)) {
1088         av_log(avctx, AV_LOG_ERROR,
1089                "invalid low delay sampling rate index %d\n",
1090                m4ac->sampling_index);
1091         return AVERROR_INVALIDDATA;
1092     }
1093
1094     skip_bits_long(gb, i);
1095
1096     switch (m4ac->object_type) {
1097     case AOT_AAC_MAIN:
1098     case AOT_AAC_LC:
1099     case AOT_AAC_SSR:
1100     case AOT_AAC_LTP:
1101     case AOT_ER_AAC_LC:
1102     case AOT_ER_AAC_LD:
1103         if ((ret = decode_ga_specific_config(ac, avctx, gb, get_bit_alignment,
1104                                             m4ac, m4ac->chan_config)) < 0)
1105             return ret;
1106         break;
1107     case AOT_ER_AAC_ELD:
1108         if ((ret = decode_eld_specific_config(ac, avctx, gb,
1109                                               m4ac, m4ac->chan_config)) < 0)
1110             return ret;
1111         break;
1112     default:
1113         avpriv_report_missing_feature(avctx,
1114                                       "Audio object type %s%d",
1115                                       m4ac->sbr == 1 ? "SBR+" : "",
1116                                       m4ac->object_type);
1117         return AVERROR(ENOSYS);
1118     }
1119
1120     ff_dlog(avctx,
1121             "AOT %d chan config %d sampling index %d (%d) SBR %d PS %d\n",
1122             m4ac->object_type, m4ac->chan_config, m4ac->sampling_index,
1123             m4ac->sample_rate, m4ac->sbr,
1124             m4ac->ps);
1125
1126     return get_bits_count(gb);
1127 }
1128
1129 static int decode_audio_specific_config(AACContext *ac,
1130                                         AVCodecContext *avctx,
1131                                         MPEG4AudioConfig *m4ac,
1132                                         const uint8_t *data, int64_t bit_size,
1133                                         int sync_extension)
1134 {
1135     int i, ret;
1136     GetBitContext gb;
1137
1138     if (bit_size < 0 || bit_size > INT_MAX) {
1139         av_log(avctx, AV_LOG_ERROR, "Audio specific config size is invalid\n");
1140         return AVERROR_INVALIDDATA;
1141     }
1142
1143     ff_dlog(avctx, "audio specific config size %d\n", (int)bit_size >> 3);
1144     for (i = 0; i < bit_size >> 3; i++)
1145         ff_dlog(avctx, "%02x ", data[i]);
1146     ff_dlog(avctx, "\n");
1147
1148     if ((ret = init_get_bits(&gb, data, bit_size)) < 0)
1149         return ret;
1150
1151     return decode_audio_specific_config_gb(ac, avctx, m4ac, &gb, 0,
1152                                            sync_extension);
1153 }
1154
1155 /**
1156  * linear congruential pseudorandom number generator
1157  *
1158  * @param   previous_val    pointer to the current state of the generator
1159  *
1160  * @return  Returns a 32-bit pseudorandom integer
1161  */
1162 static av_always_inline int lcg_random(unsigned previous_val)
1163 {
1164     union { unsigned u; int s; } v = { previous_val * 1664525u + 1013904223 };
1165     return v.s;
1166 }
1167
1168 static void reset_all_predictors(PredictorState *ps)
1169 {
1170     int i;
1171     for (i = 0; i < MAX_PREDICTORS; i++)
1172         reset_predict_state(&ps[i]);
1173 }
1174
1175 static int sample_rate_idx (int rate)
1176 {
1177          if (92017 <= rate) return 0;
1178     else if (75132 <= rate) return 1;
1179     else if (55426 <= rate) return 2;
1180     else if (46009 <= rate) return 3;
1181     else if (37566 <= rate) return 4;
1182     else if (27713 <= rate) return 5;
1183     else if (23004 <= rate) return 6;
1184     else if (18783 <= rate) return 7;
1185     else if (13856 <= rate) return 8;
1186     else if (11502 <= rate) return 9;
1187     else if (9391  <= rate) return 10;
1188     else                    return 11;
1189 }
1190
1191 static void reset_predictor_group(PredictorState *ps, int group_num)
1192 {
1193     int i;
1194     for (i = group_num - 1; i < MAX_PREDICTORS; i += 30)
1195         reset_predict_state(&ps[i]);
1196 }
1197
1198 static void aacdec_init(AACContext *ac);
1199
1200 static av_cold void aac_static_table_init(void)
1201 {
1202     static VLC_TYPE vlc_buf[304 + 270 + 550 + 300 + 328 +
1203                             294 + 306 + 268 + 510 + 366 + 462][2];
1204     for (unsigned i = 0, offset = 0; i < 11; i++) {
1205         vlc_spectral[i].table           = &vlc_buf[offset];
1206         vlc_spectral[i].table_allocated = FF_ARRAY_ELEMS(vlc_buf) - offset;
1207         ff_init_vlc_sparse(&vlc_spectral[i], 8, ff_aac_spectral_sizes[i],
1208                            ff_aac_spectral_bits[i],       sizeof(ff_aac_spectral_bits[i][0]),
1209                                                           sizeof(ff_aac_spectral_bits[i][0]),
1210                            ff_aac_spectral_codes[i],      sizeof(ff_aac_spectral_codes[i][0]),
1211                                                           sizeof(ff_aac_spectral_codes[i][0]),
1212                            ff_aac_codebook_vector_idx[i], sizeof(ff_aac_codebook_vector_idx[i][0]),
1213                                                           sizeof(ff_aac_codebook_vector_idx[i][0]),
1214                  INIT_VLC_STATIC_OVERLONG);
1215         offset += vlc_spectral[i].table_size;
1216     }
1217
1218     AAC_RENAME(ff_aac_sbr_init)();
1219
1220     ff_aac_tableinit();
1221
1222     INIT_VLC_STATIC(&vlc_scalefactors, 7,
1223                     FF_ARRAY_ELEMS(ff_aac_scalefactor_code),
1224                     ff_aac_scalefactor_bits,
1225                     sizeof(ff_aac_scalefactor_bits[0]),
1226                     sizeof(ff_aac_scalefactor_bits[0]),
1227                     ff_aac_scalefactor_code,
1228                     sizeof(ff_aac_scalefactor_code[0]),
1229                     sizeof(ff_aac_scalefactor_code[0]),
1230                     352);
1231
1232     // window initialization
1233 #if !USE_FIXED
1234     AAC_RENAME(ff_kbd_window_init)(AAC_RENAME(aac_kbd_long_960), 4.0, 960);
1235     AAC_RENAME(ff_kbd_window_init)(AAC_RENAME(aac_kbd_short_120), 6.0, 120);
1236     AAC_RENAME(ff_sine_window_init)(AAC_RENAME(sine_960), 960);
1237     AAC_RENAME(ff_sine_window_init)(AAC_RENAME(sine_120), 120);
1238     AAC_RENAME(ff_init_ff_sine_windows)(9);
1239     ff_aac_float_common_init();
1240 #else
1241     AAC_RENAME(ff_kbd_window_init)(AAC_RENAME2(aac_kbd_long_1024), 4.0, 1024);
1242     AAC_RENAME(ff_kbd_window_init)(AAC_RENAME2(aac_kbd_short_128), 6.0, 128);
1243     init_sine_windows_fixed();
1244 #endif
1245
1246     AAC_RENAME(ff_cbrt_tableinit)();
1247 }
1248
1249 static AVOnce aac_table_init = AV_ONCE_INIT;
1250
1251 static av_cold int aac_decode_init(AVCodecContext *avctx)
1252 {
1253     AACContext *ac = avctx->priv_data;
1254     int ret;
1255
1256     if (avctx->sample_rate > 96000)
1257         return AVERROR_INVALIDDATA;
1258
1259     ret = ff_thread_once(&aac_table_init, &aac_static_table_init);
1260     if (ret != 0)
1261         return AVERROR_UNKNOWN;
1262
1263     ac->avctx = avctx;
1264     ac->oc[1].m4ac.sample_rate = avctx->sample_rate;
1265
1266     aacdec_init(ac);
1267 #if USE_FIXED
1268     avctx->sample_fmt = AV_SAMPLE_FMT_S32P;
1269 #else
1270     avctx->sample_fmt = AV_SAMPLE_FMT_FLTP;
1271 #endif /* USE_FIXED */
1272
1273     if (avctx->extradata_size > 0) {
1274         if ((ret = decode_audio_specific_config(ac, ac->avctx, &ac->oc[1].m4ac,
1275                                                 avctx->extradata,
1276                                                 avctx->extradata_size * 8LL,
1277                                                 1)) < 0)
1278             return ret;
1279     } else {
1280         int sr, i;
1281         uint8_t layout_map[MAX_ELEM_ID*4][3];
1282         int layout_map_tags;
1283
1284         sr = sample_rate_idx(avctx->sample_rate);
1285         ac->oc[1].m4ac.sampling_index = sr;
1286         ac->oc[1].m4ac.channels = avctx->channels;
1287         ac->oc[1].m4ac.sbr = -1;
1288         ac->oc[1].m4ac.ps = -1;
1289
1290         for (i = 0; i < FF_ARRAY_ELEMS(ff_mpeg4audio_channels); i++)
1291             if (ff_mpeg4audio_channels[i] == avctx->channels)
1292                 break;
1293         if (i == FF_ARRAY_ELEMS(ff_mpeg4audio_channels)) {
1294             i = 0;
1295         }
1296         ac->oc[1].m4ac.chan_config = i;
1297
1298         if (ac->oc[1].m4ac.chan_config) {
1299             int ret = set_default_channel_config(ac, avctx, layout_map,
1300                 &layout_map_tags, ac->oc[1].m4ac.chan_config);
1301             if (!ret)
1302                 output_configure(ac, layout_map, layout_map_tags,
1303                                  OC_GLOBAL_HDR, 0);
1304             else if (avctx->err_recognition & AV_EF_EXPLODE)
1305                 return AVERROR_INVALIDDATA;
1306         }
1307     }
1308
1309     if (avctx->channels > MAX_CHANNELS) {
1310         av_log(avctx, AV_LOG_ERROR, "Too many channels\n");
1311         return AVERROR_INVALIDDATA;
1312     }
1313
1314 #if USE_FIXED
1315     ac->fdsp = avpriv_alloc_fixed_dsp(avctx->flags & AV_CODEC_FLAG_BITEXACT);
1316 #else
1317     ac->fdsp = avpriv_float_dsp_alloc(avctx->flags & AV_CODEC_FLAG_BITEXACT);
1318 #endif /* USE_FIXED */
1319     if (!ac->fdsp) {
1320         return AVERROR(ENOMEM);
1321     }
1322
1323     ac->random_state = 0x1f2e3d4c;
1324
1325     AAC_RENAME_32(ff_mdct_init)(&ac->mdct,       11, 1, 1.0 / RANGE15(1024.0));
1326     AAC_RENAME_32(ff_mdct_init)(&ac->mdct_ld,    10, 1, 1.0 / RANGE15(512.0));
1327     AAC_RENAME_32(ff_mdct_init)(&ac->mdct_small,  8, 1, 1.0 / RANGE15(128.0));
1328     AAC_RENAME_32(ff_mdct_init)(&ac->mdct_ltp,   11, 0, RANGE15(-2.0));
1329 #if !USE_FIXED
1330     ret = ff_mdct15_init(&ac->mdct120, 1, 3, 1.0f/(16*1024*120*2));
1331     if (ret < 0)
1332         return ret;
1333     ret = ff_mdct15_init(&ac->mdct480, 1, 5, 1.0f/(16*1024*960));
1334     if (ret < 0)
1335         return ret;
1336     ret = ff_mdct15_init(&ac->mdct960, 1, 6, 1.0f/(16*1024*960*2));
1337     if (ret < 0)
1338         return ret;
1339 #endif
1340
1341     return 0;
1342 }
1343
1344 /**
1345  * Skip data_stream_element; reference: table 4.10.
1346  */
1347 static int skip_data_stream_element(AACContext *ac, GetBitContext *gb)
1348 {
1349     int byte_align = get_bits1(gb);
1350     int count = get_bits(gb, 8);
1351     if (count == 255)
1352         count += get_bits(gb, 8);
1353     if (byte_align)
1354         align_get_bits(gb);
1355
1356     if (get_bits_left(gb) < 8 * count) {
1357         av_log(ac->avctx, AV_LOG_ERROR, "skip_data_stream_element: "overread_err);
1358         return AVERROR_INVALIDDATA;
1359     }
1360     skip_bits_long(gb, 8 * count);
1361     return 0;
1362 }
1363
1364 static int decode_prediction(AACContext *ac, IndividualChannelStream *ics,
1365                              GetBitContext *gb)
1366 {
1367     int sfb;
1368     if (get_bits1(gb)) {
1369         ics->predictor_reset_group = get_bits(gb, 5);
1370         if (ics->predictor_reset_group == 0 ||
1371             ics->predictor_reset_group > 30) {
1372             av_log(ac->avctx, AV_LOG_ERROR,
1373                    "Invalid Predictor Reset Group.\n");
1374             return AVERROR_INVALIDDATA;
1375         }
1376     }
1377     for (sfb = 0; sfb < FFMIN(ics->max_sfb, ff_aac_pred_sfb_max[ac->oc[1].m4ac.sampling_index]); sfb++) {
1378         ics->prediction_used[sfb] = get_bits1(gb);
1379     }
1380     return 0;
1381 }
1382
1383 /**
1384  * Decode Long Term Prediction data; reference: table 4.xx.
1385  */
1386 static void decode_ltp(LongTermPrediction *ltp,
1387                        GetBitContext *gb, uint8_t max_sfb)
1388 {
1389     int sfb;
1390
1391     ltp->lag  = get_bits(gb, 11);
1392     ltp->coef = ltp_coef[get_bits(gb, 3)];
1393     for (sfb = 0; sfb < FFMIN(max_sfb, MAX_LTP_LONG_SFB); sfb++)
1394         ltp->used[sfb] = get_bits1(gb);
1395 }
1396
1397 /**
1398  * Decode Individual Channel Stream info; reference: table 4.6.
1399  */
1400 static int decode_ics_info(AACContext *ac, IndividualChannelStream *ics,
1401                            GetBitContext *gb)
1402 {
1403     const MPEG4AudioConfig *const m4ac = &ac->oc[1].m4ac;
1404     const int aot = m4ac->object_type;
1405     const int sampling_index = m4ac->sampling_index;
1406     int ret_fail = AVERROR_INVALIDDATA;
1407
1408     if (aot != AOT_ER_AAC_ELD) {
1409         if (get_bits1(gb)) {
1410             av_log(ac->avctx, AV_LOG_ERROR, "Reserved bit set.\n");
1411             if (ac->avctx->err_recognition & AV_EF_BITSTREAM)
1412                 return AVERROR_INVALIDDATA;
1413         }
1414         ics->window_sequence[1] = ics->window_sequence[0];
1415         ics->window_sequence[0] = get_bits(gb, 2);
1416         if (aot == AOT_ER_AAC_LD &&
1417             ics->window_sequence[0] != ONLY_LONG_SEQUENCE) {
1418             av_log(ac->avctx, AV_LOG_ERROR,
1419                    "AAC LD is only defined for ONLY_LONG_SEQUENCE but "
1420                    "window sequence %d found.\n", ics->window_sequence[0]);
1421             ics->window_sequence[0] = ONLY_LONG_SEQUENCE;
1422             return AVERROR_INVALIDDATA;
1423         }
1424         ics->use_kb_window[1]   = ics->use_kb_window[0];
1425         ics->use_kb_window[0]   = get_bits1(gb);
1426     }
1427     ics->num_window_groups  = 1;
1428     ics->group_len[0]       = 1;
1429     if (ics->window_sequence[0] == EIGHT_SHORT_SEQUENCE) {
1430         int i;
1431         ics->max_sfb = get_bits(gb, 4);
1432         for (i = 0; i < 7; i++) {
1433             if (get_bits1(gb)) {
1434                 ics->group_len[ics->num_window_groups - 1]++;
1435             } else {
1436                 ics->num_window_groups++;
1437                 ics->group_len[ics->num_window_groups - 1] = 1;
1438             }
1439         }
1440         ics->num_windows       = 8;
1441         if (m4ac->frame_length_short) {
1442             ics->swb_offset    =  ff_swb_offset_120[sampling_index];
1443             ics->num_swb       = ff_aac_num_swb_120[sampling_index];
1444         } else {
1445             ics->swb_offset    =  ff_swb_offset_128[sampling_index];
1446             ics->num_swb       = ff_aac_num_swb_128[sampling_index];
1447         }
1448         ics->tns_max_bands     = ff_tns_max_bands_128[sampling_index];
1449         ics->predictor_present = 0;
1450     } else {
1451         ics->max_sfb           = get_bits(gb, 6);
1452         ics->num_windows       = 1;
1453         if (aot == AOT_ER_AAC_LD || aot == AOT_ER_AAC_ELD) {
1454             if (m4ac->frame_length_short) {
1455                 ics->swb_offset    =     ff_swb_offset_480[sampling_index];
1456                 ics->num_swb       =    ff_aac_num_swb_480[sampling_index];
1457                 ics->tns_max_bands =  ff_tns_max_bands_480[sampling_index];
1458             } else {
1459                 ics->swb_offset    =     ff_swb_offset_512[sampling_index];
1460                 ics->num_swb       =    ff_aac_num_swb_512[sampling_index];
1461                 ics->tns_max_bands =  ff_tns_max_bands_512[sampling_index];
1462             }
1463             if (!ics->num_swb || !ics->swb_offset) {
1464                 ret_fail = AVERROR_BUG;
1465                 goto fail;
1466             }
1467         } else {
1468             if (m4ac->frame_length_short) {
1469                 ics->num_swb    = ff_aac_num_swb_960[sampling_index];
1470                 ics->swb_offset = ff_swb_offset_960[sampling_index];
1471             } else {
1472                 ics->num_swb    = ff_aac_num_swb_1024[sampling_index];
1473                 ics->swb_offset = ff_swb_offset_1024[sampling_index];
1474             }
1475             ics->tns_max_bands = ff_tns_max_bands_1024[sampling_index];
1476         }
1477         if (aot != AOT_ER_AAC_ELD) {
1478             ics->predictor_present     = get_bits1(gb);
1479             ics->predictor_reset_group = 0;
1480         }
1481         if (ics->predictor_present) {
1482             if (aot == AOT_AAC_MAIN) {
1483                 if (decode_prediction(ac, ics, gb)) {
1484                     goto fail;
1485                 }
1486             } else if (aot == AOT_AAC_LC ||
1487                        aot == AOT_ER_AAC_LC) {
1488                 av_log(ac->avctx, AV_LOG_ERROR,
1489                        "Prediction is not allowed in AAC-LC.\n");
1490                 goto fail;
1491             } else {
1492                 if (aot == AOT_ER_AAC_LD) {
1493                     av_log(ac->avctx, AV_LOG_ERROR,
1494                            "LTP in ER AAC LD not yet implemented.\n");
1495                     ret_fail = AVERROR_PATCHWELCOME;
1496                     goto fail;
1497                 }
1498                 if ((ics->ltp.present = get_bits(gb, 1)))
1499                     decode_ltp(&ics->ltp, gb, ics->max_sfb);
1500             }
1501         }
1502     }
1503
1504     if (ics->max_sfb > ics->num_swb) {
1505         av_log(ac->avctx, AV_LOG_ERROR,
1506                "Number of scalefactor bands in group (%d) "
1507                "exceeds limit (%d).\n",
1508                ics->max_sfb, ics->num_swb);
1509         goto fail;
1510     }
1511
1512     return 0;
1513 fail:
1514     ics->max_sfb = 0;
1515     return ret_fail;
1516 }
1517
1518 /**
1519  * Decode band types (section_data payload); reference: table 4.46.
1520  *
1521  * @param   band_type           array of the used band type
1522  * @param   band_type_run_end   array of the last scalefactor band of a band type run
1523  *
1524  * @return  Returns error status. 0 - OK, !0 - error
1525  */
1526 static int decode_band_types(AACContext *ac, enum BandType band_type[120],
1527                              int band_type_run_end[120], GetBitContext *gb,
1528                              IndividualChannelStream *ics)
1529 {
1530     int g, idx = 0;
1531     const int bits = (ics->window_sequence[0] == EIGHT_SHORT_SEQUENCE) ? 3 : 5;
1532     for (g = 0; g < ics->num_window_groups; g++) {
1533         int k = 0;
1534         while (k < ics->max_sfb) {
1535             uint8_t sect_end = k;
1536             int sect_len_incr;
1537             int sect_band_type = get_bits(gb, 4);
1538             if (sect_band_type == 12) {
1539                 av_log(ac->avctx, AV_LOG_ERROR, "invalid band type\n");
1540                 return AVERROR_INVALIDDATA;
1541             }
1542             do {
1543                 sect_len_incr = get_bits(gb, bits);
1544                 sect_end += sect_len_incr;
1545                 if (get_bits_left(gb) < 0) {
1546                     av_log(ac->avctx, AV_LOG_ERROR, "decode_band_types: "overread_err);
1547                     return AVERROR_INVALIDDATA;
1548                 }
1549                 if (sect_end > ics->max_sfb) {
1550                     av_log(ac->avctx, AV_LOG_ERROR,
1551                            "Number of bands (%d) exceeds limit (%d).\n",
1552                            sect_end, ics->max_sfb);
1553                     return AVERROR_INVALIDDATA;
1554                 }
1555             } while (sect_len_incr == (1 << bits) - 1);
1556             for (; k < sect_end; k++) {
1557                 band_type        [idx]   = sect_band_type;
1558                 band_type_run_end[idx++] = sect_end;
1559             }
1560         }
1561     }
1562     return 0;
1563 }
1564
1565 /**
1566  * Decode scalefactors; reference: table 4.47.
1567  *
1568  * @param   global_gain         first scalefactor value as scalefactors are differentially coded
1569  * @param   band_type           array of the used band type
1570  * @param   band_type_run_end   array of the last scalefactor band of a band type run
1571  * @param   sf                  array of scalefactors or intensity stereo positions
1572  *
1573  * @return  Returns error status. 0 - OK, !0 - error
1574  */
1575 static int decode_scalefactors(AACContext *ac, INTFLOAT sf[120], GetBitContext *gb,
1576                                unsigned int global_gain,
1577                                IndividualChannelStream *ics,
1578                                enum BandType band_type[120],
1579                                int band_type_run_end[120])
1580 {
1581     int g, i, idx = 0;
1582     int offset[3] = { global_gain, global_gain - NOISE_OFFSET, 0 };
1583     int clipped_offset;
1584     int noise_flag = 1;
1585     for (g = 0; g < ics->num_window_groups; g++) {
1586         for (i = 0; i < ics->max_sfb;) {
1587             int run_end = band_type_run_end[idx];
1588             if (band_type[idx] == ZERO_BT) {
1589                 for (; i < run_end; i++, idx++)
1590                     sf[idx] = FIXR(0.);
1591             } else if ((band_type[idx] == INTENSITY_BT) ||
1592                        (band_type[idx] == INTENSITY_BT2)) {
1593                 for (; i < run_end; i++, idx++) {
1594                     offset[2] += get_vlc2(gb, vlc_scalefactors.table, 7, 3) - SCALE_DIFF_ZERO;
1595                     clipped_offset = av_clip(offset[2], -155, 100);
1596                     if (offset[2] != clipped_offset) {
1597                         avpriv_request_sample(ac->avctx,
1598                                               "If you heard an audible artifact, there may be a bug in the decoder. "
1599                                               "Clipped intensity stereo position (%d -> %d)",
1600                                               offset[2], clipped_offset);
1601                     }
1602 #if USE_FIXED
1603                     sf[idx] = 100 - clipped_offset;
1604 #else
1605                     sf[idx] = ff_aac_pow2sf_tab[-clipped_offset + POW_SF2_ZERO];
1606 #endif /* USE_FIXED */
1607                 }
1608             } else if (band_type[idx] == NOISE_BT) {
1609                 for (; i < run_end; i++, idx++) {
1610                     if (noise_flag-- > 0)
1611                         offset[1] += get_bits(gb, NOISE_PRE_BITS) - NOISE_PRE;
1612                     else
1613                         offset[1] += get_vlc2(gb, vlc_scalefactors.table, 7, 3) - SCALE_DIFF_ZERO;
1614                     clipped_offset = av_clip(offset[1], -100, 155);
1615                     if (offset[1] != clipped_offset) {
1616                         avpriv_request_sample(ac->avctx,
1617                                               "If you heard an audible artifact, there may be a bug in the decoder. "
1618                                               "Clipped noise gain (%d -> %d)",
1619                                               offset[1], clipped_offset);
1620                     }
1621 #if USE_FIXED
1622                     sf[idx] = -(100 + clipped_offset);
1623 #else
1624                     sf[idx] = -ff_aac_pow2sf_tab[clipped_offset + POW_SF2_ZERO];
1625 #endif /* USE_FIXED */
1626                 }
1627             } else {
1628                 for (; i < run_end; i++, idx++) {
1629                     offset[0] += get_vlc2(gb, vlc_scalefactors.table, 7, 3) - SCALE_DIFF_ZERO;
1630                     if (offset[0] > 255U) {
1631                         av_log(ac->avctx, AV_LOG_ERROR,
1632                                "Scalefactor (%d) out of range.\n", offset[0]);
1633                         return AVERROR_INVALIDDATA;
1634                     }
1635 #if USE_FIXED
1636                     sf[idx] = -offset[0];
1637 #else
1638                     sf[idx] = -ff_aac_pow2sf_tab[offset[0] - 100 + POW_SF2_ZERO];
1639 #endif /* USE_FIXED */
1640                 }
1641             }
1642         }
1643     }
1644     return 0;
1645 }
1646
1647 /**
1648  * Decode pulse data; reference: table 4.7.
1649  */
1650 static int decode_pulses(Pulse *pulse, GetBitContext *gb,
1651                          const uint16_t *swb_offset, int num_swb)
1652 {
1653     int i, pulse_swb;
1654     pulse->num_pulse = get_bits(gb, 2) + 1;
1655     pulse_swb        = get_bits(gb, 6);
1656     if (pulse_swb >= num_swb)
1657         return -1;
1658     pulse->pos[0]    = swb_offset[pulse_swb];
1659     pulse->pos[0]   += get_bits(gb, 5);
1660     if (pulse->pos[0] >= swb_offset[num_swb])
1661         return -1;
1662     pulse->amp[0]    = get_bits(gb, 4);
1663     for (i = 1; i < pulse->num_pulse; i++) {
1664         pulse->pos[i] = get_bits(gb, 5) + pulse->pos[i - 1];
1665         if (pulse->pos[i] >= swb_offset[num_swb])
1666             return -1;
1667         pulse->amp[i] = get_bits(gb, 4);
1668     }
1669     return 0;
1670 }
1671
1672 /**
1673  * Decode Temporal Noise Shaping data; reference: table 4.48.
1674  *
1675  * @return  Returns error status. 0 - OK, !0 - error
1676  */
1677 static int decode_tns(AACContext *ac, TemporalNoiseShaping *tns,
1678                       GetBitContext *gb, const IndividualChannelStream *ics)
1679 {
1680     int w, filt, i, coef_len, coef_res, coef_compress;
1681     const int is8 = ics->window_sequence[0] == EIGHT_SHORT_SEQUENCE;
1682     const int tns_max_order = is8 ? 7 : ac->oc[1].m4ac.object_type == AOT_AAC_MAIN ? 20 : 12;
1683     for (w = 0; w < ics->num_windows; w++) {
1684         if ((tns->n_filt[w] = get_bits(gb, 2 - is8))) {
1685             coef_res = get_bits1(gb);
1686
1687             for (filt = 0; filt < tns->n_filt[w]; filt++) {
1688                 int tmp2_idx;
1689                 tns->length[w][filt] = get_bits(gb, 6 - 2 * is8);
1690
1691                 if ((tns->order[w][filt] = get_bits(gb, 5 - 2 * is8)) > tns_max_order) {
1692                     av_log(ac->avctx, AV_LOG_ERROR,
1693                            "TNS filter order %d is greater than maximum %d.\n",
1694                            tns->order[w][filt], tns_max_order);
1695                     tns->order[w][filt] = 0;
1696                     return AVERROR_INVALIDDATA;
1697                 }
1698                 if (tns->order[w][filt]) {
1699                     tns->direction[w][filt] = get_bits1(gb);
1700                     coef_compress = get_bits1(gb);
1701                     coef_len = coef_res + 3 - coef_compress;
1702                     tmp2_idx = 2 * coef_compress + coef_res;
1703
1704                     for (i = 0; i < tns->order[w][filt]; i++)
1705                         tns->coef[w][filt][i] = tns_tmp2_map[tmp2_idx][get_bits(gb, coef_len)];
1706                 }
1707             }
1708         }
1709     }
1710     return 0;
1711 }
1712
1713 /**
1714  * Decode Mid/Side data; reference: table 4.54.
1715  *
1716  * @param   ms_present  Indicates mid/side stereo presence. [0] mask is all 0s;
1717  *                      [1] mask is decoded from bitstream; [2] mask is all 1s;
1718  *                      [3] reserved for scalable AAC
1719  */
1720 static void decode_mid_side_stereo(ChannelElement *cpe, GetBitContext *gb,
1721                                    int ms_present)
1722 {
1723     int idx;
1724     int max_idx = cpe->ch[0].ics.num_window_groups * cpe->ch[0].ics.max_sfb;
1725     if (ms_present == 1) {
1726         for (idx = 0; idx < max_idx; idx++)
1727             cpe->ms_mask[idx] = get_bits1(gb);
1728     } else if (ms_present == 2) {
1729         memset(cpe->ms_mask, 1, max_idx * sizeof(cpe->ms_mask[0]));
1730     }
1731 }
1732
1733 /**
1734  * Decode spectral data; reference: table 4.50.
1735  * Dequantize and scale spectral data; reference: 4.6.3.3.
1736  *
1737  * @param   coef            array of dequantized, scaled spectral data
1738  * @param   sf              array of scalefactors or intensity stereo positions
1739  * @param   pulse_present   set if pulses are present
1740  * @param   pulse           pointer to pulse data struct
1741  * @param   band_type       array of the used band type
1742  *
1743  * @return  Returns error status. 0 - OK, !0 - error
1744  */
1745 static int decode_spectrum_and_dequant(AACContext *ac, INTFLOAT coef[1024],
1746                                        GetBitContext *gb, const INTFLOAT sf[120],
1747                                        int pulse_present, const Pulse *pulse,
1748                                        const IndividualChannelStream *ics,
1749                                        enum BandType band_type[120])
1750 {
1751     int i, k, g, idx = 0;
1752     const int c = 1024 / ics->num_windows;
1753     const uint16_t *offsets = ics->swb_offset;
1754     INTFLOAT *coef_base = coef;
1755
1756     for (g = 0; g < ics->num_windows; g++)
1757         memset(coef + g * 128 + offsets[ics->max_sfb], 0,
1758                sizeof(INTFLOAT) * (c - offsets[ics->max_sfb]));
1759
1760     for (g = 0; g < ics->num_window_groups; g++) {
1761         unsigned g_len = ics->group_len[g];
1762
1763         for (i = 0; i < ics->max_sfb; i++, idx++) {
1764             const unsigned cbt_m1 = band_type[idx] - 1;
1765             INTFLOAT *cfo = coef + offsets[i];
1766             int off_len = offsets[i + 1] - offsets[i];
1767             int group;
1768
1769             if (cbt_m1 >= INTENSITY_BT2 - 1) {
1770                 for (group = 0; group < (AAC_SIGNE)g_len; group++, cfo+=128) {
1771                     memset(cfo, 0, off_len * sizeof(*cfo));
1772                 }
1773             } else if (cbt_m1 == NOISE_BT - 1) {
1774                 for (group = 0; group < (AAC_SIGNE)g_len; group++, cfo+=128) {
1775                     INTFLOAT band_energy;
1776 #if USE_FIXED
1777                     for (k = 0; k < off_len; k++) {
1778                         ac->random_state  = lcg_random(ac->random_state);
1779                         cfo[k] = ac->random_state >> 3;
1780                     }
1781
1782                     band_energy = ac->fdsp->scalarproduct_fixed(cfo, cfo, off_len);
1783                     band_energy = fixed_sqrt(band_energy, 31);
1784                     noise_scale(cfo, sf[idx], band_energy, off_len);
1785 #else
1786                     float scale;
1787
1788                     for (k = 0; k < off_len; k++) {
1789                         ac->random_state  = lcg_random(ac->random_state);
1790                         cfo[k] = ac->random_state;
1791                     }
1792
1793                     band_energy = ac->fdsp->scalarproduct_float(cfo, cfo, off_len);
1794                     scale = sf[idx] / sqrtf(band_energy);
1795                     ac->fdsp->vector_fmul_scalar(cfo, cfo, scale, off_len);
1796 #endif /* USE_FIXED */
1797                 }
1798             } else {
1799 #if !USE_FIXED
1800                 const float *vq = ff_aac_codebook_vector_vals[cbt_m1];
1801 #endif /* !USE_FIXED */
1802                 VLC_TYPE (*vlc_tab)[2] = vlc_spectral[cbt_m1].table;
1803                 OPEN_READER(re, gb);
1804
1805                 switch (cbt_m1 >> 1) {
1806                 case 0:
1807                     for (group = 0; group < (AAC_SIGNE)g_len; group++, cfo+=128) {
1808                         INTFLOAT *cf = cfo;
1809                         int len = off_len;
1810
1811                         do {
1812                             int code;
1813                             unsigned cb_idx;
1814
1815                             UPDATE_CACHE(re, gb);
1816                             GET_VLC(code, re, gb, vlc_tab, 8, 2);
1817                             cb_idx = code;
1818 #if USE_FIXED
1819                             cf = DEC_SQUAD(cf, cb_idx);
1820 #else
1821                             cf = VMUL4(cf, vq, cb_idx, sf + idx);
1822 #endif /* USE_FIXED */
1823                         } while (len -= 4);
1824                     }
1825                     break;
1826
1827                 case 1:
1828                     for (group = 0; group < (AAC_SIGNE)g_len; group++, cfo+=128) {
1829                         INTFLOAT *cf = cfo;
1830                         int len = off_len;
1831
1832                         do {
1833                             int code;
1834                             unsigned nnz;
1835                             unsigned cb_idx;
1836                             uint32_t bits;
1837
1838                             UPDATE_CACHE(re, gb);
1839                             GET_VLC(code, re, gb, vlc_tab, 8, 2);
1840                             cb_idx = code;
1841                             nnz = cb_idx >> 8 & 15;
1842                             bits = nnz ? GET_CACHE(re, gb) : 0;
1843                             LAST_SKIP_BITS(re, gb, nnz);
1844 #if USE_FIXED
1845                             cf = DEC_UQUAD(cf, cb_idx, bits);
1846 #else
1847                             cf = VMUL4S(cf, vq, cb_idx, bits, sf + idx);
1848 #endif /* USE_FIXED */
1849                         } while (len -= 4);
1850                     }
1851                     break;
1852
1853                 case 2:
1854                     for (group = 0; group < (AAC_SIGNE)g_len; group++, cfo+=128) {
1855                         INTFLOAT *cf = cfo;
1856                         int len = off_len;
1857
1858                         do {
1859                             int code;
1860                             unsigned cb_idx;
1861
1862                             UPDATE_CACHE(re, gb);
1863                             GET_VLC(code, re, gb, vlc_tab, 8, 2);
1864                             cb_idx = code;
1865 #if USE_FIXED
1866                             cf = DEC_SPAIR(cf, cb_idx);
1867 #else
1868                             cf = VMUL2(cf, vq, cb_idx, sf + idx);
1869 #endif /* USE_FIXED */
1870                         } while (len -= 2);
1871                     }
1872                     break;
1873
1874                 case 3:
1875                 case 4:
1876                     for (group = 0; group < (AAC_SIGNE)g_len; group++, cfo+=128) {
1877                         INTFLOAT *cf = cfo;
1878                         int len = off_len;
1879
1880                         do {
1881                             int code;
1882                             unsigned nnz;
1883                             unsigned cb_idx;
1884                             unsigned sign;
1885
1886                             UPDATE_CACHE(re, gb);
1887                             GET_VLC(code, re, gb, vlc_tab, 8, 2);
1888                             cb_idx = code;
1889                             nnz = cb_idx >> 8 & 15;
1890                             sign = nnz ? SHOW_UBITS(re, gb, nnz) << (cb_idx >> 12) : 0;
1891                             LAST_SKIP_BITS(re, gb, nnz);
1892 #if USE_FIXED
1893                             cf = DEC_UPAIR(cf, cb_idx, sign);
1894 #else
1895                             cf = VMUL2S(cf, vq, cb_idx, sign, sf + idx);
1896 #endif /* USE_FIXED */
1897                         } while (len -= 2);
1898                     }
1899                     break;
1900
1901                 default:
1902                     for (group = 0; group < (AAC_SIGNE)g_len; group++, cfo+=128) {
1903 #if USE_FIXED
1904                         int *icf = cfo;
1905                         int v;
1906 #else
1907                         float *cf = cfo;
1908                         uint32_t *icf = (uint32_t *) cf;
1909 #endif /* USE_FIXED */
1910                         int len = off_len;
1911
1912                         do {
1913                             int code;
1914                             unsigned nzt, nnz;
1915                             unsigned cb_idx;
1916                             uint32_t bits;
1917                             int j;
1918
1919                             UPDATE_CACHE(re, gb);
1920                             GET_VLC(code, re, gb, vlc_tab, 8, 2);
1921                             cb_idx = code;
1922
1923                             if (cb_idx == 0x0000) {
1924                                 *icf++ = 0;
1925                                 *icf++ = 0;
1926                                 continue;
1927                             }
1928
1929                             nnz = cb_idx >> 12;
1930                             nzt = cb_idx >> 8;
1931                             bits = SHOW_UBITS(re, gb, nnz) << (32-nnz);
1932                             LAST_SKIP_BITS(re, gb, nnz);
1933
1934                             for (j = 0; j < 2; j++) {
1935                                 if (nzt & 1<<j) {
1936                                     uint32_t b;
1937                                     int n;
1938                                     /* The total length of escape_sequence must be < 22 bits according
1939                                        to the specification (i.e. max is 111111110xxxxxxxxxxxx). */
1940                                     UPDATE_CACHE(re, gb);
1941                                     b = GET_CACHE(re, gb);
1942                                     b = 31 - av_log2(~b);
1943
1944                                     if (b > 8) {
1945                                         av_log(ac->avctx, AV_LOG_ERROR, "error in spectral data, ESC overflow\n");
1946                                         return AVERROR_INVALIDDATA;
1947                                     }
1948
1949                                     SKIP_BITS(re, gb, b + 1);
1950                                     b += 4;
1951                                     n = (1 << b) + SHOW_UBITS(re, gb, b);
1952                                     LAST_SKIP_BITS(re, gb, b);
1953 #if USE_FIXED
1954                                     v = n;
1955                                     if (bits & 1U<<31)
1956                                         v = -v;
1957                                     *icf++ = v;
1958 #else
1959                                     *icf++ = ff_cbrt_tab[n] | (bits & 1U<<31);
1960 #endif /* USE_FIXED */
1961                                     bits <<= 1;
1962                                 } else {
1963 #if USE_FIXED
1964                                     v = cb_idx & 15;
1965                                     if (bits & 1U<<31)
1966                                         v = -v;
1967                                     *icf++ = v;
1968 #else
1969                                     unsigned v = ((const uint32_t*)vq)[cb_idx & 15];
1970                                     *icf++ = (bits & 1U<<31) | v;
1971 #endif /* USE_FIXED */
1972                                     bits <<= !!v;
1973                                 }
1974                                 cb_idx >>= 4;
1975                             }
1976                         } while (len -= 2);
1977 #if !USE_FIXED
1978                         ac->fdsp->vector_fmul_scalar(cfo, cfo, sf[idx], off_len);
1979 #endif /* !USE_FIXED */
1980                     }
1981                 }
1982
1983                 CLOSE_READER(re, gb);
1984             }
1985         }
1986         coef += g_len << 7;
1987     }
1988
1989     if (pulse_present) {
1990         idx = 0;
1991         for (i = 0; i < pulse->num_pulse; i++) {
1992             INTFLOAT co = coef_base[ pulse->pos[i] ];
1993             while (offsets[idx + 1] <= pulse->pos[i])
1994                 idx++;
1995             if (band_type[idx] != NOISE_BT && sf[idx]) {
1996                 INTFLOAT ico = -pulse->amp[i];
1997 #if USE_FIXED
1998                 if (co) {
1999                     ico = co + (co > 0 ? -ico : ico);
2000                 }
2001                 coef_base[ pulse->pos[i] ] = ico;
2002 #else
2003                 if (co) {
2004                     co /= sf[idx];
2005                     ico = co / sqrtf(sqrtf(fabsf(co))) + (co > 0 ? -ico : ico);
2006                 }
2007                 coef_base[ pulse->pos[i] ] = cbrtf(fabsf(ico)) * ico * sf[idx];
2008 #endif /* USE_FIXED */
2009             }
2010         }
2011     }
2012 #if USE_FIXED
2013     coef = coef_base;
2014     idx = 0;
2015     for (g = 0; g < ics->num_window_groups; g++) {
2016         unsigned g_len = ics->group_len[g];
2017
2018         for (i = 0; i < ics->max_sfb; i++, idx++) {
2019             const unsigned cbt_m1 = band_type[idx] - 1;
2020             int *cfo = coef + offsets[i];
2021             int off_len = offsets[i + 1] - offsets[i];
2022             int group;
2023
2024             if (cbt_m1 < NOISE_BT - 1) {
2025                 for (group = 0; group < (int)g_len; group++, cfo+=128) {
2026                     ac->vector_pow43(cfo, off_len);
2027                     ac->subband_scale(cfo, cfo, sf[idx], 34, off_len, ac->avctx);
2028                 }
2029             }
2030         }
2031         coef += g_len << 7;
2032     }
2033 #endif /* USE_FIXED */
2034     return 0;
2035 }
2036
2037 /**
2038  * Apply AAC-Main style frequency domain prediction.
2039  */
2040 static void apply_prediction(AACContext *ac, SingleChannelElement *sce)
2041 {
2042     int sfb, k;
2043
2044     if (!sce->ics.predictor_initialized) {
2045         reset_all_predictors(sce->predictor_state);
2046         sce->ics.predictor_initialized = 1;
2047     }
2048
2049     if (sce->ics.window_sequence[0] != EIGHT_SHORT_SEQUENCE) {
2050         for (sfb = 0;
2051              sfb < ff_aac_pred_sfb_max[ac->oc[1].m4ac.sampling_index];
2052              sfb++) {
2053             for (k = sce->ics.swb_offset[sfb];
2054                  k < sce->ics.swb_offset[sfb + 1];
2055                  k++) {
2056                 predict(&sce->predictor_state[k], &sce->coeffs[k],
2057                         sce->ics.predictor_present &&
2058                         sce->ics.prediction_used[sfb]);
2059             }
2060         }
2061         if (sce->ics.predictor_reset_group)
2062             reset_predictor_group(sce->predictor_state,
2063                                   sce->ics.predictor_reset_group);
2064     } else
2065         reset_all_predictors(sce->predictor_state);
2066 }
2067
2068 static void decode_gain_control(SingleChannelElement * sce, GetBitContext * gb)
2069 {
2070     // wd_num, wd_test, aloc_size
2071     static const uint8_t gain_mode[4][3] = {
2072         {1, 0, 5},  // ONLY_LONG_SEQUENCE = 0,
2073         {2, 1, 2},  // LONG_START_SEQUENCE,
2074         {8, 0, 2},  // EIGHT_SHORT_SEQUENCE,
2075         {2, 1, 5},  // LONG_STOP_SEQUENCE
2076     };
2077
2078     const int mode = sce->ics.window_sequence[0];
2079     uint8_t bd, wd, ad;
2080
2081     // FIXME: Store the gain control data on |sce| and do something with it.
2082     uint8_t max_band = get_bits(gb, 2);
2083     for (bd = 0; bd < max_band; bd++) {
2084         for (wd = 0; wd < gain_mode[mode][0]; wd++) {
2085             uint8_t adjust_num = get_bits(gb, 3);
2086             for (ad = 0; ad < adjust_num; ad++) {
2087                 skip_bits(gb, 4 + ((wd == 0 && gain_mode[mode][1])
2088                                      ? 4
2089                                      : gain_mode[mode][2]));
2090             }
2091         }
2092     }
2093 }
2094
2095 /**
2096  * Decode an individual_channel_stream payload; reference: table 4.44.
2097  *
2098  * @param   common_window   Channels have independent [0], or shared [1], Individual Channel Stream information.
2099  * @param   scale_flag      scalable [1] or non-scalable [0] AAC (Unused until scalable AAC is implemented.)
2100  *
2101  * @return  Returns error status. 0 - OK, !0 - error
2102  */
2103 static int decode_ics(AACContext *ac, SingleChannelElement *sce,
2104                       GetBitContext *gb, int common_window, int scale_flag)
2105 {
2106     Pulse pulse;
2107     TemporalNoiseShaping    *tns = &sce->tns;
2108     IndividualChannelStream *ics = &sce->ics;
2109     INTFLOAT *out = sce->coeffs;
2110     int global_gain, eld_syntax, er_syntax, pulse_present = 0;
2111     int ret;
2112
2113     eld_syntax = ac->oc[1].m4ac.object_type == AOT_ER_AAC_ELD;
2114     er_syntax  = ac->oc[1].m4ac.object_type == AOT_ER_AAC_LC ||
2115                  ac->oc[1].m4ac.object_type == AOT_ER_AAC_LTP ||
2116                  ac->oc[1].m4ac.object_type == AOT_ER_AAC_LD ||
2117                  ac->oc[1].m4ac.object_type == AOT_ER_AAC_ELD;
2118
2119     /* This assignment is to silence a GCC warning about the variable being used
2120      * uninitialized when in fact it always is.
2121      */
2122     pulse.num_pulse = 0;
2123
2124     global_gain = get_bits(gb, 8);
2125
2126     if (!common_window && !scale_flag) {
2127         ret = decode_ics_info(ac, ics, gb);
2128         if (ret < 0)
2129             goto fail;
2130     }
2131
2132     if ((ret = decode_band_types(ac, sce->band_type,
2133                                  sce->band_type_run_end, gb, ics)) < 0)
2134         goto fail;
2135     if ((ret = decode_scalefactors(ac, sce->sf, gb, global_gain, ics,
2136                                   sce->band_type, sce->band_type_run_end)) < 0)
2137         goto fail;
2138
2139     pulse_present = 0;
2140     if (!scale_flag) {
2141         if (!eld_syntax && (pulse_present = get_bits1(gb))) {
2142             if (ics->window_sequence[0] == EIGHT_SHORT_SEQUENCE) {
2143                 av_log(ac->avctx, AV_LOG_ERROR,
2144                        "Pulse tool not allowed in eight short sequence.\n");
2145                 ret = AVERROR_INVALIDDATA;
2146                 goto fail;
2147             }
2148             if (decode_pulses(&pulse, gb, ics->swb_offset, ics->num_swb)) {
2149                 av_log(ac->avctx, AV_LOG_ERROR,
2150                        "Pulse data corrupt or invalid.\n");
2151                 ret = AVERROR_INVALIDDATA;
2152                 goto fail;
2153             }
2154         }
2155         tns->present = get_bits1(gb);
2156         if (tns->present && !er_syntax) {
2157             ret = decode_tns(ac, tns, gb, ics);
2158             if (ret < 0)
2159                 goto fail;
2160         }
2161         if (!eld_syntax && get_bits1(gb)) {
2162             decode_gain_control(sce, gb);
2163             if (!ac->warned_gain_control) {
2164                 avpriv_report_missing_feature(ac->avctx, "Gain control");
2165                 ac->warned_gain_control = 1;
2166             }
2167         }
2168         // I see no textual basis in the spec for this occurring after SSR gain
2169         // control, but this is what both reference and real implmentations do
2170         if (tns->present && er_syntax) {
2171             ret = decode_tns(ac, tns, gb, ics);
2172             if (ret < 0)
2173                 goto fail;
2174         }
2175     }
2176
2177     ret = decode_spectrum_and_dequant(ac, out, gb, sce->sf, pulse_present,
2178                                     &pulse, ics, sce->band_type);
2179     if (ret < 0)
2180         goto fail;
2181
2182     if (ac->oc[1].m4ac.object_type == AOT_AAC_MAIN && !common_window)
2183         apply_prediction(ac, sce);
2184
2185     return 0;
2186 fail:
2187     tns->present = 0;
2188     return ret;
2189 }
2190
2191 /**
2192  * Mid/Side stereo decoding; reference: 4.6.8.1.3.
2193  */
2194 static void apply_mid_side_stereo(AACContext *ac, ChannelElement *cpe)
2195 {
2196     const IndividualChannelStream *ics = &cpe->ch[0].ics;
2197     INTFLOAT *ch0 = cpe->ch[0].coeffs;
2198     INTFLOAT *ch1 = cpe->ch[1].coeffs;
2199     int g, i, group, idx = 0;
2200     const uint16_t *offsets = ics->swb_offset;
2201     for (g = 0; g < ics->num_window_groups; g++) {
2202         for (i = 0; i < ics->max_sfb; i++, idx++) {
2203             if (cpe->ms_mask[idx] &&
2204                 cpe->ch[0].band_type[idx] < NOISE_BT &&
2205                 cpe->ch[1].band_type[idx] < NOISE_BT) {
2206 #if USE_FIXED
2207                 for (group = 0; group < ics->group_len[g]; group++) {
2208                     ac->fdsp->butterflies_fixed(ch0 + group * 128 + offsets[i],
2209                                                 ch1 + group * 128 + offsets[i],
2210                                                 offsets[i+1] - offsets[i]);
2211 #else
2212                 for (group = 0; group < ics->group_len[g]; group++) {
2213                     ac->fdsp->butterflies_float(ch0 + group * 128 + offsets[i],
2214                                                ch1 + group * 128 + offsets[i],
2215                                                offsets[i+1] - offsets[i]);
2216 #endif /* USE_FIXED */
2217                 }
2218             }
2219         }
2220         ch0 += ics->group_len[g] * 128;
2221         ch1 += ics->group_len[g] * 128;
2222     }
2223 }
2224
2225 /**
2226  * intensity stereo decoding; reference: 4.6.8.2.3
2227  *
2228  * @param   ms_present  Indicates mid/side stereo presence. [0] mask is all 0s;
2229  *                      [1] mask is decoded from bitstream; [2] mask is all 1s;
2230  *                      [3] reserved for scalable AAC
2231  */
2232 static void apply_intensity_stereo(AACContext *ac,
2233                                    ChannelElement *cpe, int ms_present)
2234 {
2235     const IndividualChannelStream *ics = &cpe->ch[1].ics;
2236     SingleChannelElement         *sce1 = &cpe->ch[1];
2237     INTFLOAT *coef0 = cpe->ch[0].coeffs, *coef1 = cpe->ch[1].coeffs;
2238     const uint16_t *offsets = ics->swb_offset;
2239     int g, group, i, idx = 0;
2240     int c;
2241     INTFLOAT scale;
2242     for (g = 0; g < ics->num_window_groups; g++) {
2243         for (i = 0; i < ics->max_sfb;) {
2244             if (sce1->band_type[idx] == INTENSITY_BT ||
2245                 sce1->band_type[idx] == INTENSITY_BT2) {
2246                 const int bt_run_end = sce1->band_type_run_end[idx];
2247                 for (; i < bt_run_end; i++, idx++) {
2248                     c = -1 + 2 * (sce1->band_type[idx] - 14);
2249                     if (ms_present)
2250                         c *= 1 - 2 * cpe->ms_mask[idx];
2251                     scale = c * sce1->sf[idx];
2252                     for (group = 0; group < ics->group_len[g]; group++)
2253 #if USE_FIXED
2254                         ac->subband_scale(coef1 + group * 128 + offsets[i],
2255                                       coef0 + group * 128 + offsets[i],
2256                                       scale,
2257                                       23,
2258                                       offsets[i + 1] - offsets[i] ,ac->avctx);
2259 #else
2260                         ac->fdsp->vector_fmul_scalar(coef1 + group * 128 + offsets[i],
2261                                                     coef0 + group * 128 + offsets[i],
2262                                                     scale,
2263                                                     offsets[i + 1] - offsets[i]);
2264 #endif /* USE_FIXED */
2265                 }
2266             } else {
2267                 int bt_run_end = sce1->band_type_run_end[idx];
2268                 idx += bt_run_end - i;
2269                 i    = bt_run_end;
2270             }
2271         }
2272         coef0 += ics->group_len[g] * 128;
2273         coef1 += ics->group_len[g] * 128;
2274     }
2275 }
2276
2277 /**
2278  * Decode a channel_pair_element; reference: table 4.4.
2279  *
2280  * @return  Returns error status. 0 - OK, !0 - error
2281  */
2282 static int decode_cpe(AACContext *ac, GetBitContext *gb, ChannelElement *cpe)
2283 {
2284     int i, ret, common_window, ms_present = 0;
2285     int eld_syntax = ac->oc[1].m4ac.object_type == AOT_ER_AAC_ELD;
2286
2287     common_window = eld_syntax || get_bits1(gb);
2288     if (common_window) {
2289         if (decode_ics_info(ac, &cpe->ch[0].ics, gb))
2290             return AVERROR_INVALIDDATA;
2291         i = cpe->ch[1].ics.use_kb_window[0];
2292         cpe->ch[1].ics = cpe->ch[0].ics;
2293         cpe->ch[1].ics.use_kb_window[1] = i;
2294         if (cpe->ch[1].ics.predictor_present &&
2295             (ac->oc[1].m4ac.object_type != AOT_AAC_MAIN))
2296             if ((cpe->ch[1].ics.ltp.present = get_bits(gb, 1)))
2297                 decode_ltp(&cpe->ch[1].ics.ltp, gb, cpe->ch[1].ics.max_sfb);
2298         ms_present = get_bits(gb, 2);
2299         if (ms_present == 3) {
2300             av_log(ac->avctx, AV_LOG_ERROR, "ms_present = 3 is reserved.\n");
2301             return AVERROR_INVALIDDATA;
2302         } else if (ms_present)
2303             decode_mid_side_stereo(cpe, gb, ms_present);
2304     }
2305     if ((ret = decode_ics(ac, &cpe->ch[0], gb, common_window, 0)))
2306         return ret;
2307     if ((ret = decode_ics(ac, &cpe->ch[1], gb, common_window, 0)))
2308         return ret;
2309
2310     if (common_window) {
2311         if (ms_present)
2312             apply_mid_side_stereo(ac, cpe);
2313         if (ac->oc[1].m4ac.object_type == AOT_AAC_MAIN) {
2314             apply_prediction(ac, &cpe->ch[0]);
2315             apply_prediction(ac, &cpe->ch[1]);
2316         }
2317     }
2318
2319     apply_intensity_stereo(ac, cpe, ms_present);
2320     return 0;
2321 }
2322
2323 static const float cce_scale[] = {
2324     1.09050773266525765921, //2^(1/8)
2325     1.18920711500272106672, //2^(1/4)
2326     M_SQRT2,
2327     2,
2328 };
2329
2330 /**
2331  * Decode coupling_channel_element; reference: table 4.8.
2332  *
2333  * @return  Returns error status. 0 - OK, !0 - error
2334  */
2335 static int decode_cce(AACContext *ac, GetBitContext *gb, ChannelElement *che)
2336 {
2337     int num_gain = 0;
2338     int c, g, sfb, ret;
2339     int sign;
2340     INTFLOAT scale;
2341     SingleChannelElement *sce = &che->ch[0];
2342     ChannelCoupling     *coup = &che->coup;
2343
2344     coup->coupling_point = 2 * get_bits1(gb);
2345     coup->num_coupled = get_bits(gb, 3);
2346     for (c = 0; c <= coup->num_coupled; c++) {
2347         num_gain++;
2348         coup->type[c] = get_bits1(gb) ? TYPE_CPE : TYPE_SCE;
2349         coup->id_select[c] = get_bits(gb, 4);
2350         if (coup->type[c] == TYPE_CPE) {
2351             coup->ch_select[c] = get_bits(gb, 2);
2352             if (coup->ch_select[c] == 3)
2353                 num_gain++;
2354         } else
2355             coup->ch_select[c] = 2;
2356     }
2357     coup->coupling_point += get_bits1(gb) || (coup->coupling_point >> 1);
2358
2359     sign  = get_bits(gb, 1);
2360 #if USE_FIXED
2361     scale = get_bits(gb, 2);
2362 #else
2363     scale = cce_scale[get_bits(gb, 2)];
2364 #endif
2365
2366     if ((ret = decode_ics(ac, sce, gb, 0, 0)))
2367         return ret;
2368
2369     for (c = 0; c < num_gain; c++) {
2370         int idx  = 0;
2371         int cge  = 1;
2372         int gain = 0;
2373         INTFLOAT gain_cache = FIXR10(1.);
2374         if (c) {
2375             cge = coup->coupling_point == AFTER_IMDCT ? 1 : get_bits1(gb);
2376             gain = cge ? get_vlc2(gb, vlc_scalefactors.table, 7, 3) - 60: 0;
2377             gain_cache = GET_GAIN(scale, gain);
2378 #if USE_FIXED
2379             if ((abs(gain_cache)-1024) >> 3 > 30)
2380                 return AVERROR(ERANGE);
2381 #endif
2382         }
2383         if (coup->coupling_point == AFTER_IMDCT) {
2384             coup->gain[c][0] = gain_cache;
2385         } else {
2386             for (g = 0; g < sce->ics.num_window_groups; g++) {
2387                 for (sfb = 0; sfb < sce->ics.max_sfb; sfb++, idx++) {
2388                     if (sce->band_type[idx] != ZERO_BT) {
2389                         if (!cge) {
2390                             int t = get_vlc2(gb, vlc_scalefactors.table, 7, 3) - 60;
2391                             if (t) {
2392                                 int s = 1;
2393                                 t = gain += t;
2394                                 if (sign) {
2395                                     s  -= 2 * (t & 0x1);
2396                                     t >>= 1;
2397                                 }
2398                                 gain_cache = GET_GAIN(scale, t) * s;
2399 #if USE_FIXED
2400                                 if ((abs(gain_cache)-1024) >> 3 > 30)
2401                                     return AVERROR(ERANGE);
2402 #endif
2403                             }
2404                         }
2405                         coup->gain[c][idx] = gain_cache;
2406                     }
2407                 }
2408             }
2409         }
2410     }
2411     return 0;
2412 }
2413
2414 /**
2415  * Parse whether channels are to be excluded from Dynamic Range Compression; reference: table 4.53.
2416  *
2417  * @return  Returns number of bytes consumed.
2418  */
2419 static int decode_drc_channel_exclusions(DynamicRangeControl *che_drc,
2420                                          GetBitContext *gb)
2421 {
2422     int i;
2423     int num_excl_chan = 0;
2424
2425     do {
2426         for (i = 0; i < 7; i++)
2427             che_drc->exclude_mask[num_excl_chan++] = get_bits1(gb);
2428     } while (num_excl_chan < MAX_CHANNELS - 7 && get_bits1(gb));
2429
2430     return num_excl_chan / 7;
2431 }
2432
2433 /**
2434  * Decode dynamic range information; reference: table 4.52.
2435  *
2436  * @return  Returns number of bytes consumed.
2437  */
2438 static int decode_dynamic_range(DynamicRangeControl *che_drc,
2439                                 GetBitContext *gb)
2440 {
2441     int n             = 1;
2442     int drc_num_bands = 1;
2443     int i;
2444
2445     /* pce_tag_present? */
2446     if (get_bits1(gb)) {
2447         che_drc->pce_instance_tag  = get_bits(gb, 4);
2448         skip_bits(gb, 4); // tag_reserved_bits
2449         n++;
2450     }
2451
2452     /* excluded_chns_present? */
2453     if (get_bits1(gb)) {
2454         n += decode_drc_channel_exclusions(che_drc, gb);
2455     }
2456
2457     /* drc_bands_present? */
2458     if (get_bits1(gb)) {
2459         che_drc->band_incr            = get_bits(gb, 4);
2460         che_drc->interpolation_scheme = get_bits(gb, 4);
2461         n++;
2462         drc_num_bands += che_drc->band_incr;
2463         for (i = 0; i < drc_num_bands; i++) {
2464             che_drc->band_top[i] = get_bits(gb, 8);
2465             n++;
2466         }
2467     }
2468
2469     /* prog_ref_level_present? */
2470     if (get_bits1(gb)) {
2471         che_drc->prog_ref_level = get_bits(gb, 7);
2472         skip_bits1(gb); // prog_ref_level_reserved_bits
2473         n++;
2474     }
2475
2476     for (i = 0; i < drc_num_bands; i++) {
2477         che_drc->dyn_rng_sgn[i] = get_bits1(gb);
2478         che_drc->dyn_rng_ctl[i] = get_bits(gb, 7);
2479         n++;
2480     }
2481
2482     return n;
2483 }
2484
2485 static int decode_fill(AACContext *ac, GetBitContext *gb, int len) {
2486     uint8_t buf[256];
2487     int i, major, minor;
2488
2489     if (len < 13+7*8)
2490         goto unknown;
2491
2492     get_bits(gb, 13); len -= 13;
2493
2494     for(i=0; i+1<sizeof(buf) && len>=8; i++, len-=8)
2495         buf[i] = get_bits(gb, 8);
2496
2497     buf[i] = 0;
2498     if (ac->avctx->debug & FF_DEBUG_PICT_INFO)
2499         av_log(ac->avctx, AV_LOG_DEBUG, "FILL:%s\n", buf);
2500
2501     if (sscanf(buf, "libfaac %d.%d", &major, &minor) == 2){
2502         ac->avctx->internal->skip_samples = 1024;
2503     }
2504
2505 unknown:
2506     skip_bits_long(gb, len);
2507
2508     return 0;
2509 }
2510
2511 /**
2512  * Decode extension data (incomplete); reference: table 4.51.
2513  *
2514  * @param   cnt length of TYPE_FIL syntactic element in bytes
2515  *
2516  * @return Returns number of bytes consumed
2517  */
2518 static int decode_extension_payload(AACContext *ac, GetBitContext *gb, int cnt,
2519                                     ChannelElement *che, enum RawDataBlockType elem_type)
2520 {
2521     int crc_flag = 0;
2522     int res = cnt;
2523     int type = get_bits(gb, 4);
2524
2525     if (ac->avctx->debug & FF_DEBUG_STARTCODE)
2526         av_log(ac->avctx, AV_LOG_DEBUG, "extension type: %d len:%d\n", type, cnt);
2527
2528     switch (type) { // extension type
2529     case EXT_SBR_DATA_CRC:
2530         crc_flag++;
2531     case EXT_SBR_DATA:
2532         if (!che) {
2533             av_log(ac->avctx, AV_LOG_ERROR, "SBR was found before the first channel element.\n");
2534             return res;
2535         } else if (ac->oc[1].m4ac.frame_length_short) {
2536             if (!ac->warned_960_sbr)
2537               avpriv_report_missing_feature(ac->avctx,
2538                                             "SBR with 960 frame length");
2539             ac->warned_960_sbr = 1;
2540             skip_bits_long(gb, 8 * cnt - 4);
2541             return res;
2542         } else if (!ac->oc[1].m4ac.sbr) {
2543             av_log(ac->avctx, AV_LOG_ERROR, "SBR signaled to be not-present but was found in the bitstream.\n");
2544             skip_bits_long(gb, 8 * cnt - 4);
2545             return res;
2546         } else if (ac->oc[1].m4ac.sbr == -1 && ac->oc[1].status == OC_LOCKED) {
2547             av_log(ac->avctx, AV_LOG_ERROR, "Implicit SBR was found with a first occurrence after the first frame.\n");
2548             skip_bits_long(gb, 8 * cnt - 4);
2549             return res;
2550         } else if (ac->oc[1].m4ac.ps == -1 && ac->oc[1].status < OC_LOCKED && ac->avctx->channels == 1) {
2551             ac->oc[1].m4ac.sbr = 1;
2552             ac->oc[1].m4ac.ps = 1;
2553             ac->avctx->profile = FF_PROFILE_AAC_HE_V2;
2554             output_configure(ac, ac->oc[1].layout_map, ac->oc[1].layout_map_tags,
2555                              ac->oc[1].status, 1);
2556         } else {
2557             ac->oc[1].m4ac.sbr = 1;
2558             ac->avctx->profile = FF_PROFILE_AAC_HE;
2559         }
2560         res = AAC_RENAME(ff_decode_sbr_extension)(ac, &che->sbr, gb, crc_flag, cnt, elem_type);
2561         break;
2562     case EXT_DYNAMIC_RANGE:
2563         res = decode_dynamic_range(&ac->che_drc, gb);
2564         break;
2565     case EXT_FILL:
2566         decode_fill(ac, gb, 8 * cnt - 4);
2567         break;
2568     case EXT_FILL_DATA:
2569     case EXT_DATA_ELEMENT:
2570     default:
2571         skip_bits_long(gb, 8 * cnt - 4);
2572         break;
2573     };
2574     return res;
2575 }
2576
2577 /**
2578  * Decode Temporal Noise Shaping filter coefficients and apply all-pole filters; reference: 4.6.9.3.
2579  *
2580  * @param   decode  1 if tool is used normally, 0 if tool is used in LTP.
2581  * @param   coef    spectral coefficients
2582  */
2583 static void apply_tns(INTFLOAT coef_param[1024], TemporalNoiseShaping *tns,
2584                       IndividualChannelStream *ics, int decode)
2585 {
2586     const int mmm = FFMIN(ics->tns_max_bands, ics->max_sfb);
2587     int w, filt, m, i;
2588     int bottom, top, order, start, end, size, inc;
2589     INTFLOAT lpc[TNS_MAX_ORDER];
2590     INTFLOAT tmp[TNS_MAX_ORDER+1];
2591     UINTFLOAT *coef = coef_param;
2592
2593     if(!mmm)
2594         return;
2595
2596     for (w = 0; w < ics->num_windows; w++) {
2597         bottom = ics->num_swb;
2598         for (filt = 0; filt < tns->n_filt[w]; filt++) {
2599             top    = bottom;
2600             bottom = FFMAX(0, top - tns->length[w][filt]);
2601             order  = tns->order[w][filt];
2602             if (order == 0)
2603                 continue;
2604
2605             // tns_decode_coef
2606             AAC_RENAME(compute_lpc_coefs)(tns->coef[w][filt], order, lpc, 0, 0, 0);
2607
2608             start = ics->swb_offset[FFMIN(bottom, mmm)];
2609             end   = ics->swb_offset[FFMIN(   top, mmm)];
2610             if ((size = end - start) <= 0)
2611                 continue;
2612             if (tns->direction[w][filt]) {
2613                 inc = -1;
2614                 start = end - 1;
2615             } else {
2616                 inc = 1;
2617             }
2618             start += w * 128;
2619
2620             if (decode) {
2621                 // ar filter
2622                 for (m = 0; m < size; m++, start += inc)
2623                     for (i = 1; i <= FFMIN(m, order); i++)
2624                         coef[start] -= AAC_MUL26((INTFLOAT)coef[start - i * inc], lpc[i - 1]);
2625             } else {
2626                 // ma filter
2627                 for (m = 0; m < size; m++, start += inc) {
2628                     tmp[0] = coef[start];
2629                     for (i = 1; i <= FFMIN(m, order); i++)
2630                         coef[start] += AAC_MUL26(tmp[i], lpc[i - 1]);
2631                     for (i = order; i > 0; i--)
2632                         tmp[i] = tmp[i - 1];
2633                 }
2634             }
2635         }
2636     }
2637 }
2638
2639 /**
2640  *  Apply windowing and MDCT to obtain the spectral
2641  *  coefficient from the predicted sample by LTP.
2642  */
2643 static void windowing_and_mdct_ltp(AACContext *ac, INTFLOAT *out,
2644                                    INTFLOAT *in, IndividualChannelStream *ics)
2645 {
2646     const INTFLOAT *lwindow      = ics->use_kb_window[0] ? AAC_RENAME2(aac_kbd_long_1024) : AAC_RENAME2(sine_1024);
2647     const INTFLOAT *swindow      = ics->use_kb_window[0] ? AAC_RENAME2(aac_kbd_short_128) : AAC_RENAME2(sine_128);
2648     const INTFLOAT *lwindow_prev = ics->use_kb_window[1] ? AAC_RENAME2(aac_kbd_long_1024) : AAC_RENAME2(sine_1024);
2649     const INTFLOAT *swindow_prev = ics->use_kb_window[1] ? AAC_RENAME2(aac_kbd_short_128) : AAC_RENAME2(sine_128);
2650
2651     if (ics->window_sequence[0] != LONG_STOP_SEQUENCE) {
2652         ac->fdsp->vector_fmul(in, in, lwindow_prev, 1024);
2653     } else {
2654         memset(in, 0, 448 * sizeof(*in));
2655         ac->fdsp->vector_fmul(in + 448, in + 448, swindow_prev, 128);
2656     }
2657     if (ics->window_sequence[0] != LONG_START_SEQUENCE) {
2658         ac->fdsp->vector_fmul_reverse(in + 1024, in + 1024, lwindow, 1024);
2659     } else {
2660         ac->fdsp->vector_fmul_reverse(in + 1024 + 448, in + 1024 + 448, swindow, 128);
2661         memset(in + 1024 + 576, 0, 448 * sizeof(*in));
2662     }
2663     ac->mdct_ltp.mdct_calc(&ac->mdct_ltp, out, in);
2664 }
2665
2666 /**
2667  * Apply the long term prediction
2668  */
2669 static void apply_ltp(AACContext *ac, SingleChannelElement *sce)
2670 {
2671     const LongTermPrediction *ltp = &sce->ics.ltp;
2672     const uint16_t *offsets = sce->ics.swb_offset;
2673     int i, sfb;
2674
2675     if (sce->ics.window_sequence[0] != EIGHT_SHORT_SEQUENCE) {
2676         INTFLOAT *predTime = sce->ret;
2677         INTFLOAT *predFreq = ac->buf_mdct;
2678         int16_t num_samples = 2048;
2679
2680         if (ltp->lag < 1024)
2681             num_samples = ltp->lag + 1024;
2682         for (i = 0; i < num_samples; i++)
2683             predTime[i] = AAC_MUL30(sce->ltp_state[i + 2048 - ltp->lag], ltp->coef);
2684         memset(&predTime[i], 0, (2048 - i) * sizeof(*predTime));
2685
2686         ac->windowing_and_mdct_ltp(ac, predFreq, predTime, &sce->ics);
2687
2688         if (sce->tns.present)
2689             ac->apply_tns(predFreq, &sce->tns, &sce->ics, 0);
2690
2691         for (sfb = 0; sfb < FFMIN(sce->ics.max_sfb, MAX_LTP_LONG_SFB); sfb++)
2692             if (ltp->used[sfb])
2693                 for (i = offsets[sfb]; i < offsets[sfb + 1]; i++)
2694                     sce->coeffs[i] += (UINTFLOAT)predFreq[i];
2695     }
2696 }
2697
2698 /**
2699  * Update the LTP buffer for next frame
2700  */
2701 static void update_ltp(AACContext *ac, SingleChannelElement *sce)
2702 {
2703     IndividualChannelStream *ics = &sce->ics;
2704     INTFLOAT *saved     = sce->saved;
2705     INTFLOAT *saved_ltp = sce->coeffs;
2706     const INTFLOAT *lwindow = ics->use_kb_window[0] ? AAC_RENAME2(aac_kbd_long_1024) : AAC_RENAME2(sine_1024);
2707     const INTFLOAT *swindow = ics->use_kb_window[0] ? AAC_RENAME2(aac_kbd_short_128) : AAC_RENAME2(sine_128);
2708     int i;
2709
2710     if (ics->window_sequence[0] == EIGHT_SHORT_SEQUENCE) {
2711         memcpy(saved_ltp,       saved, 512 * sizeof(*saved_ltp));
2712         memset(saved_ltp + 576, 0,     448 * sizeof(*saved_ltp));
2713         ac->fdsp->vector_fmul_reverse(saved_ltp + 448, ac->buf_mdct + 960,     &swindow[64],      64);
2714
2715         for (i = 0; i < 64; i++)
2716             saved_ltp[i + 512] = AAC_MUL31(ac->buf_mdct[1023 - i], swindow[63 - i]);
2717     } else if (ics->window_sequence[0] == LONG_START_SEQUENCE) {
2718         memcpy(saved_ltp,       ac->buf_mdct + 512, 448 * sizeof(*saved_ltp));
2719         memset(saved_ltp + 576, 0,                  448 * sizeof(*saved_ltp));
2720         ac->fdsp->vector_fmul_reverse(saved_ltp + 448, ac->buf_mdct + 960,     &swindow[64],      64);
2721
2722         for (i = 0; i < 64; i++)
2723             saved_ltp[i + 512] = AAC_MUL31(ac->buf_mdct[1023 - i], swindow[63 - i]);
2724     } else { // LONG_STOP or ONLY_LONG
2725         ac->fdsp->vector_fmul_reverse(saved_ltp,       ac->buf_mdct + 512,     &lwindow[512],     512);
2726
2727         for (i = 0; i < 512; i++)
2728             saved_ltp[i + 512] = AAC_MUL31(ac->buf_mdct[1023 - i], lwindow[511 - i]);
2729     }
2730
2731     memcpy(sce->ltp_state,      sce->ltp_state+1024, 1024 * sizeof(*sce->ltp_state));
2732     memcpy(sce->ltp_state+1024, sce->ret,            1024 * sizeof(*sce->ltp_state));
2733     memcpy(sce->ltp_state+2048, saved_ltp,           1024 * sizeof(*sce->ltp_state));
2734 }
2735
2736 /**
2737  * Conduct IMDCT and windowing.
2738  */
2739 static void imdct_and_windowing(AACContext *ac, SingleChannelElement *sce)
2740 {
2741     IndividualChannelStream *ics = &sce->ics;
2742     INTFLOAT *in    = sce->coeffs;
2743     INTFLOAT *out   = sce->ret;
2744     INTFLOAT *saved = sce->saved;
2745     const INTFLOAT *swindow      = ics->use_kb_window[0] ? AAC_RENAME2(aac_kbd_short_128) : AAC_RENAME2(sine_128);
2746     const INTFLOAT *lwindow_prev = ics->use_kb_window[1] ? AAC_RENAME2(aac_kbd_long_1024) : AAC_RENAME2(sine_1024);
2747     const INTFLOAT *swindow_prev = ics->use_kb_window[1] ? AAC_RENAME2(aac_kbd_short_128) : AAC_RENAME2(sine_128);
2748     INTFLOAT *buf  = ac->buf_mdct;
2749     INTFLOAT *temp = ac->temp;
2750     int i;
2751
2752     // imdct
2753     if (ics->window_sequence[0] == EIGHT_SHORT_SEQUENCE) {
2754         for (i = 0; i < 1024; i += 128)
2755             ac->mdct_small.imdct_half(&ac->mdct_small, buf + i, in + i);
2756     } else {
2757         ac->mdct.imdct_half(&ac->mdct, buf, in);
2758 #if USE_FIXED
2759         for (i=0; i<1024; i++)
2760           buf[i] = (buf[i] + 4LL) >> 3;
2761 #endif /* USE_FIXED */
2762     }
2763
2764     /* window overlapping
2765      * NOTE: To simplify the overlapping code, all 'meaningless' short to long
2766      * and long to short transitions are considered to be short to short
2767      * transitions. This leaves just two cases (long to long and short to short)
2768      * with a little special sauce for EIGHT_SHORT_SEQUENCE.
2769      */
2770     if ((ics->window_sequence[1] == ONLY_LONG_SEQUENCE || ics->window_sequence[1] == LONG_STOP_SEQUENCE) &&
2771             (ics->window_sequence[0] == ONLY_LONG_SEQUENCE || ics->window_sequence[0] == LONG_START_SEQUENCE)) {
2772         ac->fdsp->vector_fmul_window(    out,               saved,            buf,         lwindow_prev, 512);
2773     } else {
2774         memcpy(                         out,               saved,            448 * sizeof(*out));
2775
2776         if (ics->window_sequence[0] == EIGHT_SHORT_SEQUENCE) {
2777             ac->fdsp->vector_fmul_window(out + 448 + 0*128, saved + 448,      buf + 0*128, swindow_prev, 64);
2778             ac->fdsp->vector_fmul_window(out + 448 + 1*128, buf + 0*128 + 64, buf + 1*128, swindow,      64);
2779             ac->fdsp->vector_fmul_window(out + 448 + 2*128, buf + 1*128 + 64, buf + 2*128, swindow,      64);
2780             ac->fdsp->vector_fmul_window(out + 448 + 3*128, buf + 2*128 + 64, buf + 3*128, swindow,      64);
2781             ac->fdsp->vector_fmul_window(temp,              buf + 3*128 + 64, buf + 4*128, swindow,      64);
2782             memcpy(                     out + 448 + 4*128, temp, 64 * sizeof(*out));
2783         } else {
2784             ac->fdsp->vector_fmul_window(out + 448,         saved + 448,      buf,         swindow_prev, 64);
2785             memcpy(                     out + 576,         buf + 64,         448 * sizeof(*out));
2786         }
2787     }
2788
2789     // buffer update
2790     if (ics->window_sequence[0] == EIGHT_SHORT_SEQUENCE) {
2791         memcpy(                     saved,       temp + 64,         64 * sizeof(*saved));
2792         ac->fdsp->vector_fmul_window(saved + 64,  buf + 4*128 + 64, buf + 5*128, swindow, 64);
2793         ac->fdsp->vector_fmul_window(saved + 192, buf + 5*128 + 64, buf + 6*128, swindow, 64);
2794         ac->fdsp->vector_fmul_window(saved + 320, buf + 6*128 + 64, buf + 7*128, swindow, 64);
2795         memcpy(                     saved + 448, buf + 7*128 + 64,  64 * sizeof(*saved));
2796     } else if (ics->window_sequence[0] == LONG_START_SEQUENCE) {
2797         memcpy(                     saved,       buf + 512,        448 * sizeof(*saved));
2798         memcpy(                     saved + 448, buf + 7*128 + 64,  64 * sizeof(*saved));
2799     } else { // LONG_STOP or ONLY_LONG
2800         memcpy(                     saved,       buf + 512,        512 * sizeof(*saved));
2801     }
2802 }
2803
2804 /**
2805  * Conduct IMDCT and windowing.
2806  */
2807 static void imdct_and_windowing_960(AACContext *ac, SingleChannelElement *sce)
2808 {
2809 #if !USE_FIXED
2810     IndividualChannelStream *ics = &sce->ics;
2811     INTFLOAT *in    = sce->coeffs;
2812     INTFLOAT *out   = sce->ret;
2813     INTFLOAT *saved = sce->saved;
2814     const INTFLOAT *swindow      = ics->use_kb_window[0] ? AAC_RENAME(aac_kbd_short_120) : AAC_RENAME(sine_120);
2815     const INTFLOAT *lwindow_prev = ics->use_kb_window[1] ? AAC_RENAME(aac_kbd_long_960) : AAC_RENAME(sine_960);
2816     const INTFLOAT *swindow_prev = ics->use_kb_window[1] ? AAC_RENAME(aac_kbd_short_120) : AAC_RENAME(sine_120);
2817     INTFLOAT *buf  = ac->buf_mdct;
2818     INTFLOAT *temp = ac->temp;
2819     int i;
2820
2821     // imdct
2822     if (ics->window_sequence[0] == EIGHT_SHORT_SEQUENCE) {
2823         for (i = 0; i < 8; i++)
2824             ac->mdct120->imdct_half(ac->mdct120, buf + i * 120, in + i * 128, 1);
2825     } else {
2826         ac->mdct960->imdct_half(ac->mdct960, buf, in, 1);
2827     }
2828
2829     /* window overlapping
2830      * NOTE: To simplify the overlapping code, all 'meaningless' short to long
2831      * and long to short transitions are considered to be short to short
2832      * transitions. This leaves just two cases (long to long and short to short)
2833      * with a little special sauce for EIGHT_SHORT_SEQUENCE.
2834      */
2835
2836     if ((ics->window_sequence[1] == ONLY_LONG_SEQUENCE || ics->window_sequence[1] == LONG_STOP_SEQUENCE) &&
2837         (ics->window_sequence[0] == ONLY_LONG_SEQUENCE || ics->window_sequence[0] == LONG_START_SEQUENCE)) {
2838         ac->fdsp->vector_fmul_window(    out,               saved,            buf,         lwindow_prev, 480);
2839     } else {
2840         memcpy(                          out,               saved,            420 * sizeof(*out));
2841
2842         if (ics->window_sequence[0] == EIGHT_SHORT_SEQUENCE) {
2843             ac->fdsp->vector_fmul_window(out + 420 + 0*120, saved + 420,      buf + 0*120, swindow_prev, 60);
2844             ac->fdsp->vector_fmul_window(out + 420 + 1*120, buf + 0*120 + 60, buf + 1*120, swindow,      60);
2845             ac->fdsp->vector_fmul_window(out + 420 + 2*120, buf + 1*120 + 60, buf + 2*120, swindow,      60);
2846             ac->fdsp->vector_fmul_window(out + 420 + 3*120, buf + 2*120 + 60, buf + 3*120, swindow,      60);
2847             ac->fdsp->vector_fmul_window(temp,              buf + 3*120 + 60, buf + 4*120, swindow,      60);
2848             memcpy(                      out + 420 + 4*120, temp, 60 * sizeof(*out));
2849         } else {
2850             ac->fdsp->vector_fmul_window(out + 420,         saved + 420,      buf,         swindow_prev, 60);
2851             memcpy(                      out + 540,         buf + 60,         420 * sizeof(*out));
2852         }
2853     }
2854
2855     // buffer update
2856     if (ics->window_sequence[0] == EIGHT_SHORT_SEQUENCE) {
2857         memcpy(                      saved,       temp + 60,         60 * sizeof(*saved));
2858         ac->fdsp->vector_fmul_window(saved + 60,  buf + 4*120 + 60, buf + 5*120, swindow, 60);
2859         ac->fdsp->vector_fmul_window(saved + 180, buf + 5*120 + 60, buf + 6*120, swindow, 60);
2860         ac->fdsp->vector_fmul_window(saved + 300, buf + 6*120 + 60, buf + 7*120, swindow, 60);
2861         memcpy(                      saved + 420, buf + 7*120 + 60,  60 * sizeof(*saved));
2862     } else if (ics->window_sequence[0] == LONG_START_SEQUENCE) {
2863         memcpy(                      saved,       buf + 480,        420 * sizeof(*saved));
2864         memcpy(                      saved + 420, buf + 7*120 + 60,  60 * sizeof(*saved));
2865     } else { // LONG_STOP or ONLY_LONG
2866         memcpy(                      saved,       buf + 480,        480 * sizeof(*saved));
2867     }
2868 #endif
2869 }
2870 static void imdct_and_windowing_ld(AACContext *ac, SingleChannelElement *sce)
2871 {
2872     IndividualChannelStream *ics = &sce->ics;
2873     INTFLOAT *in    = sce->coeffs;
2874     INTFLOAT *out   = sce->ret;
2875     INTFLOAT *saved = sce->saved;
2876     INTFLOAT *buf  = ac->buf_mdct;
2877 #if USE_FIXED
2878     int i;
2879 #endif /* USE_FIXED */
2880
2881     // imdct
2882     ac->mdct.imdct_half(&ac->mdct_ld, buf, in);
2883
2884 #if USE_FIXED
2885     for (i = 0; i < 1024; i++)
2886         buf[i] = (buf[i] + 2) >> 2;
2887 #endif /* USE_FIXED */
2888
2889     // window overlapping
2890     if (ics->use_kb_window[1]) {
2891         // AAC LD uses a low overlap sine window instead of a KBD window
2892         memcpy(out, saved, 192 * sizeof(*out));
2893         ac->fdsp->vector_fmul_window(out + 192, saved + 192, buf, AAC_RENAME2(sine_128), 64);
2894         memcpy(                     out + 320, buf + 64, 192 * sizeof(*out));
2895     } else {
2896         ac->fdsp->vector_fmul_window(out, saved, buf, AAC_RENAME2(sine_512), 256);
2897     }
2898
2899     // buffer update
2900     memcpy(saved, buf + 256, 256 * sizeof(*saved));
2901 }
2902
2903 static void imdct_and_windowing_eld(AACContext *ac, SingleChannelElement *sce)
2904 {
2905     INTFLOAT *in    = sce->coeffs;
2906     INTFLOAT *out   = sce->ret;
2907     INTFLOAT *saved = sce->saved;
2908     INTFLOAT *buf  = ac->buf_mdct;
2909     int i;
2910     const int n  = ac->oc[1].m4ac.frame_length_short ? 480 : 512;
2911     const int n2 = n >> 1;
2912     const int n4 = n >> 2;
2913     const INTFLOAT *const window = n == 480 ? AAC_RENAME(ff_aac_eld_window_480) :
2914                                            AAC_RENAME(ff_aac_eld_window_512);
2915
2916     // Inverse transform, mapped to the conventional IMDCT by
2917     // Chivukula, R.K.; Reznik, Y.A.; Devarajan, V.,
2918     // "Efficient algorithms for MPEG-4 AAC-ELD, AAC-LD and AAC-LC filterbanks,"
2919     // International Conference on Audio, Language and Image Processing, ICALIP 2008.
2920     // URL: http://ieeexplore.ieee.org/stamp/stamp.jsp?tp=&arnumber=4590245&isnumber=4589950
2921     for (i = 0; i < n2; i+=2) {
2922         INTFLOAT temp;
2923         temp =  in[i    ]; in[i    ] = -in[n - 1 - i]; in[n - 1 - i] = temp;
2924         temp = -in[i + 1]; in[i + 1] =  in[n - 2 - i]; in[n - 2 - i] = temp;
2925     }
2926 #if !USE_FIXED
2927     if (n == 480)
2928         ac->mdct480->imdct_half(ac->mdct480, buf, in, 1);
2929     else
2930 #endif
2931         ac->mdct.imdct_half(&ac->mdct_ld, buf, in);
2932
2933 #if USE_FIXED
2934     for (i = 0; i < 1024; i++)
2935       buf[i] = (buf[i] + 1) >> 1;
2936 #endif /* USE_FIXED */
2937
2938     for (i = 0; i < n; i+=2) {
2939         buf[i] = -buf[i];
2940     }
2941     // Like with the regular IMDCT at this point we still have the middle half
2942     // of a transform but with even symmetry on the left and odd symmetry on
2943     // the right
2944
2945     // window overlapping
2946     // The spec says to use samples [0..511] but the reference decoder uses
2947     // samples [128..639].
2948     for (i = n4; i < n2; i ++) {
2949         out[i - n4] = AAC_MUL31(   buf[    n2 - 1 - i] , window[i       - n4]) +
2950                       AAC_MUL31( saved[        i + n2] , window[i +   n - n4]) +
2951                       AAC_MUL31(-saved[n + n2 - 1 - i] , window[i + 2*n - n4]) +
2952                       AAC_MUL31(-saved[  2*n + n2 + i] , window[i + 3*n - n4]);
2953     }
2954     for (i = 0; i < n2; i ++) {
2955         out[n4 + i] = AAC_MUL31(   buf[              i] , window[i + n2       - n4]) +
2956                       AAC_MUL31(-saved[      n - 1 - i] , window[i + n2 +   n - n4]) +
2957                       AAC_MUL31(-saved[          n + i] , window[i + n2 + 2*n - n4]) +
2958                       AAC_MUL31( saved[2*n + n - 1 - i] , window[i + n2 + 3*n - n4]);
2959     }
2960     for (i = 0; i < n4; i ++) {
2961         out[n2 + n4 + i] = AAC_MUL31(   buf[    i + n2] , window[i +   n - n4]) +
2962                            AAC_MUL31(-saved[n2 - 1 - i] , window[i + 2*n - n4]) +
2963                            AAC_MUL31(-saved[n + n2 + i] , window[i + 3*n - n4]);
2964     }
2965
2966     // buffer update
2967     memmove(saved + n, saved, 2 * n * sizeof(*saved));
2968     memcpy( saved,       buf,     n * sizeof(*saved));
2969 }
2970
2971 /**
2972  * channel coupling transformation interface
2973  *
2974  * @param   apply_coupling_method   pointer to (in)dependent coupling function
2975  */
2976 static void apply_channel_coupling(AACContext *ac, ChannelElement *cc,
2977                                    enum RawDataBlockType type, int elem_id,
2978                                    enum CouplingPoint coupling_point,
2979                                    void (*apply_coupling_method)(AACContext *ac, SingleChannelElement *target, ChannelElement *cce, int index))
2980 {
2981     int i, c;
2982
2983     for (i = 0; i < MAX_ELEM_ID; i++) {
2984         ChannelElement *cce = ac->che[TYPE_CCE][i];
2985         int index = 0;
2986
2987         if (cce && cce->coup.coupling_point == coupling_point) {
2988             ChannelCoupling *coup = &cce->coup;
2989
2990             for (c = 0; c <= coup->num_coupled; c++) {
2991                 if (coup->type[c] == type && coup->id_select[c] == elem_id) {
2992                     if (coup->ch_select[c] != 1) {
2993                         apply_coupling_method(ac, &cc->ch[0], cce, index);
2994                         if (coup->ch_select[c] != 0)
2995                             index++;
2996                     }
2997                     if (coup->ch_select[c] != 2)
2998                         apply_coupling_method(ac, &cc->ch[1], cce, index++);
2999                 } else
3000                     index += 1 + (coup->ch_select[c] == 3);
3001             }
3002         }
3003     }
3004 }
3005
3006 /**
3007  * Convert spectral data to samples, applying all supported tools as appropriate.
3008  */
3009 static void spectral_to_sample(AACContext *ac, int samples)
3010 {
3011     int i, type;
3012     void (*imdct_and_window)(AACContext *ac, SingleChannelElement *sce);
3013     switch (ac->oc[1].m4ac.object_type) {
3014     case AOT_ER_AAC_LD:
3015         imdct_and_window = imdct_and_windowing_ld;
3016         break;
3017     case AOT_ER_AAC_ELD:
3018         imdct_and_window = imdct_and_windowing_eld;
3019         break;
3020     default:
3021         if (ac->oc[1].m4ac.frame_length_short)
3022             imdct_and_window = imdct_and_windowing_960;
3023         else
3024             imdct_and_window = ac->imdct_and_windowing;
3025     }
3026     for (type = 3; type >= 0; type--) {
3027         for (i = 0; i < MAX_ELEM_ID; i++) {
3028             ChannelElement *che = ac->che[type][i];
3029             if (che && che->present) {
3030                 if (type <= TYPE_CPE)
3031                     apply_channel_coupling(ac, che, type, i, BEFORE_TNS, AAC_RENAME(apply_dependent_coupling));
3032                 if (ac->oc[1].m4ac.object_type == AOT_AAC_LTP) {
3033                     if (che->ch[0].ics.predictor_present) {
3034                         if (che->ch[0].ics.ltp.present)
3035                             ac->apply_ltp(ac, &che->ch[0]);
3036                         if (che->ch[1].ics.ltp.present && type == TYPE_CPE)
3037                             ac->apply_ltp(ac, &che->ch[1]);
3038                     }
3039                 }
3040                 if (che->ch[0].tns.present)
3041                     ac->apply_tns(che->ch[0].coeffs, &che->ch[0].tns, &che->ch[0].ics, 1);
3042                 if (che->ch[1].tns.present)
3043                     ac->apply_tns(che->ch[1].coeffs, &che->ch[1].tns, &che->ch[1].ics, 1);
3044                 if (type <= TYPE_CPE)
3045                     apply_channel_coupling(ac, che, type, i, BETWEEN_TNS_AND_IMDCT, AAC_RENAME(apply_dependent_coupling));
3046                 if (type != TYPE_CCE || che->coup.coupling_point == AFTER_IMDCT) {
3047                     imdct_and_window(ac, &che->ch[0]);
3048                     if (ac->oc[1].m4ac.object_type == AOT_AAC_LTP)
3049                         ac->update_ltp(ac, &che->ch[0]);
3050                     if (type == TYPE_CPE) {
3051                         imdct_and_window(ac, &che->ch[1]);
3052                         if (ac->oc[1].m4ac.object_type == AOT_AAC_LTP)
3053                             ac->update_ltp(ac, &che->ch[1]);
3054                     }
3055                     if (ac->oc[1].m4ac.sbr > 0) {
3056                         AAC_RENAME(ff_sbr_apply)(ac, &che->sbr, type, che->ch[0].ret, che->ch[1].ret);
3057                     }
3058                 }
3059                 if (type <= TYPE_CCE)
3060                     apply_channel_coupling(ac, che, type, i, AFTER_IMDCT, AAC_RENAME(apply_independent_coupling));
3061
3062 #if USE_FIXED
3063                 {
3064                     int j;
3065                     /* preparation for resampler */
3066                     for(j = 0; j<samples; j++){
3067                         che->ch[0].ret[j] = (int32_t)av_clip64((int64_t)che->ch[0].ret[j]*128, INT32_MIN, INT32_MAX-0x8000)+0x8000;
3068                         if(type == TYPE_CPE)
3069                             che->ch[1].ret[j] = (int32_t)av_clip64((int64_t)che->ch[1].ret[j]*128, INT32_MIN, INT32_MAX-0x8000)+0x8000;
3070                     }
3071                 }
3072 #endif /* USE_FIXED */
3073                 che->present = 0;
3074             } else if (che) {
3075                 av_log(ac->avctx, AV_LOG_VERBOSE, "ChannelElement %d.%d missing \n", type, i);
3076             }
3077         }
3078     }
3079 }
3080
3081 static int parse_adts_frame_header(AACContext *ac, GetBitContext *gb)
3082 {
3083     int size;
3084     AACADTSHeaderInfo hdr_info;
3085     uint8_t layout_map[MAX_ELEM_ID*4][3];
3086     int layout_map_tags, ret;
3087
3088     size = ff_adts_header_parse(gb, &hdr_info);
3089     if (size > 0) {
3090         if (!ac->warned_num_aac_frames && hdr_info.num_aac_frames != 1) {
3091             // This is 2 for "VLB " audio in NSV files.
3092             // See samples/nsv/vlb_audio.
3093             avpriv_report_missing_feature(ac->avctx,
3094                                           "More than one AAC RDB per ADTS frame");
3095             ac->warned_num_aac_frames = 1;
3096         }
3097         push_output_configuration(ac);
3098         if (hdr_info.chan_config) {
3099             ac->oc[1].m4ac.chan_config = hdr_info.chan_config;
3100             if ((ret = set_default_channel_config(ac, ac->avctx,
3101                                                   layout_map,
3102                                                   &layout_map_tags,
3103                                                   hdr_info.chan_config)) < 0)
3104                 return ret;
3105             if ((ret = output_configure(ac, layout_map, layout_map_tags,
3106                                         FFMAX(ac->oc[1].status,
3107                                               OC_TRIAL_FRAME), 0)) < 0)
3108                 return ret;
3109         } else {
3110             ac->oc[1].m4ac.chan_config = 0;
3111             /**
3112              * dual mono frames in Japanese DTV can have chan_config 0
3113              * WITHOUT specifying PCE.
3114              *  thus, set dual mono as default.
3115              */
3116             if (ac->dmono_mode && ac->oc[0].status == OC_NONE) {
3117                 layout_map_tags = 2;
3118                 layout_map[0][0] = layout_map[1][0] = TYPE_SCE;
3119                 layout_map[0][2] = layout_map[1][2] = AAC_CHANNEL_FRONT;
3120                 layout_map[0][1] = 0;
3121                 layout_map[1][1] = 1;
3122                 if (output_configure(ac, layout_map, layout_map_tags,
3123                                      OC_TRIAL_FRAME, 0))
3124                     return -7;
3125             }
3126         }
3127         ac->oc[1].m4ac.sample_rate     = hdr_info.sample_rate;
3128         ac->oc[1].m4ac.sampling_index  = hdr_info.sampling_index;
3129         ac->oc[1].m4ac.object_type     = hdr_info.object_type;
3130         ac->oc[1].m4ac.frame_length_short = 0;
3131         if (ac->oc[0].status != OC_LOCKED ||
3132             ac->oc[0].m4ac.chan_config != hdr_info.chan_config ||
3133             ac->oc[0].m4ac.sample_rate != hdr_info.sample_rate) {
3134             ac->oc[1].m4ac.sbr = -1;
3135             ac->oc[1].m4ac.ps  = -1;
3136         }
3137         if (!hdr_info.crc_absent)
3138             skip_bits(gb, 16);
3139     }
3140     return size;
3141 }
3142
3143 static int aac_decode_er_frame(AVCodecContext *avctx, void *data,
3144                                int *got_frame_ptr, GetBitContext *gb)
3145 {
3146     AACContext *ac = avctx->priv_data;
3147     const MPEG4AudioConfig *const m4ac = &ac->oc[1].m4ac;
3148     ChannelElement *che;
3149     int err, i;
3150     int samples = m4ac->frame_length_short ? 960 : 1024;
3151     int chan_config = m4ac->chan_config;
3152     int aot = m4ac->object_type;
3153
3154     if (aot == AOT_ER_AAC_LD || aot == AOT_ER_AAC_ELD)
3155         samples >>= 1;
3156
3157     ac->frame = data;
3158
3159     if ((err = frame_configure_elements(avctx)) < 0)
3160         return err;
3161
3162     // The FF_PROFILE_AAC_* defines are all object_type - 1
3163     // This may lead to an undefined profile being signaled
3164     ac->avctx->profile = aot - 1;
3165
3166     ac->tags_mapped = 0;
3167
3168     if (chan_config < 0 || (chan_config >= 8 && chan_config < 11) || chan_config >= 13) {
3169         avpriv_request_sample(avctx, "Unknown ER channel configuration %d",
3170                               chan_config);
3171         return AVERROR_INVALIDDATA;
3172     }
3173     for (i = 0; i < tags_per_config[chan_config]; i++) {
3174         const int elem_type = aac_channel_layout_map[chan_config-1][i][0];
3175         const int elem_id   = aac_channel_layout_map[chan_config-1][i][1];
3176         if (!(che=get_che(ac, elem_type, elem_id))) {
3177             av_log(ac->avctx, AV_LOG_ERROR,
3178                    "channel element %d.%d is not allocated\n",
3179                    elem_type, elem_id);
3180             return AVERROR_INVALIDDATA;
3181         }
3182         che->present = 1;
3183         if (aot != AOT_ER_AAC_ELD)
3184             skip_bits(gb, 4);
3185         switch (elem_type) {
3186         case TYPE_SCE:
3187             err = decode_ics(ac, &che->ch[0], gb, 0, 0);
3188             break;
3189         case TYPE_CPE:
3190             err = decode_cpe(ac, gb, che);
3191             break;
3192         case TYPE_LFE:
3193             err = decode_ics(ac, &che->ch[0], gb, 0, 0);
3194             break;
3195         }
3196         if (err < 0)
3197             return err;
3198     }
3199
3200     spectral_to_sample(ac, samples);
3201
3202     if (!ac->frame->data[0] && samples) {
3203         av_log(avctx, AV_LOG_ERROR, "no frame data found\n");
3204         return AVERROR_INVALIDDATA;
3205     }
3206
3207     ac->frame->nb_samples = samples;
3208     ac->frame->sample_rate = avctx->sample_rate;
3209     *got_frame_ptr = 1;
3210
3211     skip_bits_long(gb, get_bits_left(gb));
3212     return 0;
3213 }
3214
3215 static int aac_decode_frame_int(AVCodecContext *avctx, void *data,
3216                                 int *got_frame_ptr, GetBitContext *gb, AVPacket *avpkt)
3217 {
3218     AACContext *ac = avctx->priv_data;
3219     ChannelElement *che = NULL, *che_prev = NULL;
3220     enum RawDataBlockType elem_type, che_prev_type = TYPE_END;
3221     int err, elem_id;
3222     int samples = 0, multiplier, audio_found = 0, pce_found = 0;
3223     int is_dmono, sce_count = 0;
3224     int payload_alignment;
3225     uint8_t che_presence[4][MAX_ELEM_ID] = {{0}};
3226
3227     ac->frame = data;
3228
3229     if (show_bits(gb, 12) == 0xfff) {
3230         if ((err = parse_adts_frame_header(ac, gb)) < 0) {
3231             av_log(avctx, AV_LOG_ERROR, "Error decoding AAC frame header.\n");
3232             goto fail;
3233         }
3234         if (ac->oc[1].m4ac.sampling_index > 12) {
3235             av_log(ac->avctx, AV_LOG_ERROR, "invalid sampling rate index %d\n", ac->oc[1].m4ac.sampling_index);
3236             err = AVERROR_INVALIDDATA;
3237             goto fail;
3238         }
3239     }
3240
3241     if ((err = frame_configure_elements(avctx)) < 0)
3242         goto fail;
3243
3244     // The FF_PROFILE_AAC_* defines are all object_type - 1
3245     // This may lead to an undefined profile being signaled
3246     ac->avctx->profile = ac->oc[1].m4ac.object_type - 1;
3247
3248     payload_alignment = get_bits_count(gb);
3249     ac->tags_mapped = 0;
3250     // parse
3251     while ((elem_type = get_bits(gb, 3)) != TYPE_END) {
3252         elem_id = get_bits(gb, 4);
3253
3254         if (avctx->debug & FF_DEBUG_STARTCODE)
3255             av_log(avctx, AV_LOG_DEBUG, "Elem type:%x id:%x\n", elem_type, elem_id);
3256
3257         if (!avctx->channels && elem_type != TYPE_PCE) {
3258             err = AVERROR_INVALIDDATA;
3259             goto fail;
3260         }
3261
3262         if (elem_type < TYPE_DSE) {
3263             if (che_presence[elem_type][elem_id]) {
3264                 int error = che_presence[elem_type][elem_id] > 1;
3265                 av_log(ac->avctx, error ? AV_LOG_ERROR : AV_LOG_DEBUG, "channel element %d.%d duplicate\n",
3266                        elem_type, elem_id);
3267                 if (error) {
3268                     err = AVERROR_INVALIDDATA;
3269                     goto fail;
3270                 }
3271             }
3272             che_presence[elem_type][elem_id]++;
3273
3274             if (!(che=get_che(ac, elem_type, elem_id))) {
3275                 av_log(ac->avctx, AV_LOG_ERROR, "channel element %d.%d is not allocated\n",
3276                        elem_type, elem_id);
3277                 err = AVERROR_INVALIDDATA;
3278                 goto fail;
3279             }
3280             samples = ac->oc[1].m4ac.frame_length_short ? 960 : 1024;
3281             che->present = 1;
3282         }
3283
3284         switch (elem_type) {
3285
3286         case TYPE_SCE:
3287             err = decode_ics(ac, &che->ch[0], gb, 0, 0);
3288             audio_found = 1;
3289             sce_count++;
3290             break;
3291
3292         case TYPE_CPE:
3293             err = decode_cpe(ac, gb, che);
3294             audio_found = 1;
3295             break;
3296
3297         case TYPE_CCE:
3298             err = decode_cce(ac, gb, che);
3299             break;
3300
3301         case TYPE_LFE:
3302             err = decode_ics(ac, &che->ch[0], gb, 0, 0);
3303             audio_found = 1;
3304             break;
3305
3306         case TYPE_DSE:
3307             err = skip_data_stream_element(ac, gb);
3308             break;
3309
3310         case TYPE_PCE: {
3311             uint8_t layout_map[MAX_ELEM_ID*4][3] = {{0}};
3312             int tags;
3313
3314             int pushed = push_output_configuration(ac);
3315             if (pce_found && !pushed) {
3316                 err = AVERROR_INVALIDDATA;
3317                 goto fail;
3318             }
3319
3320             tags = decode_pce(avctx, &ac->oc[1].m4ac, layout_map, gb,
3321                               payload_alignment);
3322             if (tags < 0) {
3323                 err = tags;
3324                 break;
3325             }
3326             if (pce_found) {
3327                 av_log(avctx, AV_LOG_ERROR,
3328                        "Not evaluating a further program_config_element as this construct is dubious at best.\n");
3329                 pop_output_configuration(ac);
3330             } else {
3331                 err = output_configure(ac, layout_map, tags, OC_TRIAL_PCE, 1);
3332                 if (!err)
3333                     ac->oc[1].m4ac.chan_config = 0;
3334                 pce_found = 1;
3335             }
3336             break;
3337         }
3338
3339         case TYPE_FIL:
3340             if (elem_id == 15)
3341                 elem_id += get_bits(gb, 8) - 1;
3342             if (get_bits_left(gb) < 8 * elem_id) {
3343                     av_log(avctx, AV_LOG_ERROR, "TYPE_FIL: "overread_err);
3344                     err = AVERROR_INVALIDDATA;
3345                     goto fail;
3346             }
3347             err = 0;
3348             while (elem_id > 0) {
3349                 int ret = decode_extension_payload(ac, gb, elem_id, che_prev, che_prev_type);
3350                 if (ret < 0) {
3351                     err = ret;
3352                     break;
3353                 }
3354                 elem_id -= ret;
3355             }
3356             break;
3357
3358         default:
3359             err = AVERROR_BUG; /* should not happen, but keeps compiler happy */
3360             break;
3361         }
3362
3363         if (elem_type < TYPE_DSE) {
3364             che_prev      = che;
3365             che_prev_type = elem_type;
3366         }
3367
3368         if (err)
3369             goto fail;
3370
3371         if (get_bits_left(gb) < 3) {
3372             av_log(avctx, AV_LOG_ERROR, overread_err);
3373             err = AVERROR_INVALIDDATA;
3374             goto fail;
3375         }
3376     }
3377
3378     if (!avctx->channels) {
3379         *got_frame_ptr = 0;
3380         return 0;
3381     }
3382
3383     multiplier = (ac->oc[1].m4ac.sbr == 1) ? ac->oc[1].m4ac.ext_sample_rate > ac->oc[1].m4ac.sample_rate : 0;
3384     samples <<= multiplier;
3385
3386     spectral_to_sample(ac, samples);
3387
3388     if (ac->oc[1].status && audio_found) {
3389         avctx->sample_rate = ac->oc[1].m4ac.sample_rate << multiplier;
3390         avctx->frame_size = samples;
3391         ac->oc[1].status = OC_LOCKED;
3392     }
3393
3394     if (multiplier)
3395         avctx->internal->skip_samples_multiplier = 2;
3396
3397     if (!ac->frame->data[0] && samples) {
3398         av_log(avctx, AV_LOG_ERROR, "no frame data found\n");
3399         err = AVERROR_INVALIDDATA;
3400         goto fail;
3401     }
3402
3403     if (samples) {
3404         ac->frame->nb_samples = samples;
3405         ac->frame->sample_rate = avctx->sample_rate;
3406     } else
3407         av_frame_unref(ac->frame);
3408     *got_frame_ptr = !!samples;
3409
3410     /* for dual-mono audio (SCE + SCE) */
3411     is_dmono = ac->dmono_mode && sce_count == 2 &&
3412                ac->oc[1].channel_layout == (AV_CH_FRONT_LEFT | AV_CH_FRONT_RIGHT);
3413     if (is_dmono) {
3414         if (ac->dmono_mode == 1)
3415             ((AVFrame *)data)->data[1] =((AVFrame *)data)->data[0];
3416         else if (ac->dmono_mode == 2)
3417             ((AVFrame *)data)->data[0] =((AVFrame *)data)->data[1];
3418     }
3419
3420     return 0;
3421 fail:
3422     pop_output_configuration(ac);
3423     return err;
3424 }
3425
3426 static int aac_decode_frame(AVCodecContext *avctx, void *data,
3427                             int *got_frame_ptr, AVPacket *avpkt)
3428 {
3429     AACContext *ac = avctx->priv_data;
3430     const uint8_t *buf = avpkt->data;
3431     int buf_size = avpkt->size;
3432     GetBitContext gb;
3433     int buf_consumed;
3434     int buf_offset;
3435     int err;
3436     int new_extradata_size;
3437     const uint8_t *new_extradata = av_packet_get_side_data(avpkt,
3438                                        AV_PKT_DATA_NEW_EXTRADATA,
3439                                        &new_extradata_size);
3440     int jp_dualmono_size;
3441     const uint8_t *jp_dualmono   = av_packet_get_side_data(avpkt,
3442                                        AV_PKT_DATA_JP_DUALMONO,
3443                                        &jp_dualmono_size);
3444
3445     if (new_extradata) {
3446         /* discard previous configuration */
3447         ac->oc[1].status = OC_NONE;
3448         err = decode_audio_specific_config(ac, ac->avctx, &ac->oc[1].m4ac,
3449                                            new_extradata,
3450                                            new_extradata_size * 8LL, 1);
3451         if (err < 0) {
3452             return err;
3453         }
3454     }
3455
3456     ac->dmono_mode = 0;
3457     if (jp_dualmono && jp_dualmono_size > 0)
3458         ac->dmono_mode =  1 + *jp_dualmono;
3459     if (ac->force_dmono_mode >= 0)
3460         ac->dmono_mode = ac->force_dmono_mode;
3461
3462     if (INT_MAX / 8 <= buf_size)
3463         return AVERROR_INVALIDDATA;
3464
3465     if ((err = init_get_bits8(&gb, buf, buf_size)) < 0)
3466         return err;
3467
3468     switch (ac->oc[1].m4ac.object_type) {
3469     case AOT_ER_AAC_LC:
3470     case AOT_ER_AAC_LTP:
3471     case AOT_ER_AAC_LD:
3472     case AOT_ER_AAC_ELD:
3473         err = aac_decode_er_frame(avctx, data, got_frame_ptr, &gb);
3474         break;
3475     default:
3476         err = aac_decode_frame_int(avctx, data, got_frame_ptr, &gb, avpkt);
3477     }
3478     if (err < 0)
3479         return err;
3480
3481     buf_consumed = (get_bits_count(&gb) + 7) >> 3;
3482     for (buf_offset = buf_consumed; buf_offset < buf_size; buf_offset++)
3483         if (buf[buf_offset])
3484             break;
3485
3486     return buf_size > buf_offset ? buf_consumed : buf_size;
3487 }
3488
3489 static av_cold int aac_decode_close(AVCodecContext *avctx)
3490 {
3491     AACContext *ac = avctx->priv_data;
3492     int i, type;
3493
3494     for (i = 0; i < MAX_ELEM_ID; i++) {
3495         for (type = 0; type < 4; type++) {
3496             if (ac->che[type][i])
3497                 AAC_RENAME(ff_aac_sbr_ctx_close)(&ac->che[type][i]->sbr);
3498             av_freep(&ac->che[type][i]);
3499         }
3500     }
3501
3502     ff_mdct_end(&ac->mdct);
3503     ff_mdct_end(&ac->mdct_small);
3504     ff_mdct_end(&ac->mdct_ld);
3505     ff_mdct_end(&ac->mdct_ltp);
3506 #if !USE_FIXED
3507     ff_mdct15_uninit(&ac->mdct120);
3508     ff_mdct15_uninit(&ac->mdct480);
3509     ff_mdct15_uninit(&ac->mdct960);
3510 #endif
3511     av_freep(&ac->fdsp);
3512     return 0;
3513 }
3514
3515 static void aacdec_init(AACContext *c)
3516 {
3517     c->imdct_and_windowing                      = imdct_and_windowing;
3518     c->apply_ltp                                = apply_ltp;
3519     c->apply_tns                                = apply_tns;
3520     c->windowing_and_mdct_ltp                   = windowing_and_mdct_ltp;
3521     c->update_ltp                               = update_ltp;
3522 #if USE_FIXED
3523     c->vector_pow43                             = vector_pow43;
3524     c->subband_scale                            = subband_scale;
3525 #endif
3526
3527 #if !USE_FIXED
3528     if(ARCH_MIPS)
3529         ff_aacdec_init_mips(c);
3530 #endif /* !USE_FIXED */
3531 }
3532 /**
3533  * AVOptions for Japanese DTV specific extensions (ADTS only)
3534  */
3535 #define AACDEC_FLAGS AV_OPT_FLAG_DECODING_PARAM | AV_OPT_FLAG_AUDIO_PARAM
3536 static const AVOption options[] = {
3537     {"dual_mono_mode", "Select the channel to decode for dual mono",
3538      offsetof(AACContext, force_dmono_mode), AV_OPT_TYPE_INT, {.i64=-1}, -1, 2,
3539      AACDEC_FLAGS, "dual_mono_mode"},
3540
3541     {"auto", "autoselection",            0, AV_OPT_TYPE_CONST, {.i64=-1}, INT_MIN, INT_MAX, AACDEC_FLAGS, "dual_mono_mode"},
3542     {"main", "Select Main/Left channel", 0, AV_OPT_TYPE_CONST, {.i64= 1}, INT_MIN, INT_MAX, AACDEC_FLAGS, "dual_mono_mode"},
3543     {"sub" , "Select Sub/Right channel", 0, AV_OPT_TYPE_CONST, {.i64= 2}, INT_MIN, INT_MAX, AACDEC_FLAGS, "dual_mono_mode"},
3544     {"both", "Select both channels",     0, AV_OPT_TYPE_CONST, {.i64= 0}, INT_MIN, INT_MAX, AACDEC_FLAGS, "dual_mono_mode"},
3545
3546     {NULL},
3547 };
3548
3549 static const AVClass aac_decoder_class = {
3550     .class_name = "AAC decoder",
3551     .item_name  = av_default_item_name,
3552     .option     = options,
3553     .version    = LIBAVUTIL_VERSION_INT,
3554 };