#pragma once
#include <iostream>
#include <opencv2/opencv.hpp>
#include <ctime>
#include <thread>
#include <stdio.h>
#include <cuda_runtime.h>
#include <locale>
#include <codecvt>
#include <json/json.h>
#include <assert.h>
#include <fstream>
#include <string>

#include "base/CvxText.h"
#include "deploy/yolov5.hpp"//目标识别
#include "deploy/DMJC.h"//堆煤检测
#include "deploy/run.h"
#include "deploy/cover.h"
#include "deploy/camera.h"
#include "deploy/belt.hpp"
#include "deploy/clasi.h"

using namespace cv;
using namespace std;
using namespace Json;

class PreProcessModel
{
public:
	//cv::Mat 
	//void drawTextWithBackground(Mat& image, const string& text, int txalign, const Point& orgxy, const Point& orgwh, Scalar bgColor);
	vector<double> fnImRecProByModelHAT(Mat& imMat, HAT& hat, std::vector<cv::Point> vertices, std::vector<cv::Point> workvertices);//人员闯入检测
	vector<double> 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);
	vector<double> fnImRecProByModelHATHC(Mat& imMat, HAT& hat, std::vector<cv::Point> vertices);//猴车检查
	vector<double> fnImRecProByModelHATSleep(Mat& imMat, HAT& hat);//睡岗
	vector<double> fnImRecProByModelFire(Mat& imMat, HAT& hat);//烟火
	
	vector<double> fnImRecProByModelBeltJC(Mat& imMat, HAT& hat, std::vector<cv::Point> vertices);//皮带检测
	vector<double> fnImRecProByModelBeltJC2(Mat& imMat, BELT& belt, std::vector<cv::Point> leftvertices, std::vector<cv::Point> rightvertices, std::vector<cv::Point> vertices);//皮带跑偏，异物检测

	double fnImRecProByModelDM(Mat& imMat, DMJC& dm, std::vector<cv::Point> vertices);//堆煤检测
	double fnImRecProByModelTSJ(Mat& imMat, std::vector<cv::Mat>& imagesTsj, int& tsjCount, RUN& rTSJ,std::queue<int>& quTSJ, std::vector<cv::Point> vertices);//提升机检测
	double fnCameraCoverRec(Mat imMat,Cover cover);//摄像头遮挡检测
	double fnCameraMoveRec(Mat imMat, Camera cam);//摄像头移动检测
	double fnBeltMoveRec(Mat& imMat, std::vector<cv::Mat>& imagesBelt, int& beltCount, RUN& rTSJ, std::vector<cv::Point> vertices);//皮带运行状态
	vector<double> fnImRecProByModelWear(Mat& imMat,  HAT& hat, clasification& cfWear, std::vector<cv::Point> vertices);//人员穿戴检测

};