SIM RFC: AI selector interface (UBC-Thunderbots#3568)

* feat: ai selector interface

* refactor: use RuntimeManager Interface

* fix: update names

* [pre-commit.ci lite] apply automatic fixes

---------

Co-authored-by: pre-commit-ci-lite[bot] <117423508+pre-commit-ci-lite[bot]@users.noreply.github.com>

Author: a-png129

src/software/thunderscope/binary_context_managers/runtime_manager.py

@@ -29,7 +29,7 @@ def fetch_installed_runtimes(self) -> list[str]:
         """
         return self.runtime_loader.fetch_installed_runtimes()
 
-    def load_existing_runtime(self, yellow_runtime: str, blue_runtime: str) -> None:
+    def load_existing_runtimes(self, yellow_runtime: str, blue_runtime: str) -> None:
         """Loads the runtimes of the specified name or throws an error upon failure.
         :param blue_runtime: name of the blue runtime to load
         :param yellow_runtime: name of the yellow runtime to load

src/software/thunderscope/gl/widgets/BUILD

@@ -43,6 +43,7 @@ py_library(
     srcs = ["gl_gamecontroller_toolbar.py"],
     deps = [
         ":gl_runtime_installer",
+        ":gl_runtime_selector",
         ":gl_toolbar",
         "//proto:import_all_protos",
         "//software/networking:ssl_proto_communication",
@@ -52,3 +53,11 @@ py_library(
         requirement("qtawesome"),
     ],
 )
+
+py_library(
+    name = "gl_runtime_selector",
+    srcs = ["gl_runtime_selector.py"],
+    deps = [
+        requirement("pyqtgraph"),
+    ],
+)

src/software/thunderscope/gl/widgets/gl_gamecontroller_toolbar.py

@@ -4,6 +4,10 @@
 from proto.ssl_gc_common_pb2 import Team as SslTeam
 from typing import Callable, override
 import webbrowser
+from software.thunderscope.binary_context_managers.runtime_manager import (
+    runtime_manager_instance,
+)
+from software.thunderscope.gl.widgets.gl_runtime_selector import GLRuntimeSelectorDialog
 from software.thunderscope.gl.widgets.gl_toolbar import GLToolbar
 from software.thunderscope.proto_unix_io import ProtoUnixIO
 from software.thunderscope.gl.widgets.gl_runtime_installer import (
@@ -105,6 +109,13 @@ def __init__(
             display_text="Install Runtimes",
         )
 
+        self.runtime_selector_button = self.__setup_icon_button(
+            qta.icon("mdi6.server"),
+            "Select runtimes for each team",
+            self.__open_runtime_selector_dialog,
+            display_text="Select Runtimes",
+        )
+
         # disable the normal start button when no play is selected
         self.normal_start_enabled = True
         self.__toggle_normal_start_button()
@@ -122,6 +133,7 @@ def __init__(
         self.layout().addStretch()
         self.__add_separator(self.layout())
         self.layout().addWidget(self.runtime_installer_button)
+        self.layout().addWidget(self.runtime_selector_button)
 
     @override
     def refresh(self) -> None:
@@ -261,11 +273,24 @@ def __send_gc_command(self, command: Command.Type, team: Team) -> None:
         command = ManualGCCommand(manual_command=Command(type=command, for_team=team))
         self.proto_unix_io.send_proto(ManualGCCommand, command)
 
-    def __open_runtime_installer_dialog(self):
+    def __open_runtime_installer_dialog(self) -> None:
         """Opens the runtime installer modal, initializing if first time"""
         if not hasattr(self, "runtime_installer_dialog"):
             self.runtime_installer_dialog = GLRuntimeInstallerDialog(
                 parent=self.parent()
             )
 
         self.runtime_installer_dialog.show()
+
+    def __open_runtime_selector_dialog(self) -> None:
+        """Opens the runtime selector dialog, initializing if first time"""
+        if not hasattr(self, "runtime_selector_dialog"):
+            options = ["Current Commit"]
+            options.extend(runtime_manager_instance.fetch_installed_runtimes())
+            self.runtime_selector_dialog = GLRuntimeSelectorDialog(
+                parent=self.parent(),
+                runtime_options=options,
+                on_runtimes_selected=runtime_manager_instance.load_existing_runtimes,
+            )
+
+        self.runtime_selector_dialog.show()

src/software/thunderscope/gl/widgets/gl_runtime_selector.py

@@ -0,0 +1,92 @@
+from pyqtgraph.Qt.QtWidgets import (
+    QDialog,
+    QWidget,
+    QVBoxLayout,
+    QHBoxLayout,
+    QLabel,
+    QPushButton,
+    QComboBox,
+)
+from typing import Callable
+
+
+class GLRuntimeSelectorDialog(QDialog):
+    """Modal that displays the selectable list of runtimes for yellow and blue teams"""
+
+    def __init__(
+        self,
+        parent: QWidget,
+        runtime_options: list[str],
+        on_runtimes_selected: Callable[[str, str], None],
+    ):
+        """Initializes the runtime selector modal, displaying the same list of installed
+        runtimes for both the blue and yellow teams.
+
+        :param parent: the modal's parent
+        :param runtime_options: the list of runtime options to display in both menus
+        :param on_runtimes_selected: the callback for runtime selection
+        """
+        super().__init__(parent)
+
+        self.on_runtimes_selected = on_runtimes_selected
+
+        self.setWindowTitle("Select Runtimes")
+        self.setModal(True)
+        self.setMinimumWidth(400)
+
+        layout = QVBoxLayout(self)
+
+        # Yellow runtime
+        layout.addWidget(QLabel("<b>Yellow Runtime</b>"))
+        self.yellow_menu = QComboBox()
+        self.yellow_menu.addItems(runtime_options)
+        self.yellow_menu.currentTextChanged.connect(self._on_yellow_changed)
+        self._yellow_selection = self.yellow_menu.currentText()
+        layout.addWidget(self.yellow_menu)
+
+        # Blue runtime
+        layout.addSpacing(10)
+        layout.addWidget(QLabel("<b>Blue Runtime</b>"))
+        self.blue_menu = QComboBox()
+        self.blue_menu.addItems(runtime_options)
+        self.blue_menu.currentTextChanged.connect(self._on_blue_changed)
+        self._blue_selection = self.blue_menu.currentText()
+        layout.addWidget(self.blue_menu)
+
+        # Restart note
+        layout.addSpacing(15)
+        restart_note = QLabel(
+            "<i>Note: Restart Thunderscope for changes to take effect.</i>"
+        )
+        layout.addWidget(restart_note)
+
+        # Done button
+        layout.addSpacing(15)
+        button_row = QHBoxLayout()
+        button_row.addStretch()
+
+        done_button = QPushButton("Done")
+        done_button.clicked.connect(self._on_done)
+        button_row.addWidget(done_button)
+
+        layout.addLayout(button_row)
+
+    def _on_yellow_changed(self, value: str) -> None:
+        """Stores currently selected runtime for yellow team
+
+        :param value: the value of the selected option
+        """
+        self._yellow_selection = value
+
+    def _on_blue_changed(self, value: str) -> None:
+        """Stores currently selected runtime for blue team
+
+        :param value: the value of the selected option
+        """
+        self._blue_selection = value
+
+    def _on_done(self) -> None:
+        """Commits the selected runtimes and closes the modal."""
+        self.on_runtimes_selected(self._yellow_selection, self._blue_selection)
+
+        self.close()