#include "TocModel.h"
+#include "Buffer.h"
+#include "BufferView.h"
+#include "Cursor.h"
+#include "DocIterator.h"
+#include "FuncRequest.h"
+#include "LyX.h"
+#include "qt_helpers.h"
+#include "TocBackend.h"
+
#include "support/debug.h"
+#include "support/lassert.h"
+
+#include <QSortFilterProxyModel>
+#include <QStandardItemModel>
+
-#include <boost/assert.hpp>
#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_;
+}
+
-TocModel::TocModel(Toc const & toc)
+QAbstractItemModel const * TocModel::model() const
{
- populate(toc);
+ if (is_sorted_)
+ return sorted_model_;
+ return model_;
}
-TocModel const & TocModel::operator=(Toc const & toc)
+void TocModel::clear()
{
- populate(toc);
- return *this;
+ model_->blockSignals(true);
+ model_->clear();
+ toc_ = make_shared<Toc>();
+ model_->blockSignals(false);
}
-TocIterator const TocModel::tocIterator(QModelIndex const & index) const
+void TocModel::sort(bool sort_it)
{
- TocMap::const_iterator map_it = toc_map_.find(index);
- BOOST_ASSERT(map_it != toc_map_.end());
- return map_it->second;
+ is_sorted_ = sort_it;
+ if (is_sorted_)
+ sorted_model_->sort(0);
}
-QModelIndex const TocModel::modelIndex(TocIterator const & it) const
+TocItem const & TocModel::tocItem(QModelIndex const & index) const
{
- ModelMap::const_iterator map_it = model_map_.find(it);
- //BOOST_ASSERT(it != model_map_.end());
+ return (*toc_)[model()->data(index, Qt::UserRole).toUInt()];
+}
+
- if (map_it == model_map_.end())
+QModelIndex TocModel::modelIndex(DocIterator const & dit) const
+{
+ if (toc_->empty())
return QModelIndex();
- return map_it->second;
+ unsigned int const toc_index = TocBackend::findItem(*toc_, dit) -
+ toc_->begin();
+
+ QModelIndexList list = model()->match(model()->index(0, 0), Qt::UserRole,
+ QVariant(toc_index), 1,
+ Qt::MatchFlags(Qt::MatchExactly | Qt::MatchRecursive));
+
+ LASSERT(!list.isEmpty(), return QModelIndex());
+ return list[0];
}
-void TocModel::clear()
+void TocModel::reset()
{
- QStandardItemModel::clear();
- toc_map_.clear();
- model_map_.clear();
- removeRows(0, rowCount());
- removeColumns(0, columnCount());
+ model_->reset();
}
-void TocModel::populate(Toc const & toc)
+void TocModel::setString(TocItem const & item, QModelIndex index)
{
- clear();
+ // Use implicit sharing of QStrings
+ QString str = toqstr(item.asString());
+ model_->setData(index, str, Qt::DisplayRole);
+ model_->setData(index, str, Qt::ToolTipRole);
+}
- if (toc.empty())
- return;
- int current_row;
- QModelIndex top_level_item;
- TocIterator iter = toc.begin();
- TocIterator end = toc.end();
+void TocModel::updateItem(DocIterator const & dit)
+{
+ QModelIndex const index = modelIndex(dit);
+ setString(tocItem(index), index);
+}
+
- insertColumns(0, 1);
+void TocModel::reset(shared_ptr<Toc const> toc)
+{
+ toc_ = toc;
+ if (toc_->empty()) {
+ maxdepth_ = 0;
+ mindepth_ = 0;
+ reset();
+ return;
+ }
+
+ model_->blockSignals(true);
+ model_->beginResetModel();
+ model_->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);
+ 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);
+ setString(item, top_level_item);
+ model_->setData(top_level_item, index, Qt::UserRole);
+
+ LYXERR(Debug::GUI, "Toc: at depth " << item.depth()
+ << ", added item " << item.asString());
+
+ populate(index, top_level_item);
+ if (index >= end)
+ break;
+ }
- // 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;
+ 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();
+}
- LYXERR(Debug::GUI, "Toc: at depth " << iter->depth()
- << ", added item " << to_utf8(iter->str()));
- populate(iter, end, top_level_item);
+void TocModel::populate(unsigned int & index, QModelIndex const & parent)
+{
+ int curdepth = (*toc_)[index].depth() + 1;
- if (iter == end)
+ 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());
+ 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;
-
- ++iter;
}
+}
+
- setHeaderData(0, Qt::Horizontal, QVariant("title"), Qt::DisplayRole);
-// emit headerDataChanged();
+int TocModel::modelDepth() const
+{
+ int const d = maxdepth_ - mindepth_;
+ LASSERT(d >= 0 && d <= 100, return 0);
+ return d;
}
-void TocModel::populate(TocIterator & iter, TocIterator const & end,
- QModelIndex const & parent)
+///////////////////////////////////////////////////////////////////////////////
+//
+// TocModels
+//
+///////////////////////////////////////////////////////////////////////////////
+
+TocModels::TocModels()
{
- int curdepth = iter->depth() + 1;
+ names_ = new TocTypeModel(this);
+ names_sorted_ = new TocModelSortProxyModel(this);
+ names_sorted_->setSourceModel(names_);
+ names_sorted_->setSortLocaleAware(true);
+ names_sorted_->sort(0);
+}
- int current_row;
- QModelIndex child_item;
- insertColumns(0, 1, parent);
- while (iter != end) {
+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();
+}
- ++iter;
- if (iter == end)
- break;
+int TocModels::depth(QString const & type)
+{
+ const_iterator it = models_.find(type);
+ if (it == models_.end())
+ return 0;
+ return it.value()->modelDepth();
+}
+
+
+QAbstractItemModel * TocModels::model(QString const & type)
+{
+ iterator it = models_.find(type);
+ if (it != models_.end())
+ return it.value()->model();
+ LYXERR0("type not found: " << type);
+ return 0;
+}
- if (iter->depth() < curdepth) {
- --iter;
- return;
- }
- 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);
+QAbstractItemModel * TocModels::nameModel()
+{
+ return names_sorted_;
+}
+
+
+QModelIndex TocModels::currentIndex(QString const & type,
+ DocIterator const & dit) const
+{
+ const_iterator it = models_.find(type);
+ if (it == models_.end())
+ return QModelIndex();
+ return it.value()->modelIndex(dit);
+}
+
+
+FuncRequest 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 FuncRequest(LFUN_NOACTION);
}
+ LASSERT(index.model() == it.value()->model(), return FuncRequest(LFUN_NOACTION));
+ TocItem const item = it.value()->tocItem(index);
+ LYXERR(Debug::GUI, "TocModels::goTo " << item.asString());
+ return 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::updateItem(QString const & type, DocIterator const & dit)
+{
+ models_[type]->updateItem(dit);
+}
+
+
+void TocModels::reset(BufferView const * 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);
+ // In the outliner, add Tocs from the master document
+ TocBackend const & backend = bv->buffer().masterBuffer()->tocBackend();
+ for (pair<string, shared_ptr<Toc>> const & toc : backend.tocs()) {
+ QString const type = toqstr(toc.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(toc.second);
+
+ // Fill in the names_ model.
+ QString const gui_name = toqstr(backend.outlinerName(toc.first));
+ 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"
projects / lyx.git / blobdiff