//
// Created by UnknownObject on 2022/11/4.
// Copyright (c) 2022 UnknownNetworkService. All rights reserved.
//

//OCR的图像预处理算法

#include "ocr_text.h"

namespace uns
{
	//像素检查，limit阈值是黑色的浓淡（文字是图片中最黑的部分）
	bool OCRSupport::PixelCheck(const cv::Vec3b &pixel)
	{
		if (pixel[0] <= limit)
			if (pixel[1] <= limit)
				if (pixel[2] <= limit)
					return true;
		return false;
	}
	
	//图片裁切
	cv::Mat OCRSupport::CutImageSize(const cv::Mat &img)
	{
		return img(roi);
	}
	
	//图片预处理，将彩色图片转换为黑白图片，确保文字是黑色的
	cv::Mat OCRSupport::ConvertImage(const cv::Mat &img)
	{
		cv::imwrite("/sdcard/MainCar/OCR/ocr_old.jpg", img);
		cv::Mat result(img.size(), img.type());
		for (int r = 0; r < img.rows; r++)
		{
			for (int c = 0; c < img.cols; c++)
			{
				if (PixelCheck(img.at<cv::Vec3b>(r, c)))
					result.at<cv::Vec3b>(r, c) = cv::Vec3b(0, 0, 0);
				else
					result.at<cv::Vec3b>(r, c) = cv::Vec3b(255, 255, 255);
			}
		}
		cv::imwrite("/sdcard/MainCar/OCR/ocr_process.jpg", result);
		return result;
	}
};

//导出的自检函数
extern "C" JNIEXPORT
jstring JNICALL Java_com_uns_maincar_cpp_1interface_EnvTest_OCRTextTest(JNIEnv *env, jclass _this)
{
	std::string version = "OCR Text Library Found, Version: " + std::string(OCR_TEXT_VERSION);
	return env->NewStringUTF(version.c_str());
}

//导出的图像预处理函数
extern "C" JNIEXPORT
jobject JNICALL
Java_com_uns_maincar_cpp_1interface_OCR_ProcessImage(JNIEnv *env, jclass _this, jobject image,
													 jboolean self_test)
{
	cv::Mat source;
	uns::OCRSupport ocr_supp;
	BitmapToMat(env, image, source);
	cv::Mat img = (self_test ? ocr_supp.ConvertImage(source) : ocr_supp.ConvertImage(
			ocr_supp.CutImageSize(source)));
	jobject bmp = GenerateBitmap(env, img.cols, img.rows);
	MatToBitmap(env, img, bmp);
	return bmp;
}