Revision control
Copy as Markdown
// Copyright (C) 2023 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only
// Qt-Security score:significant reason:default
#include "qscatter3dseries_p.h"
#include "scatteritemmodelhandler_p.h"
#include <QtGui/qquaternion.h>
#include <qtgraphs_tracepoints_p.h>
QT_BEGIN_NAMESPACE
Q_TRACE_PREFIX(qtgraphs,
"QT_BEGIN_NAMESPACE" \
"class ScatterItemModelHandler;" \
"QT_END_NAMESPACE"
)
Q_TRACE_POINT(qtgraphs, QGraphs3DScatterItemModelHandlerResolveModel_entry);
Q_TRACE_POINT(qtgraphs, QGraphs3DScatterItemModelHandlerResolveModel_exit);
Q_TRACE_POINT(qtgraphs, QGraphs3DScatterItemModelHandlerHandleDataChanged_entry);
Q_TRACE_POINT(qtgraphs, QGraphs3DScatterItemModelHandlerHandleDataChanged_exit);
ScatterItemModelHandler::ScatterItemModelHandler(QItemModelScatterDataProxy *proxy, QObject *parent)
: AbstractItemModelHandler(parent)
, m_proxy(proxy)
, m_proxyArray(0)
, m_xPosRole(noRoleIndex)
, m_yPosRole(noRoleIndex)
, m_zPosRole(noRoleIndex)
, m_rotationRole(noRoleIndex)
, m_scaleRole(noRoleIndex)
, m_haveXPosPattern(false)
, m_haveYPosPattern(false)
, m_haveZPosPattern(false)
, m_haveRotationPattern(false)
, m_haveScalePattern(false)
{}
ScatterItemModelHandler::~ScatterItemModelHandler() {}
void ScatterItemModelHandler::handleDataChanged(const QModelIndex &topLeft,
const QModelIndex &bottomRight,
const QList<int> &roles)
{
// Do nothing if full reset already pending
if (!m_fullReset) {
Q_TRACE_SCOPE(QGraphs3DScatterItemModelHandlerHandleDataChanged);
if (m_itemModel->columnCount() > 1) {
// If the data model is multi-column, do full asynchronous reset to
// simplify things
AbstractItemModelHandler::handleDataChanged(topLeft, bottomRight, roles);
} else {
int start = qMin(topLeft.row(), bottomRight.row());
int end = qMax(topLeft.row(), bottomRight.row());
QScatterDataArray array(end - start + 1);
int count = 0;
for (int i = start; i <= end; i++)
modelPosToScatterItem(i, 0, array[count++]);
m_proxy->setItems(start, array);
}
}
}
void ScatterItemModelHandler::handleRowsInserted(const QModelIndex &parent, int start, int end)
{
// Do nothing if full reset already pending
if (!m_fullReset) {
if (!m_proxy->itemCount() || m_itemModel->columnCount() > 1) {
// If inserting into an empty array, do full asynchronous reset to avoid
// multiple separate inserts when initializing the model. If the data
// model is multi-column, do full asynchronous reset to simplify things
AbstractItemModelHandler::handleRowsInserted(parent, start, end);
} else {
QScatterDataArray array(end - start + 1);
int count = 0;
for (int i = start; i <= end; i++)
modelPosToScatterItem(i, 0, array[count++]);
m_proxy->insertItems(start, array);
}
}
}
void ScatterItemModelHandler::handleRowsRemoved(const QModelIndex &parent, int start, int end)
{
Q_UNUSED(parent);
// Do nothing if full reset already pending
if (!m_fullReset) {
if (m_itemModel->columnCount() > 1) {
// If the data model is multi-column, do full asynchronous reset to
// simplify things
AbstractItemModelHandler::handleRowsRemoved(parent, start, end);
} else {
m_proxy->removeItems(start, end - start + 1);
}
}
}
static inline QQuaternion toQuaternion(const QVariant &variant)
{
if (variant.canConvert<QQuaternion>()) {
return variant.value<QQuaternion>();
} else if (variant.canConvert<QString>()) {
QString s = variant.toString();
if (!s.isEmpty()) {
bool angleAndAxis = false;
if (s.startsWith(QLatin1Char('@'))) {
angleAndAxis = true;
s = s.mid(1);
}
if (s.count(QLatin1Char(',')) == 3) {
qsizetype index = s.indexOf(QLatin1Char(','));
qsizetype index2 = s.indexOf(QLatin1Char(','), index + 1);
qsizetype index3 = s.indexOf(QLatin1Char(','), index2 + 1);
bool sGood, xGood, yGood, zGood;
float sCoord = s.left(index).toFloat(&sGood);
float xCoord = s.mid(index + 1, index2 - index - 1).toFloat(&xGood);
float yCoord = s.mid(index2 + 1, index3 - index2 - 1).toFloat(&yGood);
float zCoord = s.mid(index3 + 1).toFloat(&zGood);
if (sGood && xGood && yGood && zGood) {
if (angleAndAxis)
return QQuaternion::fromAxisAndAngle(xCoord, yCoord, zCoord, sCoord);
else
return QQuaternion(sCoord, xCoord, yCoord, zCoord);
}
}
}
}
return QQuaternion();
}
static inline QVector3D toVector3D(const QVariant &variant)
{
if (variant.canConvert<QVector3D>()) {
return variant.value<QVector3D>();
} else if (variant.canConvert<QString>()) {
QString s = variant.toString();
if (!s.isEmpty()) {
if (s.count(QLatin1Char(',')) == 2) {
qsizetype index = s.indexOf(QLatin1Char(','));
qsizetype index2 = s.indexOf(QLatin1Char(','), index + 1);
bool xGood, yGood, zGood;
float xCoord = s.left(index).toFloat(&xGood);
float yCoord = s.mid(index + 1, index2 - index - 1).toFloat(&yGood);
float zCoord = s.mid(index2 + 1).toFloat(&zGood);
if (xGood && yGood && zGood) {
return QVector3D(xCoord, yCoord, zCoord);
}
}
}
}
return QVector3D();
}
void ScatterItemModelHandler::modelPosToScatterItem(int modelRow,
int modelColumn,
QScatterDataItem &item)
{
QModelIndex index = m_itemModel->index(modelRow, modelColumn);
float xPos;
float yPos;
float zPos;
if (m_xPosRole != noRoleIndex) {
QVariant xValueVar = index.data(m_xPosRole);
if (m_haveXPosPattern)
xPos = xValueVar.toString().replace(m_xPosPattern, m_xPosReplace).toFloat();
else
xPos = xValueVar.toFloat();
} else {
xPos = 0.0f;
}
if (m_yPosRole != noRoleIndex) {
QVariant yValueVar = index.data(m_yPosRole);
if (m_haveYPosPattern)
yPos = yValueVar.toString().replace(m_yPosPattern, m_yPosReplace).toFloat();
else
yPos = yValueVar.toFloat();
} else {
yPos = 0.0f;
}
if (m_zPosRole != noRoleIndex) {
QVariant zValueVar = index.data(m_zPosRole);
if (m_haveZPosPattern)
zPos = zValueVar.toString().replace(m_zPosPattern, m_zPosReplace).toFloat();
else
zPos = zValueVar.toFloat();
} else {
zPos = 0.0f;
}
if (m_rotationRole != noRoleIndex) {
QVariant rotationVar = index.data(m_rotationRole);
if (m_haveRotationPattern) {
item.setRotation(toQuaternion(
QVariant(rotationVar.toString().replace(m_rotationPattern, m_rotationReplace))));
} else {
item.setRotation(toQuaternion(rotationVar));
}
}
item.setPosition(QVector3D(xPos, yPos, zPos));
}
QVector3D ScatterItemModelHandler::modelDataToScale(int modelRow,
int modelColumn)
{
QModelIndex index = m_itemModel->index(modelRow, modelColumn);
if (m_scaleRole != noRoleIndex) {
QVariant scaleVar = index.data(m_scaleRole);
QVector3D scale = QVector3D(1.0, 1.0, 1.0);
if (m_haveScalePattern) {
scale = toVector3D(
QVariant(scaleVar.toString().replace(m_scalePattern, m_scaleReplace)));
} else {
scale = toVector3D(scaleVar);
}
return scale;
}
return QVector3D(1.0f, 1.0f, 1.0f);
}
// Resolve entire item model into QScatterDataArray.
void ScatterItemModelHandler::resolveModel()
{
if (m_itemModel.isNull()) {
QScatterDataArray empty;
m_proxy->resetArray(empty);
m_proxyArray.clear();
return;
}
Q_TRACE(QGraphs3DScatterItemModelHandlerResolveModel_entry);
m_xPosPattern = m_proxy->xPosRolePattern();
m_yPosPattern = m_proxy->yPosRolePattern();
m_zPosPattern = m_proxy->zPosRolePattern();
m_rotationPattern = m_proxy->rotationRolePattern();
m_scalePattern = m_proxy->scaleRolePattern();
m_xPosReplace = m_proxy->xPosRoleReplace();
m_yPosReplace = m_proxy->yPosRoleReplace();
m_zPosReplace = m_proxy->zPosRoleReplace();
m_rotationReplace = m_proxy->rotationRoleReplace();
m_scaleReplace = m_proxy->scaleRoleReplace();
m_haveXPosPattern = !m_xPosPattern.namedCaptureGroups().isEmpty() && m_xPosPattern.isValid();
m_haveYPosPattern = !m_yPosPattern.namedCaptureGroups().isEmpty() && m_yPosPattern.isValid();
m_haveZPosPattern = !m_zPosPattern.namedCaptureGroups().isEmpty() && m_zPosPattern.isValid();
m_haveRotationPattern = !m_rotationPattern.namedCaptureGroups().isEmpty()
&& m_rotationPattern.isValid();
m_haveScalePattern = !m_scalePattern.namedCaptureGroups().isEmpty()
&& m_scalePattern.isValid();
QHash<int, QByteArray> roleHash = m_itemModel->roleNames();
m_xPosRole = roleHash.key(m_proxy->xPosRole().toLatin1(), noRoleIndex);
m_yPosRole = roleHash.key(m_proxy->yPosRole().toLatin1(), noRoleIndex);
m_zPosRole = roleHash.key(m_proxy->zPosRole().toLatin1(), noRoleIndex);
m_rotationRole = roleHash.key(m_proxy->rotationRole().toLatin1(), noRoleIndex);
m_scaleRole = roleHash.key(m_proxy->scaleRole().toLatin1(), noRoleIndex);
const int columnCount = m_itemModel->columnCount();
const int rowCount = m_itemModel->rowCount();
const int totalCount = rowCount * columnCount;
int runningCount = 0;
// If dimensions have changed, recreate the array
if (m_proxyArray.data() != m_proxy->series()->dataArray().data()
|| totalCount != m_proxyArray.size()) {
m_proxyArray.resize(totalCount);
if (m_scaleRole != noRoleIndex)
m_scaleArray.resize(totalCount);
}
// Parse data into newProxyArray
for (int i = 0; i < rowCount; i++) {
for (int j = 0; j < columnCount; j++) {
modelPosToScatterItem(i, j, m_proxyArray[runningCount]);
if (m_scaleRole != noRoleIndex)
m_scaleArray[runningCount] = modelDataToScale(i, j);
runningCount++;
}
}
Q_TRACE(QGraphs3DScatterItemModelHandlerResolveModel_exit);
m_proxy->resetArray(m_proxyArray);
m_proxy->resetScaleArray(m_scaleArray);
}
QT_END_NAMESPACE
#include "moc_scatteritemmodelhandler_p.cpp"