#include "PreProcessModel.h"
#include <future>

// 获取当前时间并格式化为指定格式的字符串
std::string getCurrentDateTime1(const std::string& format) {
	// 获取当前时间
	time_t now = time(0);
	// 转换为本地时间
	tm* local_time = localtime(&now);
	// 定义存储时间字符串的缓冲区
	char buffer[80];
	// 格式化时间为字符串
	strftime(buffer, sizeof(buffer), format.c_str(), local_time);
	// 返回格式化后的时间字符串
	return std::string(buffer);
}

/// <summary>
/// 根据提升机暂停和运动的比例，来最终判断是否运动
/// </summary>
/// <returns>0:运动 1:静止 -1:无法判断</returns>
int fnJudeRatio(std::queue<int> quTSJ)
{
	// 计算队列中 0 和 1 的数量
	int count0 = 0;//运动个数
	int count1 = 0;//静止个数
	int queueSize = quTSJ.size();  // 队列长度
	while (!quTSJ.empty()) {
		if (quTSJ.front() == 0) {
			count0++;
		}
		else {
			count1++;
		}
		quTSJ.pop();
	}

	//cout << getCurrentDateTime1("%Y%m%d%H%M%S") << endl;
	/*cout << "------------" << endl;
	cout << count0 << endl;
	cout << count1 << endl;*/

	// 计算比例
	double ratio0 = static_cast<double>(count0) / queueSize;
	double ratio1 = static_cast<double>(count1) / queueSize;

	if (ratio0 > 0.6)
	{
		return 0;
	}
	else if (ratio1 > 0.6)
	{
		return 1;
	}
	else
	{
		return -1;
	}
}

int ToWchar(char* &src, wchar_t* &dest, const char *locale = "zh_CN.utf8")
{
    if (src == NULL) {
        dest = NULL;
        return 0;
    }

    // 根据环境变量设置locale
    setlocale(LC_CTYPE, locale);

    // 得到转化为需要的宽字符大小
    int w_size = mbstowcs(NULL, src, 0) + 1;

    // w_size = 0 说明mbstowcs返回值为-1。即在运行过程中遇到了非法字符(很有可能使locale
    // 没有设置正确)
    if (w_size == 0) {
        dest = NULL;
        return -1;
    }

    //wcout << "w_size" << w_size << endl;
    dest = new wchar_t[w_size];
    if (!dest) {
        return -1;
    }

    int ret = mbstowcs(dest, src, strlen(src)+1);
    if (ret <= 0) {
        return -1;
    }
    return 0;
}


