customtablemodel.cpp
255 lines
| 7.5 KiB
| text/x-c
|
CppLexer
Jani Honkonen
|
r830 | /**************************************************************************** | ||
** | ||||
** Copyright (C) 2012 Digia Plc | ||||
** All rights reserved. | ||||
** For any questions to Digia, please use contact form at http://qt.digia.com | ||||
** | ||||
** This file is part of the Qt Commercial Charts Add-on. | ||||
** | ||||
** $QT_BEGIN_LICENSE$ | ||||
** Licensees holding valid Qt Commercial licenses may use this file in | ||||
** accordance with the Qt Commercial License Agreement provided with the | ||||
** Software or, alternatively, in accordance with the terms contained in | ||||
** a written agreement between you and Digia. | ||||
** | ||||
** If you have questions regarding the use of this file, please use | ||||
** contact form at http://qt.digia.com | ||||
** $QT_END_LICENSE$ | ||||
** | ||||
****************************************************************************/ | ||||
Marek Rosa
|
r519 | #include "customtablemodel.h" | ||
Marek Rosa
|
r630 | #include <QVector> | ||
Marek Rosa
|
r734 | #include <QTime> | ||
Marek Rosa
|
r519 | |||
CustomTableModel::CustomTableModel(QObject *parent) : | ||||
QAbstractTableModel(parent) | ||||
{ | ||||
Marek Rosa
|
r734 | // m_points.append(QPointF(10, 50)); | ||
// m_labels.append("Apples"); | ||||
// m_points.append(QPointF(60, 70)); | ||||
// m_labels.append("Oranges"); | ||||
// m_points.append(QPointF(110, 50)); | ||||
// m_labels.append("Bananas"); | ||||
// m_points.append(QPointF(140, 40)); | ||||
// m_labels.append("Lemons"); | ||||
// m_points.append(QPointF(200, 150)); | ||||
// m_labels.append("Plums"); | ||||
// m_points.append(QPointF(225, 75)); | ||||
Marek Rosa
|
r630 | // m_labels.append("Pearls"); | ||
Marek Rosa
|
r734 | qsrand(QTime(0,0,0).secsTo(QTime::currentTime())); | ||
Marek Rosa
|
r630 | // m_data | ||
for (int i = 0; i < 6; i++) | ||||
{ | ||||
QVector<qreal>* dataVec = new QVector<qreal>(6); | ||||
Marek Rosa
|
r735 | QVector<QColor>* colorVec = new QVector<QColor>(6); | ||
Marek Rosa
|
r630 | for (int k = 0; k < dataVec->size(); k++) | ||
Marek Rosa
|
r735 | { | ||
Marek Rosa
|
r630 | if (k%2 == 0) | ||
dataVec->replace(k, i * 50 + qrand()%20); | ||||
else | ||||
dataVec->replace(k, qrand()%100); | ||||
Marek Rosa
|
r735 | colorVec->replace(k, QColor(Qt::white)); | ||
} | ||||
Marek Rosa
|
r630 | m_data.append(dataVec); | ||
m_labels.append(QString("Row: %1").arg((i + 1))); | ||||
Marek Rosa
|
r735 | m_rowsColors.append(colorVec); | ||
Marek Rosa
|
r630 | } | ||
Marek Rosa
|
r519 | } | ||
int CustomTableModel::rowCount(const QModelIndex & parent) const | ||||
{ | ||||
Tero Ahola
|
r613 | Q_UNUSED(parent) | ||
Marek Rosa
|
r630 | // return m_points.count(); | ||
return m_data.count(); | ||||
Marek Rosa
|
r519 | } | ||
int CustomTableModel::columnCount(const QModelIndex & parent) const | ||||
{ | ||||
Tero Ahola
|
r613 | Q_UNUSED(parent) | ||
Marek Rosa
|
r630 | // return 3; | ||
return 6; | ||||
Marek Rosa
|
r519 | } | ||
QVariant CustomTableModel::headerData (int section, Qt::Orientation orientation, int role ) const | ||||
{ | ||||
if (role != Qt::DisplayRole) | ||||
return QVariant(); | ||||
if (orientation == Qt::Horizontal) | ||||
{ | ||||
switch(section) | ||||
{ | ||||
Marek Rosa
|
r630 | // case 0: | ||
// return "x"; | ||||
// case 1: | ||||
// return "y"; | ||||
// case 2: | ||||
case 6: | ||||
Marek Rosa
|
r519 | return "Fruit"; | ||
Marek Rosa
|
r527 | default: | ||
Marek Rosa
|
r630 | if (section%2 == 0) | ||
return "x"; | ||||
else | ||||
return "y"; | ||||
// return "What?"; | ||||
Marek Rosa
|
r519 | } | ||
} | ||||
else | ||||
return QString("%1").arg(section + 1); | ||||
} | ||||
QVariant CustomTableModel::data(const QModelIndex & index, int role) const | ||||
{ | ||||
if (role == Qt::DisplayRole) | ||||
{ | ||||
switch(index.column()) | ||||
{ | ||||
Marek Rosa
|
r630 | // case 0: | ||
// return m_points[index.row()].x(); | ||||
// case 1: | ||||
// return m_points[index.row()].y(); | ||||
// case 2: | ||||
case 6: | ||||
Marek Rosa
|
r519 | return m_labels[index.row()]; | ||
default: | ||||
Marek Rosa
|
r630 | return m_data[index.row()]->at(index.column()); | ||
Tero Ahola
|
r613 | break; | ||
Marek Rosa
|
r519 | } | ||
} | ||||
else if (role == Qt::EditRole) | ||||
{ | ||||
switch(index.column()) | ||||
{ | ||||
Marek Rosa
|
r630 | // case 0: | ||
// return m_points[index.row()].x(); | ||||
// case 1: | ||||
// return m_points[index.row()].y(); | ||||
// case 2: | ||||
case 6: | ||||
Marek Rosa
|
r519 | return m_labels[index.row()]; | ||
default: | ||||
Marek Rosa
|
r630 | return m_data[index.row()]->at(index.column()); | ||
Tero Ahola
|
r613 | break; | ||
Marek Rosa
|
r519 | } | ||
} | ||||
Marek Rosa
|
r735 | else if (role == Qt::BackgroundRole) | ||
{ | ||||
return m_rowsColors[index.row()]->at(index.column()); | ||||
} | ||||
Tero Ahola
|
r613 | return QVariant(); | ||
Marek Rosa
|
r519 | } | ||
bool CustomTableModel::setData ( const QModelIndex & index, const QVariant & value, int role) | ||||
{ | ||||
if (index.isValid() && role == Qt::EditRole) | ||||
{ | ||||
switch(index.column()) | ||||
{ | ||||
Marek Rosa
|
r630 | // case 0: | ||
// m_points[index.row()].setX(value.toDouble()); | ||||
// break; | ||||
// case 1: | ||||
// m_points[index.row()].setY(value.toDouble()); | ||||
// break; | ||||
// case 2: | ||||
case 6: | ||||
Marek Rosa
|
r519 | m_labels.replace(index.row(), value.toString()); | ||
break; | ||||
default: | ||||
Marek Rosa
|
r630 | m_data[index.row()]->replace(index.column(), value.toDouble()); | ||
break; | ||||
// return false; | ||||
Marek Rosa
|
r519 | } | ||
emit dataChanged(index, index); | ||||
return true; | ||||
} | ||||
Marek Rosa
|
r735 | else if (role == Qt::BackgroundRole) | ||
{ | ||||
m_rowsColors[index.row()]->replace(index.column(), value.value<QColor>()); | ||||
return true; | ||||
} | ||||
Marek Rosa
|
r519 | return false; | ||
} | ||||
Qt::ItemFlags CustomTableModel::flags ( const QModelIndex & index ) const | ||||
{ | ||||
Marek Rosa
|
r630 | // if (!index.isValid()) | ||
// return Qt::ItemIsEnabled; | ||||
return QAbstractItemModel::flags(index) | Qt::ItemIsEditable; | ||||
Marek Rosa
|
r519 | } | ||
bool CustomTableModel::insertRows ( int row, int count, const QModelIndex & parent) | ||||
{ | ||||
Tero Ahola
|
r613 | Q_UNUSED(parent) | ||
Marek Rosa
|
r546 | if (row < 0) | ||
row = 0; | ||||
Marek Rosa
|
r519 | beginInsertRows(QModelIndex(), row /*dataTable.count()*/, row + count - 1); | ||
Marek Rosa
|
r545 | for (int i = row; i < row + count; i++) | ||
Marek Rosa
|
r519 | { | ||
Marek Rosa
|
r630 | // m_points.insert(row, QPointF(10,20)); | ||
Marek Rosa
|
r735 | QVector<qreal>* dataVec = new QVector<qreal>(6); | ||
QVector<QColor>* colorVec = new QVector<QColor>(6); | ||||
Marek Rosa
|
r630 | for (int k = 0; k < dataVec->size(); k++) | ||
Marek Rosa
|
r735 | { | ||
Marek Rosa
|
r630 | if (k%2 == 0) | ||
// dataVec->replace(k, i * 50 + qrand()%20); | ||||
{ | ||||
int difference = 0; | ||||
Marek Rosa
|
r735 | if (i < m_data.size()) | ||
Marek Rosa
|
r630 | { | ||
Marek Rosa
|
r735 | if (i - 1 >= 0) | ||
Marek Rosa
|
r630 | { | ||
Marek Rosa
|
r735 | difference = (int)((qAbs(m_data[i]->at(k) - m_data[row - 1]->at(k)))/count); | ||
dataVec->replace(k, m_data[i - 1]->at(k) + qrand()%qMax(1, difference)); | ||||
Marek Rosa
|
r630 | } | ||
else | ||||
Marek Rosa
|
r734 | dataVec->replace(k, qrand()%40 + 10); | ||
Marek Rosa
|
r630 | } | ||
else | ||||
Marek Rosa
|
r735 | { | ||
if (i - 1 >= 0) | ||||
Marek Rosa
|
r734 | { | ||
Marek Rosa
|
r735 | dataVec->replace(k, m_data[i - 1]->at(k) + qrand()%40 + 10); | ||
Marek Rosa
|
r734 | } | ||
else | ||||
{ | ||||
dataVec->replace(k, qrand()%40 + 10); | ||||
} | ||||
Marek Rosa
|
r735 | } | ||
Marek Rosa
|
r630 | } | ||
else | ||||
dataVec->replace(k, qrand()%100); | ||||
Marek Rosa
|
r735 | colorVec->replace(k, QColor(Qt::white)); | ||
} | ||||
m_data.insert(i, dataVec); | ||||
m_labels.insert(i,(QString("Row: %1").arg(i + 1))); | ||||
m_rowsColors.insert(i, colorVec); | ||||
Marek Rosa
|
r519 | } | ||
endInsertRows(); | ||||
return true; | ||||
} | ||||
Marek Rosa
|
r545 | |||
bool CustomTableModel::removeRows ( int row, int count, const QModelIndex & parent) | ||||
{ | ||||
if (row > this->rowCount() - 1) | ||||
return false; | ||||
if (row < 0) | ||||
row = 0; | ||||
if (row + count > rowCount()) | ||||
return false; | ||||
beginRemoveRows(parent, row, row + count - 1); | ||||
for (int i = row; i < row + count; i++) | ||||
{ | ||||
Marek Rosa
|
r734 | // m_points.removeAt(row); | ||
Marek Rosa
|
r630 | QVector<qreal>* item = m_data.at(row); | ||
m_data.removeAt(row); | ||||
delete item; | ||||
Marek Rosa
|
r545 | m_labels.removeAt(row); | ||
} | ||||
endRemoveRows(); | ||||
return true; | ||||
} | ||||