#include "EXDataTableView.h"

#include "EXDataViewDelegate.h"

#include "ProjectManager.h"
#include <dbService/CNodeDataService.h>
#include <dbService/ClassSet.h>

#include <Widgets/Button.h>
#include <QBoxLayout>

#include <QTabWidget>
#include <QLabel>
#include <QTableView>
#include <QTableWidget>
#include <QAbstractItemModel>
#include <QStandardItemModel>
#include <QStandardItem>
#include <QHeaderView>
#include <QApplication>
#include <QDesktopWidget>

#include <QDebug>

EXDataTableView::EXDataTableView(SchemePlanManager::SchemeProcessInfo process, QWidget *parent)
    : QWidget(parent), m_process(process)
{
    m_mind1 = new CMind(this);
    m_mind2 = new CMind(this);
    initWidget();
    initLayout();
    connectSignalsAndSlots();

    m_comboDelegate = new EXDataTableComboDelegate(this);
}

SchemePlanManager::SchemeProcessInfo EXDataTableView::process() const
{
    return m_process;
}

void EXDataTableView::initWidget()
{
    m_dataTab = new QTabWidget(this);
    m_dataTab->setTabPosition(QTabWidget::South);
    m_pageLab  = new QLabel(this);
    m_previous = new PushButton("上一级指标", this);
    m_next     = new PushButton("下一级指标", this);
}

void EXDataTableView::initLayout()
{
    m_layout = new QVBoxLayout(this);
    m_layout->addWidget(m_dataTab);
    m_pageLayout = new QHBoxLayout();
    m_layout->addLayout(m_pageLayout);

    m_pageLayout->setSpacing(10);
    m_pageLayout->addStretch();
    m_pageLayout->addWidget(m_previous);
    m_pageLayout->addWidget(m_pageLab);
    m_pageLayout->addWidget(m_next);
    m_pageLayout->addStretch();
}

void EXDataTableView::connectSignalsAndSlots()
{
    connect(m_previous, &PushButton::clicked, this, &EXDataTableView::slotPrevious);
    connect(m_next, &PushButton::clicked, this, &EXDataTableView::slotNext);
    connect(m_dataTab, &QTabWidget::currentChanged, this, &EXDataTableView::slotTabCurrentChanged);
}

void EXDataTableView::setupTabWidget()
{
    /// 创建 tableView 并添加进 tabWidget
    /// 这个过程中会触发 tabWidget 的 currentChanged,
    /// 所以使用 m_isSettingTable 标记此过程, 以采取必要措施来规避一些异常操作
    m_isSettingTable = true;
    m_dataTab->clear();
    for (CNodeData n : m_mind1->nodesInLevel(m_currentPage)) {
        QTableView *t = new QTableView(m_dataTab);
        t->setAlternatingRowColors(m_mind2->nodeList().count() > 0);
        t->horizontalHeader()->setSectionResizeMode(QHeaderView::Stretch);
        t->horizontalHeader()->setStyleSheet("QHeaderView::section{background:rgb(244,244,244);color: black;}");
        t->verticalHeader()->setStyleSheet("QHeaderView::section{background:rgb(244,244,244);color: black;}");
        t->verticalHeader()->setDefaultAlignment(Qt::AlignCenter);
        t->setSelectionMode(QAbstractItemView::SingleSelection);

        m_dataTab->addTab(t, n.name);

        QStandardItemModel *model = new QStandardItemModel(t);
        t->setModel(model);
        connect(t, &QTableView::clicked, this, &EXDataTableView::itemClicked);
    }
    m_isSettingTable = false;
}

int EXDataTableView::currentPage() const
{
    return m_currentPage;
}

void EXDataTableView::setCurrentPage(int p)
{
    if (p < 1 || p >= m_mind1->levels()) {
        return;
    }
    m_currentPage = p;
    m_previous->setEnabled(p > 1);
    m_next->setEnabled(p < m_mind1->levels() - 1);
    setupTabWidget();

    m_pageLab->setText(QString("共 %1 页, 当前第 %2 页").arg(m_mind1->levels() - 1).arg(p));
    updateCurrentTable();
}

