音视频入门基础：MPEG2-PS专题（6）——FFmpeg源码中，获取PS流的视频信息的实现

=================================================================

音视频入门基础：MPEG2-PS专题系列文章：

音视频入门基础：MPEG2-PS专题（1）——MPEG2-PS官方文档下载

音视频入门基础：MPEG2-PS专题（2）——使用FFmpeg命令生成ps文件

音视频入门基础：MPEG2-PS专题（3）——MPEG2-PS格式简介

音视频入门基础：MPEG2-PS专题（4）——FFmpeg源码中，判断某文件是否为PS文件的实现

音视频入门基础：MPEG2-PS专题（5）——FFmpeg源码中，解析PS流中的PES流的实现

音视频入门基础：MPEG2-PS专题（6）——FFmpeg源码中，获取PS流的视频信息的实现

音视频入门基础：MPEG2-PS专题（7）——通过FFprobe显示PS流每个packet的信息

=================================================================

一、引言

通过FFmpeg命令可以获取到PS文件/PS流的视频压缩编码格式、色彩格式（像素格式）、分辨率、帧率信息：

./ffmpeg -i XXX.ps

本文以H.264为例讲述FFmpeg到底是从哪个地方获取到这些视频信息的。

二、视频压缩编码格式

（一）FFmpeg获取PS流的视频压缩编码格式的原理

FFmpeg获取PS文件/PS流的视频压缩编码格式，是从PES packet的有效载荷，即ES流数据中获取的。从《音视频入门基础：MPEG2-TS专题（18）——PES流简介》可以知道，PES packet的PES packet header里面存在一个stream_id属性，指定ES流的类型和编号。但是仅根据这个stream_id属性是无法判断视频压缩编码格式的：

所以要获取视频压缩编码格式，得从PES packet的有效载荷中获取。用Elecard Stream Analyzer工具可以看到，如果PS流的视频压缩编码格式为H.264，那PES packet的有效载荷中携带的就是以0x000001或0x00000001作为起始码的AnnexB格式的H.264码流（关于AnnexB可以参考：《音视频入门基础：H.264专题（3）——EBSP, RBSP和SODB》）：

所以这时候就可以通过解析PES packet的有效载荷，即ES流来获取视频压缩编码格式。下面讲解相关代码。

（二）FFmpeg获取PS流的视频压缩编码格式的实现

由《音视频入门基础：MPEG2-PS专题（5）——FFmpeg源码中，解析PS流中的PES流的实现》可以知道，FFmpeg源码中通过mpegps_read_pes_header函数解析PS流中的一个PES packet，将其PES packet header里面的信息解析出来。而在调用完mpegps_read_pes_header函数后，指针s->pb.buf_ptr会指向该PES packet的有效载荷，如果PS流的视频压缩编码格式为H.264，那就是指向以0x000001或0x00000001作为起始码的AnnexB格式的H.264码流：

然后在mpegps_read_packet函数中，会通过av_get_packet函数将s->pb.buf_ptr指向的H.264码流数据保存到pkt->data指向的缓冲区中（关于av_get_packet函数的用法可以参考：《FFmpeg源码：append_packet_chunked、av_get_packet、av_append_packet函数分析》）：

static int mpegps_read_packet(AVFormatContext *s,AVPacket *pkt)
{
//...len = mpegps_read_pes_header(s, &dummy_pos, &startcode, &pts, &dts);
//...ret = av_get_packet(s->pb, pkt, len);
//...
}

之后在probe_codec函数中，会通过语句：memcpy(pd->buf + pd->buf_size, pkt->data, pkt->size) 将上述H.264码流数据从pkt->data拷贝到pd->buf中：

