Allow providing data path when calling plugin

Author: Damir Porobic

example/IPluginOcrOther.h

@@ -27,6 +27,7 @@ class IPluginOcrOther
 public:
 	virtual ~IPluginOcrOther() = default;
 	virtual QString recognize(const QPixmap &pixmap) const = 0;
+	virtual QString recognize(const QPixmap &pixmap, const QString &dataPath) const = 0;
 	virtual QString version() const = 0;
 };
 

example/main.cpp

@@ -60,7 +60,7 @@ int main(int argc, char **argv)
 
 	textLabel->setTextInteractionFlags(Qt::TextSelectableByMouse);
 
-	QDir pluginsDir(QLatin1String("../src/"));
+	QDir pluginsDir(QLatin1String("../src"));
 
 #if  defined(_WIN32)
 	// Under Windows the 3rd Party Dlls are next to the plugin in the same directory
@@ -82,7 +82,12 @@ int main(int argc, char **argv)
 			auto pluginOcr = qobject_cast<IPluginOcrOther*>(plugin);
 			if (pluginOcr) {
 				qDebug() << "Loaded plugin " << fileName << " with version " << pluginOcr->version();
+
+#if  defined(_WIN32)
+				auto text = pluginOcr->recognize(pixmap, pluginsDir.path() + QString("\\tessdata"));
+#else
 				auto text = pluginOcr->recognize(pixmap);
+#endif
 				textLabel->setText(text);
 			} else {
 				qCritical() << "Unable to cast to interface";

src/IPluginOcr.h

@@ -27,6 +27,7 @@ class IPluginOcr
 public:
 	virtual ~IPluginOcr() = default;
 	virtual QString recognize(const QPixmap &pixmap) const = 0;
+	virtual QString recognize(const QPixmap &pixmap, const QString &dataPath) const = 0;
 	virtual QString version() const = 0;
 };
 

src/OcrWrapper.cpp

@@ -31,7 +31,17 @@ OcrWrapper::~OcrWrapper()
 
 QString OcrWrapper::recognize(const QPixmap &pixmap) const
 {
-	if (mTessApi->Init(getDataPath(), "eng") == 0)
+	return recognizeInner(pixmap, nullptr);
+}
+
+QString OcrWrapper::recognize(const QPixmap &pixmap, const QString &dataPath) const
+{
+	return recognizeInner(pixmap, dataPath.toLocal8Bit().data());
+}
+
+QString OcrWrapper::recognizeInner(const QPixmap &pixmap, char * dataPath) const
+{
+	if (mTessApi->Init(dataPath, "eng") == 0)
 	{
 		auto pix = makePixFromPixmap(pixmap);
 		mTessApi->SetImage(pix);
@@ -57,16 +67,3 @@ PIX* OcrWrapper::makePixFromPixmap(const QPixmap &pixmap)
 
 	return pixReadMemBmp((l_uint8 *)byteArray.constData(), byteArray.size());
 }
-
-const char *OcrWrapper::getDataPath()
-{
-#if defined(_WIN32)
-	// Under Windows we expect the trained data to be next to the plugin
-	// in as directory called tessdata
-	return ".\\tessdata\\";
-#else
-	// Under Linux/Unix/MacOS we expect the TESSDATA_PREFIX to point to the
-	// trained data.
-	return nullptr;
-#endif
-}

src/OcrWrapper.h

@@ -35,12 +35,14 @@ class OcrWrapper
 	~OcrWrapper();
 
 	QString recognize(const QPixmap &pixmap) const;
+	QString recognize(const QPixmap &pixmap, const QString &dataPath) const;
 
 private:
 	QSharedPointer<TessBaseAPI> mTessApi;
 
+	QString recognizeInner(const QPixmap &pixmap, char * dataPath) const;
+
 	static PIX* makePixFromPixmap(const QPixmap &pixmap);
-	static const char *getDataPath();
 };
 
 #endif //PLUGIN_OCR_OCRWRAPPER_H

src/PluginOcr.cpp

@@ -24,6 +24,11 @@ QString PluginOcr::recognize(const QPixmap &pixmap) const
 	return mOcrWrapper.recognize(pixmap);
 }
 
+QString PluginOcr::recognize(const QPixmap &pixmap, const QString &dataPath) const
+{
+	return mOcrWrapper.recognize(pixmap, dataPath);
+}
+
 QString PluginOcr::version() const
 {
 	return QLatin1String(PROJECT_VERSION);

src/PluginOcr.h

@@ -37,6 +37,7 @@ Q_OBJECT
 public:
 	~PluginOcr() override = default;
 	QString recognize(const QPixmap &pixmap) const override;
+	QString recognize(const QPixmap &pixmap, const QString &dataPath) const override;
 	QString version() const override;
 
 private: