ffmpeg的例子：制作屏幕录像

　　最近需要设计一个播放器，然后了解到ffmpeg这个东西，发现这东西应用还挺广泛的。

　　在这里要特别提一下CSDN的雷霄骅，关于ffmpeg的博客那是写的真的好，而且还开源了大量的资料。只不过天妒英才啊！听说因为过度劳累而猝死

　　本篇博客主要是学习雷神推荐的：如何用FFmpeg编写一个简单播放器

　　因为ffmpeg的版本升级，导致版本之间多少有些差异，我的FFmpeg版本为3.2.2，所以在移植第一个代码示例的过程中，出了不少幺蛾子。关于ffmpeg的安装网上到处都是，我就不说明了。

　　下面我上我修改后的示例代码：

　　开发环境：ubuntu14  64位

　　ffmpeg版本：3.2.2

　　测试文件：flv格式视频或其他格式视频

　　效果：得到5张视频截图

　　编译命令：gcc -g main.c -o test -I /usr/local/ffmpeg/include -L /usr/local/ffmpeg/lib -lavutil -lavformat -lavcodec -lz -lavutil -lswscale（说明：-g为加入gdb调试，/usr/local/ffmpeg为我的ffmpeg安装目录，具体目录看你的安装路径来设置）

#ifdef _cplusplus
extern "C" {
#endif

#include<stdio.h>
#include<libavcodec/avcodec.h>
#include<libavformat/avformat.h>
#include<libavutil/avutil.h>
#include<libswscale/swscale.h>
#include<libavutil/avutil.h>
#include<SDL2/SDL.h>


//******************函数声明***********************************
void SaveFrame(AVFrame *pFrame,int width,int height,int iFrame);

int main(int argc,char **argv)
{
    printf("~~~~~~~~design a player!
");
    AVFormatContext *pFormatCtx;
    av_register_all();
    avformat_network_init();
     pFormatCtx = avformat_alloc_context();
    //打开视频文件
//    if(avformat_open_input(&pFormatCtx,"bigbuckbunny_480x272.h265",NULL,NULL)!=0)
     if(avformat_open_input(&pFormatCtx,"test.flv",NULL,NULL)!=0)
        return -1;//不能打开文件
    if(avformat_find_stream_info(pFormatCtx,NULL)<0){
            printf("Couldn't find stream information.
");
            return -1;
        }
    //丢出一个信息,调试用的
//    av_dump_format(pFormatCtx,0,argv[1],0);
    int i,videoStream;
    AVCodecContext *pCodecCtx;
    //找到第一个视频流
    videoStream = -1;
    for(i=0;i<pFormatCtx->nb_streams;i++)
    {
        if(pFormatCtx->streams[i]->codec->codec_type ==AVMEDIA_TYPE_VIDEO)
        {
            videoStream = i;
            break;
        }
    }
    if(videoStream==-1){
            printf("Didn't find a video stream.
");
            return -1;
        }
    //从视频流中得到一个编解码上下文的指针,里面包含了编解码器的所有信息，包含了一个
    //指向真正的编解码的指针
    pCodecCtx = pFormatCtx->streams[videoStream]->codec;
    AVCodec *pCodec;
    //在视频流中找到这个解码器
    pCodec = avcodec_find_decoder(pCodecCtx->codec_id);
    if(pCodec == NULL)
    {
        fprintf(stderr,"Unsupported codec !
");
        return -1;
    }
    //Open codec
    if(avcodec_open2(pCodecCtx,pCodec,NULL)<0){
        printf("cann't open the codec!
");
        return -1;
    }
    //保存数据
    AVFrame *pFrame;
    //分配一个视频帧
    pFrame = av_frame_alloc();
    //分配一帧结构体
    AVFrame *pFrameRGB;
    pFrameRGB = av_frame_alloc();
    if(pFrameRGB == NULL){
        return -1;
    }
    //我们申请了一帧的内存，当转换的时候，我们需要用一个地方来放置原始数据，首先
    //我们需要使用avpicture_get_size来获得我们需要的大小，然后手工申请内存
    uint8_t *buffer;
    int numBytes;
    //确定需要的内存大小并分配内存
    numBytes = avpicture_get_size(AV_PIX_FMT_RGB24,pCodecCtx->width,pCodecCtx->height);
    buffer = (uint8_t*)av_mallocz(numBytes*sizeof(uint8_t));
    //指定buffer的适宜的一部分到pFrameRGB的图像平面
    //注意pFrameRGB是一个媒体帧，但是该媒体帧是AVPicture的父集
    avpicture_fill((AVPicture *)pFrameRGB, buffer, AV_PIX_FMT_RGB24,
    pCodecCtx->width, pCodecCtx->height);
    //**********读取数据*****************************************
    int frameFinished;
    AVPacket packet;
    struct SwsContext *pSwsCtx;
    i=0;
    pSwsCtx = sws_getContext(pCodecCtx->width,
            pCodecCtx->height,
            pCodecCtx->pix_fmt,
            pCodecCtx->width,
            pCodecCtx->height,
            AV_PIX_FMT_RGB24,
            SWS_BICUBIC,
            NULL,NULL,NULL
            );
     if(pSwsCtx == NULL)
       {
            fprintf(stderr, "Cannot initialize the conversion context!
");
            return -1;
        }
    while(av_read_frame(pFormatCtx,&packet) >= 0)
    {
            //如果读取的包来自视频流
            if(packet.stream_index == videoStream)
            {
                //解码帧数据
                avcodec_decode_video2(pCodecCtx,pFrame,&frameFinished,&packet);
                //我们是否得到一个视频帧
                if(frameFinished){
                    //使用RGB格式来将它转换为图片
                     sws_scale(pSwsCtx,
                                      pFrame->data, pFrame->linesize, 0, pCodecCtx->height,
                                      pFrameRGB->data, pFrameRGB->linesize);
                     //保存这个帧数据到磁盘
                     if(++i<=5)
                         SaveFrame(pFrameRGB,pCodecCtx->width,pCodecCtx->height,i);
                }
            }
            av_free_packet(&packet);
    }
    //清理操作
    av_free(buffer);
    av_free(pFrameRGB);
    av_free(pFrame);
    avcodec_close(pCodecCtx);
    avformat_close_input(&pFormatCtx);
    return EXIT_SUCCESS;
}

//*****************函数定义***********************************
void SaveFrame(AVFrame *pFrame,int width,int height,int iFrame)
{
    FILE    *pFile;
    char szFilename[32];
    int y;

    //打开文件
    sprintf(szFilename,"frame%d.ppm",iFrame);
    pFile = fopen(szFilename,"wb");
    if(pFile == NULL)
        return;
    //写头
    fprintf(pFile,"P6
%d %d
255
",width,height);

    //写 pixel 数据
    for(y=0; y<height; y++)
        fwrite(pFrame->data[0]+y*pFrame->linesize[0], width*3, 1, pFile);
    //关闭文件
    fclose(pFile);
}

#ifdef _cplusplus
}
#endif