#include"VideoSave.h"


#ifdef OPENCV
using std::cerr;
using namespace std;



cv::VideoCapture* get_capture_video_stream(const char* path) {
    
	cv::VideoCapture* cap = NULL;
	try {
    
		cap = new cv::VideoCapture(path);
	}
	catch (...) {
    
		cerr << " OpenCV exception: video-stream " << path << " can't be opened! \n";
	}
	return cap;
}



void Video::demo(const char* filename, int frame_skip, const char* out_filename)
{
    

	srand(2222222);
	cv::VideoCapture* cap = NULL;
	char* file;
	if (filename) {
    
		printf("Video file: %s\n", filename);
		cap = get_capture_video_stream(filename);

		char drive[_MAX_DRIVE];
		char dir[_MAX_DIR];
		char fname[_MAX_FNAME];
		char ext[_MAX_EXT];

		_splitpath(filename, drive, dir, fname, ext);

		printf("Drive:%s\n file name: %s\n file type: %s\n", drive, fname, ext);
		file = fname;
	}


	if (!cap) {
    
#ifdef WIN32
		printf("Check that you have copied file opencv_ffmpeg340_64.dll to the same directory where is darknet.exe \n");
#endif
		printf("Couldn't connect to webcam.\n");
		exit(EXIT_FAILURE);
	}



	//获取整个帧数
	long totalFrameNumber = cap->get(cv::CAP_PROP_FRAME_COUNT);
	cout << "Total frames:" << totalFrameNumber << "帧" << endl;


	//设置开始帧()
	long StartFrame = 1;
	cap->set(cv::CAP_PROP_POS_FRAMES, StartFrame); //从设定的帧开始读
	//cap->set(cv::CAP_PROP_POS_MSEC, double(0));  //从设定的时间开始读
	cout << "StartFrame:" << StartFrame << endl;

	//设置结束帧
	int StopFrame = 10;

	if (StopFrame < StartFrame)
	{
    
		cout << "结束帧小于开始帧，程序错误，即将退出！" << endl;
		exit(EXIT_FAILURE);
	}
	else
	{
    
		cout << "结束帧为：第" << StopFrame << "帧" << endl;
	}

	//获取帧率
	double rate = cap->get(cv::CAP_PROP_FPS);
	cout << "帧率为:" << rate << endl;


	//定义一个用来控制读取视频循环结束的变量
	bool stop = false;

	//实例化帧
	cv::Mat frame;

	//显示每一帧的窗口
	//namedWindow( "Extractedframe" );

	//两帧间的间隔时间:
	//int delay = 1000/rate;
	int delay = rate;


	//利用while循环读取帧
	//currentFrame是在循环体中控制读取到指定的帧后循环结束的变量
	long currentFrame = StartFrame;


	while (!stop)
	{
    
		//读取下一帧
		if (!cap->read(frame))
		{
    
			cout << "读取视频失败" << endl;
			exit(EXIT_FAILURE);
		}


		//cout << "正在读取第" << currentFrame << "帧" << endl;
		cv::imshow("Extractedframe", frame);

		//此处为跳帧操作
		if (currentFrame) 
		//此处为帧数间隔，修改这里就可以了 隔50帧保存一帧就是 currentframe%50 == 0
		{
    
			cout << "正在写第" << currentFrame << "帧" << endl;
			stringstream str;
			str << filename << "_" << currentFrame << ".png";        /*图片存储位置*/
			//str << file << "_" << currentFrame << ".png";


			cout << str.str() << endl;
			cv::imwrite(str.str(), frame);
		}

		waitKey(intdelay=0)当delay≤ 0时会永远等待；当delay>0时会等待delay毫秒
		当时间结束前没有按键按下时，返回值为-1；否则返回按键
		//int c = cv::waitKey(0);
		按下ESC或者到达指定的结束帧后退出读取视频
		//if ((char)c == 27 || currentFrame > StopFrame)
		//{
    
		//	stop = true;
		//}
		按下按键后会停留在当前帧，等待下一次按键
		//if (c >= 0)
		//{
    
		//	cv::waitKey(0);
		//}
		if (currentFrame > StopFrame)
		{
    
			stop = true;
		}
		currentFrame++;

	}

	//关闭视频文件
	cap->release();
	cv::waitKey(0);
	cv::destroyAllWindows();

}
#else
void demo(char* cfgfile, char* weightfile, float thresh, float hier_thresh, int cam_index, const char* filename, char** names, int classes, int avgframes,
	int frame_skip, char* prefix, char* out_filename, int mjpeg_port, int dontdraw_bbox, int json_port, int dont_show, int ext_output, int letter_box_in, int time_limit_sec, char* http_post_host,
	int benchmark, int benchmark_layers)
{
    
	fprintf(stderr, "Demo needs OpenCV for webcam images.\n");
}
#endif