]> git.lyx.org Git - lyx.git/blobdiff - src/HunspellChecker.cpp
typo
[lyx.git] / src / HunspellChecker.cpp
index a6a4f9420ce3e82efb255336dc2e1776d1125ac9..83bd586dcfc29dd8eb03288a5e44bf584fb532fd 100644 (file)
@@ -19,6 +19,8 @@
 
 #include "support/debug.h"
 #include "support/docstring_list.h"
+#include "support/filetools.h"
+#include "support/Package.h"
 #include "support/FileName.h"
 #include "support/gettext.h"
 #include "support/lassert.h"
@@ -29,6 +31,7 @@
 
 #include <map>
 #include <string>
+#include <vector>
 
 using namespace std;
 using namespace lyx::support;
@@ -39,20 +42,36 @@ namespace lyx {
 namespace {
 
 typedef map<std::string, Hunspell *> Spellers;
+typedef vector<WordLangTuple> IgnoreList;
 
 } // anon namespace
 
+
 struct HunspellChecker::Private
 {
        Private() {}
 
        ~Private();
 
+       const string dictPath(int selector);
+       bool haveLanguageFiles(string const & hpath);
+       bool haveDictionary(string const & lang, string & hpath);
+       bool haveDictionary(string const & lang);
+       Hunspell * addSpeller(string const & lang, string & hpath);
        Hunspell * addSpeller(string const & lang);
        Hunspell * speller(string const & lang);
+       /// ignored words
+       bool isIgnored(WordLangTuple const & wl) const;
 
        /// the spellers
        Spellers spellers_;
+       ///
+       IgnoreList ignored_;
+
+       /// the location below system/user directory
+       /// there the aff+dic files lookup will happen
+       const string dictDirectory(void) const { return "dict"; }
+       int maxLookupSelector(void) const { return 3; }
 };
 
 
@@ -62,67 +81,65 @@ HunspellChecker::Private::~Private()
        Spellers::iterator end = spellers_.end();
 
        for (; it != end; ++it) {
-               delete it->second;
+               if ( 0 != it->second) delete it->second;
        }
 }
 
 
