Mercurial > libavcodec.hg
annotate parser.c @ 4912:5fc99f2a111b libavcodec
move mpeg4video_parser in it's own file
| author | aurel |
|---|---|
| date | Sat, 05 May 2007 17:59:37 +0000 |
| parents | 40f3a7f2b1fd |
| children | b7bde71aa752 |
| rev | line source |
|---|---|
| 1613 | 1 /* |
| 2 * Audio and Video frame extraction | |
| 3 * Copyright (c) 2003 Fabrice Bellard. | |
| 4 * Copyright (c) 2003 Michael Niedermayer. | |
| 5 * | |
|
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3776
diff
changeset
|
6 * This file is part of FFmpeg. |
|
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3776
diff
changeset
|
7 * |
|
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3776
diff
changeset
|
8 * FFmpeg is free software; you can redistribute it and/or |
| 1613 | 9 * modify it under the terms of the GNU Lesser General Public |
| 10 * License as published by the Free Software Foundation; either | |
|
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3776
diff
changeset
|
11 * version 2.1 of the License, or (at your option) any later version. |
| 1613 | 12 * |
|
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3776
diff
changeset
|
13 * FFmpeg is distributed in the hope that it will be useful, |
| 1613 | 14 * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
| 16 * Lesser General Public License for more details. | |
| 17 * | |
| 18 * You should have received a copy of the GNU Lesser General Public | |
|
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3776
diff
changeset
|
19 * License along with FFmpeg; if not, write to the Free Software |
|
3036
0b546eab515d
Update licensing information: The FSF changed postal address.
diego
parents:
2979
diff
changeset
|
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
| 1613 | 21 */ |
| 22 #include "avcodec.h" | |
| 23 #include "mpegvideo.h" | |
| 24 #include "mpegaudio.h" | |
|
4150
2205aefb22b7
move AVCodecParser prototypes and definitions to parser.h, and move mpegvideo parser to mpeg12.c
bcoudurier
parents:
4146
diff
changeset
|
25 #include "parser.h" |
| 1613 | 26 |
| 27 AVCodecParser *av_first_parser = NULL; | |
| 28 | |
| 29 void av_register_codec_parser(AVCodecParser *parser) | |
| 30 { | |
| 31 parser->next = av_first_parser; | |
| 32 av_first_parser = parser; | |
| 33 } | |
| 34 | |
| 35 AVCodecParserContext *av_parser_init(int codec_id) | |
| 36 { | |
| 37 AVCodecParserContext *s; | |
| 38 AVCodecParser *parser; | |
| 39 int ret; | |
| 2967 | 40 |
|
2486
f2a9559db6ac
10l (array gets padded with 0 which is CODEC_ID_NONE -> parsers claim to support CODEC_ID_NONE)
michael
parents:
2480
diff
changeset
|
41 if(codec_id == CODEC_ID_NONE) |
|
f2a9559db6ac
10l (array gets padded with 0 which is CODEC_ID_NONE -> parsers claim to support CODEC_ID_NONE)
michael
parents:
2480
diff
changeset
|
42 return NULL; |
| 1613 | 43 |
| 44 for(parser = av_first_parser; parser != NULL; parser = parser->next) { | |
| 45 if (parser->codec_ids[0] == codec_id || | |
| 46 parser->codec_ids[1] == codec_id || | |
| 2348 | 47 parser->codec_ids[2] == codec_id || |
| 48 parser->codec_ids[3] == codec_id || | |
| 49 parser->codec_ids[4] == codec_id) | |
| 1613 | 50 goto found; |
| 51 } | |
| 52 return NULL; | |
| 53 found: | |
| 54 s = av_mallocz(sizeof(AVCodecParserContext)); | |
| 55 if (!s) | |
| 56 return NULL; | |
| 57 s->parser = parser; | |
| 58 s->priv_data = av_mallocz(parser->priv_data_size); | |
| 59 if (!s->priv_data) { | |
| 60 av_free(s); | |
| 61 return NULL; | |
| 62 } | |
| 63 if (parser->parser_init) { | |
| 64 ret = parser->parser_init(s); | |
| 65 if (ret != 0) { | |
| 66 av_free(s->priv_data); | |
| 67 av_free(s); | |
| 68 return NULL; | |
| 69 } | |
| 70 } | |
| 2030 | 71 s->fetch_timestamp=1; |
|
4739
9b4c5d2fb8ce
set pict_type to I type during init so parsers which dont set it get all i frames, fixes mp3 seeking
michael
parents:
4679
diff
changeset
|
72 s->pict_type = FF_I_TYPE; |
| 1613 | 73 return s; |
| 74 } | |
| 75 | |
| 3989 | 76 /** |
| 77 * | |
| 78 * @param buf input | |
| 79 * @param buf_size input length, to signal EOF, this should be 0 (so that the last frame can be output) | |
| 80 * @param pts input presentation timestamp | |
| 81 * @param dts input decoding timestamp | |
| 82 * @param poutbuf will contain a pointer to the first byte of the output frame | |
| 83 * @param poutbuf_size will contain the length of the output frame | |
| 84 * @return the number of bytes of the input bitstream used | |
| 85 * | |
| 86 * Example: | |
| 87 * @code | |
| 88 * while(in_len){ | |
| 89 * len = av_parser_parse(myparser, AVCodecContext, &data, &size, | |
| 90 * in_data, in_len, | |
| 91 * pts, dts); | |
| 92 * in_data += len; | |
| 93 * in_len -= len; | |
| 94 * | |
| 4310 | 95 * if(size) |
| 96 * decode_frame(data, size); | |
| 3989 | 97 * } |
| 98 * @endcode | |
| 99 */ | |
| 2967 | 100 int av_parser_parse(AVCodecParserContext *s, |
| 1613 | 101 AVCodecContext *avctx, |
| 2967 | 102 uint8_t **poutbuf, int *poutbuf_size, |
| 1696 | 103 const uint8_t *buf, int buf_size, |
| 104 int64_t pts, int64_t dts) | |
| 1613 | 105 { |
| 1696 | 106 int index, i, k; |
|
1694
13169235c306
added End Of File handling to return last picture for MPEG1/2/4
bellard
parents:
1681
diff
changeset
|
107 uint8_t dummy_buf[FF_INPUT_BUFFER_PADDING_SIZE]; |
| 2967 | 108 |
|
1694
13169235c306
added End Of File handling to return last picture for MPEG1/2/4
bellard
parents:
1681
diff
changeset
|
109 if (buf_size == 0) { |
|
13169235c306
added End Of File handling to return last picture for MPEG1/2/4
bellard
parents:
1681
diff
changeset
|
110 /* padding is always necessary even if EOF, so we add it here */ |
|
13169235c306
added End Of File handling to return last picture for MPEG1/2/4
bellard
parents:
1681
diff
changeset
|
111 memset(dummy_buf, 0, sizeof(dummy_buf)); |
|
13169235c306
added End Of File handling to return last picture for MPEG1/2/4
bellard
parents:
1681
diff
changeset
|
112 buf = dummy_buf; |
| 1696 | 113 } else { |
| 114 /* add a new packet descriptor */ | |
| 115 k = (s->cur_frame_start_index + 1) & (AV_PARSER_PTS_NB - 1); | |
| 116 s->cur_frame_start_index = k; | |
| 117 s->cur_frame_offset[k] = s->cur_offset; | |
| 118 s->cur_frame_pts[k] = pts; | |
| 119 s->cur_frame_dts[k] = dts; | |
| 120 | |
| 121 /* fill first PTS/DTS */ | |
| 2030 | 122 if (s->fetch_timestamp){ |
| 123 s->fetch_timestamp=0; | |
| 1696 | 124 s->last_pts = pts; |
| 125 s->last_dts = dts; | |
| 4846 | 126 s->last_offset = 0; |
| 2107 | 127 s->cur_frame_pts[k] = |
| 128 s->cur_frame_dts[k] = AV_NOPTS_VALUE; | |
| 1696 | 129 } |
|
1694
13169235c306
added End Of File handling to return last picture for MPEG1/2/4
bellard
parents:
1681
diff
changeset
|
130 } |
|
13169235c306
added End Of File handling to return last picture for MPEG1/2/4
bellard
parents:
1681
diff
changeset
|
131 |
| 1613 | 132 /* WARNING: the returned index can be negative */ |
| 133 index = s->parser->parser_parse(s, avctx, poutbuf, poutbuf_size, buf, buf_size); | |
|
4122
daae66c03857
Replace most of the %lld and %llx by their (cleaner) PRI*64 counterparts.
diego
parents:
4104
diff
changeset
|
134 //av_log(NULL, AV_LOG_DEBUG, "parser: in:%"PRId64", %"PRId64", out:%"PRId64", %"PRId64", in:%d out:%d id:%d\n", pts, dts, s->last_pts, s->last_dts, buf_size, *poutbuf_size, avctx->codec_id); |
| 1613 | 135 /* update the file pointer */ |
| 136 if (*poutbuf_size) { | |
| 1696 | 137 /* fill the data for the current frame */ |
| 1613 | 138 s->frame_offset = s->last_frame_offset; |
| 1696 | 139 s->pts = s->last_pts; |
| 140 s->dts = s->last_dts; | |
| 4846 | 141 s->offset = s->last_offset; |
| 2967 | 142 |
| 1696 | 143 /* offset of the next frame */ |
| 1613 | 144 s->last_frame_offset = s->cur_offset + index; |
| 1696 | 145 /* find the packet in which the new frame starts. It |
| 146 is tricky because of MPEG video start codes | |
| 147 which can begin in one packet and finish in | |
| 148 another packet. In the worst case, an MPEG | |
| 149 video start code could be in 4 different | |
| 150 packets. */ | |
| 151 k = s->cur_frame_start_index; | |
| 152 for(i = 0; i < AV_PARSER_PTS_NB; i++) { | |
| 153 if (s->last_frame_offset >= s->cur_frame_offset[k]) | |
| 154 break; | |
| 155 k = (k - 1) & (AV_PARSER_PTS_NB - 1); | |
| 156 } | |
| 2030 | 157 |
| 1696 | 158 s->last_pts = s->cur_frame_pts[k]; |
| 159 s->last_dts = s->cur_frame_dts[k]; | |
| 4846 | 160 s->last_offset = s->last_frame_offset - s->cur_frame_offset[k]; |
| 2967 | 161 |
| 2030 | 162 /* some parsers tell us the packet size even before seeing the first byte of the next packet, |
| 163 so the next pts/dts is in the next chunk */ | |
| 164 if(index == buf_size){ | |
| 165 s->fetch_timestamp=1; | |
| 166 } | |
| 1613 | 167 } |
| 168 if (index < 0) | |
| 169 index = 0; | |
| 170 s->cur_offset += index; | |
| 171 return index; | |
| 172 } | |
| 173 | |
| 2777 | 174 /** |
| 175 * | |
| 176 * @return 0 if the output buffer is a subset of the input, 1 if it is allocated and must be freed | |
|
3421
b7826511f7b6
AVBitStreamFilter (some thingy which can modify the bitstream like add or remove global headers or change the headers or ...)
michael
parents:
3395
diff
changeset
|
177 * @deprecated use AVBitstreamFilter |
| 2777 | 178 */ |
|
2769
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
179 int av_parser_change(AVCodecParserContext *s, |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
180 AVCodecContext *avctx, |
| 2967 | 181 uint8_t **poutbuf, int *poutbuf_size, |
|
2769
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
182 const uint8_t *buf, int buf_size, int keyframe){ |
| 2967 | 183 |
|
2769
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
184 if(s && s->parser->split){ |
| 2777 | 185 if((avctx->flags & CODEC_FLAG_GLOBAL_HEADER) || (avctx->flags2 & CODEC_FLAG2_LOCAL_HEADER)){ |
|
2769
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
186 int i= s->parser->split(avctx, buf, buf_size); |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
187 buf += i; |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
188 buf_size -= i; |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
189 } |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
190 } |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
191 |
|
2864
95bac7109ff0
Kill some compiler warnings. Compiled code verified identical after changes.
mru
parents:
2846
diff
changeset
|
192 /* cast to avoid warning about discarding qualifiers */ |
|
95bac7109ff0
Kill some compiler warnings. Compiled code verified identical after changes.
mru
parents:
2846
diff
changeset
|
193 *poutbuf= (uint8_t *) buf; |
|
2769
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
194 *poutbuf_size= buf_size; |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
195 if(avctx->extradata){ |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
196 if( (keyframe && (avctx->flags2 & CODEC_FLAG2_LOCAL_HEADER)) |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
197 /*||(s->pict_type != I_TYPE && (s->flags & PARSER_FLAG_DUMP_EXTRADATA_AT_NOKEY))*/ |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
198 /*||(? && (s->flags & PARSER_FLAG_DUMP_EXTRADATA_AT_BEGIN)*/){ |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
199 int size= buf_size + avctx->extradata_size; |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
200 *poutbuf_size= size; |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
201 *poutbuf= av_malloc(size + FF_INPUT_BUFFER_PADDING_SIZE); |
| 2967 | 202 |
|
2769
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
203 memcpy(*poutbuf, avctx->extradata, avctx->extradata_size); |
| 2777 | 204 memcpy((*poutbuf) + avctx->extradata_size, buf, buf_size + FF_INPUT_BUFFER_PADDING_SIZE); |
|
2769
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
205 return 1; |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
206 } |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
207 } |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
208 |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
209 return 0; |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
210 } |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
211 |
| 1613 | 212 void av_parser_close(AVCodecParserContext *s) |
| 213 { | |
| 214 if (s->parser->parser_close) | |
| 215 s->parser->parser_close(s); | |
| 216 av_free(s->priv_data); | |
| 217 av_free(s); | |
| 218 } | |
| 219 | |
| 220 /*****************************************************/ | |
| 221 | |
| 222 /** | |
| 223 * combines the (truncated) bitstream to a complete frame | |
| 224 * @returns -1 if no complete frame could be created | |
| 225 */ | |
| 1988 | 226 int ff_combine_frame(ParseContext *pc, int next, uint8_t **buf, int *buf_size) |
| 1613 | 227 { |
| 228 #if 0 | |
| 229 if(pc->overread){ | |
| 230 printf("overread %d, state:%X next:%d index:%d o_index:%d\n", pc->overread, pc->state, next, pc->index, pc->overread_index); | |
| 231 printf("%X %X %X %X\n", (*buf)[0], (*buf)[1],(*buf)[2],(*buf)[3]); | |
| 232 } | |
| 233 #endif | |
| 234 | |
| 4795 | 235 /* Copy overread bytes from last frame into buffer. */ |
| 1613 | 236 for(; pc->overread>0; pc->overread--){ |
| 237 pc->buffer[pc->index++]= pc->buffer[pc->overread_index++]; | |
| 238 } | |
| 2386 | 239 |
| 240 /* flush remaining if EOF */ | |
| 241 if(!*buf_size && next == END_NOT_FOUND){ | |
| 242 next= 0; | |
| 243 } | |
| 244 | |
| 1613 | 245 pc->last_index= pc->index; |
| 246 | |
| 247 /* copy into buffer end return */ | |
| 248 if(next == END_NOT_FOUND){ | |
| 249 pc->buffer= av_fast_realloc(pc->buffer, &pc->buffer_size, (*buf_size) + pc->index + FF_INPUT_BUFFER_PADDING_SIZE); | |
| 250 | |
| 251 memcpy(&pc->buffer[pc->index], *buf, *buf_size); | |
| 252 pc->index += *buf_size; | |
| 253 return -1; | |
| 254 } | |
| 255 | |
| 256 *buf_size= | |
| 257 pc->overread_index= pc->index + next; | |
| 2967 | 258 |
| 1613 | 259 /* append to buffer */ |
| 260 if(pc->index){ | |
| 261 pc->buffer= av_fast_realloc(pc->buffer, &pc->buffer_size, next + pc->index + FF_INPUT_BUFFER_PADDING_SIZE); | |
| 262 | |
| 263 memcpy(&pc->buffer[pc->index], *buf, next + FF_INPUT_BUFFER_PADDING_SIZE ); | |
| 264 pc->index = 0; | |
| 265 *buf= pc->buffer; | |
| 266 } | |
| 267 | |
| 268 /* store overread bytes */ | |
| 269 for(;next < 0; next++){ | |
| 270 pc->state = (pc->state<<8) | pc->buffer[pc->last_index + next]; | |
| 271 pc->overread++; | |
| 272 } | |
| 273 | |
| 274 #if 0 | |
| 275 if(pc->overread){ | |
| 276 printf("overread %d, state:%X next:%d index:%d o_index:%d\n", pc->overread, pc->state, next, pc->index, pc->overread_index); | |
| 277 printf("%X %X %X %X\n", (*buf)[0], (*buf)[1],(*buf)[2],(*buf)[3]); | |
| 278 } | |
| 279 #endif | |
| 280 | |
| 281 return 0; | |
| 282 } | |
| 283 | |
| 1988 | 284 void ff_parse_close(AVCodecParserContext *s) |
| 1613 | 285 { |
| 1988 | 286 ParseContext *pc = s->priv_data; |
| 1613 | 287 |
| 288 av_free(pc->buffer); | |
| 1988 | 289 } |
| 290 | |
|
4150
2205aefb22b7
move AVCodecParser prototypes and definitions to parser.h, and move mpegvideo parser to mpeg12.c
bcoudurier
parents:
4146
diff
changeset
|
291 void ff_parse1_close(AVCodecParserContext *s) |
| 1988 | 292 { |
| 293 ParseContext1 *pc1 = s->priv_data; | |
| 294 | |
| 295 av_free(pc1->pc.buffer); | |
| 296 av_free(pc1->enc); | |
| 1613 | 297 } |
| 298 | |
| 299 /*************************/ | |
| 300 | |
|
4175
b3328ed50a5e
make mpeg4video_split public as ff_mpeg4video_split
stefang
parents:
4150
diff
changeset
|
301 int ff_mpeg4video_split(AVCodecContext *avctx, |
|
2769
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
302 const uint8_t *buf, int buf_size) |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
303 { |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
304 int i; |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
305 uint32_t state= -1; |
| 2967 | 306 |
|
2769
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
307 for(i=0; i<buf_size; i++){ |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
308 state= (state<<8) | buf[i]; |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
309 if(state == 0x1B3 || state == 0x1B6) |
| 2777 | 310 return i-3; |
|
2769
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
311 } |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
312 return 0; |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
313 } |
|
1394b45a7bf4
support changing in bitstream global headers into extradata style and back
michael
parents:
2637
diff
changeset
|
314 |
| 1613 | 315 /*************************/ |
| 316 | |
|
3455
cc4b4ea83e29
--enable/disable parsers. Warning: some combinations are broken.
mru
parents:
3432
diff
changeset
|
317 #ifdef CONFIG_MPEGAUDIO_PARSER |
| 1613 | 318 typedef struct MpegAudioParseContext { |
| 2979 | 319 uint8_t inbuf[MPA_MAX_CODED_FRAME_SIZE]; /* input buffer */ |
| 1613 | 320 uint8_t *inbuf_ptr; |
| 321 int frame_size; | |
| 322 int free_format_frame_size; | |
| 323 int free_format_next_header; | |
|
2470
06aafb585f69
require a few valid and equal mp3 headers for resync
michael
parents:
2389
diff
changeset
|
324 uint32_t header; |
|
06aafb585f69
require a few valid and equal mp3 headers for resync
michael
parents:
2389
diff
changeset
|
325 int header_count; |
| 1613 | 326 } MpegAudioParseContext; |
| 327 | |
| 328 #define MPA_HEADER_SIZE 4 | |
| 329 | |
| 330 /* header + layer + bitrate + freq + lsf/mpeg25 */ | |
|
2522
e25782262d7d
kill warnings patch by (M?ns Rullg?rd <mru inprovide com>)
michael
parents:
2486
diff
changeset
|
331 #undef SAME_HEADER_MASK /* mpegaudio.h defines different version */ |
| 1613 | 332 #define SAME_HEADER_MASK \ |
| 2480 | 333 (0xffe00000 | (3 << 17) | (3 << 10) | (3 << 19)) |
| 1613 | 334 |
| 335 static int mpegaudio_parse_init(AVCodecParserContext *s1) | |
| 336 { | |
| 337 MpegAudioParseContext *s = s1->priv_data; | |
| 338 s->inbuf_ptr = s->inbuf; | |
| 339 return 0; | |
| 340 } | |
| 341 | |
| 342 static int mpegaudio_parse(AVCodecParserContext *s1, | |
| 343 AVCodecContext *avctx, | |
| 2967 | 344 uint8_t **poutbuf, int *poutbuf_size, |
| 1613 | 345 const uint8_t *buf, int buf_size) |
| 346 { | |
| 347 MpegAudioParseContext *s = s1->priv_data; | |
|
2470
06aafb585f69
require a few valid and equal mp3 headers for resync
michael
parents:
2389
diff
changeset
|
348 int len, ret, sr; |
| 1613 | 349 uint32_t header; |
| 350 const uint8_t *buf_ptr; | |
| 351 | |
| 352 *poutbuf = NULL; | |
| 353 *poutbuf_size = 0; | |
| 354 buf_ptr = buf; | |
| 355 while (buf_size > 0) { | |
| 2979 | 356 len = s->inbuf_ptr - s->inbuf; |
| 357 if (s->frame_size == 0) { | |
| 1613 | 358 /* special case for next header for first frame in free |
| 359 format case (XXX: find a simpler method) */ | |
| 360 if (s->free_format_next_header != 0) { | |
| 361 s->inbuf[0] = s->free_format_next_header >> 24; | |
| 362 s->inbuf[1] = s->free_format_next_header >> 16; | |
| 363 s->inbuf[2] = s->free_format_next_header >> 8; | |
| 364 s->inbuf[3] = s->free_format_next_header; | |
| 365 s->inbuf_ptr = s->inbuf + 4; | |
| 366 s->free_format_next_header = 0; | |
| 367 goto got_header; | |
| 368 } | |
| 2979 | 369 /* no header seen : find one. We need at least MPA_HEADER_SIZE |
| 1613 | 370 bytes to parse it */ |
|
3639
949bc256f1e3
dont copy frame if the whole mp1/2/3 frame is available in one piece in the input
michael
parents:
3456
diff
changeset
|
371 len = FFMIN(MPA_HEADER_SIZE - len, buf_size); |
| 2979 | 372 if (len > 0) { |
| 373 memcpy(s->inbuf_ptr, buf_ptr, len); | |
| 374 buf_ptr += len; | |
| 375 buf_size -= len; | |
| 376 s->inbuf_ptr += len; | |
| 377 } | |
| 378 if ((s->inbuf_ptr - s->inbuf) >= MPA_HEADER_SIZE) { | |
| 1613 | 379 got_header: |
| 2979 | 380 header = (s->inbuf[0] << 24) | (s->inbuf[1] << 16) | |
| 381 (s->inbuf[2] << 8) | s->inbuf[3]; | |
| 1613 | 382 |
|
4104
04ff8026d9c0
dont set the sampling rate just because 1 mp3 packet header says so (fixes playback speed on some old mencoder generated avis which where then dumped to mp3)
michael
parents:
3989
diff
changeset
|
383 ret = mpa_decode_header(avctx, header, &sr); |
| 1613 | 384 if (ret < 0) { |
|
2470
06aafb585f69
require a few valid and equal mp3 headers for resync
michael
parents:
2389
diff
changeset
|
385 s->header_count= -2; |
| 2979 | 386 /* no sync found : move by one byte (inefficient, but simple!) */ |
| 387 memmove(s->inbuf, s->inbuf + 1, s->inbuf_ptr - s->inbuf - 1); | |
| 388 s->inbuf_ptr--; | |
| 4652 | 389 dprintf(avctx, "skip %x\n", header); |
| 1613 | 390 /* reset free format frame size to give a chance |
| 391 to get a new bitrate */ | |
| 392 s->free_format_frame_size = 0; | |
| 2979 | 393 } else { |
|
2470
06aafb585f69
require a few valid and equal mp3 headers for resync
michael
parents:
2389
diff
changeset
|
394 if((header&SAME_HEADER_MASK) != (s->header&SAME_HEADER_MASK) && s->header) |
|
06aafb585f69
require a few valid and equal mp3 headers for resync
michael
parents:
2389
diff
changeset
|
395 s->header_count= -3; |
|
06aafb585f69
require a few valid and equal mp3 headers for resync
michael
parents:
2389
diff
changeset
|
396 s->header= header; |
|
06aafb585f69
require a few valid and equal mp3 headers for resync
michael
parents:
2389
diff
changeset
|
397 s->header_count++; |
| 1613 | 398 s->frame_size = ret; |
| 2967 | 399 |
| 1613 | 400 #if 0 |
| 401 /* free format: prepare to compute frame size */ | |
| 2979 | 402 if (decode_header(s, header) == 1) { |
| 403 s->frame_size = -1; | |
| 1613 | 404 } |
| 405 #endif | |
| 2979 | 406 } |
|
4104
04ff8026d9c0
dont set the sampling rate just because 1 mp3 packet header says so (fixes playback speed on some old mencoder generated avis which where then dumped to mp3)
michael
parents:
3989
diff
changeset
|
407 if(s->header_count > 1) |
|
04ff8026d9c0
dont set the sampling rate just because 1 mp3 packet header says so (fixes playback speed on some old mencoder generated avis which where then dumped to mp3)
michael
parents:
3989
diff
changeset
|
408 avctx->sample_rate= sr; |
| 2979 | 409 } |
| 2967 | 410 } else |
| 1613 | 411 #if 0 |
| 412 if (s->frame_size == -1) { | |
| 413 /* free format : find next sync to compute frame size */ | |
| 2979 | 414 len = MPA_MAX_CODED_FRAME_SIZE - len; |
| 415 if (len > buf_size) | |
| 416 len = buf_size; | |
| 1613 | 417 if (len == 0) { |
| 2979 | 418 /* frame too long: resync */ |
| 1613 | 419 s->frame_size = 0; |
| 2979 | 420 memmove(s->inbuf, s->inbuf + 1, s->inbuf_ptr - s->inbuf - 1); |
| 421 s->inbuf_ptr--; | |
| 1613 | 422 } else { |
| 423 uint8_t *p, *pend; | |
| 424 uint32_t header1; | |
| 425 int padding; | |
| 426 | |
| 427 memcpy(s->inbuf_ptr, buf_ptr, len); | |
| 428 /* check for header */ | |
| 429 p = s->inbuf_ptr - 3; | |
| 430 pend = s->inbuf_ptr + len - 4; | |
| 431 while (p <= pend) { | |
| 432 header = (p[0] << 24) | (p[1] << 16) | | |
| 433 (p[2] << 8) | p[3]; | |
| 434 header1 = (s->inbuf[0] << 24) | (s->inbuf[1] << 16) | | |
| 435 (s->inbuf[2] << 8) | s->inbuf[3]; | |
| 436 /* check with high probability that we have a | |
| 437 valid header */ | |
| 438 if ((header & SAME_HEADER_MASK) == | |
| 439 (header1 & SAME_HEADER_MASK)) { | |
| 440 /* header found: update pointers */ | |
| 441 len = (p + 4) - s->inbuf_ptr; | |
| 442 buf_ptr += len; | |
| 443 buf_size -= len; | |
| 444 s->inbuf_ptr = p; | |
| 445 /* compute frame size */ | |
| 446 s->free_format_next_header = header; | |
| 447 s->free_format_frame_size = s->inbuf_ptr - s->inbuf; | |
| 448 padding = (header1 >> 9) & 1; | |
| 449 if (s->layer == 1) | |
| 450 s->free_format_frame_size -= padding * 4; | |
| 451 else | |
| 452 s->free_format_frame_size -= padding; | |
| 4652 | 453 dprintf(avctx, "free frame size=%d padding=%d\n", |
| 1613 | 454 s->free_format_frame_size, padding); |
| 455 decode_header(s, header1); | |
| 456 goto next_data; | |
| 457 } | |
| 458 p++; | |
| 459 } | |
| 460 /* not found: simply increase pointers */ | |
| 461 buf_ptr += len; | |
| 462 s->inbuf_ptr += len; | |
| 463 buf_size -= len; | |
| 464 } | |
| 2979 | 465 } else |
| 1613 | 466 #endif |
| 467 if (len < s->frame_size) { | |
| 468 if (s->frame_size > MPA_MAX_CODED_FRAME_SIZE) | |
| 469 s->frame_size = MPA_MAX_CODED_FRAME_SIZE; | |
|
3639
949bc256f1e3
dont copy frame if the whole mp1/2/3 frame is available in one piece in the input
michael
parents:
3456
diff
changeset
|
470 len = FFMIN(s->frame_size - len, buf_size); |
| 2979 | 471 memcpy(s->inbuf_ptr, buf_ptr, len); |
| 472 buf_ptr += len; | |
| 473 s->inbuf_ptr += len; | |
| 474 buf_size -= len; | |
| 475 } | |
|
3639
949bc256f1e3
dont copy frame if the whole mp1/2/3 frame is available in one piece in the input
michael
parents:
3456
diff
changeset
|
476 |
|
949bc256f1e3
dont copy frame if the whole mp1/2/3 frame is available in one piece in the input
michael
parents:
3456
diff
changeset
|
477 if(s->frame_size > 0 && buf_ptr - buf == s->inbuf_ptr - s->inbuf |
|
949bc256f1e3
dont copy frame if the whole mp1/2/3 frame is available in one piece in the input
michael
parents:
3456
diff
changeset
|
478 && buf_size + buf_ptr - buf >= s->frame_size){ |
|
949bc256f1e3
dont copy frame if the whole mp1/2/3 frame is available in one piece in the input
michael
parents:
3456
diff
changeset
|
479 if(s->header_count > 0){ |
|
949bc256f1e3
dont copy frame if the whole mp1/2/3 frame is available in one piece in the input
michael
parents:
3456
diff
changeset
|
480 *poutbuf = buf; |
|
949bc256f1e3
dont copy frame if the whole mp1/2/3 frame is available in one piece in the input
michael
parents:
3456
diff
changeset
|
481 *poutbuf_size = s->frame_size; |
|
949bc256f1e3
dont copy frame if the whole mp1/2/3 frame is available in one piece in the input
michael
parents:
3456
diff
changeset
|
482 } |
|
949bc256f1e3
dont copy frame if the whole mp1/2/3 frame is available in one piece in the input
michael
parents:
3456
diff
changeset
|
483 buf_ptr = buf + s->frame_size; |
|
949bc256f1e3
dont copy frame if the whole mp1/2/3 frame is available in one piece in the input
michael
parents:
3456
diff
changeset
|
484 s->inbuf_ptr = s->inbuf; |
|
949bc256f1e3
dont copy frame if the whole mp1/2/3 frame is available in one piece in the input
michael
parents:
3456
diff
changeset
|
485 s->frame_size = 0; |
|
949bc256f1e3
dont copy frame if the whole mp1/2/3 frame is available in one piece in the input
michael
parents:
3456
diff
changeset
|
486 break; |
|
949bc256f1e3
dont copy frame if the whole mp1/2/3 frame is available in one piece in the input
michael
parents:
3456
diff
changeset
|
487 } |
|
949bc256f1e3
dont copy frame if the whole mp1/2/3 frame is available in one piece in the input
michael
parents:
3456
diff
changeset
|
488 |
| 1613 | 489 // next_data: |
| 2967 | 490 if (s->frame_size > 0 && |
| 1613 | 491 (s->inbuf_ptr - s->inbuf) >= s->frame_size) { |
|
2470
06aafb585f69
require a few valid and equal mp3 headers for resync
michael
parents:
2389
diff
changeset
|
492 if(s->header_count > 0){ |
|
06aafb585f69
require a few valid and equal mp3 headers for resync
michael
parents:
2389
diff
changeset
|
493 *poutbuf = s->inbuf; |
|
06aafb585f69
require a few valid and equal mp3 headers for resync
michael
parents:
2389
diff
changeset
|
494 *poutbuf_size = s->inbuf_ptr - s->inbuf; |
|
06aafb585f69
require a few valid and equal mp3 headers for resync
michael
parents:
2389
diff
changeset
|
495 } |
| 2979 | 496 s->inbuf_ptr = s->inbuf; |
| 497 s->frame_size = 0; | |
| 498 break; | |
| 499 } | |
| 1613 | 500 } |
| 501 return buf_ptr - buf; | |
| 502 } | |
|
3455
cc4b4ea83e29
--enable/disable parsers. Warning: some combinations are broken.
mru
parents:
3432
diff
changeset
|
503 #endif /* CONFIG_MPEGAUDIO_PARSER */ |
| 1613 | 504 |
|
3455
cc4b4ea83e29
--enable/disable parsers. Warning: some combinations are broken.
mru
parents:
3432
diff
changeset
|
505 #if defined(CONFIG_AC3_PARSER) || defined(CONFIG_AAC_PARSER) |
| 3098 | 506 /* also used for ADTS AAC */ |
| 1613 | 507 typedef struct AC3ParseContext { |
| 508 uint8_t *inbuf_ptr; | |
| 509 int frame_size; | |
| 3098 | 510 int header_size; |
| 511 int (*sync)(const uint8_t *buf, int *channels, int *sample_rate, | |
| 512 int *bit_rate, int *samples); | |
|
3344
f9d739057d6c
The AAC frame header uses 13 bits for the frame size, so the buffer should
mru
parents:
3104
diff
changeset
|
513 uint8_t inbuf[8192]; /* input buffer */ |
| 1613 | 514 } AC3ParseContext; |
| 515 | |
| 516 #define AC3_HEADER_SIZE 7 | |
| 3104 | 517 #define AAC_HEADER_SIZE 7 |
| 3059 | 518 |
|
3455
cc4b4ea83e29
--enable/disable parsers. Warning: some combinations are broken.
mru
parents:
3432
diff
changeset
|
519 #ifdef CONFIG_AC3_PARSER |
|
4397
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
520 |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
521 static const uint8_t eac3_blocks[4] = { |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
522 1, 2, 3, 6 |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
523 }; |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
524 |
|
3455
cc4b4ea83e29
--enable/disable parsers. Warning: some combinations are broken.
mru
parents:
3432
diff
changeset
|
525 #endif /* CONFIG_AC3_PARSER */ |
| 3059 | 526 |
|
3455
cc4b4ea83e29
--enable/disable parsers. Warning: some combinations are broken.
mru
parents:
3432
diff
changeset
|
527 #ifdef CONFIG_AAC_PARSER |
| 3456 | 528 static const int aac_sample_rates[16] = { |
| 3098 | 529 96000, 88200, 64000, 48000, 44100, 32000, |
| 530 24000, 22050, 16000, 12000, 11025, 8000, 7350 | |
| 531 }; | |
| 532 | |
| 3456 | 533 static const int aac_channels[8] = { |
| 3098 | 534 0, 1, 2, 3, 4, 5, 6, 8 |
| 535 }; | |
|
3455
cc4b4ea83e29
--enable/disable parsers. Warning: some combinations are broken.
mru
parents:
3432
diff
changeset
|
536 #endif |
| 3098 | 537 |
|
3455
cc4b4ea83e29
--enable/disable parsers. Warning: some combinations are broken.
mru
parents:
3432
diff
changeset
|
538 #ifdef CONFIG_AC3_PARSER |
|
4855
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
539 int ff_ac3_parse_header(const uint8_t buf[7], AC3HeaderInfo *hdr) |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
540 { |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
541 GetBitContext gbc; |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
542 |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
543 memset(hdr, 0, sizeof(*hdr)); |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
544 |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
545 init_get_bits(&gbc, buf, 54); |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
546 |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
547 hdr->sync_word = get_bits(&gbc, 16); |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
548 if(hdr->sync_word != 0x0B77) |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
549 return -1; |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
550 |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
551 /* read ahead to bsid to make sure this is AC-3, not E-AC-3 */ |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
552 hdr->bsid = show_bits_long(&gbc, 29) & 0x1F; |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
553 if(hdr->bsid > 10) |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
554 return -2; |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
555 |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
556 hdr->crc1 = get_bits(&gbc, 16); |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
557 hdr->fscod = get_bits(&gbc, 2); |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
558 if(hdr->fscod == 3) |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
559 return -3; |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
560 |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
561 hdr->frmsizecod = get_bits(&gbc, 6); |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
562 if(hdr->frmsizecod > 37) |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
563 return -4; |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
564 |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
565 skip_bits(&gbc, 5); // skip bsid, already got it |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
566 |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
567 hdr->bsmod = get_bits(&gbc, 3); |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
568 hdr->acmod = get_bits(&gbc, 3); |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
569 if((hdr->acmod & 1) && hdr->acmod != 1) { |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
570 hdr->cmixlev = get_bits(&gbc, 2); |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
571 } |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
572 if(hdr->acmod & 4) { |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
573 hdr->surmixlev = get_bits(&gbc, 2); |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
574 } |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
575 if(hdr->acmod == 2) { |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
576 hdr->dsurmod = get_bits(&gbc, 2); |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
577 } |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
578 hdr->lfeon = get_bits1(&gbc); |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
579 |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
580 hdr->halfratecod = FFMAX(hdr->bsid, 8) - 8; |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
581 hdr->sample_rate = ff_ac3_freqs[hdr->fscod] >> hdr->halfratecod; |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
582 hdr->bit_rate = (ff_ac3_bitratetab[hdr->frmsizecod>>1] * 1000) >> hdr->halfratecod; |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
583 hdr->channels = ff_ac3_channels[hdr->acmod] + hdr->lfeon; |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
584 hdr->frame_size = ff_ac3_frame_sizes[hdr->frmsizecod][hdr->fscod] * 2; |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
585 |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
586 return 0; |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
587 } |
|
40f3a7f2b1fd
Move AC3 header parsing code together with the rest of the AC3 parsing code.
diego
parents:
4846
diff
changeset
|
588 |
| 3059 | 589 static int ac3_sync(const uint8_t *buf, int *channels, int *sample_rate, |
| 3098 | 590 int *bit_rate, int *samples) |
| 3059 | 591 { |
| 4648 | 592 int err; |
| 593 unsigned int fscod, acmod, bsid, lfeon; | |
|
4397
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
594 unsigned int strmtyp, substreamid, frmsiz, fscod2, numblkscod; |
| 3059 | 595 GetBitContext bits; |
| 4648 | 596 AC3HeaderInfo hdr; |
| 3059 | 597 |
| 4648 | 598 err = ff_ac3_parse_header(buf, &hdr); |
| 3059 | 599 |
| 4648 | 600 if(err < 0 && err != -2) |
|
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3059
diff
changeset
|
601 return 0; |
| 3059 | 602 |
| 4648 | 603 bsid = hdr.bsid; |
| 4501 | 604 if(bsid <= 10) { /* Normal AC-3 */ |
| 4648 | 605 *sample_rate = hdr.sample_rate; |
| 606 *bit_rate = hdr.bit_rate; | |
| 607 *channels = hdr.channels; | |
| 608 *samples = AC3_FRAME_SIZE; | |
| 609 return hdr.frame_size; | |
| 4501 | 610 } else if (bsid > 10 && bsid <= 16) { /* Enhanced AC-3 */ |
| 4648 | 611 init_get_bits(&bits, &buf[2], (AC3_HEADER_SIZE-2) * 8); |
|
4397
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
612 strmtyp = get_bits(&bits, 2); |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
613 substreamid = get_bits(&bits, 3); |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
614 |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
615 if (strmtyp != 0 || substreamid != 0) |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
616 return 0; /* Currently don't support additional streams */ |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
617 |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
618 frmsiz = get_bits(&bits, 11) + 1; |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
619 fscod = get_bits(&bits, 2); |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
620 if (fscod == 3) { |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
621 fscod2 = get_bits(&bits, 2); |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
622 numblkscod = 3; |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
623 |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
624 if(fscod2 == 3) |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
625 return 0; |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
626 |
|
4645
056127e5df89
remove redundancy in AC-3 parser by using common tables from ac3tab.h
jbr
parents:
4501
diff
changeset
|
627 *sample_rate = ff_ac3_freqs[fscod2] / 2; |
|
4397
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
628 } else { |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
629 numblkscod = get_bits(&bits, 2); |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
630 |
|
4645
056127e5df89
remove redundancy in AC-3 parser by using common tables from ac3tab.h
jbr
parents:
4501
diff
changeset
|
631 *sample_rate = ff_ac3_freqs[fscod]; |
|
4397
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
632 } |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
633 |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
634 acmod = get_bits(&bits, 3); |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
635 lfeon = get_bits1(&bits); |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
636 |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
637 *samples = eac3_blocks[numblkscod] * 256; |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
638 *bit_rate = frmsiz * (*sample_rate) * 16 / (*samples); |
|
4645
056127e5df89
remove redundancy in AC-3 parser by using common tables from ac3tab.h
jbr
parents:
4501
diff
changeset
|
639 *channels = ff_ac3_channels[acmod] + lfeon; |
|
4397
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
640 |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
641 return frmsiz * 2; |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
642 } |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
643 |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
644 /* Unsupported bitstream version */ |
|
acb9faabab8d
Allows the AC3 parser to read the frame size and codec parameters from E-AC3 streams,
gpoirier
parents:
4310
diff
changeset
|
645 return 0; |
| 3059 | 646 } |
|
3455
cc4b4ea83e29
--enable/disable parsers. Warning: some combinations are broken.
mru
parents:
3432
diff
changeset
|
647 #endif /* CONFIG_AC3_PARSER */ |
| 1613 | 648 |
|
3455
cc4b4ea83e29
--enable/disable parsers. Warning: some combinations are broken.
mru
parents:
3432
diff
changeset
|
649 #ifdef CONFIG_AAC_PARSER |
| 3098 | 650 static int aac_sync(const uint8_t *buf, int *channels, int *sample_rate, |
| 651 int *bit_rate, int *samples) | |
| 652 { | |
| 653 GetBitContext bits; | |
| 654 int size, rdb, ch, sr; | |
| 655 | |
| 656 init_get_bits(&bits, buf, AAC_HEADER_SIZE * 8); | |
| 657 | |
| 658 if(get_bits(&bits, 12) != 0xfff) | |
| 659 return 0; | |
| 660 | |
| 3104 | 661 skip_bits1(&bits); /* id */ |
| 662 skip_bits(&bits, 2); /* layer */ | |
| 663 skip_bits1(&bits); /* protection_absent */ | |
| 664 skip_bits(&bits, 2); /* profile_objecttype */ | |
| 665 sr = get_bits(&bits, 4); /* sample_frequency_index */ | |
| 3098 | 666 if(!aac_sample_rates[sr]) |
| 667 return 0; | |
| 3104 | 668 skip_bits1(&bits); /* private_bit */ |
| 669 ch = get_bits(&bits, 3); /* channel_configuration */ | |
| 3098 | 670 if(!aac_channels[ch]) |
| 671 return 0; | |
| 3104 | 672 skip_bits1(&bits); /* original/copy */ |
| 673 skip_bits1(&bits); /* home */ | |
| 3098 | 674 |
| 675 /* adts_variable_header */ | |
| 3104 | 676 skip_bits1(&bits); /* copyright_identification_bit */ |
| 677 skip_bits1(&bits); /* copyright_identification_start */ | |
| 678 size = get_bits(&bits, 13); /* aac_frame_length */ | |
| 679 skip_bits(&bits, 11); /* adts_buffer_fullness */ | |
| 680 rdb = get_bits(&bits, 2); /* number_of_raw_data_blocks_in_frame */ | |
| 3098 | 681 |
| 682 *channels = aac_channels[ch]; | |
| 683 *sample_rate = aac_sample_rates[sr]; | |
| 684 *samples = (rdb + 1) * 1024; | |
| 685 *bit_rate = size * 8 * *sample_rate / *samples; | |
| 686 | |
| 687 return size; | |
| 688 } | |
|
3455
cc4b4ea83e29
--enable/disable parsers. Warning: some combinations are broken.
mru
parents:
3432
diff
changeset
|
689 #endif /* CONFIG_AAC_PARSER */ |
| 3098 | 690 |
|
3455
cc4b4ea83e29
--enable/disable parsers. Warning: some combinations are broken.
mru
parents:
3432
diff
changeset
|
691 #ifdef CONFIG_AC3_PARSER |
| 1613 | 692 static int ac3_parse_init(AVCodecParserContext *s1) |
| 693 { | |
| 694 AC3ParseContext *s = s1->priv_data; | |
| 695 s->inbuf_ptr = s->inbuf; | |
| 3098 | 696 s->header_size = AC3_HEADER_SIZE; |
| 697 s->sync = ac3_sync; | |
| 1613 | 698 return 0; |
| 699 } | |
|
3455
cc4b4ea83e29
--enable/disable parsers. Warning: some combinations are broken.
mru
parents:
3432
diff
changeset
|
700 #endif |
| 1613 | 701 |
|
3455
cc4b4ea83e29
--enable/disable parsers. Warning: some combinations are broken.
mru
parents:
3432
diff
changeset
|
702 #ifdef CONFIG_AAC_PARSER |
| 3098 | 703 static int aac_parse_init(AVCodecParserContext *s1) |
| 704 { | |
| 705 AC3ParseContext *s = s1->priv_data; | |
| 706 s->inbuf_ptr = s->inbuf; | |
| 707 s->header_size = AAC_HEADER_SIZE; | |
| 708 s->sync = aac_sync; | |
| 709 return 0; | |
| 710 } | |
|
3455
cc4b4ea83e29
--enable/disable parsers. Warning: some combinations are broken.
mru
parents:
3432
diff
changeset
|
711 #endif |
| 3098 | 712 |
| 713 /* also used for ADTS AAC */ | |
| 1613 | 714 static int ac3_parse(AVCodecParserContext *s1, |
| 715 AVCodecContext *avctx, | |
| 2967 | 716 uint8_t **poutbuf, int *poutbuf_size, |
| 1613 | 717 const uint8_t *buf, int buf_size) |
| 718 { | |
| 719 AC3ParseContext *s = s1->priv_data; | |
| 720 const uint8_t *buf_ptr; | |
| 3098 | 721 int len, sample_rate, bit_rate, channels, samples; |
| 1613 | 722 |
| 723 *poutbuf = NULL; | |
| 724 *poutbuf_size = 0; | |
| 725 | |
| 726 buf_ptr = buf; | |
| 727 while (buf_size > 0) { | |
| 728 len = s->inbuf_ptr - s->inbuf; | |
| 729 if (s->frame_size == 0) { | |
| 3098 | 730 /* no header seen : find one. We need at least s->header_size |
| 731 bytes to parse it */ | |
| 732 len = FFMIN(s->header_size - len, buf_size); | |
| 3082 | 733 |
| 1613 | 734 memcpy(s->inbuf_ptr, buf_ptr, len); |
| 735 buf_ptr += len; | |
| 736 s->inbuf_ptr += len; | |
| 737 buf_size -= len; | |
| 3098 | 738 if ((s->inbuf_ptr - s->inbuf) == s->header_size) { |
| 739 len = s->sync(s->inbuf, &channels, &sample_rate, &bit_rate, | |
| 740 &samples); | |
| 1613 | 741 if (len == 0) { |
| 742 /* no sync found : move by one byte (inefficient, but simple!) */ | |
| 3098 | 743 memmove(s->inbuf, s->inbuf + 1, s->header_size - 1); |
| 1613 | 744 s->inbuf_ptr--; |
| 745 } else { | |
| 2979 | 746 s->frame_size = len; |
| 1613 | 747 /* update codec info */ |
| 748 avctx->sample_rate = sample_rate; | |
| 1987 | 749 /* set channels,except if the user explicitly requests 1 or 2 channels, XXX/FIXME this is a bit ugly */ |
| 3098 | 750 if(avctx->codec_id == CODEC_ID_AC3){ |
| 751 if(avctx->channels!=1 && avctx->channels!=2){ | |
| 752 avctx->channels = channels; | |
| 753 } | |
| 754 } else { | |
|
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3059
diff
changeset
|
755 avctx->channels = channels; |
| 1987 | 756 } |
| 2979 | 757 avctx->bit_rate = bit_rate; |
| 3098 | 758 avctx->frame_size = samples; |
| 1613 | 759 } |
| 760 } | |
| 3082 | 761 } else { |
| 762 len = FFMIN(s->frame_size - len, buf_size); | |
| 1613 | 763 |
| 764 memcpy(s->inbuf_ptr, buf_ptr, len); | |
| 765 buf_ptr += len; | |
| 766 s->inbuf_ptr += len; | |
| 767 buf_size -= len; | |
| 3082 | 768 |
| 769 if(s->inbuf_ptr - s->inbuf == s->frame_size){ | |
| 770 *poutbuf = s->inbuf; | |
| 771 *poutbuf_size = s->frame_size; | |
| 772 s->inbuf_ptr = s->inbuf; | |
| 773 s->frame_size = 0; | |
| 774 break; | |
| 775 } | |
| 1613 | 776 } |
| 777 } | |
| 778 return buf_ptr - buf; | |
| 779 } | |
|
3455
cc4b4ea83e29
--enable/disable parsers. Warning: some combinations are broken.
mru
parents:
3432
diff
changeset
|
780 #endif /* CONFIG_AC3_PARSER || CONFIG_AAC_PARSER */ |
| 1613 | 781 |
|
3455
cc4b4ea83e29
--enable/disable parsers. Warning: some combinations are broken.
mru
parents:
3432
diff
changeset
|
782 #ifdef CONFIG_MPEGAUDIO_PARSER |
| 1613 | 783 AVCodecParser mpegaudio_parser = { |
| 784 { CODEC_ID_MP2, CODEC_ID_MP3 }, | |
| 785 sizeof(MpegAudioParseContext), | |
| 786 mpegaudio_parse_init, | |
| 787 mpegaudio_parse, | |
| 788 NULL, | |
| 789 }; | |
|
3455
cc4b4ea83e29
--enable/disable parsers. Warning: some combinations are broken.
mru
parents:
3432
diff
changeset
|
790 #endif |
|
cc4b4ea83e29
--enable/disable parsers. Warning: some combinations are broken.
mru
parents:
3432
diff
changeset
|
791 #ifdef CONFIG_AC3_PARSER |
| 1613 | 792 AVCodecParser ac3_parser = { |
| 793 { CODEC_ID_AC3 }, | |
| 794 sizeof(AC3ParseContext), | |
| 795 ac3_parse_init, | |
| 796 ac3_parse, | |
| 797 NULL, | |
| 798 }; | |
|
3455
cc4b4ea83e29
--enable/disable parsers. Warning: some combinations are broken.
mru
parents:
3432
diff
changeset
|
799 #endif |
|
cc4b4ea83e29
--enable/disable parsers. Warning: some combinations are broken.
mru
parents:
3432
diff
changeset
|
800 #ifdef CONFIG_AAC_PARSER |
| 3098 | 801 AVCodecParser aac_parser = { |
| 802 { CODEC_ID_AAC }, | |
| 803 sizeof(AC3ParseContext), | |
| 804 aac_parse_init, | |
| 805 ac3_parse, | |
| 806 NULL, | |
| 807 }; | |
|
3455
cc4b4ea83e29
--enable/disable parsers. Warning: some combinations are broken.
mru
parents:
3432
diff
changeset
|
808 #endif |
