72 lines
2.0 KiB
C++
72 lines
2.0 KiB
C++
#include "recognizerworker.h"
|
|
#include "debug.h"
|
|
#include "task.h"
|
|
#include "tesseract.h"
|
|
|
|
RecognizeWorker::~RecognizeWorker() = default;
|
|
|
|
void RecognizeWorker::handle(const TaskPtr &task)
|
|
{
|
|
SOFT_ASSERT(task, return );
|
|
SOFT_ASSERT(task->isValid(), return );
|
|
SOFT_ASSERT(!tessdataPath_.isEmpty(), return );
|
|
|
|
LTRACE() << "Start recognize" << task->captured;
|
|
auto result = task;
|
|
|
|
if (!engines_.count(task->sourceLanguage)) {
|
|
LTRACE() << "Create OCR engine" << task->sourceLanguage;
|
|
|
|
auto engine = std::make_unique<Tesseract>(task->sourceLanguage,
|
|
tessdataPath_, tesseractLibrary_);
|
|
|
|
if (!engine->isValid()) {
|
|
result->error = tr("Failed to init OCR engine: %1").arg(engine->error());
|
|
emit finished(result);
|
|
return;
|
|
}
|
|
|
|
engines_.emplace(task->sourceLanguage, std::move(engine));
|
|
LTRACE() << "Added OCR engine" << task->sourceLanguage;
|
|
}
|
|
|
|
auto &engine = engines_[task->sourceLanguage];
|
|
SOFT_ASSERT(engine->isValid(), return );
|
|
|
|
result->recognized = engine->recognize(task->captured);
|
|
if (result->recognized.isEmpty())
|
|
result->error = engine->error();
|
|
|
|
lastGenerations_[task->sourceLanguage] = task->generation;
|
|
removeUnused(task->generation);
|
|
|
|
emit finished(result);
|
|
}
|
|
|
|
void RecognizeWorker::reset(const QString &tessdataPath,
|
|
const QString &tesseractLibrary)
|
|
{
|
|
if (tessdataPath_ == tessdataPath && tesseractLibrary_ == tesseractLibrary)
|
|
return;
|
|
|
|
tessdataPath_ = tessdataPath;
|
|
tesseractLibrary_ = tesseractLibrary;
|
|
engines_.clear();
|
|
LTRACE() << "Cleared OCR engines";
|
|
}
|
|
|
|
void RecognizeWorker::removeUnused(Generation current)
|
|
{
|
|
const auto keepGenerations = 10;
|
|
for (auto it = lastGenerations_.begin(), end = lastGenerations_.end();
|
|
it != end;) {
|
|
if (current - it->second < keepGenerations) {
|
|
++it;
|
|
continue;
|
|
}
|
|
engines_.erase(it->first);
|
|
LTRACE() << "Removed unused OCR engine" << it->first;
|
|
it = lastGenerations_.erase(it);
|
|
}
|
|
}
|