/////////////////////////////////////////////////////////////////////////////////
//
//  Copyright (C) 2017     Statoil ASA
// 
//  ResInsight is free software: you can redistribute it and/or modify
//  it under the terms of the GNU General Public License as published by
//  the Free Software Foundation, either version 3 of the License, or
//  (at your option) any later version.
// 
//  ResInsight is distributed in the hope that it will be useful, but WITHOUT ANY
//  WARRANTY; without even the implied warranty of MERCHANTABILITY or
//  FITNESS FOR A PARTICULAR PURPOSE.
// 
//  See the GNU General Public License at <http://www.gnu.org/licenses/gpl.html> 
//  for more details.
//
/////////////////////////////////////////////////////////////////////////////////

#include "RiuFlowCharacteristicsPlot.h"

#include "RiaColorTables.h"

#include "RimFlowCharacteristicsPlot.h"

#include "RiuLineSegmentQwtPlotCurve.h"
#include "RiuQwtPlotWheelZoomer.h"
#include "RiuQwtPlotZoomer.h"
#include "RiuResultQwtPlot.h"
#include "RiuSummaryQwtPlot.h"

#include "cvfBase.h"
#include "cvfColor3.h"

#include "qwt_date.h"
#include "qwt_legend.h"
#include "qwt_plot.h"
#include "qwt_plot_zoomer.h"
#include "qwt_symbol.h"

#include <QBoxLayout>
#include <QContextMenuEvent>
#include <QDateTime>
#include <QLabel>
#include <QMenu>


