302 lines
13 KiB
C++
302 lines
13 KiB
C++
#include <QDir>
|
|
#include <QSettings>
|
|
|
|
#include "LanguageHelper.h"
|
|
#include "Settings.h"
|
|
#include "StAssert.h"
|
|
|
|
LanguageHelper::LanguageHelper () {
|
|
init ();
|
|
}
|
|
|
|
QStringList LanguageHelper::availableOcrLanguagesUi () const {
|
|
QStringList uiItems;
|
|
foreach (const QString &item, availableOcrLanguages_) {
|
|
uiItems << ocrCodeToUi (item);
|
|
}
|
|
uiItems.sort ();
|
|
return uiItems;
|
|
}
|
|
|
|
const QStringList &LanguageHelper::availableOcrLanguages () const {
|
|
return availableOcrLanguages_;
|
|
}
|
|
|
|
QStringList LanguageHelper::availableOcrLanguages (const QString &path) const {
|
|
QDir dir (path + "/tessdata/");
|
|
if (!dir.exists ()) {
|
|
return QStringList ();
|
|
}
|
|
QStringList items;
|
|
QStringList files = dir.entryList (QStringList () << "*.traineddata", QDir::Files);
|
|
foreach (const QString &file, files) {
|
|
QString lang = file.left (file.indexOf ("."));
|
|
items << lang;
|
|
}
|
|
return items;
|
|
}
|
|
|
|
QStringList LanguageHelper::availableOcrLanguagesUi (const QString &path) const {
|
|
QStringList uiItems, items;
|
|
items = availableOcrLanguages (path);
|
|
foreach (const QString &item, items) {
|
|
uiItems << ocrCodeToUi (item);
|
|
}
|
|
uiItems.sort ();
|
|
return uiItems;
|
|
}
|
|
|
|
QStringList LanguageHelper::translateLanguagesUi () const {
|
|
QStringList uiItems = translateLanguages_.keys ();
|
|
uiItems.sort ();
|
|
return uiItems;
|
|
}
|
|
|
|
QStringList LanguageHelper::translateLanguages () const {
|
|
return translateLanguages_.values ();
|
|
}
|
|
|
|
QString LanguageHelper::translateCodeToUi (const QString &text) const {
|
|
return translateLanguages_.key (text, text);
|
|
}
|
|
|
|
QString LanguageHelper::translateUiToCode (const QString &text) const {
|
|
return translateLanguages_.value (text, text);
|
|
}
|
|
|
|
QString LanguageHelper::ocrCodeToUi (const QString &text) const {
|
|
return ocrLanguages_.key (text, text);
|
|
}
|
|
|
|
QString LanguageHelper::ocrUiToCode (const QString &text) const {
|
|
return ocrLanguages_.value (text, text);
|
|
}
|
|
|
|
QString LanguageHelper::ocrToTranslateCodes (const QString &text) const {
|
|
QString ocrUi = ocrCodeToUi (text);
|
|
QString translateCode = translateUiToCode (ocrUi);
|
|
if (translateCode == ocrUi) {
|
|
translateCode = "auto";
|
|
}
|
|
return translateCode;
|
|
}
|
|
|
|
QString LanguageHelper::translateToOcrCodes (const QString &text) const {
|
|
QString translateUi = translateCodeToUi (text);
|
|
QString ocrCode = ocrUiToCode (translateUi);
|
|
if (translateUi == ocrCode) {
|
|
return QString ();
|
|
}
|
|
return ocrCode;
|
|
}
|
|
|
|
void LanguageHelper::init () {
|
|
initOcrLanguages ();
|
|
initTranslateLanguages ();
|
|
updateAvailableOcrLanguages ();
|
|
}
|
|
|
|
void LanguageHelper::updateAvailableOcrLanguages () {
|
|
availableOcrLanguages_.clear ();
|
|
QSettings settings;
|
|
settings.beginGroup (settings_names::recogntionGroup);
|
|
QString tessDataPlace = settings.value (settings_names::tessDataPlace,
|
|
settings_values::tessDataPlace).toString ();
|
|
availableOcrLanguages_ = availableOcrLanguages (tessDataPlace);
|
|
}
|
|
|
|
void LanguageHelper::updateMenu (QMenu *menu, const QStringList &languages, int groupSize) const {
|
|
ST_ASSERT (menu != NULL);
|
|
menu->clear ();
|
|
if (languages.isEmpty ()) {
|
|
return;
|
|
}
|
|
|
|
if (languages.size () <= groupSize) {
|
|
foreach (const QString &language, languages) {
|
|
menu->addAction (language);
|
|
}
|
|
}
|
|
else {
|
|
int subIndex = groupSize;
|
|
QMenu *subMenu = NULL;
|
|
QString prevLetter;
|
|
foreach (const QString &language, languages) {
|
|
QString curLetter = language.left (1);
|
|
if (++subIndex >= groupSize && prevLetter != curLetter) {
|
|
if (subMenu != NULL) {
|
|
subMenu->setTitle (subMenu->title () + " - " + prevLetter);
|
|
}
|
|
subMenu = menu->addMenu (curLetter);
|
|
subIndex = 0;
|
|
}
|
|
prevLetter = curLetter;
|
|
subMenu->addAction (language);
|
|
}
|
|
subMenu->setTitle (subMenu->title () + " - " + prevLetter);
|
|
}
|
|
}
|
|
|
|
void LanguageHelper::initTranslateLanguages () {
|
|
translateLanguages_.insert (QObject::tr ("Afrikaans"),"af");
|
|
translateLanguages_.insert (QObject::tr ("Albanian"),"sq");
|
|
translateLanguages_.insert (QObject::tr ("Arabic"),"ar");
|
|
translateLanguages_.insert (QObject::tr ("Armenian"),"hy");
|
|
translateLanguages_.insert (QObject::tr ("Azerbaijani"),"az");
|
|
translateLanguages_.insert (QObject::tr ("Basque"),"eu");
|
|
translateLanguages_.insert (QObject::tr ("Belarusian"),"be");
|
|
translateLanguages_.insert (QObject::tr ("Bulgarian"),"bg");
|
|
translateLanguages_.insert (QObject::tr ("Catalan"),"ca");
|
|
translateLanguages_.insert (QObject::tr ("Chinese (Simplified)"),"zh-CN");
|
|
translateLanguages_.insert (QObject::tr ("Chinese (Traditional)"),"zh-TW");
|
|
translateLanguages_.insert (QObject::tr ("Croatian"),"hr");
|
|
translateLanguages_.insert (QObject::tr ("Czech"),"cs");
|
|
translateLanguages_.insert (QObject::tr ("Danish"),"da");
|
|
translateLanguages_.insert (QObject::tr ("Dutch"),"nl");
|
|
translateLanguages_.insert (QObject::tr ("English"),"en");
|
|
translateLanguages_.insert (QObject::tr ("Estonian"),"et");
|
|
translateLanguages_.insert (QObject::tr ("Filipino"),"tl");
|
|
translateLanguages_.insert (QObject::tr ("Finnish"),"fi");
|
|
translateLanguages_.insert (QObject::tr ("French"),"fr");
|
|
translateLanguages_.insert (QObject::tr ("Galician"),"gl");
|
|
translateLanguages_.insert (QObject::tr ("Georgian"),"ka");
|
|
translateLanguages_.insert (QObject::tr ("German"),"de");
|
|
translateLanguages_.insert (QObject::tr ("Greek"),"el");
|
|
translateLanguages_.insert (QObject::tr ("Haitian Creole"),"ht");
|
|
translateLanguages_.insert (QObject::tr ("Hebrew"),"iw");
|
|
translateLanguages_.insert (QObject::tr ("Hindi"),"hi");
|
|
translateLanguages_.insert (QObject::tr ("Hungarian"),"hu");
|
|
translateLanguages_.insert (QObject::tr ("Icelandic"),"is");
|
|
translateLanguages_.insert (QObject::tr ("Indonesian"),"id");
|
|
translateLanguages_.insert (QObject::tr ("Irish"),"ga");
|
|
translateLanguages_.insert (QObject::tr ("Italian"),"it");
|
|
translateLanguages_.insert (QObject::tr ("Japanese"),"ja");
|
|
translateLanguages_.insert (QObject::tr ("Korean"),"ko");
|
|
translateLanguages_.insert (QObject::tr ("Latvian"),"lv");
|
|
translateLanguages_.insert (QObject::tr ("Lithuanian"),"lt");
|
|
translateLanguages_.insert (QObject::tr ("Macedonian"),"mk");
|
|
translateLanguages_.insert (QObject::tr ("Malay"),"ms");
|
|
translateLanguages_.insert (QObject::tr ("Maltese"),"mt");
|
|
translateLanguages_.insert (QObject::tr ("Norwegian"),"no");
|
|
translateLanguages_.insert (QObject::tr ("Persian"),"fa");
|
|
translateLanguages_.insert (QObject::tr ("Polish"),"pl");
|
|
translateLanguages_.insert (QObject::tr ("Portuguese"),"pt");
|
|
translateLanguages_.insert (QObject::tr ("Romanian"),"ro");
|
|
translateLanguages_.insert (QObject::tr ("Russian"),"ru");
|
|
translateLanguages_.insert (QObject::tr ("Serbian"),"sr");
|
|
translateLanguages_.insert (QObject::tr ("Slovak"),"sk");
|
|
translateLanguages_.insert (QObject::tr ("Slovenian"),"sl");
|
|
translateLanguages_.insert (QObject::tr ("Spanish"),"es");
|
|
translateLanguages_.insert (QObject::tr ("Swahili"),"sw");
|
|
translateLanguages_.insert (QObject::tr ("Swedish"),"sv");
|
|
translateLanguages_.insert (QObject::tr ("Thai"),"th");
|
|
translateLanguages_.insert (QObject::tr ("Turkish"),"tr");
|
|
translateLanguages_.insert (QObject::tr ("Ukrainian"),"uk");
|
|
translateLanguages_.insert (QObject::tr ("Urdu"),"ur");
|
|
translateLanguages_.insert (QObject::tr ("Vietnamese"),"vi");
|
|
translateLanguages_.insert (QObject::tr ("Welsh"),"cy");
|
|
translateLanguages_.insert (QObject::tr ("Yiddish"),"yi");
|
|
}
|
|
|
|
void LanguageHelper::initOcrLanguages () {
|
|
ocrLanguages_.insert (QObject::tr ("Ancient Greek"),"grc");
|
|
ocrLanguages_.insert (QObject::tr ("Esperanto alternative"),"epo_alt");
|
|
ocrLanguages_.insert (QObject::tr ("English"),"eng");
|
|
ocrLanguages_.insert (QObject::tr ("Ukrainian"),"ukr");
|
|
ocrLanguages_.insert (QObject::tr ("Turkish"),"tur");
|
|
ocrLanguages_.insert (QObject::tr ("Thai"),"tha");
|
|
ocrLanguages_.insert (QObject::tr ("Tagalog"),"tgl");
|
|
ocrLanguages_.insert (QObject::tr ("Telugu"),"tel");
|
|
ocrLanguages_.insert (QObject::tr ("Tamil"),"tam");
|
|
ocrLanguages_.insert (QObject::tr ("Swedish"),"swe");
|
|
ocrLanguages_.insert (QObject::tr ("Swahili"),"swa");
|
|
ocrLanguages_.insert (QObject::tr ("Serbian"),"srp");
|
|
ocrLanguages_.insert (QObject::tr ("Albanian"),"sqi");
|
|
ocrLanguages_.insert (QObject::tr ("Spanish"),"spa");
|
|
ocrLanguages_.insert (QObject::tr ("Slovenian"),"slv");
|
|
ocrLanguages_.insert (QObject::tr ("Slovakian"),"slk");
|
|
ocrLanguages_.insert (QObject::tr ("Romanian"),"ron");
|
|
ocrLanguages_.insert (QObject::tr ("Portuguese"),"por");
|
|
ocrLanguages_.insert (QObject::tr ("Polish"),"pol");
|
|
ocrLanguages_.insert (QObject::tr ("Norwegian"),"nor");
|
|
ocrLanguages_.insert (QObject::tr ("Dutch"),"nld");
|
|
ocrLanguages_.insert (QObject::tr ("Malay"),"msa");
|
|
ocrLanguages_.insert (QObject::tr ("Maltese"),"mlt");
|
|
ocrLanguages_.insert (QObject::tr ("Macedonian"),"mkd");
|
|
ocrLanguages_.insert (QObject::tr ("Malayalam"),"mal");
|
|
ocrLanguages_.insert (QObject::tr ("Lithuanian"),"lit");
|
|
ocrLanguages_.insert (QObject::tr ("Latvian"),"lav");
|
|
ocrLanguages_.insert (QObject::tr ("Korean"),"kor");
|
|
ocrLanguages_.insert (QObject::tr ("Kannada"),"kan");
|
|
ocrLanguages_.insert (QObject::tr ("Italian"),"ita");
|
|
ocrLanguages_.insert (QObject::tr ("Icelandic"),"isl");
|
|
ocrLanguages_.insert (QObject::tr ("Indonesian"),"ind");
|
|
ocrLanguages_.insert (QObject::tr ("Cherokee"),"chr");
|
|
ocrLanguages_.insert (QObject::tr ("Hungarian"),"hun");
|
|
ocrLanguages_.insert (QObject::tr ("Croatian"),"hrv");
|
|
ocrLanguages_.insert (QObject::tr ("Hindi"),"hin");
|
|
ocrLanguages_.insert (QObject::tr ("Hebrew"),"heb");
|
|
ocrLanguages_.insert (QObject::tr ("Galician"),"glg");
|
|
ocrLanguages_.insert (QObject::tr ("Middle French (ca. 1400-1600)"),"frm");
|
|
ocrLanguages_.insert (QObject::tr ("Frankish"),"frk");
|
|
ocrLanguages_.insert (QObject::tr ("French"),"fra");
|
|
ocrLanguages_.insert (QObject::tr ("Finnish"),"fin");
|
|
ocrLanguages_.insert (QObject::tr ("Basque"),"eus");
|
|
ocrLanguages_.insert (QObject::tr ("Estonian"),"est");
|
|
ocrLanguages_.insert (QObject::tr ("Math / equation"),"equ");
|
|
ocrLanguages_.insert (QObject::tr ("Esperanto"),"epo");
|
|
ocrLanguages_.insert (QObject::tr ("Middle English (1100-1500)"),"enm");
|
|
ocrLanguages_.insert (QObject::tr ("Greek"),"ell");
|
|
ocrLanguages_.insert (QObject::tr ("German"),"deu");
|
|
ocrLanguages_.insert (QObject::tr ("Danish"),"dan");
|
|
ocrLanguages_.insert (QObject::tr ("Czech"),"ces");
|
|
ocrLanguages_.insert (QObject::tr ("Catalan"),"cat");
|
|
ocrLanguages_.insert (QObject::tr ("Bulgarian"),"bul");
|
|
ocrLanguages_.insert (QObject::tr ("Bengali"),"ben");
|
|
ocrLanguages_.insert (QObject::tr ("Belarusian"),"bel");
|
|
ocrLanguages_.insert (QObject::tr ("Azerbaijani"),"aze");
|
|
ocrLanguages_.insert (QObject::tr ("Arabic"),"ara");
|
|
ocrLanguages_.insert (QObject::tr ("Afrikaans"),"afr");
|
|
ocrLanguages_.insert (QObject::tr ("Japanese"),"jpn");
|
|
ocrLanguages_.insert (QObject::tr ("Chinese (Simplified)"),"chi_sim");
|
|
ocrLanguages_.insert (QObject::tr ("Chinese (Traditional)"),"chi_tra");
|
|
ocrLanguages_.insert (QObject::tr ("Russian"),"rus");
|
|
ocrLanguages_.insert (QObject::tr ("Vietnamese"),"vie");
|
|
ocrLanguages_.insert (QObject::tr ("Amharic"),"amh");
|
|
ocrLanguages_.insert (QObject::tr ("Assamese"),"asm");
|
|
ocrLanguages_.insert (QObject::tr ("Tibetan"),"bod");
|
|
ocrLanguages_.insert (QObject::tr ("Bosnian"),"bos");
|
|
ocrLanguages_.insert (QObject::tr ("Cebuano"),"ceb");
|
|
ocrLanguages_.insert (QObject::tr ("Welsh"),"cym");
|
|
ocrLanguages_.insert (QObject::tr ("Dzongkha"),"dzo");
|
|
ocrLanguages_.insert (QObject::tr ("Persian"),"fas");
|
|
ocrLanguages_.insert (QObject::tr ("Irish"),"gle");
|
|
ocrLanguages_.insert (QObject::tr ("Gujarati"),"guj");
|
|
ocrLanguages_.insert (QObject::tr ("Haitian"),"hat");
|
|
ocrLanguages_.insert (QObject::tr ("Inuktitut"),"iku");
|
|
ocrLanguages_.insert (QObject::tr ("Javanese"),"jav");
|
|
ocrLanguages_.insert (QObject::tr ("Georgian"),"kat");
|
|
ocrLanguages_.insert (QObject::tr ("Kazakh"),"kaz");
|
|
ocrLanguages_.insert (QObject::tr ("Khmer"),"khm");
|
|
ocrLanguages_.insert (QObject::tr ("Kirghiz"),"kir");
|
|
ocrLanguages_.insert (QObject::tr ("Kurdish"),"kur");
|
|
ocrLanguages_.insert (QObject::tr ("Lao"),"lao");
|
|
ocrLanguages_.insert (QObject::tr ("Latin"),"lat");
|
|
ocrLanguages_.insert (QObject::tr ("Marathi"),"mar");
|
|
ocrLanguages_.insert (QObject::tr ("Burmese"),"mya");
|
|
ocrLanguages_.insert (QObject::tr ("Nepali"),"nep");
|
|
ocrLanguages_.insert (QObject::tr ("Oriya"),"ori");
|
|
ocrLanguages_.insert (QObject::tr ("Panjabi"),"pan");
|
|
ocrLanguages_.insert (QObject::tr ("Pushto"),"pus");
|
|
ocrLanguages_.insert (QObject::tr ("Sanskrit"),"san");
|
|
ocrLanguages_.insert (QObject::tr ("Sinhala"),"sin");
|
|
ocrLanguages_.insert (QObject::tr ("Syriac"),"syr");
|
|
ocrLanguages_.insert (QObject::tr ("Tajik"),"tgk");
|
|
ocrLanguages_.insert (QObject::tr ("Tigrinya"),"tir");
|
|
ocrLanguages_.insert (QObject::tr ("Uighur"),"uig");
|
|
ocrLanguages_.insert (QObject::tr ("Urdu"),"urd");
|
|
ocrLanguages_.insert (QObject::tr ("Uzbek"),"uzb");
|
|
ocrLanguages_.insert (QObject::tr ("Yiddish"),"yid");
|
|
|
|
}
|