Add well target generator cases to RimProject::allCases.

OPM · Dec 20, 2024 · b92109f · b92109f
1 parent 2f76412
commit b92109f
Show file tree

Hide file tree

Showing 5 changed files with 26 additions and 1 deletion.
diff --git a/ApplicationLibCode/ProjectDataModel/RimEclipseCaseEnsemble.cpp b/ApplicationLibCode/ProjectDataModel/RimEclipseCaseEnsemble.cpp
@@ -116,6 +116,14 @@ std::vector<RimEclipseCase*> RimEclipseCaseEnsemble::cases() const
     return m_caseCollection->reservoirs.childrenByType();
 }
 
+//--------------------------------------------------------------------------------------------------
+///
+//--------------------------------------------------------------------------------------------------
+std::vector<RimWellTargetCandidatesGenerator*> RimEclipseCaseEnsemble::wellTargetsGenerators() const
+{
+    return m_wellTargetGenerators.childrenByType();
+}
+
 //--------------------------------------------------------------------------------------------------
 ///
 //--------------------------------------------------------------------------------------------------

diff --git a/ApplicationLibCode/ProjectDataModel/RimEclipseCaseEnsemble.h b/ApplicationLibCode/ProjectDataModel/RimEclipseCaseEnsemble.h
@@ -56,7 +56,8 @@ class RimEclipseCaseEnsemble : public RimNamedObject
 
     RimEclipseViewCollection* viewCollection() const;
 
-    void addWellTargetsGenerator( RimWellTargetCandidatesGenerator* generator );
+    void                                           addWellTargetsGenerator( RimWellTargetCandidatesGenerator* generator );
+    std::vector<RimWellTargetCandidatesGenerator*> wellTargetsGenerators() const;
 
     void addStatisticsContourMap( RimStatisticsContourMap* statisticsContourMap );
 

diff --git a/ApplicationLibCode/ProjectDataModel/RimProject.cpp b/ApplicationLibCode/ProjectDataModel/RimProject.cpp
@@ -102,6 +102,7 @@
 #include "RimWellPath.h"
 #include "RimWellPathCollection.h"
 
+#include "RimWellTargetCandidatesGenerator.h"
 #include "VerticalFlowPerformance/RimVfpDataCollection.h"
 #include "VerticalFlowPerformance/RimVfpPlotCollection.h"
 
@@ -605,6 +606,11 @@ std::vector<RimCase*> RimProject::allGridCases() const
                 {
                     cases.push_back( acase );
                 }
+
+                for ( RimWellTargetCandidatesGenerator* generator : ensemble->wellTargetsGenerators() )
+                {
+                    if ( auto ensembleStatisticsCase = generator->ensembleStatisticsCase() ) cases.push_back( ensembleStatisticsCase );
+                }
             }
 
             RimGeoMechModels* geomModels = oilField->geoMechModels();

diff --git a/ApplicationLibCode/ProjectDataModel/RimWellTargetCandidatesGenerator.cpp b/ApplicationLibCode/ProjectDataModel/RimWellTargetCandidatesGenerator.cpp
@@ -423,3 +423,11 @@ void RimWellTargetCandidatesGenerator::updateResultDefinition()
     RimEclipseCase* eclipseCase = firstCase();
     if ( eclipseCase ) m_resultDefinition->setEclipseCase( eclipseCase );
 }
+
+//--------------------------------------------------------------------------------------------------
+///
+//--------------------------------------------------------------------------------------------------
+RimEclipseCase* RimWellTargetCandidatesGenerator::ensembleStatisticsCase() const
+{
+    return m_ensembleStatisticsCase;
+}
diff --git a/ApplicationLibCode/ProjectDataModel/RimWellTargetCandidatesGenerator.h b/ApplicationLibCode/ProjectDataModel/RimWellTargetCandidatesGenerator.h
@@ -42,6 +42,8 @@ class RimWellTargetCandidatesGenerator : public caf::PdmObject
 
     void updateResultDefinition();
 
+    RimEclipseCase* ensembleStatisticsCase() const;
+
 protected:
     void fieldChangedByUi( const caf::PdmFieldHandle* changedField, const QVariant& oldValue, const QVariant& newValue ) override;
     void defineEditorAttribute( const caf::PdmFieldHandle* field, QString uiConfigName, caf::PdmUiEditorAttribute* attribute ) override;