src/frontends/controllers/ControlSpellchecker.C

   1 /**
   2  * \file ControlSpellchecker.C
   3  * Copyright 2001 the LyX Team
   4  * Read the file COPYING
   5  *
   6  * \author Edwin Leuven <leuven@fee.uva.nl>
   7  */
   8
   9 #include <config.h>
  10
  11 #ifdef __GNUG__
  12 #pragma implementation
  13 #endif
  14
  15 #include "ControlSpellchecker.h"
  16 #include "buffer.h"
  17 #include "lyxrc.h"
  18 #include "BufferView.h"
  19 #include "lyxtext.h"
  20 #include "gettext.h"
  21 #include "language.h"
  22
  23 #include "ViewBase.h"
  24 #include "Dialogs.h"
  25 #include "Liason.h"
  26
  27 #include "frontends/LyXView.h"
  28 #include "frontends/Alert.h"
  29
  30 #include "support/lstrings.h"
  31
  32 #include "ispell.h"
  33 #ifdef USE_PSPELL
  34 # include "pspell.h"
  35 #endif
  36
  37 ControlSpellchecker::ControlSpellchecker(LyXView & lv, Dialogs & d)
  38         : ControlDialogBD(lv, d),
  39           rtl_(false), newval_(0.0), oldval_(0), newvalue_(0), count_(0),
  40           stop_(false), speller_(0)
  41 {}
  42
  43
  44 void ControlSpellchecker::setParams()
  45 {
  46         if (!speller_) {
  47                 // create spell object
  48                 string tmp;
  49 #ifdef USE_PSPELL
  50                 if (lyxrc.use_pspell) {
  51                         tmp = (lyxrc.isp_use_alt_lang) ?
  52                                 lyxrc.isp_alt_lang : lv_.buffer()->params.language->code();
  53
  54                         speller_ = new PSpell(lv_.view()->buffer()->params, tmp);
  55                 } else {
  56 #endif
  57                         tmp = (lyxrc.isp_use_alt_lang) ?
  58                                 lyxrc.isp_alt_lang : lv_.buffer()->params.language->lang();
  59
  60                         speller_ = new ISpell(lv_.view()->buffer()->params, tmp);
  61 #ifdef USE_PSPELL
  62                 }
  63 #endif
  64
  65                 if (lyxrc.isp_use_alt_lang) {
  66                         Language const * lang = languages.getLanguage(tmp);
  67                         if (lang)
  68                                 rtl_ = lang->RightToLeft();
  69                 } else {
  70                         rtl_ = lv_.buffer()->params.language->RightToLeft();
  71                 }
  72
  73                 if (!speller_->error().empty()) {
  74                         clearParams();
  75                         return;
  76                 }
  77         }
  78 }
  79
  80
  81 void ControlSpellchecker::check()
  82 {
  83         SpellBase::Result res = SpellBase::OK;
  84         stop_ = false;
  85
  86         // clear any old selection
  87         LyXText * text = lv_.view()->getLyXText();
  88         lv_.view()->toggleSelection(true);
  89         lv_.view()->update(text, BufferView::SELECT);
  90
  91         while ((res == SpellBase::OK || res == SpellBase::IGNORE) && !stop_) {
  92                 word_ = lv_.view()->nextWord(newval_);
  93
  94                 if (word_.empty()) {
  95                         clearParams();
  96                         break;
  97                 }
  98
  99                 ++count_;
 100
 101                 // Update slider if and only if value has changed
 102                 newvalue_ = int(100.0*newval_);
 103                 if (newvalue_!= oldval_) {
 104                         oldval_ = newvalue_;
 105                         // set progress bar
 106                         view().partialUpdate(0);
 107                 }
 108
 109                 if (!speller_->alive()) clearParams();
 110
 111                 res = speller_->check(word_);
 112         }
 113
 114         if (!stop_ && !word_.empty())
 115                 lv_.view()->selectLastWord();
 116
 117         // set suggestions
 118         if (res != SpellBase::OK && res != SpellBase::IGNORE) {
 119                 view().partialUpdate(1);
 120         }
 121 }
 122
 123
 124 void ControlSpellchecker::replace(string const & replacement)
 125 {
 126         lv_.view()->replaceWord(replacement);
 127         check();
 128 }
 129
 130
 131 void ControlSpellchecker::replaceAll(string const & replacement)
 132 {
 133         // TODO: add to list
 134         replace(replacement);
 135 }
 136
 137
 138 void ControlSpellchecker::insert()
 139 {
 140         speller_->insert(word_);
 141         check();
 142 }
 143
 144
 145 string ControlSpellchecker::getSuggestion()
 146 {
 147         string miss(speller_->nextMiss());
 148
 149         if (rtl_)
 150                 std::reverse(miss.begin(), miss.end());
 151
 152         return miss;
 153 }
 154
 155
 156 string ControlSpellchecker::getWord()
 157 {
 158         string tmp = word_;
 159         if (rtl_)
 160                 std::reverse(tmp.begin(), tmp.end());
 161         return tmp;
 162 }
 163
 164
 165 void ControlSpellchecker::ignoreAll()
 166 {
 167         speller_->accept(word_);
 168         check();
 169 }
 170
 171
 172 void ControlSpellchecker::stop()
 173 {
 174         stop_ = true;
 175         lv_.view()->endOfSpellCheck();
 176 }
 177
 178
 179 void ControlSpellchecker::clearParams()
 180 {
 181         if (!speller_) return;
 182
 183         if (speller_->alive()) {
 184                 speller_->close();
 185                 message_ = tostr(count_);
 186                 if (count_ != 1) {
 187                         message_ += _(" words checked.");
 188
 189                 } else {
 190                         message_ += _(" word checked.");
 191                 }
 192                 message_ = "\n" + message_;
 193                 message_ = _("Spellchecking completed! ") + message_;
 194
 195         } else {
 196                 message_ = speller_->error();
 197                 speller_->cleanUp();
 198                 if (message_.empty())
 199                     message_ = _("The spell checker has died for some reason.\n"
 200                                  "Maybe it has been killed.");
 201
 202                 // make sure that the dialog is not launched
 203                 emergency_exit_ = true;
 204                 Alert::alert("The spellchecker has failed", message_);
 205         }
 206
 207         delete speller_;
 208
 209         lv_.view()->endOfSpellCheck();
 210
 211         // show closing message if any words were checked.
 212         if (count_ > 0)
 213                 view().partialUpdate(2);
 214
 215         // reset values to initial
 216         rtl_ = false;
 217         word_.erase();
 218         newval_ = 0.0;
 219         oldval_ = 0;
 220         newvalue_ = 0;
 221         count_ = 0;
 222         message_.erase();
 223         stop_ = false;
 224         speller_ = 0;
 225 }