Leahnaya/SceneSwitcher
1
0
Fork 0
mirror of https://github.com/WarmUpTill/SceneSwitcher.git synced 2026-08-12 04:11:06 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add tesseract config file support

Browse Source
This commit is contained in:
warmuptill
2025-05-13 08:59:21 +02:00
committed by WarmUpTill
parent 5490fabf92
commit 0e5f56b562
15 changed files with 384 additions and 284 deletions
Download Patch File Download Diff File Expand all files Collapse all files

77
plugins/video/parameter-wrappers.cpp
View File

@@ -1,4 +1,5 @@
#include "parameter-wrappers.hpp"
#include "log-helper.hpp"
#include <QFileInfo>
#include <source-helpers.hpp>
@@ -253,14 +254,13 @@ OCRParameters::OCRParameters(const OCRParameters &other)
colorThreshold(other.colorThreshold),
pageSegMode(other.pageSegMode),
tesseractBasePath(other.tesseractBasePath),
languageCode(other.languageCode)
languageCode(other.languageCode),
useConfig(other.useConfig),
configFile(other.configFile)
{
if (!initDone) {
Setup();
}
if (initDone) {
ocr->SetPageSegMode(pageSegMode);
}
}
OCRParameters &OCRParameters::operator=(const OCRParameters &other)
@@ -272,12 +272,11 @@ OCRParameters &OCRParameters::operator=(const OCRParameters &other)
pageSegMode = other.pageSegMode;
tesseractBasePath = other.tesseractBasePath;
languageCode = other.languageCode;
useConfig = other.useConfig;
configFile = other.configFile;
if (!initDone) {
Setup();
}
if (initDone) {
ocr->SetPageSegMode(pageSegMode);
}
return *this;
}
@@ -288,6 +287,8 @@ bool OCRParameters::Save(obs_data_t *obj) const
regex.Save(data);
tesseractBasePath.Save(data, "tesseractBasePath");
languageCode.Save(data, "language");
obs_data_set_bool(data, "useConfig", useConfig);
obs_data_set_string(data, "configFile", configFile.c_str());
SaveColor(data, "textColor", color);
colorThreshold.Save(data, "colorThreshold");
obs_data_set_int(data, "pageSegMode", static_cast<int>(pageSegMode));
@@ -311,6 +312,8 @@ bool OCRParameters::Load(obs_data_t *obj)
obs_current_module())) +
"/res/ocr/";
}
useConfig = obs_data_get_bool(data, "useConfig");
configFile = obs_data_get_string(data, "configFile");
color = LoadColor(data, "textColor");
if (obs_data_has_user_value(data, "version")) {
@@ -320,9 +323,6 @@ bool OCRParameters::Load(obs_data_t *obj)
obs_data_get_int(data, "pageSegMode"));
obs_data_release(data);
if (initDone) {
ocr->SetPageSegMode(pageSegMode);
}
Setup();
return true;
}
@@ -330,7 +330,9 @@ bool OCRParameters::Load(obs_data_t *obj)
void OCRParameters::SetPageMode(tesseract::PageSegMode mode)
{
pageSegMode = mode;
ocr->SetPageSegMode(mode);
if (ocr) {
ocr->SetPageSegMode(mode);
}
}
bool OCRParameters::SetLanguageCode(const std::string &value)
@@ -343,7 +345,6 @@ bool OCRParameters::SetLanguageCode(const std::string &value)
}
languageCode = value;
Setup();
ocr->SetPageSegMode(pageSegMode);
return true;
}
@@ -363,7 +364,6 @@ bool OCRParameters::SetTesseractBasePath(const std::string &value)
}
tesseractBasePath = value;
Setup();
ocr->SetPageSegMode(pageSegMode);
return true;
}
@@ -372,6 +372,18 @@ std::string OCRParameters::GetTesseractBasePath() const
return tesseractBasePath;
}
void OCRParameters::EnableCustomConfig(bool enable)
{
useConfig = enable;
Setup();
}
void OCRParameters::SetCustomConfigFile(const std::string &filename)
{
configFile = filename;
Setup();
}
void OCRParameters::Setup()
{
ocr = std::make_unique<tesseract::TessBaseAPI>();
@@ -383,18 +395,41 @@ void OCRParameters::Setup()
const std::string dataPath = std::string(tesseractBasePath) + "/";
const std::string modelFile =
std::string(languageCode) + ".traineddata";
const auto fullPath = QString::fromStdString(dataPath) +
QString::fromStdString(modelFile);
QFileInfo fileInfo(fullPath);
if (!fileInfo.exists(fullPath)) {
const auto modelFullPath = QString::fromStdString(dataPath) +
QString::fromStdString(modelFile);
QFileInfo modelFileInfo(modelFullPath);
if (!modelFileInfo.exists(modelFullPath)) {
initDone = false;
blog(LOG_WARNING,
"cannot init tesseract! Model path does not exists: %s",
modelFileInfo.absoluteFilePath().toStdString().c_str());
return;
}
auto configPath = QString::fromStdString(configFile);
QFileInfo configFileInfo(configPath);
bool configFileExists = configFileInfo.exists(configPath);
bool setupWithConfig = useConfig;
if (useConfig && !configFileExists) {
blog(LOG_WARNING,
"tesseract config file will be ignored! File does not exists: %s",
configFileInfo.absoluteFilePath().toStdString().c_str());
setupWithConfig = false;
}
char *configs[] = {configFile.data()};
if (ocr->Init(dataPath.c_str(), languageCode.c_str(),
tesseract::OEM_DEFAULT,
setupWithConfig ? configs : nullptr,
setupWithConfig ? 1 : 0, nullptr, nullptr, false) != 0) {
blog(LOG_WARNING, "tesseract init failed!");
initDone = false;
return;
}
if (ocr->Init(dataPath.c_str(), languageCode.c_str()) != 0) {
initDone = false;
return;
}
ocr->SetPageSegMode(pageSegMode);
initDone = true;
}