#include "dpBarsChart.h"
#include "dataManager.h"

dpBarsChart::dpBarsChart(QWidget* parent)
    :dpBaseChart(parent)
{
    setAttribute(Qt::WA_TranslucentBackground,true);

    m_pCustomPlot = new QCustomPlot(this);
    /* m_pBarsGroup = new QCPBarsGroup(m_pCustomPlot);
    m_pBarsGroup->setSpacingType(QCPBarsGroup::stAbsolute);
    m_pBarsGroup->setSpacing(2); */
    initQCP();

    QBoxLayout* mainLayout = new QBoxLayout(QBoxLayout::LeftToRight);
    mainLayout->setContentsMargins(0, 1, 0, 0);
    mainLayout->addWidget(m_pCustomPlot);
    setLayout(mainLayout);

    connect(DataManager::instance(), &DataManager::dataUpdated, this, &dpBarsChart::onSignal_dataUpdated);
}

dpBarsChart::~dpBarsChart()
{
}

void dpBarsChart::initQCP()
{
    m_chartStyle.bgColor = Qt::transparent;
    m_chartStyle.axisColor = QColor(87, 100, 120);
    m_chartStyle.labelColor = QColor(250, 250, 250);
    m_chartStyle.labelFont = QFont("黑体", 12);
    m_chartStyle.tickColor = QColor(87, 100, 120);
    m_chartStyle.tickLabelColor = QColor(250, 250, 250);
    m_chartStyle.tickLabelFont = QFont("黑体", 12);
    m_chartStyle.gridPen = QPen(QColor(87, 100, 120), 1, Qt::DotLine);

    m_pCustomPlot->setInteractions(QCP::iNone);
    /*m_pCustomPlot->axisRect()->setRangeDrag(Qt::Horizontal); //只允许x轴方向的拖拽
    double zoomFactor = m_pCustomPlot->axisRect()->rangeZoomFactor(Qt::Horizontal);
    m_pCustomPlot->axisRect()->setRangeZoomFactor(zoomFactor, 1); //只做x轴的缩放*/
    //m_pCustomPlot->xAxis->setTicks(false);
    m_pCustomPlot->xAxis->setSubTicks(false);
    //m_pCustomPlot->yAxis->setTicks(false);
    m_pCustomPlot->yAxis->setSubTicks(false);
    //m_pCustomPlot->yAxis->setVisible(false);
    //m_pCustomPlot->yAxis2->setTicks(false);
    m_pCustomPlot->yAxis2->setSubTicks(false);
    m_pCustomPlot->yAxis2->setVisible(false);
    //背景颜色
    m_pCustomPlot->setBackground(QBrush(m_chartStyle.bgColor));
    //坐标轴颜色
    m_pCustomPlot->xAxis->setBasePen(m_chartStyle.axisColor);
    m_pCustomPlot->xAxis2->setBasePen(m_chartStyle.axisColor);
    m_pCustomPlot->yAxis->setBasePen(m_chartStyle.axisColor);
    m_pCustomPlot->yAxis2->setBasePen(m_chartStyle.axisColor);
    //坐标刻度颜色
    m_pCustomPlot->xAxis->setTickPen(QPen(m_chartStyle.tickColor));
    m_pCustomPlot->yAxis->setTickPen(QPen(m_chartStyle.tickColor));
    m_pCustomPlot->yAxis2->setTickPen(QPen(m_chartStyle.tickColor));
    //坐标刻度Label颜色
    m_pCustomPlot->xAxis->setTickLabels(false);
    m_pCustomPlot->xAxis->setTickLabelColor(m_chartStyle.tickLabelColor);
    m_pCustomPlot->xAxis->setTickLabelFont(m_chartStyle.tickLabelFont);
    m_pCustomPlot->yAxis->setTickLabels(false);
    m_pCustomPlot->yAxis->setTickLabelColor(m_chartStyle.tickLabelColor);
    m_pCustomPlot->yAxis->setTickLabelFont(m_chartStyle.tickLabelFont);
    m_pCustomPlot->yAxis2->setTickLabelColor(m_chartStyle.tickLabelColor);
    m_pCustomPlot->yAxis2->setTickLabelFont(m_chartStyle.tickLabelFont);
    //网格线颜色
    m_pCustomPlot->xAxis->grid()->setVisible(false);
    m_pCustomPlot->xAxis->grid()->setPen(m_chartStyle.gridPen);
    m_pCustomPlot->xAxis->grid()->setZeroLinePen(m_chartStyle.gridPen);
    //m_pCustomPlot->xAxis2->grid()->setPen(m_chartStyle.gridPen);
    m_pCustomPlot->yAxis->grid()->setVisible(false);
    m_pCustomPlot->yAxis->grid()->setPen(m_chartStyle.gridPen);
    m_pCustomPlot->yAxis->grid()->setZeroLinePen(m_chartStyle.gridPen);
    m_pCustomPlot->yAxis2->grid()->setPen(m_chartStyle.gridPen);

    //Legend
    m_pCustomPlot->legend->setBrush(QBrush(QColor(255,255,255,12))); //背景透明
    m_pCustomPlot->legend->setBorderPen(QPen(QColor(255,255,255,0))); //边框透明
    m_pCustomPlot->legend->setFont(m_chartStyle.labelFont);
    m_pCustomPlot->legend->setTextColor( m_chartStyle.labelColor);
}

void dpBarsChart::setTimeRange(TimeUnit unit)
{}

void dpBarsChart::setDateTime(const QDateTime& dateTime)
{}

