annotate parser.c @ 2497:69adfbbdcdeb libavcodec

- samples from mplayer ftp in the "adv" profile seem to have profile=2, which isn't the advanced one; and indeed, using adv. profile parser fails. Using normal parser works, and that's what is done - attempt at taking care of stride for NORM2 bitplane decoding - duplication of much code from msmpeg4.c; this code isn't yet used, but goes down as far as the block layer (mainly Transform Type stuff, the remains are wild editing without checking). Unusable yet, and lacks the AC decoding (but a step further in bitstream parsing) patch by anonymous
author michael
date Fri, 04 Feb 2005 02:20:38 +0000
parents f2a9559db6ac
children e25782262d7d
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
1 /*
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
2 * Audio and Video frame extraction
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
3 * Copyright (c) 2003 Fabrice Bellard.
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
4 * Copyright (c) 2003 Michael Niedermayer.
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
5 *
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
6 * This library is free software; you can redistribute it and/or
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
7 * modify it under the terms of the GNU Lesser General Public
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
8 * License as published by the Free Software Foundation; either
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
9 * version 2 of the License, or (at your option) any later version.
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
10 *
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
11 * This library is distributed in the hope that it will be useful,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
14 * Lesser General Public License for more details.
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
15 *
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
16 * You should have received a copy of the GNU Lesser General Public
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
17 * License along with this library; if not, write to the Free Software
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
19 */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
20 #include "avcodec.h"
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
21 #include "mpegvideo.h"
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
22 #include "mpegaudio.h"
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
23
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
24 AVCodecParser *av_first_parser = NULL;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
25
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
26 void av_register_codec_parser(AVCodecParser *parser)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
27 {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
28 parser->next = av_first_parser;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
29 av_first_parser = parser;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
30 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
31
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
32 AVCodecParserContext *av_parser_init(int codec_id)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
33 {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
34 AVCodecParserContext *s;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
35 AVCodecParser *parser;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
36 int ret;
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
37
f2a9559db6ac 10l (array gets padded with 0 which is CODEC_ID_NONE -> parsers claim to support CODEC_ID_NONE)
michael
parents: 2480
diff changeset
38 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
39 return NULL;
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
40
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
41 for(parser = av_first_parser; parser != NULL; parser = parser->next) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
42 if (parser->codec_ids[0] == codec_id ||
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
43 parser->codec_ids[1] == codec_id ||
2348
d02fb928ca44 pnm parser
michael
parents: 2270
diff changeset
44 parser->codec_ids[2] == codec_id ||
d02fb928ca44 pnm parser
michael
parents: 2270
diff changeset
45 parser->codec_ids[3] == codec_id ||
d02fb928ca44 pnm parser
michael
parents: 2270
diff changeset
46 parser->codec_ids[4] == codec_id)
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
47 goto found;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
48 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
49 return NULL;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
50 found:
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
51 s = av_mallocz(sizeof(AVCodecParserContext));
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
52 if (!s)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
53 return NULL;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
54 s->parser = parser;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
55 s->priv_data = av_mallocz(parser->priv_data_size);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
56 if (!s->priv_data) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
57 av_free(s);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
58 return NULL;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
59 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
60 if (parser->parser_init) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
61 ret = parser->parser_init(s);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
62 if (ret != 0) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
63 av_free(s->priv_data);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
64 av_free(s);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
65 return NULL;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
66 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
67 }
2030
f796043935f3 mpeg audio timestamp fix
michael
parents: 2024
diff changeset
68 s->fetch_timestamp=1;
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
69 return s;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
70 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
71
1694
13169235c306 added End Of File handling to return last picture for MPEG1/2/4
bellard
parents: 1681
diff changeset
72 /* NOTE: buf_size == 0 is used to signal EOF so that the last frame
13169235c306 added End Of File handling to return last picture for MPEG1/2/4
bellard
parents: 1681
diff changeset
73 can be returned if necessary */
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
74 int av_parser_parse(AVCodecParserContext *s,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
75 AVCodecContext *avctx,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
76 uint8_t **poutbuf, int *poutbuf_size,
1696
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
77 const uint8_t *buf, int buf_size,
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
78 int64_t pts, int64_t dts)
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
79 {
1696
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
80 int index, i, k;
1694
13169235c306 added End Of File handling to return last picture for MPEG1/2/4
bellard
parents: 1681
diff changeset
81 uint8_t dummy_buf[FF_INPUT_BUFFER_PADDING_SIZE];
13169235c306 added End Of File handling to return last picture for MPEG1/2/4
bellard
parents: 1681
diff changeset
82
13169235c306 added End Of File handling to return last picture for MPEG1/2/4
bellard
parents: 1681
diff changeset
83 if (buf_size == 0) {
13169235c306 added End Of File handling to return last picture for MPEG1/2/4
bellard
parents: 1681
diff changeset
84 /* 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
85 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
86 buf = dummy_buf;
1696
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
87 } else {
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
88 /* add a new packet descriptor */
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
89 k = (s->cur_frame_start_index + 1) & (AV_PARSER_PTS_NB - 1);
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
90 s->cur_frame_start_index = k;
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
91 s->cur_frame_offset[k] = s->cur_offset;
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
92 s->cur_frame_pts[k] = pts;
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
93 s->cur_frame_dts[k] = dts;
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
94
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
95 /* fill first PTS/DTS */
2030
f796043935f3 mpeg audio timestamp fix
michael
parents: 2024
diff changeset
96 if (s->fetch_timestamp){
f796043935f3 mpeg audio timestamp fix
michael
parents: 2024
diff changeset
97 s->fetch_timestamp=0;
1696
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
98 s->last_pts = pts;
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
99 s->last_dts = dts;
2107
bec4623c2201 timestamp duplication bugfix
michael
parents: 2030
diff changeset
100 s->cur_frame_pts[k] =
bec4623c2201 timestamp duplication bugfix
michael
parents: 2030
diff changeset
101 s->cur_frame_dts[k] = AV_NOPTS_VALUE;
1696
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
102 }
1694
13169235c306 added End Of File handling to return last picture for MPEG1/2/4
bellard
parents: 1681
diff changeset
103 }
13169235c306 added End Of File handling to return last picture for MPEG1/2/4
bellard
parents: 1681
diff changeset
104
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
105 /* WARNING: the returned index can be negative */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
106 index = s->parser->parser_parse(s, avctx, poutbuf, poutbuf_size, buf, buf_size);
2107
bec4623c2201 timestamp duplication bugfix
michael
parents: 2030
diff changeset
107 //av_log(NULL, AV_LOG_DEBUG, "parser: in:%lld, %lld, out:%lld, %lld, in:%d out:%d id:%d\n", pts, dts, s->last_pts, s->last_dts, buf_size, *poutbuf_size, avctx->codec_id);
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
108 /* update the file pointer */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
109 if (*poutbuf_size) {
1696
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
110 /* fill the data for the current frame */
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
111 s->frame_offset = s->last_frame_offset;
1696
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
112 s->pts = s->last_pts;
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
113 s->dts = s->last_dts;
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
114
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
115 /* offset of the next frame */
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
116 s->last_frame_offset = s->cur_offset + index;
1696
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
117 /* find the packet in which the new frame starts. It
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
118 is tricky because of MPEG video start codes
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
119 which can begin in one packet and finish in
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
120 another packet. In the worst case, an MPEG
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
121 video start code could be in 4 different
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
122 packets. */
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
123 k = s->cur_frame_start_index;
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
124 for(i = 0; i < AV_PARSER_PTS_NB; i++) {
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
125 if (s->last_frame_offset >= s->cur_frame_offset[k])
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
126 break;
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
127 k = (k - 1) & (AV_PARSER_PTS_NB - 1);
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
128 }
2030
f796043935f3 mpeg audio timestamp fix
michael
parents: 2024
diff changeset
129
1696
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
130 s->last_pts = s->cur_frame_pts[k];
f5af91b8be17 pts and dts support in parser API
bellard
parents: 1694
diff changeset
131 s->last_dts = s->cur_frame_dts[k];
2030
f796043935f3 mpeg audio timestamp fix
michael
parents: 2024
diff changeset
132
f796043935f3 mpeg audio timestamp fix
michael
parents: 2024
diff changeset
133 /* some parsers tell us the packet size even before seeing the first byte of the next packet,
f796043935f3 mpeg audio timestamp fix
michael
parents: 2024
diff changeset
134 so the next pts/dts is in the next chunk */
f796043935f3 mpeg audio timestamp fix
michael
parents: 2024
diff changeset
135 if(index == buf_size){
f796043935f3 mpeg audio timestamp fix
michael
parents: 2024
diff changeset
136 s->fetch_timestamp=1;
f796043935f3 mpeg audio timestamp fix
michael
parents: 2024
diff changeset
137 }
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
138 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
139 if (index < 0)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
140 index = 0;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
141 s->cur_offset += index;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
142 return index;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
143 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
144
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
145 void av_parser_close(AVCodecParserContext *s)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
146 {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
147 if (s->parser->parser_close)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
148 s->parser->parser_close(s);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
149 av_free(s->priv_data);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
150 av_free(s);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
151 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
152
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
153 /*****************************************************/
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
154
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
155 //#define END_NOT_FOUND (-100)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
156
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
157 #define PICTURE_START_CODE 0x00000100
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
158 #define SEQ_START_CODE 0x000001b3
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
159 #define EXT_START_CODE 0x000001b5
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
160 #define SLICE_MIN_START_CODE 0x00000101
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
161 #define SLICE_MAX_START_CODE 0x000001af
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
162
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
163 typedef struct ParseContext1{
1988
b5753525f9a8 remove duplicated find_frame_end() code
michael
parents: 1987
diff changeset
164 ParseContext pc;
b5753525f9a8 remove duplicated find_frame_end() code
michael
parents: 1987
diff changeset
165 /* XXX/FIXME PC1 vs. PC */
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
166 /* MPEG2 specific */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
167 int frame_rate;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
168 int progressive_sequence;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
169 int width, height;
1614
6c82ef97d3e6 also parse extradata for MPEG4
bellard
parents: 1613
diff changeset
170
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
171 /* XXX: suppress that, needed by MPEG4 */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
172 MpegEncContext *enc;
1614
6c82ef97d3e6 also parse extradata for MPEG4
bellard
parents: 1613
diff changeset
173 int first_picture;
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
174 } ParseContext1;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
175
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
176 /**
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
177 * combines the (truncated) bitstream to a complete frame
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
178 * @returns -1 if no complete frame could be created
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
179 */
1988
b5753525f9a8 remove duplicated find_frame_end() code
michael
parents: 1987
diff changeset
180 int ff_combine_frame(ParseContext *pc, int next, uint8_t **buf, int *buf_size)
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
181 {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
182 #if 0
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
183 if(pc->overread){
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
184 printf("overread %d, state:%X next:%d index:%d o_index:%d\n", pc->overread, pc->state, next, pc->index, pc->overread_index);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
185 printf("%X %X %X %X\n", (*buf)[0], (*buf)[1],(*buf)[2],(*buf)[3]);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
186 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
187 #endif
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
188
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
189 /* copy overreaded bytes from last frame into buffer */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
190 for(; pc->overread>0; pc->overread--){
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
191 pc->buffer[pc->index++]= pc->buffer[pc->overread_index++];
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
192 }
2386
8d1983254e28 flush remaining data from parser at EOF
michael
parents: 2348
diff changeset
193
8d1983254e28 flush remaining data from parser at EOF
michael
parents: 2348
diff changeset
194 /* flush remaining if EOF */
8d1983254e28 flush remaining data from parser at EOF
michael
parents: 2348
diff changeset
195 if(!*buf_size && next == END_NOT_FOUND){
8d1983254e28 flush remaining data from parser at EOF
michael
parents: 2348
diff changeset
196 next= 0;
8d1983254e28 flush remaining data from parser at EOF
michael
parents: 2348
diff changeset
197 }
8d1983254e28 flush remaining data from parser at EOF
michael
parents: 2348
diff changeset
198
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
199 pc->last_index= pc->index;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
200
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
201 /* copy into buffer end return */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
202 if(next == END_NOT_FOUND){
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
203 pc->buffer= av_fast_realloc(pc->buffer, &pc->buffer_size, (*buf_size) + pc->index + FF_INPUT_BUFFER_PADDING_SIZE);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
204
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
205 memcpy(&pc->buffer[pc->index], *buf, *buf_size);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
206 pc->index += *buf_size;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
207 return -1;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
208 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
209
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
210 *buf_size=
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
211 pc->overread_index= pc->index + next;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
212
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
213 /* append to buffer */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
214 if(pc->index){
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
215 pc->buffer= av_fast_realloc(pc->buffer, &pc->buffer_size, next + pc->index + FF_INPUT_BUFFER_PADDING_SIZE);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
216
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
217 memcpy(&pc->buffer[pc->index], *buf, next + FF_INPUT_BUFFER_PADDING_SIZE );
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
218 pc->index = 0;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
219 *buf= pc->buffer;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
220 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
221
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
222 /* store overread bytes */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
223 for(;next < 0; next++){
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
224 pc->state = (pc->state<<8) | pc->buffer[pc->last_index + next];
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
225 pc->overread++;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
226 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
227
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
228 #if 0
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
229 if(pc->overread){
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
230 printf("overread %d, state:%X next:%d index:%d o_index:%d\n", pc->overread, pc->state, next, pc->index, pc->overread_index);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
231 printf("%X %X %X %X\n", (*buf)[0], (*buf)[1],(*buf)[2],(*buf)[3]);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
232 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
233 #endif
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
234
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
235 return 0;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
236 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
237
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
238 static int find_start_code(const uint8_t **pbuf_ptr, const uint8_t *buf_end)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
239 {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
240 const uint8_t *buf_ptr;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
241 unsigned int state=0xFFFFFFFF, v;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
242 int val;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
243
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
244 buf_ptr = *pbuf_ptr;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
245 while (buf_ptr < buf_end) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
246 v = *buf_ptr++;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
247 if (state == 0x000001) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
248 state = ((state << 8) | v) & 0xffffff;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
249 val = state;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
250 goto found;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
251 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
252 state = ((state << 8) | v) & 0xffffff;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
253 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
254 val = -1;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
255 found:
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
256 *pbuf_ptr = buf_ptr;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
257 return val;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
258 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
259
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
260 /* XXX: merge with libavcodec ? */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
261 #define MPEG1_FRAME_RATE_BASE 1001
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
262
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
263 static const int frame_rate_tab[16] = {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
264 0,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
265 24000,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
266 24024,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
267 25025,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
268 30000,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
269 30030,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
270 50050,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
271 60000,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
272 60060,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
273 // Xing's 15fps: (9)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
274 15015,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
275 // libmpeg3's "Unofficial economy rates": (10-13)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
276 5005,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
277 10010,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
278 12012,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
279 15015,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
280 // random, just to avoid segfault !never encode these
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
281 25025,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
282 25025,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
283 };
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
284
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
285 static void mpegvideo_extract_headers(AVCodecParserContext *s,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
286 AVCodecContext *avctx,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
287 const uint8_t *buf, int buf_size)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
288 {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
289 ParseContext1 *pc = s->priv_data;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
290 const uint8_t *buf_end;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
291 int32_t start_code;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
292 int frame_rate_index, ext_type, bytes_left;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
293 int frame_rate_ext_n, frame_rate_ext_d;
2119
f8948ed6553a field pic timestamp fix
michael
parents: 2107
diff changeset
294 int picture_structure, top_field_first, repeat_first_field, progressive_frame;
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
295 int horiz_size_ext, vert_size_ext;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
296
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
297 s->repeat_pict = 0;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
298 buf_end = buf + buf_size;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
299 while (buf < buf_end) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
300 start_code = find_start_code(&buf, buf_end);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
301 bytes_left = buf_end - buf;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
302 switch(start_code) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
303 case PICTURE_START_CODE:
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
304 if (bytes_left >= 2) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
305 s->pict_type = (buf[1] >> 3) & 7;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
306 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
307 break;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
308 case SEQ_START_CODE:
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
309 if (bytes_left >= 4) {
2269
535b7dfee202 lowres fixes for the parser
michael
parents: 2120
diff changeset
310 pc->width = (buf[0] << 4) | (buf[1] >> 4);
535b7dfee202 lowres fixes for the parser
michael
parents: 2120
diff changeset
311 pc->height = ((buf[1] & 0x0f) << 8) | buf[2];
2270
21f450be6cb5 lowres width/height cleanup 3rd try
michael
parents: 2269
diff changeset
312 avcodec_set_dimensions(avctx, pc->width, pc->height);
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
313 frame_rate_index = buf[3] & 0xf;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
314 pc->frame_rate = avctx->frame_rate = frame_rate_tab[frame_rate_index];
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
315 avctx->frame_rate_base = MPEG1_FRAME_RATE_BASE;
1681
27a272442d6b fill codec info
bellard
parents: 1614
diff changeset
316 avctx->codec_id = CODEC_ID_MPEG1VIDEO;
27a272442d6b fill codec info
bellard
parents: 1614
diff changeset
317 avctx->sub_id = 1;
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
318 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
319 break;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
320 case EXT_START_CODE:
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
321 if (bytes_left >= 1) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
322 ext_type = (buf[0] >> 4);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
323 switch(ext_type) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
324 case 0x1: /* sequence extension */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
325 if (bytes_left >= 6) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
326 horiz_size_ext = ((buf[1] & 1) << 1) | (buf[2] >> 7);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
327 vert_size_ext = (buf[2] >> 5) & 3;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
328 frame_rate_ext_n = (buf[5] >> 5) & 3;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
329 frame_rate_ext_d = (buf[5] & 0x1f);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
330 pc->progressive_sequence = buf[1] & (1 << 3);
2389
429c1eedeee9 fix timestamp prediction for low_delay mpeg streams
michael
parents: 2386
diff changeset
331 avctx->has_b_frames= buf[5] >> 7;
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
332
2269
535b7dfee202 lowres fixes for the parser
michael
parents: 2120
diff changeset
333 pc->width |=(horiz_size_ext << 12);
535b7dfee202 lowres fixes for the parser
michael
parents: 2120
diff changeset
334 pc->height |=( vert_size_ext << 12);
2270
21f450be6cb5 lowres width/height cleanup 3rd try
michael
parents: 2269
diff changeset
335 avcodec_set_dimensions(avctx, pc->width, pc->height);
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
336 avctx->frame_rate = pc->frame_rate * (frame_rate_ext_n + 1);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
337 avctx->frame_rate_base = MPEG1_FRAME_RATE_BASE * (frame_rate_ext_d + 1);
1681
27a272442d6b fill codec info
bellard
parents: 1614
diff changeset
338 avctx->codec_id = CODEC_ID_MPEG2VIDEO;
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
339 avctx->sub_id = 2; /* forces MPEG2 */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
340 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
341 break;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
342 case 0x8: /* picture coding extension */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
343 if (bytes_left >= 5) {
2120
872ac88d4e60 1000000l
michael
parents: 2119
diff changeset
344 picture_structure = buf[2]&3;
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
345 top_field_first = buf[3] & (1 << 7);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
346 repeat_first_field = buf[3] & (1 << 1);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
347 progressive_frame = buf[4] & (1 << 7);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
348
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
349 /* check if we must repeat the frame */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
350 if (repeat_first_field) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
351 if (pc->progressive_sequence) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
352 if (top_field_first)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
353 s->repeat_pict = 4;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
354 else
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
355 s->repeat_pict = 2;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
356 } else if (progressive_frame) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
357 s->repeat_pict = 1;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
358 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
359 }
2119
f8948ed6553a field pic timestamp fix
michael
parents: 2107
diff changeset
360
f8948ed6553a field pic timestamp fix
michael
parents: 2107
diff changeset
361 /* the packet only represents half a frame
f8948ed6553a field pic timestamp fix
michael
parents: 2107
diff changeset
362 XXX,FIXME maybe find a different solution */
f8948ed6553a field pic timestamp fix
michael
parents: 2107
diff changeset
363 if(picture_structure != 3)
f8948ed6553a field pic timestamp fix
michael
parents: 2107
diff changeset
364 s->repeat_pict = -1;
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
365 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
366 break;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
367 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
368 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
369 break;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
370 case -1:
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
371 goto the_end;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
372 default:
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
373 /* we stop parsing when we encounter a slice. It ensures
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
374 that this function takes a negligible amount of time */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
375 if (start_code >= SLICE_MIN_START_CODE &&
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
376 start_code <= SLICE_MAX_START_CODE)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
377 goto the_end;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
378 break;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
379 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
380 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
381 the_end: ;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
382 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
383
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
384 static int mpegvideo_parse(AVCodecParserContext *s,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
385 AVCodecContext *avctx,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
386 uint8_t **poutbuf, int *poutbuf_size,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
387 const uint8_t *buf, int buf_size)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
388 {
1988
b5753525f9a8 remove duplicated find_frame_end() code
michael
parents: 1987
diff changeset
389 ParseContext1 *pc1 = s->priv_data;
b5753525f9a8 remove duplicated find_frame_end() code
michael
parents: 1987
diff changeset
390 ParseContext *pc= &pc1->pc;
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
391 int next;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
392
1988
b5753525f9a8 remove duplicated find_frame_end() code
michael
parents: 1987
diff changeset
393 next= ff_mpeg1_find_frame_end(pc, buf, buf_size);
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
394
1988
b5753525f9a8 remove duplicated find_frame_end() code
michael
parents: 1987
diff changeset
395 if (ff_combine_frame(pc, next, (uint8_t **)&buf, &buf_size) < 0) {
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
396 *poutbuf = NULL;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
397 *poutbuf_size = 0;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
398 return buf_size;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
399 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
400 /* we have a full frame : we just parse the first few MPEG headers
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
401 to have the full timing information. The time take by this
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
402 function should be negligible for uncorrupted streams */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
403 mpegvideo_extract_headers(s, avctx, buf, buf_size);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
404 #if 0
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
405 printf("pict_type=%d frame_rate=%0.3f repeat_pict=%d\n",
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
406 s->pict_type, (double)avctx->frame_rate / avctx->frame_rate_base, s->repeat_pict);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
407 #endif
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
408
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
409 *poutbuf = (uint8_t *)buf;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
410 *poutbuf_size = buf_size;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
411 return next;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
412 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
413
1988
b5753525f9a8 remove duplicated find_frame_end() code
michael
parents: 1987
diff changeset
414 void ff_parse_close(AVCodecParserContext *s)
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
415 {
1988
b5753525f9a8 remove duplicated find_frame_end() code
michael
parents: 1987
diff changeset
416 ParseContext *pc = s->priv_data;
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
417
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
418 av_free(pc->buffer);
1988
b5753525f9a8 remove duplicated find_frame_end() code
michael
parents: 1987
diff changeset
419 }
b5753525f9a8 remove duplicated find_frame_end() code
michael
parents: 1987
diff changeset
420
b5753525f9a8 remove duplicated find_frame_end() code
michael
parents: 1987
diff changeset
421 static void parse1_close(AVCodecParserContext *s)
b5753525f9a8 remove duplicated find_frame_end() code
michael
parents: 1987
diff changeset
422 {
b5753525f9a8 remove duplicated find_frame_end() code
michael
parents: 1987
diff changeset
423 ParseContext1 *pc1 = s->priv_data;
b5753525f9a8 remove duplicated find_frame_end() code
michael
parents: 1987
diff changeset
424
b5753525f9a8 remove duplicated find_frame_end() code
michael
parents: 1987
diff changeset
425 av_free(pc1->pc.buffer);
b5753525f9a8 remove duplicated find_frame_end() code
michael
parents: 1987
diff changeset
426 av_free(pc1->enc);
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
427 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
428
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
429 /*************************/
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
430
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
431 /* used by parser */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
432 /* XXX: make it use less memory */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
433 static int av_mpeg4_decode_header(AVCodecParserContext *s1,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
434 AVCodecContext *avctx,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
435 const uint8_t *buf, int buf_size)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
436 {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
437 ParseContext1 *pc = s1->priv_data;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
438 MpegEncContext *s = pc->enc;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
439 GetBitContext gb1, *gb = &gb1;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
440 int ret;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
441
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
442 s->avctx = avctx;
1614
6c82ef97d3e6 also parse extradata for MPEG4
bellard
parents: 1613
diff changeset
443 s->current_picture_ptr = &s->current_picture;
6c82ef97d3e6 also parse extradata for MPEG4
bellard
parents: 1613
diff changeset
444
6c82ef97d3e6 also parse extradata for MPEG4
bellard
parents: 1613
diff changeset
445 if (avctx->extradata_size && pc->first_picture){
6c82ef97d3e6 also parse extradata for MPEG4
bellard
parents: 1613
diff changeset
446 init_get_bits(gb, avctx->extradata, avctx->extradata_size*8);
6c82ef97d3e6 also parse extradata for MPEG4
bellard
parents: 1613
diff changeset
447 ret = ff_mpeg4_decode_picture_header(s, gb);
6c82ef97d3e6 also parse extradata for MPEG4
bellard
parents: 1613
diff changeset
448 }
6c82ef97d3e6 also parse extradata for MPEG4
bellard
parents: 1613
diff changeset
449
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
450 init_get_bits(gb, buf, 8 * buf_size);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
451 ret = ff_mpeg4_decode_picture_header(s, gb);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
452 if (s->width) {
2270
21f450be6cb5 lowres width/height cleanup 3rd try
michael
parents: 2269
diff changeset
453 avcodec_set_dimensions(avctx, s->width, s->height);
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
454 }
1614
6c82ef97d3e6 also parse extradata for MPEG4
bellard
parents: 1613
diff changeset
455 pc->first_picture = 0;
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
456 return ret;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
457 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
458
2024
f65d87bfdd5a some of the warning fixes by (Michael Roitzsch <mroi at users dot sourceforge dot net>)
michael
parents: 1988
diff changeset
459 static int mpeg4video_parse_init(AVCodecParserContext *s)
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
460 {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
461 ParseContext1 *pc = s->priv_data;
1614
6c82ef97d3e6 also parse extradata for MPEG4
bellard
parents: 1613
diff changeset
462
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
463 pc->enc = av_mallocz(sizeof(MpegEncContext));
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
464 if (!pc->enc)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
465 return -1;
1614
6c82ef97d3e6 also parse extradata for MPEG4
bellard
parents: 1613
diff changeset
466 pc->first_picture = 1;
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
467 return 0;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
468 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
469
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
470 static int mpeg4video_parse(AVCodecParserContext *s,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
471 AVCodecContext *avctx,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
472 uint8_t **poutbuf, int *poutbuf_size,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
473 const uint8_t *buf, int buf_size)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
474 {
1988
b5753525f9a8 remove duplicated find_frame_end() code
michael
parents: 1987
diff changeset
475 ParseContext *pc = s->priv_data;
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
476 int next;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
477
1988
b5753525f9a8 remove duplicated find_frame_end() code
michael
parents: 1987
diff changeset
478 next= ff_mpeg4_find_frame_end(pc, buf, buf_size);
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
479
1988
b5753525f9a8 remove duplicated find_frame_end() code
michael
parents: 1987
diff changeset
480 if (ff_combine_frame(pc, next, (uint8_t **)&buf, &buf_size) < 0) {
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
481 *poutbuf = NULL;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
482 *poutbuf_size = 0;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
483 return buf_size;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
484 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
485 av_mpeg4_decode_header(s, avctx, buf, buf_size);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
486
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
487 *poutbuf = (uint8_t *)buf;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
488 *poutbuf_size = buf_size;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
489 return next;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
490 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
491
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
492 /*************************/
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
493
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
494 typedef struct MpegAudioParseContext {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
495 uint8_t inbuf[MPA_MAX_CODED_FRAME_SIZE]; /* input buffer */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
496 uint8_t *inbuf_ptr;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
497 int frame_size;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
498 int free_format_frame_size;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
499 int free_format_next_header;
2470
06aafb585f69 require a few valid and equal mp3 headers for resync
michael
parents: 2389
diff changeset
500 uint32_t header;
06aafb585f69 require a few valid and equal mp3 headers for resync
michael
parents: 2389
diff changeset
501 int header_count;
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
502 } MpegAudioParseContext;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
503
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
504 #define MPA_HEADER_SIZE 4
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
505
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
506 /* header + layer + bitrate + freq + lsf/mpeg25 */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
507 #define SAME_HEADER_MASK \
2480
5252700f61df 10000l vbr mp3 fix
michael
parents: 2470
diff changeset
508 (0xffe00000 | (3 << 17) | (3 << 10) | (3 << 19))
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
509
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
510 static int mpegaudio_parse_init(AVCodecParserContext *s1)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
511 {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
512 MpegAudioParseContext *s = s1->priv_data;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
513 s->inbuf_ptr = s->inbuf;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
514 return 0;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
515 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
516
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
517 static int mpegaudio_parse(AVCodecParserContext *s1,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
518 AVCodecContext *avctx,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
519 uint8_t **poutbuf, int *poutbuf_size,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
520 const uint8_t *buf, int buf_size)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
521 {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
522 MpegAudioParseContext *s = s1->priv_data;
2470
06aafb585f69 require a few valid and equal mp3 headers for resync
michael
parents: 2389
diff changeset
523 int len, ret, sr;
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
524 uint32_t header;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
525 const uint8_t *buf_ptr;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
526
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
527 *poutbuf = NULL;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
528 *poutbuf_size = 0;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
529 buf_ptr = buf;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
530 while (buf_size > 0) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
531 len = s->inbuf_ptr - s->inbuf;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
532 if (s->frame_size == 0) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
533 /* special case for next header for first frame in free
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
534 format case (XXX: find a simpler method) */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
535 if (s->free_format_next_header != 0) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
536 s->inbuf[0] = s->free_format_next_header >> 24;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
537 s->inbuf[1] = s->free_format_next_header >> 16;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
538 s->inbuf[2] = s->free_format_next_header >> 8;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
539 s->inbuf[3] = s->free_format_next_header;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
540 s->inbuf_ptr = s->inbuf + 4;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
541 s->free_format_next_header = 0;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
542 goto got_header;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
543 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
544 /* no header seen : find one. We need at least MPA_HEADER_SIZE
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
545 bytes to parse it */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
546 len = MPA_HEADER_SIZE - len;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
547 if (len > buf_size)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
548 len = buf_size;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
549 if (len > 0) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
550 memcpy(s->inbuf_ptr, buf_ptr, len);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
551 buf_ptr += len;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
552 buf_size -= len;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
553 s->inbuf_ptr += len;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
554 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
555 if ((s->inbuf_ptr - s->inbuf) >= MPA_HEADER_SIZE) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
556 got_header:
2470
06aafb585f69 require a few valid and equal mp3 headers for resync
michael
parents: 2389
diff changeset
557 sr= avctx->sample_rate;
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
558 header = (s->inbuf[0] << 24) | (s->inbuf[1] << 16) |
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
559 (s->inbuf[2] << 8) | s->inbuf[3];
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
560
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
561 ret = mpa_decode_header(avctx, header);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
562 if (ret < 0) {
2470
06aafb585f69 require a few valid and equal mp3 headers for resync
michael
parents: 2389
diff changeset
563 s->header_count= -2;
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
564 /* no sync found : move by one byte (inefficient, but simple!) */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
565 memmove(s->inbuf, s->inbuf + 1, s->inbuf_ptr - s->inbuf - 1);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
566 s->inbuf_ptr--;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
567 dprintf("skip %x\n", header);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
568 /* reset free format frame size to give a chance
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
569 to get a new bitrate */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
570 s->free_format_frame_size = 0;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
571 } else {
2470
06aafb585f69 require a few valid and equal mp3 headers for resync
michael
parents: 2389
diff changeset
572 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
573 s->header_count= -3;
06aafb585f69 require a few valid and equal mp3 headers for resync
michael
parents: 2389
diff changeset
574 s->header= header;
06aafb585f69 require a few valid and equal mp3 headers for resync
michael
parents: 2389
diff changeset
575 s->header_count++;
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
576 s->frame_size = ret;
2470
06aafb585f69 require a few valid and equal mp3 headers for resync
michael
parents: 2389
diff changeset
577
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
578 #if 0
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
579 /* free format: prepare to compute frame size */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
580 if (decode_header(s, header) == 1) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
581 s->frame_size = -1;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
582 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
583 #endif
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
584 }
2470
06aafb585f69 require a few valid and equal mp3 headers for resync
michael
parents: 2389
diff changeset
585 if(s->header_count <= 0)
06aafb585f69 require a few valid and equal mp3 headers for resync
michael
parents: 2389
diff changeset
586 avctx->sample_rate= sr; //FIXME ugly
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
587 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
588 } else
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
589 #if 0
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
590 if (s->frame_size == -1) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
591 /* free format : find next sync to compute frame size */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
592 len = MPA_MAX_CODED_FRAME_SIZE - len;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
593 if (len > buf_size)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
594 len = buf_size;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
595 if (len == 0) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
596 /* frame too long: resync */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
597 s->frame_size = 0;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
598 memmove(s->inbuf, s->inbuf + 1, s->inbuf_ptr - s->inbuf - 1);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
599 s->inbuf_ptr--;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
600 } else {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
601 uint8_t *p, *pend;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
602 uint32_t header1;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
603 int padding;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
604
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
605 memcpy(s->inbuf_ptr, buf_ptr, len);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
606 /* check for header */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
607 p = s->inbuf_ptr - 3;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
608 pend = s->inbuf_ptr + len - 4;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
609 while (p <= pend) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
610 header = (p[0] << 24) | (p[1] << 16) |
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
611 (p[2] << 8) | p[3];
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
612 header1 = (s->inbuf[0] << 24) | (s->inbuf[1] << 16) |
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
613 (s->inbuf[2] << 8) | s->inbuf[3];
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
614 /* check with high probability that we have a
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
615 valid header */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
616 if ((header & SAME_HEADER_MASK) ==
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
617 (header1 & SAME_HEADER_MASK)) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
618 /* header found: update pointers */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
619 len = (p + 4) - s->inbuf_ptr;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
620 buf_ptr += len;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
621 buf_size -= len;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
622 s->inbuf_ptr = p;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
623 /* compute frame size */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
624 s->free_format_next_header = header;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
625 s->free_format_frame_size = s->inbuf_ptr - s->inbuf;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
626 padding = (header1 >> 9) & 1;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
627 if (s->layer == 1)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
628 s->free_format_frame_size -= padding * 4;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
629 else
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
630 s->free_format_frame_size -= padding;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
631 dprintf("free frame size=%d padding=%d\n",
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
632 s->free_format_frame_size, padding);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
633 decode_header(s, header1);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
634 goto next_data;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
635 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
636 p++;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
637 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
638 /* not found: simply increase pointers */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
639 buf_ptr += len;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
640 s->inbuf_ptr += len;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
641 buf_size -= len;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
642 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
643 } else
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
644 #endif
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
645 if (len < s->frame_size) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
646 if (s->frame_size > MPA_MAX_CODED_FRAME_SIZE)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
647 s->frame_size = MPA_MAX_CODED_FRAME_SIZE;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
648 len = s->frame_size - len;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
649 if (len > buf_size)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
650 len = buf_size;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
651 memcpy(s->inbuf_ptr, buf_ptr, len);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
652 buf_ptr += len;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
653 s->inbuf_ptr += len;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
654 buf_size -= len;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
655 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
656 // next_data:
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
657 if (s->frame_size > 0 &&
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
658 (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
659 if(s->header_count > 0){
06aafb585f69 require a few valid and equal mp3 headers for resync
michael
parents: 2389
diff changeset
660 *poutbuf = s->inbuf;
06aafb585f69 require a few valid and equal mp3 headers for resync
michael
parents: 2389
diff changeset
661 *poutbuf_size = s->inbuf_ptr - s->inbuf;
06aafb585f69 require a few valid and equal mp3 headers for resync
michael
parents: 2389
diff changeset
662 }
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
663 s->inbuf_ptr = s->inbuf;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
664 s->frame_size = 0;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
665 break;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
666 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
667 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
668 return buf_ptr - buf;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
669 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
670
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
671 #ifdef CONFIG_AC3
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
672 extern int a52_syncinfo (const uint8_t * buf, int * flags,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
673 int * sample_rate, int * bit_rate);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
674
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
675 typedef struct AC3ParseContext {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
676 uint8_t inbuf[4096]; /* input buffer */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
677 uint8_t *inbuf_ptr;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
678 int frame_size;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
679 int flags;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
680 } AC3ParseContext;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
681
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
682 #define AC3_HEADER_SIZE 7
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
683 #define A52_LFE 16
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
684
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
685 static int ac3_parse_init(AVCodecParserContext *s1)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
686 {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
687 AC3ParseContext *s = s1->priv_data;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
688 s->inbuf_ptr = s->inbuf;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
689 return 0;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
690 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
691
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
692 static int ac3_parse(AVCodecParserContext *s1,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
693 AVCodecContext *avctx,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
694 uint8_t **poutbuf, int *poutbuf_size,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
695 const uint8_t *buf, int buf_size)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
696 {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
697 AC3ParseContext *s = s1->priv_data;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
698 const uint8_t *buf_ptr;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
699 int len, sample_rate, bit_rate;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
700 static const int ac3_channels[8] = {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
701 2, 1, 2, 3, 3, 4, 4, 5
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
702 };
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
703
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
704 *poutbuf = NULL;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
705 *poutbuf_size = 0;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
706
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
707 buf_ptr = buf;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
708 while (buf_size > 0) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
709 len = s->inbuf_ptr - s->inbuf;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
710 if (s->frame_size == 0) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
711 /* no header seen : find one. We need at least 7 bytes to parse it */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
712 len = AC3_HEADER_SIZE - len;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
713 if (len > buf_size)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
714 len = buf_size;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
715 memcpy(s->inbuf_ptr, buf_ptr, len);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
716 buf_ptr += len;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
717 s->inbuf_ptr += len;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
718 buf_size -= len;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
719 if ((s->inbuf_ptr - s->inbuf) == AC3_HEADER_SIZE) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
720 len = a52_syncinfo(s->inbuf, &s->flags, &sample_rate, &bit_rate);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
721 if (len == 0) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
722 /* no sync found : move by one byte (inefficient, but simple!) */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
723 memmove(s->inbuf, s->inbuf + 1, AC3_HEADER_SIZE - 1);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
724 s->inbuf_ptr--;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
725 } else {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
726 s->frame_size = len;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
727 /* update codec info */
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
728 avctx->sample_rate = sample_rate;
1987
d9e067853051 >2 channels decoding fix
michael
parents: 1696
diff changeset
729 /* set channels,except if the user explicitly requests 1 or 2 channels, XXX/FIXME this is a bit ugly */
d9e067853051 >2 channels decoding fix
michael
parents: 1696
diff changeset
730 if(avctx->channels!=1 && avctx->channels!=2){
d9e067853051 >2 channels decoding fix
michael
parents: 1696
diff changeset
731 avctx->channels = ac3_channels[s->flags & 7];
d9e067853051 >2 channels decoding fix
michael
parents: 1696
diff changeset
732 if (s->flags & A52_LFE)
d9e067853051 >2 channels decoding fix
michael
parents: 1696
diff changeset
733 avctx->channels++;
d9e067853051 >2 channels decoding fix
michael
parents: 1696
diff changeset
734 }
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
735 avctx->bit_rate = bit_rate;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
736 avctx->frame_size = 6 * 256;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
737 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
738 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
739 } else if (len < s->frame_size) {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
740 len = s->frame_size - len;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
741 if (len > buf_size)
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
742 len = buf_size;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
743
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
744 memcpy(s->inbuf_ptr, buf_ptr, len);
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
745 buf_ptr += len;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
746 s->inbuf_ptr += len;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
747 buf_size -= len;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
748 } else {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
749 *poutbuf = s->inbuf;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
750 *poutbuf_size = s->frame_size;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
751 s->inbuf_ptr = s->inbuf;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
752 s->frame_size = 0;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
753 break;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
754 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
755 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
756 return buf_ptr - buf;
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
757 }
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
758 #endif
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
759
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
760 AVCodecParser mpegvideo_parser = {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
761 { CODEC_ID_MPEG1VIDEO, CODEC_ID_MPEG2VIDEO },
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
762 sizeof(ParseContext1),
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
763 NULL,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
764 mpegvideo_parse,
1988
b5753525f9a8 remove duplicated find_frame_end() code
michael
parents: 1987
diff changeset
765 parse1_close,
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
766 };
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
767
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
768 AVCodecParser mpeg4video_parser = {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
769 { CODEC_ID_MPEG4 },
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
770 sizeof(ParseContext1),
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
771 mpeg4video_parse_init,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
772 mpeg4video_parse,
1988
b5753525f9a8 remove duplicated find_frame_end() code
michael
parents: 1987
diff changeset
773 parse1_close,
1613
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
774 };
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
775
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
776 AVCodecParser mpegaudio_parser = {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
777 { CODEC_ID_MP2, CODEC_ID_MP3 },
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
778 sizeof(MpegAudioParseContext),
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
779 mpegaudio_parse_init,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
780 mpegaudio_parse,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
781 NULL,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
782 };
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
783
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
784 #ifdef CONFIG_AC3
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
785 AVCodecParser ac3_parser = {
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
786 { CODEC_ID_AC3 },
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
787 sizeof(AC3ParseContext),
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
788 ac3_parse_init,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
789 ac3_parse,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
790 NULL,
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
791 };
0279c6c61f11 new audio/video parser API
bellard
parents:
diff changeset
792 #endif