workspace/plasma/shells/common/kcategorizeditemsviewmodels.cpp

   1 /*
   2  *   Copyright (C) 2007 Ivan Cukic <ivan.cukic+kde@gmail.com>
   3  *
   4  *   This program is free software; you can redistribute it and/or modify
   5  *   it under the terms of the GNU Library/Lesser General Public License
   6  *   version 2, or (at your option) any later version, as published by the
   7  *   Free Software Foundation
   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 Library/Lesser General Public
  15  *   License 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 "kcategorizeditemsviewmodels_p.h"
  21 #include <klocale.h>
  22
  23 #define COLUMN_COUNT 4
  24
  25 namespace KCategorizedItemsViewModels {
  26
  27 // AbstractItem
  28
  29 QString AbstractItem::name() const
  30 {
  31     return text();
  32 }
  33
  34 QString AbstractItem::description() const
  35 {
  36     return "";
  37 }
  38
  39 bool AbstractItem::isFavorite() const
  40 {
  41     return passesFiltering(Filter("favorite", true));
  42 }
  43
  44 int AbstractItem::running() const
  45 {
  46     return 0;
  47 }
  48
  49 bool AbstractItem::matches(const QString &pattern) const
  50 {
  51     return
  52         name().contains(pattern, Qt::CaseInsensitive) ||
  53         description().contains(pattern, Qt::CaseInsensitive);
  54 }
  55
  56 // DefaultFilterModel
  57
  58 DefaultFilterModel::DefaultFilterModel(QObject *parent) :
  59     QStandardItemModel(0, 1, parent)
  60 {
  61     setHeaderData(1, Qt::Horizontal, i18n("Filters"));
  62 }
  63
  64 void DefaultFilterModel::addFilter(const QString &caption, const Filter &filter, const KIcon &icon)
  65 {
  66     QList<QStandardItem *> newRow;
  67     QStandardItem *item = new QStandardItem(caption);
  68     item->setData(qVariantFromValue<Filter>(filter));
  69     if (!icon.isNull()) {
  70         item->setIcon(icon);
  71     }
  72
  73     newRow << item;
  74     appendRow(newRow);
  75 }
  76
  77 void DefaultFilterModel::addSeparator(const QString &caption)
  78 {
  79     QList<QStandardItem *> newRow;
  80     QStandardItem *item = new QStandardItem(caption);
  81     item->setEnabled(false);
  82
  83     newRow << item;
  84     appendRow(newRow);
  85 }
  86
  87 // DefaultItemFilterProxyModel
  88
  89 DefaultItemFilterProxyModel::DefaultItemFilterProxyModel(QObject *parent) :
  90     QSortFilterProxyModel(parent), m_innerModel(parent)
  91 {
  92 }
  93
  94 void DefaultItemFilterProxyModel::setSourceModel(QAbstractItemModel *sourceModel)
  95 {
  96     QStandardItemModel *model = qobject_cast<QStandardItemModel*>(sourceModel);
  97
  98     if (!model) {
  99         kWarning() << "Expecting a QStandardItemModel!";
 100         return;
 101     }
 102
 103     m_innerModel.setSourceModel(model);
 104     QSortFilterProxyModel::setSourceModel(&m_innerModel);
 105 }
 106
 107 QStandardItemModel *DefaultItemFilterProxyModel::sourceModel() const
 108 {
 109     return m_innerModel.sourceModel();
 110 }
 111
 112 int DefaultItemFilterProxyModel::columnCount(const QModelIndex &index) const
 113 {
 114     Q_UNUSED(index);
 115     return COLUMN_COUNT;
 116 }
 117
 118 QVariant DefaultItemFilterProxyModel::data(const QModelIndex &index, int role) const
 119 {
 120     return m_innerModel.data(index, (index.column() == 1), role);
 121 }
 122
 123 bool DefaultItemFilterProxyModel::filterAcceptsRow(int sourceRow,
 124         const QModelIndex &sourceParent) const
 125 {
 126     QStandardItemModel *model = (QStandardItemModel *) sourceModel();
 127
 128     QModelIndex index = sourceModel()->index(sourceRow, 0, sourceParent);
 129
 130     AbstractItem *item = (AbstractItem *) model->itemFromIndex(index);
 131     //kDebug() << "ITEM " << (item ? "IS NOT " : "IS") << " NULL\n";
 132
 133     return
 134         (m_filter.first.isEmpty() || item->passesFiltering(m_filter)) &&
 135         (m_searchPattern.isEmpty() || item->matches(m_searchPattern));
 136 }
 137
 138 bool DefaultItemFilterProxyModel::lessThan(const QModelIndex &left,
 139         const QModelIndex &right) const
 140 {
 141     return
 142         sourceModel()->data(left).toString().localeAwareCompare(
 143             sourceModel()->data(right).toString()) < 0;
 144 }
 145
 146 void DefaultItemFilterProxyModel::setSearch(const QString &pattern)
 147 {
 148     m_searchPattern = pattern;
 149     invalidateFilter();
 150     emit searchTermChanged(pattern);
 151 }
 152
 153 void DefaultItemFilterProxyModel::setFilter(const Filter &filter)
 154 {
 155     m_filter = filter;
 156     invalidateFilter();
 157 }
 158
 159 // DefaultItemFilterProxyModel::InnerProxyModel
 160
 161 DefaultItemFilterProxyModel::InnerProxyModel::InnerProxyModel(QObject *parent) :
 162     QAbstractItemModel(parent), m_sourceModel(NULL)
 163 {
 164 }
 165
 166 Qt::ItemFlags DefaultItemFilterProxyModel::InnerProxyModel::flags(const QModelIndex &index) const
 167 {
 168     if (!m_sourceModel) {
 169         return 0;
 170     }
 171     return m_sourceModel->flags(index);
 172 }
 173
 174 QVariant DefaultItemFilterProxyModel::InnerProxyModel::data(
 175     const QModelIndex &index, bool favoriteColumn, int role) const
 176 {
 177     Q_UNUSED(favoriteColumn);
 178     return data(index, role);
 179 }
 180
 181 QVariant DefaultItemFilterProxyModel::InnerProxyModel::data(
 182         const QModelIndex &index, int role) const
 183 {
 184     if (!m_sourceModel) {
 185         return QVariant();
 186     }
 187     return m_sourceModel->data(index, role);
 188 }
 189
 190 QVariant DefaultItemFilterProxyModel::InnerProxyModel::headerData(
 191     int section, Qt::Orientation orientation, int role) const
 192 {
 193     Q_UNUSED(orientation);
 194     Q_UNUSED(role);
 195     return QVariant(section);
 196 }
 197
 198 int DefaultItemFilterProxyModel::InnerProxyModel::rowCount(const QModelIndex &parent) const
 199 {
 200     if (!m_sourceModel) {
 201         return 0;
 202     }
 203     return m_sourceModel->rowCount(parent);
 204 }
 205
 206 bool DefaultItemFilterProxyModel::InnerProxyModel::setData(
 207     const QModelIndex &index, const QVariant &value, int role)
 208 {
 209     if (!m_sourceModel) {
 210         return false;
 211     }
 212     return m_sourceModel->setData(index, value, role);
 213 }
 214
 215 bool DefaultItemFilterProxyModel::InnerProxyModel::setHeaderData(
 216     int section, Qt::Orientation orientation, const QVariant &value, int role)
 217 {
 218     Q_UNUSED(section);
 219     Q_UNUSED(value);
 220     Q_UNUSED(orientation);
 221     Q_UNUSED(role);
 222     return false;
 223 }
 224
 225 QModelIndex DefaultItemFilterProxyModel::InnerProxyModel::index(
 226     int row, int column, const QModelIndex &parent) const
 227 {
 228     Q_UNUSED(column);
 229     if (!m_sourceModel) {
 230         return QModelIndex();
 231     }
 232     return m_sourceModel->index(row, 0, parent);
 233 }
 234
 235 QModelIndex DefaultItemFilterProxyModel::InnerProxyModel::parent(const QModelIndex &index) const
 236 {
 237     if (!m_sourceModel) {
 238         return QModelIndex();
 239     }
 240     return m_sourceModel->parent(index);
 241 }
 242
 243 QMimeData *DefaultItemFilterProxyModel::InnerProxyModel::mimeData(
 244     const QModelIndexList &indexes) const
 245 {
 246     if (!m_sourceModel) {
 247         return NULL;
 248     }
 249     return m_sourceModel->mimeData(indexes);
 250 }
 251
 252 int DefaultItemFilterProxyModel::InnerProxyModel::columnCount(const QModelIndex &index) const
 253 {
 254     Q_UNUSED(index);
 255     return COLUMN_COUNT;
 256 }
 257
 258 void DefaultItemFilterProxyModel::InnerProxyModel::setSourceModel(QStandardItemModel *sourceModel)
 259 {
 260     m_sourceModel = sourceModel;
 261 }
 262
 263 QStandardItemModel *DefaultItemFilterProxyModel::InnerProxyModel::sourceModel() const
 264 {
 265     return m_sourceModel;
 266 }
 267
 268 // DefaultItemModel
 269
 270 DefaultItemModel::DefaultItemModel(QObject *parent) :
 271     QStandardItemModel(parent) {}
 272 }