// 在图像上绘制带有背景的文本
void drawTextWithBackground(Mat& image, const string& text, int txalign, const Point& orgxy, const Point& orgwh, Scalar bgColor)
{
	try
	{
		CvxText chinese = CvxText("SimHei.ttf");
		//设置位置和字体参数
		Scalar textColor(255, 255, 255);//白色文本
		//Scalar bgColor(0, 255, 0);//绿色背景
		int fontFace = FONT_HERSHEY_SCRIPT_SIMPLEX;
		double fontScale = 0.2;
		int thickness = 1;

		// 获取文字长度大小
		Size textSize = getTextSize(text, fontFace, fontScale, thickness, 0);

		//计算文本背景矩形框的大小
		//Rect backgroundRect(orgxy.x, orgxy.y - textSize.height - 30, orgwh.x, textSize.height + 30);
		if (text.length() >= 6)
		{
			Rect backgroundRect(orgxy.x, orgxy.y - textSize.height - 30, orgwh.x + textSize.width + 30, textSize.height + 30);
			//绘制文本背景
			cv::rectangle(image, backgroundRect, bgColor, FILLED);
		}
		else
		{
			Rect backgroundRect(orgxy.x, orgxy.y - textSize.height - 30, orgwh.x, textSize.height + 30);
			//绘制文本背景
			cv::rectangle(image, backgroundRect, bgColor, FILLED);
		}
		//绘制文本背景
		//rectangle(image, backgroundRect, bgColor, FILLED);

		//CvxText chinese(ttfFont);
		//设置字体的大小 / 空白比例 / 间隔比较 / 旋转角度
		//Scalar size(30, 0.5, 0, 0);
		Scalar size(25, 0, 0.1, 0);
		float p = 1.0f; //设置字体的透明度
		//设置字体
		chinese.setFont(nullptr, &size, nullptr, &p);

		switch (txalign)
		{
		case 1://文字左对齐	
			//绘制文本
			//chinese.putText(image, text.c_str(), cv::Point(orgxy.x + 5, (orgxy.y - (orgwh.y / 2) - 2)), textColor);
			chinese.putText(image, text.c_str(), cv::Point(orgxy.x + 5, orgxy.y - 8), textColor);
			break;
		case 2://文字居中
			//绘制文本
			chinese.putText(image, text.c_str(), cv::Point(orgxy.x + 5 + (orgwh.x - textSize.width) / 2, (orgxy.y - 2)), textColor);

			break;
		case 3://文字右对齐
			//绘制文本
			chinese.putText(image, text.c_str(), cv::Point(orgxy.x + 5 + (orgwh.x - textSize.width), (orgxy.y - 2)), textColor);

			break;
		default://默认不支持中文
			//绘制文本
			putText(image, text.c_str(), cv::Point(orgxy.x + (orgwh.x - textSize.width) / 2, (orgxy.y - (orgwh.y / 2) - 2)), fontFace, 1, textColor, 2);
			break;
		}
	}
	catch (const std::exception& ex)
	{
		std::string errorMessage = "drawTextWithBackground-";
		errorMessage += ex.what();
		cout << errorMessage << endl;
		throw std::runtime_error(errorMessage);
	}

}
// 在图像上绘制带有背景的文本
void drawTextWithBackgroundWstr(Mat& image, const string& _text, int txalign, const Point& orgxy, const Point& orgwh, Scalar bgColor)
{
	try
	{
		

		std::wstring_convert<std::codecvt_utf8_utf16<wchar_t>> converter;
    		std::wstring text = converter.from_bytes(_text);

		CvxText chinese = CvxText("SimHei.ttf");
		//设置位置和字体参数
		Scalar textColor(255, 255, 255);//白色文本
		//Scalar bgColor(0, 255, 0);//绿色背景
		int fontFace = FONT_HERSHEY_SCRIPT_SIMPLEX;
		double fontScale = 0.2;
		int thickness = 1;

		// 获取文字长度大小
		Size textSize = getTextSize(_text, fontFace, fontScale, thickness, 0);

		//计算文本背景矩形框的大小
		//Rect backgroundRect(orgxy.x, orgxy.y - textSize.height - 30, orgwh.x, textSize.height + 30);
		if (text.length() >= 6)
		{
			Rect backgroundRect(orgxy.x, orgxy.y - textSize.height - 30, orgwh.x + textSize.width + 30, textSize.height + 30);
			//绘制文本背景
			cv::rectangle(image, backgroundRect, bgColor, FILLED);
		}
		else
		{
			Rect backgroundRect(orgxy.x, orgxy.y - textSize.height - 30, orgwh.x, textSize.height + 30);
			//绘制文本背景
			cv::rectangle(image, backgroundRect, bgColor, FILLED);
		}
		//绘制文本背景
		//rectangle(image, backgroundRect, bgColor, FILLED);

		//CvxText chinese(ttfFont);
		//设置字体的大小 / 空白比例 / 间隔比较 / 旋转角度
		//Scalar size(30, 0.5, 0, 0);
		Scalar size(25, 0, 0.1, 0);
		float p = 1.0f; //设置字体的透明度
		//设置字体
		chinese.setFont(nullptr, &size, nullptr, &p);

		switch (txalign)
		{
		case 1://文字左对齐	
			//绘制文本
			//chinese.putText(image, text.c_str(), cv::Point(orgxy.x + 5, (orgxy.y - (orgwh.y / 2) - 2)), textColor);
			chinese.putText(image, text.c_str(), cv::Point(orgxy.x + 5, orgxy.y - 8), textColor);
			break;
		case 2://文字居中
			//绘制文本
			chinese.putText(image, text.c_str(), cv::Point(orgxy.x + 5 + (orgwh.x - textSize.width) / 2, (orgxy.y - 2)), textColor);

			break;
		case 3://文字右对齐
			//绘制文本
			chinese.putText(image, text.c_str(), cv::Point(orgxy.x + 5 + (orgwh.x - textSize.width), (orgxy.y - 2)), textColor);

			break;
		default://默认不支持中文
			//绘制文本
			putText(image, _text.c_str(), cv::Point(orgxy.x + (orgwh.x - textSize.width) / 2, (orgxy.y - (orgwh.y / 2) - 2)), fontFace, 1, textColor, 2);
			break;
		}
	}
	catch (const std::exception& ex)
	{
		std::string errorMessage = "drawTextWithBackground-";
		errorMessage += ex.what();
		cout << errorMessage << endl;
		throw std::runtime_error(errorMessage);
	}

}
//皮带运行状态检测 modelId = 1
double PreProcessModel::fnBeltMoveRec(Mat& imMat, std::vector<cv::Mat>& imagesBelt, int& beltCount, RUN& rBelt, std::vector<cv::Point> vertices)
{
	try
	{
		int result = 0;//0运行 1停止 -1未检查

		//定义接受帧
		cv::Mat frame = imMat;
		// 在视频帧上添加矩形框  
		rectangle(frame, cv::Point(vertices[0].x, vertices[0].y), cv::Point(vertices[1].x, vertices[1].y), cv::Scalar(255, 255, 0), 1);

		if (imagesBelt.size() >= 3)//长度大于3自动去掉头部帧
		{
			imagesBelt.erase(imagesBelt.begin());
		}

		if (imagesBelt.size() < 3 && (beltCount % 10 == 0))//取第间隔10帧取一次
		{
			imagesBelt.push_back(frame);
		}

		if (imagesBelt.size() == 3)//取满3帧，调用模型
		{
			//result = rBelt.detect(imagesBelt);
			imagesBelt.clear();
			beltCount++;
			return result;
		}

		if (beltCount > 10000)
			beltCount = 0;

		beltCount = beltCount + 1;

		frame.release();
		return -1;
	}
	catch (const std::exception& ex)
	{
		std::string errorMessage = "fnBeltMoveRec-";
		errorMessage += ex.what();
		cout << errorMessage << endl;
		throw std::runtime_error(errorMessage);
	}
}

/// <summary>
/// 提升机变动检测 modelId = 2
/// </summary>
/// <param name="imMat">视频流Mat</param>
/// <param name="imMatF">提升机用帧队列</param> 
/// <param name="hat">模型引用</param>
double PreProcessModel::fnImRecProByModelTSJ(Mat& imMat, std::vector<cv::Mat>& imagesTsj, int& tsjCount, RUN& rTSJ, std::queue<int>& quTSJ, std::vector<cv::Point> vertices)
{
	try
	{
		int lastresult = -1;//最终判断结果 0运行 1停止 -1 无法判断
		int result = 0;//0运行 1停止 -1无法判断

		//定义接受帧
		cv::Mat frame = imMat;
		// 在视频帧上添加矩形框  
		rectangle(frame, cv::Point(vertices[0].x, vertices[0].y), cv::Point(vertices[1].x, vertices[1].y), cv::Scalar(255, 255, 0), 2);

		//if (imagesTsj.size() >= 3)//长度大于3自动去掉头部帧
		//{
		//	imagesTsj.erase(imagesTsj.begin());
		//}

		if (imagesTsj.size() < 3 && (tsjCount % 20 == 0))//取第间隔10帧取一次，帧率为15，间隔30帧，也就是2秒
		{
			imagesTsj.push_back(frame);
		}

		if (imagesTsj.size() == 3)//取满3帧，调用模型
		{
			result = rTSJ.detect(imagesTsj);

			std::string text;
			if (result == 0) {
				text = "run";
			}
			else {
				text = "stop";
			}
			//显示结果
			cv::putText(frame, text, cv::Point(vertices[0].x, vertices[0].y - 5), 2, 4, cv::Scalar(255, 255, 255), 4);

			imagesTsj.clear();
			tsjCount++;

			//把判断结果存入队列
			if (quTSJ.size() < 30)
			{
				quTSJ.push(result);
			}
			else
			{
				//足够30个结果，就进行总判断处理
				lastresult = fnJudeRatio(quTSJ);
				quTSJ.pop();//去掉头部第一个元素
			}

			return lastresult;//返回结果
		}

		if (tsjCount > 10000)
			tsjCount = 0;

		tsjCount = tsjCount + 1;

		frame.release();

		return lastresult;
	}
	catch (const std::exception& ex)
	{
		imagesTsj.clear();// 清空帧容器
		tsjCount = 0;// 计数归零
		std::string errorMessage = "fnImRecProByModelTSJ-";
		errorMessage += ex.what();
		cout << errorMessage << endl;
		throw std::runtime_error(errorMessage);
	}
}