//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
RiuFlowCharacteristicsPlot::RiuFlowCharacteristicsPlot(RimFlowCharacteristicsPlot* plotDefinition, QWidget* parent) 
    :   m_plotDefinition(plotDefinition),
        QFrame(parent)
{
    Q_ASSERT(m_plotDefinition);
    
    QGridLayout* mainLayout = new QGridLayout();
    this->setLayout(mainLayout);
    this->layout()->setMargin(3);
    this->layout()->setSpacing(3);

    // White background
    QPalette pal = this->palette();
    pal.setColor(QPalette::Background, Qt::white);
    this->setAutoFillBackground(true);
    this->setPalette(pal);

    m_lorenzPlot = new QwtPlot(this);
    m_flowCapVsStorageCapPlot = new QwtPlot(this);
    m_sweepEffPlot = new QwtPlot(this);

    mainLayout->addWidget(m_lorenzPlot, 0 ,0, 1, 2);
    mainLayout->addWidget(m_flowCapVsStorageCapPlot, 1, 0);
    mainLayout->addWidget(m_sweepEffPlot, 1, 1);

    RiuSummaryQwtPlot::setCommonPlotBehaviour(m_lorenzPlot);
    new RiuQwtPlotWheelZoomer(m_lorenzPlot);
    addWindowZoom(m_lorenzPlot);
    RiuSummaryQwtPlot::enableDateBasedBottomXAxis(m_lorenzPlot);
    m_lorenzPlot->setTitle("Lorenz Coefficient");

    RiuSummaryQwtPlot::setCommonPlotBehaviour(m_sweepEffPlot);
    new RiuQwtPlotWheelZoomer(m_sweepEffPlot);
    addWindowZoom(m_sweepEffPlot);
    m_sweepEffPlot->setTitle("Sweep Efficiency");

    RiuSummaryQwtPlot::setCommonPlotBehaviour(m_flowCapVsStorageCapPlot);
    new RiuQwtPlotWheelZoomer(m_flowCapVsStorageCapPlot);
    addWindowZoom(m_flowCapVsStorageCapPlot);
    m_flowCapVsStorageCapPlot->setTitle("Flow Capacity vs Storage Capacity");
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
void RiuFlowCharacteristicsPlot::addWindowZoom(QwtPlot* plot)
{
    auto zoomer = new RiuQwtPlotZoomer(plot->canvas());
    zoomer->setRubberBandPen(QColor(Qt::black));
    zoomer->setTrackerMode(QwtPicker::AlwaysOff);
    zoomer->setTrackerPen(QColor(Qt::black));
    zoomer->initMousePattern(1);
}
//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
RiuFlowCharacteristicsPlot::~RiuFlowCharacteristicsPlot()
{
    if (m_plotDefinition)
    {
        m_plotDefinition->handleMdiWindowClosed();
    }
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
void RiuFlowCharacteristicsPlot::setLorenzCurve(const QStringList& dateTimeStrings, const std::vector<QDateTime>& dateTimes, const std::vector<double>& timeHistoryValues)
{
    initializeColors(dateTimes);

    m_lorenzPlot->detachItems(QwtPlotItem::Rtti_PlotCurve, true); 

    for (size_t tsIdx = 0; tsIdx < dateTimes.size(); ++tsIdx)
    {
        if (timeHistoryValues[tsIdx] == HUGE_VAL) continue;

        QDateTime dateTime = dateTimes[tsIdx];
        double timeHistoryValue = timeHistoryValues[tsIdx];

        QString curveName = dateTimeStrings[static_cast<int>(tsIdx)];

        RiuFlowCharacteristicsPlot::addCurveWithLargeSymbol(m_lorenzPlot, curveName, m_dateToColorMap[dateTime], dateTime, timeHistoryValue);
    }

    m_lorenzPlot->replot();
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
void RiuFlowCharacteristicsPlot::addCurveWithLargeSymbol(QwtPlot* plot, const QString& curveName, const QColor& color, const QDateTime& dateTime, double timeHistoryValue)
{
    auto curve = createEmptyCurve(plot, curveName, color);

    QwtSymbol::Style style = QwtSymbol::Diamond;
    QwtSymbol* symbol = new QwtSymbol(style);

    symbol->setSize(15, 15);
    symbol->setColor(color);

    curve->setSymbol(symbol);

    // Add date and value twice to avoid a cross as symbol generated by RiuLineSegmentQwtPlotCurve

    std::vector<QDateTime> dateTimes;
    dateTimes.push_back(dateTime);
    dateTimes.push_back(dateTime);

    std::vector<double> timeHistoryValues;
    timeHistoryValues.push_back(timeHistoryValue);
    timeHistoryValues.push_back(timeHistoryValue);

    curve->setSamplesFromDateAndValues(dateTimes, timeHistoryValues, false);
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
RiuLineSegmentQwtPlotCurve* RiuFlowCharacteristicsPlot::createEmptyCurve(QwtPlot* plot, const QString& curveName, const QColor& curveColor )
{
    RiuLineSegmentQwtPlotCurve* plotCurve = new RiuLineSegmentQwtPlotCurve(curveName);
   
    plotCurve->setTitle(curveName);
    plotCurve->setPen(QPen(curveColor));
    plotCurve->attach(plot);
    return plotCurve;
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
void RiuFlowCharacteristicsPlot::addFlowCapStorageCapCurve(const QDateTime& dateTime, const std::vector<double>& xVals, const std::vector<double>& yVals)
{
    CVF_ASSERT(!m_dateToColorMap.empty());

    RiuLineSegmentQwtPlotCurve* plotCurve = createEmptyCurve(m_flowCapVsStorageCapPlot, dateTime.toString(), m_dateToColorMap[dateTime]);
    plotCurve->setSamplesFromTimeAndValues(xVals, yVals, false);
    m_flowCapVsStorageCapPlot->replot();
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
void RiuFlowCharacteristicsPlot::addSweepEfficiencyCurve(const QDateTime& dateTime, const std::vector<double>& xVals, const std::vector<double>& yVals)
{
    CVF_ASSERT(!m_dateToColorMap.empty());

    RiuLineSegmentQwtPlotCurve* plotCurve = createEmptyCurve(m_sweepEffPlot, dateTime.toString(),  m_dateToColorMap[dateTime]);
    plotCurve->setSamplesFromTimeAndValues(xVals, yVals, false);

    m_sweepEffPlot->replot();
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
void RiuFlowCharacteristicsPlot::removeAllCurves()
{
    m_lorenzPlot->detachItems(QwtPlotItem::Rtti_PlotCurve, true); 
    m_lorenzPlot->replot();
    m_sweepEffPlot->detachItems(QwtPlotItem::Rtti_PlotCurve, true); 
    m_sweepEffPlot->replot();
    m_flowCapVsStorageCapPlot->detachItems(QwtPlotItem::Rtti_PlotCurve, true); 
    m_flowCapVsStorageCapPlot->replot();
    m_dateToColorMap.clear();
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
void zoomAllInPlot(QwtPlot * plot)
{
    plot->setAxisAutoScale(QwtPlot::xBottom, true);
    plot->setAxisAutoScale(QwtPlot::yLeft, true);
    plot->replot();
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
void RiuFlowCharacteristicsPlot::zoomAll()
{
    zoomAllInPlot(m_lorenzPlot);
    zoomAllInPlot(m_sweepEffPlot);
    zoomAllInPlot(m_flowCapVsStorageCapPlot);
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
void RiuFlowCharacteristicsPlot::showLegend(bool show)
{
    if (show)
    {
        // Will be released in plot destructor or when a new legend is set
        QwtLegend* legend = new QwtLegend(m_lorenzPlot);
        m_lorenzPlot->insertLegend(legend, QwtPlot::BottomLegend);
    }
    else
    {
        m_lorenzPlot->insertLegend(nullptr);
    }
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
RimFlowCharacteristicsPlot* RiuFlowCharacteristicsPlot::ownerPlotDefinition()
{
    return m_plotDefinition;
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
RimViewWindow* RiuFlowCharacteristicsPlot::ownerViewWindow() const
{
    return m_plotDefinition;
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
QSize RiuFlowCharacteristicsPlot::minimumSizeHint() const
{
    return QSize(0, 100);
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
QSize RiuFlowCharacteristicsPlot::sizeHint() const
{
    return QSize(0, 0);
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
void RiuFlowCharacteristicsPlot::setDefaults()
{

}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
void RiuFlowCharacteristicsPlot::initializeColors(const std::vector<QDateTime>& dateTimes)
{
    CVF_ASSERT(m_dateToColorMap.empty());

    const caf::ColorTable& palette = RiaColorTables::timestepsPaletteColors();
    cvf::Color3ubArray colorArray = caf::ColorTable::interpolateColorArray(palette.color3ubArray(), dateTimes.size());

    for (size_t tsIdx = 0; tsIdx < dateTimes.size(); ++tsIdx)
    {
        m_dateToColorMap[dateTimes[tsIdx]] = QColor( colorArray[tsIdx].r(), colorArray[tsIdx].g(), colorArray[tsIdx].b());
    }
}