summaryrefslogtreecommitdiffstats
path: root/examples/charts/chartsgallery/modeldatamodel.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'examples/charts/chartsgallery/modeldatamodel.cpp')
-rw-r--r--examples/charts/chartsgallery/modeldatamodel.cpp92
1 files changed, 92 insertions, 0 deletions
diff --git a/examples/charts/chartsgallery/modeldatamodel.cpp b/examples/charts/chartsgallery/modeldatamodel.cpp
new file mode 100644
index 00000000..af5bc9dd
--- /dev/null
+++ b/examples/charts/chartsgallery/modeldatamodel.cpp
@@ -0,0 +1,92 @@
+// Copyright (C) 2023 The Qt Company Ltd.
+// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only
+
+#include "modeldatamodel.h"
+
+#include <QColor>
+#include <QList>
+#include <QRandomGenerator>
+#include <QRect>
+
+ModelDataModel::ModelDataModel(QObject *parent) :
+ QAbstractTableModel(parent)
+{
+ m_columnCount = 4;
+ m_rowCount = 15;
+
+ // m_data
+ for (int i = 0; i < m_rowCount; i++) {
+ auto dataList = new QList<qreal>(m_columnCount);
+ for (int k = 0; k < dataList->size(); k++) {
+ if (k % 2 == 0)
+ dataList->replace(k, i * 50 + QRandomGenerator::global()->bounded(20));
+ else
+ dataList->replace(k, QRandomGenerator::global()->bounded(100));
+ }
+ m_data.append(dataList);
+ }
+}
+
+int ModelDataModel::rowCount(const QModelIndex &parent) const
+{
+ Q_UNUSED(parent);
+ return m_data.count();
+}
+
+int ModelDataModel::columnCount(const QModelIndex &parent) const
+{
+ Q_UNUSED(parent);
+ return m_columnCount;
+}
+
+QVariant ModelDataModel::headerData(int section, Qt::Orientation orientation, int role) const
+{
+ if (role != Qt::DisplayRole)
+ return QVariant();
+
+ if (orientation == Qt::Horizontal) {
+ if (section % 2 == 0)
+ return "x";
+ else
+ return "y";
+ } else {
+ return QString("%1").arg(section + 1);
+ }
+}
+
+QVariant ModelDataModel::data(const QModelIndex &index, int role) const
+{
+ if (role == Qt::DisplayRole) {
+ return m_data[index.row()]->at(index.column());
+ } else if (role == Qt::EditRole) {
+ return m_data[index.row()]->at(index.column());
+ } else if (role == Qt::BackgroundRole) {
+ for (const QRect &rect : m_mapping) {
+ if (rect.contains(index.column(), index.row()))
+ return QColor(m_mapping.key(rect));
+ }
+ // cell not mapped return white color
+ return QColor(Qt::white);
+ }
+ return QVariant();
+}
+
+bool ModelDataModel::setData(const QModelIndex &index, const QVariant &value, int role)
+{
+ if (index.isValid() && role == Qt::EditRole) {
+ m_data[index.row()]->replace(index.column(), value.toDouble());
+ emit dataChanged(index, index);
+ return true;
+ }
+ return false;
+}
+
+Qt::ItemFlags ModelDataModel::flags(const QModelIndex &index) const
+{
+ return QAbstractItemModel::flags(index) | Qt::ItemIsEditable;
+}
+
+void ModelDataModel::addMapping(const QString &color, const QRect &area)
+{
+ m_mapping.insert(color, area);
+}