gnh1201/welsonjs
1
0
Fork 0
mirror of https://github.com/gnh1201/welsonjs.git synced 2025-05-13 21:21:03 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update ScreenMatch.cs

Browse Source
This commit is contained in:
Namhyeon Go 2024-09-02 15:30:20 +09:00
parent e70c152494
commit 2f4b95313f
1 changed files with 17 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
WelsonJS.Toolkit/WelsonJS.Service/ScreenMatch.cs
View File

@ -283,6 +283,8 @@ public class ScreenMatch
parent.Log($"Trying match the template {image.Tag as string} on the screen {i}..."); parent.Log($"Trying match the template {image.Tag as string} on the screen {i}...");
string filename = image.Tag as string; string filename = image.Tag as string;
int imageWidth = image.Width;
int imageHeight = image.Height;
Bitmap _mainImage; Bitmap _mainImage;
if (filename.StartsWith("binary_")) if (filename.StartsWith("binary_"))
@ -300,7 +302,7 @@ public class ScreenMatch
string text; string text;
if (isOCR128) if (isOCR128)
{ {
text = OCR((Bitmap)mainImage.Clone(), matchPosition.X, matchPosition.Y, 128, 128); text = OCR((Bitmap)mainImage.Clone(), matchPosition.X, matchPosition.Y, imageWidth, imageHeight, 128, 128);
} }
else else
{ {
@ -332,18 +334,30 @@ public class ScreenMatch
return results; return results;
} }
public string OCR(Bitmap bitmap, int x, int y, int w, int h) public string OCR(Bitmap bitmap, int x, int y, int a, int b, int w, int h)
{ {
if (bitmap == null)
{
throw new ArgumentNullException(nameof(bitmap), "Bitmap cannot be null.");
}
string text = ""; string text = "";
// Adjust coordinates
x = x + (a / 2);
y = y + (b / 2);
// Set range of crop image
int cropX = Math.Max(x - w / 2, 0); int cropX = Math.Max(x - w / 2, 0);
int cropY = Math.Max(y - h / 2, 0); int cropY = Math.Max(y - h / 2, 0);
int cropWidth = Math.Min(w, bitmap.Width - cropX); int cropWidth = Math.Min(w, bitmap.Width - cropX);
int cropHeight = Math.Min(h, bitmap.Height - cropY); int cropHeight = Math.Min(h, bitmap.Height - cropY);
Rectangle cropArea = new Rectangle(cropX, cropY, cropWidth, cropHeight); Rectangle cropArea = new Rectangle(cropX, cropY, cropWidth, cropHeight);
// Crop image
Bitmap croppedBitmap = bitmap.Clone(cropArea, bitmap.PixelFormat); Bitmap croppedBitmap = bitmap.Clone(cropArea, bitmap.PixelFormat);
// OCR
using (var engine = new TesseractEngine(tesseractDataPath, tesseractLanguage, EngineMode.Default)) using (var engine = new TesseractEngine(tesseractDataPath, tesseractLanguage, EngineMode.Default))
{ {
using (var img = PixConverter.ToPix(croppedBitmap)) using (var img = PixConverter.ToPix(croppedBitmap))