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

#include "RiaApplication.h"

#include "RigEclipseCaseData.h"

#include "RimEclipseCase.h"
#include "RimProject.h"
#include "RimWellPlotTools.h"

#include "cafPdmUiOrdering.h"

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
std::vector<const RigWellPath*> RiaSimWellBranchTools::simulationWellBranches( const QString& simWellName,
                                                                               bool useAutoDetectionOfBranches )
{
    RiaApplication* app  = RiaApplication::instance();
    RimProject*     proj = app->project();

    // Find first case containing the specified simulation well
    std::vector<RimEclipseCase*> simCases;
    proj->descendantsIncludingThisOfType( simCases );

    auto caseItr = std::find_if( simCases.begin(), simCases.end(), [&simWellName]( const RimEclipseCase* eclCase ) {
        const auto& eclData = eclCase->eclipseCaseData();
        return eclData != nullptr && eclData->hasSimulationWell( simWellName );
    } );
    RimEclipseCase*     eclipseCase = caseItr != simCases.end() ? *caseItr : nullptr;
    RigEclipseCaseData* eclCaseData = eclipseCase != nullptr ? eclipseCase->eclipseCaseData() : nullptr;
    return eclCaseData != nullptr ? eclCaseData->simulationWellBranches( simWellName, false, useAutoDetectionOfBranches )
                                  : std::vector<const RigWellPath*>();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
QList<caf::PdmOptionItemInfo>
    RiaSimWellBranchTools::valueOptionsForBranchIndexField( const std::vector<const RigWellPath*>& simulationWellPaths )
{
    QList<caf::PdmOptionItemInfo> options;

    size_t branchCount = simulationWellPaths.size();
    if ( simulationWellPaths.size() == 0 )
    {
        options.push_front( caf::PdmOptionItemInfo( "None", -1 ) );
    }
    else
    {
        for ( int bIdx = 0; bIdx < static_cast<int>( branchCount ); ++bIdx )
        {
            options.push_back(
                caf::PdmOptionItemInfo( "Branch " + QString::number( bIdx + 1 ), QVariant::fromValue( bIdx ) ) );
        }
    }

    return options;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RiaSimWellBranchTools::appendSimWellBranchFieldsIfRequiredFromWellName( caf::PdmUiOrdering* uiOrdering,
                                                                             const QString&      wellPathOrSimWellName,
                                                                             const caf::PdmField<bool>& branchDetectionField,
                                                                             const caf::PdmField<int>& branchIndexField )
{
    if ( !RimWellPlotTools::hasAssociatedWellPath( wellPathOrSimWellName ) )
    {
        const QString simWellName = RimWellPlotTools::simWellName( wellPathOrSimWellName );

        RiaSimWellBranchTools::appendSimWellBranchFieldsIfRequiredFromSimWellName( uiOrdering,
                                                                                   simWellName,
                                                                                   branchDetectionField,
                                                                                   branchIndexField );
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RiaSimWellBranchTools::appendSimWellBranchFieldsIfRequiredFromSimWellName( caf::PdmUiOrdering* uiOrdering,
                                                                                const QString&      simWellName,
                                                                                const caf::PdmField<bool>& branchDetectionField,
                                                                                const caf::PdmField<int>& branchIndexField )
{
    if ( RiaSimWellBranchTools::simulationWellBranches( simWellName, true ).size() > 1 )
    {
        uiOrdering->add( &branchDetectionField );

        if ( RiaSimWellBranchTools::simulationWellBranches( simWellName, branchDetectionField ).size() > 1 )
        {
            uiOrdering->add( &branchIndexField );
        }
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
int RiaSimWellBranchTools::clampBranchIndex( const QString& simWellName, int branchIndexValue, bool branchDetection )
{
    auto branches = RiaSimWellBranchTools::simulationWellBranches( simWellName, branchDetection );

    if ( branches.size() == 0 )
    {
        return -1;
    }

    int maxIndexValue = static_cast<int>( branches.size() ) - 1;
    if ( branchIndexValue > maxIndexValue )
    {
        branchIndexValue = maxIndexValue;
    }

    return branchIndexValue;
}