(转)ffmpeg 中 av_read_frame_internal分析

作者: chenwei1983    时间: 2012-3-5 04:21 PM
标题: ffmpeg 中 av_read_frame_internal分析                            原出处:http://www.chinavideo.org/viewthread.php?action=printable&tid=13846
av_read_frame_internal 在ffmpeg中实现了将format格式的packet,最终转换成一帧帧的es流packet，并解析填充了packet的pts,dts，等信息，为最终解码提供了重要的数据,av_read_frame_internal,调用av_read_packet,每次只读取一个包，然后直到parser完这个包的所有数据，才开始读取下一个包，parser完的数据被保存在parser结构的数据缓冲中，这样即使av_read_packet读取的下一包和前一包的流不一样，由于parser也不一样，所以实现了av_read_frame_internal这个函数调用，可以解析出不同流的es流,而av_read_frame_internal函数除非出错否则必须解析出一帧数据才能返回
static int av_read_frame_internal(AVFormatContext *s, AVPacket *pkt)
{
    AVStream *st;
    int len, ret, i;

av_init_packet(pkt);

for(;;) {
        //s->cur_st记录现在的stream指针，如果真，就需要接着分析packet数据，否则需要读包
        st = s->cur_st;
        if (st) {
        //当流不需要parser或者parser为空时,即没有注册这种parser，就不要parser 包了,输出原始包数据
            if (!st->need_parsing || !st->parser) {
                /* no parsing needed: we just output the packet as is */
                /* raw data support */
                *pkt = s->cur_pkt;
                compute_pkt_fields(s, st, NULL, pkt);//计算包的，其它信息
                s->cur_st = NULL; //当输出一个帧数据时，需要将现在的流置成NULL,这是因为下一个包并不一定和现在是同一个流，下次调用av_read_frame_internal 时，也能够直接进入读包的那个分支
                break;
                      //当现在的数据长度大于0,并且不是丢掉所有的包条件下,进去parser
            } else if (s->cur_len > 0 && st->discard < AVDISCARD_ALL) {
                //如果数据足够parser出一帧就会返回,注意ffmpeg 的parser设计得非常好，它并不需要每次从头将所有的流parser一遍，而只要每次parser新的数据就可以了,直到parser出一个完整帧出来
                len = av_parser_parse(st->parser, st->codec, &pkt->data, &pkt->size,
                                      s->cur_ptr, s->cur_len,
                                      s->cur_pkt.pts, s->cur_pkt.dts);
                s->cur_pkt.pts = AV_NOPTS_VALUE;
                s->cur_pkt.dts = AV_NOPTS_VALUE;
                /* increment read pointer */
               //如果一个原始包里有多个帧，那就需要调用多次av_read_frame_internal 
                s->cur_ptr += len;
                s->cur_len -= len;

/* return packet if any */
                //当parser出一帧后，pkt->size大小为真
                if (pkt->size) {
                got_packet:
                                    //填充pkt的其它信息
                    pkt->pos = s->cur_pkt.pos;              // Isn't quite accurate but close.
                    pkt->duration = 0;
                    pkt->stream_index = st->index;
                    pkt->pts = st->parser->pts;
                    pkt->dts = st->parser->dts;
//注意这里使用这个destruct函数的含义，因为这里 pkt->data的指针并没有实际分配内存，而是指向了
parser结构中数据buffer,所以这里后面需要处理
                    pkt->destruct = av_destruct_packet_nofree;
                    compute_pkt_fields(s, st, st->parser, pkt);
                    //当需要产生通用的index时，并且是关键帧，那么需要将这一帧添加到index表中
                    if((s->iformat->flags & AVFMT_GENERIC_INDEX) && pkt->flags & PKT_FLAG_KEY){
                        ff_reduce_index(s, st->index);
                        av_add_index_entry(st, st->parser->frame_offset, pkt->dts,
                                           0, 0, AVINDEX_KEYFRAME);
                    }

break;
                }
            } else {//s->cur_len == 0 ，当 s->cur_ptr  数据解析完了，则需要重新读取一个原始包来分析
                /* free packet */
                av_free_packet(&s->cur_pkt);
                s->cur_st = NULL;
            }
        } else {//s->cur_st 为NULL时,读取一个原始包
            /* read next packet */
            ret = av_read_packet(s, &s->cur_pkt);
            if (ret < 0) {
                if (ret == AVERROR(EAGAIN))
                    return ret;
                /* return the last frames, if any */
                                //输出最后的包
                for(i = 0; i < s->nb_streams; i++) {
                    st = s->streams[i];
                    if (st->parser && st->need_parsing) {
                        av_parser_parse(st->parser, st->codec,
                                        &pkt->data, &pkt->size,
                                        NULL, 0,
                                        AV_NOPTS_VALUE, AV_NOPTS_VALUE);
                        if (pkt->size)
                            goto got_packet;
                    }
                }
                /* no more packets: really terminate parsing */
                return ret;
            }

if(s->cur_pkt.pts != AV_NOPTS_VALUE &&
               s->cur_pkt.dts != AV_NOPTS_VALUE &&
               s->cur_pkt.pts < s->cur_pkt.dts){//如果pts < dts 出错，返回
                av_log(s, AV_LOG_WARNING, "Invalid timestamps stream=%d, pts=%"PRId64", dts=%"PRId64", size=%d\n",
                    s->cur_pkt.stream_index,
                    s->cur_pkt.pts,
                    s->cur_pkt.dts,
                    s->cur_pkt.size);
//                av_free_packet(&s->cur_pkt);
//                return -1;
            }

st = s->streams[s->cur_pkt.stream_index];
            if(s->debug & FF_FDEBUG_TS)
                av_log(s, AV_LOG_DEBUG, "av_read_packet stream=%d, pts=%"PRId64", dts=%"PRId64", size=%d,  flags=%d\n",
                    s->cur_pkt.stream_index,
                    s->cur_pkt.pts,
                    s->cur_pkt.dts,
                    s->cur_pkt.size,
                    s->cur_pkt.flags);

s->cur_st = st;//读取一个包后，设置好cur_ptr,cur_len,以便解析
            s->cur_ptr = s->cur_pkt.data;//
            s->cur_len = s->cur_pkt.size;//
// 当准备好解析数据时，如果parser为空则创建parser
            if (st->need_parsing && !st->parser) {
                st->parser = av_parser_init(st->codec->codec_id);
                if (!st->parser) {
                    /* no parser available: just output the raw packets */
                    st->need_parsing = AVSTREAM_PARSE_NONE;
                }else if(st->need_parsing == AVSTREAM_PARSE_HEADERS){
                    st->parser->flags |= PARSER_FLAG_COMPLETE_FRAMES;
                }
                if(st->parser && (s->iformat->flags & AVFMT_GENERIC_INDEX)){
                    st->parser->next_frame_offset=
                    st->parser->cur_offset= s->cur_pkt.pos;
                }
            }
        }
    }
    if(s->debug & FF_FDEBUG_TS)
        av_log(s, AV_LOG_DEBUG, "av_read_frame_internal stream=%d, pts=%"PRId64", dts=%"PRId64", size=%d, flags=%d\n",
            pkt->stream_index,
            pkt->pts,
            pkt->dts,
            pkt->size,
            pkt->flags);

return 0;
}