视频编码技术——Linux下ffmpeg编程实现摄像头捕捉

Posted 2021-07-25 赵新政

1 打开设备

对于FFMpeg编程，首先要打开相关设备，这里我们打开的是linux下的webCam，也就是笔记本摄像头 。在Linux下的ffmpeg里面，需要梳理两个东西：

a  Video4linux2(简称V4L2)，是linux中关于视频设备的内核驱动

b  /dev/video0  /dev/video1 都是代表目标录制设备

其中帧率跟图像大小都是需要进行设置的，然后按照如下代码进行设备打开。

static AVFormatContext* open_dev() {
	//注册所有设备
	avdevice_register_all();
	//获得格式
	const AVInputFormat* iformat = av_find_input_format("video4linux2");

	//打开设备
	AVFormatContext* fmt_ctx = NULL;
	//const char* device_name = "hw:0,0";//这个是音频
	const char* device_name = "/dev/video0";
	AVDictionary* options = NULL;

	//视频就需要做options了
	av_dict_set(&options, "video_size", "640*480", 0);
	av_dict_set(&options, "framerate", "30", 0);

	int ret = avformat_open_input(&fmt_ctx, device_name, iformat, &options);
	
	//判断是否成功
	char errors[1024] = {0, };
	if(ret < 0) {
		av_strerror(ret, errors, 1024);
		std::cout << "Faile to open Device : " <<errors<<std::endl;
		return NULL;
	}
	std::cout<<"Success open Device"<<std::endl;
	return fmt_ctx;
}

2 进行录制保存

其中注意，在linux下直接用pkt的size是可以的。但是在很多系统上，比如mac，pkt的size会有些许变动，每个平台的实现都有不同，所以在遇到很多问题的时候，需要自己注意这点，可以用计算的每一帧数据大小代替

 

void record_video(AVFormatContext* fmt_ctx) {
int ret = -1;
	//准备好文件
	FILE* out_file = fopen("video.yuv", "wb+");
	//读取数据
	AVPacket pkt;
	int count = 0;
	av_init_packet(&pkt);
	while((ret = av_read_frame(fmt_ctx, &pkt)) == 0) {
		
		std::cout<<pkt.size<<std::endl;
		av_log(NULL, AV_LOG_INFO, "packet size is %d(%p), count = %d\\n", 
				pkt.size, pkt.data, count);
		//写入文件
		fwrite(pkt.data, pkt.size, 1, out_file);
		//将文件缓冲区当中数据直接刷进去
		fflush(out_file);
		//每一次读取都会生成空间,所以需要release
		av_packet_unref(&pkt);
		if(count++ > 8000) {
			break;
		}
	}
	
	fclose(out_file);
}