using PaddleOCRSharp;

namespace OCRTest02;

public class OCRHelper
{
    /// <summary>
    /// 初始化OCR引擎
    /// </summary>
    /// <returns></returns>
    public static PaddleOCREngine initOcrEngine(string exePath)
    {
        //OCR參數(shù)
        OCRParameter oCRParameter = new OCRParameter();
        oCRParameter.cpu_math_library_num_threads = 6;//預(yù)測并發(fā)線程數(shù),CPU預(yù)測時的線程數(shù)，在機器核數(shù)充足的情況下，該值越大，預(yù)測速度越快；默認10
        oCRParameter.enable_mkldnn = true;//web部署該值建議設(shè)置為0,否則出錯，內(nèi)存如果使用很大，建議該值也設(shè)置為0.
        oCRParameter.cls = true; //是否執(zhí)行文字方向分類；默認false
        oCRParameter.use_angle_cls = true;//是否開啟方向檢測，用于檢測識別180旋轉(zhuǎn)
        oCRParameter.det_db_score_mode = false;//是否使用多段線，即文字區(qū)域是用多段線還是用矩形，
        oCRParameter.det_db_unclip_ratio = 1.6f;
        //OCR配置
        OCRModelConfig config = new OCRModelConfig();
        string modelPathroot = Path.Join(exePath,"ch_PP-OCRv4");
        config.det_infer = modelPathroot + @"\ch_PP-OCRv4_det_infer";
        config.cls_infer = modelPathroot + @"\ch_ppocr_mobile_v2.0_cls_infer";
        config.rec_infer = modelPathroot + @"\ch_PP-OCRv4_rec_infer";
        config.keys = modelPathroot + @"\ppocr_keys.txt";
        //初始化OCR引擎
        return new PaddleOCREngine(config, oCRParameter);
    }
}

using OCRTest02;
var exePath = AppDomain.CurrentDomain.BaseDirectory;
var engine = OCRHelper.initOcrEngine(exePath);
Console.WriteLine("初始化成功");
var filePath = "F:\\Desktop\\韋小寶.png";
var imagebyte = File.ReadAllBytes(filePath);
var ocrRes = engine.DetectText(imagebyte);

foreach (var textBlock in ocrRes.TextBlocks)
{
    Console.Write(string.Join(";",textBlock.BoxPoints));
    Console.Write("\t"+textBlock.Text);
    Console.Write("\t"+textBlock.Score+"\n");
}

using System.Text.RegularExpressions;
using OCRTest02;
using SixLabors.ImageSharp;
using SixLabors.ImageSharp.Drawing.Processing;
using SixLabors.ImageSharp.Processing;

var exePath = AppDomain.CurrentDomain.BaseDirectory;
var engine = OCRHelper.initOcrEngine(exePath);
Console.WriteLine("初始化成功");
var filePath = "F:\\Desktop\\韋小寶.jpg";
var imagebyte = File.ReadAllBytes(filePath);
var ocrRes = engine.DetectText(imagebyte);

string name = "";
string id = "";
var sfzImg = Image.Load(filePath);

foreach (var textBlock in ocrRes.TextBlocks)
{
    //Console.WriteLine(textBlock.Text);
    
    #region 繪制包圍矩形
    List<PointF> points = new List<PointF>();
    foreach (var point in textBlock.BoxPoints)
    {
        points.Add(new PointF(point.X,point.Y));
    }
    sfzImg.Mutate(x=>x.DrawPolygon(new SolidPen(Color.HotPink,2.0f),points.ToArray()));
    #endregion
    
    if (string.IsNullOrWhiteSpace(textBlock.Text))
    {
        continue;
    }
    var tmpText = textBlock.Text.Trim();
    if (tmpText.Contains("姓名"))
    {
        tmpText = tmpText.Replace("姓名", "");
        name = tmpText;
    }

    var idMatch = new Regex(@"^[1-9]\d{5}(18|19|([23]\d))\d{2}((0[1-9])|(10|11|12))(([0-2][1-9])|10|20|30|31)\d{3}[0-9Xx]$");
    if (idMatch.IsMatch(tmpText))
    {
        id = tmpText;
    }
}

Console.WriteLine($"姓名:{name},身份證號:{id}");
var now = DateTime.Now;
var resultName = now.ToString("yyyy_MM_dd_HH_mm_ss");
sfzImg.Save($"{resultName}.jpg");
Console.WriteLine("保存成功");