Revision control
++ // 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"