#include "DataCollectionWidget.h"

#include "DataTableWidget.h"
#include "ConfigExpertWidget.h"

#include "dbService/SchemeProcessService.h"
#include "dbService/ClassSet.h"
#include "dbService/CNodeDataService.h"

#include <CNode.h>

#include <QTabWidget>
#include <QBoxLayout>
#include <QMap>

#include <QDebug>

DataCollectionWidget::DataCollectionWidget(ProjectInfo *proj, QWidget *parent) : EvalWidget(proj, parent)
{
    setTitle("评估数据采集");
    m_configExpert = new ConfigExpertWidget(this);
    m_contentLayout->addWidget(m_configExpert);
}

void DataCollectionWidget::setType(int type)
{
    EvalWidget::setType(type);
    setupTabWidget();
}

void DataCollectionWidget::setupTabWidget()
{
    m_tab->clear();

    QMap<int, QList<CNodeData>> nodeListMap;
    for (int i : indexList()) {
        // 获取指标体系数据
        QList<CNodeData> nodeList;
        bool ret = CNodeDataService().QueryAll(nodeList, m_proj->id, i);
        if (ret) {
            nodeListMap[i] = nodeList;
        } else {
            return;
        }

        // 获取方案规划数据
        QList<SchemePlanManager::SchemeProcessInfo> processList;
        ret = SchemeProcessService().QueryAllByProjectIdAndIndexType(processList, m_proj->id, i);
        if (ret == false) {
            return;
        }

        for (SchemePlanManager::SchemeProcessInfo process : processList) {
            if (process.dSource >= 0) {
                DataTableWidget *m = new DataTableWidget(process, this);
                m->mind1()->setNodeList(nodeListMap[i]);
                if (i == ProjectManager::TechIndex) {
                    m->mind2()->setNodeList(nodeListMap[ProjectManager::AbilityIndex]);
                }
                m->setCurrentPage(1);

                ProjectManager::IndexType t = (ProjectManager::IndexType)i;
                QString indexName           = ProjectManager::nameOfIndexType(t);
                QString processName         = SchemePlanManager::processName(process);
                m_tab->addTab(m, indexName + " - " + processName);
            }
        }
    }
}