/////////////////////////////////////////////////////////////////////////////////
//
//  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 "RifWellPathFormationsImporter.h"
#include "RifWellPathFormationReader.h"

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
cvf::ref<RigWellPathFormations> RifWellPathFormationsImporter::readWellPathFormations( const QString& formationFilePath, const QString& wellName )
{
    readAllWellPathFormations( formationFilePath );
    if ( m_fileNameToWellPathFormationMap[formationFilePath].find( wellName ) != m_fileNameToWellPathFormationMap[formationFilePath].end() )
    {
        return m_fileNameToWellPathFormationMap[formationFilePath][wellName];
    }
    else
    {
        return nullptr;
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
cvf::ref<RigWellPathFormations> RifWellPathFormationsImporter::reloadWellPathFormations( const QString& formationFilePath,
                                                                                         const QString& wellName )
{
    m_fileNameToWellPathFormationMap.erase( formationFilePath );

    return readWellPathFormations( formationFilePath, wellName );
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
std::map<QString, cvf::ref<RigWellPathFormations>> RifWellPathFormationsImporter::readWellPathFormationsFromPath( const QString& filePath )
{
    // If we have the file in the map, assume it is already read.
    if ( m_fileNameToWellPathFormationMap.find( filePath ) != m_fileNameToWellPathFormationMap.end() )
    {
        return m_fileNameToWellPathFormationMap[filePath];
    }

    std::map<QString, cvf::ref<RigWellPathFormations>> wellPathToFormationMap =
        RifWellPathFormationReader::readWellFormationsToGeometry( filePath );

    m_fileNameToWellPathFormationMap[filePath] = wellPathToFormationMap;

    return wellPathToFormationMap;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RifWellPathFormationsImporter::reloadAllWellPathFormations()
{
    std::vector<QString> allFilePaths;
    for ( auto it = m_fileNameToWellPathFormationMap.begin(); it != m_fileNameToWellPathFormationMap.end(); it++ )
    {
        allFilePaths.push_back( it->first );
    }

    m_fileNameToWellPathFormationMap.clear();

    for ( const QString& filePath : allFilePaths )
    {
        readWellPathFormationsFromPath( filePath );
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RifWellPathFormationsImporter::readAllWellPathFormations( const QString& filePath )
{
    // If we have the file in the map, assume it is already read.
    if ( m_fileNameToWellPathFormationMap.find( filePath ) != m_fileNameToWellPathFormationMap.end() )
    {
        return;
    }

    std::map<QString, cvf::ref<RigWellPathFormations>> wellPathToFormationMap =
        RifWellPathFormationReader::readWellFormationsToGeometry( filePath );

    m_fileNameToWellPathFormationMap[filePath] = wellPathToFormationMap;
}