FFMPEG系列课程读取视频帧

Posted 2020-08-25

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

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

本文出自 “夏曹俊” 博客，谢绝转载！