ffmpeg簡單視頻播放器

參考學習雷霄驊的最簡單的基於FFMPEG視頻播放器。

在移植代碼到我的linux環境中時總出現問題，所以干脆把雷神的代碼從頭到尾分析並移植過來調試。

開發環境：

　　操作系統：ubuntu14

　　ffmpeg版本：3.2.2

　　sdl版本：2

代碼我是自己參照雷神的代碼敲的，在這過程中，我發現寫代碼還是要加打印，沒有什么能一拿過來就可以成功運行。尤其是在自己設計接口的時候，返回值最好是返回運行結果，需要傳出的內存通過參數來傳出。

對於錯誤的判斷打印還是需要詳細一些，以后在寫比較大的程序時要養成加log的習慣，這對調試程序還是有很大幫助的。

下面插入代碼：

說明：該程序運行一段時間后會變成黑白，具體原因再查

編譯與運行：

gcc -g main.c -o test -I /usr/local/ffmpeg/include -L /usr/local/ffmpeg/lib -lavutil -lavformat -lavcodec -lz -lavutil -lswscale -L /usr/lib/x86_64-linux-gnu -lSDL2 -lSDL2main
（其中路徑按照自己的安裝路徑來 -g 是加GDB調試）

./test

#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<libavutil/imgutils.h>
#include<SDL2/SDL.h>

//是否將YUV420P內容輸出到文件
#define OUTPUT_YUV420P 0
//要播放的文件路徑
#define filename "/home/sns/cuc_ieschool.flv"
//要輸出YUV420P內容的文件路徑
#define outfilename "/home/sns/output.yuv"

int main(int argc, char **argv)
{
    //變量定義*********************************************************************
    AVFormatContext *pFormatCtx;
    int i, videoStream;
    AVCodecContext *pCodecCtx;
    AVCodec *pCodec;
    AVFrame *pFrame;
    AVFrame *pFrameYUV;
    uint8_t *buffer;
    int numBytes;

    SDL_Window *screen;
    SDL_Renderer *sdlRender;
    SDL_Texture *sdlTexture;
    SDL_Rect sdlRect;
    int frameFinished;
    AVPacket packet;
    i = 0;
    struct SwsContext *img_convert_ctx;
    int err_code;
    char buf[1024];
    FILE *fp_yuv;
    int y_size;
    //*******************************************************************************
    av_register_all();
    //1、打開視頻文件*************************************************
    pFormatCtx = avformat_alloc_context();
    err_code = avformat_open_input(&pFormatCtx, filename, NULL, NULL);
    if (err_code != 0)
    {//打開文件失敗
        av_strerror(err_code, buf, 1024);
        printf("coundn't open the file!,error code = %d(%s)\n", err_code, buf);
        return -1;
    }
    if (avformat_find_stream_info(pFormatCtx, NULL) < 0)
    {
        printf("Couldn't find stream information.\n");
        return -1;
    }
    //2、找到第一個視頻流****************************
    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;
    }
    /* 3、從視頻流中得到一個編解碼上下文,里面包含了編解碼器的所有信息和一個
    指向真正的編解碼器     ,然后我們找到這個解碼器*/
    pCodecCtx = pFormatCtx->streams[videoStream]->codec;
    pCodec = avcodec_find_decoder(pCodecCtx->codec_id);
    if (pCodec == NULL)
    {
        fprintf(stderr, "Unsupported codec !\n");
        return -1;
    }
    //4、打開該編解碼器
    if (avcodec_open2(pCodecCtx, pCodec, NULL) < 0)
    {
        printf("cann't open the codec!\n");
        return -1;
    }
    //5、分配兩個視頻幀，一個保存得到的原始視頻幀，一個保存為指定格式的視頻幀（該幀通過原始幀轉換得來）
    pFrame = av_frame_alloc();
    if (pFrame == NULL)
    {
        printf("pFrame alloc fail!\n");
        return -1;
    }
    pFrameYUV = av_frame_alloc();
    if (pFrameYUV == NULL)
    {
        printf("pFrameYUV alloc fail!\n");
        return -1;
    }
    //6、得到一幀視頻截圖的內存大小並分配內存，並將YUV數據填充進去
    numBytes = av_image_get_buffer_size(AV_PIX_FMT_YUV420P, pCodecCtx->width,
            pCodecCtx->height,1);
    buffer = (uint8_t*) av_mallocz(numBytes * sizeof(uint8_t));
    if (!buffer)
    {
        printf("numBytes :%d , buffer malloc 's mem \n", numBytes);
        return -1;
    }
    //打印信息
    printf("--------------- File Information ----------------\n");
    av_dump_format(pFormatCtx, 0, filename, 0);
    printf("-------------------------------------------------\n");
    av_image_fill_arrays(pFrameYUV->data, pFrameYUV->linesize,buffer,
            AV_PIX_FMT_YUV420P,pCodecCtx->width, pCodecCtx->height,1);
    //7、得到指定轉換格式的上下文**********************************
    img_convert_ctx = sws_getContext(pCodecCtx->width, pCodecCtx->height,
            pCodecCtx->pix_fmt, pCodecCtx->width, pCodecCtx->height,
            AV_PIX_FMT_YUV420P,
            SWS_BICUBIC,
            NULL, NULL, NULL);
    if (img_convert_ctx == NULL)
    {
        fprintf(stderr, "Cannot initialize the conversion context!\n");
        return -1;
    }
    //***********************************************************