/// <summary>
/// 人员识别，安全帽识别 modelId = 3
/// </summary>
/// <param name="imMat">视频流Mat</param>
/// <param name="hat">模型引用</param>
vector<double> PreProcessModel::fnImRecProByModelHAT(Mat& imMat, HAT& hat, std::vector<cv::Point> vertices, std::vector<cv::Point> workvertices)
{
	try
	{
		vector<double> result;
		string titleperson = "人 员";

		double distance = -1;//0:危险区域边界线上 >1:危险区域内 <1:危险区域外
		double workdistance = -1;//0:工作区域边界线上 >1:工作区域内 <1:工作区域外
		double workpersons = 0;//工作区域人的数量
		double wearHat = 0;//0:全部佩戴安全帽 1：未佩戴安全帽的人的数量

		//定义接受帧
		cv::Mat frame = imMat;

		//获取视频帧的宽度和高度和帧率
		int frame_width = frame.cols;
		int frame_height = frame.rows;
		double frameRate = 25;

		// 创建一个空的边界框向量
		std::vector<Box> boxes;
		// 调用 detect 函数获取检测到的边界框
		hat.detect(frame, boxes);

		//多边形区域
		//std::vector<cv::Point> vertices = stringToPoints(area);

		// 绘制多边形
		cv::polylines(frame, vertices, true, cv::Scalar(0, 0, 255), 1);

		// 定义要测试的点
		cv::Point testPoint;
		cv::Scalar sl;//标框颜色

		// 使用索引遍历boxes
		for (int i = 0; i < boxes.size(); i++) {
			
			switch (boxes[i].class_id)
			{
			case 0:
				//如果标注了危险区域
				if (vertices.size()>0)
				{
					distance = cv::pointPolygonTest(vertices, testPoint, false);//危险区域判断
					if (distance > 0)//危险区域内有个人,画红色
					{
						sl = Scalar(0, 0, 255);
						titleperson = "闯 入";
					}
					else
					{
						sl = Scalar(0, 255, 0);
						titleperson = "人 员";
					}
				}			

				//如果标注了工作区域
				if (workvertices.size()>0)
				{
					workdistance = cv::pointPolygonTest(workvertices, testPoint, false);//脱岗离岗判断


					if (workdistance >= 0)//工作区域内有个人
					{
						workpersons++;
					}
				}

				//标注人形
				drawTextWithBackgroundWstr(frame, titleperson, 2, cv::Point(boxes[i].x, boxes[i].y), cv::Point(boxes[i].width, boxes[i].height), sl);
				//在视频帧上添加矩形框 
				rectangle(frame, cv::Point(boxes[i].x, boxes[i].y), cv::Point((boxes[i].x + boxes[i].width), (boxes[i].y + boxes[i].height)), sl, 1);

				break;
			//case 1:
			//	//标注帽子
			//	drawTextWithBackgroundWstr(frame, "安全帽", 2, cv::Point(boxes[i].x, boxes[i].y), cv::Point(boxes[i].width, boxes[i].height), cv::Scalar(0, 255, 0));
			//	//在视频帧上添加矩形框 
			//	rectangle(frame, cv::Point(boxes[i].x, boxes[i].y), cv::Point((boxes[i].x + boxes[i].width), (boxes[i].y + boxes[i].height)), cv::Scalar(0, 255, 0), 1);

			//	break;
			case 2:

				//如果标注了工作区域
				if (workvertices.size() > 0)
				{
					workdistance = cv::pointPolygonTest(workvertices, testPoint, false);//脱岗离岗判断

					if (workdistance >= 0)//工作区域内有个人
					{
						workpersons++;
					}
				}

				//wearHat++;//未佩戴安全帽的人的数量

				//标注头部
				drawTextWithBackgroundWstr(frame, "头", 2, cv::Point(boxes[i].x, boxes[i].y), cv::Point(boxes[i].width, boxes[i].height), cv::Scalar(0, 255, 0));
				rectangle(frame, cv::Point(boxes[i].x, boxes[i].y), cv::Point((boxes[i].x + boxes[i].width), (boxes[i].y + boxes[i].height)), cv::Scalar(0, 255, 0), 1);

				break;
			default:
				break;
			}
		}
		
		if (workpersons <= 0 && workvertices.size() > 0)
		{
			drawTextWithBackgroundWstr(frame,"离 岗", 2, cv::Point(workvertices[0].x, workvertices[0].y), cv::Point(workvertices[1].x - workvertices[0].x, workvertices[1].y - workvertices[0].y), Scalar(255, 255, 0));

			// 绘制多边形，工作区域
			cv::polylines(frame, workvertices, true, cv::Scalar(255, 255, 0), 2);
		}		

		frame.release();

		result.push_back(distance);//有人进入危险区域
		result.push_back(wearHat);//未带安全帽人员数量
		result.push_back(workpersons);//工作区域人员数量

		return result;
	}
	catch (const std::exception& ex)
	{
		std::string errorMessage = "fnImRecProByModelHAT-";
		errorMessage += ex.what();
		cout << errorMessage << endl;
		throw std::runtime_error(errorMessage);
	}
}
vector<double> PreProcessModel::fnImRecProByModelHAT1(Mat& imMat, HAT& hat, std::vector<cv::Point> vertices, std::vector<cv::Point> workvertices,std::vector<std::vector<int>> c_list,std::string lable_title,std::string color_result)
{
	try
	{
		vector<double> result;
		string titleperson = "人 员";

		double distance = -1;//0:危险区域边界线上 >1:危险区域内 <1:危险区域外
		double workdistance = -1;//0:工作区域边界线上 >1:工作区域内 <1:工作区域外
		double workpersons = 0;//工作区域人的数量
		double wearHat = 0;//0:全部佩戴安全帽 1：未佩戴安全帽的人的数量

		//定义接受帧
		cv::Mat frame = imMat;

		//获取视频帧的宽度和高度和帧率
		int frame_width = frame.cols;
		int frame_height = frame.rows;
		double frameRate = 25;

		// 创建一个空的边界框向量
		std::vector<Box> boxes;
		// 调用 detect 函数获取检测到的边界框
		hat.detect(frame, boxes);

		//多边形区域
		//std::vector<cv::Point> vertices = stringToPoints(area);
		// 绘制多边形
		cv::polylines(frame, vertices, true, cv::Scalar(0, 0, 255), 1);
		drawTextWithBackgroundWstr(frame, lable_title, 1, cv::Point(vertices[0].x, vertices[0].y), cv::Point(150, 40), Scalar(c_list[0][0], c_list[0][1], c_list[0][2]));

		// 定义要测试的点
		cv::Point testPoint;
		cv::Scalar sl;//标框颜色

		// 使用索引遍历boxes
		for (int i = 0; i < boxes.size(); i++) {
			
			switch (boxes[i].class_id)
			{
			case 0:
				//如果标注了危险区域
				if (vertices.size()>0)
				{
					distance = cv::pointPolygonTest(vertices, testPoint, false);//危险区域判断
					if (distance > 0)//危险区域内有个人,画红色
					{
						if(color_result == "false")
						{
							sl = Scalar(255, 0, 0);
						}
						else
						{
							sl = Scalar(0, 0, 255);
						}
						titleperson = "闯 入";
					}
					else
					{
						if(color_result == "false")
						{
							sl = Scalar(255, 0, 0);
						}
						else
						{
							sl = Scalar(0, 0, 255);
						}
						titleperson = "人 员";
					}
				}			

				//如果标注了工作区域
				if (workvertices.size()>0)
				{
					workdistance = cv::pointPolygonTest(workvertices, testPoint, false);//脱岗离岗判断


					if (workdistance >= 0)//工作区域内有个人
					{
						workpersons++;
					}
				}

				//标注人形
				drawTextWithBackgroundWstr(frame, titleperson, 2, cv::Point(boxes[i].x, boxes[i].y), cv::Point(boxes[i].width, boxes[i].height), sl);
				//在视频帧上添加矩形框 
				rectangle(frame, cv::Point(boxes[i].x, boxes[i].y), cv::Point((boxes[i].x + boxes[i].width), (boxes[i].y + boxes[i].height)), sl, 1);

				break;
			//case 1:
			//	//标注帽子
			//	drawTextWithBackgroundWstr(frame, "安全帽", 2, cv::Point(boxes[i].x, boxes[i].y), cv::Point(boxes[i].width, boxes[i].height), cv::Scalar(0, 255, 0));
			//	//在视频帧上添加矩形框 
			//	rectangle(frame, cv::Point(boxes[i].x, boxes[i].y), cv::Point((boxes[i].x + boxes[i].width), (boxes[i].y + boxes[i].height)), cv::Scalar(0, 255, 0), 1);

			//	break;
			case 2:

				//如果标注了工作区域
				if (workvertices.size() > 0)
				{
					workdistance = cv::pointPolygonTest(workvertices, testPoint, false);//脱岗离岗判断

					if (workdistance >= 0)//工作区域内有个人
					{
						workpersons++;
					}
				}

				//wearHat++;//未佩戴安全帽的人的数量

				//标注头部
				drawTextWithBackgroundWstr(frame, "头", 2, cv::Point(boxes[i].x, boxes[i].y), cv::Point(boxes[i].width, boxes[i].height), cv::Scalar(0, 255, 0));
				rectangle(frame, cv::Point(boxes[i].x, boxes[i].y), cv::Point((boxes[i].x + boxes[i].width), (boxes[i].y + boxes[i].height)), cv::Scalar(0, 255, 0), 1);

				break;
			default:
				break;
			}
		}
		
		if (workpersons <= 0 && workvertices.size() > 0)
		{
			drawTextWithBackgroundWstr(frame,"离 岗", 2, cv::Point(workvertices[0].x, workvertices[0].y), cv::Point(workvertices[1].x - workvertices[0].x, workvertices[1].y - workvertices[0].y), Scalar(255, 255, 0));

			// 绘制多边形，工作区域
			cv::polylines(frame, workvertices, true, cv::Scalar(255, 255, 0), 2);
		}		

		frame.release();

		result.push_back(distance);//有人进入危险区域
		result.push_back(wearHat);//未带安全帽人员数量
		result.push_back(workpersons);//工作区域人员数量

		return result;
	}
	catch (const std::exception& ex)
	{
		std::string errorMessage = "fnImRecProByModelHAT-";
		errorMessage += ex.what();
		cout << errorMessage << endl;
		throw std::runtime_error(errorMessage);
	}
}

