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

#include "RifSummaryReaderInterface.h"
#include "RimSummaryCase.h"
#include "RimSummaryCaseCollection.h"

#include "cafAssert.h"

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RiaSummaryCurveDefinition::RiaSummaryCurveDefinition()
    : m_summaryCase( nullptr )
    , m_ensemble( nullptr )
    , m_isEnsembleCurve( false )
{
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RiaSummaryCurveDefinition::RiaSummaryCurveDefinition( RimSummaryCase*                 summaryCase,
                                                      const RifEclipseSummaryAddress& summaryAddress,
                                                      bool                            isEnsembleCurve )
    : m_summaryCase( summaryCase )
    , m_summaryAddress( summaryAddress )
    , m_ensemble( nullptr )
    , m_isEnsembleCurve( isEnsembleCurve )
{
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RiaSummaryCurveDefinition::RiaSummaryCurveDefinition( RimSummaryCaseCollection* ensemble, const RifEclipseSummaryAddress& summaryAddress )
    : m_summaryCase( nullptr )
    , m_summaryAddress( summaryAddress )
    , m_ensemble( ensemble )
    , m_isEnsembleCurve( true )
{
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RimSummaryCase* RiaSummaryCurveDefinition::summaryCase() const
{
    return m_summaryCase;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RimSummaryCaseCollection* RiaSummaryCurveDefinition::ensemble() const
{
    return m_ensemble;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
const RifEclipseSummaryAddress& RiaSummaryCurveDefinition::summaryAddress() const
{
    return m_summaryAddress;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
bool RiaSummaryCurveDefinition::isEnsembleCurve() const
{
    return m_isEnsembleCurve;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RiaSummaryCurveDefinition::setSummaryAddress( const RifEclipseSummaryAddress& address )
{
    m_summaryAddress = address;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RiaSummaryCurveDefinition::resultValues( const RiaSummaryCurveDefinition& curveDefinition, gsl::not_null<std::vector<double>*> values )
{
    if ( !curveDefinition.summaryAddress().isValid() ) return;
    if ( !curveDefinition.summaryCase() ) return;

    RifSummaryReaderInterface* reader = curveDefinition.summaryCase()->summaryReader();
    if ( !reader ) return;

    reader->values( curveDefinition.summaryAddress(), values );
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
std::vector<time_t> RiaSummaryCurveDefinition::timeSteps( const RiaSummaryCurveDefinition& curveDefinition )
{
    if ( !curveDefinition.summaryAddress().isValid() ) return {};
    if ( !curveDefinition.summaryCase() ) return {};

    RifSummaryReaderInterface* reader = curveDefinition.summaryCase()->summaryReader();
    if ( !reader ) return {};

    return reader->timeSteps( curveDefinition.summaryAddress() );
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
QString RiaSummaryCurveDefinition::curveDefinitionText() const
{
    QString caseName;
    if ( summaryCase() )
        caseName = summaryCase()->displayCaseName();
    else if ( ensemble() )
        caseName = ensemble()->name();

    return RiaSummaryCurveDefinition::curveDefinitionText( caseName, summaryAddress() );
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
QString RiaSummaryCurveDefinition::curveDefinitionText( const QString& caseName, const RifEclipseSummaryAddress& summaryAddress )
{
    QString txt;

    txt += caseName;
    txt += ", ";

    txt += QString::fromStdString( summaryAddress.uiText() );

    return txt;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
bool RiaSummaryCurveDefinition::operator<( const RiaSummaryCurveDefinition& other ) const
{
    if ( m_ensemble != other.ensemble() )
    {
        QString ensembleName;
        QString otherEnsembleName;

        if ( m_ensemble )
        {
            ensembleName = m_ensemble->name();
        }

        if ( other.ensemble() )
        {
            otherEnsembleName = other.ensemble()->name();
        }

        // First check if names are different to ensure stable alphabetic sort
        if ( ensembleName != otherEnsembleName )
        {
            return ensembleName < otherEnsembleName;
        }

        // Use pointer address, sorting will be be unstable
        return m_ensemble < other.ensemble();
    }

    if ( m_summaryCase != other.summaryCase() )
    {
        QString summaryCaseName;
        QString otherSummaryCaseName;

        if ( m_summaryCase )
        {
            summaryCaseName = m_summaryCase->displayCaseName();
        }
        if ( other.summaryCase() )
        {
            otherSummaryCaseName = other.summaryCase()->displayCaseName();
        }

        // First check if names are different to ensure stable alphabetic sort
        if ( summaryCaseName != otherSummaryCaseName )
        {
            return summaryCaseName < otherSummaryCaseName;
        }

        // Use pointer address, sorting will be be unstable
        return m_summaryCase < other.summaryCase();
    }

    if ( m_summaryAddress != other.summaryAddress() )
    {
        return ( m_summaryAddress < other.summaryAddress() );
    }

    return m_isEnsembleCurve < other.isEnsembleCurve();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RiaSummaryCurveDefinitionAnalyser::setCurveDefinitions( const std::vector<RiaSummaryCurveDefinition>& curveDefs )
{
    m_singleSummaryCases.clear();
    m_ensembles.clear();
    m_vectorNames.clear();
    m_summaryAdresses.clear();

    for ( const auto& curveDef : curveDefs )
    {
        bool valid = false;
        if ( curveDef.ensemble() && curveDef.isEnsembleCurve() )
        {
            m_ensembles.insert( curveDef.ensemble() );
            valid = true;
        }
        else if ( curveDef.summaryCase() )
        {
            m_singleSummaryCases.insert( curveDef.summaryCase() );

            if ( curveDef.summaryCase()->ensemble() )
            {
                m_ensembles.insert( curveDef.summaryCase()->ensemble() );
            }
            valid = true;
        }

        if ( valid )
        {
            const RifEclipseSummaryAddress& address = curveDef.summaryAddress();

            m_vectorNames.insert( address.vectorName() );
            m_summaryAdresses.insert( address );
        }
    }
}
