ac3_parser.c 6.64 KB
Newer Older
1
/*
2
 * AC-3 parser
3 4
 * Copyright (c) 2003 Fabrice Bellard
 * Copyright (c) 2003 Michael Niedermayer
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
 *
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * FFmpeg is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with FFmpeg; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

#include "parser.h"
#include "ac3_parser.h"
#include "aac_ac3_parser.h"
26
#include "get_bits.h"
27 28 29 30 31 32 33 34 35


#define AC3_HEADER_SIZE 7


static const uint8_t eac3_blocks[4] = {
    1, 2, 3, 6
};

36 37

int ff_ac3_parse_header(GetBitContext *gbc, AC3HeaderInfo *hdr)
38
{
39
    int frame_size_code;
40 41 42

    memset(hdr, 0, sizeof(*hdr));

43
    hdr->sync_word = get_bits(gbc, 16);
44
    if(hdr->sync_word != 0x0B77)
45
        return AAC_AC3_PARSE_ERROR_SYNC;
46

47
    /* read ahead to bsid to distinguish between AC-3 and E-AC-3 */
48
    hdr->bitstream_id = show_bits_long(gbc, 29) & 0x1F;
49
    if(hdr->bitstream_id > 16)
50
        return AAC_AC3_PARSE_ERROR_BSID;
51

52 53
    hdr->num_blocks = 6;

54
    /* set default mix levels */
55 56
    hdr->center_mix_level   = 1;  // -4.5dB
    hdr->surround_mix_level = 1;  // -6.0dB
57

58
    if(hdr->bitstream_id <= 10) {
59
        /* Normal AC-3 */
60 61
        hdr->crc1 = get_bits(gbc, 16);
        hdr->sr_code = get_bits(gbc, 2);
62
        if(hdr->sr_code == 3)
63
            return AAC_AC3_PARSE_ERROR_SAMPLE_RATE;
64

65
        frame_size_code = get_bits(gbc, 6);
66
        if(frame_size_code > 37)
67
            return AAC_AC3_PARSE_ERROR_FRAME_SIZE;
68

69 70 71 72 73
        skip_bits(gbc, 5); // skip bsid, already got it

        skip_bits(gbc, 3); // skip bitstream mode
        hdr->channel_mode = get_bits(gbc, 3);

74
        if(hdr->channel_mode == AC3_CHMODE_STEREO) {
75 76 77
            skip_bits(gbc, 2); // skip dsurmod
        } else {
            if((hdr->channel_mode & 1) && hdr->channel_mode != AC3_CHMODE_MONO)
78
                hdr->center_mix_level = get_bits(gbc, 2);
79
            if(hdr->channel_mode & 4)
80
                hdr->surround_mix_level = get_bits(gbc, 2);
81
        }
82
        hdr->lfe_on = get_bits1(gbc);
83 84 85 86 87 88

        hdr->sr_shift = FFMAX(hdr->bitstream_id, 8) - 8;
        hdr->sample_rate = ff_ac3_sample_rate_tab[hdr->sr_code] >> hdr->sr_shift;
        hdr->bit_rate = (ff_ac3_bitrate_tab[frame_size_code>>1] * 1000) >> hdr->sr_shift;
        hdr->channels = ff_ac3_channels_tab[hdr->channel_mode] + hdr->lfe_on;
        hdr->frame_size = ff_ac3_frame_size_tab[frame_size_code][hdr->sr_code] * 2;
89
        hdr->frame_type = EAC3_FRAME_TYPE_AC3_CONVERT; //EAC3_FRAME_TYPE_INDEPENDENT;
90
        hdr->substreamid = 0;
91 92 93
    } else {
        /* Enhanced AC-3 */
        hdr->crc1 = 0;
94
        hdr->frame_type = get_bits(gbc, 2);
95
        if(hdr->frame_type == EAC3_FRAME_TYPE_RESERVED)
96
            return AAC_AC3_PARSE_ERROR_FRAME_TYPE;
97

98
        hdr->substreamid = get_bits(gbc, 3);
99

100
        hdr->frame_size = (get_bits(gbc, 11) + 1) << 1;
101
        if(hdr->frame_size < AC3_HEADER_SIZE)
102
            return AAC_AC3_PARSE_ERROR_FRAME_SIZE;
103

104
        hdr->sr_code = get_bits(gbc, 2);
105
        if (hdr->sr_code == 3) {
106
            int sr_code2 = get_bits(gbc, 2);
107
            if(sr_code2 == 3)
108
                return AAC_AC3_PARSE_ERROR_SAMPLE_RATE;
109 110 111
            hdr->sample_rate = ff_ac3_sample_rate_tab[sr_code2] / 2;
            hdr->sr_shift = 1;
        } else {
112
            hdr->num_blocks = eac3_blocks[get_bits(gbc, 2)];
113 114 115 116
            hdr->sample_rate = ff_ac3_sample_rate_tab[hdr->sr_code];
            hdr->sr_shift = 0;
        }

117 118
        hdr->channel_mode = get_bits(gbc, 3);
        hdr->lfe_on = get_bits1(gbc);
119 120

        hdr->bit_rate = (uint32_t)(8.0 * hdr->frame_size * hdr->sample_rate /
121
                        (hdr->num_blocks * 256.0));
122 123
        hdr->channels = ff_ac3_channels_tab[hdr->channel_mode] + hdr->lfe_on;
    }
