182 lines
4.5 KiB
C++
182 lines
4.5 KiB
C++
#include "spellchecker.h"
|
|
|
|
#ifdef USE_HUNSPELL
|
|
#include "hunspell.hxx"
|
|
#endif
|
|
#include <QTextCodec>
|
|
#include <QFile>
|
|
#include <QDebug>
|
|
#include <QLocale>
|
|
|
|
#include <QRegularExpression>
|
|
#include <QGuiApplication>
|
|
#include <QDir>
|
|
|
|
SpellChecker::SpellChecker(QObject *parent)
|
|
: QObject(parent)
|
|
, m_userDict("userDict_")
|
|
#ifdef USE_HUNSPELL
|
|
, m_hunspell(nullptr)
|
|
#endif
|
|
{
|
|
|
|
}
|
|
|
|
SpellChecker::~SpellChecker()
|
|
{
|
|
#ifdef USE_HUNSPELL
|
|
delete m_hunspell;
|
|
#endif
|
|
}
|
|
|
|
bool SpellChecker::spell(const QString &word)
|
|
{
|
|
#ifdef USE_HUNSPELL
|
|
return m_hunspell->spell(m_codec->fromUnicode(word).toStdString());
|
|
#else
|
|
return true;
|
|
#endif
|
|
}
|
|
|
|
bool SpellChecker::isInit() const
|
|
{
|
|
#ifdef USE_HUNSPELL
|
|
return !m_hunspell;
|
|
#endif
|
|
return false;
|
|
}
|
|
|
|
void SpellChecker::initHunspell()
|
|
{
|
|
#ifdef USE_HUNSPELL
|
|
if (m_hunspell) {
|
|
delete m_hunspell;
|
|
}
|
|
|
|
QString dictFile = QGuiApplication::applicationDirPath() + "/dictionaries/" + m_lang + "/index.dic";
|
|
QString affixFile = QGuiApplication::applicationDirPath() + "/dictionaries/" + m_lang + "/index.aff";
|
|
QByteArray dictFilePathBA = dictFile.toLocal8Bit();
|
|
QByteArray affixFilePathBA = affixFile.toLocal8Bit();
|
|
m_hunspell = new Hunspell(affixFilePathBA.constData(),
|
|
dictFilePathBA.constData());
|
|
|
|
// detect encoding analyzing the SET option in the affix file
|
|
auto encoding = QStringLiteral("ISO8859-15");
|
|
QFile _affixFile(affixFile);
|
|
if (_affixFile.open(QIODevice::ReadOnly)) {
|
|
QTextStream stream(&_affixFile);
|
|
QRegularExpression enc_detector(
|
|
QStringLiteral("^\\s*SET\\s+([A-Z0-9\\-]+)\\s*"),
|
|
QRegularExpression::CaseInsensitiveOption);
|
|
QString sLine;
|
|
QRegularExpressionMatch match;
|
|
while (!stream.atEnd()) {
|
|
sLine = stream.readLine();
|
|
if (sLine.isEmpty()) { continue; }
|
|
match = enc_detector.match(sLine);
|
|
if (match.hasMatch()) {
|
|
encoding = match.captured(1);
|
|
qDebug() << "Encoding set to " + encoding;
|
|
break;
|
|
}
|
|
}
|
|
_affixFile.close();
|
|
}
|
|
m_codec = QTextCodec::codecForName(encoding.toLatin1().constData());
|
|
|
|
QString userDict = m_userDict + m_lang + ".txt";
|
|
|
|
if (!userDict.isEmpty()) {
|
|
QFile userDictonaryFile(userDict);
|
|
if (userDictonaryFile.open(QIODevice::ReadOnly)) {
|
|
QTextStream stream(&userDictonaryFile);
|
|
for (QString word = stream.readLine();
|
|
!word.isEmpty();
|
|
word = stream.readLine())
|
|
ignoreWord(word);
|
|
userDictonaryFile.close();
|
|
} else {
|
|
qWarning() << "User dictionary in " << userDict
|
|
<< "could not be opened";
|
|
}
|
|
} else {
|
|
qDebug() << "User dictionary not set.";
|
|
}
|
|
#endif
|
|
}
|
|
|
|
QVariantList SpellChecker::suggest(const QString &word)
|
|
{
|
|
int numSuggestions = 0;
|
|
QVariantList suggestions;
|
|
#ifdef USE_HUNSPELL
|
|
std::vector<std::string> wordlist;
|
|
wordlist = m_hunspell->suggest(m_codec->fromUnicode(word).toStdString());
|
|
|
|
numSuggestions = static_cast<int>(wordlist.size());
|
|
if (numSuggestions > 0) {
|
|
suggestions.reserve(numSuggestions);
|
|
for (int i = 0; i < numSuggestions; i++) {
|
|
suggestions << m_codec->toUnicode(
|
|
QByteArray::fromStdString(wordlist[i]));
|
|
}
|
|
}
|
|
#endif
|
|
|
|
return suggestions;
|
|
}
|
|
|
|
void SpellChecker::ignoreWord(const QString &word)
|
|
{
|
|
#ifdef USE_HUNSPELL
|
|
m_hunspell->add(m_codec->fromUnicode(word).constData());
|
|
#endif
|
|
}
|
|
|
|
void SpellChecker::addToUserWordlist(const QString &word)
|
|
{
|
|
#ifdef USE_HUNSPELL
|
|
QString userDict = m_userDict + m_lang + ".txt";
|
|
if (!userDict.isEmpty()) {
|
|
QFile userDictonaryFile(userDict);
|
|
if (userDictonaryFile.open(QIODevice::Append)) {
|
|
QTextStream stream(&userDictonaryFile);
|
|
stream << word << "\n";
|
|
userDictonaryFile.close();
|
|
} else {
|
|
qWarning() << "User dictionary in " << userDict
|
|
<< "could not be opened for appending a new word";
|
|
}
|
|
} else {
|
|
qDebug() << "User dictionary not set.";
|
|
}
|
|
#endif
|
|
}
|
|
|
|
const QString& SpellChecker::lang() const
|
|
{
|
|
return m_lang;
|
|
}
|
|
|
|
void SpellChecker::setLang(const QString& lang)
|
|
{
|
|
if (m_lang != lang) {
|
|
m_lang = lang;
|
|
initHunspell();
|
|
emit langChanged();
|
|
}
|
|
}
|
|
|
|
const QString& SpellChecker::userDict() const
|
|
{
|
|
return m_userDict;
|
|
}
|
|
|
|
void SpellChecker::setUserDict(const QString& userDict)
|
|
{
|
|
if (m_userDict != userDict) {
|
|
m_userDict = userDict;
|
|
emit userDictChanged();
|
|
}
|
|
}
|