#include "TocModel.h"
+#include "Buffer.h"
+#include "BufferView.h"
+#include "Cursor.h"
+#include "DocIterator.h"
+#include "FuncRequest.h"
+#include "LyX.h"
+#include "TocBackend.h"
+
#include "support/debug.h"
+#include "support/lassert.h"
+
+#include <QSortFilterProxyModel>
-#include <boost/assert.hpp>
#include <climits>
using namespace std;
namespace lyx {
namespace frontend {
-typedef std::pair<QModelIndex, TocIterator> TocPair;
+/// A QStandardItemModel that gives access to the reset methods.
+/// This is needed in order to fix http://www.lyx.org/trac/ticket/3740
+class TocTypeModel : public QStandardItemModel
+{
+public:
+ ///
+ TocTypeModel(QObject * parent) : QStandardItemModel(parent)
+ {}
+ ///
+ void reset()
+ {
+ QStandardItemModel::reset();
+ }
+ ///
+ void beginResetModel()
+ {
+ #if QT_VERSION >= 0x040600
+ QStandardItemModel::beginResetModel();
+ #endif
+ }
+ ///
+ void endResetModel()
+ {
+ #if QT_VERSION >= 0x040600
+ QStandardItemModel::endResetModel();
+ #else
+ QStandardItemModel::reset();
+ #endif
+ }
+};
+
+///////////////////////////////////////////////////////////////////////////////
+//
+// TocModel
+//
+///////////////////////////////////////////////////////////////////////////////
-TocIterator TocModel::tocIterator(QModelIndex const & index) const
+TocModel::TocModel(QObject * parent)
+ : model_(new TocTypeModel(parent)),
+ sorted_model_(new QSortFilterProxyModel(parent)),
+ is_sorted_(false), maxdepth_(0), mindepth_(0)
{
- TocMap::const_iterator map_it = toc_map_.find(index);
- BOOST_ASSERT(map_it != toc_map_.end());
- return map_it->second;
+#if QT_VERSION >= 0x040300
+ sorted_model_->setSortLocaleAware(true);
+#endif
+ sorted_model_->setSourceModel(model_);
}
-QModelIndex TocModel::modelIndex(TocIterator const & it) const
+QAbstractItemModel * TocModel::model()
{
- ModelMap::const_iterator map_it = model_map_.find(it);
- //BOOST_ASSERT(it != model_map_.end());
+ if (is_sorted_)
+ return sorted_model_;
+ return model_;
+}
- if (map_it == model_map_.end())
- return QModelIndex();
- return map_it->second;
+QAbstractItemModel const * TocModel::model() const
+{
+ if (is_sorted_)
+ return sorted_model_;
+ return model_;
}
void TocModel::clear()
{
- QStandardItemModel::clear();
- toc_map_.clear();
- model_map_.clear();
- removeRows(0, rowCount());
- removeColumns(0, columnCount());
+ model_->blockSignals(true);
+ model_->clear();
+ model_->blockSignals(false);
}
-void TocModel::populate(Toc const & toc)
+void TocModel::sort(bool sort_it)
{
- clear();
+ is_sorted_ = sort_it;
+ if (is_sorted_)
+ sorted_model_->sort(0);
+}
- if (toc.empty())
- return;
- int current_row;
- QModelIndex top_level_item;
- TocIterator iter = toc.begin();
- TocIterator end = toc.end();
+TocItem const & TocModel::tocItem(QModelIndex const & index) const
+{
+ return (*toc_)[model()->data(index, Qt::UserRole).toUInt()];
+}
- insertColumns(0, 1);
- maxdepth_ = 0;
- mindepth_ = INT_MAX;
- while (iter != end) {
- maxdepth_ = max(maxdepth_, iter->depth());
- mindepth_ = min(mindepth_, iter->depth());
- current_row = rowCount();
- insertRows(current_row, 1);
- top_level_item = QStandardItemModel::index(current_row, 0);
- //setData(top_level_item, toqstr(iter->str()));
- setData(top_level_item, toqstr(iter->str()), Qt::DisplayRole);
+QModelIndex TocModel::modelIndex(DocIterator const & dit) const
+{
+ if (toc_->empty())
+ return QModelIndex();
- // This looks like a gcc bug, in principle this should work:
- //toc_map_[top_level_item] = iter;
- // but it crashes with gcc-4.1 and 4.0.2
- toc_map_.insert( TocPair(top_level_item, iter) );
- model_map_[iter] = top_level_item;
+ unsigned int const toc_index = toc_->item(dit) - toc_->begin();
- LYXERR(Debug::GUI, "Toc: at depth " << iter->depth()
- << ", added item " << to_utf8(iter->str()));
+ QModelIndexList list = model()->match(model()->index(0, 0), Qt::UserRole,
+ QVariant(toc_index), 1,
+ Qt::MatchFlags(Qt::MatchExactly | Qt::MatchRecursive));
- populate(iter, end, top_level_item);
+ LASSERT(!list.isEmpty(), return QModelIndex());
+ return list[0];
+}
+
+
+void TocModel::reset()
+{
+ model_->reset();
+}
- if (iter == end)
- break;
- ++iter;
+void TocModel::updateItem(DocIterator const & dit)
+{
+ QModelIndex index = modelIndex(dit);
+ TocItem const & toc_item = tocItem(index);
+ model_->setData(index, toqstr(toc_item.str()), Qt::DisplayRole);
+ model_->setData(index, toqstr(toc_item.tooltip()), Qt::ToolTipRole);
+}
+
+
+void TocModel::reset(Toc const & toc)
+{
+ toc_ = &toc;
+ if (toc_->empty()) {
+ maxdepth_ = 0;
+ mindepth_ = 0;
+ reset();
+ return;
}
- setHeaderData(0, Qt::Horizontal, QVariant("title"), Qt::DisplayRole);
-// emit headerDataChanged();
+ model_->blockSignals(true);
+ model_->beginResetModel();
+ model_->insertColumns(0, 1);
+ maxdepth_ = 0;
+ mindepth_ = INT_MAX;
+
+ 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());
+ int current_row = model_->rowCount();
+ model_->insertRows(current_row, 1);
+ QModelIndex top_level_item = model_->index(current_row, 0);
+ model_->setData(top_level_item, toqstr(item.str()), Qt::DisplayRole);
+ model_->setData(top_level_item, index, Qt::UserRole);
+ model_->setData(top_level_item, toqstr(item.tooltip()), Qt::ToolTipRole);
+
+ LYXERR(Debug::GUI, "Toc: at depth " << item.depth()
+ << ", added item " << item.str());
+
+ populate(index, top_level_item);
+ if (index >= end)
+ break;
+ }
+
+ model_->setHeaderData(0, Qt::Horizontal, QVariant("title"), Qt::DisplayRole);
+ if (is_sorted_)
+ sorted_model_->sort(0);
+ model_->blockSignals(false);
+ model_->endResetModel();
}
-void TocModel::populate(TocIterator & iter, TocIterator const & end,
- QModelIndex const & parent)
+void TocModel::populate(unsigned int & index, QModelIndex const & parent)
{
- int curdepth = iter->depth() + 1;
+ int curdepth = (*toc_)[index].depth() + 1;
int current_row;
QModelIndex child_item;
+ model_->insertColumns(0, 1, parent);
+
+ size_t end = toc_->size();
+ ++index;
+ for (; index != end; ++index) {
+ TocItem const & item = (*toc_)[index];
+ if (item.depth() < curdepth) {
+ --index;
+ return;
+ }
+ maxdepth_ = max(maxdepth_, item.depth());
+ mindepth_ = min(mindepth_, item.depth());
+ current_row = model_->rowCount(parent);
+ model_->insertRows(current_row, 1, parent);
+ child_item = model_->index(current_row, 0, parent);
+ model_->setData(child_item, toqstr(item.str()), Qt::DisplayRole);
+ model_->setData(child_item, index, Qt::UserRole);
+ model_->setData(child_item, toqstr(item.tooltip()), Qt::ToolTipRole);
+ populate(index, child_item);
+ if (index >= end)
+ break;
+ }
+}
- insertColumns(0, 1, parent);
- while (iter != end) {
- ++iter;
+int TocModel::modelDepth() const
+{
+ int const d = maxdepth_ - mindepth_;
+ LASSERT(d >= 0 && d <= 100, return 0);
+ return d;
+}
- if (iter == end)
- break;
- if (iter->depth() < curdepth) {
- --iter;
- return;
- }
+///////////////////////////////////////////////////////////////////////////////
+//
+// TocModels
+//
+///////////////////////////////////////////////////////////////////////////////
+
+TocModels::TocModels()
+ : bv_(0)
+{
+ names_ = new TocTypeModel(this);
+ names_sorted_ = new TocModelSortProxyModel(this);
+ names_sorted_->setSourceModel(names_);
+#if QT_VERSION >= 0x040300
+ names_sorted_->setSortLocaleAware(true);
+#endif
+ names_sorted_->sort(0);
+}
- maxdepth_ = max(maxdepth_, iter->depth());
- mindepth_ = min(mindepth_, iter->depth());
- current_row = rowCount(parent);
- insertRows(current_row, 1, parent);
- child_item = QStandardItemModel::index(current_row, 0, parent);
- //setData(child_item, toqstr(iter->str()));
- setData(child_item, toqstr(iter->str()), Qt::DisplayRole);
-
- // This looks like a gcc bug, in principle this should work:
- //toc_map_[child_item] = iter;
- // but it crashes with gcc-4.1 and 4.0.2
- toc_map_.insert( TocPair(child_item, iter) );
- model_map_[iter] = child_item;
- populate(iter, end, child_item);
+
+void TocModels::clear()
+{
+ names_->blockSignals(true);
+ names_->clear();
+ names_->blockSignals(false);
+ iterator end = models_.end();
+ for (iterator it = models_.begin(); it != end; ++it)
+ it.value()->clear();
+}
+
+
+int TocModels::depth(QString const & type)
+{
+ const_iterator it = models_.find(type);
+ if (!bv_ || it == models_.end())
+ return 0;
+ return it.value()->modelDepth();
+}
+
+
+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;
+}
+
+
+QAbstractItemModel * TocModels::nameModel()
+{
+ return names_sorted_;
+}
+
+
+QModelIndex TocModels::currentIndex(QString const & type) const
+{
+ const_iterator it = models_.find(type);
+ if (!bv_ || it == models_.end())
+ return QModelIndex();
+ return it.value()->modelIndex(bv_->cursor());
+}
+
+
+void TocModels::goTo(QString const & type, QModelIndex const & index) const
+{
+ const_iterator it = models_.find(type);
+ if (it == models_.end() || !index.isValid()) {
+ LYXERR(Debug::GUI, "TocModels::goTo(): QModelIndex is invalid!");
+ return;
}
+ LASSERT(index.model() == it.value()->model(), return);
+ TocItem const item = it.value()->tocItem(index);
+ LYXERR(Debug::GUI, "TocModels::goTo " << item.str());
+ dispatch(item.action());
}
-int TocModel::modelDepth() 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::updateBackend() const
+{
+ bv_->buffer().masterBuffer()->tocBackend().update();
+ bv_->buffer().structureChanged();
+}
+
+
+void TocModels::updateItem(QString const & type, DocIterator const & dit)
+{
+ models_[type]->updateItem(dit);
+}
+
+
+void TocModels::reset(BufferView const * bv)
+{
+ bv_ = bv;
+ clear();
+ 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 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::isSorted(QString const & type) const
+{
+ const_iterator it = models_.find(type);
+ if (it == models_.end()) {
+ LYXERR0("type not found: " << type);
+ return false;
+ }
+ return it.value()->isSorted();
+}
+
+
+void TocModels::sort(QString const & type, bool sort_it)
{
- return maxdepth_ - mindepth_;
+ 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"