void dpBarsChart::viewHistoricalData(const QDateTime& dateTime)
{}

void dpBarsChart::synchronizeConfigData(const configurationResults& cfg)
{
    //先清除现有图形(直方图不像曲线一样有连续的过程展现，所以不采用动态更新，每次都清除、重新创建)
    //m_pBarsGroup->clear();
    m_pCustomPlot->clearPlottables();
    m_bars.clear();
    m_barsGroups.clear();

    bool groupByType = cfg.groupByType;

    //设定朝向
    QCPAxis* keyAxis = nullptr;
    QCPAxis* valueAxis = nullptr;
    if(cfg.barOrientataion == Vertical)
    {
        keyAxis = m_pCustomPlot->xAxis;
        valueAxis = m_pCustomPlot->yAxis;
    }
    else
    {
        keyAxis = m_pCustomPlot->yAxis;
        valueAxis = m_pCustomPlot->xAxis;
    }
    if(!keyAxis || !valueAxis)
        return;

    QVector<double> ticks;
    int tick = 0;
    QVector<QString> tickLabels;

    QHash<RealTimeDataType, QSet<QString>> groupedBars; //按数据类型对bar进行分组，以实现可以按组放置以及设置颜色等操作
    for(int i = 0; i < cfg.m_pModel_dataSource->rowCount(); i++)
    {
        QString itemText = cfg.m_pModel_dataSource->item(i, 0)->text();
        QString stationID = cfg.m_pModel_dataSource->item(i, 0)->data(Qt::UserRole + itemRole_stationID).toString();
        QString compoentID = cfg.m_pModel_dataSource->item(i, 0)->data(Qt::UserRole + itemRole_componentID).toString();
        QString pointID = cfg.m_pModel_dataSource->item(i, 0)->data(Qt::UserRole + itemRole_pointID).toString();
        QString dataID = stationID + "-" + compoentID + "-" + pointID;
        QVariant colorData = cfg.m_pModel_dataSource->item(i, 0)->data(Qt::DecorationRole);
        RealTimeDataType dataType = (RealTimeDataType)cfg.m_pModel_dataSource->item(i, 0)->data(Qt::UserRole + itemRole_dataType).toInt();

        Bars bars;
        bars.name = itemText;
        bars.dataID = dataID;
        bars.dataType = dataType;
        bars.color = colorData.value<QColor>();
        QCPBars* qBars = new QCPBars(keyAxis, valueAxis);
        qBars->setName(bars.name);
        qBars->setPen(bars.color); //边框颜色
        qBars->setBrush(bars.color); //填充颜色
        bars.qBars = qBars;

        if(groupByType)
        {
            groupedBars[dataType].insert(dataID);
            //每一个类型创建一个QCPBarsGroup对象
            QCPBarsGroup* barsGroup = nullptr;
            if(!m_barsGroups.contains(dataType))
            {
                barsGroup = new QCPBarsGroup(m_pCustomPlot);
                m_barsGroups.insert(dataType, barsGroup);
            }
        }
        else
        {
            ticks << ++tick;
            tickLabels << itemText;
            bars.keys << tick;
        }

        m_bars.insert(dataID, bars);
    }

    if(groupByType)
    {
        for(auto it = groupedBars.begin(); it != groupedBars.end(); ++it)
        {
            ticks.push_back(++tick);
            switch(it.key())
            {
            case power:
                tickLabels.push_back("功率");
                break;
            case voltage:
                tickLabels.push_back("电压");
                break;
            case current:
                tickLabels.push_back("电流");
                break;
            case temperature:
                tickLabels.push_back("温度");
                break;
            default:
                tickLabels.push_back("未定义");
                break;
            }

            for(auto bars_it = it.value().begin(); bars_it != it.value().end(); ++bars_it)
            {
                QString dataID = *bars_it;
                if(m_bars.contains(dataID))
                    m_bars[dataID].keys.push_back(tick);
            }
        }
    }

    //Axis
    if(groupByType)
        keyAxis->setTickLabels(true);
    else
        keyAxis->setTickLabels(false);

    QSharedPointer<QCPAxisTickerText> textTicker(new QCPAxisTickerText);
    textTicker->addTicks(ticks, tickLabels);
    keyAxis->setTicker(textTicker);
    keyAxis->setRange(0, ticks.count() + 1);

    valueAxis->setTickLabels(true);
    valueAxis->setRange(0, 11);

    for(auto it = m_bars.begin(); it != m_bars.end(); ++it)
    {
        if(it.value().qBars.isNull())
            continue;

        //随机模拟值
        QVector<double> values;
        double randomVaule = 1 + QRandomGenerator::global()->generateDouble() * (8 - 1);
        values << randomVaule;

        Bars bar = it.value();
        bar.qBars->setData(bar.keys, values);
        if(groupByType && m_barsGroups.value(bar.dataType))
        {
            bar.qBars->setWidthType(QCPBars::wtPlotCoords);
            //bar.qBars->setWidth(bar.qBars->width() / groupedBars.value(bar.dataType).count());
            bar.qBars->setWidth(bar.qBars->width() * ticks.count() / m_bars.count());
            bar.qBars->setBarsGroup(m_barsGroups.value(bar.dataType));
        }
    }

    //Legend
    if(cfg.showLegend && m_bars.count() > 0)
        m_pCustomPlot->legend->setVisible(true);
    else
        m_pCustomPlot->legend->setVisible(false);

    m_pCustomPlot->replot();
}

void dpBarsChart::onSignal_dataUpdated(const QString& dataKey, const QVariant& data, const QDateTime& timestamp)
{}