1 files changed, 125 insertions, 0 deletions
diff --git a/examples/datavisualization/qmlsurfacegallery/datasource.cpp b/examples/datavisualization/qmlsurfacegallery/datasource.cpp
new file mode 100644
index 00000000..7aeea178
--- /dev/null
+++ b/examples/datavisualization/qmlsurfacegallery/datasource.cpp
@@ -0,0 +1,125 @@
+// Copyright (C) 2023 The Qt Company Ltd.
+// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR BSD-3-Clause
+
+#include "datasource.h"
+#include <QtCore/qmath.h>
+#include <QtCore/qrandom.h>
+
+#include <algorithm>
+
+DataSource::DataSource(QObject *parent) :
+    QObject(parent)
+{
+    //! [3]
+    qRegisterMetaType<QSurface3DSeries *>();
+    //! [3]
+}
+
+DataSource::~DataSource()
+{
+    clearData();
+}
+
+void DataSource::generateData(int cacheCount, int rowCount, int columnCount,
+                              float xMin, float xMax,
+                              float yMin, float yMax,
+                              float zMin, float zMax)
+{
+    if (!cacheCount || !rowCount || !columnCount)
+        return;
+
+    clearData();
+
+    // Re-create the cache array
+    m_data.resize(cacheCount);
+    for (int i = 0; i < cacheCount; ++i) {
+        QSurfaceDataArray &array = m_data[i];
+        array.reserve(rowCount);
+        for (int j = 0; j < rowCount; ++j)
+            array.append(new QSurfaceDataRow(columnCount));
+    }
+
+    float xRange = xMax - xMin;
+    float yRange = yMax - yMin;
+    float zRange = zMax - zMin;
+    int cacheIndexStep = columnCount / cacheCount;
+    float cacheStep = float(cacheIndexStep) * xRange / float(columnCount);
+
+    //! [0]
+    // Populate caches
+    auto *generator = QRandomGenerator::global();
+    for (int i = 0; i < cacheCount; ++i) {
+        QSurfaceDataArray &cache = m_data[i];
+        float cacheXAdjustment = cacheStep * i;
+        float cacheIndexAdjustment = cacheIndexStep * i;
+        for (int j = 0; j < rowCount; ++j) {
+            QSurfaceDataRow &row = *(cache[j]);
+            float rowMod = (float(j)) / float(rowCount);
+            float yRangeMod = yRange * rowMod;
+            float zRangeMod = zRange * rowMod;
+            float z = zRangeMod + zMin;
+            qreal rowColWaveAngleMul = M_PI * M_PI * rowMod;
+            float rowColWaveMul = yRangeMod * 0.2f;
+            for (int k = 0; k < columnCount; k++) {
+                float colMod = (float(k)) / float(columnCount);
+                float xRangeMod = xRange * colMod;
+                float x = xRangeMod + xMin + cacheXAdjustment;
+                float colWave = float(qSin((2.0 * M_PI * colMod) - (1.0 / 2.0 * M_PI)) + 1.0);
+                float y = (colWave * ((float(qSin(rowColWaveAngleMul * colMod) + 1.0))))
+                        * rowColWaveMul
+                        + generator->bounded(0.15f) * yRangeMod;
+
+                int index = k + cacheIndexAdjustment;
+                if (index >= columnCount) {
+                    // Wrap over
+                    index -= columnCount;
+                    x -= xRange;
+                }
+                row[index] = QVector3D(x, y, z);
+            }
+        }
+    }
+    //! [0]
+}
+
+void DataSource::update(QSurface3DSeries *series)
+{
+    if (series && !m_data.isEmpty()) {
+        //! [1]
+        // Each iteration uses data from a different cached array
+        if (++m_index >= m_data.size())
+            m_index = 0;
+
+        const QSurfaceDataArray &array = m_data.at(m_index);
+        int newRowCount = array.size();
+        int newColumnCount = array.at(0)->size();
+
+        // If the first time or the dimensions of the cache array have changed,
+        // reconstruct the reset array
+        if (!m_resetArray || series->dataProxy()->rowCount() != newRowCount
+                || series->dataProxy()->columnCount() != newColumnCount) {
+            m_resetArray = new QSurfaceDataArray();
+            m_resetArray->reserve(newRowCount);
+            for (int i = 0; i < newRowCount; ++i)
+                m_resetArray->append(new QSurfaceDataRow(newColumnCount));
+        }
+
+        // Copy items from our cache to the reset array
+        for (int i = 0; i < newRowCount; ++i) {
+            const QSurfaceDataRow &sourceRow = *(array.at(i));
+            QSurfaceDataRow &row = *(*m_resetArray)[i];
+            std::copy(sourceRow.cbegin(), sourceRow.cend(), row.begin());
+        }
+
+        // Notify the proxy that data has changed
+        series->dataProxy()->resetArray(m_resetArray);
+        //! [1]
+    }
+}
+
+void DataSource::clearData()
+{
+    for (auto &array : std::as_const(m_data))
+        qDeleteAll(array);
+    m_data.clear();
+}