git.sesse.net Git - ffmpeg/blob - 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->parser->parser_close)
 218         s->parser->parser_close(s);
 219     av_free(s->priv_data);
 220     av_free(s);
 221 }
 222
 223 /*****************************************************/
 224
 225 /**
 226  * combines the (truncated) bitstream to a complete frame
 227  * @returns -1 if no complete frame could be created
 228  */
 229 int ff_combine_frame(ParseContext *pc, int next, const uint8_t **buf, int *buf_size)
 230 {
 231 #if 0
 232     if(pc->overread){
 233         printf("overread %d, state:%X next:%d index:%d o_index:%d\n", pc->overread, pc->state, next, pc->index, pc->overread_index);
 234         printf("%X %X %X %X\n", (*buf)[0], (*buf)[1],(*buf)[2],(*buf)[3]);
 235     }
 236 #endif
 237
 238     /* Copy overread bytes from last frame into buffer. */
 239     for(; pc->overread>0; pc->overread--){
 240         pc->buffer[pc->index++]= pc->buffer[pc->overread_index++];
 241     }
 242
 243     /* flush remaining if EOF */
 244     if(!*buf_size && next == END_NOT_FOUND){
 245         next= 0;
 246     }
 247
 248     pc->last_index= pc->index;
 249
 250     /* copy into buffer end return */
 251     if(next == END_NOT_FOUND){
 252         pc->buffer= av_fast_realloc(pc->buffer, &pc->buffer_size, (*buf_size) + pc->index + FF_INPUT_BUFFER_PADDING_SIZE);
 253
 254         memcpy(&pc->buffer[pc->index], *buf, *buf_size);
 255         pc->index += *buf_size;
 256         return -1;
 257     }
 258
 259     *buf_size=
 260     pc->overread_index= pc->index + next;
 261
 262     /* append to buffer */
 263     if(pc->index){
 264         pc->buffer= av_fast_realloc(pc->buffer, &pc->buffer_size, next + pc->index + FF_INPUT_BUFFER_PADDING_SIZE);
 265
 266         memcpy(&pc->buffer[pc->index], *buf, next + FF_INPUT_BUFFER_PADDING_SIZE );
 267         pc->index = 0;
 268         *buf= pc->buffer;
 269     }
 270
 271     /* store overread bytes */
 272     for(;next < 0; next++){
 273         pc->state = (pc->state<<8) | pc->buffer[pc->last_index + next];
 274         pc->overread++;
 275     }
 276
 277 #if 0
 278     if(pc->overread){
 279         printf("overread %d, state:%X next:%d index:%d o_index:%d\n", pc->overread, pc->state, next, pc->index, pc->overread_index);
 280         printf("%X %X %X %X\n", (*buf)[0], (*buf)[1],(*buf)[2],(*buf)[3]);
 281     }
 282 #endif
 283
 284     return 0;
 285 }
 286
 287 void ff_parse_close(AVCodecParserContext *s)
 288 {
 289     ParseContext *pc = s->priv_data;
 290
 291     av_free(pc->buffer);
 292 }
 293
 294 void ff_parse1_close(AVCodecParserContext *s)
 295 {
 296     ParseContext1 *pc1 = s->priv_data;
 297
 298     av_free(pc1->pc.buffer);
 299     av_free(pc1->enc);
 300 }
 301
 302 /*************************/
 303
 304 int ff_mpeg4video_split(AVCodecContext *avctx,
 305                            const uint8_t *buf, int buf_size)
 306 {
 307     int i;
 308     uint32_t state= -1;
 309
 310     for(i=0; i<buf_size; i++){
 311         state= (state<<8) | buf[i];
 312         if(state == 0x1B3 || state == 0x1B6)
 313             return i-3;
 314     }
 315     return 0;
 316 }