124 125 126
    hdr->channel_layout = ff_ac3_channel_layout_tab[hdr->channel_mode];
    if (hdr->lfe_on)
        hdr->channel_layout |= CH_LOW_FREQUENCY;
127 128 129 130

    return 0;
}

131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159
int ff_ac3_parse_header_full(GetBitContext *gbc, AC3HeaderInfo *hdr){
    int ret, i;
    ret = ff_ac3_parse_header(gbc, hdr);
    if(!ret){
        if(hdr->bitstream_id>10){
            /* Enhanced AC-3 */
            skip_bits(gbc, 5); // skip bitstream id

            /* skip dialog normalization and compression gain */
            for (i = 0; i < (hdr->channel_mode ? 1 : 2); i++) {
                skip_bits(gbc, 5); // skip dialog normalization
                if (get_bits1(gbc)) {
                    skip_bits(gbc, 8); //skip Compression gain word
                }
            }
            /* dependent stream channel map */
            if (hdr->frame_type == EAC3_FRAME_TYPE_DEPENDENT && get_bits1(gbc)) {
                    hdr->channel_map = get_bits(gbc, 16); //custom channel map
                    return 0;
            }
        }
        //default channel map based on acmod and lfeon
        hdr->channel_map = ff_eac3_default_chmap[hdr->channel_mode];
        if(hdr->lfe_on)
            hdr->channel_map |= AC3_CHMAP_LFE;
    }
    return ret;
}

160 161
static int ac3_sync(uint64_t state, AACAC3ParseContext *hdr_info,
        int *need_next_header, int *new_frame_start)
162 163
{
    int err;
164 165 166 167
    union {
        uint64_t u64;
        uint8_t  u8[8];
    } tmp = { be2me_64(state) };
168
    AC3HeaderInfo hdr;
169
    GetBitContext gbc;
170

171
    init_get_bits(&gbc, tmp.u8+8-AC3_HEADER_SIZE, 54);
172
    err = ff_ac3_parse_header(&gbc, &hdr);
173

174
    if(err < 0)
175 176
        return 0;

177 178 179
    hdr_info->sample_rate = hdr.sample_rate;
    hdr_info->bit_rate = hdr.bit_rate;
    hdr_info->channels = hdr.channels;
180
    hdr_info->channel_layout = hdr.channel_layout;
181
    hdr_info->samples = hdr.num_blocks * 256;
182 183
    if(hdr.bitstream_id>10)
        hdr_info->codec_id = CODEC_ID_EAC3;
184
    else if (hdr_info->codec_id == CODEC_ID_NONE)
185
        hdr_info->codec_id = CODEC_ID_AC3;
186

187 188
    *need_next_header = (hdr.frame_type != EAC3_FRAME_TYPE_AC3_CONVERT);
    *new_frame_start  = (hdr.frame_type != EAC3_FRAME_TYPE_DEPENDENT);
189
    return hdr.frame_size;
190 191
}

192
static av_cold int ac3_parse_init(AVCodecParserContext *s1)
193
{
194
    AACAC3ParseContext *s = s1->priv_data;
195 196 197 198 199 200 201
    s->header_size = AC3_HEADER_SIZE;
    s->sync = ac3_sync;
    return 0;
}


AVCodecParser ac3_parser = {
202
    { CODEC_ID_AC3, CODEC_ID_EAC3 },
203
    sizeof(AACAC3ParseContext),
204
    ac3_parse_init,
205
    ff_aac_ac3_parse,
David Liu's avatar
David Liu committed
206
    ff_parse_close,
207
};