kdelibs/kdeui/tests/proxymodeltestsuite/indexfinder.h
2014-11-13 01:04:59 +02:00

84 lines
2.3 KiB
C++

/*
Copyright (c) 2009 Stephen Kelly <steveire@gmail.com>
This library is free software; you can redistribute it and/or modify it
under the terms of the GNU Library General Public License as published by
the Free Software Foundation; either version 2 of the License, or (at your
option) any later version.
This library 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 Library General Public
License for more details.
You should have received a copy of the GNU Library General Public License
along with this library; see the file COPYING.LIB. If not, write to the
Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
02110-1301, USA.
*/
#ifndef INDEXFINDER_H
#define INDEXFINDER_H
#include <QModelIndex>
class IndexFinder
{
public:
IndexFinder(QList<int> rows = QList<int>()) : m_rows(rows), m_model(0) {}
IndexFinder(const QAbstractItemModel *model, QList<int> rows = QList<int>() )
: m_rows(rows), m_model(model)
{
Q_ASSERT(model);
}
QModelIndex getIndex() const
{
if(!m_model)
return QModelIndex();
static const int col = 0;
QModelIndex parent = QModelIndex();
QListIterator<int> i(m_rows);
while (i.hasNext())
{
parent = m_model->index(i.next(), col, parent);
Q_ASSERT(parent.isValid());
}
return parent;
}
static IndexFinder indexToIndexFinder(const QModelIndex &_idx)
{
if (!_idx.isValid())
return IndexFinder();
QList<int> list;
QModelIndex idx = _idx;
while (idx.isValid())
{
list.prepend(idx.row());
idx = idx.parent();
}
return IndexFinder(_idx.model(), list);
}
bool operator==( const IndexFinder &other ) const
{
return (m_rows == other.m_rows && m_model == other.m_model );
}
QList<int> rows() const { return m_rows; }
void appendRow(int row) { m_rows.append(row); }
void setRows( const QList<int> &rows ) { m_rows = rows; }
void setModel(QAbstractItemModel *model) { m_model = model; }
private:
QList<int> m_rows;
const QAbstractItemModel * m_model;
};
Q_DECLARE_METATYPE( IndexFinder )
#endif