ffmprg Api 示例1--保存指定时间段的视频数据为yuv格式

最新推荐文章于 2022-04-03 10:09:18 发布

轻飘风扬

最新推荐文章于 2022-04-03 10:09:18 发布

阅读量2.2k

点赞数

分类专栏：【音视频】

本文链接：https://blog.csdn.net/sjin_1314/article/details/41323131

版权

【音视频】专栏收录该内容

14 篇文章 1 订阅

订阅专栏

利用ffmpeg截取视频播放文件数据并保存为ｙｕｖ格式。

代码完全参考了这个链接的文章，很详细，在centos进行编译完全没有问题　。

看下面的代码吧：

/*File : testYuv.c
 *Auth : sjin
 *Date : 20141120
 *Mail : 413977243@qq.com
 */

#include <libavformat/avformat.h>
#include <libavcodec/avcodec.h>
#include <libavutil/avutil.h>
#include <libavutil/parseutils.h>

 int main(int argc, char *argv[])
{
	const char *szInfile = "BeatIt.mp4";
	const char *szStartTime = "00:1:55.179";
	const char *szEndTime = "00:4:20.110";
	unsigned int i;
	int j = 0, k = 0;
	int width, height;
	int videoStream = -1;
 
	AVFormatContext *pFmtCtx = NULL;
	AVCodecContext  *pCodecCtx = NULL;
	AVCodec         *pCodec = NULL;
	AVFrame         *pFrame = NULL;
	AVPacket        packet;
	int             getframe;
	AVRational avR, avTimeUint;
	AVDictionary    *optionsDict = NULL;
 
	FILE *pFile;
	char szFilename[32];
 
	double  lframeRate;
	int64_t lStartTime, lEndTime; //millisecond(毫秒)=1/1000 s
	int64_t pos;
	int iStartFrame, iEndFrame;
	int isFirst = 1;//is the first keyframe(the frame av_seek_frame finds)
 
	av_register_all();
 
	//convert time string "hh:mm:ss.xxx" to millisecond
	if (av_parse_time(&lStartTime, szStartTime, 1) < 0) //lStartTime now in microsecond(1/1000 000 s)
	{
		printf("parse start time error!\n");
		return -1;
	}
	pos = lStartTime;
	lStartTime /= 1000;
 
	if (av_parse_time(&lEndTime, szEndTime, 1) < 0)
	{
		printf("parse end time error!\n");
		return -1;
	}
	lEndTime /= 1000;
 
	if(avformat_open_input(&pFmtCtx, szInfile, NULL, NULL) != 0){
		return -1;
    }

	if(avformat_find_stream_info(pFmtCtx, NULL) < 0){
		return -1;
    }

	av_dump_format(pFmtCtx, 0, szInfile, 0);
 
	for(i=0; i<pFmtCtx->nb_streams; i++){
		if(pFmtCtx->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO){
            //帧率  分子/分母
			lframeRate = pFmtCtx->streams[i]->avg_frame_rate.num / pFmtCtx->streams[i]->avg_frame_rate.den;

            //时间基数
			avTimeUint.den = pFmtCtx->streams[i]->time_base.den;
			avTimeUint.num = pFmtCtx->streams[i]->time_base.num;
			width = pFmtCtx->streams[i]->codec->width;
			height = pFmtCtx->streams[i]->codec->height;
			//calc start and end frame number
			iStartFrame = (int)(lStartTime * lframeRate / 1000 + 0.5); 
			iEndFrame = (int)(lEndTime * lframeRate / 1000 + 0.5);

            printf("avTimeUint.den:%d,avTimeUint.num:%d\n",avTimeUint.den,avTimeUint.num);
            printf("lframeRate:%f,width:%d,height:%d,lStartTime:%d,lEndTime:%d,iStartFrame:%d,iEndFrame:%d\n",
                   lframeRate,width,height,lStartTime,lEndTime,iStartFrame,iEndFrame);
 
			videoStream=i;
			break;
		}
	}
	if(videoStream==-1)
		return -1;
 
    //获得编解码器上下文句柄，
	pCodecCtx = pFmtCtx->streams[videoStream]->codec;
 
	// 查找音视频解码器
	pCodec = avcodec_find_decoder(pCodecCtx->codec_id);
	if(pCodec == NULL){
		printf("Codec not found!\n");
		return -1;
	}
	// 打开解码器
	if(avcodec_open2(pCodecCtx, pCodec, &optionsDict) < 0)
		return -1;
 
    //保存解压后的帧数据
	pFrame = avcodec_alloc_frame();
 
    //打开保存YUV的文件
	sprintf(szFilename, "output.yuv");//output file
	pFile = fopen(szFilename, "wb");
	if(pFile == NULL)
		return -1;
 
	avR.num = 1;
	avR.den = AV_TIME_BASE;
 
    //视频时间定位（DTS解码时间戳，PTS显示时间戳）
    //返回定位到开始时间的时间基
	pos = av_rescale_q(pos, avR, pFmtCtx->streams[videoStream]->time_base);
 
	//跳转到指定时间的视频时刻
	if(av_seek_frame(pFmtCtx, videoStream, pos, AVSEEK_FLAG_BACKWARD) < 0)
	{
		printf("seek failed!\n");
		return -1;
	}
    //情况缓存
	avcodec_flush_buffers(pFmtCtx->streams[videoStream]->codec);
 
	while(av_read_frame(pFmtCtx, &packet)>=0)
	{
		if(packet.stream_index == videoStream)
		{
			//解码
			avcodec_decode_video2(pCodecCtx, pFrame, &getframe, &packet);
			if(getframe)
			{
				//save yuv to disk
				if(isFirst)
				{
					k = lframeRate * pFrame->pkt_pts * avTimeUint.num / avTimeUint.den;
					isFirst = 0;
                    if(pFrame->pict_type == AV_PICTURE_TYPE_SP){
                        printf("is P frame .........\n");
                    }
				}
				if(k >= iStartFrame && k<= iEndFrame)
				{
				    for(j=0; j<height; j++)
				    	fwrite(pFrame->data[0] + j * pFrame->linesize[0], 1, width, pFile);
				    for(j=0; j<height/2; j++)
				    	fwrite(pFrame->data[1] + j * pFrame->linesize[1], 1, width/2, pFile);
				    for(j=0; j<height/2; j++)
				    	fwrite(pFrame->data[2] + j * pFrame->linesize[2], 1, width/2, pFile);
				}
 
			}
			k++;
			if(k > iEndFrame)
				break;
		}
		//Free the packet that was allocated by av_read_frame
		av_free_packet(&packet);
	}//while
 
	printf("Resolution: %dx%d", width, height);
 
	fflush(pFile);
	fclose(pFile);
	//Free the YUV frame
	av_free(pFrame);
	//Close the codec
	avcodec_close(pCodecCtx);
	//Close the video file
	avformat_close_input(&pFmtCtx);
	return 0;
}

参考资料：

1、关于视频显示时间及PTS相关的知识