#8370: Add Project.summary_ensembles and summary_plot_collection

ApplicationLibCode/ProjectDataModelCommands/RimcProject.cpp

@@ -28,6 +28,7 @@
 #include "RiaPreferencesGrid.h"
 #include "RiaQStringFormatter.h"
 #include "RiaRegressionTestRunner.h"
+#include "Summary/RiaSummaryTools.h"
 
 #include "ExportCommands/RicSnapshotAllPlotsToFileFeature.h"
 #include "ExportCommands/RicSnapshotAllViewsToFileFeature.h"
@@ -54,6 +55,7 @@
 #include "RimProject.h"
 #include "RimSummaryCase.h"
 #include "RimSummaryCaseMainCollection.h"
+#include "RimSummaryEnsemble.h"
 #include "RimSurfaceCollection.h"
 #include "RimTools.h"
 #include "RimValveTemplateCollection.h"
@@ -132,6 +134,64 @@ QString RimProject_importSummaryCase::classKeywordReturnedType() const
     return RimFileSummaryCase::classKeywordStatic();
 }
 
+CAF_PDM_OBJECT_METHOD_SOURCE_INIT( RimProject, RimProject_importSummaryEnsemble, "importSummaryEnsemble" );
+
+//--------------------------------------------------------------------------------------------------
+///
+//--------------------------------------------------------------------------------------------------
+RimProject_importSummaryEnsemble::RimProject_importSummaryEnsemble( caf::PdmObjectHandle* self )
+    : PdmObjectCreationMethod( self )
+{
+    CAF_PDM_InitObject( "Import Summary Ensemble", "", "", "Import summary files as one ensemble" );
+
+    CAF_PDM_InitScriptableFieldNoDefault( &m_fileNames, "FileNames", "", "", "", "Summary files (SMSPEC), one for each realization" );
+    CAF_PDM_InitScriptableFieldNoDefault( &m_ensembleName, "EnsembleName", "", "", "", "Ensemble name. Automatically generated if empty" );
+}
+
+//--------------------------------------------------------------------------------------------------
+///
+//--------------------------------------------------------------------------------------------------
+std::expected<caf::PdmObjectHandle*, QString> RimProject_importSummaryEnsemble::execute()
+{
+    QStringList summaryFileNames;
+    for ( const QString& fileName : m_fileNames() )
+    {
+        QString absolutePath = fileName;
+        if ( !QFileInfo( absolutePath ).exists() )
+        {
+            absolutePath = QDir( RiaApplication::instance()->startDir() ).absoluteFilePath( fileName );
+        }
+        summaryFileNames.push_back( absolutePath );
+    }
+
+    if ( summaryFileNames.isEmpty() ) return std::unexpected( "No summary files specified." );
+
+    RiaEnsembleImportTools::CreateConfig createConfig{ .fileType = RiaDefines::FileType::SMSPEC, .ensembleOrGroup = true, .allowDialogs = false };
+    auto newCases = RiaEnsembleImportTools::createSummaryCasesFromFiles( summaryFileNames, createConfig );
+    if ( newCases.empty() ) return std::unexpected( "Unable to import any of the summary files." );
+
+    auto* mainCollection = RiaSummaryTools::summaryCaseMainCollection();
+    if ( !mainCollection ) return std::unexpected( "No summary case collection found." );
+
+    auto ensemble = mainCollection->addEnsemble( newCases, m_ensembleName(), true );
+
+    // The name template is only used when auto naming is disabled
+    if ( !m_ensembleName().isEmpty() ) ensemble->setAutoName( false );
+    RiaSummaryTools::updateSummaryEnsembleNames();
+
+    mainCollection->updateConnectedEditors();
+
+    return ensemble;
+}
+
+//--------------------------------------------------------------------------------------------------
+///
+//--------------------------------------------------------------------------------------------------
+QString RimProject_importSummaryEnsemble::classKeywordReturnedType() const
+{
+    return RimSummaryEnsemble::classKeywordStatic();
+}
+
 CAF_PDM_OBJECT_METHOD_SOURCE_INIT( RimProject, RimProject_summaryCase, "summaryCase" );
 
 //--------------------------------------------------------------------------------------------------

ApplicationLibCode/ProjectDataModelCommands/RimcProject.h

@@ -52,6 +52,24 @@ private:
     caf::PdmField<QString> m_fileName;
 };
 
+//==================================================================================================
+///
+//==================================================================================================
+class RimProject_importSummaryEnsemble : public caf::PdmObjectCreationMethod
+{
+    CAF_PDM_HEADER_INIT;
+
+public:
+    RimProject_importSummaryEnsemble( caf::PdmObjectHandle* self );
+
+    std::expected<caf::PdmObjectHandle*, QString> execute() override;
+    QString                                       classKeywordReturnedType() const override;
+
+private:
+    caf::PdmField<std::vector<QString>> m_fileNames;
+    caf::PdmField<QString>              m_ensembleName;
+};
+
 //==================================================================================================
 ///
 //==================================================================================================