Add Sumo summary ensemble and summary case

Add feature to create ensemble from a sumo data source
This commit is contained in:
Magne Sjaastad
2024-07-05 09:06:16 +02:00
parent d751ce4ce0
commit 31d653bce7
28 changed files with 1517 additions and 11 deletions

View File

@@ -0,0 +1,11 @@
set(SOURCE_GROUP_HEADER_FILES
${CMAKE_CURRENT_LIST_DIR}/RimCloudDataSourceCollection.h
)
set(SOURCE_GROUP_SOURCE_FILES
${CMAKE_CURRENT_LIST_DIR}/RimCloudDataSourceCollection.cpp
)
list(APPEND CODE_HEADER_FILES ${SOURCE_GROUP_HEADER_FILES})
list(APPEND CODE_SOURCE_FILES ${SOURCE_GROUP_SOURCE_FILES})

View File

@@ -0,0 +1,233 @@
/////////////////////////////////////////////////////////////////////////////////
//
// Copyright (C) 2024 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 "RimCloudDataSourceCollection.h"
#include "RiaApplication.h"
#include "RimOilField.h"
#include "RimProject.h"
#include "Sumo/RimSummarySumoDataSource.h"
#include "RiuPlotMainWindowTools.h"
#include "cafPdmUiPushButtonEditor.h"
#include "cafPdmUiTreeSelectionEditor.h"
CAF_PDM_SOURCE_INIT( RimCloudDataSourceCollection, "RimCloudDataSourceCollection" );
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RimCloudDataSourceCollection::RimCloudDataSourceCollection()
{
CAF_PDM_InitObject( "Cloud Data", ":/cloud-and-server.svg" );
CAF_PDM_InitFieldNoDefault( &m_sumoFieldName, "SumoFieldId", "Field Id" );
CAF_PDM_InitFieldNoDefault( &m_sumoCaseId, "SumoCaseId", "Case Id" );
m_sumoCaseId.uiCapability()->setUiEditorTypeName( caf::PdmUiTreeSelectionEditor::uiEditorTypeName() );
CAF_PDM_InitFieldNoDefault( &m_sumoEnsembleNames, "SumoEnsembleNames", "Ensembles" );
m_sumoEnsembleNames.uiCapability()->setUiEditorTypeName( caf::PdmUiTreeSelectionEditor::uiEditorTypeName() );
CAF_PDM_InitFieldNoDefault( &m_addEnsembles, "AddEnsembles", "" );
caf::PdmUiPushButtonEditor::configureEditorLabelLeft( &m_addEnsembles );
CAF_PDM_InitFieldNoDefault( &m_sumoDataSources, "SumoDataSources", "Sumo Data Sources" );
m_sumoConnector = RiaApplication::instance()->makeSumoConnector();
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RimCloudDataSourceCollection* RimCloudDataSourceCollection::instance()
{
return RimProject::current()->activeOilField()->cloudDataCollection();
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
std::vector<RimSummarySumoDataSource*> RimCloudDataSourceCollection::sumoDataSources() const
{
return m_sumoDataSources.childrenByType();
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimCloudDataSourceCollection::fieldChangedByUi( const caf::PdmFieldHandle* changedField, const QVariant& oldValue, const QVariant& newValue )
{
if ( changedField == &m_sumoFieldName )
{
m_sumoCaseId = "";
m_sumoEnsembleNames.v().clear();
m_sumoConnector->requestCasesForFieldBlocking( m_sumoFieldName );
}
else if ( changedField == &m_sumoCaseId )
{
m_sumoEnsembleNames.v().clear();
}
if ( changedField == &m_addEnsembles )
{
addEnsemble();
m_addEnsembles = false;
}
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
QList<caf::PdmOptionItemInfo> RimCloudDataSourceCollection::calculateValueOptions( const caf::PdmFieldHandle* fieldNeedingOptions )
{
QList<caf::PdmOptionItemInfo> options;
if ( fieldNeedingOptions == &m_sumoFieldName )
{
if ( m_sumoConnector->assets().empty() )
{
m_sumoConnector->requestAssetsBlocking();
}
for ( const auto& asset : m_sumoConnector->assets() )
{
if ( m_sumoFieldName().isEmpty() )
{
m_sumoFieldName = asset.name;
}
options.push_back( { asset.name, asset.name } );
}
}
else if ( fieldNeedingOptions == &m_sumoCaseId && !m_sumoFieldName().isEmpty() )
{
if ( m_sumoConnector->cases().empty() )
{
m_sumoConnector->requestCasesForFieldBlocking( m_sumoFieldName );
}
for ( const auto& sumoCase : m_sumoConnector->cases() )
{
options.push_back( { sumoCase.name, sumoCase.caseId.get() } );
}
}
else if ( fieldNeedingOptions == &m_sumoEnsembleNames && !m_sumoCaseId().isEmpty() )
{
if ( m_sumoConnector->ensembleNamesForCase( SumoCaseId( m_sumoCaseId ) ).empty() )
{
m_sumoConnector->requestEnsembleByCasesIdBlocking( SumoCaseId( m_sumoCaseId ) );
}
for ( const auto& name : m_sumoConnector->ensembleNamesForCase( SumoCaseId( m_sumoCaseId ) ) )
{
options.push_back( { name, name } );
}
}
return options;
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimCloudDataSourceCollection::defineUiOrdering( QString uiConfigName, caf::PdmUiOrdering& uiOrdering )
{
uiOrdering.add( &m_sumoFieldName );
uiOrdering.add( &m_sumoCaseId );
uiOrdering.add( &m_sumoEnsembleNames );
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimCloudDataSourceCollection::defineEditorAttribute( const caf::PdmFieldHandle* field,
QString uiConfigName,
caf::PdmUiEditorAttribute* attribute )
{
if ( field == &m_addEnsembles )
{
if ( auto attrib = dynamic_cast<caf::PdmUiPushButtonEditorAttribute*>( attribute ) )
{
attrib->m_buttonText = "Add Ensemble(s)";
}
}
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimCloudDataSourceCollection::addEnsemble()
{
RimSummarySumoDataSource* objectToSelect = nullptr;
auto sumoCaseId = SumoCaseId( m_sumoCaseId );
for ( const auto& ensembleName : m_sumoEnsembleNames() )
{
bool createNewDataSource = true;
for ( const auto dataSource : sumoDataSources() )
{
if ( dataSource->caseId() == sumoCaseId && dataSource->ensembleName() == ensembleName )
{
createNewDataSource = false;
break;
}
}
if ( !createNewDataSource )
{
continue;
}
QString caseName;
for ( const auto& sumoCase : m_sumoConnector->cases() )
{
if ( sumoCase.caseId == sumoCaseId )
{
caseName = sumoCase.name;
break;
}
}
m_sumoConnector->requestRealizationIdsForEnsembleBlocking( sumoCaseId, ensembleName );
m_sumoConnector->requestVectorNamesForEnsembleBlocking( sumoCaseId, ensembleName );
auto realizationIds = m_sumoConnector->realizationIds();
auto vectorNames = m_sumoConnector->vectorNames();
auto dataSource = new RimSummarySumoDataSource();
dataSource->setCaseId( sumoCaseId );
dataSource->setCaseName( caseName );
dataSource->setEnsembleName( ensembleName );
dataSource->setRealizationIds( realizationIds );
dataSource->setVectorNames( vectorNames );
dataSource->updateName();
objectToSelect = dataSource;
m_sumoDataSources.push_back( dataSource );
}
uiCapability()->updateAllRequiredEditors();
if ( objectToSelect )
{
RiuPlotMainWindowTools::setExpanded( objectToSelect );
RiuPlotMainWindowTools::selectAsCurrentItem( objectToSelect );
}
}

View File

@@ -0,0 +1,65 @@
/////////////////////////////////////////////////////////////////////////////////
//
// Copyright (C) 2024 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.
//
/////////////////////////////////////////////////////////////////////////////////
#pragma once
#include "cafPdmChildArrayField.h"
#include "cafPdmField.h"
#include "cafPdmObject.h"
#include "Cloud/RiaSumoConnector.h"
#include <QPointer>
#include <QString>
class RimSummarySumoDataSource;
//==================================================================================================
///
///
//==================================================================================================
class RimCloudDataSourceCollection : public caf::PdmObject
{
CAF_PDM_HEADER_INIT;
public:
RimCloudDataSourceCollection();
static RimCloudDataSourceCollection* instance();
std::vector<RimSummarySumoDataSource*> sumoDataSources() const;
private:
void fieldChangedByUi( const caf::PdmFieldHandle* changedField, const QVariant& oldValue, const QVariant& newValue ) override;
QList<caf::PdmOptionItemInfo> calculateValueOptions( const caf::PdmFieldHandle* fieldNeedingOptions ) override;
void defineUiOrdering( QString uiConfigName, caf::PdmUiOrdering& uiOrdering ) override;
void defineEditorAttribute( const caf::PdmFieldHandle* field, QString uiConfigName, caf::PdmUiEditorAttribute* attribute ) override;
void addEnsemble();
private:
caf::PdmField<QString> m_sumoFieldName;
caf::PdmField<QString> m_sumoCaseId;
caf::PdmField<std::vector<QString>> m_sumoEnsembleNames;
caf::PdmField<bool> m_addEnsembles;
caf::PdmChildArrayField<RimSummarySumoDataSource*> m_sumoDataSources;
QPointer<RiaSumoConnector> m_sumoConnector;
};