void EXDataTableView::updateCurrentTable()
{
    int c             = m_dataTab->currentIndex();
    QTableView *table = (QTableView *)m_dataTab->widget(c);

    if (table == nullptr || table->model() == nullptr || m_isSettingTable) {
        return;
    }

    m_hNodes.clear();
    m_vNodes.clear();
    m_values.clear();

    QStandardItemModel *model = (QStandardItemModel *)table->model();

    /// 设置顶部水平方向标题
    int hIndex         = -1;
    int dimensionIndex = -1;  // 量纲所在列
    int typeIndex      = -1;  // 指标类型所在列

    // 指标
    CNodeData n            = m_mind1->nodesInLevel(m_currentPage)[c];
    QList<CNodeData> hList = m_mind1->subNodes(n);
    // 以下情况需要显示指标
    // 导入权重分析数据
    // 导入需求分析评估的评估数据
    if (m_process.type == SchemePlanManager::ImportWeightData || m_process.indexType == ProjectManager::TechIndex) {
        for (CNodeData node : hList) {
            QStandardItem *item = new QStandardItem(node.name);
            item->setToolTip(node.remark);
            model->setHorizontalHeaderItem(++hIndex, item);
            m_hNodes.append(node.name);
        }
    }
    // 导入方案优选评估或效能评估的评估数据时, 需要显示量纲
    if (m_process.type == SchemePlanManager::ImportEvalData
        && (m_process.indexType == ProjectManager::OptimalIndex
            || m_process.indexType == ProjectManager::EfficiencyIndex)) {
        QStandardItem *item = new QStandardItem("指标量纲");
        model->setHorizontalHeaderItem(++hIndex, item);
        dimensionIndex = hIndex;
    }
    // 导入方案优选评估的评估数据时, 需要显示指标类型
    if (m_process.type == SchemePlanManager::ImportEvalData && m_process.indexType == ProjectManager::OptimalIndex) {
        QStandardItem *item = new QStandardItem("指标类型");
        model->setHorizontalHeaderItem(++hIndex, item);
        typeIndex = hIndex;
    }

    /// 设置左侧垂直方向标题
    QList<CNodeData> vList;
    // 导入权重分析的专家数据时, 显示指标
    if (m_process.type == SchemePlanManager::ImportWeightData) {
        vList = hList;
    }
    // 导入评估数据时, 显示最后一级指标
    if (m_process.type == SchemePlanManager::ImportEvalData) {
        if (m_process.indexType == ProjectManager::TechIndex) {
            vList = m_mind2->leaves();
        } else {
            vList = m_mind1->leaves();
        }
    }
    for (int i = 0; i < vList.count(); i++) {
        CNodeData node      = vList[i];
        QStandardItem *item = new QStandardItem(QString("  %1  ").arg(node.name));
        item->setToolTip(node.remark);
        model->setVerticalHeaderItem(i, item);
        table->setRowHeight(i, 35);
        m_vNodes.append(node.name);
    }

    /// 填充量纲和指标类型
    for (int i = 0; i < vList.count(); i++) {
        CNodeData vNode  = vList[i];
        QStandardItem *d = new QStandardItem();
        d->setData(Qt::AlignCenter, Qt::TextAlignmentRole);  // 单元格文字居中
        if (dimensionIndex >= 0) {
            d->setText(vNode.dimension);
            model->setItem(i, dimensionIndex, d);
        }

        if (typeIndex >= 0) {
            model->setItem(i, typeIndex, d);
        }
        table->setRowHeight(i, 35);
    }

    if (m_process.type == SchemePlanManager::ImportWeightData) {
        for (int i = 0; i < vList.count(); i++) {
            CNodeData vNode = vList[i];
            for (int j = 0; j < hList.count(); j++) {
                CNodeData hNode     = hList[i];
                QStandardItem *item = new QStandardItem();
                item->setData(Qt::AlignCenter, Qt::TextAlignmentRole);
                item->setEditable(false);
                if (i == j) {
                    item->setText("1");
                }
                if (i >= j) {
                    item->setBackground(QBrush(QColor("lightgray")));
                }
                model->setItem(i, j, item);
            }
        }
    }
    if (m_process.type == SchemePlanManager::ImportEvalData || m_process.indexType == ProjectManager::TechIndex) {
        for (int i = 0; i < vList.count(); i++) {
            for (int j = 0; j < hList.count(); j++) {
                QStandardItem *item = new QStandardItem();
                item->setData(Qt::AlignCenter, Qt::TextAlignmentRole);
                //                item->setEditable(false);
                model->setItem(i, j, item);
                table->setItemDelegate(m_comboDelegate);
            }
        }
    }
}

CMind *EXDataTableView::mind1() const
{
    return m_mind1;
}

CMind *EXDataTableView::mind2() const
{
    return m_mind2;
}

void EXDataTableView::editItemData(const QModelIndex &index, const QString &val)
{
    int c                     = m_dataTab->currentIndex();
    QTableView *table         = (QTableView *)m_dataTab->widget(c);
    QStandardItemModel *model = (QStandardItemModel *)table->model();
    model->itemFromIndex(index)->setText(val);
    QString symmetry;
    if (val.startsWith("1/")) {
        symmetry = val.split("/")[1];
    } else {
        if (val == "0" || val == "1") {
            symmetry = val;
        } else {
            symmetry = "1/" + val;
        }
    }
    model->item(index.column(), index.row())->setText(symmetry);
}

void EXDataTableView::slotPrevious()
{
    setCurrentPage(m_currentPage - 1);
}

void EXDataTableView::slotNext()
{
    setCurrentPage(m_currentPage + 1);
}

void EXDataTableView::slotTabCurrentChanged(int c)
{
    Q_UNUSED(c)
    updateCurrentTable();
}

void EXDataTableView::itemClicked(const QModelIndex &index)
{
    if (index.row() >= index.column()) {
        return;
    }

    if (m_process.type != SchemePlanManager::ImportWeightData) {
        return;
    }

    QStringList l = { "1/9", "1/7", "1/5", "1/3", "1", "3", "5", "7", "9" };

    QTableView *table         = (QTableView *)sender();
    QStandardItemModel *model = (QStandardItemModel *)table->model();
    //    SchemeBar *scheme =
    //            new SchemeBar(model->item(index.row(), 0)->text(),
    //            model->horizontalHeaderItem(index.column())->text(), l);
    SchemeBar *scheme = new SchemeBar(model->verticalHeaderItem(index.row())->text(),
                                      model->horizontalHeaderItem(index.column())->text(), l);
    scheme->setModal(true);
    scheme->setAttribute(Qt::WA_DeleteOnClose);

    connect(scheme, &SchemeBar::setValue, [=](QString val) { editItemData(index, val); });

    scheme->show();
    QPoint p = QCursor::pos();
    if (p.x() + scheme->width() + 10 >= QApplication::desktop()->width()) {
        p.setX(QApplication::desktop()->width() - 10 - scheme->width());
    }
    scheme->move(p);
}