ffmpeg获取RTSP视频流信息

#include "FFPlayVeido.h"

 

FFPlayVedio::FFPlayVedio(HWND hwd)

{

    char sdl_var[64];

    sprintf(sdl_var, "SDL_WINDOWID=%d", hwd); //主窗口句柄   

    putenv(sdl_var);

    //初始化

    av_register_all();

    avformat_network_init();

 

    if (SDL_Init(SDL_INIT_VIDEO | SDL_INIT_AUDIO | SDL_INIT_TIMER)) {

        printf("Could not initialize SDL - %s\n", SDL_GetError());

        return;

    }

}

FFPlayVedio::~FFPlayVedio()

{

 

}

 

void FFPlayVedio::OnInit(HWND hwd)

{

 

}

void FFPlayVedio::GetVedioPlay( char *name)

{

 

    AVFormatContext *pFormatCtx;

    int             i, videoindex;

    AVCodecContext  *pCodecCtx;

    AVCodec         *pCodec;

    char filepath[] = "rtsp://admin:123456@192.168.1.252:554/mpeg4cif";

 

    pFormatCtx = avformat_alloc_context();

 

    if (avformat_open_input(&pFormatCtx, filepath, NULL, NULL) != 0){

        printf("Couldn't open input stream.（无法打开输入流）\n");

        return ;

    }

    if (av_find_stream_info(pFormatCtx) < 0)

    {

        printf("Couldn't find stream information.（无法获取流信息）\n");

        return ;

    }

    videoindex = -1;

    for (i = 0; i < pFormatCtx->nb_streams; i )

    if (pFormatCtx->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO)

    {

        videoindex = i;

        break;

    }

    if (videoindex == -1)

    {

        printf("Didn't find a video stream.（没有找到视频流）\n");

        return ;

    }

    pCodecCtx = pFormatCtx->streams[videoindex]->codec;

    pCodec = avcodec_find_decoder(pCodecCtx->codec_id);

    if (pCodec == NULL)

    {

        printf("Codec not found.（没有找到解码器）\n");

        return ;

    }

    if (avcodec_open2(pCodecCtx, pCodec, NULL) < 0)

    {

        printf("Could not open codec.（无法打开解码器）\n");

        return ;

    }

    AVFrame *pFrame, *pFrameYUV;

    pFrame = avcodec_alloc_frame();

    pFrameYUV = avcodec_alloc_frame();

    uint8_t *out_buffer = (uint8_t *)av_malloc(avpicture_get_size(PIX_FMT_YUV420P, pCodecCtx->width, pCodecCtx->height));

    avpicture_fill((AVPicture *)pFrameYUV, out_buffer, PIX_FMT_YUV420P, pCodecCtx->width, pCodecCtx->height);

    //SDL---------------------------

 

    int screen_w = 0, screen_h = 0;

    //SDL 2.0 Support for multiple windows

    SDL_Window *screen;

    screen_w = pCodecCtx->width;

    screen_h = pCodecCtx->height;

    screen = SDL_CreateWindow("Simplest ffmpeg player's Window", SDL_WINDOWPOS_UNDEFINED, SDL_WINDOWPOS_UNDEFINED,

        screen_w, screen_h,

        SDL_WINDOW_OPENGL);

 

    if (!screen) {

        printf("SDL: could not create window - exiting:%s\n", SDL_GetError());

        return ;

    }

 

    SDL_Renderer* sdlRenderer = SDL_CreateRenderer(screen, -1, 0);

    //IYUV: Y   U   V  (3 planes)

    //YV12: Y   V   U  (3 planes)

    SDL_Texture* sdlTexture = SDL_CreateTexture(sdlRenderer, SDL_PIXELFORMAT_IYUV, SDL_TEXTUREACCESS_STREAMING, pCodecCtx->width, pCodecCtx->height);

 

    SDL_Rect sdlRect;

    sdlRect.x = 0;

    sdlRect.y = 0;

    sdlRect.w = screen_w;

    sdlRect.h = screen_h;

    //SDL End----------------------

    int ret, got_picture;

    AVPacket *packet = (AVPacket *)av_malloc(sizeof(AVPacket));

    //Output Info-----------------------------

    printf("File Information（文件信息）---------------------\n");

    av_dump_format(pFormatCtx, 0, filepath, 0);

    printf("-------------------------------------------------\n");

 

#if OUTPUT_YUV420P 

    FILE *fp_yuv = fopen("output.yuv", "wb ");

#endif  

 

    struct SwsContext *img_convert_ctx;

    img_convert_ctx = sws_getContext(pCodecCtx->width, pCodecCtx->height, pCodecCtx->pix_fmt, pCodecCtx->width, pCodecCtx->height, PIX_FMT_YUV420P, SWS_BICUBIC, NULL, NULL, NULL);

    //------------------------------

    while (av_read_frame(pFormatCtx, packet) >= 0)

    {

        if (packet->stream_index == videoindex)

        {

            ret = avcodec_decode_video2(pCodecCtx, pFrame, &got_picture, packet);

            if (ret < 0)

            {

                printf("Decode Error.（解码错误）\n");

                return ;

            }

            if (got_picture)

            {

                sws_scale(img_convert_ctx, (const uint8_t* const*)pFrame->data, pFrame->linesize, 0, pCodecCtx->height, pFrameYUV->data, pFrameYUV->linesize);

 

#if OUTPUT_YUV420P

                int 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---------------------------

                SDL_UpdateTexture(sdlTexture, &sdlRect, pFrameYUV->data[0], pFrameYUV->linesize[0]);

                SDL_RenderClear(sdlRenderer);

                SDL_RenderCopy(sdlRenderer, sdlTexture, &sdlRect, &sdlRect);

                SDL_RenderPresent(sdlRenderer);

                //SDL End-----------------------

                //Delay 40ms

                SDL_Delay(40);

            }

        }

        av_free_packet(packet);

    }

    sws_freeContext(img_convert_ctx);

 

#if OUTPUT_YUV420P 

    fclose(fp_yuv);

#endif 

 

    SDL_Quit();

 

    av_free(out_buffer);

    av_free(pFrameYUV);

    avcodec_close(pCodecCtx);

    avformat_close_input(&pFormatCtx);

 

    return ;

}

void FFPlayVedio::StopVedio()

{

 

}