///摄像头遮挡算法 modelId = 4
double PreProcessModel::fnCameraCoverRec(Mat imMat, Cover cover)
{
	try
	{
		cv::Mat image, standard;
		image = imMat;//被判断图
		int result = cover.detect(image);
		return result;//返回1是遮挡，0是不遮挡
	}
	catch (const std::exception& ex)
	{
		std::string errorMessage = "fnCameraCoverRec-";
		errorMessage += ex.what();
		cout << errorMessage << endl;
		throw std::runtime_error(errorMessage);
	}
}

///摄像头移动算法 modelId = 5
double PreProcessModel::fnCameraMoveRec(Mat imMat, Camera cam)
{
	try
	{
		cv::Mat image, standard;
		image = imMat;//被判断图
		standard = cam.stdMat;//参考图
		int result = cam.detect(image, standard);
		return result;//返回1是移动，0是不移动
	}
	catch (const std::exception& ex)
	{
		std::string errorMessage = "fnCameraMoveRec-";
		errorMessage += ex.what();
		cout << errorMessage << endl;
		throw std::runtime_error(errorMessage);
	}
}

/// <summary>
/// 堆煤检测,返回煤炭占比 modelId = 6
/// </summary>
/// <param name="imMat">视频流Mat</param>
/// <param name="dm">模型引用</param>
double PreProcessModel::fnImRecProByModelDM(Mat& imMat, DMJC& dm, std::vector<cv::Point> vertices)
{
	try
	{
		double result = 0;
		// 定义接受帧
		cv::Mat frame = imMat;		
		// 调用煤流模型后的返回图片
		cv::Mat frameout;	

		// 外层矩形区域
		cv::Rect rect(vertices[0].x, vertices[0].y, vertices[2].x - vertices[0].x, vertices[2].y - vertices[0].y);
		// 截取外层矩形区域
		cv::Mat croppedFrame = frame(rect).clone();	

		//result = dm.detect(frame);// 返回指定区域煤占比	
		result = dm.detect_pic(croppedFrame, frameout);
		// 合并图片
		// 在大图像上定义ROI（感兴趣区域）
		cv::Mat roi = frame(cv::Rect(vertices[0].x, vertices[0].y, frameout.cols, frameout.rows));

		// 将小图像复制到ROI
		frameout.copyTo(roi);

		// 煤炭占比
		drawTextWithBackgroundWstr(frame, std::to_string(result), 0, cv::Point(vertices[0].x, vertices[0].y), cv::Point(vertices[1].x - vertices[0].x, vertices[1].y - vertices[0].y), Scalar(255, 255, 0));

		// 在视频帧上添加矩形框  
		//rectangle(frame, cv::Point(vertices[0].x, vertices[0].y), cv::Point(vertices[2].x, vertices[3].y), cv::Scalar(255, 255, 0), 2);

		croppedFrame.release();
		frameout.release();
		return result;
	}
	catch (const std::exception& ex)
	{
		std::string errorMessage = "fnImRecProByModelDM-";
		errorMessage += ex.what();
		cout << errorMessage << endl;
		throw std::runtime_error(errorMessage);
	}
}

