ffmpeg源码简析（六）编码-av_write_frame()，av_write_trailer()

1.av_write_frame()

av_write_frame()用于输出一帧视音频数据，它的声明位于libavformat\avformat.h，如下所示。

int av_write_frame(AVFormatContext *s, AVPacket *pkt); 

简单解释一下它的参数的含义：    s：用于输出的AVFormatContext。    pkt：等待输出的AVPacket。函数正常执行后返回值等于0。

av_write_frame()的定义位于libavformat\mux.c，如下所示。

    int av_write_frame(AVFormatContext *s, AVPacket *pkt)      {          int ret;          ret = check_packet(s, pkt);          if (ret < 0)              return ret;          //Packet为NULL，Flush Encoder          if (!pkt) {              if (s->oformat->flags & AVFMT_ALLOW_FLUSH) {                  ret = s->oformat->write_packet(s, NULL);                  if (s->flush_packets && s->pb && s->pb->error >= 0 && s->flags & AVFMT_FLAG_FLUSH_PACKETS)                      avio_flush(s->pb);                  if (ret >= 0 && s->pb && s->pb->error < 0)                      ret = s->pb->error;                  return ret;              }              return 1;          }          ret = compute_pkt_fields2(s, s->streams[pkt->stream_index], pkt);          if (ret < 0 && !(s->oformat->flags & AVFMT_NOTIMESTAMPS))              return ret;          //写入          ret = write_packet(s, pkt);          if (ret >= 0 && s->pb && s->pb->error < 0)              ret = s->pb->error;          if (ret >= 0)              s->streams[pkt->stream_index]->nb_frames++;          return ret;      }  

从源代码可以看出，av_write_frame()主要完成了以下几步工作：
（1）调用check_packet()做一些简单的检测
（2）调用compute_pkt_fields2()设置AVPacket的一些属性值
（3）调用write_packet()写入数据

下面分别看一下这几个函数功能。
check_packet()
check_packet()定义位于libavformat\mux.c，如下所示。

    static int check_packet(AVFormatContext *s, AVPacket *pkt)      {          if (!pkt)              return 0;          if (pkt->stream_index < 0 || pkt->stream_index >= s->nb_streams) {              av_log(s, AV_LOG_ERROR, "Invalid packet stream index: %d\n",                     pkt->stream_index);              return AVERROR(EINVAL);          }          if (s->streams[pkt->stream_index]->codec->codec_type == AVMEDIA_TYPE_ATTACHMENT) {              av_log(s, AV_LOG_ERROR, "Received a packet for an attachment stream.\n");              return AVERROR(EINVAL);          }          return 0;      }  

从代码中可以看出，check_packet()的功能比较简单：首先检查一下输入的AVPacket是否为空，如果为空，则是直接返回；然后检查一下AVPacket的stream_index（标记了该AVPacket所属的AVStream）设置是否正常，如果为负数或者大于AVStream的个数，则返回错误信息；

compute_pkt_fields2()
compute_pkt_fields2()主要有两方面的功能：一方面用于计算AVPacket的duration， dts等信息；另一方面用于检查pts、dts这些参数的合理性（例如PTS是否一定大于DTS）

AVOutputFormat->write_packet()
write_packet()函数最关键的地方就是调用了AVOutputFormat中写入数据的方法。如果AVPacket中的flag标记中包含AV_PKT_FLAG_UNCODED_FRAME，就会调用AVOutputFormat的write_uncoded_frame()函数；如果不包含那个标记，就会调用write_packet()函数。write_packet()实际上是一个函数指针，指向特定的AVOutputFormat中的实现函数。

2.av_write_trailer()

av_write_trailer()用于输出文件尾，它的声明位于libavformat\avformat.h，如下所示。

int av_write_trailer(AVFormatContext *s);  

它只需要指定一个参数，即用于输出的AVFormatContext。
函数正常执行后返回值等于0。
av_write_trailer()的定义位于libavformat\mux.c，如下所示。

    int av_write_trailer(AVFormatContext *s)      {          int ret, i;          for (;; ) {              AVPacket pkt;              ret = interleave_packet(s, &pkt, NULL, 1);              if (ret < 0)                  goto fail;              if (!ret)                  break;              //写入AVPacket              ret = write_packet(s, &pkt);              if (ret >= 0)                  s->streams[pkt.stream_index]->nb_frames++;              av_free_packet(&pkt);              if (ret < 0)                  goto fail;              if(s->pb && s->pb->error)                  goto fail;          }      fail:          //写文件尾          if (s->oformat->write_trailer)              if (ret >= 0) {              ret = s->oformat->write_trailer(s);              } else {                  s->oformat->write_trailer(s);              }          if (s->pb)             avio_flush(s->pb);          if (ret == 0)             ret = s->pb ? s->pb->error : 0;          for (i = 0; i < s->nb_streams; i++) {              av_freep(&s->streams[i]->priv_data);              av_freep(&s->streams[i]->index_entries);          }          if (s->oformat->priv_class)              av_opt_free(s->priv_data);          av_freep(&s->priv_data);          return ret;      }  

从源代码可以看出av_write_trailer()主要完成了以下两步工作：

（1）循环调用interleave_packet()以及write_packet()，将还未输出的AVPacket输出出来。

（2）调用AVOutputFormat的write_trailer()，输出文件尾。

AVOutputFormat->write_trailer()
AVOutputFormat的write_trailer()是一个函数指针，指向特定的AVOutputFormat中的实现函数。