src/selectionmanager.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2008 by Peter Penz <peter.penz@gmx.at>                  *
   3  *                                                                         *
   4  *   This program is free software; you can redistribute it and/or modify  *
   5  *   it under the terms of the GNU General Public License as published by  *
   6  *   the Free Software Foundation; either version 2 of the License, or     *
   7  *   (at your option) any later version.                                   *
   8  *                                                                         *
   9  *   This program is distributed in the hope that it will be useful,       *
  10  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  11  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  12  *   GNU General Public License for more details.                          *
  13  *                                                                         *
  14  *   You should have received a copy of the GNU General Public License     *
  15  *   along with this program; if not, write to the                         *
  16  *   Free Software Foundation, Inc.,                                       *
  17  *   51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA            *
  18  ***************************************************************************/
  19
  20 #include "selectionmanager.h"
  21
  22 #include "dolphinmodel.h"
  23 #include "selectiontoggle.h"
  24 #include <kdirmodel.h>
  25 #include <kiconeffect.h>
  26
  27 #include <QAbstractButton>
  28 #include <QAbstractItemView>
  29 #include <QAbstractProxyModel>
  30 #include <QApplication>
  31 #include <QModelIndex>
  32 #include <QPainter>
  33 #include <QPaintEvent>
  34 #include <QRect>
  35 #include <QTimeLine>
  36
  37 SelectionManager::SelectionManager(QAbstractItemView* parent) :
  38     QObject(parent),
  39     m_view(parent),
  40     m_toggle(0),
  41     m_connected(false)
  42 {
  43     connect(parent, SIGNAL(entered(const QModelIndex&)),
  44             this, SLOT(slotEntered(const QModelIndex&)));
  45     connect(parent, SIGNAL(viewportEntered()),
  46             this, SLOT(slotViewportEntered()));
  47     m_toggle = new SelectionToggle(m_view->viewport());
  48     m_toggle->setCheckable(true);
  49     m_toggle->hide();
  50     connect(m_toggle, SIGNAL(clicked(bool)),
  51             this, SLOT(setItemSelected(bool)));
  52 }
  53
  54 SelectionManager::~SelectionManager()
  55 {
  56 }
  57
  58 void SelectionManager::reset()
  59 {
  60     m_toggle->reset();
  61 }
  62
  63 void SelectionManager::slotEntered(const QModelIndex& index)
  64 {
  65     m_toggle->hide();
  66     const bool showToggle = index.isValid() &&
  67                             (index.column() == DolphinModel::Name) &&
  68                             (QApplication::mouseButtons() == Qt::NoButton);
  69     if (showToggle) {
  70         m_toggle->setUrl(urlForIndex(index));
  71
  72         if (!m_connected) {
  73             connect(m_view->model(), SIGNAL(rowsRemoved(const QModelIndex&, int, int)),
  74                     this, SLOT(slotRowsRemoved(const QModelIndex&, int, int)));
  75             connect(m_view->selectionModel(),
  76                     SIGNAL(selectionChanged(const QItemSelection&, const QItemSelection&)),
  77                     this,
  78                     SLOT(slotSelectionChanged(const QItemSelection&, const QItemSelection&)));
  79             m_connected = true;
  80         }
  81
  82         const QRect rect = m_view->visualRect(index);
  83
  84         // align the toggle on the bottom left of the item
  85         const int gap = 2;
  86         const int x = rect.left() + gap;
  87         const int y = rect.top() + gap;
  88         m_toggle->move(QPoint(x, y));
  89
  90         // increase the size of the toggle for large items
  91         if (rect.height() >= KIconLoader::SizeEnormous) {
  92             m_toggle->resize(KIconLoader::SizeMedium, KIconLoader::SizeMedium);
  93         } else if (rect.height() >= KIconLoader::SizeHuge) {
  94             m_toggle->resize(KIconLoader::SizeSmallMedium, KIconLoader::SizeSmallMedium);
  95         } else {
  96             m_toggle->resize(KIconLoader::SizeSmall, KIconLoader::SizeSmall);
  97         }
  98
  99         QItemSelectionModel* selModel = m_view->selectionModel();
 100         m_toggle->setChecked(selModel->isSelected(index));
 101         m_toggle->show();
 102     } else {
 103         m_toggle->setUrl(KUrl());
 104         disconnect(m_view->model(), SIGNAL(rowsRemoved(const QModelIndex&, int, int)),
 105                    this, SLOT(slotRowsRemoved(const QModelIndex&, int, int)));
 106         disconnect(m_view->selectionModel(),
 107                    SIGNAL(selectionChanged(const QItemSelection&, const QItemSelection&)),
 108                    this,
 109                    SLOT(slotSelectionChanged(const QItemSelection&, const QItemSelection&)));
 110         m_connected = false;
 111     }
 112 }
 113
 114 void SelectionManager::slotViewportEntered()
 115 {
 116     m_toggle->hide();
 117 }
 118
 119 void SelectionManager::setItemSelected(bool selected)
 120 {
 121     emit selectionChanged();
 122
 123     if (!m_toggle->url().isEmpty()) {
 124         const QModelIndex index = indexForUrl(m_toggle->url());
 125         if (index.isValid()) {
 126             QItemSelectionModel* selModel = m_view->selectionModel();
 127             if (selected) {
 128                 selModel->select(index, QItemSelectionModel::Select);
 129             } else {
 130                 selModel->select(index, QItemSelectionModel::Deselect);
 131             }
 132             selModel->setCurrentIndex(index, QItemSelectionModel::Current);
 133         }
 134     }
 135 }
 136
 137 void SelectionManager::slotRowsRemoved(const QModelIndex& parent, int start, int end)
 138 {
 139     Q_UNUSED(parent);
 140     Q_UNUSED(start);
 141     Q_UNUSED(end);
 142     m_toggle->hide();
 143 }
 144
 145 void SelectionManager::slotSelectionChanged(const QItemSelection& selected,
 146                                             const QItemSelection& deselected)
 147 {
 148     // The selection has been changed outside the scope of the selection manager
 149     // (e. g. by the rubberband or the "Select All" action). Take care updating
 150     // the state of the toggle button.
 151     if (!m_toggle->url().isEmpty()) {
 152         const QModelIndex index = indexForUrl(m_toggle->url());
 153         if (index.isValid()) {
 154             if (selected.contains(index)) {
 155                 m_toggle->setChecked(true);
 156             }
 157
 158             if (deselected.contains(index)) {
 159                 m_toggle->setChecked(false);
 160             }
 161         }
 162     }
 163 }
 164
 165 KUrl SelectionManager::urlForIndex(const QModelIndex& index) const
 166 {
 167     QAbstractProxyModel* proxyModel = static_cast<QAbstractProxyModel*>(m_view->model());
 168     KDirModel* dirModel = static_cast<KDirModel*>(proxyModel->sourceModel());
 169     const QModelIndex dirIndex = proxyModel->mapToSource(index);
 170     return dirModel->itemForIndex(dirIndex).url();
 171 }
 172
 173 const QModelIndex SelectionManager::indexForUrl(const KUrl& url) const
 174 {
 175     QAbstractProxyModel* proxyModel = static_cast<QAbstractProxyModel*>(m_view->model());
 176     KDirModel* dirModel = static_cast<KDirModel*>(proxyModel->sourceModel());
 177     const QModelIndex dirIndex = dirModel->indexForUrl(url);
 178     return proxyModel->mapFromSource(dirIndex);
 179 }
 180
 181 #include "selectionmanager.moc"