static int probe_codec(AVFormatContext *s, AVStream *st, const AVPacket *pkt)
{
//...if (sti->request_probe > 0) {//...AVProbeData *const pd = &sti->probe_data;int end;av_log(s, AV_LOG_DEBUG, "probing stream %d pp:%d\n", st->index, sti->probe_packets);--sti->probe_packets;if (pkt) {uint8_t *new_buf = av_realloc(pd->buf, pd->buf_size+pkt->size+AVPROBE_PADDING_SIZE);if (!new_buf) {av_log(s, AV_LOG_WARNING,"Failed to reallocate probe buffer for stream %d\n",st->index);goto no_packet;}pd->buf = new_buf;memcpy(pd->buf + pd->buf_size, pkt->data, pkt->size);pd->buf_size += pkt->size;memset(pd->buf + pd->buf_size, 0, AVPROBE_PADDING_SIZE);}//...}
//...
}

然后probe_codec函数中会调用set_codec_from_probe_data函数，set_codec_from_probe_data函数的定义如下：

static int set_codec_from_probe_data(AVFormatContext *s, AVStream *st,AVProbeData *pd)
{static const struct {const char *name;enum AVCodecID id;enum AVMediaType type;} fmt_id_type[] = {{ "aac",        AV_CODEC_ID_AAC,          AVMEDIA_TYPE_AUDIO    },{ "ac3",        AV_CODEC_ID_AC3,          AVMEDIA_TYPE_AUDIO    },{ "aptx",       AV_CODEC_ID_APTX,         AVMEDIA_TYPE_AUDIO    },{ "dts",        AV_CODEC_ID_DTS,          AVMEDIA_TYPE_AUDIO    },{ "dvbsub",     AV_CODEC_ID_DVB_SUBTITLE, AVMEDIA_TYPE_SUBTITLE },{ "dvbtxt",     AV_CODEC_ID_DVB_TELETEXT, AVMEDIA_TYPE_SUBTITLE },{ "eac3",       AV_CODEC_ID_EAC3,         AVMEDIA_TYPE_AUDIO    },{ "h264",       AV_CODEC_ID_H264,         AVMEDIA_TYPE_VIDEO    },{ "hevc",       AV_CODEC_ID_HEVC,         AVMEDIA_TYPE_VIDEO    },{ "loas",       AV_CODEC_ID_AAC_LATM,     AVMEDIA_TYPE_AUDIO    },{ "m4v",        AV_CODEC_ID_MPEG4,        AVMEDIA_TYPE_VIDEO    },{ "mjpeg_2000", AV_CODEC_ID_JPEG2000,     AVMEDIA_TYPE_VIDEO    },{ "mp3",        AV_CODEC_ID_MP3,          AVMEDIA_TYPE_AUDIO    },{ "mpegvideo",  AV_CODEC_ID_MPEG2VIDEO,   AVMEDIA_TYPE_VIDEO    },{ "truehd",     AV_CODEC_ID_TRUEHD,       AVMEDIA_TYPE_AUDIO    },{ "evc",        AV_CODEC_ID_EVC,          AVMEDIA_TYPE_VIDEO    },{ "vvc",        AV_CODEC_ID_VVC,          AVMEDIA_TYPE_VIDEO    },{ 0 }};int score;const AVInputFormat *fmt = av_probe_input_format3(pd, 1, &score);FFStream *const sti = ffstream(st);if (fmt) {av_log(s, AV_LOG_DEBUG,"Probe with size=%d, packets=%d detected %s with score=%d\n",pd->buf_size, s->max_probe_packets - sti->probe_packets,fmt->name, score);for (int i = 0; fmt_id_type[i].name; i++) {if (!strcmp(fmt->name, fmt_id_type[i].name)) {if (fmt_id_type[i].type != AVMEDIA_TYPE_AUDIO &&st->codecpar->sample_rate)continue;if (sti->request_probe > score &&st->codecpar->codec_id != fmt_id_type[i].id)continue;st->codecpar->codec_id   = fmt_id_type[i].id;st->codecpar->codec_type = fmt_id_type[i].type;sti->need_context_update = 1;return score;}}}return 0;
}

