EasyOCR-CPP/LibEasyOCR-CPP/OCRConfig.h

#pragma once
#include <string>
#include "LibEasyOCR-CPP-Export.h"

namespace uns
{
	namespace easyocr
	{
		enum class LIBEASYOCRCPP_API GPUUsage
		{
			ForceGPU,     // 强制使用GPU（失败报错）
			PreferGPU,    // 优先GPU，失败回退
			CPUOnly       // 仅使用CPU
		};

		enum class LIBEASYOCRCPP_API CharsetType
		{
			EN,		//英语
			EN_CH	//英语和中文
		};
	}

	class LIBEASYOCRCPP_API OCRConfig
	{
	private:
		easyocr::GPUUsage gpu_usage;
		easyocr::CharsetType language;
		std::wstring detect_model_path;
		std::wstring recognize_model_path;

	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);
	};

	extern OCRConfig G_OCRConfig;
}