-namespace {
-bool haveLanguageFiles(string const & hpath)
+bool HunspellChecker::Private::haveLanguageFiles(string const & hpath)
 {
        FileName const affix(hpath + ".aff");
        FileName const dict(hpath + ".dic");
-       if (!affix.isReadableFile()) {
-               // FIXME: We should indicate somehow that this language is not
-               // supported.
-               LYXERR(Debug::FILES, "Hunspell affix file " << affix << " does not exist");
-               return false;
-       }
-       if (!dict.isReadableFile()) {
-               LYXERR(Debug::FILES, "Hunspell dictionary file " << dict << " does not exist");
-               return false;
-       }
-       return true;
+       return affix.isReadableFile() && dict.isReadableFile();
 }
+
+
+const string HunspellChecker::Private::dictPath(int selector)
+{
+       switch (selector) {
+       case 2:
+               return addName(lyx::support::package().system_support().absFileName(),dictDirectory());
+               break;
+       case 1:
+               return addName(lyx::support::package().user_support().absFileName(),dictDirectory());
+               break;
+       default:
+               return lyxrc.hunspelldir_path;
+       }
 }
 
 
-Hunspell * HunspellChecker::Private::addSpeller(string const & lang)
+bool HunspellChecker::Private::haveDictionary(string const & lang, string & hpath)
 {
-       string hunspell_path = external_path(lyxrc.hunspelldir_path);
-       LYXERR(Debug::FILES, "hunspell path: " << hunspell_path);
-       if (hunspell_path.empty()) {
-               // FIXME We'd like to issue a better error message here, but there seems
-               // to be a problem about thread safety, or something of the sort. If
-               // we issue the message using frontend::Alert, then the code comes
-               // back through here while the box is waiting, and causes some kind
-               // of crash. 
-               static bool warned = false;
-               if (!warned) {
-                       warned = true;
-                       LYXERR0("Hunspell path not set.");
-                       //frontend::Alert::error(_("Hunspell Path Not Found"), 
-                       //              _("You must set the Hunspell dictionary path in Tools>Preferences>Paths."));
-               }
-               return 0;
+       if (hpath.empty()) {
+               return false;
        }
 
-       hunspell_path += "/" + lang;
-       if (!haveLanguageFiles(hunspell_path)) {
+       LYXERR(Debug::FILES, "check hunspell path: " << hpath << " for language " << lang);
+       string h_path = addName(hpath, lang);
+       if (!haveLanguageFiles(h_path)) {
                // try with '_' replaced by '-'
-               hunspell_path = subst(hunspell_path, '_', '-');
-               if (!haveLanguageFiles(hunspell_path)) {
+               h_path = addName(hpath, subst(lang, '_', '-'));
+               if (!haveLanguageFiles(h_path)) {
                        // FIXME: We should indicate somehow that this language is not
                        // supported, probably by popping a warning. But we'll need to
                        // remember which warnings we've issued.
-                       return 0;
+                       return false;
                }
        }
-       FileName const affix(hunspell_path + ".aff");
-       FileName const dict(hunspell_path + ".dic");
-       Hunspell * h = new Hunspell(affix.absFilename().c_str(), dict.absFilename().c_str());
-       spellers_[lang] = h;
-       return h;
+       hpath = h_path;
+       return true;
+}
+
+
+bool HunspellChecker::Private::haveDictionary(string const & lang)
+{
+       bool result = false;
+       for ( int p = 0; !result && p < maxLookupSelector(); p++ ) {
+               string lpath = dictPath(p);
+               result = haveDictionary(lang, lpath);
+       }
+       return result;
 }
 
 
@@ -136,6 +153,46 @@ Hunspell * HunspellChecker::Private::speller(string const & lang)
 }
 
 
+Hunspell * HunspellChecker::Private::addSpeller(string const & lang,string & path)
+{
+       if (!haveDictionary(lang, path)) {
+               spellers_[lang] = 0;
+               return 0;
+       }
+
+       FileName const affix(path + ".aff");
+       FileName const dict(path + ".dic");
+       Hunspell * h = new Hunspell(affix.absFileName().c_str(), dict.absFileName().c_str());
+       LYXERR(Debug::FILES, "Hunspell speller for langage " << lang << " at " << dict << " found");
+       spellers_[lang] = h;
+       return h;
+}
+
+
+Hunspell * HunspellChecker::Private::addSpeller(string const & lang)
+{
+       Hunspell * h = 0;
+       for ( int p = 0; p < maxLookupSelector() && 0 == h; p++ ) {
+               string lpath = dictPath(p);
+               h = addSpeller(lang, lpath);
+       }
+       return h;
+}
+
+
+bool HunspellChecker::Private::isIgnored(WordLangTuple const & wl) const
+{
+       IgnoreList::const_iterator it = ignored_.begin();
+       for (; it != ignored_.end(); ++it) {
+               if ((*it).lang()->code() != wl.lang()->code())
+                       continue;
+               if ((*it).word() == wl.word())
+                       return true;
+       }
+       return false;
+}
+
+
 HunspellChecker::HunspellChecker(): d(new Private)
 {
 }
@@ -149,13 +206,19 @@ HunspellChecker::~HunspellChecker()
 
 SpellChecker::Result HunspellChecker::check(WordLangTuple const & wl)
 {
-       string const word_to_check = to_utf8(wl.word());
-       Hunspell * h = d->speller(wl.lang_code());
+       if (d->isIgnored(wl))
+               return WORD_OK;
+
+       Hunspell * h = d->speller(wl.lang()->code());
        if (!h)
-               return OK;
+               return WORD_OK;
        int info;
+
+       string const encoding = h->get_dic_encoding();
+       string const word_to_check = to_iconv_encoding(wl.word(), encoding);
+       
        if (h->spell(word_to_check.c_str(), &info))
-               return OK;
+               return WORD_OK;
 
        if (info & SPELL_COMPOUND) {
                // FIXME: What to do with that?
@@ -170,19 +233,27 @@ SpellChecker::Result HunspellChecker::check(WordLangTuple const & wl)
 }
 
 
+void HunspellChecker::advanceChangeNumber()
+{
+       nextChangeNumber();
+}
+
+
 void HunspellChecker::insert(WordLangTuple const & wl)
 {
        string const word_to_check = to_utf8(wl.word());
-       Hunspell * h = d->speller(wl.lang_code());
+       Hunspell * h = d->speller(wl.lang()->code());
        if (!h)
                return;
        h->add(word_to_check.c_str());
+       advanceChangeNumber();
 }
 
 
-void HunspellChecker::accept(WordLangTuple const &)
+void HunspellChecker::accept(WordLangTuple const & wl)
 {
-       // FIXME: not implemented!
+       d->ignored_.push_back(wl);
+       advanceChangeNumber();
 }
 
 
@@ -190,20 +261,29 @@ void HunspellChecker::suggest(WordLangTuple const & wl,
        docstring_list & suggestions)
 {
        suggestions.clear();
-       string const word_to_check = to_utf8(wl.word());
-       Hunspell * h = d->speller(wl.lang_code());
+       Hunspell * h = d->speller(wl.lang()->code());
        if (!h)
                return;
+       string const encoding = h->get_dic_encoding();
+       string const word_to_check = to_iconv_encoding(wl.word(), encoding);
        char ** suggestion_list;
        int const suggestion_number = h->suggest(&suggestion_list, word_to_check.c_str());
        if (suggestion_number <= 0)
                return;
        for (int i = 0; i != suggestion_number; ++i)
-               suggestions.push_back(from_utf8(suggestion_list[i]));
+               suggestions.push_back(from_iconv_encoding(suggestion_list[i], encoding));
        h->free_list(&suggestion_list, suggestion_number);
 }
 
 
+bool HunspellChecker::hasDictionary(Language const * lang) const
+{
+       if (!lang)
+               return false;
+       return (d->haveDictionary(lang->code()));
+}
+
+
 docstring const HunspellChecker::error()
 {
        return docstring();