#include "customtablemodel.h"

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

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

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

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:
            return "Fruit";
        default:
            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:
            return m_labels[index.row()];
        default:
            return QVariant();
        }
    }
    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:
            return m_labels[index.row()];
        default:
            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:
            m_labels.replace(index.row(), value.toString());
            break;
        default:
            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)
{
    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());
        m_labels.insert(row,(""));
    }
    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);
        m_labels.removeAt(row);
    }
    endRemoveRows();
    return true;
}