/////////////////////////////////////////////////////////////////////////////////
//
//  Copyright (C) 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 "RimFormationNames.h"

#include "RiaLogging.h"
#include "RifFormationNamesReader.h"
#include "RigFormationNames.h"
#include "RiuMessageDialog.h"

#include "Rim3dView.h"
#include "RimCase.h"
#include "RimTools.h"
#include "RimWellLogTrack.h"

#include "RiuPlotMainWindowTools.h"

#include "cafAssert.h"
#include "cafPdmFieldScriptingCapability.h"
#include "cafPdmObjectScriptingCapability.h"
#include "cafPdmUiFilePathEditor.h"

#include <QFile>
#include <QFileInfo>

CAF_PDM_SOURCE_INIT( RimFormationNames, "FormationNames" );

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RimFormationNames::RimFormationNames()
{
    CAF_PDM_InitScriptableObjectWithNameAndComment( "Formation Names",
                                                    ":/Formations16x16.png",
                                                    "",
                                                    "",
                                                    "FormationNames",
                                                    "Formation names imported from a .lyr or similar file" );

    CAF_PDM_InitScriptableFieldNoDefault( &m_formationNamesFileName, "FormationNamesFileName", "File Name" );

    m_formationNamesFileName.uiCapability()->setUiEditorTypeName( caf::PdmUiFilePathEditor::uiEditorTypeName() );

    setDeletable( true );
}

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

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimFormationNames::fieldChangedByUi( const caf::PdmFieldHandle* changedField, const QVariant& oldValue, const QVariant& newValue )
{
    if ( &m_formationNamesFileName == changedField )
    {
        updateUiTreeName();
        if ( auto result = readFormationNamesFile(); !result )
        {
            RiuMessageDialog::showError( nullptr, "Formation Names", result.error() );
        }
        updateConnectedViews();
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimFormationNames::defineUiTreeOrdering( caf::PdmUiTreeOrdering& uiTreeOrdering, QString uiConfigName /*= ""*/ )
{
    updateUiTreeName();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimFormationNames::setFileName( const QString& fileName )
{
    m_formationNamesFileName = fileName;

    updateUiTreeName();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
QString RimFormationNames::fileName()
{
    return m_formationNamesFileName().path();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
QString RimFormationNames::shortName()
{
    QFileInfo fnameFileInfo( m_formationNamesFileName().path() );
    return fnameFileInfo.baseName();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
const RigFormationNames* RimFormationNames::formationNamesData() const
{
    return m_formationNamesData.get();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimFormationNames::updateConnectedViews()
{
    std::vector<RimCase*> objects = objectsWithReferringPtrFieldsOfType<RimCase>();

    for ( RimCase* caseObj : objects )
    {
        if ( caseObj )
        {
            caseObj->updateFormationNamesData();

            std::vector<RimWellLogTrack*> tracks = caseObj->objectsWithReferringPtrFieldsOfType<RimWellLogTrack>();
            for ( RimWellLogTrack* track : tracks )
            {
                // The track may be referring to the case for other reasons than formations.
                if ( track->formationCase() == caseObj )
                {
                    track->loadDataAndUpdate();
                }
            }
        }
    }
    RiuPlotMainWindowTools::refreshToolbars();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
std::expected<void, QString> RimFormationNames::readFormationNamesFile()
{
    auto result = RifFormationNamesReader::readFormationNamesFile( m_formationNamesFileName().path() );
    if ( !result )
    {
        m_formationNamesData.reset();
        return std::unexpected( result.error() );
    }

    m_formationNamesData = std::make_unique<RigFormationNames>( std::move( *result ) );
    return {};
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
QString RimFormationNames::layerZoneTableFileName()
{
    return "layer_zone_table.txt";
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimFormationNames::updateUiTreeName()
{
    uiCapability()->setUiName( shortName() );
}