首页 > 代码库 > 视频解码图片

视频解码图片

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;   }

视频解码图片