/// <summary>
/// 皮带跑偏，大块异物 modelId = 7-1
/// </summary>
/// <param name="imMat">视频流Mat</param>
/// <param name="hat">模型引用</param>
vector<double> PreProcessModel::fnImRecProByModelBeltJC(Mat& imMat, HAT& hat, std::vector<cv::Point> vertices)
{
	try
	{
		vector<double> result;
		cv::Scalar sGreen = Scalar(0, 255, 0);;//绿色框
		cv::Scalar sRed = Scalar(0, 0, 255);//红色框	
		cv::Scalar sPurple = Scalar(255, 0, 255);//紫色框	

		//定义接受帧
		cv::Mat frame = imMat;

		//获取视频帧的宽度和高度和帧率
		int frame_width = frame.cols;
		int frame_height = frame.rows;
		double frameRate = 25;

		// 创建一个空的边界框向量
		std::vector<Box> boxes;
		// 调用 detect 函数获取检测到的边界框
		hat.detect(frame, boxes);	

		//多边形区域
		//std::vector<cv::Point> vertices = stringToPoints(area);
		// 绘制多边形
		cv::polylines(frame, vertices, true, sGreen, 2);

		// 定义要测试的点
		cv::Point testPoint;
		cv::Scalar sl;//标框颜色

		// 使用索引遍历boxes
		for (int i = 0; i < boxes.size(); i++) {
			switch (boxes[i].class_id)
			{
			case 0:
				//人形中心点
				testPoint.x = boxes[i].x + boxes[i].width / 2;
				testPoint.y = boxes[i].y + boxes[i].height / 2;

				if (cv::pointPolygonTest(vertices, testPoint, false) > 0)//区域内有个人,画红色
				{
					//标注大媒块
					drawTextWithBackgroundWstr(frame, "大 煤 块", 2, cv::Point(boxes[i].x, boxes[i].y), cv::Point(boxes[i].width, boxes[i].height), sPurple);
					//在视频帧上添加矩形框 
					rectangle(frame, cv::Point(boxes[i].x, boxes[i].y), cv::Point((boxes[i].x + boxes[i].width), (boxes[i].y + boxes[i].height)), sPurple, 1);
				}				

				break;
			case 2:
			case 3:
			case 4:
			case 5:
				//人形中心点
				testPoint.x = boxes[i].x + boxes[i].width / 2;
				testPoint.y = boxes[i].y + boxes[i].height / 2;

				if (cv::pointPolygonTest(vertices, testPoint, false) > 0)//区域内有个人,画红色
				{
					//标注异物
					drawTextWithBackgroundWstr(frame, "异物", 2, cv::Point(boxes[i].x, boxes[i].y), cv::Point(boxes[i].width, boxes[i].height), sRed);
					//在视频帧上添加矩形框 
					rectangle(frame, cv::Point(boxes[i].x, boxes[i].y), cv::Point((boxes[i].x + boxes[i].width), (boxes[i].y + boxes[i].height)), sRed, 1);
				}
				break;
			default:
				break;
			}
		}

		frame.release();

		return result;
	}
	catch (const std::exception& ex)
	{
		std::string errorMessage = "fnImRecProByModelBeltJC-";
		errorMessage += ex.what();
		cout << errorMessage << endl;
		throw std::runtime_error(errorMessage);
	}
}

