// DLLUsageDemo.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。
//

#include <iostream>
#include "LibEasyOCR-CPP.h"
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgproc/imgproc.hpp>

int main()
{
	std::wstring detect_model = L"..\\EasyOCR-CPP\\DetectionModel.onnx";
	std::wstring recognize_model = L"..\\EasyOCR-CPP\\RecognitionModel-EN.onnx";
	std::string image = "..\\..\\vin.jpg";

	uns::EasyOCR_CPP::GlobalOCRConfig().SetDetectModelPath(detect_model);
	uns::EasyOCR_CPP::GlobalOCRConfig().SetGPUUsage(uns::easyocr::GPUUsage::CPUOnly);
	uns::EasyOCR_CPP::SetRecognitionModel(recognize_model, uns::easyocr::CharsetType::EN);

	if (!uns::EasyOCR_CPP::InitDetectionModel())
	{
		std::cout << "Detector Init Failure!" << std::endl;
		return -1;
	}
	if (!uns::EasyOCR_CPP::InitRecognitionModel())
	{
		std::cout << "Recognizer Init Failure!" << std::endl;
		return -2;
	}

	cv::Mat img = uns::EOCR_SupportTools::ReadImage(image, true);
	if (img.empty())
	{
		std::cerr << "Failed to load image: " << image << std::endl;
		return -1;
	}

	auto results = uns::EasyOCR_CPP::FullAuto(img);
	for (const auto& [index, info] : results)
	{
		const auto& [text, conf, rect] = info;
		printf("Result %d: Text = {%s}, Box = [%d, %d, %d, %d], Confidence = %.3f\n", (int)index, uns::EOCR_SupportTools::WtoA(text).c_str(), rect.tl().x, rect.tl().y, rect.br().x, rect.br().y, conf);
	}

	uns::EasyOCR_CPP::CleanupOCR();
	return 0;
}