indent
[ffmpeg.git] / libavcodec / parser.c
1 /*
2  * Audio and Video frame extraction
3  * Copyright (c) 2003 Fabrice Bellard.
4  * Copyright (c) 2003 Michael Niedermayer.
5  *
6  * This file is part of FFmpeg.
7  *
8  * FFmpeg is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * FFmpeg is distributed in the hope that it will be useful,
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
19  * License along with FFmpeg; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22
23 #include "parser.h"
24
25 AVCodecParser *av_first_parser = NULL;
26
27 AVCodecParser* av_parser_next(AVCodecParser *p){
28     if(p) return p->next;
29     else  return av_first_parser;
30 }
31
32 void av_register_codec_parser(AVCodecParser *parser)
33 {
34     parser->next = av_first_parser;
35     av_first_parser = parser;
36 }
37
38 AVCodecParserContext *av_parser_init(int codec_id)
39 {
40     AVCodecParserContext *s;
41     AVCodecParser *parser;
42     int ret;
43
44     if(codec_id == CODEC_ID_NONE)
45         return NULL;
46
47     for(parser = av_first_parser; parser != NULL; parser = parser->next) {
48         if (parser->codec_ids[0] == codec_id ||
49             parser->codec_ids[1] == codec_id ||
50             parser->codec_ids[2] == codec_id ||
51             parser->codec_ids[3] == codec_id ||
52             parser->codec_ids[4] == codec_id)
53             goto found;
54     }
55     return NULL;
56  found:
57     s = av_mallocz(sizeof(AVCodecParserContext));
58     if (!s)
59         return NULL;
60     s->parser = parser;
61     s->priv_data = av_mallocz(parser->priv_data_size);
62     if (!s->priv_data) {
63         av_free(s);
64         return NULL;
65     }
66     if (parser->parser_init) {
67         ret = parser->parser_init(s);
68         if (ret != 0) {
69             av_free(s->priv_data);
70             av_free(s);
71             return NULL;
72         }
73     }
74     s->fetch_timestamp=1;
75     s->pict_type = FF_I_TYPE;
76     return s;
77 }
78
79 /**
80  *
81  * @param buf           input
82  * @param buf_size      input length, to signal EOF, this should be 0 (so that the last frame can be output)
83  * @param pts           input presentation timestamp
84  * @param dts           input decoding timestamp
85  * @param poutbuf       will contain a pointer to the first byte of the output frame
86  * @param poutbuf_size  will contain the length of the output frame
87  * @return the number of bytes of the input bitstream used
88  *
89  * Example:
90  * @code
91  *   while(in_len){
92  *       len = av_parser_parse(myparser, AVCodecContext, &data, &size,
93  *                                       in_data, in_len,
94  *                                       pts, dts);
95  *       in_data += len;
96  *       in_len  -= len;
97  *
98  *       if(size)
99  *          decode_frame(data, size);
100  *   }
101  * @endcode
102  */
103 int av_parser_parse(AVCodecParserContext *s,
104                     AVCodecContext *avctx,
105                     uint8_t **poutbuf, int *poutbuf_size,
106                     const uint8_t *buf, int buf_size,
107                     int64_t pts, int64_t dts)
108 {
109     int index, i, k;
110     uint8_t dummy_buf[FF_INPUT_BUFFER_PADDING_SIZE];
111
112     if (buf_size == 0) {
113         /* padding is always necessary even if EOF, so we add it here */
114         memset(dummy_buf, 0, sizeof(dummy_buf));
115         buf = dummy_buf;
116     } else {
117         /* add a new packet descriptor */
118         k = (s->cur_frame_start_index + 1) & (AV_PARSER_PTS_NB - 1);
119         s->cur_frame_start_index = k;
120         s->cur_frame_offset[k] = s->cur_offset;
121         s->cur_frame_pts[k] = pts;
122         s->cur_frame_dts[k] = dts;
123
124         /* fill first PTS/DTS */
125         if (s->fetch_timestamp){
126             s->fetch_timestamp=0;
127             s->last_pts = pts;
128             s->last_dts = dts;
129             s->last_offset = 0;
130             s->cur_frame_pts[k] =
131             s->cur_frame_dts[k] = AV_NOPTS_VALUE;
132         }
133     }
134
135     /* WARNING: the returned index can be negative */
136     index = s->parser->parser_parse(s, avctx, (const uint8_t **)poutbuf, poutbuf_size, buf, buf_size);
137 //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);
138     /* update the file pointer */
139     if (*poutbuf_size) {
140         /* fill the data for the current frame */
141         s->frame_offset = s->last_frame_offset;
142         s->pts = s->last_pts;
143         s->dts = s->last_dts;
144         s->offset = s->last_offset;
145
146         /* offset of the next frame */
147         s->last_frame_offset = s->cur_offset + index;
148         /* find the packet in which the new frame starts. It
149            is tricky because of MPEG video start codes
150            which can begin in one packet and finish in
151            another packet. In the worst case, an MPEG
152            video start code could be in 4 different
153            packets. */
154         k = s->cur_frame_start_index;
155         for(i = 0; i < AV_PARSER_PTS_NB; i++) {
156             if (s->last_frame_offset >= s->cur_frame_offset[k])
157                 break;
158             k = (k - 1) & (AV_PARSER_PTS_NB - 1);
159         }
160
161         s->last_pts = s->cur_frame_pts[k];
162         s->last_dts = s->cur_frame_dts[k];
163         s->last_offset = s->last_frame_offset - s->cur_frame_offset[k];
164
165         /* some parsers tell us the packet size even before seeing the first byte of the next packet,
166            so the next pts/dts is in the next chunk */
167         if(index == buf_size){
168             s->fetch_timestamp=1;
169         }
170     }
171     if (index < 0)
172         index = 0;
173     s->cur_offset += index;
174     return index;
175 }
176
177 /**
178  *
179  * @return 0 if the output buffer is a subset of the input, 1 if it is allocated and must be freed
180  * @deprecated use AVBitstreamFilter
181  */
182 int av_parser_change(AVCodecParserContext *s,
183                      AVCodecContext *avctx,
184                      uint8_t **poutbuf, int *poutbuf_size,
185                      const uint8_t *buf, int buf_size, int keyframe){
186
187     if(s && s->parser->split){
188         if((avctx->flags & CODEC_FLAG_GLOBAL_HEADER) || (avctx->flags2 & CODEC_FLAG2_LOCAL_HEADER)){
189             int i= s->parser->split(avctx, buf, buf_size);
190             buf += i;
191             buf_size -= i;
192         }
193     }
194
195     /* cast to avoid warning about discarding qualifiers */
196     *poutbuf= (uint8_t *) buf;
197     *poutbuf_size= buf_size;
198     if(avctx->extradata){
199         if(  (keyframe && (avctx->flags2 & CODEC_FLAG2_LOCAL_HEADER))
200             /*||(s->pict_type != FF_I_TYPE && (s->flags & PARSER_FLAG_DUMP_EXTRADATA_AT_NOKEY))*/
201             /*||(? && (s->flags & PARSER_FLAG_DUMP_EXTRADATA_AT_BEGIN)*/){
202             int size= buf_size + avctx->extradata_size;
203             *poutbuf_size= size;
204             *poutbuf= av_malloc(size + FF_INPUT_BUFFER_PADDING_SIZE);
205
206             memcpy(*poutbuf, avctx->extradata, avctx->extradata_size);
207             memcpy((*poutbuf) + avctx->extradata_size, buf, buf_size + FF_INPUT_BUFFER_PADDING_SIZE);
208             return 1;
209         }
210     }
211
212     return 0;
213 }
214
215 void av_parser_close(AVCodecParserContext *s)
216 {
217     if(s){
218         if (s->parser->parser_close)
219             s->parser->parser_close(s);
220         av_free(s->priv_data);
221         av_free(s);
222     }
223 }
224
225 /*****************************************************/
226
227 /**
228  * combines the (truncated) bitstream to a complete frame
229  * @returns -1 if no complete frame could be created, AVERROR(ENOMEM) if there was a memory allocation error
230  */
231 int ff_combine_frame(ParseContext *pc, int next, const uint8_t **buf, int *buf_size)
232 {
233 #if 0
234     if(pc->overread){
235         printf("overread %d, state:%X next:%d index:%d o_index:%d\n", pc->overread, pc->state, next, pc->index, pc->overread_index);
236         printf("%X %X %X %X\n", (*buf)[0], (*buf)[1],(*buf)[2],(*buf)[3]);
237     }
238 #endif
239
240     /* Copy overread bytes from last frame into buffer. */
241     for(; pc->overread>0; pc->overread--){
242         pc->buffer[pc->index++]= pc->buffer[pc->overread_index++];
243     }
244
245     /* flush remaining if EOF */
246     if(!*buf_size && next == END_NOT_FOUND){
247         next= 0;
248     }
249
250     pc->last_index= pc->index;
251
252     /* copy into buffer end return */
253     if(next == END_NOT_FOUND){
254         void* new_buffer = av_fast_realloc(pc->buffer, &pc->buffer_size, (*buf_size) + pc->index + FF_INPUT_BUFFER_PADDING_SIZE);
255
256         if(!new_buffer)
257             return AVERROR(ENOMEM);
258         pc->buffer = new_buffer;
259         memcpy(&pc->buffer[pc->index], *buf, *buf_size);
260         pc->index += *buf_size;
261         return -1;
262     }
263
264     *buf_size=
265     pc->overread_index= pc->index + next;
266
267     /* append to buffer */
268     if(pc->index){
269         void* new_buffer = av_fast_realloc(pc->buffer, &pc->buffer_size, next + pc->index + FF_INPUT_BUFFER_PADDING_SIZE);
270
271         if(!new_buffer)
272             return AVERROR(ENOMEM);
273         pc->buffer = new_buffer;
274         memcpy(&pc->buffer[pc->index], *buf, next + FF_INPUT_BUFFER_PADDING_SIZE );
275         pc->index = 0;
276         *buf= pc->buffer;
277     }
278
279     /* store overread bytes */
280     for(;next < 0; next++){
281         pc->state = (pc->state<<8) | pc->buffer[pc->last_index + next];
282         pc->overread++;
283     }
284
285 #if 0
286     if(pc->overread){
287         printf("overread %d, state:%X next:%d index:%d o_index:%d\n", pc->overread, pc->state, next, pc->index, pc->overread_index);
288         printf("%X %X %X %X\n", (*buf)[0], (*buf)[1],(*buf)[2],(*buf)[3]);
289     }
290 #endif
291
292     return 0;
293 }
294
295 void ff_parse_close(AVCodecParserContext *s)
296 {
297     ParseContext *pc = s->priv_data;
298
299     av_free(pc->buffer);
300 }
301
302 void ff_parse1_close(AVCodecParserContext *s)
303 {
304     ParseContext1 *pc1 = s->priv_data;
305
306     av_free(pc1->pc.buffer);
307     av_free(pc1->enc);
308 }
309
310 /*************************/
311
312 int ff_mpeg4video_split(AVCodecContext *avctx,
313                            const uint8_t *buf, int buf_size)
314 {
315     int i;
316     uint32_t state= -1;
317
318     for(i=0; i<buf_size; i++){
319         state= (state<<8) | buf[i];
320         if(state == 0x1B3 || state == 0x1B6)
321             return i-3;
322     }
323     return 0;
324 }