#pragma once
#include <map>
#include <vector>
#include <string>
#include <opencv2/core/core.hpp>

#ifdef LIBEASYOCRCPP_EXPORTS
#define LIBEASYOCRCPP_API _declspec(dllexport)
#else
#define LIBEASYOCRCPP_API _declspec(dllimport)
#ifdef _DEBUG
#pragma comment(lib, "LibEasyOCR-CPPd.lib")
#else
#pragma comment(lib, "LibEasyOCR-CPP.lib")
#endif
#endif

namespace uns
{
	namespace easyocr
	{
		enum class LIBEASYOCRCPP_API GPUUsage
		{
			ForceGPU,
			PreferGPU,
			CPUOnly
		};

		enum class LIBEASYOCRCPP_API CharsetType
		{
			EN,
			EN_CH
		};

		using EOCR_Result = std::tuple<std::wstring, float>;
		using EOCR_Result_Single = std::tuple<std::wstring, float, cv::Rect>;
		using EOCR_ResultSet = std::map<size_t, EOCR_Result_Single>;
		using EOCRD_Rects = std::vector<std::vector<cv::Point2f>>;
	}

	class LIBEASYOCRCPP_API OCRConfig
	{
	public:
		OCRConfig();
		OCRConfig(const OCRConfig& obj) = delete;
		OCRConfig(const std::wstring& detect_model, const std::wstring& reco_model, easyocr::CharsetType language, easyocr::GPUUsage gpu = easyocr::GPUUsage::PreferGPU);

	public:
		easyocr::GPUUsage GetGPUUsage() const;
		easyocr::CharsetType GetLanguage() const;

		std::wstring GetDetectModelPath() const;
		std::wstring GetRecognizeModelPath() const;

	public:
		void SetGPUUsage(easyocr::GPUUsage usage);
		void SetLanguage(easyocr::CharsetType type);
		void SetDetectModelPath(const std::wstring& path);
		void SetRecognizeModelPath(const std::wstring& path);
	};

	class LIBEASYOCRCPP_API EasyOCR_CPP
	{
	public:
		static OCRConfig& GlobalOCRConfig();
		static void SetRecognitionModel(const std::wstring& reco_model, easyocr::CharsetType language);

		static bool CleanupOCR();
		static bool InitDetectionModel();
		static bool InitRecognitionModel();

		static easyocr::EOCRD_Rects Detect(const cv::Mat& img);
		static easyocr::EOCR_Result Recognize(const cv::Mat& img);
		static easyocr::EOCR_ResultSet Recognize(const cv::Mat& img, const easyocr::EOCRD_Rects& rects);

		static easyocr::EOCR_ResultSet FullAuto(const cv::Mat& img);
	};

	class LIBEASYOCRCPP_API EOCR_SupportTools
	{
	public:
		static std::string WtoA(const std::wstring& wstr);
		static std::wstring AtoW(const std::string& str);
		static std::string AtoUTF8(const std::string& str);
		static std::string UTF8toA(const std::string& utf8);

		static cv::Mat ReadImage(const std::string& file, bool blur = false, int blur_size = 3);
		static cv::Mat ReadImage(const std::wstring& file, bool blur = false, int blur_size = 3);
	};
}