class MultipleDataAssimilationPanel(SimulationConfigPanel): analysis_module_name = "STD_ENKF" def __init__(self, facade: LibresFacade, notifier: ErtNotifier): SimulationConfigPanel.__init__(self, MultipleDataAssimilation) layout = QFormLayout() case_selector = CaseSelector(facade, notifier) layout.addRow("Current case:", case_selector) run_path_label = QLabel(f"<b>{facade.run_path}</b>") addHelpToWidget(run_path_label, "config/simulation/runpath") layout.addRow("Runpath:", run_path_label) number_of_realizations_label = QLabel( f"<b>{facade.get_ensemble_size()}</b>") addHelpToWidget(number_of_realizations_label, "config/ensemble/num_realizations") layout.addRow(QLabel("Number of realizations:"), number_of_realizations_label) self._target_case_format_model = TargetCaseModel(facade, notifier, format_mode=True) self._target_case_format_field = StringBox( self._target_case_format_model, "config/simulation/target_case_format") self._target_case_format_field.setValidator(ProperNameFormatArgument()) layout.addRow("Target case format:", self._target_case_format_field) self.weights = MultipleDataAssimilation.default_weights self._createInputForWeights(layout) self._iter_field = StringBox( IterValueModel(notifier), "config/simulation/iter_num", ) self._iter_field.setValidator(IntegerArgument(from_value=0), ) layout.addRow("Start iteration:", self._iter_field) self._analysis_module_edit = AnalysisModuleEdit( facade, module_name=MultipleDataAssimilationPanel.analysis_module_name, help_link="config/analysis/analysis_module", ) layout.addRow("Analysis module:", self._analysis_module_edit) self._active_realizations_model = ActiveRealizationsModel(facade) self._active_realizations_field = StringBox( self._active_realizations_model, "config/simulation/active_realizations") self._active_realizations_field.setValidator( RangeStringArgument(facade.get_ensemble_size())) layout.addRow("Active realizations:", self._active_realizations_field) self._target_case_format_field.getValidationSupport( ).validationChanged.connect( # noqa self.simulationConfigurationChanged) self._active_realizations_field.getValidationSupport( ).validationChanged.connect( # noqa self.simulationConfigurationChanged) self._relative_iteration_weights_box.getValidationSupport( ).validationChanged.connect( # noqa self.simulationConfigurationChanged) self.setLayout(layout) def _createInputForWeights(self, layout): relative_iteration_weights_model = ValueModel(self.weights) self._relative_iteration_weights_box = StringBox( relative_iteration_weights_model, help_link="config/simulation/iteration_weights", continuous_update=True, ) self._relative_iteration_weights_box.setValidator( NumberListStringArgument()) layout.addRow("Relative Weights:", self._relative_iteration_weights_box) relative_iteration_weights_model.valueChanged.connect(self.setWeights) normalized_weights_model = ValueModel() normalized_weights_widget = ActiveLabel( normalized_weights_model, help_link="config/simulation/iteration_weights") layout.addRow("Normalized weights:", normalized_weights_widget) def updateVisualizationOfNormalizedWeights(): if self._relative_iteration_weights_box.isValid(): weights = MultipleDataAssimilation.parseWeights( relative_iteration_weights_model.getValue()) normalized_weights = MultipleDataAssimilation.normalizeWeights( weights) normalized_weights_model.setValue(", ".join( f"{x:.2f}" for x in normalized_weights)) else: normalized_weights_model.setValue("The weights are invalid!") self._relative_iteration_weights_box.getValidationSupport( ).validationChanged.connect( # noqa updateVisualizationOfNormalizedWeights) updateVisualizationOfNormalizedWeights( ) # To normalize the default weights def isConfigurationValid(self): return (self._target_case_format_field.isValid() and self._active_realizations_field.isValid() and self._relative_iteration_weights_box.isValid()) def getSimulationArguments(self): return Arguments( mode="es_mda", target_case=self._target_case_format_model.getValue(), realizations=self._active_realizations_field.text(), weights=self.weights, start_iteration=int(self._iter_field.model.getValue()), ) def setWeights(self, weights): str_weights = str(weights) print(f"Weights changed: {str_weights}") self.weights = str_weights
class EnsembleExperimentPanel(SimulationConfigPanel): def __init__(self, ert: EnKFMain, notifier: ErtNotifier): self.ert = ert self.facade = LibresFacade(ert) super().__init__(EnsembleExperiment) self.setObjectName("Ensemble_experiment_panel") layout = QFormLayout() self._case_selector = CaseSelector(self.facade, notifier) layout.addRow("Current case:", self._case_selector) run_path_label = QLabel(f"<b>{self.facade.run_path}</b>") addHelpToWidget(run_path_label, "config/simulation/runpath") layout.addRow("Runpath:", run_path_label) number_of_realizations_label = QLabel( f"<b>{self.facade.get_ensemble_size()}</b>") addHelpToWidget(number_of_realizations_label, "config/ensemble/num_realizations") layout.addRow(QLabel("Number of realizations:"), number_of_realizations_label) self._active_realizations_field = StringBox( ActiveRealizationsModel(self.facade), "config/simulation/active_realizations", ) self._active_realizations_field.setValidator( RangeStringArgument(self.facade.get_ensemble_size()), ) layout.addRow("Active realizations", self._active_realizations_field) self._iter_field = StringBox( IterValueModel(notifier), "config/simulation/iter_num", ) self._iter_field.setValidator(IntegerArgument(from_value=0), ) layout.addRow("Iteration", self._iter_field) self.setLayout(layout) self._active_realizations_field.getValidationSupport( ).validationChanged.connect( # noqa self.simulationConfigurationChanged) self._case_selector.currentIndexChanged.connect( self._realizations_from_fs) self._realizations_from_fs() # update with the current case def isConfigurationValid(self): return self._active_realizations_field.isValid( ) and self._iter_field.isValid() def getSimulationArguments(self): return Arguments( mode="ensemble_experiment", iter_num=int(self._iter_field.text()), realizations=self._active_realizations_field.text(), ) def _realizations_from_fs(self): case = str(self._case_selector.currentText()) mask = get_runnable_realizations_mask(self.ert, case) self._active_realizations_field.model.setValueFromMask(mask)
class IteratedEnsembleSmootherPanel(SimulationConfigPanel): analysis_module_name = "IES_ENKF" def __init__(self, facade: LibresFacade, notifier: ErtNotifier): self.facade = facade self.notifier = notifier SimulationConfigPanel.__init__(self, IteratedEnsembleSmoother) layout = QFormLayout() case_selector = CaseSelector(self.facade, notifier) layout.addRow("Current case:", case_selector) run_path_label = QLabel(f"<b>{self.facade.run_path}</b>") addHelpToWidget(run_path_label, "config/simulation/runpath") layout.addRow("Runpath:", run_path_label) number_of_realizations_label = QLabel( f"<b>{self.facade.get_ensemble_size()}</b>") addHelpToWidget(number_of_realizations_label, "config/ensemble/num_realizations") layout.addRow(QLabel("Number of realizations:"), number_of_realizations_label) # The num_iterations_spinner does not track any external changes (will # that ever happen?) self._num_iterations_spinner = QSpinBox() self._num_iterations_spinner.setMinimum(1) self._num_iterations_spinner.setMaximum(100) self._num_iterations_spinner.setValue( self.facade.get_number_of_iterations()) addHelpToWidget(self._num_iterations_spinner, "config/simulation/number_of_iterations") self._num_iterations_spinner.valueChanged[int].connect( self.setNumberIterations) layout.addRow("Number of iterations:", self._num_iterations_spinner) self._iterated_target_case_format_model = TargetCaseModel( self.facade, notifier, format_mode=True) self._iterated_target_case_format_field = StringBox( self._iterated_target_case_format_model, "config/simulation/iterated_target_case_format", ) self._iterated_target_case_format_field.setValidator( ProperNameFormatArgument()) layout.addRow("Target case format:", self._iterated_target_case_format_field) self._analysis_module_edit = AnalysisModuleEdit( self.facade, module_name=IteratedEnsembleSmootherPanel.analysis_module_name, help_link="config/analysis/analysis_module", ) layout.addRow("Analysis module:", self._analysis_module_edit) self._active_realizations_model = ActiveRealizationsModel(self.facade) self._active_realizations_field = StringBox( self._active_realizations_model, "config/simulation/active_realizations") self._active_realizations_field.setValidator( RangeStringArgument(self.facade.get_ensemble_size())) layout.addRow("Active realizations", self._active_realizations_field) self._iterated_target_case_format_field.getValidationSupport( ).validationChanged.connect( # noqa self.simulationConfigurationChanged) self._active_realizations_field.getValidationSupport( ).validationChanged.connect( # noqa self.simulationConfigurationChanged) self.setLayout(layout) def setNumberIterations(self, iteration_count): if iteration_count != self.facade.get_number_of_iterations(): self.facade.get_analysis_config().getAnalysisIterConfig( ).setNumIterations(iteration_count) self.notifier.emitErtChange() def isConfigurationValid(self): return (self._iterated_target_case_format_field.isValid() and self._active_realizations_field.isValid()) def getSimulationArguments(self): return Arguments( mode="iterative_ensemble_smoother", target_case=self._iterated_target_case_format_model.getValue(), realizations=self._active_realizations_field.text(), num_iterations=self._num_iterations_spinner.value(), )
class EnsembleSmootherPanel(SimulationConfigPanel): analysis_module_name = "STD_ENKF" def __init__(self, ert: EnKFMain, notifier: ErtNotifier): super().__init__(EnsembleSmoother) self.ert = ert facade = LibresFacade(ert) layout = QFormLayout() self._case_selector = CaseSelector(facade, notifier) layout.addRow("Current case:", self._case_selector) run_path_label = QLabel(f"<b>{facade.run_path}</b>") addHelpToWidget(run_path_label, "config/simulation/runpath") layout.addRow("Runpath:", run_path_label) number_of_realizations_label = QLabel( f"<b>{facade.get_ensemble_size()}</b>") addHelpToWidget(number_of_realizations_label, "config/ensemble/num_realizations") layout.addRow(QLabel("Number of realizations:"), number_of_realizations_label) self._target_case_model = TargetCaseModel(facade, notifier) self._target_case_field = StringBox(self._target_case_model, "config/simulation/target_case") self._target_case_field.setValidator(ProperNameArgument()) layout.addRow("Target case:", self._target_case_field) self._analysis_module_edit = AnalysisModuleEdit( facade, module_name=EnsembleSmootherPanel.analysis_module_name, help_link="config/analysis/analysis_module", ) layout.addRow("Analysis module:", self._analysis_module_edit) active_realizations_model = ActiveRealizationsModel(facade) self._active_realizations_field = StringBox( active_realizations_model, "config/simulation/active_realizations") self._active_realizations_field.setValidator( RangeStringArgument(facade.get_ensemble_size())) layout.addRow("Active realizations", self._active_realizations_field) self.setLayout(layout) self._target_case_field.getValidationSupport( ).validationChanged.connect(self.simulationConfigurationChanged) self._active_realizations_field.getValidationSupport( ).validationChanged.connect( # noqa self.simulationConfigurationChanged) self._case_selector.currentIndexChanged.connect( self._realizations_from_fs) self._realizations_from_fs() # update with the current case def isConfigurationValid(self): return (self._target_case_field.isValid() and self._active_realizations_field.isValid()) def getSimulationArguments(self): arguments = Arguments( mode="ensemble_smoother", target_case=self._target_case_model.getValue(), realizations=self._active_realizations_field.text(), ) return arguments def _realizations_from_fs(self): case = str(self._case_selector.currentText()) mask = get_runnable_realizations_mask(self.ert, case) self._active_realizations_field.model.setValueFromMask(mask)