Mirrors/SubtitleEdit
1
0
Fork 0
mirror of https://github.com/SubtitleEdit/subtitleedit.git synced 2024-11-23 03:33:18 +01:00
Code Issues Projects Releases Wiki Activity

Optimize "Binary image compare" OCR method a little

Browse Source
This commit is contained in:
Nikolaj Olsson 2017-12-14 20:20:28 +01:00
parent 8c56fb1b58
commit 1f9ed3682a
3 changed files with 28 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
libse/NikseBitmap.cs
View File

@ -855,6 +855,11 @@ namespace Nikse.SubtitleEdit.Core
return _bitmapData[(x * 4) + (y * _widthX4) + 3];
}
public int GetAlpha(int index)
{
return _bitmapData[index];
}
public Color GetPixel(int x, int y)
{
_pixelAddress = (x * 4) + (y * _widthX4);

28
src/Logic/NikseBitmapImageSplitter.cs
View File

@ -12,7 +12,7 @@ namespace Nikse.SubtitleEdit.Logic
if (a.A < 120 && b.A < 120)
return true; // transparent
var alphaDiff = (int)Math.Abs(a.A - b.A);
var alphaDiff = Math.Abs(a.A - b.A);
if (alphaDiff > 50)
return false; // different alpha levels
@ -983,14 +983,17 @@ namespace Nikse.SubtitleEdit.Logic
{
int different = 0;
int maxDiff = bmp1.Width * bmp1.Height / 5;
for (int x = 0; x < bmp1.Width; x++)
int w4 = bmp2.Width * 4;
for (int y = 0; y < bmp1.Height; y++)
{
for (int y = 0; y < bmp1.Height; y++)
var alpha = y * w4 + 3;
var pixel = y * bmp1.Width;
for (int x = 0; x < bmp1.Width; x++)
{
//if (!IsColorClose(bmp1.GetPixel(x, y), bmp2.GetPixel(x, y), 20))
if (bmp1.GetPixel(x, y) > 0 && bmp2.GetAlpha(x, y) < 100)
if (bmp1.GetPixel(pixel) > 0 && bmp2.GetAlpha(alpha) < 100)
different++;
pixel++;
alpha += 4;
}
if (different > maxDiff)
return different + 10;
@ -1002,18 +1005,23 @@ namespace Nikse.SubtitleEdit.Logic
{
int different = 0;
int maxDiff = bmp1.Width * bmp1.Height / 5;
for (int x = 1; x < bmp1.Width; x++)
int w4 = bmp1.Width * 4;
for (int y = 1; y < bmp1.Height; y++)
{
for (int y = 1; y < bmp1.Height; y++)
var alpha = y * w4 + 7;
var pixel = y * bmp2.Width + 1;
for (int x = 1; x < bmp1.Width; x++)
{
if (bmp1.GetAlpha(x, y) < 100 && bmp2.GetPixel(x, y) > 0)
if (bmp1.GetAlpha(alpha) < 100 && bmp2.GetPixel(pixel) > 0)
different++;
pixel++;
alpha += 4;
}
if (different > maxDiff)
return different + 10;
}
return different;
}
}
}

5
src/Logic/OCR/Binary/BinaryOcrBitmap.cs
View File

@ -191,6 +191,11 @@ namespace Nikse.SubtitleEdit.Logic.Ocr.Binary
return _colors[Width * y + x];
}
public int GetPixel(int index)
{
return _colors[index];
}
public void SetPixel(int x, int y, int c)
{
_colors[Width * y + x] = (byte)c;