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!\n");
    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.\n");
            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.\n");
            return -1;
        }
    //從視頻流中得到一個編解碼上下文的指針,里面包含了編解碼器的所有信息，包含了一個
    //指向真正的編解碼的指針
    pCodecCtx = pFormatCtx->streams[videoStream]->codec;
    AVCodec *pCodec;
    //在視頻流中找到這個解碼器
    pCodec = avcodec_find_decoder(pCodecCtx->codec_id);
    if(pCodec == NULL)
    {
        fprintf(stderr,"Unsupported codec !\n");
        return -1;
    }
    //Open codec
    if(avcodec_open2(pCodecCtx,pCodec,NULL)<0){
        printf("cann't open the codec!\n");
        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!\n");
            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\n%d %d\n255\n",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