Reflect failed realizations after ert restart

equinor · Apr 12, 2024 · ced2fc2 · ced2fc2
1 parent 4496033
commit ced2fc2
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/src/ert/gui/simulation/evaluate_ensemble_panel.py b/src/ert/gui/simulation/evaluate_ensemble_panel.py
@@ -47,6 +47,7 @@ def __init__(self, ensemble_size: int, run_path: str, notifier: ErtNotifier):
         self._active_realizations_field.setValidator(
             RangeStringArgument(ensemble_size),
         )
+        self._realizations_from_fs()
         layout.addRow("Active realizations", self._active_realizations_field)
 
         self.setLayout(layout)
@@ -55,6 +56,7 @@ def __init__(self, ensemble_size: int, run_path: str, notifier: ErtNotifier):
             self.simulationConfigurationChanged
         )
         self._ensemble_selector.ensemble_populated.connect(self._realizations_from_fs)
+        self._ensemble_selector.currentIndexChanged.connect(self._realizations_from_fs)
 
     def isConfigurationValid(self):
         return self._active_realizations_field.isValid()

diff --git a/src/ert/gui/simulation/multiple_data_assimilation_panel.py b/src/ert/gui/simulation/multiple_data_assimilation_panel.py
@@ -86,6 +86,8 @@ def __init__(
             self._active_realizations_model, "config/simulation/active_realizations"
         )
         self._active_realizations_field.setValidator(RangeStringArgument(ensemble_size))
+        self._ensemble_selector = EnsembleSelector(notifier)
+        self._realizations_from_fs()
         layout.addRow("Active realizations:", self._active_realizations_field)
 
         self._restart_box = QCheckBox("")
@@ -94,8 +96,8 @@ def __init__(
         self._restart_box.setEnabled(False)
         layout.addRow("Restart run:", self._restart_box)
 
-        self._ensemble_selector = EnsembleSelector(notifier)
         self._ensemble_selector.ensemble_populated.connect(self.restart_run_toggled)
+        self._ensemble_selector.currentIndexChanged.connect(self._realizations_from_fs)
         layout.addRow("Restart from:", self._ensemble_selector)
 
         self._target_ensemble_format_field.getValidationSupport().validationChanged.connect(  # noqa
@@ -179,3 +181,11 @@ def getSimulationArguments(self):
 
     def setWeights(self, weights):
         self.weights = str(weights)
+
+    def _realizations_from_fs(self):
+        ensemble = str(self._ensemble_selector.currentText())
+        if ensemble:
+            mask = self.notifier.storage.get_ensemble_by_name(
+                ensemble
+            ).get_realization_mask_with_parameters()
+            self._active_realizations_field.model.setValueFromMask(mask)