/////////////////////////////////////////////////////////////////////////////////
//
//  Copyright (C) 2011-2012 Statoil ASA, Ceetron AS
//
//  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 "RimWellPathImport.h"

#include "RimOilFieldEntry.h"
#include "RimOilRegionEntry.h"
#include "RimTools.h"
#include "RimWellPath.h"
#include "RimWellPathCollection.h"

#include "cafPdmUiTreeViewEditor.h"

#include <QFileInfo>

namespace caf
{
template <>
void caf::AppEnum<RimWellPathImport::UtmFilterEnum>::setUp()
{
    addItem( RimWellPathImport::UTM_FILTER_OFF, "UTM_FILTER_OFF", "Off" );
    addItem( RimWellPathImport::UTM_FILTER_PROJECT, "UTM_FILTER_PROJECT", "Project" );
    addItem( RimWellPathImport::UTM_FILTER_CUSTOM, "UTM_FILTER_CUSTOM", "Custom" );
    setDefault( RimWellPathImport::UTM_FILTER_PROJECT );
}

} // End namespace caf

CAF_PDM_SOURCE_INIT( RimWellPathImport, "RimWellPathImport" );

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RimWellPathImport::RimWellPathImport()
{
    CAF_PDM_InitObject( "RimWellPathImport", "", "", "" );

    CAF_PDM_InitField( &wellTypeSurvey, "WellTypeSurvey", true, "Survey", "", "", "" );
    CAF_PDM_InitField( &wellTypePlans, "WellTypePlans", true, "Plans", "", "", "" );

    caf::AppEnum<RimWellPathImport::UtmFilterEnum> defaultUtmMode = UTM_FILTER_OFF;
    CAF_PDM_InitField( &utmFilterMode, "UtmMode", defaultUtmMode, "Utm Filter", "", "", "" );

    CAF_PDM_InitField( &north, "UtmNorth", 0.0, "North", "", "", "" );
    CAF_PDM_InitField( &south, "UtmSouth", 0.0, "South", "", "", "" );
    CAF_PDM_InitField( &east, "UtmEast", 0.0, "East", "", "", "" );
    CAF_PDM_InitField( &west, "UtmWest", 0.0, "West", "", "", "" );

    CAF_PDM_InitFieldNoDefault( &regions, "Regions", "", "", "", "" );
    regions.uiCapability()->setUiHidden( true );
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimWellPathImport::updateRegions( const QStringList& regionStrings,
                                       const QStringList& fieldStrings,
                                       const QStringList& edmIds )
{
    assert( regionStrings.size() == fieldStrings.size() && regionStrings.size() == edmIds.size() );

    std::vector<RimOilRegionEntry*> regionsToRemove;

    // Remove regions and fields not present in last request
    for ( size_t regionIdx = 0; regionIdx < this->regions.size(); regionIdx++ )
    {
        if ( !regionStrings.contains( this->regions[regionIdx]->name ) )
        {
            regionsToRemove.push_back( this->regions[regionIdx] );
        }
        else
        {
            std::vector<RimOilFieldEntry*> fieldsToRemove;

            for ( size_t fIdx = 0; fIdx < this->regions[regionIdx]->fields.size(); fIdx++ )
            {
                if ( !fieldStrings.contains( this->regions[regionIdx]->fields[fIdx]->name ) )
                {
                    fieldsToRemove.push_back( this->regions[regionIdx]->fields[fIdx] );
                }
            }

            for ( size_t i = 0; i < fieldsToRemove.size(); i++ )
            {
                this->regions[regionIdx]->fields.removeChildObject( fieldsToRemove[i] );

                delete fieldsToRemove[i];
            }
        }
    }

    for ( size_t i = 0; i < regionsToRemove.size(); i++ )
    {
        this->regions.removeChildObject( regionsToRemove[i] );

        delete regionsToRemove[i];
    }

    for ( int i = 0; i < regionStrings.size(); i++ )
    {
        RimOilRegionEntry* oilRegionEntry = nullptr;
        RimOilFieldEntry*  oilFieldEntry  = nullptr;

        for ( size_t regionIdx = 0; regionIdx < this->regions.size(); regionIdx++ )
        {
            if ( this->regions[regionIdx]->name == regionStrings[i] )
            {
                oilRegionEntry = this->regions[regionIdx];

                for ( size_t fIdx = 0; fIdx < this->regions[regionIdx]->fields.size(); fIdx++ )
                {
                    if ( this->regions[regionIdx]->fields[fIdx]->edmId == edmIds[i] )
                    {
                        oilFieldEntry = this->regions[regionIdx]->fields[fIdx];
                    }
                }
            }
        }

        if ( !oilRegionEntry )
        {
            oilRegionEntry       = new RimOilRegionEntry;
            oilRegionEntry->name = regionStrings[i];

            this->regions.push_back( oilRegionEntry );
        }

        assert( oilRegionEntry );

        if ( !oilFieldEntry )
        {
            oilFieldEntry        = new RimOilFieldEntry;
            oilFieldEntry->name  = fieldStrings[i];
            oilFieldEntry->edmId = edmIds[i];

            oilRegionEntry->fields.push_back( oilFieldEntry );
        }
    }

    updateFieldVisibility();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimWellPathImport::initAfterRead()
{
    updateFieldVisibility();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimWellPathImport::updateFieldVisibility()
{
    if ( utmFilterMode == UTM_FILTER_CUSTOM )
    {
        north.uiCapability()->setUiReadOnly( false );
        south.uiCapability()->setUiReadOnly( false );
        east.uiCapability()->setUiReadOnly( false );
        west.uiCapability()->setUiReadOnly( false );
    }
    else
    {
        north.uiCapability()->setUiReadOnly( true );
        south.uiCapability()->setUiReadOnly( true );
        east.uiCapability()->setUiReadOnly( true );
        west.uiCapability()->setUiReadOnly( true );
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimWellPathImport::fieldChangedByUi( const caf::PdmFieldHandle* changedField,
                                          const QVariant&            oldValue,
                                          const QVariant&            newValue )
{
    if ( changedField == &utmFilterMode )
    {
        updateFieldVisibility();
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimWellPathImport::defineObjectEditorAttribute( QString uiConfigName, caf::PdmUiEditorAttribute* attribute )
{
    caf::PdmUiTreeViewEditorAttribute* myAttr = dynamic_cast<caf::PdmUiTreeViewEditorAttribute*>( attribute );
    if ( myAttr )
    {
        QStringList colHeaders;
        colHeaders << "Region";
        myAttr->columnHeaders = colHeaders;
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RimWellPathImport::~RimWellPathImport()
{
    regions.deleteAllChildObjects();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimWellPathImport::updateFilePaths()
{
    QString wellPathsFolderPath = RimTools::getCacheRootDirectoryPathFromProject();
    wellPathsFolderPath += "_wellpaths";

    for ( size_t regionIdx = 0; regionIdx < this->regions.size(); regionIdx++ )
    {
        for ( size_t fIdx = 0; fIdx < this->regions[regionIdx]->fields.size(); fIdx++ )
        {
            RimOilFieldEntry* oilField = this->regions[regionIdx]->fields[fIdx];

            QFileInfo fi( oilField->wellsFilePath );

            QString newWellsFilePath = wellPathsFolderPath + "/" + fi.fileName();
            oilField->wellsFilePath  = newWellsFilePath;

            for ( size_t wIdx = 0; wIdx < oilField->wells.size(); wIdx++ )
            {
                RimWellPathEntry* rimWellPathEntry = oilField->wells[wIdx];

                QFileInfo fiWell( rimWellPathEntry->wellPathFilePath );

                QString newFilePath                = wellPathsFolderPath + "/" + fiWell.fileName();
                rimWellPathEntry->wellPathFilePath = newFilePath;
            }
        }
    }
}