objecttypefilterproxymodel.h

Go to the documentation of this file.

/*
  objecttypefilterproxymodel.h

  This file is part of GammaRay, the Qt application inspection and
  manipulation tool.

  Copyright (C) 2010-2015 Klarälvdalens Datakonsult AB, a KDAB Group company, info@kdab.com
  Author: Volker Krause <volker.krause@kdab.com>

  Licensees holding valid commercial KDAB GammaRay licenses may use this file in
  accordance with GammaRay Commercial License Agreement provided with the Software.

  Contact info@kdab.com if any conditions of this licensing are not clear to you.

  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, see <http://www.gnu.org/licenses/>.
*/
#ifndef GAMMARAY_OBJECTTYPEFILTERPROXYMODEL_H
#define GAMMARAY_OBJECTTYPEFILTERPROXYMODEL_H

#include "objectmodelbase.h"

#include <QSortFilterProxyModel>

namespace GammaRay {

class ObjectFilterProxyModelBase : public QSortFilterProxyModel
{
  public:
    explicit ObjectFilterProxyModelBase(QObject *parent = 0) : QSortFilterProxyModel(parent)
    {
      setDynamicSortFilter(true);
    }

  protected:
    bool filterAcceptsRow(int source_row, const QModelIndex &source_parent) const
    {
      const QModelIndex source_index = sourceModel()->index(source_row, 0, source_parent);
      if (!source_index.isValid()) {
        return false;
      }

      QObject *obj = source_index.data(ObjectModel::ObjectRole).value<QObject*>();
      if (!obj || !filterAcceptsObject(obj)) {
        return false;
      }

      return QSortFilterProxyModel::filterAcceptsRow(source_row, source_parent);
    }

    virtual bool filterAcceptsObject(QObject *object) const = 0;
};

template <typename T>
class ObjectTypeFilterProxyModel : public ObjectFilterProxyModelBase
{
  public:
    explicit ObjectTypeFilterProxyModel(QObject *parent = 0)
      : ObjectFilterProxyModelBase(parent)
    {
    }

  protected:
    virtual bool filterAcceptsObject(QObject *object) const
    {
      return qobject_cast<T*>(object);
    }
};

}

#endif // GAMMARAY_OBJECTTYPEFILTERPROXYMODEL_H