/////////////////////////////////////////////////////////////////////////////////
//
//  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 "RicSummaryCurveCalculatorDialog.h"

#include "RicSummaryCurveCalculatorUi.h"
#include "RicSummaryCurveCalculatorWidgetCreator.h"

#include "RimSummaryCalculation.h"
#include "RimSummaryCalculationCollection.h"

#include "RiuTools.h"

#include <QDialogButtonBox>
#include <QMessageBox>
#include <QVBoxLayout>

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RicSummaryCurveCalculatorDialog::RicSummaryCurveCalculatorDialog( QWidget* parent )
    : QDialog( parent, RiuTools::defaultDialogFlags() )
{
    setWindowTitle( "Summary Curve Calculator" );
    resize( 1200, 800 );

    setUp();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RicSummaryCurveCalculatorDialog::~RicSummaryCurveCalculatorDialog()
{
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RicSummaryCurveCalculatorDialog::setCalculationAndUpdateUi( RimSummaryCalculation* calculation )
{
    m_summaryCalcEditor->calculator()->setCurrentCalculation( calculation );

    m_summaryCalcEditor->updateUi();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RicSummaryCurveCalculatorDialog::slotTryCloseDialog()
{
    RimSummaryCalculationCollection* calculationCollection = RicSummaryCurveCalculatorUi::calculationCollection();

    if ( dirtyCount() > 0 )
    {
        QMessageBox msgBox( this );
        msgBox.setIcon( QMessageBox::Question );

        QString questionText = QString( "Detected calculation expression text modifications." );

        msgBox.setText( questionText );
        msgBox.setInformativeText( "Do you want to trigger calculation?" );
        msgBox.setStandardButtons( QMessageBox::Yes | QMessageBox::No );

        int ret = msgBox.exec();
        if ( ret == QMessageBox::No )
        {
            reject();
        }
        else if ( ret == QMessageBox::Yes )
        {
            for ( auto c : calculationCollection->calculations() )
            {
                if ( c->isDirty() )
                {
                    c->calculate();
                    c->updateDependentCurvesAndPlots();
                }
            }

            if ( dirtyCount() > 0 )
            {
                return;
            }
        }
        else
        {
            return;
        }
    }

    accept();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RicSummaryCurveCalculatorDialog::setUp()
{
    QVBoxLayout* mainLayout = new QVBoxLayout( this );
    mainLayout->setContentsMargins( 0, 0, 0, 0 );

    m_summaryCalcEditor =
        std::unique_ptr<RicSummaryCurveCalculatorWidgetCreator>( new RicSummaryCurveCalculatorWidgetCreator() );
    mainLayout->addWidget( m_summaryCalcEditor->getOrCreateWidget( this ) );

    QDialogButtonBox* buttonBox = new QDialogButtonBox( QDialogButtonBox::Close );
    connect( buttonBox, SIGNAL( rejected() ), this, SLOT( slotTryCloseDialog() ) );

    mainLayout->addWidget( buttonBox );

    m_summaryCalcEditor->updateUi();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
size_t RicSummaryCurveCalculatorDialog::dirtyCount() const
{
    size_t count = 0;

    RimSummaryCalculationCollection* calculationCollection = RicSummaryCurveCalculatorUi::calculationCollection();
    for ( auto c : calculationCollection->calculations() )
    {
        if ( c->isDirty() )
        {
            count++;
        }
    }

    return count;
}