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

#include "RiaLogging.h"
#include "RiaQDateTimeTools.h"

#include "RifEclipseSummaryAddress.h"
#include "RifEclipseUserDataParserTools.h"
#include "RifKeywordVectorParser.h"

#include "cafUtils.h"

#include <QDateTime>
#include <QFile>
#include <QStringList>
#include <QTextStream>

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RifKeywordVectorUserData::RifKeywordVectorUserData()
{
}

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

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
bool RifKeywordVectorUserData::parse( const QString& data, const QString& customWellName )
{
    m_allResultAddresses.clear();
    m_timeSteps.clear();

    m_parser = std::unique_ptr<RifKeywordVectorParser>( new RifKeywordVectorParser( data ) );
    if ( !m_parser )
    {
        RiaLogging::error( QString( "Failed to parse file" ) );

        return false;
    }

    std::vector<std::map<QString, QString>> keyValuePairVector;
    for ( const KeywordBasedVector& keywordVector : m_parser->keywordBasedVectors() )
    {
        std::map<QString, QString> keyValuePairs;

        for ( auto s : keywordVector.header )
        {
            QString ss = QString::fromStdString( s );

            QStringList entries = ss.split( " " );
            if ( entries.size() == 2 )
            {
                keyValuePairs[entries[0]] = entries[1];
            }
        }

        keyValuePairVector.push_back( keyValuePairs );
    }

    // Find all time vectors

    std::map<QString, size_t> mapFromOriginToTimeStepIndex;

    for ( size_t i = 0; i < keyValuePairVector.size(); i++ )
    {
        const std::map<QString, QString>& keyValuePairs = keyValuePairVector[i];

        if ( isTimeHeader( keyValuePairs ) )
        {
            std::vector<time_t> ts;

            {
                QDateTime startDate;
                QString   startDateString = valueForKey( keyValuePairs, "STARTDATE" );
                if ( !startDateString.isEmpty() )
                {
                    QString dateFormatString = valueForKey( keyValuePairs, "DATEFORMAT" );
                    if ( dateFormatString.isEmpty() )
                    {
                        dateFormatString = "DD MM YYYY";
                    }

                    startDate = RiaQDateTimeTools::fromString( startDateString, dateFormatString );
                }
                else
                {
                    startDate = RiaQDateTimeTools::epoch();
                }

                QString unitText = valueForKey( keyValuePairs, "UNITS" );
                if ( unitText == "DAY" || unitText == "DAYS" )
                {
                    for ( const auto& timeStepValue : m_parser->keywordBasedVectors()[i].values )
                    {
                        QDateTime dateTime = RiaQDateTimeTools::addDays( startDate, timeStepValue );
                        ts.push_back( dateTime.toSecsSinceEpoch() );
                    }
                }
                else if ( unitText == "YEAR" || unitText == "YEARS" )
                {
                    for ( const auto& timeStepValue : m_parser->keywordBasedVectors()[i].values )
                    {
                        QDateTime dateTime = RiaQDateTimeTools::fromYears( timeStepValue );
                        ts.push_back( dateTime.toSecsSinceEpoch() );
                    }
                }
            }

            m_timeSteps.push_back( ts );

            QString originText = valueForKey( keyValuePairs, "ORIGIN" );

            mapFromOriginToTimeStepIndex[originText] = m_timeSteps.size() - 1;
        }
    }

    // Find all data vectors having a reference to a time step vector

    for ( size_t i = 0; i < keyValuePairVector.size(); i++ )
    {
        const std::map<QString, QString>& keyValuePairs = keyValuePairVector[i];

        if ( !isTimeHeader( keyValuePairs ) )
        {
            if ( isVectorHeader( keyValuePairs ) )
            {
                QString originText            = valueForKey( keyValuePairs, "ORIGIN" );
                auto    timeStepIndexIterator = mapFromOriginToTimeStepIndex.find( originText );
                if ( timeStepIndexIterator != mapFromOriginToTimeStepIndex.end() )
                {
                    QString vectorText = valueForKey( keyValuePairs, "VECTOR" );

                    QString wellName = originText;
                    if ( customWellName.size() > 0 )
                    {
                        wellName = customWellName;
                    }

                    auto addr = RifEclipseSummaryAddress::wellAddress( vectorText.toStdString(), wellName.toStdString(), -1 );
                    m_allResultAddresses.insert( addr );

                    m_mapFromAddressToTimeIndex[addr]   = timeStepIndexIterator->second;
                    m_mapFromAddressToVectorIndex[addr] = i;
                }
            }
        }
    }

    return true;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
std::pair<bool, std::vector<double>> RifKeywordVectorUserData::values( const RifEclipseSummaryAddress& resultAddress ) const
{
    auto search = m_mapFromAddressToVectorIndex.find( resultAddress );
    if ( search == m_mapFromAddressToVectorIndex.end() ) return { false, {} };

    std::vector<double> values;
    for ( const auto& v : m_parser->keywordBasedVectors()[search->second].values )
    {
        values.push_back( v );
    }

    return { true, values };
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
std::vector<time_t> RifKeywordVectorUserData::timeSteps( const RifEclipseSummaryAddress& resultAddress ) const
{
    auto timeIndexIterator = m_mapFromAddressToTimeIndex.find( resultAddress );
    if ( timeIndexIterator != m_mapFromAddressToTimeIndex.end() )
    {
        return m_timeSteps[timeIndexIterator->second];
    }

    return {};
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
std::string RifKeywordVectorUserData::unitName( const RifEclipseSummaryAddress& resultAddress ) const
{
    return "";
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RiaDefines::EclipseUnitSystem RifKeywordVectorUserData::unitSystem() const
{
    return RiaDefines::EclipseUnitSystem::UNITS_UNKNOWN;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
bool RifKeywordVectorUserData::isTimeHeader( const std::map<QString, QString>& header )
{
    for ( const auto& keyValue : header )
    {
        if ( keyValue.first == "VECTOR" && keyValue.second == "YEARX" )
        {
            return true;
        }
    }

    return false;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
bool RifKeywordVectorUserData::isVectorHeader( const std::map<QString, QString>& header )
{
    for ( const auto& keyValue : header )
    {
        if ( keyValue.first == "VECTOR" )
        {
            return true;
        }
    }

    return false;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
QString RifKeywordVectorUserData::valueForKey( const std::map<QString, QString>& header, const QString& key )
{
    auto it = header.find( key );
    if ( it != header.end() )
    {
        return it->second;
    }

    return "";
}