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

Add more trace messages

Browse Source
This commit is contained in:
Gres 2020-04-24 19:22:38 +03:00
parent 3e1ab494d9
commit 5556d7aae2
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/ocr/tesseract.cpp
View File

@ -94,11 +94,13 @@ static Pix *prepareImage(const QImage &image)
{ {
auto pix = convertImage(image); auto pix = convertImage(image);
SOFT_ASSERT(pix, return nullptr); SOFT_ASSERT(pix, return nullptr);
LTRACE() << "Converted Pix" << pix;
auto gray = pixConvertRGBToGray(pix, 0.0, 0.0, 0.0); auto gray = pixConvertRGBToGray(pix, 0.0, 0.0, 0.0);
LTRACE() << "Created gray Pix" << gray;
SOFT_ASSERT(gray, return nullptr); SOFT_ASSERT(gray, return nullptr);
pixDestroy(&pix); pixDestroy(&pix);
LTRACE() << "Removed original Pix"; LTRACE() << "Removed converted Pix";
auto scaleSource = gray; auto scaleSource = gray;
auto scaled = scaleSource; auto scaled = scaleSource;
@ -138,16 +140,19 @@ void Tesseract::init(const LanguageId &language, const QString &tessdataPath)
SOFT_ASSERT(!engine_, return ); SOFT_ASSERT(!engine_, return );
engine_ = std::make_unique<tesseract::TessBaseAPI>(); engine_ = std::make_unique<tesseract::TessBaseAPI>();
LTRACE() << "Created Tesseract api" << engine_.get();
const auto tesseractName = LanguageCodes::tesseract(language); const auto tesseractName = LanguageCodes::tesseract(language);
auto result = auto result =
engine_->Init(qPrintable(tessdataPath), qPrintable(tesseractName), engine_->Init(qPrintable(tessdataPath), qPrintable(tesseractName),
tesseract::OEM_DEFAULT); tesseract::OEM_DEFAULT);
LTRACE() << "Inited Tesseract api" << result;
if (result == 0) if (result == 0)
return; return;
error_ = QObject::tr("init failed"); error_ = QObject::tr("init failed");
engine_.reset(); engine_.reset();
LTRACE() << "Cleared Tesseract api";
} }
const QString &Tesseract::error() const const QString &Tesseract::error() const
@ -187,12 +192,14 @@ QString Tesseract::recognize(const QPixmap &source)
error_.clear(); error_.clear();
Pix *image = prepareImage(source.toImage()); Pix *image = prepareImage(source.toImage());
SOFT_ASSERT(image != NULL, return {}); SOFT_ASSERT(image, return {});
LTRACE() << "Preprocessed Pix for OCR"; LTRACE() << "Preprocessed Pix for OCR" << image;
engine_->SetImage(image); engine_->SetImage(image);
LTRACE() << "Set Pix to engine";
char *outText = engine_->GetUTF8Text(); char *outText = engine_->GetUTF8Text();
LTRACE() << "Received recognized text"; LTRACE() << "Received recognized text";
engine_->Clear(); engine_->Clear();
LTRACE() << "Cleared engine";
cleanupImage(&image); cleanupImage(&image);
LTRACE() << "Cleared preprocessed Pix"; LTRACE() << "Cleared preprocessed Pix";