#include "Cursor.h"
#include "DocIterator.h"
#include "FuncRequest.h"
-#include "LyXFunc.h"
+#include "LyX.h"
+#include "qt_helpers.h"
#include "TocBackend.h"
-#include "support/convert.h"
#include "support/debug.h"
#include "support/lassert.h"
+#include <QSortFilterProxyModel>
+#include <QStandardItemModel>
+
+
#include <climits>
using namespace std;
namespace lyx {
namespace frontend {
+/// A QStandardItemModel that gives access to the reset methods.
+/// This is needed in order to fix http://www.lyx.org/trac/ticket/3740
+// FIXME: Better appropriately subclass QStandardItemModel and implement
+// the toc-specific reset methods there.
+class TocTypeModel : public QStandardItemModel
+{
+public:
+ ///
+ TocTypeModel(QObject * parent) : QStandardItemModel(parent)
+ {}
+ ///
+ void reset()
+ {
+ QStandardItemModel::beginResetModel();
+ QStandardItemModel::endResetModel();
+ }
+ ///
+ void beginResetModel()
+ {
+ QStandardItemModel::beginResetModel();
+ }
+ ///
+ void endResetModel()
+ {
+ QStandardItemModel::endResetModel();
+ }
+};
+
+
+///////////////////////////////////////////////////////////////////////////////
+//
+// TocModel
+//
+///////////////////////////////////////////////////////////////////////////////
+
+TocModel::TocModel(QObject * parent)
+ : model_(new TocTypeModel(parent)),
+ sorted_model_(new QSortFilterProxyModel(parent)),
+ is_sorted_(false), toc_(new Toc()),
+ maxdepth_(0), mindepth_(0)
+{
+ sorted_model_->setSortLocaleAware(true);
+ sorted_model_->setSourceModel(model_);
+}
+
+
+QAbstractItemModel * TocModel::model()
+{
+ if (is_sorted_)
+ return sorted_model_;
+ return model_;
+}
+
+
+QAbstractItemModel const * TocModel::model() const
+{
+ if (is_sorted_)
+ return sorted_model_;
+ return model_;
+}
+
+
+void TocModel::clear()
+{
+ model_->blockSignals(true);
+ model_->clear();
+ toc_ = make_shared<Toc>();
+ model_->blockSignals(false);
+}
+
+
+void TocModel::sort(bool sort_it)
+{
+ is_sorted_ = sort_it;
+ if (is_sorted_)
+ sorted_model_->sort(0);
+}
+
TocItem const & TocModel::tocItem(QModelIndex const & index) const
{
- return toc_[data(index, Qt::UserRole).toUInt()];
+ return (*toc_)[model()->data(index, Qt::UserRole).toUInt()];
}
QModelIndex TocModel::modelIndex(DocIterator const & dit) const
{
- size_t const toc_index = toc_.item(dit) - toc_.begin();
+ if (toc_->empty())
+ return QModelIndex();
+
+ unsigned int const toc_index = TocBackend::findItem(*toc_, dit) -
+ toc_->begin();
- QModelIndexList list = match(index(0, 0), Qt::UserRole,
+ QModelIndexList list = model()->match(model()->index(0, 0), Qt::UserRole,
QVariant(toc_index), 1,
Qt::MatchFlags(Qt::MatchExactly | Qt::MatchRecursive));
}
-void TocModel::clear()
+void TocModel::reset()
+{
+ model_->reset();
+}
+
+
+void TocModel::setString(TocItem const & item, QModelIndex index)
+{
+ // Use implicit sharing of QStrings
+ QString str = toqstr(item.asString());
+ model_->setData(index, str, Qt::DisplayRole);
+ model_->setData(index, str, Qt::ToolTipRole);
+}
+
+
+void TocModel::updateItem(DocIterator const & dit)
{
- QStandardItemModel::clear();
- removeRows(0, rowCount());
- removeColumns(0, columnCount());
+ QModelIndex const index = modelIndex(dit);
+ setString(tocItem(index), index);
}
-TocModel::TocModel(Toc const & toc): toc_(toc)
+void TocModel::reset(shared_ptr<Toc const> toc)
{
- if (toc_.empty())
+ toc_ = toc;
+ if (toc_->empty()) {
+ maxdepth_ = 0;
+ mindepth_ = 0;
+ reset();
return;
- int current_row;
- QModelIndex top_level_item;
- insertColumns(0, 1);
+ }
+
+ model_->blockSignals(true);
+ model_->beginResetModel();
+ model_->insertColumns(0, 1);
maxdepth_ = 0;
mindepth_ = INT_MAX;
- size_t end = toc.size();
- for (size_t index = 0; index != end; ++index) {
- TocItem const & item = toc_[index];
+ size_t end = toc_->size();
+ for (unsigned int index = 0; index != end; ++index) {
+ TocItem const & item = (*toc_)[index];
maxdepth_ = max(maxdepth_, item.depth());
mindepth_ = min(mindepth_, item.depth());
- current_row = rowCount();
- insertRows(current_row, 1);
- top_level_item = QStandardItemModel::index(current_row, 0);
- setData(top_level_item, toqstr(item.str()), Qt::DisplayRole);
- setData(top_level_item, index, Qt::UserRole);
+ int current_row = model_->rowCount();
+ model_->insertRows(current_row, 1);
+ QModelIndex top_level_item = model_->index(current_row, 0);
+ setString(item, top_level_item);
+ model_->setData(top_level_item, index, Qt::UserRole);
LYXERR(Debug::GUI, "Toc: at depth " << item.depth()
- << ", added item " << item.str());
+ << ", added item " << item.asString());
populate(index, top_level_item);
if (index >= end)
break;
}
- setHeaderData(0, Qt::Horizontal, QVariant("title"), Qt::DisplayRole);
-// emit headerDataChanged();
+ model_->setHeaderData(0, Qt::Horizontal, QVariant("title"), Qt::DisplayRole);
+ sorted_model_->setSourceModel(model_);
+ if (is_sorted_)
+ sorted_model_->sort(0);
+ model_->blockSignals(false);
+ model_->endResetModel();
}
-void TocModel::populate(size_t & index, QModelIndex const & parent)
+void TocModel::populate(unsigned int & index, QModelIndex const & parent)
{
- int curdepth = toc_[index].depth() + 1;
+ int curdepth = (*toc_)[index].depth() + 1;
- int current_row;
QModelIndex child_item;
- insertColumns(0, 1, parent);
+ model_->insertColumns(0, 1, parent);
- size_t end = toc_.size();
+ size_t end = toc_->size();
++index;
for (; index != end; ++index) {
- TocItem const & item = toc_[index];
+ TocItem const & item = (*toc_)[index];
if (item.depth() < curdepth) {
--index;
return;
}
maxdepth_ = max(maxdepth_, item.depth());
mindepth_ = min(mindepth_, item.depth());
- current_row = rowCount(parent);
- insertRows(current_row, 1, parent);
- child_item = QStandardItemModel::index(current_row, 0, parent);
- setData(child_item, toqstr(item.str()), Qt::DisplayRole);
- setData(child_item, index, Qt::UserRole);
+ int current_row = model_->rowCount(parent);
+ model_->insertRows(current_row, 1, parent);
+ child_item = model_->index(current_row, 0, parent);
+ setString(item, child_item);
+ model_->setData(child_item, index, Qt::UserRole);
populate(index, child_item);
if (index >= end)
break;
int TocModel::modelDepth() const
{
- return maxdepth_ - mindepth_;
+ int const d = maxdepth_ - mindepth_;
+ LASSERT(d >= 0 && d <= 100, return 0);
+ return d;
}
///////////////////////////////////////////////////////////////////////////////
-// TocModels implementation.
+//
+// TocModels
+//
///////////////////////////////////////////////////////////////////////////////
-void TocModels::clear()
+
+TocModels::TocModels()
+ : bv_(0)
{
- types_.clear();
- type_names_.clear();
- const unsigned int size = models_.size();
- for (unsigned int i = 0; i < size; ++i) {
- delete models_[i];
- }
- models_.clear();
+ names_ = new TocTypeModel(this);
+ names_sorted_ = new TocModelSortProxyModel(this);
+ names_sorted_->setSourceModel(names_);
+ names_sorted_->setSortLocaleAware(true);
+ names_sorted_->sort(0);
}
-int TocModels::depth(int type)
+void TocModels::clear()
{
- if (type < 0)
- return 0;
- return models_[type]->modelDepth();
+ names_->blockSignals(true);
+ names_->clear();
+ names_->blockSignals(false);
+ iterator end = models_.end();
+ for (iterator it = models_.begin(); it != end; ++it)
+ it.value()->clear();
}
-QStandardItemModel * TocModels::model(int type)
+int TocModels::depth(QString const & type)
{
- if (type < 0)
+ const_iterator it = models_.find(type);
+ if (!bv_ || it == models_.end())
return 0;
+ return it.value()->modelDepth();
+}
+
- if (models_.empty()) {
- LYXERR(Debug::GUI, "TocModels::tocModel(): no types available ");
+QAbstractItemModel * TocModels::model(QString const & type)
+{
+ if (!bv_)
return 0;
- }
+ iterator it = models_.find(type);
+ if (it != models_.end())
+ return it.value()->model();
+ LYXERR0("type not found: " << type);
+ return 0;
+}
- LYXERR(Debug::GUI, "TocModels: type " << type
- << " models_.size() " << models_.size());
- LASSERT(type >= 0 && type < int(models_.size()), /**/);
- return models_[type];
+QAbstractItemModel * TocModels::nameModel()
+{
+ return names_sorted_;
}
-QModelIndex TocModels::currentIndex(int type) const
+QModelIndex TocModels::currentIndex(QString const & type) const
{
- if (type < 0 || !bv_)
+ const_iterator it = models_.find(type);
+ if (!bv_ || it == models_.end())
return QModelIndex();
- return models_[type]->modelIndex(bv_->cursor());
+ return it.value()->modelIndex(bv_->cursor());
}
-void TocModels::goTo(int type, QModelIndex const & index) const
+void TocModels::goTo(QString const & type, QModelIndex const & index) const
{
- if (type < 0 || !index.isValid()
- || index.model() != models_[type]) {
+ const_iterator it = models_.find(type);
+ if (it == models_.end() || !index.isValid()) {
LYXERR(Debug::GUI, "TocModels::goTo(): QModelIndex is invalid!");
return;
}
-
- LASSERT(type >= 0 && type < int(models_.size()), /**/);
- TocItem const item = models_[type]->tocItem(index);
- LYXERR(Debug::GUI, "TocModels::goTo " << item.str());
+ LASSERT(index.model() == it.value()->model(), return);
+ TocItem const item = it.value()->tocItem(index);
+ LYXERR(Debug::GUI, "TocModels::goTo " << item.asString());
dispatch(item.action());
}
-void TocModels::updateBackend() const
+TocItem const TocModels::currentItem(QString const & type,
+ QModelIndex const & index) const
+{
+ const_iterator it = models_.find(type);
+ if (it == models_.end() || !index.isValid()) {
+ LYXERR(Debug::GUI, "TocModels::currentItem(): QModelIndex is invalid!");
+ return TocItem();
+ }
+ LASSERT(index.model() == it.value()->model(), return TocItem());
+
+ return it.value()->tocItem(index);
+}
+
+
+void TocModels::updateItem(QString const & type, DocIterator const & dit)
{
- bv_->buffer().masterBuffer()->tocBackend().update();
- bv_->buffer().structureChanged();
+ models_[type]->updateItem(dit);
}
{
bv_ = bv;
clear();
- if (!bv_)
+ if (!bv_) {
+ iterator end = models_.end();
+ for (iterator it = models_.begin(); it != end; ++it)
+ it.value()->reset();
+ names_->reset();
return;
+ }
+ names_->blockSignals(true);
+ names_->beginResetModel();
+ names_->insertColumns(0, 1);
TocList const & tocs = bv_->buffer().masterBuffer()->tocBackend().tocs();
TocList::const_iterator it = tocs.begin();
- TocList::const_iterator end = tocs.end();
- for (; it != end; ++it) {
- types_.push_back(toqstr(it->first));
- type_names_.push_back(guiName(it->first, bv->buffer().params()));
- models_.push_back(new TocModel(it->second));
+ TocList::const_iterator toc_end = tocs.end();
+ for (; it != toc_end; ++it) {
+ QString const type = toqstr(it->first);
+
+ // First, fill in the toc models.
+ iterator mod_it = models_.find(type);
+ if (mod_it == models_.end())
+ mod_it = models_.insert(type, new TocModel(this));
+ mod_it.value()->reset(it->second);
+
+ // Fill in the names_ model.
+ QString const gui_name = guiName(it->first, bv->buffer().params());
+ int const current_row = names_->rowCount();
+ names_->insertRows(current_row, 1);
+ QModelIndex const index = names_->index(current_row, 0);
+ names_->setData(index, gui_name, Qt::DisplayRole);
+ names_->setData(index, type, Qt::UserRole);
}
+ names_->blockSignals(false);
+ names_->endResetModel();
}
-bool TocModels::canOutline(int type) const
+bool TocModels::isSorted(QString const & type) const
{
- if (type < 0 || type >= types_.size())
+ const_iterator it = models_.find(type);
+ if (it == models_.end()) {
+ LYXERR0("type not found: " << type);
return false;
- return types_[type] == "tableofcontents";
+ }
+ return it.value()->isSorted();
}
-int TocModels::decodeType(QString const & str) const
+void TocModels::sort(QString const & type, bool sort_it)
{
- QString new_type;
- if (str.contains("tableofcontents")) {
- new_type = "tableofcontents";
- } else if (str.contains("floatlist")) {
- if (str.contains("\"figure"))
- new_type = "figure";
- else if (str.contains("\"table"))
- new_type = "table";
- else if (str.contains("\"algorithm"))
- new_type = "algorithm";
- } else if (!str.isEmpty()) {
- new_type = str;
- } else {
- // Default to Outliner.
- new_type = "tableofcontents";
- }
- int const type = types_.indexOf(new_type);
- if (type != -1)
- return type;
- // If everything else fails, settle on the table of contents which is
- // guaranted to exist.
- return types_.indexOf("tableofcontents");
+ iterator it = models_.find(type);
+ if (it == models_.end())
+ LYXERR0("type not found: " << type);
+ else
+ it.value()->sort(sort_it);
}
} // namespace frontend
} // namespace lyx
-#include "TocModel_moc.cpp"
+#include "moc_TocModel.cpp"