OCVProc.h

#pragma once
#include "Helpers.h"
#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/videoio.hpp"
#include "opencv2/core.hpp"
#include "opencv2/calib3d.hpp"
#include <chrono>

#include <wx/wxprec.h>
#ifndef WX_PRECOMP
#include <wx/wx.h>
#endif
constexpr auto defaultIP = "http://192.168.x.x:4747/video";

class OCVProc
{
private:
	const std::vector<cv::Point2f> transformPointsIMG = {
		cv::Point(0,0),
		cv::Point(480,0),
		cv::Point(480,480),
		cv::Point(0,480)
	};
	const std::vector<cv::Point2f> transformPointsMCU = {
		cv::Point(0,255),
		cv::Point(255,255),
		cv::Point(255,0),
		cv::Point(0,0)
	};
	HDC drawingDC = 0;
	wxPanel* streamCanvas = nullptr;
	bool stream = false;
	bool mirror = false;
	int rotation = 3;
	cv::Mat framePreProc,afterTransform, framePostProc;
	cv::Size cameraRes;
	cv::VideoCapture camera;
	std::thread cameraStream;
	void InitCameraRes();
	void previewLoop();
	std::vector<std::vector<cv::Point>> FindQuadsInMat(cv::Mat& mat);
	double angle(cv::Point pt1, cv::Point pt2, cv::Point pt0);
	bool isQuad(std::vector<cv::Point> contours);
	void removeDoubleQuads(std::vector<std::vector<cv::Point>>& quads);
	void drawMatToDC(cv::Mat& mat);
public:
	static cv::Point2f quadCenter(std::vector<cv::Point> corners);
	void SetStreamCanvas(wxPanel* canvas);
	bool OpenCamera(int selectedCameraIndex);
	bool OpenCamera(std::string cameraIP);
	bool IsStreaming();
	bool IsOpen();
	wxSize GetCameraResolution();
	void rotateStream();
	void mirrorStream();
	void StartCameraStream();
	void StopCameraStream();
	std::vector<cv::Point2f> ProcessImage();
	void sortCorners(std::vector<cv::Point>& corners);
	std::vector<cv::Point> removeBiggestQuad(std::vector<std::vector<cv::Point>>& quads);
	OCVProc();
	~OCVProc();
};