102 lines
2.5 KiB
C++
102 lines
2.5 KiB
C++
#include "Recognizer.h"
|
|
|
|
//#include <tesseract/baseapi.h>
|
|
|
|
#include <QDebug>
|
|
#include <QSettings>
|
|
|
|
#include "Settings.h"
|
|
|
|
Recognizer::Recognizer(QObject *parent) :
|
|
QObject(parent),
|
|
engine_ (NULL), imageScale_ (0)
|
|
{
|
|
applySettings ();
|
|
}
|
|
|
|
void Recognizer::applySettings()
|
|
{
|
|
QSettings settings;
|
|
settings.beginGroup (settings_names::recogntionGroup);
|
|
|
|
tessDataDir_ = settings.value (settings_names::tessDataPlace,
|
|
settings_values::tessDataPlace).toString ();
|
|
if (tessDataDir_.right (1) != "/")
|
|
{
|
|
tessDataDir_ += "/";
|
|
}
|
|
ocrLanguage_ = settings.value (settings_names::ocrLanguage,
|
|
settings_values::ocrLanguage).toString ();
|
|
imageScale_ = settings.value (settings_names::imageScale,
|
|
settings_values::imageScale).toInt ();
|
|
|
|
initEngine ();
|
|
}
|
|
|
|
bool Recognizer::initEngine()
|
|
{
|
|
// if (tessDataDir_.isEmpty () || ocrLanguage_.isEmpty ())
|
|
// {
|
|
// emit error (tr ("Неверные параметры для OCR"));
|
|
// return false;
|
|
// }
|
|
// if (engine_ != NULL)
|
|
// {
|
|
// delete engine_;
|
|
// }
|
|
// engine_ = new tesseract::TessBaseAPI();
|
|
// int result = engine_->Init(qPrintable (tessDataDir_), qPrintable (ocrLanguage_),
|
|
// tesseract::OEM_DEFAULT);
|
|
// if (result != 0)
|
|
// {
|
|
// emit error (tr ("Ошибка инициализации OCR: %1").arg (result));
|
|
// delete engine_;
|
|
// engine_ = NULL;
|
|
// return false;
|
|
// }
|
|
// return true;
|
|
}
|
|
|
|
void Recognizer::recognize(ProcessingItem item)
|
|
{
|
|
Q_ASSERT (!item.source.isNull ());
|
|
item.recognized = "test rec";
|
|
emit recognized (item);
|
|
return;
|
|
// if (engine_ == NULL)
|
|
// {
|
|
// if (!initEngine ())
|
|
// {
|
|
// return;
|
|
// }
|
|
// }
|
|
|
|
// QPixmap scaled = pixmap;
|
|
// if (imageScale_ > 0)
|
|
// {
|
|
// scaled = pixmap.scaledToHeight (pixmap.height () * imageScale_,
|
|
// Qt::SmoothTransformation);
|
|
// }
|
|
// QImage image = scaled.toImage ();
|
|
// const int bytesPerPixel = image.depth () / 8;
|
|
// engine_->SetImage (image.bits (), image.width (), image.height (),
|
|
// bytesPerPixel, image.bytesPerLine ());
|
|
|
|
// char* outText = engine_->GetUTF8Text();
|
|
// engine_->Clear();
|
|
|
|
// QString result (outText);
|
|
// result = result.trimmed();
|
|
// if (!result.isEmpty ())
|
|
// {
|
|
// item.recognized = result;
|
|
// emit recognized (result);
|
|
// emit recognized (pixmap, result);
|
|
// }
|
|
// else
|
|
// {
|
|
// emit error (tr ("Текст не распознан."));
|
|
// }
|
|
// delete [] outText;
|
|
}
|