ffmpeg实现将视频存储为图片jpg
============================================================================
ffmpeg视频H264压缩，rtsp推流课程教学视频：
https://edu.csdn.net/course/detail/27795
课件中里面提供源码可以直接下载运行！
============================================================================
一、 添加库 和 lib 以及头文件
在.pro中添加
INCLUDEPATH += $$PWD/ffmpeg/include
LIBS += $$PWD/ffmpeg/lib/avcodec.lib \
$$PWD/ffmpeg/lib/avdevice.lib\
$$PWD/ffmpeg/lib/avfilter.lib\
$$PWD/ffmpeg/lib/avformat.lib\
$$PWD/ffmpeg/lib/avutil.lib\
$$PWD/ffmpeg/lib/postproc.lib\
$$PWD/ffmpeg/lib/swresample.lib\
$$PWD/ffmpeg/lib/swscale.lib在main.c中添加
#include
#include
#include
using namespace std;
extern "C"
{
#include "libavcodec/avcodec.h"
#include "libavformat/avformat.h"
#include "libswscale/swscale.h"
#include "libavdevice/avdevice.h"
#include "libavutil/pixfmt.h"
} 二、FFMPEG初始化操作
av_register_all(); //初始化FFMPEG 调用了这个才能正常适用编码器和解码器
//=========================== 创建AVFormatContext结构体 ===============================//
//分配一个AVFormatContext，FFMPEG所有的操作都要通过这个AVFormatContext来进行
AVFormatContext *pFormatCtx = avformat_alloc_context();
//==================================== 打开文件 ======================================//
char *file_path = "E:/QT/test_ffmpegSavePic/ffmpeg/test.avi";//这里必须使用左斜杠
int ret = avformat_open_input(&pFormatCtx, file_path, NULL, NULL);
if(ret != 0)
{
cout << "open error!" << endl;
return -1;
}
//循环查找视频中包含的流信息，直到找到视频类型的流
//便将其记录下来 保存到videoStream变量中
int i;
int videoStream;
//=================================== 获取视频流信息 ===================================//
if (avformat_find_stream_info(pFormatCtx, NULL) < 0)
{
cout << "Could't find stream infomation." << endl;
return -1;
}
videoStream = -1;
for (i = 0; i < pFormatCtx->nb_streams; i++)
{
if (pFormatCtx->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO)
{
videoStream = i;
}
}
//如果videoStream为-1 说明没有找到视频流
if (videoStream == -1)
{
cout << "Didn't find a video stream." << endl;
return -1;
}
//================================= 查找解码器 ===================================//
AVCodecContext* pCodecCtx = pFormatCtx->streams[videoStream]->codec;
AVCodec* pCodec = avcodec_find_decoder(pCodecCtx->codec_id);
if (pCodec == NULL)
{
cout << "Codec not found." << endl;
return -1;
}
//================================ 打开解码器 ===================================//
if (avcodec_open2(pCodecCtx, pCodec, NULL) < 0)// 具体采用什么解码器ffmpeg经过封装 我们无须知道
{
cout << "Could not open codec." << endl;
return -1;
}
//================================ 设置数据转换参数 ================================//
SwsContext * img_convert_ctx;
img_convert_ctx = sws_getContext(pCodecCtx->width, pCodecCtx->height, pCodecCtx->pix_fmt, //源地址长宽以及数据格式
pCodecCtx->width, pCodecCtx->height,AV_PIX_FMT_YUVJ420P, //目的地址长宽以及数据格式
SWS_BICUBIC, NULL, NULL, NULL);//算法类型 AV_PIX_FMT_YUVJ420P AV_PIX_FMT_BGR24
//==================================== 分配空间 ==================================//
//一帧图像数据大小
int numBytes = avpicture_get_size(AV_PIX_FMT_YUVJ420P, pCodecCtx->width,pCodecCtx->height);
unsigned char *out_buffer;
out_buffer = (unsigned char *) av_malloc(numBytes * sizeof(unsigned char));
AVFrame * pFrame;
pFrame = av_frame_alloc();
AVFrame * pFrameRGB;
pFrameRGB = av_frame_alloc();
avpicture_fill((AVPicture *) pFrameRGB, out_buffer, AV_PIX_FMT_YUVJ420P,pCodecCtx->width, pCodecCtx->height);
//会将pFrameRGB的数据按RGB格式自动"关联"到buffer 即pFrameRGB中的数据改变了 out_buffer中的数据也会相应的改变
//=========================== 分配AVPacket结构体 ===============================//
int y_size = pCodecCtx->width * pCodecCtx->height;
AVPacket *packet = (AVPacket *) malloc(sizeof(AVPacket)); //分配一个packet
av_new_packet(packet, y_size); //分配packet的数据
三、循环采集视频流数据 将其转换为图片
//=========================== 读取视频信息 ===============================//
if (av_read_frame(pFormatCtx, packet) < 0) //读取的是一帧视频 数据存入一个AVPacket的结构中
{
cout << "read error." << endl;
return -1;
}
//此时数据存储在packet中
//视频解码函数 解码之后的数据存储在 pFrame中
ret = avcodec_decode_video2(pCodecCtx, pFrame, &got_picture, packet);
if (ret < 0)
{
cout << "decode error." << endl;
return -1;
}
//转换一帧图像
sws_scale(img_convert_ctx,pFrame->data, pFrame->linesize, 0, pCodecCtx->height, //源
pFrameRGB->data, pFrameRGB->linesize); //目的
SaveAsJPEG(pFrameRGB, pCodecCtx->width,pCodecCtx->height,index++); //保存图片
四、 实现保存jpg图片函数
/*==================================================================================
* 将AVFrame(YUV420格式)保存为JPEG格式的图片
===================================================================================*/
int SaveAsJPEG(AVFrame* pFrame, int width, int height, int index)
{
// 输出文件路径
char out_file[MAX_PATH] = {0};
sprintf_s(out_file, sizeof(out_file), "%s%d.jpg", "E:/QT/test_ffmpegSavePic/ffmpeg/output/", index);
// 分配AVFormatContext对象
AVFormatContext* pFormatCtx = avformat_alloc_context();
// 设置输出文件格式
pFormatCtx->oformat = av_guess_format("mjpeg", NULL, NULL);
// 创建并初始化一个和该url相关的AVIOContext
if( avio_open(&pFormatCtx->pb, out_file, AVIO_FLAG_READ_WRITE) < 0)
{
printf("Couldn't open output file.");
return -1;
}
// 构建一个新stream
AVStream* pAVStream = avformat_new_stream(pFormatCtx, 0);
if( pAVStream == NULL )
{
return -1;
}
// 设置该stream的信息
AVCodecContext* pCodecCtx = pAVStream->codec;
pCodecCtx->codec_id = pFormatCtx->oformat->video_codec;
pCodecCtx->codec_type = AVMEDIA_TYPE_VIDEO;
pCodecCtx->pix_fmt = AV_PIX_FMT_YUVJ420P;
pCodecCtx->width = width;
pCodecCtx->height = height;
pCodecCtx->time_base.num = 1;
pCodecCtx->time_base.den = 25;
//打印输出相关信息
av_dump_format(pFormatCtx, 0, out_file, 1);
//================================== 查找编码器 ==================================//
AVCodec* pCodec = avcodec_find_encoder(pCodecCtx->codec_id);
if( !pCodec )
{
printf("Codec not found.");
return -1;
}
// 设置pCodecCtx的解码器为pCodec
if( avcodec_open2(pCodecCtx, pCodec, NULL) < 0 )
{
printf("Could not open codec.");
return -1;
}
//================================Write Header ===============================//
avformat_write_header(pFormatCtx, NULL);
int y_size = pCodecCtx->width * pCodecCtx->height;
//==================================== 编码 ==================================//
// 给AVPacket分配足够大的空间
AVPacket pkt;
av_new_packet(&pkt, y_size * 3);
//
int got_picture = 0;
int ret = avcodec_encode_video2(pCodecCtx, &pkt, pFrame, &got_picture);
if( ret < 0 )
{
printf("Encode Error.\n");
return -1;
}
if( got_picture == 1 )
{
pkt.stream_index = pAVStream->index;
ret = av_write_frame(pFormatCtx, &pkt);
}
av_free_packet(&pkt);
//Write Trailer
av_write_trailer(pFormatCtx);
if( pAVStream )
{
avcodec_close(pAVStream->codec);
}
avio_close(pFormatCtx->pb);
avformat_free_context(pFormatCtx);
return 0;
}
============================================================================
ffmpeg视频H264压缩，rtsp推流课程教学视频：
https://edu.csdn.net/course/detail/27795
课件中里面提供源码可以直接下载运行！
============================================================================
=======================================================================
最近新开的公众号，文章正在一篇篇的更新，
公众号名称：玩转电子世界
各位朋友有什么问题了可以直接在上面提问，我会一一进行解答的。
跟着阳光非宅男，一步步走进电子的世界。
关注之后回复 资料下载 关键词可以获得免费海量的视频学习资料下载~~！
已共享的学习视频资料，共享资料正在不断更新中。
共享ffmpeg视频学习资料：
=======================================================================