Added button for simulated experiments

Author: andreas-el

src/ert/gui/main_window.py

@@ -1,10 +1,11 @@
 from __future__ import annotations
 
+import datetime
 import functools
 import webbrowser
 from typing import TYPE_CHECKING, Dict, Optional
 
-from qtpy.QtCore import QSize, Qt, Signal
+from qtpy.QtCore import QSize, Qt, Signal, Slot
 from qtpy.QtGui import QCloseEvent, QCursor, QIcon
 from qtpy.QtWidgets import (
     QAction,
@@ -22,6 +23,7 @@
 from ert.gui.ertnotifier import ErtNotifier
 from ert.gui.find_ert_info import find_ert_info
 from ert.gui.simulation import ExperimentPanel
+from ert.gui.simulation.run_dialog import RunDialog
 from ert.gui.tools.event_viewer import EventViewerTool, GUILogHandler
 from ert.gui.tools.export import ExportTool
 from ert.gui.tools.load_results import LoadResultsTool
@@ -80,8 +82,8 @@ def __init__(
         self.log_handler = log_handler
 
         self.setWindowTitle(f"ERT - {config_file} - {find_ert_info()}")
+        self.dialog_panels = []
         self.central_panels = []
-        self.central_panels_index = 0
 
         self.plugin_manager = plugin_manager
         self.central_widget = QFrame(self)
@@ -105,8 +107,16 @@ def __init__(
             self.notifier,
             self.ert_config.model_config.num_realizations,
         )
+
         self._create_sidebar_button(self._manage_experiments_tool)
 
+        self.results_button = self._create_sidebar_button()
+        self.results_button.setIcon(QIcon("img:in_progress.svg"))
+        self.results_button.setToolTip("Show Results")
+        self.results_button.setEnabled(False)
+        menu = QMenu()
+        self.results_button.setMenu(menu)
+
         self.vbox_layout.addStretch()
         self.central_layout.addWidget(self.side_frame)
 
@@ -117,6 +127,43 @@ def __init__(
         self.__add_tools_menu()
         self.__add_help_menu()
 
+    @Slot(object)
+    def slot_add_widget(self, run_dialog: RunDialog):
+        print("hello!")
+        for widget in self.central_panels:
+            widget.setVisible(False)
+
+        self.dialog_panels.append(run_dialog)
+
+        run_dialog.setParent(self)
+        self.central_layout.addWidget(run_dialog)
+        self.results_button.setEnabled(True)
+        date_time = datetime.datetime.utcnow().strftime("%Y-%d-%m %H:%M:%S")
+        act = self.results_button.menu().addAction(date_time)
+        act.setProperty("index", len(self.dialog_panels) - 1)
+        act.triggered.connect(self.select_dialog_panel)
+
+    def select_dialog_panel(self):
+        actor = self.sender()
+        index = int(actor.property("index"))
+
+        for w in self.central_panels:
+            w.hide()
+
+        for i in range(len(self.dialog_panels)):
+            should_be_visible = i == index
+            self.dialog_panels[i].setVisible(should_be_visible)
+
+    def select_widget(self) -> None:
+        index = 0
+
+        for w in self.dialog_panels:
+            w.hide()
+
+        for i in range(len(self.central_panels)):
+            should_be_visible = i == index
+            self.central_panels[i].setVisible(should_be_visible)
+
     def post_init(self):
         experiment_panel = ExperimentPanel(
             self.ert_config,
@@ -127,6 +174,8 @@ def post_init(self):
         self.central_layout.addWidget(experiment_panel)
         self.central_panels.append(experiment_panel)
 
+        experiment_panel.experiment_started.connect(self.slot_add_widget)
+
         plugin_handler = PluginHandler(
             self.notifier,
             [wfj for wfj in self.ert_config.workflow_jobs.values() if wfj.is_plugin()],
@@ -150,23 +199,27 @@ def _create_sidebar_button(self, tool: Optional[Tool] = None) -> QPushButton:
             button.setToolTip(tool.getName())
             button.clicked.connect(tool.trigger)
         self.vbox_layout.addWidget(button)
-        button.setProperty("INDEX", self.central_panels_index)
-        self.central_panels_index += 1
+        button.setProperty("index", len(self.central_panels) - 1)
         return button
 
     def add_experiment_button(self) -> None:
         button = self._create_sidebar_button()
         button.setIcon(QIcon("img:play_circle_outlined.svg"))
         button.setToolTip("Start Simulation")
-        button.clicked.connect(self.toggle_visibility)
+        button.clicked.connect(self.select_widget)
 
         menu = QMenu()
-        menu.addAction("Single Test Run")
-        menu.addAction("Ensemble Experiment")
-        menu.addAction("Manual Update")
-        menu.addAction("ES MDA")
-        menu.addAction("Ensemble Smoother")
         menu.setStyleSheet(MENU_ITEM_STYLE_SHEET)
+
+        for sim_mode in [
+            "Single test run",
+            "Ensemble Experiment",
+            "Manual Update",
+            "ES MDA",
+            "Ensemble Smoother",
+        ]:
+            act = menu.addAction(sim_mode)
+            act.triggered.connect(self.select_widget)
         button.setMenu(menu)
 
     def toggle_visibility(self) -> None:

src/ert/gui/simulation/experiment_panel.py

@@ -211,9 +211,6 @@ def getExperimentName(self) -> str:
         """Get the experiment name as provided by the user. Defaults to run mode if not set."""
         return self.get_experiment_arguments().experiment_name
 
-    # def set_main_frame(self, main_frame: ErtMainWindow) -> None:
-    #     self._main_frame = main_frame
-
     def run_experiment(self) -> None:
         args = self.get_experiment_arguments()
         QApplication.setOverrideCursor(Qt.CursorShape.WaitCursor)
@@ -301,13 +298,8 @@ def run_experiment(self) -> None:
         self.experiment_started.emit(dialog)
         dialog.produce_clipboard_debug_info.connect(self.populate_clipboard_debug_info)
         dialog.run_experiment()
-
-        # if self._main_frame:
-        #     dialog.setParent(self._main_frame)
-        #     self._main_frame.setWidget(dialog)
-
-        dialog.show()
-        self.hide()
+        # dialog.show()
+        # self.hide()
 
         def exit_handler() -> None:
             self.run_button.setText(EXPERIMENT_READY_TO_RUN_BUTTON_MESSAGE)

src/ert/gui/tools/manage_experiments/manage_experiments_tool.py

@@ -2,7 +2,6 @@
 
 from typing import TYPE_CHECKING, Optional
 
-from qtpy.QtCore import Qt
 from qtpy.QtGui import QIcon
 
 from ert.gui.tools import Tool
@@ -26,8 +25,5 @@ def trigger(self) -> None:
             self._manage_experiments_panel = ManageExperimentsPanel(
                 self.ert_config, self.notifier, self.ensemble_size
             )
-            self._manage_experiments_panel.setWindowModality(
-                Qt.WindowModality.ApplicationModal
-            )
 
         self._manage_experiments_panel.show()