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 "baritemmodelhandler_p.h"
#include "qbar3dseries_p.h"
#include <qtgraphs_tracepoints_p.h>
QT_BEGIN_NAMESPACE
Q_TRACE_PREFIX(qtgraphs,
"QT_BEGIN_NAMESPACE" \
"class BarItemModelHandler;" \
"QT_END_NAMESPACE"
)
Q_TRACE_POINT(qtgraphs, QGraphs3DBarItemModelHandlerResolveModel_modelCategories_entry);
Q_TRACE_POINT(qtgraphs, QGraphs3DBarItemModelHandlerResolveModel_modelCategories_exit);
Q_TRACE_POINT(qtgraphs, QGraphs3DBarItemModelHandlerResolveModel_noModelCategories_entry);
Q_TRACE_POINT(qtgraphs, QGraphs3DBarItemModelHandlerResolveModel_noModelCategories_exit);
Q_TRACE_POINT(qtgraphs, QGraphs3DBarItemModelHandleDataChanged_entry, bool useModelCategories);
Q_TRACE_POINT(qtgraphs, QGraphs3DBarItemModelHandleDataChanged_exit);
Q_TRACE_POINT(qtgraphs, QGraphs3DBarItemModelHandlerResolveModel_labelGen_entry);
Q_TRACE_POINT(qtgraphs, QGraphs3DBarItemModelHandlerResolveModel_labelGen_exit);
BarItemModelHandler::BarItemModelHandler(QItemModelBarDataProxy *proxy, QObject *parent)
: AbstractItemModelHandler(parent)
, m_proxy(proxy)
, m_proxyArray(0)
, m_columnCount(0)
, m_valueRole(noRoleIndex)
, m_rotationRole(noRoleIndex)
, m_haveValuePattern(false)
, m_haveRotationPattern(false)
{}
BarItemModelHandler::~BarItemModelHandler() {}
void BarItemModelHandler::handleDataChanged(const QModelIndex &topLeft,
const QModelIndex &bottomRight,
const QList<int> &roles)
{
// Do nothing if full reset already pending
if (!m_fullReset) {
Q_TRACE_SCOPE(QGraphs3DBarItemModelHandleDataChanged, m_proxy->useModelCategories());
if (!m_proxy->useModelCategories()) {
// If the data model doesn't directly map rows and columns, we cannot
// optimize
AbstractItemModelHandler::handleDataChanged(topLeft, bottomRight, roles);
} else {
int startRow = qMin(topLeft.row(), bottomRight.row());
int endRow = qMax(topLeft.row(), bottomRight.row());
int startCol = qMin(topLeft.column(), bottomRight.column());
int endCol = qMax(topLeft.column(), bottomRight.column());
for (int i = startRow; i <= endRow; i++) {
for (int j = startCol; j <= endCol; j++) {
QModelIndex index = m_itemModel->index(i, j);
QBarDataItem item;
QVariant valueVar = index.data(m_valueRole);
float value;
if (m_haveValuePattern)
value = valueVar.toString().replace(m_valuePattern, m_valueReplace).toFloat();
else
value = valueVar.toFloat();
item.setValue(value);
if (m_rotationRole != noRoleIndex) {
QVariant rotationVar = index.data(m_rotationRole);
float rotation;
if (m_haveRotationPattern) {
rotation = rotationVar.toString()
.replace(m_rotationPattern, m_rotationReplace)
.toFloat();
} else {
rotation = rotationVar.toFloat();
}
item.setRotation(rotation);
}
m_proxy->setItem(i, j, item);
}
}
}
}
}
// Resolve entire item model into QBarDataArray.
void BarItemModelHandler::resolveModel()
{
if (m_itemModel.isNull()) {
m_proxy->resetArray();
return;
}
if (!m_proxy->useModelCategories()
&& (m_proxy->rowRole().isEmpty() || m_proxy->columnRole().isEmpty())) {
m_proxy->resetArray();
return;
}
// Value and rotation patterns can be reused on single item changes,
// so store them to member variables.
QRegularExpression rowPattern(m_proxy->rowRolePattern());
QRegularExpression colPattern(m_proxy->columnRolePattern());
m_valuePattern = m_proxy->valueRolePattern();
m_rotationPattern = m_proxy->rotationRolePattern();
QString rowReplace = m_proxy->rowRoleReplace();
QString colReplace = m_proxy->columnRoleReplace();
m_valueReplace = m_proxy->valueRoleReplace();
m_rotationReplace = m_proxy->rotationRoleReplace();
bool haveRowPattern = !rowPattern.namedCaptureGroups().isEmpty() && rowPattern.isValid();
bool haveColPattern = !colPattern.namedCaptureGroups().isEmpty() && colPattern.isValid();
m_haveValuePattern = !m_valuePattern.namedCaptureGroups().isEmpty() && m_valuePattern.isValid();
m_haveRotationPattern = !m_rotationPattern.namedCaptureGroups().isEmpty()
&& m_rotationPattern.isValid();
QStringList rowLabels;
QStringList columnLabels;
QHash<int, QByteArray> roleHash = m_itemModel->roleNames();
// Default value role to display role if no mapping
m_valueRole = roleHash.key(m_proxy->valueRole().toLatin1(), Qt::DisplayRole);
m_rotationRole = roleHash.key(m_proxy->rotationRole().toLatin1(), noRoleIndex);
int rowCount = m_itemModel->rowCount();
int columnCount = m_itemModel->columnCount();
if (m_proxy->useModelCategories()) {
if (!m_proxy->series())
return;
Q_TRACE(QGraphs3DBarItemModelHandlerResolveModel_modelCategories_entry);
// If dimensions have changed, recreate the array
if (m_proxyArray.data() != m_proxy->series()->dataArray().data()
|| columnCount != m_columnCount || rowCount != m_proxyArray.size()) {
m_proxyArray.reserve(rowCount);
for (int i = 0; i < rowCount; i++)
m_proxyArray.append(QBarDataRow(columnCount));
}
for (int i = 0; i < rowCount; i++) {
QBarDataRow &newProxyRow = m_proxyArray[i];
for (int j = 0; j < columnCount; j++) {
QModelIndex index = m_itemModel->index(i, j);
QVariant valueVar = index.data(m_valueRole);
float value;
if (m_haveValuePattern)
value = valueVar.toString().replace(m_valuePattern, m_valueReplace).toFloat();
else
value = valueVar.toFloat();
newProxyRow[j].setValue(value);
if (m_rotationRole != noRoleIndex) {
QVariant rotationVar = index.data(m_rotationRole);
float rotation;
if (m_haveRotationPattern) {
rotation = rotationVar.toString()
.replace(m_rotationPattern, m_rotationReplace)
.toFloat();
} else {
rotation = rotationVar.toFloat();
}
newProxyRow[j].setRotation(rotation);
}
}
}
Q_TRACE(QGraphs3DBarItemModelHandlerResolveModel_modelCategories_exit);
Q_TRACE_SCOPE(QGraphs3DBarItemModelHandlerResolveModel_labelGen)
// Generate labels from headers if using model rows/columns
for (int i = 0; i < rowCount; i++)
rowLabels << m_itemModel->headerData(i, Qt::Vertical).toString();
for (int i = 0; i < columnCount; i++)
columnLabels << m_itemModel->headerData(i, Qt::Horizontal).toString();
m_columnCount = columnCount;
} else {
Q_TRACE_SCOPE(QGraphs3DBarItemModelHandlerResolveModel_noModelCategories)
int rowRole = roleHash.key(m_proxy->rowRole().toLatin1());
int columnRole = roleHash.key(m_proxy->columnRole().toLatin1());
bool generateRows = m_proxy->autoRowCategories();
bool generateColumns = m_proxy->autoColumnCategories();
QStringList rowList;
QStringList columnList;
// For detecting duplicates in categories generation, using QHashes should
// be faster than simple QStringList::contains() check.
QHash<QString, bool> rowListHash;
QHash<QString, bool> columnListHash;
// Sort values into rows and columns
typedef QHash<QString, float> ColumnValueMap;
QHash<QString, ColumnValueMap> itemValueMap;
QHash<QString, ColumnValueMap> itemRotationMap;
bool cumulative = m_proxy->multiMatchBehavior()
== QItemModelBarDataProxy::MultiMatchBehavior::Average
|| m_proxy->multiMatchBehavior()
== QItemModelBarDataProxy::MultiMatchBehavior::Cumulative;
bool countMatches = m_proxy->multiMatchBehavior()
== QItemModelBarDataProxy::MultiMatchBehavior::Average;
bool takeFirst = m_proxy->multiMatchBehavior()
== QItemModelBarDataProxy::MultiMatchBehavior::First;
QHash<QString, QHash<QString, int>> *matchCountMap = 0;
if (countMatches)
matchCountMap = new QHash<QString, QHash<QString, int>>;
for (int i = 0; i < rowCount; i++) {
for (int j = 0; j < columnCount; j++) {
QModelIndex index = m_itemModel->index(i, j);
QString rowRoleStr = index.data(rowRole).toString();
if (haveRowPattern)
rowRoleStr.replace(rowPattern, rowReplace);
QString columnRoleStr = index.data(columnRole).toString();
if (haveColPattern)
columnRoleStr.replace(colPattern, colReplace);
QVariant valueVar = index.data(m_valueRole);
float value;
if (m_haveValuePattern)
value = valueVar.toString().replace(m_valuePattern, m_valueReplace).toFloat();
else
value = valueVar.toFloat();
if (countMatches)
(*matchCountMap)[rowRoleStr][columnRoleStr]++;
if (cumulative) {
itemValueMap[rowRoleStr][columnRoleStr] += value;
} else {
if (takeFirst && itemValueMap.contains(rowRoleStr)) {
if (itemValueMap.value(rowRoleStr).contains(columnRoleStr))
continue; // We already have a value for this row/column combo
}
itemValueMap[rowRoleStr][columnRoleStr] = value;
}
if (m_rotationRole != noRoleIndex) {
QVariant rotationVar = index.data(m_rotationRole);
float rotation;
if (m_haveRotationPattern) {
rotation = rotationVar.toString()
.replace(m_rotationPattern, m_rotationReplace)
.toFloat();
} else {
rotation = rotationVar.toFloat();
}
if (cumulative) {
itemRotationMap[rowRoleStr][columnRoleStr] += rotation;
} else {
// We know we are in take last mode if we get here,
// as take first mode skips to next loop already earlier
itemRotationMap[rowRoleStr][columnRoleStr] = rotation;
}
}
if (generateRows && !rowListHash.value(rowRoleStr, false)) {
rowListHash.insert(rowRoleStr, true);
rowList << rowRoleStr;
}
if (generateColumns && !columnListHash.value(columnRoleStr, false)) {
columnListHash.insert(columnRoleStr, true);
columnList << columnRoleStr;
}
}
}
if (generateRows)
m_proxy->d_func()->m_rowCategories = rowList;
else
rowList = m_proxy->rowCategories();
if (generateColumns)
m_proxy->d_func()->m_columnCategories = columnList;
else
columnList = m_proxy->columnCategories();
// If dimensions have changed, recreate the array
if (m_proxyArray.data() != m_proxy->series()->dataArray().data()
|| columnList.size() != m_columnCount || rowList.size() != m_proxyArray.size()) {
m_proxyArray.clear();
m_proxyArray.reserve(rowList.size());
for (int i = 0; i < rowList.size(); i++)
m_proxyArray.append(QBarDataRow(columnList.size()));
}
// Create new data array from itemValueMap
for (int i = 0; i < rowList.size(); i++) {
QString rowKey = rowList.at(i);
QBarDataRow &newProxyRow = m_proxyArray[i];
for (int j = 0; j < columnList.size(); j++) {
float value = itemValueMap[rowKey][columnList.at(j)];
if (countMatches)
value /= float((*matchCountMap)[rowKey][columnList.at(j)]);
newProxyRow[j].setValue(value);
if (m_rotationRole != noRoleIndex) {
float angle = itemRotationMap[rowKey][columnList.at(j)];
if (countMatches)
angle /= float((*matchCountMap)[rowKey][columnList.at(j)]);
newProxyRow[j].setRotation(angle);
}
}
}
if (!m_proxy->series()->customRowLabels().isEmpty())
rowLabels = m_proxy->series()->customRowLabels();
else
rowLabels = rowList;
if (!m_proxy->series()->customColumnLabels().isEmpty())
columnLabels = m_proxy->series()->customColumnLabels();
else
columnLabels = columnList;
m_columnCount = columnList.size();
delete matchCountMap;
}
m_proxy->resetArray(m_proxyArray, rowLabels, columnLabels);
}
QT_END_NAMESPACE
#include "moc_baritemmodelhandler_p.cpp"