背景建模技术（四）：视频分析（VideoAnalysis）模块

视频分析模块主要包含两个函数，一个是VideoAnalysis::setup（....），其主要功能就是确定测试的视频是视频文件或摄像头输入亦或是采用命令行参数；第二个函数是VideoAnalysis::start()，其主要功能初始化视频处理、设置视频获取方式以及开始视频捕获功能等。

1、VideoAnalysis::setup（....）

该函数的代码如下：

bool VideoAnalysis::setup(int argc, const char **argv)
	{
		bool flag = false;

		const char* keys =
							"{hp|help|false|Print help message}"
							"{uf|use_file|false|Use video file}"
							"{fn|filename||Specify video file}"
							"{uc|use_cam|false|Use camera}"
							"{ca|camera|0|Specify camera index}"
							"{co|use_comp|false|Use mask comparator}"
							"{st|stopAt|0|Frame number to stop}"
							"{im|imgref||Specify image file}" ;
		
		cv::CommandLineParser cmd(argc, argv, keys);

		////////////use_command
		if (argc <= 1 || cmd.get<bool>("help") == true)
		{
			cout << "Usage: " << argv[0] << " [options]" << endl;
			cout << "Avaible options:" << endl;
			cmd.printParams();
			return false;
		}

		////////////use_file
		use_file = cmd.get<bool>("use_file");
		if (use_file)
		{
			filename = cmd.get<string>("filename");

			if (filename.empty())
			{
				cout << "Specify filename" << endl;
				return false;
			}

			flag = true;
		}

		////////////use_camera
		use_camera = cmd.get<bool>("use_cam");
		if (use_camera)
		{
			cameraIndex = cmd.get<int>("camera");
			flag = true;
		}

		////////////use_comp
		if (flag == true)
		{
			use_comp = cmd.get<bool>("use_comp");
			if (use_comp)
			{
				frameToStop = cmd.get<int>("stopAt");
				imgref = cmd.get<string>("imgref");

				if (imgref.empty())
				{
					cout << "Specify image reference" << endl;
					return false;
				}
			}
		}

		return flag;
	}

它的主要流程如下图所示：

2、VideoAnalysis::start()

该函数的代码如下：

void VideoAnalysis::start()
	{
		//cout << "Press 'ESC' to stop..." << endl;

		do
		{
			videoCapture = new VideoCapture;
			frameProcessor = new FrameProcessor;

			frameProcessor->init();
			frameProcessor->frameToStop = frameToStop;
			frameProcessor->imgref = imgref;

			videoCapture->setFrameProcessor(frameProcessor);///setFrameProcessor

			if (use_file)
				videoCapture->setVideo(filename);///setVideo

			if (use_camera)
				videoCapture->setCamera(cameraIndex);///setCamera

			videoCapture->start();///start

			if (use_file || use_camera)
				break;

			frameProcessor->finish();

			int key = cvWaitKey(500);
			if (key == KEY_ESC)
				break;

			delete frameProcessor;
			delete videoCapture;

		} 
		while (1);

		delete frameProcessor;
		delete videoCapture;
	}

它的主要流程如下图所示：