#187 Put an icon on the overall application (naff, but better than having the default)

Author: dmh23

src/prototype_ui/main.py

@@ -1,5 +1,4 @@
 import sys
-import time
 
 from PyQt6.QtCore import Qt
 from PyQt6.QtWidgets import QMainWindow, QApplication, QProgressBar, QFileDialog
@@ -12,6 +11,7 @@
 from src.graphics.track_analysis_canvas import TrackAnalysisCanvas
 from src.tracks.tracks import get_all_tracks
 from src.ui.open_file_dialog import OpenFileDialog
+from ui.icons import get_custom_icon
 
 
 class MainWindow(QMainWindow):
@@ -26,7 +26,8 @@ def __init__(self):
 
         self.setMinimumSize(200, 200)
         self.resize(1800, 400)
-        self.setWindowTitle("Example")
+        self.setWindowTitle("DeepRacer Guru")
+        self.setWindowIcon(get_custom_icon("window_icon"))
 
         # Status Bar
         self._please_wait = PleaseWait(self.statusBar(), self.set_busy_cursor, self.set_normal_cursor)

src/ui/actions.py

@@ -1,36 +1,31 @@
 # v4 UI STATUS - BRAND NEW
 # ************************
 
-import os
-
-from PyQt6.QtGui import QAction, QIcon
+from PyQt6.QtGui import QAction
 from PyQt6.QtWidgets import QStyle
 
-
-ICON_DIRECTORY = os.path.join("../icons")
-ICON_FILE_EXTENSION = ".png"
+from ui.icons import get_custom_icon
 
 
 class Actions:
     def __init__(self, style: QStyle):
         self._style = style
-        assert os.path.isdir(ICON_DIRECTORY)
 
         # REAL
         self.open_file = QAction("Open")
         self.open_file.setShortcut("Ctrl+O")
         self.open_file.setStatusTip("Open log file(s)")
-        self.open_file.setIcon(self.get_custom_icon("open_file"))
+        self.open_file.setIcon(get_custom_icon("open_file"))
 
         self.file_info = QAction("Info")
         self.file_info.setShortcut("Ctrl+I")
         self.file_info.setStatusTip("Get information about the currently open log file(s)")
-        self.file_info.setIcon(self.get_custom_icon("file_info"))
+        self.file_info.setIcon(get_custom_icon("file_info"))
 
         self.change_log_directory = QAction("Directory")
         self.change_log_directory.setShortcut("Ctrl+D")
         self.change_log_directory.setStatusTip("Change log file source directory")
-        self.change_log_directory.setIcon(self.get_custom_icon("change_log_directory"))
+        self.change_log_directory.setIcon(get_custom_icon("change_log_directory"))
 
         self.set_file_options = QAction("Options")
         self.set_file_options.setStatusTip("Set special options for opening log file(s)")
@@ -39,11 +34,4 @@ def __init__(self, style: QStyle):
         self.exit.setStatusTip("Exit application")
 
         # Example of how to set a standard icon when I need to ....
-
         #         self.change_log_directory.setIcon(style.standardIcon(QStyle.StandardPixmap.SP_DirIcon))
-
-    @staticmethod
-    def get_custom_icon(icon_name: str):
-        file_path = os.path.join(ICON_DIRECTORY, icon_name + ".png")
-        assert os.path.isfile(file_path)
-        return QIcon(file_path)

src/ui/icons.py

@@ -0,0 +1,13 @@
+import os
+
+from PyQt6.QtGui import QIcon
+
+
+ICON_DIRECTORY = os.path.join("../icons")
+ICON_FILE_EXTENSION = ".png"
+
+
+def get_custom_icon(icon_name: str):
+    file_path = os.path.join(ICON_DIRECTORY, icon_name + ".png")
+    assert os.path.isfile(file_path)
+    return QIcon(file_path)