objectlistmodel.cpp

   1 /*
   2   objectlistmodel.cpp
   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 #include "objectlistmodel.h"
  25
  26 #include "readorwritelocker.h"
  27
  28 #include <QThread>
  29 #include "probe.h"
  30
  31 using namespace GammaRay;
  32
  33 ObjectListModel::ObjectListModel(QObject *parent)
  34   : ObjectModelBase< QAbstractTableModel >(parent),
  35     m_lock(QReadWriteLock::Recursive)
  36 {
  37 }
  38
  39 QVariant ObjectListModel::data(const QModelIndex &index, int role) const
  40 {
  41   ReadOrWriteLocker lock(&m_lock);
  42   if (index.row() >= 0 && index.row() < m_objects.size()) {
  43     QObject *obj = m_objects.at(index.row());
  44     if (obj) {
  45       return dataForObject(obj, index, role);
  46     }
  47   }
  48   return QVariant();
  49 }
  50
  51 int ObjectListModel::columnCount(const QModelIndex &parent) const
  52 {
  53   if (parent.isValid()) {
  54     return 0;
  55   }
  56   return ObjectModelBase<QAbstractTableModel>::columnCount(parent);
  57 }
  58
  59 int ObjectListModel::rowCount(const QModelIndex &parent) const
  60 {
  61   if (parent.isValid()) {
  62     return 0;
  63   }
  64
  65   ReadOrWriteLocker lock(&m_lock);
  66   return m_objects.size();
  67 }
  68
  69 void ObjectListModel::objectAdded(QObject *obj)
  70 {
  71   // when called from background, delay into foreground, otherwise call directly
  72   QMetaObject::invokeMethod(this, "objectAddedMainThread", Qt::AutoConnection,
  73                             Q_ARG(QObject *, obj));
  74 }
  75
  76 void ObjectListModel::objectAddedMainThread(QObject *obj)
  77 {
  78   ReadOrWriteLocker objectLock(Probe::instance()->objectLock());
  79   if (!Probe::instance()->isValidObject(obj)) {
  80     return;
  81   }
  82
  83   QWriteLocker lock(&m_lock);
  84   if (m_objects.contains(obj)) {
  85     return;
  86   }
  87
  88   beginInsertRows(QModelIndex(), m_objects.size(), m_objects.size());
  89   m_objects << obj;
  90   endInsertRows();
  91 }
  92
  93 void ObjectListModel::objectRemoved(QObject *obj)
  94 {
  95   if (thread() != QThread::currentThread()) {
  96     // invalidate data
  97     QWriteLocker lock(&m_lock);
  98     const int index = m_objects.indexOf(obj);
  99     if (index != -1) {
 100       m_objects[index] = 0;
 101     }
 102   }
 103
 104   // when called from background, delay into foreground, otherwise call directly
 105   QMetaObject::invokeMethod(this, "objectRemovedMainThread", Qt::AutoConnection,
 106                             Q_ARG(QObject *, obj));
 107 }
 108
 109 void ObjectListModel::objectRemovedMainThread(QObject *obj)
 110 {
 111   QWriteLocker lock(&m_lock);
 112
 113   for (int i = 0; i < m_objects.size(); ++i) {
 114     if (!m_objects.at(i) || m_objects.at(i) == obj) {
 115       beginRemoveRows(QModelIndex(), i, i);
 116       m_objects.remove(i);
 117       endRemoveRows();
 118     }
 119   }
 120 }
 121
 122 #include "objectlistmodel.moc"