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

#include "RiaApplication.h"
#include "RiaSeismicDefines.h"

#include "RimOilField.h"
#include "RimProject.h"
#include "RimSeismicData.h"
#include "RimSeismicDataCollection.h"
#include "RimSeismicSection.h"
#include "RimSeismicSectionCollection.h"
#include "RimSeismicView.h"
#include "RimSeismicViewCollection.h"

#include "Riu3DMainWindowTools.h"

#include "cafPdmUiPropertyViewDialog.h"
#include "cafSelectionManagerTools.h"
#include "cafUtils.h"

#include <QAction>
#include <QMessageBox>

CAF_CMD_SOURCE_INIT( RicNewSeismicViewFeature, "RicNewSeismicViewFeature" );

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
bool RicNewSeismicViewFeature::isCommandEnabled() const
{
    auto  proj     = RimProject::current();
    auto& seisColl = proj->activeOilField()->seismicDataCollection();
    return ( seisColl && !seisColl->isEmpty() );
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RicNewSeismicViewFeature::onActionTriggered( bool isChecked )
{
    std::vector<caf::PdmUiItem*> uiItems;
    caf::SelectionManager::instance()->selectedItems( uiItems );

    RimSeismicData* selectedData = nullptr;

    if ( !uiItems.empty() )
    {
        selectedData = dynamic_cast<RimSeismicData*>( uiItems[0] );
    }

    createSeismicView( selectedData );
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RicNewSeismicViewFeature::setupActionLook( QAction* actionToSetup )
{
    actionToSetup->setIcon( QIcon( ":/SeismicView16x16.png" ) );
    actionToSetup->setText( "New Seismic View" );
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RimSeismicView* RicNewSeismicViewFeature::createInitialViewIfNeeded( RimSeismicDataInterface* seisData )
{
    auto proj = RimProject::current();

    std::vector<Rim3dView*> views;
    proj->allViews( views );
    if ( !views.empty() ) return nullptr;

    return createSeismicView( seisData );
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RimSeismicView* RicNewSeismicViewFeature::createSeismicView( RimSeismicDataInterface* seisData )
{
    auto  proj         = RimProject::current();
    auto& seisViewColl = proj->activeOilField()->seismicViewCollection();
    auto& seisDataColl = proj->activeOilField()->seismicDataCollection();
    if ( !seisViewColl || !seisDataColl ) return nullptr;

    if ( ( seisData == nullptr ) && !seisDataColl->isEmpty() )
    {
        seisData = seisDataColl->seismicData()[0];
    }

    if ( seisData )
    {
        auto view = seisViewColl->addView( seisData, RiaDefines::SeismicSectionType::SS_INLINE );

        if ( view )
        {
            seisViewColl->updateAllRequiredEditors();
            view->scheduleCreateDisplayModelAndRedraw();

            if ( view->seismicSectionCollection()->size() > 0 )
            {
                auto section = view->seismicSectionCollection()->seismicSections()[0];
                Riu3DMainWindowTools::selectAsCurrentItem( section );
            }
            else
            {
                Riu3DMainWindowTools::selectAsCurrentItem( view );
            }
        }

        return view;
    }

    return nullptr;
}