src/frontends/qt4/GuiSpellchecker.cpp

   1 /**
   2  * \file GuiSpellchecker.cpp
   3  * This file is part of LyX, the document processor.
   4  * Licence details can be found in the file COPYING.
   5  *
   6  * \author John Levon
   7  * \author Edwin Leuven
   8  * \author Abdelrazak Younes
   9  *
  10  * Full author contact details are available in file CREDITS.
  11  */
  12
  13 #include <config.h>
  14
  15 #include "GuiSpellchecker.h"
  16 #include "GuiApplication.h"
  17
  18 #include "qt_helpers.h"
  19
  20 #include "ui_SpellcheckerUi.h"
  21
  22 #include "Buffer.h"
  23 #include "BufferParams.h"
  24 #include "BufferView.h"
  25 #include "buffer_funcs.h"
  26 #include "Cursor.h"
  27 #include "CutAndPaste.h"
  28 #include "FuncRequest.h"
  29 #include "Language.h"
  30 #include "LyX.h"
  31 #include "LyXRC.h"
  32 #include "lyxfind.h"
  33 #include "Paragraph.h"
  34 #include "WordLangTuple.h"
  35
  36 #include "support/debug.h"
  37 #include "support/docstring.h"
  38 #include "support/docstring_list.h"
  39 #include "support/ExceptionMessage.h"
  40 #include "support/gettext.h"
  41 #include "support/lstrings.h"
  42 #include "support/textutils.h"
  43
  44 #include <QListWidgetItem>
  45 #include <QKeyEvent>
  46
  47 #include "SpellChecker.h"
  48
  49 #include "frontends/alert.h"
  50
  51 using namespace std;
  52 using namespace lyx::support;
  53
  54 namespace lyx {
  55 namespace frontend {
  56
  57
  58 struct GuiSpellchecker::Private
  59 {
  60         Private() : progress_(0), count_(0) {}
  61         Ui::SpellcheckerUi ui;
  62         /// current word being checked and lang code
  63         WordLangTuple word_;
  64         /// values for progress
  65         int total_;
  66         int progress_;
  67         /// word count
  68         int count_;
  69 };
  70
  71
  72 GuiSpellchecker::GuiSpellchecker(GuiView & lv)
  73         : DockView(lv, "spellchecker", qt_("Spellchecker"),
  74         Qt::RightDockWidgetArea), d(new GuiSpellchecker::Private)
  75 {
  76         d->ui.setupUi(this);
  77
  78         connect(d->ui.suggestionsLW, SIGNAL(itemDoubleClicked(QListWidgetItem*)),
  79                 this, SLOT(on_replacePB_clicked()));
  80
  81         // language
  82         QAbstractItemModel * language_model = guiApp->languageModel();
  83         // FIXME: it would be nice if sorting was enabled/disabled via a checkbox.
  84         language_model->sort(0);
  85         d->ui.languageCO->setModel(language_model);
  86
  87         d->ui.wordED->setReadOnly(true);
  88
  89         d->ui.suggestionsLW->installEventFilter(this);
  90 }
  91
  92
  93 GuiSpellchecker::~GuiSpellchecker()
  94 {
  95         delete d;
  96 }
  97
  98
  99 void GuiSpellchecker::on_closePB_clicked()
 100 {
 101         close();
 102 }
 103
 104
 105 bool GuiSpellchecker::eventFilter(QObject *obj, QEvent *event)
 106 {
 107         if (obj == d->ui.suggestionsLW && event->type() == QEvent::KeyPress) {
 108                 QKeyEvent *e = static_cast<QKeyEvent *> (event);
 109                 if (e->key() == Qt::Key_Enter || e->key() == Qt::Key_Return) {
 110                         on_suggestionsLW_itemClicked(d->ui.suggestionsLW->currentItem());
 111                         on_replacePB_clicked();
 112                         return true;
 113                 } else if (e->key() == Qt::Key_Right) {
 114                         on_suggestionsLW_itemClicked(d->ui.suggestionsLW->currentItem());
 115                         return true;
 116                 }
 117         }
 118         // standard event processing
 119         return QWidget::eventFilter(obj, event);
 120 }
 121
 122
 123 void GuiSpellchecker::on_suggestionsLW_itemClicked(QListWidgetItem * item)
 124 {
 125         if (d->ui.replaceCO->count() != 0)
 126                 d->ui.replaceCO->setItemText(0, item->text());
 127         else
 128                 d->ui.replaceCO->addItem(item->text());
 129
 130         d->ui.replaceCO->setCurrentIndex(0);
 131 }
 132
 133
 134 void GuiSpellchecker::on_replaceCO_highlighted(const QString & str)
 135 {
 136         QListWidget * lw = d->ui.suggestionsLW;
 137         if (lw->currentItem() && lw->currentItem()->text() == str)
 138                 return;
 139
 140         for (int i = 0; i != lw->count(); ++i) {
 141                 if (lw->item(i)->text() == str) {
 142                         lw->setCurrentRow(i);
 143                         break;
 144                 }
 145         }
 146 }
 147
 148
 149 void GuiSpellchecker::updateView()
 150 {
 151         if (hasFocus())
 152                 check();
 153 }
 154
 155
 156 void GuiSpellchecker::on_languageCO_activated(int index)
 157 {
 158         string const lang =
 159                 fromqstr(d->ui.languageCO->itemData(index).toString());
 160         if (d->word_.lang()->lang() == lang)
 161                 // nothing changed
 162                 return;
 163         dispatch(FuncRequest(LFUN_LANGUAGE, lang));
 164         check();
 165 }
 166
 167
 168 void GuiSpellchecker::on_ignoreAllPB_clicked()
 169 {
 170         /// replace all occurances of word
 171         theSpellChecker()->accept(d->word_);
 172         check();
 173 }
 174
 175
 176 void GuiSpellchecker::on_addPB_clicked()
 177 {
 178         /// insert word in personal dictionary
 179         theSpellChecker()->insert(d->word_);
 180         check();
 181 }
 182
 183
 184 void GuiSpellchecker::on_ignorePB_clicked()
 185 {
 186         dispatch(FuncRequest(LFUN_CHAR_FORWARD));
 187         check();
 188 }
 189
 190
 191 void GuiSpellchecker::on_findNextPB_clicked()
 192 {
 193         docstring const data = find2string(
 194                                 qstring_to_ucs4(d->ui.wordED->text()),
 195                                 true, true, true);
 196         dispatch(FuncRequest(LFUN_WORD_FIND, data));
 197 }
 198
 199
 200 void GuiSpellchecker::on_replacePB_clicked()
 201 {
 202         docstring const replacement = qstring_to_ucs4(d->ui.replaceCO->currentText());
 203
 204         LYXERR(Debug::GUI, "Replace (" << replacement << ")");
 205         BufferView * bv = const_cast<BufferView *>(bufferview());
 206         if (!bv->cursor().inTexted())
 207                 return;
 208         cap::replaceSelectionWithString(bv->cursor(), replacement, true);
 209         bv->buffer().markDirty();
 210         // If we used an LFUN, we would not need that
 211         bv->processUpdateFlags(Update::Force | Update::FitCursor);
 212         // fix up the count
 213         --d->count_;
 214         check();
 215 }
 216
 217
 218 void GuiSpellchecker::on_replaceAllPB_clicked()
 219 {
 220         docstring const data = replace2string(
 221                 qstring_to_ucs4(d->ui.replaceCO->currentText()),
 222                 qstring_to_ucs4(d->ui.wordED->text()),
 223                 true, true, true, true);
 224         dispatch(FuncRequest(LFUN_WORD_REPLACE, data));
 225         check(); // continue spellchecking
 226 }
 227
 228
 229 void GuiSpellchecker::updateSuggestions(docstring_list & words)
 230 {
 231         QString const suggestion = toqstr(d->word_.word());
 232         d->ui.wordED->setText(suggestion);
 233         QListWidget * lw = d->ui.suggestionsLW;
 234         lw->clear();
 235
 236         if (words.empty()) {
 237                 on_suggestionsLW_itemClicked(new QListWidgetItem(suggestion));
 238                 return;
 239         }
 240         for (size_t i = 0; i != words.size(); ++i)
 241                 lw->addItem(toqstr(words[i]));
 242
 243         on_suggestionsLW_itemClicked(lw->item(0));
 244         lw->setCurrentRow(0);
 245 }
 246
 247
 248 bool GuiSpellchecker::initialiseParams(string const &)
 249 {
 250         LYXERR(Debug::GUI, "Spellchecker::initialiseParams");
 251
 252         if (!theSpellChecker())
 253                 return false;
 254
 255         DocIterator const begin = doc_iterator_begin(&buffer());
 256         Cursor const & cur = bufferview()->cursor();
 257         d->progress_ = countWords(begin, cur);
 258         d->total_ = d->progress_ + countWords(cur, doc_iterator_end(&buffer()));
 259         d->count_ = 0;
 260         return true;
 261 }
 262
 263
 264 void GuiSpellchecker::check()
 265 {
 266         LYXERR(Debug::GUI, "Check the spelling of a word");
 267
 268         DocIterator from = bufferview()->cursor();
 269         DocIterator to;
 270         WordLangTuple word_lang;
 271         docstring_list suggestions;
 272
 273         int progress;
 274         try {
 275                 progress = buffer().spellCheck(from, to, word_lang, suggestions);
 276         } catch (ExceptionMessage const & message) {
 277                 if (message.type_ == WarningException) {
 278                         Alert::warning(message.title_, message.details_);
 279                         close();
 280                         return;
 281                 }
 282                 throw message;
 283         }
 284         LYXERR(Debug::GUI, "Found word \"" << word_lang.word() << "\"");
 285         d->count_ += progress;
 286         d->progress_ += progress;
 287
 288         // end of document
 289         if (from == doc_iterator_end(&buffer())) {
 290                 showSummary();
 291                 return;
 292         }
 293         if (!isVisible())
 294                 show();
 295
 296         d->word_ = word_lang;
 297
 298         int const progress_bar = d->total_
 299                 ? int(100.0 * float(d->progress_)/d->total_) : 100;
 300         LYXERR(Debug::GUI, "Updating spell progress.");
 301         // set progress bar
 302         d->ui.spellcheckPR->setValue(progress_bar);
 303         // set suggestions
 304         updateSuggestions(suggestions);
 305         // set language
 306         int const pos = d->ui.languageCO->findData(toqstr(word_lang.lang()->lang()));
 307         if (pos != -1)
 308                 d->ui.languageCO->setCurrentIndex(pos);
 309
 310         // FIXME: if we used a lfun like in find/replace, dispatch would do
 311         // that for us
 312         int const size = to.pos() - from.pos();
 313         BufferView * bv = const_cast<BufferView *>(bufferview());
 314         bv->putSelectionAt(from, size, false);
 315 }
 316
 317
 318 void GuiSpellchecker::showSummary()
 319 {
 320         if (d->count_ == 0) {
 321                 close();
 322                 return;
 323         }
 324
 325         docstring message;
 326         if (d->count_ != 1)
 327                 message = bformat(_("%1$d words checked."), d->count_);
 328         else
 329                 message = _("One word checked.");
 330
 331         close();
 332         Alert::information(_("Spelling check completed"), message);
 333 }
 334
 335
 336 Dialog * createGuiSpellchecker(GuiView & lv) { return new GuiSpellchecker(lv); }
 337
 338 } // namespace frontend
 339 } // namespace lyx
 340
 341 #include "moc_GuiSpellchecker.cpp"