mpegaudio_parser.c 4.22 KB
Newer Older
1 2
/*
 * MPEG Audio 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
 *
 * 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 "mpegaudio.h"
25
#include "mpegaudiodecheader.h"
26 27 28


typedef struct MpegAudioParseContext {
29
    ParseContext pc;
30 31 32 33 34 35 36 37 38 39 40 41
    int frame_size;
    uint32_t header;
    int header_count;
} MpegAudioParseContext;

#define MPA_HEADER_SIZE 4

/* header + layer + bitrate + freq + lsf/mpeg25 */
#undef SAME_HEADER_MASK /* mpegaudio.h defines different version */
#define SAME_HEADER_MASK \
   (0xffe00000 | (3 << 17) | (3 << 10) | (3 << 19))

42 43
/* useful helper to get mpeg audio stream infos. Return -1 if error in
   header, otherwise the coded frame size in bytes */
44
int ff_mpa_decode_header(AVCodecContext *avctx, uint32_t head, int *sample_rate, int *channels, int *frame_size, int *bit_rate)
45
{
46
    MPADecodeHeader s1, *s = &s1;
47 48 49 50

    if (ff_mpa_check_header(head) != 0)
        return -1;

51
    if (ff_mpegaudio_decode_header(s, head) != 0) {
52 53 54 55 56
        return -1;
    }

    switch(s->layer) {
    case 1:
57
        avctx->codec_id = CODEC_ID_MP1;
58
        *frame_size = 384;
59 60
        break;
    case 2:
61
        avctx->codec_id = CODEC_ID_MP2;
62
        *frame_size = 1152;
63 64 65
        break;
    default:
    case 3:
66
        avctx->codec_id = CODEC_ID_MP3;
67
        if (s->lsf)
68
            *frame_size = 576;
69
        else
70
            *frame_size = 1152;
71 72 73 74
        break;
    }

    *sample_rate = s->sample_rate;
75 76
    *channels = s->nb_channels;
    *bit_rate = s->bit_rate;
77 78 79 80
    avctx->sub_id = s->layer;
    return s->frame_size;
}

81 82
static int mpegaudio_parse(AVCodecParserContext *s1,
                           AVCodecContext *avctx,
83
                           const uint8_t **poutbuf, int *poutbuf_size,
84 85 86
                           const uint8_t *buf, int buf_size)
{
    MpegAudioParseContext *s = s1->priv_data;
87 88 89 90 91 92 93 94 95 96 97 98 99 100
    ParseContext *pc = &s->pc;
    uint32_t state= pc->state;
    int i;
    int next= END_NOT_FOUND;

    for(i=0; i<buf_size; ){
        if(s->frame_size){
            int inc= FFMIN(buf_size - i, s->frame_size);
            i += inc;
            s->frame_size -= inc;

            if(!s->frame_size){
                next= i;
                break;
101
            }
102 103 104 105 106
        }else{
            while(i<buf_size){
                int ret, sr, channels, bit_rate, frame_size;

                state= (state<<8) + buf[i++];
107

108 109
                ret = ff_mpa_decode_header(avctx, state, &sr, &channels, &frame_size, &bit_rate);
                if (ret < 4) {
110 111
                    s->header_count= -2;
                } else {
112
                    if((state&SAME_HEADER_MASK) != (s->header&SAME_HEADER_MASK) && s->header)
113
                        s->header_count= -3;
114
                    s->header= state;
115
                    s->header_count++;
116
                    s->frame_size = ret-4;
117

118
                    if(s->header_count > 1){
Måns Rullgård's avatar
Måns Rullgård committed
119
                        avctx->sample_rate= sr;
120 121 122 123
                        avctx->channels   = channels;
                        avctx->frame_size = frame_size;
                        avctx->bit_rate   = bit_rate;
                    }
124
                    break;
125
                }
126 127
            }
        }
128
    }
129

130 131 132 133 134
    pc->state= state;
    if (ff_combine_frame(pc, next, &buf, &buf_size) < 0) {
        *poutbuf = NULL;
        *poutbuf_size = 0;
        return buf_size;
135
    }
136 137 138 139

    *poutbuf = buf;
    *poutbuf_size = buf_size;
    return next;
140 141 142 143
}


AVCodecParser mpegaudio_parser = {
144
    { CODEC_ID_MP1, CODEC_ID_MP2, CODEC_ID_MP3 },
145
    sizeof(MpegAudioParseContext),
146
    NULL,
147
    mpegaudio_parse,
148
    ff_parse_close,
149
};