+}
+
+
+void GuiCitation::updateControls()
+{
+ BiblioInfo const & bi = bibInfo();
+ updateControls(bi);
+}
+
+
+// The main point of separating this out is that the fill*() methods
+// called in update() do not need to be called for INTERNAL updates,
+// such as when addPB is pressed, as the list of fields, entries, etc,
+// will not have changed. At the moment, however, the division between
+// fillStyles() and updateStyle() doesn't lend itself to dividing the
+// two methods, though they should be divisible.
+void GuiCitation::updateControls(BiblioInfo const & bi)
+{
+ if (selectionManager->selectedFocused()) {
+ if (selectedLV->selectionModel()->selectedIndexes().isEmpty())
+ updateInfo(bi, availableLV->currentIndex());
+ else
+ updateInfo(bi, selectedLV->currentIndex());
+ } else {
+ if (availableLV->selectionModel()->selectedIndexes().isEmpty())
+ updateInfo(bi, QModelIndex());
+ else
+ updateInfo(bi, availableLV->currentIndex());
+ }
+ setButtons();
+
+ textBeforeED->setText(toqstr(params_["before"]));
+ textAfterED->setText(toqstr(params_["after"]));
+ fillStyles(bi);
+ updateStyle();
+}
+
+
+void GuiCitation::updateFormatting(CiteStyle currentStyle)
+{
+ CiteEngine const engine = citeEngine();
+ bool const natbib_engine =
+ engine == ENGINE_NATBIB_AUTHORYEAR ||
+ engine == ENGINE_NATBIB_NUMERICAL;
+ bool const basic_engine = engine == ENGINE_BASIC;
+
+ bool const haveSelection =
+ selectedLV->model()->rowCount() > 0;
+
+ bool const isNocite = currentStyle == NOCITE;
+
+ bool const isCiteyear =
+ currentStyle == CITEYEAR ||
+ currentStyle == CITEYEARPAR;
+
+ fulllistCB->setEnabled(natbib_engine && haveSelection && !isNocite
+ && !isCiteyear);
+ forceuppercaseCB->setEnabled(natbib_engine && haveSelection
+ && !isNocite && !isCiteyear);
+ textBeforeED->setEnabled(!basic_engine && haveSelection && !isNocite);
+ textBeforeLA->setEnabled(!basic_engine && haveSelection && !isNocite);
+ textAfterED->setEnabled(haveSelection && !isNocite);
+ textAfterLA->setEnabled(haveSelection && !isNocite);
+ citationStyleCO->setEnabled(haveSelection);
+ citationStyleLA->setEnabled(haveSelection);
+}
+
+
+void GuiCitation::updateStyle()
+{
+ string const & command = params_.getCmdName();
+
+ // Find the style of the citekeys
+ vector<CiteStyle> const & styles = citeStyles_;
+ CitationStyle const cs = citationStyleFromString(command);
+
+ vector<CiteStyle>::const_iterator cit =
+ std::find(styles.begin(), styles.end(), cs.style);
+
+ // restore the latest natbib style
+ if (style_ >= 0 && style_ < citationStyleCO->count())
+ citationStyleCO->setCurrentIndex(style_);
+ else
+ citationStyleCO->setCurrentIndex(0);
+
+ if (cit != styles.end()) {
+ int const i = int(cit - styles.begin());
+ citationStyleCO->setCurrentIndex(i);
+ fulllistCB->setChecked(cs.full);
+ forceuppercaseCB->setChecked(cs.forceUpperCase);
+ } else {
+ fulllistCB->setChecked(false);
+ forceuppercaseCB->setChecked(false);
+ }
+ updateFormatting(cs.style);
+}
+
+// This one needs to be called whenever citationStyleCO needs
+// to be updated---and this would be on anything that changes the
+// selection in selectedLV, or on a general update.
+void GuiCitation::fillStyles(BiblioInfo const & bi)
+{
+ int const oldIndex = citationStyleCO->currentIndex();
+
+ citationStyleCO->clear();
+
+ QStringList selected_keys = selected_model_.stringList();
+ if (selected_keys.empty()) {
+ citationStyleCO->setEnabled(false);
+ citationStyleLA->setEnabled(false);
+ return;
+ }
+
+ int curr = selectedLV->model()->rowCount() - 1;
+ if (curr < 0)
+ return;
+
+ if (!selectedLV->selectionModel()->selectedIndexes().empty())
+ curr = selectedLV->selectionModel()->selectedIndexes()[0].row();
+
+ QStringList sty = citationStyles(bi, curr);
+
+ citationStyleCO->setEnabled(!sty.isEmpty());
+ citationStyleLA->setEnabled(!sty.isEmpty());
+
+ if (sty.isEmpty())
+ return;
+
+ citationStyleCO->insertItems(0, sty);
+
+ if (oldIndex != -1 && oldIndex < citationStyleCO->count())
+ citationStyleCO->setCurrentIndex(oldIndex);
+}
+
+
+void GuiCitation::fillFields(BiblioInfo const & bi)
+{
+ fieldsCO->blockSignals(true);
+ int const oldIndex = fieldsCO->currentIndex();
+ fieldsCO->clear();
+ QStringList const fields = to_qstring_list(bi.getFields());
+ fieldsCO->insertItem(0, qt_("All fields"));
+ fieldsCO->insertItem(1, qt_("Keys"));
+ fieldsCO->insertItems(2, fields);
+ if (oldIndex != -1 && oldIndex < fieldsCO->count())
+ fieldsCO->setCurrentIndex(oldIndex);
+ fieldsCO->blockSignals(false);
+}
+
+
+void GuiCitation::fillEntries(BiblioInfo const & bi)
+{
+ entriesCO->blockSignals(true);
+ int const oldIndex = entriesCO->currentIndex();
+ entriesCO->clear();
+ QStringList const entries = to_qstring_list(bi.getEntries());
+ entriesCO->insertItem(0, qt_("All entry types"));
+ entriesCO->insertItems(1, entries);
+ if (oldIndex != -1 && oldIndex < entriesCO->count())
+ entriesCO->setCurrentIndex(oldIndex);
+ entriesCO->blockSignals(false);
+}
+
+
+bool GuiCitation::isSelected(const QModelIndex & idx)
+{
+ QString const str = idx.data().toString();
+ return selected_model_.stringList().contains(str);
+}
+
+
+void GuiCitation::setButtons()
+{
+ selectionManager->update();
+ int const srows = selectedLV->model()->rowCount();
+ applyPB->setEnabled(srows > 0);
+ okPB->setEnabled(srows > 0);
+}
+
+
+void GuiCitation::updateInfo(BiblioInfo const & bi, QModelIndex const & idx)
+{
+ if (!idx.isValid() || bi.empty()) {
+ infoML->document()->clear();
+ return;
+ }
+
+ QString const keytxt = toqstr(
+ bi.getInfo(qstring_to_ucs4(idx.data().toString())));
+ infoML->document()->setPlainText(keytxt);
+}
+
+
+void GuiCitation::findText(QString const & text, bool reset)
+{
+ //"All Fields" and "Keys" are the first two
+ int index = fieldsCO->currentIndex() - 2;
+ BiblioInfo const & bi = bibInfo();
+ vector<docstring> const & fields = bi.getFields();
+ docstring field;
+
+ if (index <= -1 || index >= int(fields.size()))
+ //either "All Fields" or "Keys" or an invalid value
+ field = from_ascii("");
+ else
+ field = fields[index];
+
+ //Was it "Keys"?
+ bool const onlyKeys = index == -1;
+
+ //"All Entry Types" is first.
+ index = entriesCO->currentIndex() - 1;
+ vector<docstring> const & entries = bi.getEntries();
+ docstring entry_type;
+ if (index < 0 || index >= int(entries.size()))
+ entry_type = from_ascii("");
+ else
+ entry_type = entries[index];
+
+ bool const case_sentitive = caseCB->checkState();
+ bool const reg_exp = regexCB->checkState();
+ findKey(bi, text, onlyKeys, field, entry_type,
+ case_sentitive, reg_exp, reset);
+ //FIXME
+ //It'd be nice to save and restore the current selection in
+ //availableLV. Currently, we get an automatic reset, since the
+ //model is reset.
+
+ updateControls(bi);
+}
+
+
+void GuiCitation::on_fieldsCO_currentIndexChanged(int /*index*/)
+{
+ findText(findLE->text(), true);
+}
+
+
+void GuiCitation::on_entriesCO_currentIndexChanged(int /*index*/)
+{
+ findText(findLE->text(), true);
+}
+
+
+void GuiCitation::on_citationStyleCO_currentIndexChanged(int index)
+{
+ if (index >= 0 && index < citationStyleCO->count()) {
+ vector<CiteStyle> const & styles = citeStyles_;
+ updateFormatting(styles[index]);
+ }
+}
+
+
+void GuiCitation::on_findLE_textChanged(const QString & text)
+{
+ bool const searchAsWeGo = (asTypeCB->checkState() == Qt::Checked);
+ searchPB->setDisabled(text.isEmpty() || searchAsWeGo);
+ if (!text.isEmpty()) {
+ if (searchAsWeGo)
+ findText(findLE->text());
+ return;
+ }
+ findText(findLE->text());
+ findLE->setFocus();
+}
+
+void GuiCitation::on_searchPB_clicked()
+{
+ findText(findLE->text(), true);
+}
+
+
+void GuiCitation::on_caseCB_stateChanged(int)
+{
+ findText(findLE->text());
+}
+
+
+void GuiCitation::on_regexCB_stateChanged(int)
+{
+ findText(findLE->text());
+}
+
+
+void GuiCitation::on_asTypeCB_stateChanged(int)
+{
+ bool const searchAsWeGo = (asTypeCB->checkState() == Qt::Checked);
+ searchPB->setDisabled(findLE->text().isEmpty() || searchAsWeGo);
+ if (searchAsWeGo)
+ findText(findLE->text(), true);
+}
+
+
+void GuiCitation::changed()
+{
+ setButtons();
+}
+
+
+void GuiCitation::apply(int const choice, bool full, bool force,
+ QString before, QString after)
+{
+ if (cited_keys_.isEmpty())
+ return;
+
+ vector<CiteStyle> const & styles = citeStyles_;
+ if (styles[choice] == NOCITE) {
+ full = false;
+ force = false;
+ before.clear();
+ after.clear();
+ }
+
+ CitationStyle s;
+ s.style = styles[choice];
+ s.full = full;
+ s.forceUpperCase = force;
+ string const command = citationStyleToString(s);
+
+ params_.setCmdName(command);
+ params_["key"] = qstring_to_ucs4(cited_keys_.join(","));
+ params_["before"] = qstring_to_ucs4(before);
+ params_["after"] = qstring_to_ucs4(after);
+ dispatchParams();
+}
+
+
+void GuiCitation::clearSelection()
+{
+ cited_keys_.clear();
+ selected_model_.setStringList(cited_keys_);