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)
", err_code, buf);
        return -1;
    }
    if (avformat_find_stream_info(pFormatCtx, NULL) < 0)
    {
        printf("Couldn't find stream information.
");
        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.
");
        return -1;
    }
    /* 3、从视频流中得到一个编解码上下文,里面包含了编解码器的所有信息和一个
    指向真正的编解码器     ,然后我们找到这个解码器*/
    pCodecCtx = pFormatCtx->streams[videoStream]->codec;
    pCodec = avcodec_find_decoder(pCodecCtx->codec_id);
    if (pCodec == NULL)
    {
        fprintf(stderr, "Unsupported codec !
");
        return -1;
    }
    //4、打开该编解码器
    if (avcodec_open2(pCodecCtx, pCodec, NULL) < 0)
    {
        printf("cann't open the codec!
");
        return -1;
    }
    //5、分配两个视频帧，一个保存得到的原始视频帧，一个保存为指定格式的视频帧（该帧通过原始帧转换得来）
    pFrame = av_frame_alloc();
    if (pFrame == NULL)
    {
        printf("pFrame alloc fail!
");
        return -1;
    }
    pFrameYUV = av_frame_alloc();
    if (pFrameYUV == NULL)
    {
        printf("pFrameYUV alloc fail!
");
        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 
", numBytes);
        return -1;
    }
    //打印信息
    printf("--------------- File Information ----------------
");
    av_dump_format(pFormatCtx, 0, filename, 0);
    printf("-------------------------------------------------
");
    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!
");
        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
", 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
",SDL_GetError());
        return -1;
    }
    //对该window创建一个2D渲染上下文
    sdlRender = SDL_CreateRenderer(screen, -1, 0);
    if (!sdlRender)
    {
        fprintf(stderr, "SDL:cound not create render :    %s
", 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
", 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!
");
                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