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

#include "RimEclipseView.h"
#include "RimProject.h"
#include "RimTools.h"
#include "RimWellPathCollection.h"

#include "RivWellPathPartMgr.h"

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RivWellPathsPartMgr::RivWellPathsPartMgr( Rim3dView* view )
    : m_rimView( view )
{
}

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

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RivWellPathsPartMgr::appendStaticGeometryPartsToModel( cvf::ModelBasicList*              model,
                                                            const caf::DisplayCoordTransform* displayCoordTransform,
                                                            double                            characteristicCellSize,
                                                            const cvf::BoundingBox&           wellPathClipBoundingBox )
{
    if ( !isWellPathVisible() ) return;

    createPartManagersIfRequired();

    for ( auto& partMgr : m_wellPathsPartMgrs )
    {
        partMgr->appendStaticGeometryPartsToModel( model, displayCoordTransform, characteristicCellSize, wellPathClipBoundingBox );
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RivWellPathsPartMgr::appendStaticFracturePartsToModel( cvf::ModelBasicList* model, const cvf::BoundingBox& wellPathClipBoundingBox )
{
    if ( !isWellPathVisible() ) return;

    createPartManagersIfRequired();

    for ( auto& partMgr : m_wellPathsPartMgrs )
    {
        partMgr->appendStaticFracturePartsToModel( model, wellPathClipBoundingBox );
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RivWellPathsPartMgr::appendDynamicGeometryPartsToModel( cvf::ModelBasicList*              model,
                                                             size_t                            timeStepIndex,
                                                             const caf::DisplayCoordTransform* displayCoordTransform,
                                                             double                            characteristicCellSize,
                                                             const cvf::BoundingBox&           wellPathClipBoundingBox )
{
    createPartManagersIfRequired();

    for ( auto& partMgr : m_wellPathsPartMgrs )
    {
        partMgr->appendDynamicGeometryPartsToModel( model, timeStepIndex, displayCoordTransform, characteristicCellSize, wellPathClipBoundingBox );
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RivWellPathsPartMgr::clearGeometryCache()
{
    m_wellPathsPartMgrs.clear();
    m_mapFromViewToIndex.clear();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RivWellPathsPartMgr::scheduleGeometryRegen()
{
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RivWellPathsPartMgr::createPartManagersIfRequired()
{
    RimProject* proj      = RimProject::current();
    auto        wellPaths = proj->allWellPaths();

    if ( m_wellPathsPartMgrs.size() != wellPaths.size() )
    {
        clearGeometryCache();

        for ( auto wellPath : wellPaths )
        {
            RivWellPathPartMgr* wppm = new RivWellPathPartMgr( wellPath, m_rimView );
            m_wellPathsPartMgrs.push_back( wppm );
            m_mapFromViewToIndex[wellPath] = wppm;
        }
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
bool RivWellPathsPartMgr::isWellPathVisible() const
{
    auto wellPathColl = RimTools::wellPathCollection();

    if ( !wellPathColl->isActive() ) return false;
    if ( wellPathColl->wellPathVisibility() == RimWellPathCollection::FORCE_ALL_OFF ) return false;

    return true;
}