104 lines
2.7 KiB
C++
104 lines
2.7 KiB
C++
|
#include "ResourceProxyModel.h"
|
||
|
|
|
||
|
|
#include <QItemSelectionRange>
|
||
|
|
|
||
|
|
#include "Resource.h"
|
||
|
|
#include "ResourceObserver.h"
|
||
|
|
|
||
|
|
//Q_DECLARE_METATYPE(QVector<int>)
|
||
|
|
|
||
|
|
class ModelResourceObserver : public ResourceObserver
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
explicit ModelResourceObserver(const QModelIndex &index, const int role)
|
||
|
|
: m_index(index), m_role(role)
|
||
|
|
{
|
||
|
|
qRegisterMetaType<QVector<int>>("QVector<int>");
|
||
|
|
}
|
||
|
|
|
||
|
|
void resourceUpdated() override
|
||
|
|
{
|
||
|
|
if (m_index.isValid())
|
||
|
|
{
|
||
|
|
QMetaObject::invokeMethod(const_cast<QAbstractItemModel *>(m_index.model()),
|
||
|
|
"dataChanged", Qt::QueuedConnection,
|
||
|
|
Q_ARG(QModelIndex, m_index), Q_ARG(QModelIndex, m_index), Q_ARG(QVector<int>, QVector<int>() << m_role));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private:
|
||
|
|
QPersistentModelIndex m_index;
|
||
|
|
int m_role;
|
||
|
|
};
|
||
|
|
|
||
|
|
ResourceProxyModel::ResourceProxyModel(const int resultTypeId, QObject *parent)
|
||
|
|
: QIdentityProxyModel(parent), m_resultTypeId(resultTypeId)
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
QVariant ResourceProxyModel::data(const QModelIndex &proxyIndex, int role) const
|
||
|
|
{
|
||
|
|
const QModelIndex mapped = mapToSource(proxyIndex);
|
||
|
|
if (mapped.isValid() && role == Qt::DecorationRole && !mapToSource(proxyIndex).data(role).toString().isEmpty())
|
||
|
|
{
|
||
|
|
if (!m_resources.contains(mapped))
|
||
|
|
{
|
||
|
|
Resource::Ptr res = Resource::create(mapToSource(proxyIndex).data(role).toString())
|
||
|
|
->applyTo(new ModelResourceObserver(proxyIndex, role));
|
||
|
|
|
||
|
|
const QVariant placeholder = mapped.data(PlaceholderRole);
|
||
|
|
if (!placeholder.isNull() && placeholder.type() == QVariant::String)
|
||
|
|
{
|
||
|
|
res->placeholder(Resource::create(placeholder.toString()));
|
||
|
|
}
|
||
|
|
|
||
|
|
m_resources.insert(mapped, res);
|
||
|
|
}
|
||
|
|
|
||
|
|
return m_resources.value(mapped)->getResourceInternal(m_resultTypeId);
|
||
|
|
}
|
||
|
|
return mapped.data(role);
|
||
|
|
}
|
||
|
|
|
||
|
|
void ResourceProxyModel::setSourceModel(QAbstractItemModel *model)
|
||
|
|
{
|
||
|
|
if (sourceModel())
|
||
|
|
{
|
||
|
|
disconnect(sourceModel(), 0, this, 0);
|
||
|
|
}
|
||
|
|
if (model)
|
||
|
|
{
|
||
|
|
connect(model, &QAbstractItemModel::dataChanged, this, [this](const QModelIndex &tl, const QModelIndex &br, const QVector<int> &roles)
|
||
|
|
{
|
||
|
|
if (roles.contains(Qt::DecorationRole) || roles.isEmpty())
|
||
|
|
{
|
||
|
|
const QItemSelectionRange range(tl, br);
|
||
|
|
for (const QModelIndex &index : range.indexes())
|
||
|
|
{
|
||
|
|
m_resources.remove(index);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else if (roles.contains(PlaceholderRole))
|
||
|
|
{
|
||
|
|
const QItemSelectionRange range(tl, br);
|
||
|
|
for (const QModelIndex &index : range.indexes())
|
||
|
|
{
|
||
|
|
if (m_resources.contains(index))
|
||
|
|
{
|
||
|
|
const QVariant placeholder = index.data(PlaceholderRole);
|
||
|
|
if (!placeholder.isNull() && placeholder.type() == QVariant::String)
|
||
|
|
{
|
||
|
|
m_resources.value(index)->placeholder(Resource::create(placeholder.toString()));
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
m_resources.value(index)->placeholder(nullptr);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}
|
||
|
|
QIdentityProxyModel::setSourceModel(model);
|
||
|
|
}
|