ffmpeg解码视频存为BMP文件
分类&＃xff1a; ffmpeg2011-07-28 12:13 8人阅读 评论(0) 收藏 举报
view plain
- #include
- #include
- #include
- #include
- #pragma once
- #ifdef __cplusplus
- extern "C" {
- #endif
- #include
- #include
- #include
- #ifdef __cplusplus
- }
- #endif
- //定义BMP文件头
- #ifndef _WINGDI_
- #define _WINGDI_
- typedef struct tagBITMAPFILEHEADER {
- WORD bfType;
- DWORD bfSize;
- WORD bfReserved1;
- WORD bfReserved2;
- DWORD bfOffBits;
- } BITMAPFILEHEADER, FAR *LPBITMAPFILEHEADER, *PBITMAPFILEHEADER;
- typedef struct tagBITMAPINFOHEADER{
- DWORD biSize;
- LONG biWidth;
- LONG biHeight;
- WORD biPlanes;
- WORD biBitCount;
- DWORD biCompression;
- DWORD biSizeImage;
- LONG biXPelsPerMeter;
- LONG biYPelsPerMeter;
- DWORD biClrUsed;
- DWORD biClrImportant;
- } BITMAPINFOHEADER, FAR *LPBITMAPINFOHEADER, *PBITMAPINFOHEADER;
- #endif
- //保存BMP文件的函数
- void SaveAsBMP (AVFrame *pFrameRGB, int width, int height, int index, int bpp)
- {
- char buf[5] &＃61; {0};
- BITMAPFILEHEADER bmpheader;
- BITMAPINFOHEADER bmpinfo;
- FILE *fp;
- char *filename &＃61; new char[255];
- //文件存放路径&＃xff0c;根据自己的修改
- sprintf_s(filename,255,"%s%d.bmp","D:/My Documents/Visual Studio 2008/Projects/WriteVideo/Debug/test",index);
- if ( (fp&＃61;fopen(filename,"wb&＃43;")) &＃61;&＃61; NULL )
- {
- printf ("open file failed!\n");
- return;
- }
- bmpheader.bfType &＃61; 0x4d42;
- bmpheader.bfReserved1 &＃61; 0;
- bmpheader.bfReserved2 &＃61; 0;
- bmpheader.bfOffBits &＃61; sizeof(BITMAPFILEHEADER) &＃43; sizeof(BITMAPINFOHEADER);
- bmpheader.bfSize &＃61; bmpheader.bfOffBits &＃43; width*height*bpp/8;
- bmpinfo.biSize &＃61; sizeof(BITMAPINFOHEADER);
- bmpinfo.biWidth &＃61; width;
- bmpinfo.biHeight &＃61; height;
- bmpinfo.biPlanes &＃61; 1;
- bmpinfo.biBitCount &＃61; bpp;
- bmpinfo.biCompression &＃61; BI_RGB;
- bmpinfo.biSizeImage &＃61; (width*bpp&＃43;31)/32*4*height;
- bmpinfo.biXPelsPerMeter &＃61; 100;
- bmpinfo.biYPelsPerMeter &＃61; 100;
- bmpinfo.biClrUsed &＃61; 0;
- bmpinfo.biClrImportant &＃61; 0;
- fwrite (&bmpheader, sizeof(bmpheader), 1, fp);
- fwrite (&bmpinfo, sizeof(bmpinfo), 1, fp);
- fwrite (pFrameRGB->data[0], width*height*bpp/8, 1, fp);
- fclose(fp);
- }
- //主函数
- int main (void)
- {
- unsigned int i &＃61; 0, videoStream &＃61; -1;
- AVCodecContext *pCodecCtx;
- AVFormatContext *pFormatCtx;
- AVCodec *pCodec;
- AVFrame *pFrame, *pFrameRGB;
- struct SwsContext *pSwsCtx;
- const char *filename &＃61; "D:/My Documents/Visual Studio 2008/Projects/WriteVideo/Debug/DELTA.MPG";
- AVPacket packet;
- int frameFinished;
- int PictureSize;
- uint8_t *buf;
- //注册编解码器
- av_register_all();
- //打开视频文件
- if ( av_open_input_file(&pFormatCtx, filename, NULL, 0, NULL) !&＃61; 0 )
- {
- printf ("av open input file failed!\n");
- exit (1);
- }
- //获取流信息
- if ( av_find_stream_info(pFormatCtx) < 0 )
- {
- printf ("av find stream info failed!\n");
- exit (1);
- }
- //获取视频流
- for ( i&＃61;0; i
nb_streams; i&＃43;&＃43; ) - if ( pFormatCtx->streams[i]->codec->codec_type &＃61;&＃61; CODEC_TYPE_VIDEO )
- {
- videoStream &＃61; i;
- break;
- }
- if (videoStream &＃61;&＃61; -1)
- {
- printf ("find video stream failed!\n");
- exit (1);
- }
- pCodecCtx &＃61; pFormatCtx->streams[videoStream]->codec;
- pCodec &＃61; avcodec_find_decoder (pCodecCtx->codec_id);
- if (pCodec &＃61;&＃61; NULL)
- {
- printf ("avcode find decoder failed!\n");
- exit (1);
- }
- //打开解码器
- if ( avcodec_open(pCodecCtx, pCodec)<0 )
- {
- printf ("avcode open failed!\n");
- exit (1);
- }
- //为每帧图像分配内存
- pFrame &＃61; avcodec_alloc_frame();
- pFrameRGB &＃61; avcodec_alloc_frame();
- if ( (pFrame&＃61;&＃61;NULL)||(pFrameRGB&＃61;&＃61;NULL) )
- {
- printf("avcodec alloc frame failed!\n");
- exit (1);
- }
- PictureSize &＃61; avpicture_get_size (PIX_FMT_BGR24, pCodecCtx->width, pCodecCtx->height);
- buf &＃61; (uint8_t*)av_malloc(PictureSize);
- if ( buf &＃61;&＃61; NULL )
- {
- printf( "av malloc failed!\n");
- exit(1);
- }
- avpicture_fill ( (AVPicture *)pFrameRGB, buf, PIX_FMT_BGR24, pCodecCtx->width, pCodecCtx->height);
- //设置图像转换上下文
- pSwsCtx &＃61; sws_getContext (pCodecCtx->width,
- pCodecCtx->height,
- pCodecCtx->pix_fmt,
- pCodecCtx->width,
- pCodecCtx->height,
- PIX_FMT_BGR24,
- SWS_BICUBIC,
- NULL, NULL, NULL);
- i &＃61; 0;
- while(av_read_frame(pFormatCtx, &packet) >&＃61; 0)
- {
- if(packet.stream_index&＃61;&＃61;videoStream)
- {
- avcodec_decode_video(pCodecCtx, pFrame, &frameFinished,
- packet.data, packet.size);
- if(frameFinished)
- {
- //反转图像 &＃xff0c;否则生成的图像是上下调到的
- pFrame->data[0] &＃43;&＃61; pFrame->linesize[0] * (pCodecCtx->height - 1);
- pFrame->linesize[0] *&＃61; -1;
- pFrame->data[1] &＃43;&＃61; pFrame->linesize[1] * (pCodecCtx->height / 2 - 1);
- pFrame->linesize[1] *&＃61; -1;
- pFrame->data[2] &＃43;&＃61; pFrame->linesize[2] * (pCodecCtx->height / 2 - 1);
- pFrame->linesize[2] *&＃61; -1;
- //转换图像格式&＃xff0c;将解压出来的YUV420P的图像转换为BRG24的图像
- sws_scale (pSwsCtx, pFrame->data, pFrame->linesize, 0, pCodecCtx->height, pFrameRGB->data, pFrameRGB->linesize);
- SaveAsBMP (pFrameRGB, pCodecCtx->width, pCodecCtx->height, i&＃43;&＃43;, 24);
- }
- }
- av_free_packet(&packet);
- }
- sws_freeContext (pSwsCtx);
- av_free (pFrame);
- av_free (pFrameRGB);
- avcodec_close (pCodecCtx);
- av_close_input_file (pFormatCtx);
- return 0;
- }
京公网安备 11010802041100号