Files
ResInsight/ApplicationLibCode/Commands/EnsembleFileSetCommands/RicCreateReservoirGridEnsembleFromFileSetFeature.cpp
T

129 lines
4.7 KiB
C++

/////////////////////////////////////////////////////////////////////////////////
//
// Copyright (C) 2026 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 "RicCreateReservoirGridEnsembleFromFileSetFeature.h"
#include "RiaLogging.h"
#include "RiaQStringFormatter.h"
#include "EnsembleFileSet/RimEnsembleFileSet.h"
#include "RimEclipseCaseCollection.h"
#include "RimEclipseView.h"
#include "RimOilField.h"
#include "RimProject.h"
#include "RimReservoirGridEnsemble.h"
#include "Riu3DMainWindowTools.h"
#include "cafProgressInfo.h"
#include "cafSelectionManagerTools.h"
#include <QAction>
#include <QFileInfo>
CAF_CMD_SOURCE_INIT( RicCreateReservoirGridEnsembleFromFileSetFeature, "RicCreateReservoirGridEnsembleFromFileSetFeature" );
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
bool RicCreateReservoirGridEnsembleFromFileSetFeature::isCommandEnabled() const
{
return !caf::selectedObjectsByType<RimEnsembleFileSet*>().empty();
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RicCreateReservoirGridEnsembleFromFileSetFeature::onActionTriggered( bool isChecked )
{
std::vector<RimEnsembleFileSet*> selectedFileSets = caf::selectedObjectsByType<RimEnsembleFileSet*>();
if ( selectedFileSets.empty() ) return;
RimProject* project = RimProject::current();
RimEclipseCaseCollection* eclipseCaseColl = project->activeOilField()->analysisModels();
for ( RimEnsembleFileSet* fileSet : selectedFileSets )
{
// Get grid file paths from the file set
QStringList gridFiles = fileSet->createPaths( ".EGRID" );
if ( gridFiles.empty() )
{
gridFiles = fileSet->createPaths( ".GRID" );
}
// Filter out non-existing files
QStringList existingGridFiles;
for ( const QString& filePath : gridFiles )
{
if ( QFileInfo::exists( filePath ) )
{
existingGridFiles.append( filePath );
}
else
{
RiaLogging::warning( std::format( "Grid file does not exist: {}", filePath ) );
}
}
gridFiles = existingGridFiles;
if ( gridFiles.empty() )
{
RiaLogging::warning( std::format( "No existing grid files found for ensemble '{}'", fileSet->name() ) );
continue;
}
// Always create RimReservoirGridEnsemble
RiaLogging::info(
QString( "Creating Reservoir Grid Ensemble for '%1' with %2 grid files." ).arg( fileSet->name() ).arg( gridFiles.size() ).toStdString() );
RimReservoirGridEnsemble* gridEnsemble = new RimReservoirGridEnsemble();
gridEnsemble->setEnsembleFileSet( fileSet );
gridEnsemble->createGridCasesFromEnsembleFileSet();
project->assignIdToCaseGroup( gridEnsemble );
eclipseCaseColl->reservoirGridEnsembles.push_back( gridEnsemble );
gridEnsemble->loadDataAndUpdate();
// Create view for first case if available
auto allCases = gridEnsemble->cases();
if ( !allCases.empty() )
{
RimEclipseView* view = gridEnsemble->addViewForCase( allCases[0] );
if ( view )
{
view->loadDataAndUpdate();
}
}
Riu3DMainWindowTools::selectAsCurrentItem( gridEnsemble );
}
eclipseCaseColl->updateAllRequiredEditors();
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RicCreateReservoirGridEnsembleFromFileSetFeature::setupActionLook( QAction* actionToSetup )
{
actionToSetup->setIcon( QIcon( ":/GridCaseGroup16x16.png" ) );
actionToSetup->setText( "Create Reservoir Grid Ensemble" );
}