#if OUTPUT_YUV420P
    fp_yuv = fopen(outfilename, "wb+");
#endif
    //8、SDL初始化和創建多重windows等准備工作
    if (SDL_Init(SDL_INIT_AUDIO | SDL_INIT_TIMER | SDL_INIT_VIDEO))
    {
        fprintf(stderr, "Could not initialize SDL - %s\n", SDL_GetError());
        return -1;
    }
    //使用SDL_CreateWindow代替SDL_SetVideoMode
    //創建一個給定高度和寬度、位置和標示的windows。
    screen = SDL_CreateWindow("Simplest ffmpeg player's Window",
    SDL_WINDOWPOS_UNDEFINED, SDL_WINDOWPOS_UNDEFINED, pCodecCtx->width,
            pCodecCtx->height, SDL_WINDOW_OPENGL);
    if (!screen)
    {
        fprintf(stderr, "SDL: could not create window - exiting - %s\n",SDL_GetError());
        return -1;
    }
    //對該window創建一個2D渲染上下文
    sdlRender = SDL_CreateRenderer(screen, -1, 0);
    if (!sdlRender)
    {
        fprintf(stderr, "SDL:cound not create render :    %s\n", SDL_GetError());
        return -1;
    }
    //Create a texture for a rendering context.
    //為一個渲染上下文創建一個紋理
    //IYUV: Y + U + V  (3 planes)
    //YV12: Y + V + U  (3 planes)
    sdlTexture = SDL_CreateTexture(sdlRender, SDL_PIXELFORMAT_IYUV,
            SDL_TEXTUREACCESS_STREAMING, pCodecCtx->width, pCodecCtx->height);
    if (!sdlTexture)
    {
        fprintf(stderr, "SDL:cound not create Texture :    %s\n", SDL_GetError());
        return -1;
    }
    //建立一個矩形變量，提供后面使用
    sdlRect.x = 0;
    sdlRect.y = 0;
    sdlRect.w = pCodecCtx->width;
    sdlRect.h = pCodecCtx->height;
    //9、正式開始讀取數據*****************************************
    while (av_read_frame(pFormatCtx, &packet) >= 0)
    {
        //如果讀取的包來自視頻流
        if (packet.stream_index == videoStream)
        {
            //從包中得到解碼后的幀
            if (avcodec_decode_video2(pCodecCtx, pFrame, &frameFinished,&packet) < 0)
            {
                printf("Decode Error!\n");
                return -1;
            }
            //如果確定完成得到該視頻幀
            if (frameFinished)
            {
                //轉換幀數據格式
                sws_scale(img_convert_ctx, pFrame->data, pFrame->linesize, 0,
                        pCodecCtx->height,
                        pFrameYUV->data,
                        pFrameYUV->linesize);
#if OUTPUT_YUV420P
                y_size = pCodecCtx->width * pCodecCtx->height;
                fwrite(pFrameYUV->data[0], 1, y_size, fp_yuv);    //Y
                fwrite(pFrameYUV->data[1], 1, y_size / 4, fp_yuv);  //U
                fwrite(pFrameYUV->data[2], 1, y_size / 4, fp_yuv);  //V
#endif
                //SDL顯示~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
#if 0
                SDL_UpdateTexture(sdlTexture,NULL,pFrameYUV->data[0],pFrameYUV->linesize[0]);
#else
                SDL_UpdateYUVTexture(sdlTexture, &sdlRect, pFrameYUV->data[0],
                        pFrameYUV->linesize[0], pFrameYUV->data[1],
                        pFrameYUV->linesize[1], pFrameYUV->data[2],
                        pFrameYUV->linesize[2]);
#endif
                SDL_RenderClear(sdlRender);
                SDL_RenderCopy(sdlRender, sdlTexture, NULL, &sdlRect);
                SDL_RenderPresent(sdlRender);
                //結束SDL~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
                SDL_Delay(40);//延時
            }
        }
        av_free_packet(&packet);//釋放讀出來的包
    }
    //**************************************************************************************
    //10、釋放分配的內存或關閉文件等操作
#if OUTPUT_YUV420P
    fclose(fp_yuv);
#endif
    sws_freeContext(img_convert_ctx);
    SDL_Quit();
    av_free(buffer);
    av_free(pFrame);
    av_free(pFrameYUV);
    avcodec_close(pCodecCtx);
    avformat_close_input(&pFormatCtx);
    return EXIT_SUCCESS;
}

#ifdef _cplusplus
}
#endif