首页 > 代码库 > 视频解码图片
视频解码图片
extern "C"{ #include <libavdevice/avdevice.h> #include <libavcodec/avcodec.h> #include <libavformat/avformat.h> #include <libswscale/swscale.h> #include <libavutil/avutil.h>}#pragma comment( lib, "avcodec.lib" )#pragma comment( lib, "avdevice.lib" )#pragma comment( lib, "avformat.lib" )#pragma comment( lib, "avutil.lib" )#pragma comment( lib, "swscale.lib" )#include <windows.h>static int av_create_bmp(char* filename,uint8_t *pRGBBuffer,int width,int height,int bpp){ BITMAPFILEHEADER bmpheader; BITMAPINFO bmpinfo; FILE *fp; fp = fopen(filename,"wb"); if(!fp)return -1; bmpheader.bfType = (‘M‘<<8)|‘B‘; bmpheader.bfReserved1 = 0; bmpheader.bfReserved2 = 0; bmpheader.bfOffBits = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER); bmpheader.bfSize = bmpheader.bfOffBits + width*height*bpp/8; bmpinfo.bmiHeader.biSize = sizeof(BITMAPINFOHEADER); bmpinfo.bmiHeader.biWidth = width; bmpinfo.bmiHeader.biHeight = height; bmpinfo.bmiHeader.biPlanes = 1; bmpinfo.bmiHeader.biBitCount = bpp; bmpinfo.bmiHeader.biCompression = BI_RGB; bmpinfo.bmiHeader.biSizeImage = 0; bmpinfo.bmiHeader.biXPelsPerMeter = 100; bmpinfo.bmiHeader.biYPelsPerMeter = 100; bmpinfo.bmiHeader.biClrUsed = 0; bmpinfo.bmiHeader.biClrImportant = 0; uint8_t* pBuffer = new uint8_t[width * bpp / 8]; int byteWidth = width * bpp / 8; for( int h = 0; h < height / 2; ++h ) { memcpy( pBuffer, pRGBBuffer + h * byteWidth, byteWidth ); memcpy( pRGBBuffer + h * byteWidth, pRGBBuffer + (height - h - 1) * byteWidth, byteWidth ); memcpy( pRGBBuffer + (height - h - 1) * byteWidth, pBuffer, byteWidth ); } delete[] pBuffer; fwrite(&bmpheader,sizeof(BITMAPFILEHEADER),1,fp); fwrite(&bmpinfo.bmiHeader,sizeof(BITMAPINFOHEADER),1,fp); fwrite(pRGBBuffer,width*height*bpp/8,1,fp); fclose(fp); return 0;}int main() { SwsContext *pSWSCtx; AVFormatContext *pFormatCtx; const char *filename="F:\\ACG\\缘之空\\[SumiSora][Yosuga_no_Sora][BDRip][CM01][720P].mp4"; int i,videoStream,y_size; AVCodecContext *pCodecCtx; AVFrame *pFrame; AVFrame *pFrameRGB; int numBytes,frameFinished; uint8_t *buffer; static AVPacket packet; av_register_all(); if(av_open_input_file(&pFormatCtx, filename, NULL, 0, NULL)!=0) printf("error!\n"); if(av_find_stream_info(pFormatCtx)<0) printf("error!\n"); dump_format(pFormatCtx, 0, filename, false); videoStream=-1; for(i=0; i<pFormatCtx->nb_streams; i++) if(pFormatCtx->streams[i]->codec->codec_type==CODEC_TYPE_VIDEO) { videoStream=i; continue; } if(videoStream==-1) printf("error!\n");// Didn‘t find a video stream pCodecCtx=pFormatCtx->streams[videoStream]->codec; AVCodec *pCodec; pCodec=avcodec_find_decoder(pCodecCtx->codec_id); if(pCodec==NULL) printf("error!\n"); if(avcodec_open(pCodecCtx, pCodec)<0) printf("error!\n"); pFrame=avcodec_alloc_frame(); pFrameRGB = avcodec_alloc_frame(); numBytes=avpicture_get_size(PIX_FMT_BGR24, pCodecCtx->width,pCodecCtx->height); buffer=new uint8_t[numBytes]; avpicture_fill((AVPicture *)pFrameRGB, buffer, PIX_FMT_BGR24,pCodecCtx->width, pCodecCtx->height); pSWSCtx = sws_getContext(pCodecCtx->width, pCodecCtx->height, pCodecCtx->pix_fmt, pCodecCtx->width, pCodecCtx->height, PIX_FMT_BGR24, SWS_BICUBIC, NULL, NULL, NULL); i=0; while(av_read_frame(pFormatCtx,&packet)>=0) { if(packet.stream_index==videoStream) { avcodec_decode_video(pCodecCtx, pFrame, &frameFinished,packet.data, packet.size); if(frameFinished) { if(pFrame->key_frame==1 || pFrame->pict_type == FF_P_TYPE)//这里取到关键帧数据 { sws_scale(pSWSCtx, pFrame->data, pFrame->linesize,0, pCodecCtx->height, pFrameRGB->data, pFrameRGB->linesize); i++; static char path[100] = {0}; sprintf( path, "E:/video/img_%d.bmp", i - 1); av_create_bmp( path,pFrameRGB->data[0],pCodecCtx->width,pCodecCtx->height,24); } } } av_free_packet(&packet); } av_free(pFrameRGB); av_free(pFrame); sws_freeContext(pSWSCtx); avcodec_close(pCodecCtx); av_close_input_file(pFormatCtx); return 0; }
视频解码图片
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。