aji/ScreenTranslator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor scaling

Browse Source
This commit is contained in:
Gres 2020-03-09 13:36:32 +03:00
parent 4005f94766
commit deb1bee2c7
1 changed files with 45 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

75
src/ocr/tesseract.cpp
View File

@ -108,43 +108,58 @@ static QImage convertImage(Pix &image)
return result; return result;
} }
static double getScale(Pix *source)
{
SOFT_ASSERT(source, return -1.0);
const auto xRes = pixGetXRes(source);
const auto yRes = pixGetYRes(source);
if (xRes * yRes == 0)
return -1.0;
const auto preferredScale = std::max(300.0 / std::min(xRes, yRes), 1.0);
if (preferredScale <= 1.0)
return -1.0;
const auto MAX_INT16 = 0x7fff;
const auto maxScaleX = MAX_INT16 / double(source->w);
const auto scaleX = std::min(preferredScale, maxScaleX);
const auto maxScaleY = MAX_INT16 / double(source->h);
const auto scaleY = std::min(preferredScale, maxScaleY);
auto scale = std::min(scaleX, scaleY);
const auto availableMemory = getFreeMemory() * 0.95;
if (availableMemory < 1)
return -1.0;
const auto actualSize = source->w * source->h * source->d / 8;
const auto maxScaleMemory = availableMemory / actualSize;
scale = std::min(scale, maxScaleMemory);
return scale;
}
static Pix *prepareImage(const QImage &image) static Pix *prepareImage(const QImage &image)
{ {
Pix *pix = convertImage(image); auto pix = convertImage(image);
SOFT_ASSERT(pix != NULL, return nullptr); SOFT_ASSERT(pix, return nullptr);
Pix *gray = pixConvertRGBToGray(pix, 0.0, 0.0, 0.0); auto gray = pixConvertRGBToGray(pix, 0.0, 0.0, 0.0);
SOFT_ASSERT(gray != NULL, return nullptr); SOFT_ASSERT(gray, return nullptr);
pixDestroy(&pix); pixDestroy(&pix);
Pix *scaled = gray; auto scaleSource = gray;
const auto xRes = pixGetXRes(gray); auto scaled = scaleSource;
const auto yRes = pixGetYRes(gray);
if (xRes * yRes != 0) {
const auto preferredScale = std::max(300.0 / std::min(xRes, yRes), 1.0);
if (preferredScale > 1.0) {
const auto MAX_INT16 = 0x7fff;
float maxScaleX = MAX_INT16 / double(gray->w);
float scaleX = std::min(float(preferredScale), maxScaleX);
float maxScaleY = MAX_INT16 / double(gray->h);
float scaleY = std::min(float(preferredScale), maxScaleY);
float scale = std::min(scaleX, scaleY);
qint64 availableMemory = getFreeMemory() * 0.95; if (const auto scale = getScale(scaleSource); scale > 1.0) {
if (availableMemory > 0) { scaled = pixScale(scaleSource, scale, scale);
qint32 actualSize = gray->w * gray->h * gray->d / 8; if (!scaled)
float maxScaleMemory = float(availableMemory) / actualSize; scaled = scaleSource;
scale = std::min(scale, maxScaleMemory);
}
scaled = pixScale(gray, scale, scale);
if (scaled == NULL) {
scaled = gray;
}
}
}
if (scaled != gray) {
pixDestroy(&gray);
} }
if (scaled != scaleSource)
pixDestroy(&scaleSource);
return scaled; return scaled;
} }