Merge pull request #186 from CWRUbotix/carp-animation

Carp animation

Author: benjaminwp18

src/surface/gui/gui/images/CWRUbotix Logo.png renamed to src/surface/gui/gui/images/cwrubotix_logo.png

@@ -7,6 +7,7 @@
 from gui.widgets.heartbeat import HeartbeatWidget
 from gui.widgets.ip_widget import IPWidget
 from gui.widgets.logger import Logger
+from gui.widgets.tabs.carp_model_tab import CarpModelTab
 from gui.widgets.tabs.general_debug_tab import GeneralDebugTab
 from gui.widgets.tabs.photosphere_tab import PhotosphereTab
 from gui.widgets.tabs.shipwreck import ShipwreckTab
@@ -22,7 +23,7 @@ class OperatorApp(App):
     def __init__(self) -> None:
         super().__init__('operator_gui_node')
 
-        self.setWindowTitle('Operator GUI - CWRUbotix ROV 2024')
+        self.setWindowTitle('Operator GUI - CWRUbotix ROV 2025')
 
         # Main tab
         main_tab = QWidget()
@@ -56,6 +57,7 @@ def __init__(self) -> None:
         self.tabs.addTab(main_tab, 'Main')
         self.tabs.addTab(GeneralDebugTab(), 'General Debug')
         self.tabs.addTab(PhotosphereTab(), 'Photosphere')
+        self.tabs.addTab(CarpModelTab(), 'Carp Model')
         self.shipwreck_tab = ShipwreckTab()
         self.tabs.addTab(self.shipwreck_tab, SHIPWRECK_TEXT)
         self.tabs.currentChanged.connect(self.changed_tabs)

src/surface/gui/gui/images/illinois_river_map_0.png

@@ -0,0 +1,105 @@
+from pathlib import Path
+
+from ament_index_python.packages import get_package_share_directory
+from PyQt6.QtCore import Qt, QTimer
+from PyQt6.QtGui import QColor, QFont, QPainter, QPen, QPixmap
+from PyQt6.QtWidgets import QComboBox, QHBoxLayout, QLabel, QPushButton, QVBoxLayout, QWidget
+
+START_YEAR = 2016
+END_YEAR = 2025
+
+
+class AnimationLabel(QLabel):
+    def __init__(self, year_list: list[int], interval: int = 2000) -> None:
+        super().__init__()
+        self.year = START_YEAR
+        self.year_list = year_list
+
+        map_path = str(
+            Path(get_package_share_directory('gui')) / 'images' / 'illinois_river_map_0.png'
+        )
+        map_pixmap = QPixmap(map_path)
+        self.setPixmap(
+            map_pixmap.scaled(
+                750,
+                750,
+                Qt.AspectRatioMode.KeepAspectRatio,
+                Qt.TransformationMode.SmoothTransformation,
+            )
+        )
+
+        self.timer = QTimer(self)
+        self.timer.timeout.connect(self.update_frame)
+        self.timer.start(interval)  # in milliseconds
+
+    def update_frame(self) -> None:
+        region = 0
+        for year in self.year_list:
+            if self.year >= year:
+                region += 1
+
+        map_path = str(
+            Path(get_package_share_directory('gui'))
+            / 'images'
+            / ('illinois_river_map_' + str(region) + '.png')
+        )
+
+        map_pixmap = QPixmap(map_path)
+
+        painter = QPainter(map_pixmap)
+        pen = QPen(QColor('black'))
+        painter.setFont(QFont('Arial', 50))
+        painter.setPen(pen)
+        painter.drawText(20, 60, str(self.year))
+        painter.end()
+        self.setPixmap(
+            map_pixmap.scaled(
+                750,
+                750,
+                Qt.AspectRatioMode.KeepAspectRatio,
+                Qt.TransformationMode.SmoothTransformation,
+            )
+        )
+
+        self.year += 1
+        self.year = min(self.year, END_YEAR)
+
+
+class CarpAnimation(QWidget):
+    def __init__(self) -> None:
+        super().__init__()
+
+        self.root_layout = QHBoxLayout()
+
+        input_layout = QVBoxLayout()
+
+        self.dropdown_list: list[QComboBox] = []
+
+        for _ in range(5):
+            dropdown = QComboBox()
+            # List of years [2016,2025]
+            options = list(map(str, range(2016, 2026)))
+            options.append('9999')
+            dropdown.addItems(options)
+            dropdown.setCurrentIndex(len(options) - 1)
+            self.dropdown_list.append(dropdown)
+            input_layout.addWidget(dropdown)
+
+        self.setLayout(self.root_layout)
+
+        show_button = QPushButton('Show Animation', None)
+
+        input_layout.addWidget(show_button)
+        input_layout.addStretch()
+
+        self.root_layout.addLayout(input_layout)
+
+        show_button.clicked.connect(self.show_animation)
+
+        self.animation_label: AnimationLabel | None = None
+
+    def show_animation(self) -> None:
+        self.root_layout.removeWidget(self.animation_label)
+        self.year_list = [int(dropdown.currentText()) for dropdown in self.dropdown_list]
+        self.animation_label = AnimationLabel(self.year_list)
+        self.root_layout.addWidget(self.animation_label)

src/surface/gui/gui/images/illinois_river_map_1.png

@@ -13,7 +13,7 @@ def __init__(self) -> None:
         root_layout = QHBoxLayout()
         self.setLayout(root_layout)
 
-        logo_path = str(Path(get_package_share_directory('gui')) / 'images' / 'CWRUbotix Logo.png')
+        logo_path = str(Path(get_package_share_directory('gui')) / 'images' / 'cwrubotix_logo.png')
 
         logo_pixmap = QPixmap(logo_path)