/// <summary>
/// 皮带跑偏，大块异物,新处理方法 modelId = 7-2
/// </summary>
/// <param name="imMat">视频流Mat</param>
/// <param name="hat">模型引用</param>
vector<double> PreProcessModel::fnImRecProByModelBeltJC2(Mat& imMat, BELT& belt, std::vector<cv::Point> leftvertices, std::vector<cv::Point> rightvertices,std::vector<cv::Point> vertices)
{
	try
	{
		vector<double> result;
		int bigcoal = 0;//0:无大煤块 1:有大煤块
		int unbalance = 1;//0:偏载 1:不偏载
		int foreign = 0;//0:无异物 1:有异物

		cv::Scalar sGreen = Scalar(0, 255, 0);;//绿色框
		cv::Scalar sRed = Scalar(0, 0, 255);//红色框	
		cv::Scalar sPurple = Scalar(255, 0, 255);//紫色框	

		// 画图
		cv::Scalar roller_color(238, 130, 238); // 托辊区域线颜色
		cv::Scalar roller_color2(203, 192, 255); // 托辊框的颜色
		cv::Scalar belt_color(255, 0, 255); // 皮带区域线颜色
		cv::Scalar belt_color2(0, 0, 255); // 异物框的颜色

		//定义接受帧
		cv::Mat frame = imMat;

		//获取视频帧的宽度和高度和帧率
		int frame_width = frame.cols;
		int frame_height = frame.rows;
		double frameRate = 25;

		// 创建一个空的边界框向量
		std::vector<Box> boxes;
		// 调用 detect 函数获取检测到的边界框
		unbalance = belt.detect(frame, boxes);

		//多边形区域
		//std::vector<cv::Point> vertices = stringToPoints(area);
		// 绘制多边形
		//cv::polylines(frame, vertices, true, sGreen, 2);

		polylines(frame, leftvertices, true, roller_color, 5, 8, 0);
		polylines(frame, rightvertices, true, roller_color, 5, 8, 0);
		polylines(frame, vertices, true, belt_color, 5, 8, 0);

		// 定义要测试的点
		cv::Point testPoint;
		cv::Scalar sl;//标框颜色

		// 使用索引遍历boxes
		for (int i = 0; i < boxes.size(); i++) 
		{

			// 设置文本起始坐标（左下角坐标）
			cv::Point textOrg(boxes[i].x - boxes[i].width / 2, boxes[i].y - boxes[i].height / 2 - 10);

			switch (boxes[i].class_id)
			{
			case 0://大煤块
				//中心点
				testPoint.x = boxes[i].x + boxes[i].width / 2;
				testPoint.y = boxes[i].y + boxes[i].height / 2;

				if (cv::pointPolygonTest(vertices, testPoint, false) > 0)//进入区域后再画框
				{
					//标注大煤块
					drawTextWithBackgroundWstr(frame, "大煤块", 2, cv::Point(boxes[i].x, boxes[i].y), cv::Point(boxes[i].width, boxes[i].height), sPurple);
					//在视频帧上添加矩形框 
					rectangle(frame, cv::Point(boxes[i].x, boxes[i].y), cv::Point((boxes[i].x + boxes[i].width), (boxes[i].y + boxes[i].height)), sPurple, 1);
				}

				bigcoal = 1;
				break;
			case 1:
				cv::rectangle(frame, cv::Point(boxes[i].x - boxes[i].width / 2, boxes[i].y - boxes[i].height / 2), cv::Point(boxes[i].x + boxes[i].width / 2, boxes[i].y + boxes[i].height / 2), roller_color2, 3);
				//标注托辊
				drawTextWithBackgroundWstr(frame, "托辊", 2, cv::Point(boxes[i].x - boxes[i].width / 2, boxes[i].y - boxes[i].height / 2), cv::Point(boxes[i].width, boxes[i].height), sPurple);
				break;
			case 2:
			case 3:
			case 4:
			case 5:
				//中心点
				testPoint.x = boxes[i].x + boxes[i].width / 2;
				testPoint.y = boxes[i].y + boxes[i].height / 2;

				if (cv::pointPolygonTest(vertices, testPoint, false) > 0)//区域内,画红色
				{
					//标注异物
					drawTextWithBackgroundWstr(frame, "异物", 2, cv::Point(boxes[i].x, boxes[i].y), cv::Point(boxes[i].width, boxes[i].height), sRed);
					//在视频帧上添加矩形框 
					rectangle(frame, cv::Point(boxes[i].x, boxes[i].y), cv::Point((boxes[i].x + boxes[i].width), (boxes[i].y + boxes[i].height)), sRed, 1);
				}

				foreign = 1;
				break;
			default:
				break;
			}
		}

		frame.release();

		//foreign = 1;//test

		result.push_back(bigcoal);//大煤块
		result.push_back(unbalance);//偏载
		result.push_back(foreign);//异物		

		return result;
	}
	catch (const std::exception& ex)
	{
		std::string errorMessage = "fnImRecProByModelBeltJC2-";
		errorMessage += ex.what();
		cout << errorMessage << endl;
		throw std::runtime_error(errorMessage);
	}
}

/// <summary>
/// 乘猴车携带大件 modelId = 8
/// </summary>
/// <param name="imMat">视频流Mat</param>
/// <param name="hat">模型引用</param>
vector<double> PreProcessModel::fnImRecProByModelHATHC(Mat& imMat, HAT& hat, std::vector<cv::Point> vertices)
{
	try
	{
		vector<double> result;

		//定义接受帧
		cv::Mat frame = imMat;

		//获取视频帧的宽度和高度和帧率
		int frame_width = frame.cols;
		int frame_height = frame.rows;
		double frameRate = 25;

		// 创建一个空的边界框向量
		std::vector<Box> boxes;
		// 调用 detect 函数获取检测到的边界框
		hat.detect(frame, boxes);

		//多边形区域
		//std::vector<cv::Point> vertices = stringToPoints(area);

		// 绘制多边形
		//cv::polylines(frame, vertices, true, cv::Scalar(0, 0, 255), 2);

		// 定义要测试的点
		cv::Scalar sl;//标框颜色

		// 使用索引遍历boxes
		for (int i = 0; i < boxes.size(); i++) {
			
			switch (boxes[i].class_id)
			{
			case 0:					
				sl = Scalar(0, 0, 255);

				//标注人形
				drawTextWithBackgroundWstr(frame, "携 带 大 件", 2, cv::Point(boxes[i].x, boxes[i].y), cv::Point(boxes[i].width, boxes[i].height), sl);

				//在视频帧上添加矩形框 
				rectangle(frame, cv::Point(boxes[i].x, boxes[i].y), cv::Point((boxes[i].x + boxes[i].width), (boxes[i].y + boxes[i].height)), sl, 1);

				break;
			//case 1:
			//	//标注帽子
			//	drawTextWithBackgroundWstr(frame, "安全帽", 2, cv::Point(boxes[i].x, boxes[i].y), cv::Point(boxes[i].width, boxes[i].height), cv::Scalar(0, 255, 0));
			//	//在视频帧上添加矩形框 
			//	rectangle(frame, cv::Point(boxes[i].x, boxes[i].y), cv::Point((boxes[i].x + boxes[i].width), (boxes[i].y + boxes[i].height)), cv::Scalar(0, 255, 0), 1);

			//	break;
			//case 2:

			//	wearHat++;//未佩戴安全帽的人的数量
			//	//标注头部
			//	drawTextWithBackgroundWstr(frame, "头", 3, cv::Point(boxes[i].x, boxes[i].y), cv::Point(boxes[i].width, boxes[i].height), cv::Scalar(0, 255, 0));
			//	rectangle(frame, cv::Point(boxes[i].x, boxes[i].y), cv::Point((boxes[i].x + boxes[i].width), (boxes[i].y + boxes[i].height)), cv::Scalar(0, 255, 0), 1);

			//	break;
			default:
				break;
			}
		}

		frame.release();
		
		result.push_back(boxes.size());//返回携带大件人数

		return result;
	}
	catch (const std::exception& ex)
	{
		std::string errorMessage = "fnImRecProByModelHAT-";
		errorMessage += ex.what();
		cout << errorMessage << endl;
		throw std::runtime_error(errorMessage);
	}
}

