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

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");

    // m_data
    for (int i = 0; i < 6; i++)
    {
        QVector<qreal>* dataVec = new QVector<qreal>(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);
        m_data.append(dataVec);
        m_labels.append(QString("Row: %1").arg((i + 1)));
    }
}

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;
        }
    }
    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;
    }
    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);
        for (int k = 0; k < dataVec->size(); k++)
            if (k%2 == 0)
                //                dataVec->replace(k, i * 50 + qrand()%20);
            {
                int difference = 0;
                if (row < m_data.size())
                {
                    if (row - 1 >= 0)
                    {
                        difference = (int)(qAbs(m_data[row]->at(k) - m_data[row - 1]->at(k)));
                        dataVec->replace(k, m_data[row - 1]->at(k) + qrand()%qMax(1, difference));
                    }
                    else
                        dataVec->replace(k, qrand()%20);
                }
                else
                    dataVec->replace(k, m_data[row - 1]->at(k) + qrand()%20);
            }
            else
                dataVec->replace(k, qrand()%100);
        m_data.insert(row, dataVec);
        m_labels.insert(row,(QString("Row: %1").arg(row + 1)));
    }
    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;
}