ResInsight/ApplicationLibCode/Commands/SeismicCommands/RicNewSeismicViewFeature.cpp

135 lines
4.6 KiB
C++

/////////////////////////////////////////////////////////////////////////////////
//
// 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;
}