首页 > 代码库 > FFMPEG系列课程(二)读取视频帧

FFMPEG系列课程(二)读取视频帧

再来介绍下如何读取视频帧,打开视频参考前面的文章,首先需要创建一个帧的存放对象

AVPacket pkt;

memset(&pkt, 0, sizeof(AVPacket)); 

     再通过 int err = av_read_frame(ic, &pkt); 读取帧数据,其中ic是之前打开的视频流句柄。读取视频帧后要注意一点av_read_frame会自动给视频帧分配空间,返回值0表示成功。

    可以通过 av_packet_unref(&pkt);删除packet的控制,注意这里删除的不是packet对象的空间,而是packet对象指向的数据空间。

   通过判断pkt的stream_index 成员可以判断此帧数据是不是视频

    if (pkt.stream_index == videoStream),

    videoStream是打开视频时存放的视频流索引,大部分文件都是0,参考前一篇文章。

  1. extern "C"  

  2. {  

  3. #include<libavformat/avformat.h>  

  4. }  

  5. //引用lib库,也可以在项目中设置,打开视频只需要用到这三个库  

  6. #pragma comment(lib,"avformat.lib")  

  7. #pragma comment(lib,"avutil.lib")  

  8. #pragma comment(lib,"avcodec.lib")  

  9. #include <iostream>  

  10. using namespace std;  

  11. int main(int argc, char *argv[])  

  12. {  

  13.     //初始化所以ffmpeg的解码器  

  14.     av_register_all();  

  15.   

  16.   

  17.     char path[1024] = "video.mp4";  

  18.     //用来存放打开的视频流信息  

  19.     AVFormatContext *ic = NULL;  

  20.   

  21.   

  22.     //用来存储视频流索引  

  23.     int videoStream = 0;  

  24.   

  25.   

  26.     //打开视频播放流  

  27.     //path参数表示打开的视频路径,这个路径可以包括各种视频文件  

  28.     //也包括rtsp和http网络视频流  

  29.     //第三个参数表示传入的视频格式,我这边不传递有FFmpeg内部获取  

  30.     //最后一个参数是设置,我们这里也不传递  

  31.     int re = avformat_open_input(&ic, path, 0, 0);  

  32.     if (re != 0)  

  33.     {  

  34.         //获取到FFmpeg的错误信息  

  35.         char errorbuf[1024] = { 0 };  

  36.         av_strerror(re, errorbuf, sizeof(errorbuf));  

  37.         printf("open %s failed: %s\n", path, errorbuf);  

  38.         return -1;  

  39.     }  

  40.   

  41.   

  42.     //遍历视频流,里面包含音频流,视频流,或者字母流,我们这里只处理视频  

  43.     for (int i = 0; i < ic->nb_streams; i++)  

  44.     {  

  45.         AVCodecContext *enc = ic->streams[i]->codec;  

  46.         //确认是视频流  

  47.         if (enc->codec_type == AVMEDIA_TYPE_VIDEO)  

  48.         {  

  49.             //存放视频流索引,后面的代码要用到  

  50.             videoStream = i;  

  51.   

  52.   

  53.             //找到解码器,比如H264,解码器的信息也是ffmpeg内部获取的  

  54.             AVCodec *codec = avcodec_find_decoder(enc->codec_id);  

  55.             if (!codec)  

  56.             {  

  57.                 printf("video code not find!\n");  

  58.                 return -2;  

  59.             }  

  60.   

  61.   

  62.             //打开视频解码器,打开音频解码器用的也是同一个函数  

  63.             int err = avcodec_open2(enc, codec, NULL);  

  64.             if (err != 0)  

  65.             {  

  66.                 char buf[1024] = { 0 };  

  67.                 av_strerror(err, buf, sizeof(buf));  

  68.                 printf(buf);  

  69.                 return -3;  

  70.             }  

  71.         }  

  72.     }  

  73.   

  74.   

  75.     //创建视频帧对象  

  76.     AVPacket pkt;  

  77.     memset(&pkt, 0, sizeof(AVPacket));  

  78.   

  79.   

  80.     for (;;)  

  81.     {  

  82.         //读取视频帧  

  83.         int err = av_read_frame(ic, &pkt);  

  84.         if (err != 0)  

  85.         {  

  86.             //读到结尾或者出错  

  87.             char buf[1024] = { 0 };  

  88.             av_strerror(err, buf, sizeof(buf));  

  89.             printf(buf);  

  90.             break;  

  91.         }  

  92.   

  93.   

  94.         //判断是否是音频  

  95.         if (pkt.stream_index == videoStream)  

  96.         {  

  97.             printf("video pkt size is %d\n", pkt.size);  

  98.             //视频处理,解码或者存储  

  99.         }  

  100.   

  101.   

  102.         //视频帧数据空间  

  103.         av_packet_unref(&pkt);  

  104.     }  

  105. }  

更多的资料也可以关注我51CTO上的视频课程

夏老师的课堂 http://edu.51cto.com/lecturer/index/user_id-12016059.html

手把手教您开发视频播放器

http://edu.51cto.com/course/course_id-8059.html


技术分享


本文出自 “夏曹俊” 博客,谢绝转载!

FFMPEG系列课程(二)读取视频帧