可以看到set_codec_from_probe_data函数中调用了av_probe_input_format3函数来推测pd->buf中的码流的格式。关于av_probe_input_format3函数的用法可以参考：《FFmpeg源码：av_probe_input_format3函数和AVInputFormat结构体分析（FFmpeg源码5.0.3版本）》。对于H.264裸流，av_probe_input_format3函数中就是调用了h264_probe函数来检测这段码流是否为AnnexB格式的H.264裸流，具体可以参考：《音视频入门基础：H.264专题（16）——FFmpeg源码中，判断某文件是否为H.264裸流文件的实现》。

判断出这段码流为H.264裸流后，set_codec_from_probe_data函数中会执行语句：st->codecpar->codec_id = fmt_id_type[i].id，让AVCodecParameters的codec_id得到视频压缩编码格式：

static int set_codec_from_probe_data(AVFormatContext *s, AVStream *st,AVProbeData *pd)
{
//...if (fmt) {
//...for (int i = 0; fmt_id_type[i].name; i++) {if (!strcmp(fmt->name, fmt_id_type[i].name)) {if (fmt_id_type[i].type != AVMEDIA_TYPE_AUDIO &&st->codecpar->sample_rate)continue;if (sti->request_probe > score &&st->codecpar->codec_id != fmt_id_type[i].id)continue;st->codecpar->codec_id   = fmt_id_type[i].id;st->codecpar->codec_type = fmt_id_type[i].type;sti->need_context_update = 1;return score;}}}return 0;
}

然后在set_codec_from_probe_data函数外部，通过avcodec_parameters_to_context函数将AVCodecParameters的codec_id赋值给AVCodecContext的codec_id：

int avcodec_parameters_to_context(AVCodecContext *codec,const AVCodecParameters *par)
{
//...codec->codec_id   = par->codec_id;
//...
}

然后在dump_stream_format函数中，通过avcodec_string函数中的语句：codec_name = avcodec_get_name(enc->codec_id) 拿到AVCodecContext的codec_id对应的视频压缩编码格式名称。最后再在dump_stream_format函数中将视频压缩编码格式打印出来：

void avcodec_string(char *buf, int buf_size, AVCodecContext *enc, int encode)
{
//...codec_name = avcodec_get_name(enc->codec_id);
//...
}

所以FFmpeg获取PS文件/PS流的视频压缩编码格式，是从PES packet的有效载荷，即ES流数据中获取的：

三、视频压缩编码格式的profile

如果PS文件/PS流中的视频压缩编码格式为H.264，FFmpeg获取其视频压缩编码格式的profile，是通过SPS的profile_idc属性获取到的，具体可以参考：《音视频入门基础：H.264专题（17）——FFmpeg源码中，获取H.264视频的profile的实现》：

四、视频的色彩格式

如果PS文件/PS流中的视频压缩编码格式为H.264，FFmpeg获取其视频的色彩格式，是通过SPS中的属性chroma_format_idc获取到的，具体可以参考：《音视频入门基础：H.264专题（13）——FFmpeg源码中通过SPS属性获取视频色彩格式的实现》：

五、视频分辨率

如果PS文件/PS流中的视频压缩编码格式为H.264，FFmpeg获取其视频分辨率，是通过SPS中的属性获取的，具体可以参考：《音视频入门基础：H.264专题（12）——FFmpeg源码中通过SPS属性计算视频分辨率的实现》：

六、视频码率

由于PS文件/PS流的文件格式（包括TS Header、PES packet header）不包含视频码率信息，所以无法通过FFmpeg直接获取到其视频码率。与之对应，由于FLV文件的Script Tag中包含视频码率信息，所以FFmpeg可以直接打印FLV文件的视频码率，具体可以参考：《音视频入门基础：FLV专题（24）——FFmpeg源码中，获取FLV文件视频信息的实现》。