/// <summary>
/// 睡岗处理 modelId = 9
/// </summary>
/// <param name="imMat">视频流Mat</param>
/// <param name="hat">模型引用</param>
vector<double> PreProcessModel::fnImRecProByModelHATSleep(Mat& imMat, HAT& hat)
{
	try
	{
		vector<double> result;
		//定义接受帧
		cv::Mat frame = imMat;

		//获取视频帧的宽度和高度和帧率
		int frame_width = frame.cols;
		int frame_height = frame.rows;

		// 创建一个空的边界框向量
		std::vector<Box> boxes;
		// 调用 detect 函数获取检测到的边界框
		hat.detect(frame, boxes);

		// 定义要测试的点
		cv::Scalar sl;//标框颜色

		// 使用索引遍历boxes
		for (int i = 0; i < boxes.size(); i++) {

			sl = Scalar(0, 0, 255);

			//标注人形
			drawTextWithBackgroundWstr(frame, "睡 岗", 2, cv::Point(boxes[i].x, boxes[i].y), cv::Point(boxes[i].width, boxes[i].height), sl);

			//在视频帧上添加矩形框 
			rectangle(frame, cv::Point(boxes[i].x, boxes[i].y), cv::Point((boxes[i].x + boxes[i].width), (boxes[i].y + boxes[i].height)), sl, 1);
		}

		frame.release();

		result.push_back(boxes.size());//返回睡岗人员的数量

		return result;
	}
	catch (const std::exception& ex)
	{
		std::string errorMessage = "fnImRecProByModelHATSleep-";
		errorMessage += ex.what();
		cout << errorMessage << endl;
		throw std::runtime_error(errorMessage);
	}
}

/// <summary>
/// 人员穿戴
/// </summary>
/// <param name="imMat">视频流Mat</param>
/// <param name="hat">模型引用</param>

