objecttypefilterproxymodel.h

   1 /*
   2   objecttypefilterproxymodel.h
   3
   4   This file is part of GammaRay, the Qt application inspection and
   5   manipulation tool.
   6
   7   Copyright (C) 2010-2011 Klarälvdalens Datakonsult AB, a KDAB Group company, info@kdab.com
   8   Author: Volker Krause <volker.krause@kdab.com>
   9
  10   This program is free software; you can redistribute it and/or modify
  11   it under the terms of the GNU General Public License as published by
  12   the Free Software Foundation, either version 2 of the License, or
  13   (at your option) any later version.
  14
  15   This program is distributed in the hope that it will be useful,
  16   but WITHOUT ANY WARRANTY; without even the implied warranty of
  17   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18   GNU General Public License for more details.
  19
  20   You should have received a copy of the GNU General Public License
  21   along with this program.  If not, see <http://www.gnu.org/licenses/>.
  22 */
  23
  24 #ifndef GAMMARAY_OBJECTTYPEFILTERPROXYMODEL_H
  25 #define GAMMARAY_OBJECTTYPEFILTERPROXYMODEL_H
  26
  27 #include <QSortFilterProxyModel>
  28
  29 #include "objectmodelbase.h"
  30
  31 namespace GammaRay {
  32
  33 class ObjectFilterProxyModelBase : public QSortFilterProxyModel
  34 {
  35   public:
  36     explicit ObjectFilterProxyModelBase(QObject *parent = 0) : QSortFilterProxyModel(parent)
  37     {
  38       setDynamicSortFilter(true);
  39     }
  40
  41   protected:
  42     bool filterAcceptsRow(int source_row, const QModelIndex &source_parent) const
  43     {
  44       const QModelIndex source_index = sourceModel()->index(source_row, 0, source_parent);
  45       if (!source_index.isValid()) {
  46         return false;
  47       }
  48
  49       QObject *obj = source_index.data(ObjectModel::ObjectRole).value<QObject*>();
  50       Q_ASSERT(obj);
  51       if (!filterAcceptsObject(obj)) {
  52         return false;
  53       }
  54
  55       return QSortFilterProxyModel::filterAcceptsRow(source_row, source_parent);
  56     }
  57
  58     virtual bool filterAcceptsObject(QObject *object) const = 0;
  59 };
  60
  61 template <typename T>
  62 class ObjectTypeFilterProxyModel : public ObjectFilterProxyModelBase
  63 {
  64   public:
  65     explicit ObjectTypeFilterProxyModel(QObject *parent = 0)
  66       : ObjectFilterProxyModelBase(parent) {}
  67
  68   protected:
  69     virtual bool filterAcceptsObject(QObject *object) const {
  70       return qobject_cast<T*>(object);
  71     }
  72 };
  73
  74 }
  75
  76 #endif // GAMMARAY_OBJECTTYPEFILTERPROXYMODEL_H