| /**************************************************************************** |
| ** |
| ** Copyright (C) 2016 The Qt Company Ltd. |
| ** Contact: https://www.qt.io/licensing/ |
| ** |
| ** This file is part of the Qt Charts module of the Qt Toolkit. |
| ** |
| ** $QT_BEGIN_LICENSE:GPL$ |
| ** Commercial License Usage |
| ** Licensees holding valid commercial Qt licenses may use this file in |
| ** accordance with the commercial license agreement provided with the |
| ** Software or, alternatively, in accordance with the terms contained in |
| ** a written agreement between you and The Qt Company. For licensing terms |
| ** and conditions see https://www.qt.io/terms-conditions. For further |
| ** information use the contact form at https://www.qt.io/contact-us. |
| ** |
| ** GNU General Public License Usage |
| ** Alternatively, this file may be used under the terms of the GNU |
| ** General Public License version 3 or (at your option) any later version |
| ** approved by the KDE Free Qt Foundation. The licenses are as published by |
| ** the Free Software Foundation and appearing in the file LICENSE.GPL3 |
| ** included in the packaging of this file. Please review the following |
| ** information to ensure the GNU General Public License requirements will |
| ** be met: https://www.gnu.org/licenses/gpl-3.0.html. |
| ** |
| ** $QT_END_LICENSE$ |
| ** |
| ****************************************************************************/ |
| |
| #include <QtCharts/QPieModelMapper> |
| #include <private/qpiemodelmapper_p.h> |
| #include <QtCharts/QPieSeries> |
| #include <QtCharts/QPieSlice> |
| #include <QtCore/QAbstractItemModel> |
| |
| QT_CHARTS_BEGIN_NAMESPACE |
| |
| QPieModelMapper::QPieModelMapper(QObject *parent) |
| : QObject(parent), |
| d_ptr(new QPieModelMapperPrivate(this)) |
| { |
| } |
| |
| QAbstractItemModel *QPieModelMapper::model() const |
| { |
| Q_D(const QPieModelMapper); |
| return d->m_model; |
| } |
| |
| void QPieModelMapper::setModel(QAbstractItemModel *model) |
| { |
| if (model == 0) |
| return; |
| |
| Q_D(QPieModelMapper); |
| if (d->m_model) { |
| disconnect(d->m_model, 0, d, 0); |
| } |
| |
| d->m_model = model; |
| d->initializePieFromModel(); |
| // connect signals from the model |
| connect(d->m_model, SIGNAL(modelReset()), d, SLOT(initializePieFromModel())); |
| connect(d->m_model, SIGNAL(dataChanged(QModelIndex,QModelIndex)), d, SLOT(modelUpdated(QModelIndex,QModelIndex))); |
| connect(d->m_model, SIGNAL(rowsInserted(QModelIndex,int,int)), d, SLOT(modelRowsAdded(QModelIndex,int,int))); |
| connect(d->m_model, SIGNAL(rowsRemoved(QModelIndex,int,int)), d, SLOT(modelRowsRemoved(QModelIndex,int,int))); |
| connect(d->m_model, SIGNAL(columnsInserted(QModelIndex,int,int)), d, SLOT(modelColumnsAdded(QModelIndex,int,int))); |
| connect(d->m_model, SIGNAL(columnsRemoved(QModelIndex,int,int)), d, SLOT(modelColumnsRemoved(QModelIndex,int,int))); |
| connect(d->m_model, SIGNAL(destroyed()), d, SLOT(handleModelDestroyed())); |
| } |
| |
| QPieSeries *QPieModelMapper::series() const |
| { |
| Q_D(const QPieModelMapper); |
| return d->m_series; |
| } |
| |
| void QPieModelMapper::setSeries(QPieSeries *series) |
| { |
| Q_D(QPieModelMapper); |
| if (d->m_series) { |
| disconnect(d->m_series, 0, d, 0); |
| } |
| |
| if (series == 0) |
| return; |
| |
| d->m_series = series; |
| d->initializePieFromModel(); |
| // connect the signals from the series |
| connect(d->m_series, SIGNAL(added(QList<QPieSlice*>)), d, SLOT(slicesAdded(QList<QPieSlice*>))); |
| connect(d->m_series, SIGNAL(removed(QList<QPieSlice*>)), d, SLOT(slicesRemoved(QList<QPieSlice*>))); |
| connect(d->m_series, SIGNAL(destroyed()), d, SLOT(handleSeriesDestroyed())); |
| } |
| |
| /*! |
| Defines which row/column of the model contains the first slice value. |
| Minimal and default value is: 0 |
| */ |
| int QPieModelMapper::first() const |
| { |
| Q_D(const QPieModelMapper); |
| return d->m_first; |
| } |
| |
| /*! |
| Sets which row/column of the model contains the \a first slice value. |
| Minimal and default value is: 0 |
| */ |
| void QPieModelMapper::setFirst(int first) |
| { |
| Q_D(QPieModelMapper); |
| d->m_first = qMax(first, 0); |
| d->initializePieFromModel(); |
| } |
| |
| /*! |
| Defines the number of rows/columns of the model that are mapped as the data for QPieSeries |
| Minimal and default value is: -1 (count limited by the number of rows/columns in the model) |
| */ |
| int QPieModelMapper::count() const |
| { |
| Q_D(const QPieModelMapper); |
| return d->m_count; |
| } |
| |
| /*! |
| Defines the \a count of rows/columns of the model that are mapped as the data for QPieSeries |
| Minimal and default value is: -1 (count limited by the number of rows/columns in the model) |
| */ |
| void QPieModelMapper::setCount(int count) |
| { |
| Q_D(QPieModelMapper); |
| d->m_count = qMax(count, -1); |
| d->initializePieFromModel(); |
| } |
| |
| /*! |
| Returns the orientation that is used when QPieModelMapper accesses the model. |
| This mean whether the consecutive values/labels of the pie are read from row (Qt::Horizontal) |
| or from columns (Qt::Vertical) |
| */ |
| Qt::Orientation QPieModelMapper::orientation() const |
| { |
| Q_D(const QPieModelMapper); |
| return d->m_orientation; |
| } |
| |
| /*! |
| Returns the \a orientation that is used when QPieModelMapper accesses the model. |
| This mean whether the consecutive values/labels of the pie are read from row (Qt::Horizontal) |
| or from columns (Qt::Vertical) |
| */ |
| void QPieModelMapper::setOrientation(Qt::Orientation orientation) |
| { |
| Q_D(QPieModelMapper); |
| d->m_orientation = orientation; |
| d->initializePieFromModel(); |
| } |
| |
| /*! |
| Returns which section of the model is kept in sync with the values of the pie's slices |
| */ |
| int QPieModelMapper::valuesSection() const |
| { |
| Q_D(const QPieModelMapper); |
| return d->m_valuesSection; |
| } |
| |
| /*! |
| Sets the model section that is kept in sync with the pie slices values. |
| Parameter \a valuesSection specifies the section of the model. |
| */ |
| void QPieModelMapper::setValuesSection(int valuesSection) |
| { |
| Q_D(QPieModelMapper); |
| d->m_valuesSection = qMax(-1, valuesSection); |
| d->initializePieFromModel(); |
| } |
| |
| /*! |
| Returns which section of the model is kept in sync with the labels of the pie's slices |
| */ |
| int QPieModelMapper::labelsSection() const |
| { |
| Q_D(const QPieModelMapper); |
| return d->m_labelsSection; |
| } |
| |
| /*! |
| Sets the model section that is kept in sync with the pie slices labels. |
| Parameter \a labelsSection specifies the section of the model. |
| */ |
| void QPieModelMapper::setLabelsSection(int labelsSection) |
| { |
| Q_D(QPieModelMapper); |
| d->m_labelsSection = qMax(-1, labelsSection); |
| d->initializePieFromModel(); |
| } |
| |
| /////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// |
| |
| QPieModelMapperPrivate::QPieModelMapperPrivate(QPieModelMapper *q) : |
| QObject(q), |
| m_series(0), |
| m_model(0), |
| m_first(0), |
| m_count(-1), |
| m_orientation(Qt::Vertical), |
| m_valuesSection(-1), |
| m_labelsSection(-1), |
| m_seriesSignalsBlock(false), |
| m_modelSignalsBlock(false), |
| q_ptr(q) |
| { |
| } |
| |
| void QPieModelMapperPrivate::blockModelSignals(bool block) |
| { |
| m_modelSignalsBlock = block; |
| } |
| |
| void QPieModelMapperPrivate::blockSeriesSignals(bool block) |
| { |
| m_seriesSignalsBlock = block; |
| } |
| |
| |
| QPieSlice *QPieModelMapperPrivate::pieSlice(QModelIndex index) const |
| { |
| if (!index.isValid()) |
| return 0; // index is invalid |
| |
| if (m_orientation == Qt::Vertical && (index.column() == m_valuesSection || index.column() == m_labelsSection)) { |
| if (index.row() >= m_first && (m_count == - 1 || index.row() < m_first + m_count)) { |
| if (m_model->index(index.row(), m_valuesSection).isValid() && m_model->index(index.row(), m_labelsSection).isValid()) |
| return m_series->slices().at(index.row() - m_first); |
| else |
| return 0; |
| } |
| } else if (m_orientation == Qt::Horizontal && (index.row() == m_valuesSection || index.row() == m_labelsSection)) { |
| if (index.column() >= m_first && (m_count == - 1 || index.column() < m_first + m_count)) { |
| if (m_model->index(m_valuesSection, index.column()).isValid() && m_model->index(m_labelsSection, index.column()).isValid()) |
| return m_series->slices().at(index.column() - m_first); |
| else |
| return 0; |
| } |
| } |
| return 0; // This part of model has not been mapped to any slice |
| } |
| |
| QModelIndex QPieModelMapperPrivate::valueModelIndex(int slicePos) |
| { |
| if (m_count != -1 && slicePos >= m_count) |
| return QModelIndex(); // invalid |
| |
| if (m_orientation == Qt::Vertical) |
| return m_model->index(slicePos + m_first, m_valuesSection); |
| else |
| return m_model->index(m_valuesSection, slicePos + m_first); |
| } |
| |
| QModelIndex QPieModelMapperPrivate::labelModelIndex(int slicePos) |
| { |
| if (m_count != -1 && slicePos >= m_count) |
| return QModelIndex(); // invalid |
| |
| if (m_orientation == Qt::Vertical) |
| return m_model->index(slicePos + m_first, m_labelsSection); |
| else |
| return m_model->index(m_labelsSection, slicePos + m_first); |
| } |
| |
| bool QPieModelMapperPrivate::isLabelIndex(QModelIndex index) const |
| { |
| if (m_orientation == Qt::Vertical && index.column() == m_labelsSection) |
| return true; |
| else if (m_orientation == Qt::Horizontal && index.row() == m_labelsSection) |
| return true; |
| |
| return false; |
| } |
| |
| bool QPieModelMapperPrivate::isValueIndex(QModelIndex index) const |
| { |
| if (m_orientation == Qt::Vertical && index.column() == m_valuesSection) |
| return true; |
| else if (m_orientation == Qt::Horizontal && index.row() == m_valuesSection) |
| return true; |
| |
| return false; |
| } |
| |
| void QPieModelMapperPrivate::slicesAdded(QList<QPieSlice *> slices) |
| { |
| if (m_seriesSignalsBlock) |
| return; |
| |
| if (slices.count() == 0) |
| return; |
| |
| int firstIndex = m_series->slices().indexOf(slices.at(0)); |
| if (firstIndex == -1) |
| return; |
| |
| if (m_count != -1) |
| m_count += slices.count(); |
| |
| for (int i = firstIndex; i < firstIndex + slices.count(); i++) { |
| m_slices.insert(i, slices.at(i - firstIndex)); |
| connect(slices.at(i - firstIndex), SIGNAL(labelChanged()), this, SLOT(sliceLabelChanged())); |
| connect(slices.at(i - firstIndex), SIGNAL(valueChanged()), this, SLOT(sliceValueChanged())); |
| } |
| |
| blockModelSignals(); |
| if (m_orientation == Qt::Vertical) |
| m_model->insertRows(firstIndex + m_first, slices.count()); |
| else |
| m_model->insertColumns(firstIndex + m_first, slices.count()); |
| |
| for (int i = firstIndex; i < firstIndex + slices.count(); i++) { |
| m_model->setData(valueModelIndex(i), slices.at(i - firstIndex)->value()); |
| m_model->setData(labelModelIndex(i), slices.at(i - firstIndex)->label()); |
| } |
| blockModelSignals(false); |
| } |
| |
| void QPieModelMapperPrivate::slicesRemoved(QList<QPieSlice *> slices) |
| { |
| if (m_seriesSignalsBlock) |
| return; |
| |
| if (slices.count() == 0) |
| return; |
| |
| int firstIndex = m_slices.indexOf(slices.at(0)); |
| if (firstIndex == -1) |
| return; |
| |
| if (m_count != -1) |
| m_count -= slices.count(); |
| |
| for (int i = firstIndex + slices.count() - 1; i >= firstIndex; i--) |
| m_slices.removeAt(i); |
| |
| blockModelSignals(); |
| if (m_orientation == Qt::Vertical) |
| m_model->removeRows(firstIndex + m_first, slices.count()); |
| else |
| m_model->removeColumns(firstIndex + m_first, slices.count()); |
| blockModelSignals(false); |
| } |
| |
| void QPieModelMapperPrivate::sliceLabelChanged() |
| { |
| if (m_seriesSignalsBlock) |
| return; |
| |
| blockModelSignals(); |
| QPieSlice *slice = qobject_cast<QPieSlice *>(QObject::sender()); |
| m_model->setData(labelModelIndex(m_series->slices().indexOf(slice)), slice->label()); |
| blockModelSignals(false); |
| } |
| |
| void QPieModelMapperPrivate::sliceValueChanged() |
| { |
| if (m_seriesSignalsBlock) |
| return; |
| |
| blockModelSignals(); |
| QPieSlice *slice = qobject_cast<QPieSlice *>(QObject::sender()); |
| m_model->setData(valueModelIndex(m_series->slices().indexOf(slice)), slice->value()); |
| blockModelSignals(false); |
| } |
| |
| void QPieModelMapperPrivate::handleSeriesDestroyed() |
| { |
| m_series = 0; |
| } |
| |
| void QPieModelMapperPrivate::modelUpdated(QModelIndex topLeft, QModelIndex bottomRight) |
| { |
| if (m_model == 0 || m_series == 0) |
| return; |
| |
| if (m_modelSignalsBlock) |
| return; |
| |
| blockSeriesSignals(); |
| QModelIndex index; |
| QPieSlice *slice; |
| for (int row = topLeft.row(); row <= bottomRight.row(); row++) { |
| for (int column = topLeft.column(); column <= bottomRight.column(); column++) { |
| index = topLeft.sibling(row, column); |
| slice = pieSlice(index); |
| if (slice) { |
| if (isValueIndex(index)) |
| slice->setValue(m_model->data(index, Qt::DisplayRole).toReal()); |
| if (isLabelIndex(index)) |
| slice->setLabel(m_model->data(index, Qt::DisplayRole).toString()); |
| } |
| } |
| } |
| blockSeriesSignals(false); |
| } |
| |
| |
| void QPieModelMapperPrivate::modelRowsAdded(QModelIndex parent, int start, int end) |
| { |
| Q_UNUSED(parent); |
| if (m_modelSignalsBlock) |
| return; |
| |
| blockSeriesSignals(); |
| if (m_orientation == Qt::Vertical) |
| insertData(start, end); |
| else if (start <= m_valuesSection || start <= m_labelsSection) // if the changes affect the map - reinitialize the pie |
| initializePieFromModel(); |
| blockSeriesSignals(false); |
| } |
| |
| void QPieModelMapperPrivate::modelRowsRemoved(QModelIndex parent, int start, int end) |
| { |
| Q_UNUSED(parent); |
| if (m_modelSignalsBlock) |
| return; |
| |
| blockSeriesSignals(); |
| if (m_orientation == Qt::Vertical) |
| removeData(start, end); |
| else if (start <= m_valuesSection || start <= m_labelsSection) // if the changes affect the map - reinitialize the pie |
| initializePieFromModel(); |
| blockSeriesSignals(false); |
| } |
| |
| void QPieModelMapperPrivate::modelColumnsAdded(QModelIndex parent, int start, int end) |
| { |
| Q_UNUSED(parent); |
| if (m_modelSignalsBlock) |
| return; |
| |
| blockSeriesSignals(); |
| if (m_orientation == Qt::Horizontal) |
| insertData(start, end); |
| else if (start <= m_valuesSection || start <= m_labelsSection) // if the changes affect the map - reinitialize the pie |
| initializePieFromModel(); |
| blockSeriesSignals(false); |
| } |
| |
| void QPieModelMapperPrivate::modelColumnsRemoved(QModelIndex parent, int start, int end) |
| { |
| Q_UNUSED(parent); |
| if (m_modelSignalsBlock) |
| return; |
| |
| blockSeriesSignals(); |
| if (m_orientation == Qt::Horizontal) |
| removeData(start, end); |
| else if (start <= m_valuesSection || start <= m_labelsSection) // if the changes affect the map - reinitialize the pie |
| initializePieFromModel(); |
| blockSeriesSignals(false); |
| } |
| |
| void QPieModelMapperPrivate::handleModelDestroyed() |
| { |
| m_model = 0; |
| } |
| |
| void QPieModelMapperPrivate::insertData(int start, int end) |
| { |
| if (m_model == 0 || m_series == 0) |
| return; |
| |
| if (m_count != -1 && start >= m_first + m_count) { |
| return; |
| } else { |
| int addedCount = end - start + 1; |
| if (m_count != -1 && addedCount > m_count) |
| addedCount = m_count; |
| int first = qMax(start, m_first); |
| int last = qMin(first + addedCount - 1, m_orientation == Qt::Vertical ? m_model->rowCount() - 1 : m_model->columnCount() - 1); |
| for (int i = first; i <= last; i++) { |
| QModelIndex valueIndex = valueModelIndex(i - m_first); |
| QModelIndex labelIndex = labelModelIndex(i - m_first); |
| if (valueIndex.isValid() && labelIndex.isValid()) { |
| QPieSlice *slice = new QPieSlice; |
| slice->setValue(m_model->data(valueIndex, Qt::DisplayRole).toDouble()); |
| slice->setLabel(m_model->data(labelIndex, Qt::DisplayRole).toString()); |
| connect(slice, SIGNAL(labelChanged()), this, SLOT(sliceLabelChanged())); |
| connect(slice, SIGNAL(valueChanged()), this, SLOT(sliceValueChanged())); |
| m_series->insert(i - m_first, slice); |
| m_slices.insert(i - m_first, slice); |
| } |
| } |
| |
| // remove excess of slices (abouve m_count) |
| if (m_count != -1 && m_series->slices().size() > m_count) |
| for (int i = m_series->slices().size() - 1; i >= m_count; i--) { |
| m_series->remove(m_series->slices().at(i)); |
| m_slices.removeAt(i); |
| } |
| } |
| } |
| |
| void QPieModelMapperPrivate::removeData(int start, int end) |
| { |
| if (m_model == 0 || m_series == 0) |
| return; |
| |
| int removedCount = end - start + 1; |
| if (m_count != -1 && start >= m_first + m_count) { |
| return; |
| } else { |
| int toRemove = qMin(m_series->slices().size(), removedCount); // first find how many items can actually be removed |
| int first = qMax(start, m_first); // get the index of the first item that will be removed. |
| int last = qMin(first + toRemove - 1, m_series->slices().size() + m_first - 1); // get the index of the last item that will be removed. |
| for (int i = last; i >= first; i--) { |
| m_series->remove(m_series->slices().at(i - m_first)); |
| m_slices.removeAt(i - m_first); |
| } |
| |
| if (m_count != -1) { |
| int itemsAvailable; // check how many are available to be added |
| if (m_orientation == Qt::Vertical) |
| itemsAvailable = m_model->rowCount() - m_first - m_series->slices().size(); |
| else |
| itemsAvailable = m_model->columnCount() - m_first - m_series->slices().size(); |
| int toBeAdded = qMin(itemsAvailable, m_count - m_series->slices().size()); // add not more items than there is space left to be filled. |
| int currentSize = m_series->slices().size(); |
| if (toBeAdded > 0) |
| for (int i = m_series->slices().size(); i < currentSize + toBeAdded; i++) { |
| QModelIndex valueIndex = valueModelIndex(i - m_first); |
| QModelIndex labelIndex = labelModelIndex(i - m_first); |
| if (valueIndex.isValid() && labelIndex.isValid()) { |
| QPieSlice *slice = new QPieSlice; |
| slice->setValue(m_model->data(valueIndex, Qt::DisplayRole).toDouble()); |
| slice->setLabel(m_model->data(labelIndex, Qt::DisplayRole).toString()); |
| m_series->insert(i, slice); |
| m_slices.insert(i, slice); |
| } |
| } |
| } |
| } |
| } |
| |
| void QPieModelMapperPrivate::initializePieFromModel() |
| { |
| if (m_model == 0 || m_series == 0) |
| return; |
| |
| blockSeriesSignals(); |
| // clear current content |
| m_series->clear(); |
| m_slices.clear(); |
| |
| // create the initial slices set |
| int slicePos = 0; |
| QModelIndex valueIndex = valueModelIndex(slicePos); |
| QModelIndex labelIndex = labelModelIndex(slicePos); |
| while (valueIndex.isValid() && labelIndex.isValid()) { |
| QPieSlice *slice = new QPieSlice; |
| slice->setLabel(m_model->data(labelIndex, Qt::DisplayRole).toString()); |
| slice->setValue(m_model->data(valueIndex, Qt::DisplayRole).toDouble()); |
| connect(slice, SIGNAL(labelChanged()), this, SLOT(sliceLabelChanged())); |
| connect(slice, SIGNAL(valueChanged()), this, SLOT(sliceValueChanged())); |
| m_series->append(slice); |
| m_slices.append(slice); |
| slicePos++; |
| valueIndex = valueModelIndex(slicePos); |
| labelIndex = labelModelIndex(slicePos); |
| } |
| blockSeriesSignals(false); |
| } |
| |
| QT_CHARTS_END_NAMESPACE |
| |
| #include "moc_qpiemodelmapper_p.cpp" |
| #include "moc_qpiemodelmapper.cpp" |