Window and toolbars settings save and restore

laserstudio/laserstudio.py

@@ -1,5 +1,5 @@
 #!/usr/bin/python3
-from PyQt6.QtCore import Qt, QKeyCombination
+from PyQt6.QtCore import Qt, QKeyCombination, QSettings
 from PyQt6.QtGui import QColor, QShortcut, QKeySequence
 from PyQt6.QtWidgets import (
     QMainWindow,
@@ -43,6 +43,9 @@ def __init__(self, config: Optional[dict]):
         if config is None:
             config = {}
 
+        # User settings
+        self.settings = QSettings("ledger", "laserstudio")
+
         # Instantiate all instruments
         self.instruments = Instruments(config)
 
@@ -169,6 +172,20 @@ def __init__(self, config: Optional[dict]):
         )
         shortcut.activated.connect(self.handle_go_next)
 
+        # Restore docks are previous session
+        geometry = self.settings.value("geometry")
+        if geometry is not None:
+            self.restoreGeometry(geometry)
+        window_state = self.settings.value("window-state")
+        if window_state is not None:
+            self.restoreState(window_state)
+
+    def closeEvent(self, event):
+        """Saves user settings before closing the application."""
+        self.settings.setValue("geometry", self.saveGeometry())
+        self.settings.setValue("window-state", self.saveState())
+        super().closeEvent(event)
+
     def handle_go_next(self) -> dict:
         """Go Next operation.
         Triggers the instruments to perform changes to go to next step of scan.

laserstudio/widgets/toolbars/cameranittoolbar.py

@@ -28,6 +28,7 @@ def __init__(self, laser_studio: "LaserStudio"):
         self.laser_studio = laser_studio
 
         super().__init__("NIT Camera parameters", laser_studio)
+        self.setObjectName("toolbar-camera-nit")  # For settings save and restore
         self.setAllowedAreas(
             Qt.ToolBarArea.LeftToolBarArea
             | Qt.ToolBarArea.RightToolBarArea

laserstudio/widgets/toolbars/cameratoolbar.py

@@ -27,6 +27,7 @@ def __init__(self, laser_studio: "LaserStudio"):
         assert laser_studio.instruments.camera is not None
         self.camera = laser_studio.instruments.camera
         super().__init__("Camera parameters", laser_studio)
+        self.setObjectName("toolbar-camera")  # For settings save and restore
         self.setAllowedAreas(
             Qt.ToolBarArea.LeftToolBarArea
             | Qt.ToolBarArea.RightToolBarArea

laserstudio/widgets/toolbars/laserdrivertoolbar.py

@@ -17,7 +17,7 @@ def __init__(self, laser_studio: "LaserStudio", laser_num: int):
         assert laser_num < len(laser_studio.instruments.lasers)
         self.laser = laser_studio.instruments.lasers[laser_num]
         assert isinstance(self.laser, LaserDriverInstrument)
-        super().__init__(f"Laser {laser_num} (Donjon Driver)", laser_studio)
+        super().__init__(f"Laser {laser_num} (Donjon Driver)", f"toolbar-laser-donjon-{laser_num}")
         self.setAllowedAreas(
             Qt.ToolBarArea.LeftToolBarArea | Qt.ToolBarArea.RightToolBarArea
         )

laserstudio/widgets/toolbars/lasertoolbar.py

@@ -2,4 +2,10 @@
 
 
 class LaserToolbar(QToolBar):
-    pass
+    def __init__(self, title: str, object_name: str):
+        """
+        :param title: Toolbar title, as displayed in the context menus.
+        :param object_name: Unique object name for Qt, used for settings save and restore.
+        """
+        super().__init__(title)
+        super().setObjectName(object_name)  # For settings save and restore

laserstudio/widgets/toolbars/maintoolbar.py

@@ -12,6 +12,7 @@
 class MainToolbar(QToolBar):
     def __init__(self, laser_studio: "LaserStudio"):
         super().__init__("Main", laser_studio)
+        self.setObjectName("toolbar-main")  # For settings save and restore
         group = laser_studio.viewer_buttons_group
         self.setAllowedAreas(
             Qt.ToolBarArea.LeftToolBarArea | Qt.ToolBarArea.RightToolBarArea

laserstudio/widgets/toolbars/pdmtoolbar.py

@@ -24,12 +24,11 @@ def __init__(self, laser_studio: "LaserStudio", laser_num: int):
         assert laser_num < len(laser_studio.instruments.lasers)
         self.laser = laser_studio.instruments.lasers[laser_num]
         assert isinstance(self.laser, PDMInstrument)
-        super().__init__(f"Laser {laser_num} (PDM)", laser_studio)
+        super().__init__(f"Laser {laser_num} (PDM)", f"toolbar-laser-pdm-{laser_num}")
         self.setAllowedAreas(
             Qt.ToolBarArea.LeftToolBarArea | Qt.ToolBarArea.RightToolBarArea
         )
 
-
         self.setFloatable(True)
         w = QPushButton(self)
         w.setToolTip("On/Off Laser")

laserstudio/widgets/toolbars/picturetoolbar.py

@@ -15,6 +15,7 @@
 class PictureToolbar(QToolBar):
     def __init__(self, laser_studio: "LaserStudio"):
         super().__init__("Background picture", laser_studio)
+        self.setObjectName("toolbar-background-picture")  # For settings save and restore
         group = laser_studio.viewer_buttons_group
         self.setAllowedAreas(Qt.ToolBarArea.TopToolBarArea)
         self.setFloatable(True)

laserstudio/widgets/toolbars/scantoolbar.py

@@ -16,6 +16,7 @@
 class ScanToolbar(QToolBar):
     def __init__(self, laser_studio: "LaserStudio"):
         super().__init__("Scanning Zones", laser_studio)
+        self.setObjectName("toolbar-scanning")  # For settings save and restore
         group = laser_studio.viewer_buttons_group
         self.setAllowedAreas(Qt.ToolBarArea.TopToolBarArea)
         self.setFloatable(True)

laserstudio/widgets/toolbars/stagetoolbar.py

@@ -26,6 +26,7 @@ def __init__(self, laser_studio: "LaserStudio"):
         assert laser_studio.instruments.stage is not None
         self.stage = laser_studio.instruments.stage
         super().__init__("Stage control", laser_studio)
+        self.setObjectName("toolbar-stage")  # For settings save and restore
         group = laser_studio.viewer_buttons_group
         self.setAllowedAreas(
             Qt.ToolBarArea.LeftToolBarArea

laserstudio/widgets/toolbars/zoomtoolbar.py

@@ -13,6 +13,7 @@
 class ZoomToolbar(QToolBar):
     def __init__(self, laser_studio: "LaserStudio"):
         super().__init__("Zoom control", laser_studio)
+        self.setObjectName("toolbar-zoom")  # For settings save and restore
         self.setAllowedAreas(Qt.ToolBarArea.TopToolBarArea)
         self.setFloatable(True)