--- /dev/null
+/***************************************************************************
+ * Copyright (C) 2005-07 by The Quassel Team *
+ * devel@quassel-irc.org *
+ * *
+ * This program is free software; you can redistribute it and/or modify *
+ * it under the terms of the GNU General Public License as published by *
+ * the Free Software Foundation; either version 2 of the License, or *
+ * (at your option) any later version. *
+ * *
+ * This program is distributed in the hope that it will be useful, *
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of *
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
+ * GNU General Public License for more details. *
+ * *
+ * You should have received a copy of the GNU General Public License *
+ * along with this program; if not, write to the *
+ * Free Software Foundation, Inc., *
+ * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. *
+ ***************************************************************************/
+
+#include "mappedselectionmodel.h"
+
+#include <QItemSelectionModel>
+#include <QAbstractItemModel>
+#include <QAbstractProxyModel>
+#include <QDebug>
+
+MappedSelectionModel::MappedSelectionModel(QAbstractItemModel *model)
+ : QItemSelectionModel(model)
+{
+ _isProxyModel = (bool)proxyModel();
+ connect(this, SIGNAL(currentChanged(QModelIndex, QModelIndex)),
+ this, SLOT(_currentChanged(QModelIndex, QModelIndex)));
+ connect(this, SIGNAL(selectionChanged(QItemSelection, QItemSelection)),
+ this, SLOT(_selectionChanged(QItemSelection, QItemSelection)));
+}
+
+MappedSelectionModel::~MappedSelectionModel() {
+}
+
+const QAbstractItemModel *MappedSelectionModel::baseModel() const {
+ if(isProxyModel())
+ return proxyModel()->sourceModel();
+ else
+ return model();
+}
+
+const QAbstractProxyModel *MappedSelectionModel::proxyModel() const {
+ return qobject_cast<const QAbstractProxyModel *>(model());
+}
+
+QModelIndex MappedSelectionModel::mapFromSource(const QModelIndex &sourceIndex) {
+ if(isProxyModel())
+ return proxyModel()->mapFromSource(sourceIndex);
+ else
+ return sourceIndex;
+}
+
+QItemSelection MappedSelectionModel::mapFromSource(const QItemSelection &sourceSelection) {
+ if(isProxyModel()) {
+ QItemSelection mappedSelection;
+ foreach(QItemSelectionRange range, sourceSelection) {
+ QModelIndex topleft = mapFromSource(range.topLeft());
+ QModelIndex bottomright = mapFromSource(range.bottomRight());
+ if(topleft.isValid() && bottomright.isValid())
+ mappedSelection << QItemSelectionRange(topleft, bottomright);
+ else
+ Q_ASSERT(!topleft.isValid() && !bottomright.isValid());
+ }
+ return mappedSelection;
+ } else {
+ return sourceSelection;
+ }
+}
+
+QModelIndex MappedSelectionModel::mapToSource(const QModelIndex &proxyIndex) {
+ if(isProxyModel())
+ return proxyModel()->mapToSource(proxyIndex);
+ else
+ return proxyIndex;
+}
+
+QItemSelection MappedSelectionModel::mapToSource(const QItemSelection &proxySelection) {
+ if(isProxyModel()) {
+ QItemSelection mappedSelection;
+ foreach(QItemSelectionRange range, proxySelection) {
+ mappedSelection << QItemSelectionRange(mapToSource(range.topLeft()), mapToSource(range.bottomRight()));
+ }
+ return mappedSelection;
+ } else {
+ return proxySelection;
+ }
+}
+
+void MappedSelectionModel::mappedSelect(const QModelIndex &index, QItemSelectionModel::SelectionFlags command) {
+ QModelIndex mappedIndex = mapFromSource(index);
+ if(!isSelected(mappedIndex))
+ select(mappedIndex, command);
+}
+
+void MappedSelectionModel::mappedSelect(const QItemSelection &selection, QItemSelectionModel::SelectionFlags command) {
+ QItemSelection mappedSelection = mapFromSource(selection);
+ if(mappedSelection != QItemSelectionModel::selection())
+ select(mappedSelection, command);
+}
+
+void MappedSelectionModel::mappedSetCurrentIndex(const QModelIndex &index, QItemSelectionModel::SelectionFlags command) {
+ QModelIndex mappedIndex = mapFromSource(index);
+ if(mappedIndex == currentIndex())
+ return;
+
+ if(mappedIndex.isValid())
+ setCurrentIndex(mappedIndex, command);
+ else if(hasSelection())
+ setCurrentIndex(currentIndex(), QItemSelectionModel::Clear);
+}
+
+
+void MappedSelectionModel::_currentChanged(const QModelIndex ¤t, const QModelIndex &previous) {
+ Q_UNUSED(previous)
+ emit mappedCurrentChanged(mapToSource(current));
+}
+
+void MappedSelectionModel::_selectionChanged(const QItemSelection &selected, const QItemSelection &deselected) {
+ Q_UNUSED(selected)
+ Q_UNUSED(deselected)
+ emit mappedSelectionChanged(mapToSource(QItemSelectionModel::selection()));
+}
+