（转载）ffmpe---实现将视频存储为图片jpg

原文出自：https://mp.weixin.qq.com/s/z1LXwRVPlDqxn_p3DIFOag

#include <iostream>
#include <stdio.h>
#include <Windows.h>

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"
}
/*==================================================================================
*                  将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);
    sprintf_s(out_file, sizeof(out_file), "%s%d.jpg", "../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 (pFrame){
    //    av_frame_unref(pFrame);
    //}

    if (pAVStream)
    {
        avcodec_close(pAVStream->codec);
    }
    avio_close(pFormatCtx->pb);
    avformat_free_context(pFormatCtx);

    return 0;
}
//没有完全源代码，自己修改过的
void FFmpegDemo::on_pictureBtn2_clicked()
{
    //1.FFMPEG初始化操作
    av_register_all(); //初始化FFMPEG  调用了这个才能正常适用编码器和解码器
    //=========================== 创建AVFormatContext结构体 ===============================//
    //分配一个AVFormatContext，FFMPEG所有的操作都要通过这个AVFormatContext来进行
    AVFormatContext *pFormatCtx = avformat_alloc_context();
    //==================================== 打开文件 ======================================//
    char *file_path = "./2.mp4";//这里必须使用左斜杠
    int ret = avformat_open_input(&pFormatCtx, file_path, NULL, NULL);
    if (ret != 0)
    {
        cout << "open error!" << endl;
        return ;
    }

    //循环查找视频中包含的流信息，直到找到视频类型的流
    //便将其记录下来 保存到videoStream变量中
    int i;
    int videoStream;

    //=================================== 获取视频流信息 ===================================//
    if (avformat_find_stream_info(pFormatCtx, NULL) < 0)
    {
        cout << "Could't find stream infomation." << endl;
        return ;
    }

    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 ;
    }

    //=================================  查找解码器 ===================================//
    AVCodecContext* pCodecCtx = pFormatCtx->streams[videoStream]->codec;
    AVCodec* pCodec = avcodec_find_decoder(pCodecCtx->codec_id);
    if (pCodec == NULL)
    {
        cout << "Codec not found." << endl;
        return ;
    }

    //================================  打开解码器 ===================================//
    if (avcodec_open2(pCodecCtx, pCodec, NULL) < 0)// 具体采用什么解码器ffmpeg经过封装  我们无须知道
    {
        cout << "Could not open codec." << endl;
        return ;
    }

    //================================ 设置数据转换参数 ================================//
    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的数据

    //2.循环采集视频流数据  将其转换为图片

    //===========================  读取视频信息 ===============================//
    int index = 0;
    //读取的是一帧视频  数据存入一个AVPacket的结构中  
    while (av_read_frame(pFormatCtx, packet) >= 0)
    {
        //此时数据存储在packet中
        if (packet && packet->stream_index == videoStream)
        {
            //视频解码函数  解码之后的数据存储在 pFrame中
            int got_picture = 0;
            ret = avcodec_decode_video2(pCodecCtx, pFrame, &got_picture, packet);
            if (ret < 0)
            {
                cout << "decode error." << endl;
                return;
            }


            //转换一帧图像
            sws_scale(img_convert_ctx, pFrame->data, pFrame->linesize, 0,
                pCodecCtx->height,  //源
                pFrameRGB->data, pFrameRGB->linesize); //目的

            SaveAsJPEG(pFrameRGB, pCodecCtx->width, pCodecCtx->height, index++); //保存图片
        }

    }
    free(packet);
    av_frame_free(&pFrameRGB);
    av_frame_free(&pFrame);
    avcodec_close(pCodecCtx);
    avformat_free_context(pFormatCtx);


}