赞
踩
- 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");
-
- videoStream=-1;
-
- for(i=0; i<pFormatCtx->nb_streams; i++)
- if(pFormatCtx->streams[i]->codec->codec_type==CODEC_TYPE_VIDEO)
- {
- videoStream=i;
- break;
- }
- if(videoStream==-1)
- printf("error!\n");// Didn't find a video stream
-
- // 得到视频流编码上下文的指针
- pCodecCtx=pFormatCtx->streams[videoStream]->codec;
用两种方式,一是利用ffmpeg提供的可执行文件进行提取,另外就是用ffmpeg的sdk,进行开发。我下面说一下如何使用ffmpeg sdk进行提取(假设把提取的关键帧保存成bmp,源文件名是sample.mpg):
首先获取文件中的视频流:
然后选择解码器进行解码:
- 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_RGB24,pCodecCtx->width, pCodecCtx->height);
- pSWSCtx = sws_getContext(pCodecCtx->width, pCodecCtx->height, pCodecCtx->pix_fmt, pCodecCtx->width, pCodecCtx->height, PIX_FMT_RGB24, 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) // 这就是关键帧
- {
- sws_scale(pSWSCtx, pFrame->data, pFrame->linesize,0, pCodecCtx->height, pFrameRGB->data, pFrameRGB->linesize);
- // 保存到磁盘
- char pic[200];
- sprintf(pic,"pic%d.bmp",i);
- i++;
- av_create_bmp(pic,pFrameRGB->data[0],pCodecCtx->width,pCodecCtx->height,24);
- }
- }
- }
- av_free_packet(&packet);
- }
最后,释放资源和句柄
- // 释放 RGB 图象
- av_free(pFrameRGB);
- // 释放YUV 帧
- av_free(pFrame);
-
- sws_freeContext(pSWSCtx);
-
-
- // 关闭解码器(codec)
- avcodec_close(pCodecCtx);
-
- // 关闭视频文件
- av_close_input_file(pFormatCtx);
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。