vector<double> PreProcessModel::fnImRecProByModelWear(Mat& imMat, HAT& hatWear, clasification& cfWear,std::vector<cv::Point> vertices)
{
	try
	{
		vector<double> result;
		string title = "人员";

		double checkpersons = 0;//检测区域正在检测的人员数量,默认是0
		double resclothes = 0;//工作服 0：不穿工作服 1：穿工作服
		double reshat = 0;//安全吗 0：不带安全帽 1：带安全帽
		double respager = 0;//自救器 0：不携带自救器 1：带自救器
		double resshoes = 0;//鞋 0：不穿鞋 1：穿鞋
		//定义接受帧
		cv::Mat frame = imMat;

		// 创建一个空的边界框向量
		std::vector<Box> boxes;
		// 调用 detect 函数获取检测到的边界框
		hatWear.detect(frame, boxes);		

		// 定义要测试的点
		cv::Point testPoint;
		cv::Scalar sl = Scalar(0, 255, 0);//默认标框颜色

		// 设置原图中四点的坐标
		std::vector<cv::Point2f> srcPoints;
		// 设置目标图像的大小和四个点的坐标
		std::vector<cv::Point2f> dstPoints;
		// 计算透视变换矩阵
		cv::Mat transform;
		// 应用透视变换
		cv::Mat dst;

		// 使用索引遍历boxes
		for (int i = 0; i < boxes.size(); i++) 
		{
			//标框中心点
			testPoint.x = boxes[i].x + boxes[i].width / 2;
			testPoint.y = boxes[i].y + boxes[i].height / 2;					

			switch (boxes[i].class_id)
			{
			case 0://人				

				if (cv::pointPolygonTest(vertices, testPoint, false) > 0)//进入检测区域内有人
				{
					title = "人";
					//personflg = true;//检测到人进入检测区域
					checkpersons = 1;

					//标注人形
					drawTextWithBackgroundWstr(frame, title, 2, cv::Point(boxes[i].x, boxes[i].y), cv::Point(boxes[i].width, boxes[i].height), sl);
					//在视频帧上添加矩形框 
					rectangle(frame, cv::Point(boxes[i].x, boxes[i].y), cv::Point((boxes[i].x + boxes[i].width), (boxes[i].y + boxes[i].height)), sl, 1);


					//截取人员图片处理
					srcPoints.push_back(cv::Point2f(boxes[i].x, boxes[i].y)); // 左上角
					srcPoints.push_back(cv::Point2f(boxes[i].x + boxes[i].width, boxes[i].y)); // 右上角
					srcPoints.push_back(cv::Point2f(boxes[i].x + boxes[i].width, boxes[i].y + boxes[i].height)); // 右下角
					srcPoints.push_back(cv::Point2f(boxes[i].x, boxes[i].y + boxes[i].height)); // 左下角

					dstPoints.push_back(cv::Point2f(0, 0));
					dstPoints.push_back(cv::Point2f(boxes[i].width, 0));
					dstPoints.push_back(cv::Point2f(boxes[i].width, boxes[i].height));
					dstPoints.push_back(cv::Point2f(0, boxes[i].height));

					// 计算透视变换矩阵
					transform = cv::getPerspectiveTransform(srcPoints, dstPoints);
					cv::warpPerspective(frame, dst, transform, cv::Size(boxes[i].width, boxes[i].height));									
				
				}					

				break;
			case 1://安全帽
				
				if (cv::pointPolygonTest(vertices, testPoint, false) > 0)//是否在检测区域内
				{
					title = "安全帽";
					reshat = 1;

					//标注帽子
					drawTextWithBackgroundWstr(frame, title, 2, cv::Point(boxes[i].x, boxes[i].y), cv::Point(boxes[i].width, boxes[i].height), cv::Scalar(0, 255, 0));
					//在视频帧上添加矩形框 
					rectangle(frame, cv::Point(boxes[i].x, boxes[i].y), cv::Point((boxes[i].x + boxes[i].width), (boxes[i].y + boxes[i].height)), cv::Scalar(0, 255, 0), 1);
				}
				
				break;
			//case 2://头部				

			//	//标注头部
			//	drawTextWithBackgroundWstr(frame, "头", 2, cv::Point(boxes[i].x, boxes[i].y), cv::Point(boxes[i].width, boxes[i].height), cv::Scalar(0, 255, 0));
			//	rectangle(frame, cv::Point(boxes[i].x, boxes[i].y), cv::Point((boxes[i].x + boxes[i].width), (boxes[i].y + boxes[i].height)), cv::Scalar(0, 255, 0), 1);

			//	break;
			case 3://自救器
				
				if (cv::pointPolygonTest(vertices, testPoint, false) > 0)//是否在检测区域内
				{
					title = "自救器";
					respager = 1;

					//标注自救器
					drawTextWithBackgroundWstr(frame, title, 2, cv::Point(boxes[i].x, boxes[i].y), cv::Point(boxes[i].width, boxes[i].height), cv::Scalar(0, 255, 0));
					rectangle(frame, cv::Point(boxes[i].x, boxes[i].y), cv::Point((boxes[i].x + boxes[i].width), (boxes[i].y + boxes[i].height)), cv::Scalar(0, 255, 0), 1);
				}
				
				break;
			case 4://鞋				

				if (cv::pointPolygonTest(vertices, testPoint, false) > 0)//是否在检测区域内
				{
					title = "鞋";
					resshoes = 1;

					//标注鞋
					drawTextWithBackgroundWstr(frame, title, 2, cv::Point(boxes[i].x, boxes[i].y), cv::Point(boxes[i].width, boxes[i].height), cv::Scalar(0, 255, 0));
					rectangle(frame, cv::Point(boxes[i].x, boxes[i].y), cv::Point((boxes[i].x + boxes[i].width), (boxes[i].y + boxes[i].height)), cv::Scalar(0, 255, 0), 1);
				}

				break;
			default:
				break;
			}
		}

		if (!dst.empty())
		{			
			if (cfWear.detect(dst) == 0)
			{
				resclothes = 1;//穿了工作服
			}
			else
			{
				resclothes = 0;//未穿工作服
			}

			//cv::imshow("Cropped Image", dst);
			//waitKey(10);
		}		

		// 绘制多边形检测区域
		cv::polylines(frame, vertices, true, cv::Scalar(0, 0, 255), 2);

		//cv::imshow("Cropped Image", frame);
		//waitKey(10);
		frame.release();

		result.push_back(checkpersons);//检测区域人员数量
		result.push_back(reshat);//安全帽
		result.push_back(respager);//自救器
		result.push_back(resshoes);//鞋
		result.push_back(resclothes);//工作服

		return result;
	}
	catch (const std::exception& ex)
	{
		std::string errorMessage = "fnImRecProByModelHATWear-";
		errorMessage += ex.what();
		cout << errorMessage << endl;
		throw std::runtime_error(errorMessage);
	}
}

/// <summary>
/// 烟火
/// </summary>
/// <param name="imMat">视频流Mat</param>
/// <param name="hat">模型引用</param>
vector<double> PreProcessModel::fnImRecProByModelFire(Mat& imMat, HAT& hat)
{
	try
	{
		vector<double> result;
		double fire = 0;// 火焰数量
		double smoke = 0;//烟数量
		//定义接受帧
		cv::Mat frame = imMat;

		//获取视频帧的宽度和高度和帧率
		int frame_width = frame.cols;
		int frame_height = frame.rows;

		// 创建一个空的边界框向量
		std::vector<Box> boxes;
		// 调用 detect 函数获取检测到的边界框
		hat.detect(frame, boxes);

		// 定义要测试的点
		cv::Scalar sl;//标框颜色
		sl = Scalar(0, 0, 255);

		// 使用索引遍历boxes
		for (int i = 0; i < boxes.size(); i++) 
		{

			switch (boxes[i].class_id)
			{
			case 0:
				fire++;
				//标注火
				drawTextWithBackgroundWstr(frame, "火", 2, cv::Point(boxes[i].x, boxes[i].y), cv::Point(boxes[i].width, boxes[i].height), sl);

				//在视频帧上添加矩形框 
				rectangle(frame, cv::Point(boxes[i].x, boxes[i].y), cv::Point((boxes[i].x + boxes[i].width), (boxes[i].y + boxes[i].height)), sl, 1);
				break;
			case 1:
				smoke++;
				//标注烟
				drawTextWithBackgroundWstr(frame, "烟", 2, cv::Point(boxes[i].x, boxes[i].y), cv::Point(boxes[i].width, boxes[i].height), sl);

				//在视频帧上添加矩形框 
				rectangle(frame, cv::Point(boxes[i].x, boxes[i].y), cv::Point((boxes[i].x + boxes[i].width), (boxes[i].y + boxes[i].height)), sl, 1);
				break;
			default:
				break;
			}			
		}

		frame.release();

		result.push_back(fire);//返回火数量
		result.push_back(smoke);//返回烟数量

		return result;
	}
	catch (const std::exception& ex)
	{
		std::string errorMessage = "fnImRecProByModelHATSleep-";
		errorMessage += ex.what();
		cout << errorMessage << endl;
		throw std::runtime_error(errorMessage);
	}
}