/****************************************************************************
**
** 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$
**
****************************************************************************/

#include "customtablemodel.h"
#include <QVector>
#include <QTime>

CustomTableModel::CustomTableModel(QObject *parent) :
    QAbstractTableModel(parent)
{
    //    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));
    //    m_labels.append("Pearls");

    qsrand(QTime(0,0,0).secsTo(QTime::currentTime()));

    // m_data
    for (int i = 0; i < 6; i++)
    {
        QVector<qreal>* dataVec = new QVector<qreal>(6);
        QVector<QColor>* colorVec = new QVector<QColor>(6);
        for (int k = 0; k < dataVec->size(); k++)
        {
            if (k%2 == 0)
                dataVec->replace(k, i * 50 + qrand()%20);
            else
                dataVec->replace(k, qrand()%100);
            colorVec->replace(k, QColor(Qt::white));
        }
        m_data.append(dataVec);
        m_labels.append(QString("Row: %1").arg((i + 1)));
        m_rowsColors.append(colorVec);
    }
}

int CustomTableModel::rowCount(const QModelIndex & parent) const
{
    Q_UNUSED(parent)
    //    return m_points.count();
    return m_data.count();
}

int CustomTableModel::columnCount(const QModelIndex & parent) const
{
    Q_UNUSED(parent)
    //    return 3;
    return 6;
}

QVariant CustomTableModel::headerData (int section, Qt::Orientation orientation, int role ) const
{
    if (role != Qt::DisplayRole)
        return QVariant();

    if (orientation == Qt::Horizontal)
    {
        switch(section)
        {
        //        case 0:
        //            return "x";
        //        case 1:
        //            return "y";
        //        case 2:
        case 6:
            return "Fruit";
        default:
            if (section%2 == 0)
                return "x";
            else
                return "y";
            //            return "What?";
        }
    }
    else
        return QString("%1").arg(section + 1);
}

QVariant CustomTableModel::data(const QModelIndex & index, int role) const
{
    if (role == Qt::DisplayRole)
    {
        switch(index.column())
        {
        //        case 0:
        //            return m_points[index.row()].x();
        //        case 1:
        //            return m_points[index.row()].y();
        //        case 2:
        case 6:
            return m_labels[index.row()];
        default:
            return m_data[index.row()]->at(index.column());
            break;
        }
    }
    else if (role == Qt::EditRole)
    {
        switch(index.column())
        {
        //        case 0:
        //            return m_points[index.row()].x();
        //        case 1:
        //            return m_points[index.row()].y();
        //        case 2:
        case 6:
            return m_labels[index.row()];
        default:
            return m_data[index.row()]->at(index.column());
            break;
        }
    }
    else if (role == Qt::BackgroundRole)
        {
            return m_rowsColors[index.row()]->at(index.column());
        }
    return QVariant();
}

bool CustomTableModel::setData ( const QModelIndex & index, const QVariant & value, int role)
{
    if (index.isValid() && role == Qt::EditRole)
    {
        switch(index.column())
        {
        //        case 0:
        //            m_points[index.row()].setX(value.toDouble());
        //            break;
        //        case 1:
        //            m_points[index.row()].setY(value.toDouble());
        //            break;
        //        case 2:
        case 6:
            m_labels.replace(index.row(), value.toString());
            break;
        default:
            m_data[index.row()]->replace(index.column(), value.toDouble());
            break;
            //            return false;
        }
        emit dataChanged(index, index);
        return true;
    }
    else if (role == Qt::BackgroundRole)
        {
            m_rowsColors[index.row()]->replace(index.column(), value.value<QColor>());
            return true;
        }
    return false;
}

Qt::ItemFlags CustomTableModel::flags ( const QModelIndex & index ) const
{
    //    if (!index.isValid())
    //                return Qt::ItemIsEnabled;
    return QAbstractItemModel::flags(index) | Qt::ItemIsEditable;
}

bool CustomTableModel::insertRows ( int row, int count, const QModelIndex & parent)
{
    Q_UNUSED(parent)

    if (row < 0)
        row = 0;
    beginInsertRows(QModelIndex(), row /*dataTable.count()*/, row + count - 1);
    for (int i = row; i < row + count; i++)
    {
        //        m_points.insert(row, QPointF(10,20));
        QVector<qreal>* dataVec = new QVector<qreal>(6);        
        QVector<QColor>* colorVec = new QVector<QColor>(6);
        for (int k = 0; k < dataVec->size(); k++)
        {
            if (k%2 == 0)
                //                dataVec->replace(k, i * 50 + qrand()%20);
            {
                int difference = 0;
                if (i < m_data.size())
                {
                    if (i - 1 >= 0)
                    {
                        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));
                    }
                    else
                        dataVec->replace(k, qrand()%40 + 10);
                }
                else
                {
                    if (i - 1 >= 0)
                    {
                        dataVec->replace(k, m_data[i - 1]->at(k) + qrand()%40 + 10);
                    }
                    else
                    {
                        dataVec->replace(k, qrand()%40 + 10);
                    }
                }
            }
            else
                dataVec->replace(k, qrand()%100);
            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);
    }
    endInsertRows();
    return true;
}

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++)
    {
        //        m_points.removeAt(row);
        QVector<qreal>* item = m_data.at(row);
        m_data.removeAt(row);
        delete item;
        m_labels.removeAt(row);
    }
    endRemoveRows();
    return true;
}