#include <QColor>
#include "mytablemodel.h"


MyTableModel::MyTableModel(QObject* parent)
    :QAbstractTableModel(parent)
{
}

int MyTableModel::rowCount(const QModelIndex & /*parent*/) const
{
    return m_modelData.size();
}

int MyTableModel::columnCount(const QModelIndex & /*parent*/) const
{
    return AppData::getInstance()->lstDataTableHeaderText.count();
}

QVariant MyTableModel::data(const QModelIndex& index, int role) const
{
    if (!index.isValid())
        return QVariant();

    if (role == Qt::DisplayRole || role == Qt::EditRole)
    {
        const int row = index.row();
        switch (index.column())
        {
        case 0:
        {
            if (m_modelData.at(row).status == STATUS_NORMAL) return tr("Normal");
            if (m_modelData.at(row).status == STATUS_INFO) return tr("INFO");
            if (m_modelData.at(row).status == STATUS_WARN) return tr("WARN");
            if (m_modelData.at(row).status == STATUS_ERROR) return tr("ERROR");
        }
        case 1:
        {
            return m_modelData.at(row).ParameterName.c_str();
        }
        case 2:
            return m_modelData.at(row).value.c_str();
        case 3:
            return m_modelData.at(row).unit.c_str();
        case 4:
            return m_modelData.at(row).sampleTime.c_str();
        }
    }

    //对齐处理
    if (role == Qt::TextAlignmentRole)
    {
        //const int row = index.row();
        switch (index.column())
        {
        case 1:
        case 2:
        case 3:
            return QVariant(Qt::AlignLeft|Qt::AlignVCenter);
            break;
        //case 4:
        //    return QVariant(Qt::AlignRight|Qt::AlignVCenter);
        //    break;
        default:
            return Qt::AlignCenter;
            break;
        }
    }

    if (role == Qt::BackgroundRole)
    {
        const int row = index.row();
        switch (index.column())
        {
        case 0: //状态
        case 2: //数值
        {
            if (m_modelData.at(row).status == STATUS_WARN)
                return QVariant(QColor(Qt::yellow));
            else if (m_modelData.at(row).status == STATUS_ERROR)
                return QVariant(QColor(Qt::red));
        }
        }
    }

    return QVariant();
}

QVariant MyTableModel::headerData(int section, Qt::Orientation orientation, int role) const
{
    if (role == Qt::DisplayRole && orientation == Qt::Horizontal)
    {
        if (section < m_headeList.size())
        {
            return m_headeList[section];
        }
    }
    return QAbstractItemModel::headerData(section, orientation, role);
}

const TableData *MyTableModel::getModelData() const
{
    return &m_modelData;
}

bool MyTableModel::setData(const QModelIndex& index, const QVariant& value, int role)
{
    //beginResetModel();
    if (!index.isValid())
        return false;

    //endResetModel();  //在结束前添加此函数
    return QAbstractTableModel::setData(index, value, role);

}

Qt::ItemFlags MyTableModel::flags(const QModelIndex &index) const
{
    return Qt::ItemIsEditable | QAbstractTableModel::flags(index);
}

void MyTableModel::setModelData(TableData data)
{
    //修改数据前后，用beginResetModel()和endResetModel()
    beginResetModel();    //通过这个告诉我要开始修改model了
    m_modelData.clear();

    for (int i = 0; i < data.count(); ++i)
    {
        m_modelData.append(data[i]);
    }
    endResetModel();    //通过这个告诉我修改model结束了

}

void MyTableModel::clearModelData()
{
    m_modelData.clear();
}

void MyTableModel::setHeadData(QStringList i_list)
{
    m_headeList=i_list;
}