Implemented loading and saving of pulse sequence

src/nqrduck_pulseprogrammer/controller.py

@@ -1,6 +1,8 @@
 import logging
+import json
 from PyQt6.QtCore import pyqtSlot
 from nqrduck.module.module_controller import ModuleController
+from nqrduck_spectrometer.pulsesequence import PulseSequence
 
 logger = logging.getLogger(__name__)
 
@@ -36,3 +38,23 @@ class PulseProgrammerController(ModuleController):
                 event.duration = float(duration)
                 break
         self.module.model.events_changed.emit()
+
+    def save_pulse_sequence(self, path):
+        logger.debug("Saving pulse sequence to %s", path)
+        sequence = self.module.model.pulse_sequence.dump_sequence_data()
+        with open(path, "w") as file:
+            file.write(json.dumps(sequence))
+        
+
+    def load_pulse_sequence(self, path):
+        logger.debug("Loading pulse sequence from %s", path)
+        sequence = None
+        with open(path, "r") as file:
+            sequence = file.read()
+
+        sequence = json.loads(sequence)
+
+        loaded_sequence = PulseSequence.load_sequence(sequence, self.module.model.pulse_parameter_options)
+        
+        self.module.model.pulse_sequence = loaded_sequence
+        self.module.model.events_changed.emit()

src/nqrduck_pulseprogrammer/view.py

@@ -2,7 +2,7 @@ import logging
 import functools
 from collections import OrderedDict
 from PyQt6.QtGui import QIcon
-from PyQt6.QtWidgets import QTableWidget, QVBoxLayout, QPushButton, QHBoxLayout, QLabel, QDialog, QLineEdit, QDialogButtonBox, QWidget, QCheckBox, QToolButton
+from PyQt6.QtWidgets import QTableWidget, QVBoxLayout, QPushButton, QHBoxLayout, QLabel, QDialog, QLineEdit, QDialogButtonBox, QWidget, QCheckBox, QToolButton, QFileDialog
 from PyQt6.QtCore import pyqtSlot, pyqtSignal
 from nqrduck.module.module_view import ModuleView
 from nqrduck_spectrometer.pulseparameters import BooleanOption, NumericOption
@@ -21,6 +21,7 @@ class PulseProgrammerView(ModuleView):
         self.module.model.pulse_parameter_options_changed.connect(self.on_pulse_parameter_options_changed)
         
     def setup_ui(self):
+
         # Create pulse table
         title = QLabel("Pulse Sequence: %s" % self.module.model.pulse_sequence.name)
         # Make title bold
@@ -39,12 +40,23 @@ class PulseProgrammerView(ModuleView):
         table_layout.addStretch(1)
         # Add button for new event
         self.new_event_button = QPushButton("New event")
+        self.new_event_button.clicked.connect(self.on_new_event_button_clicked)
+        button_layout.addWidget(self.new_event_button)
+
+        # Add button for save pulse sequence
+        self.save_pulse_sequence_button = QPushButton("Save pulse sequence")
+        self.save_pulse_sequence_button.clicked.connect(self.on_save_button_clicked)
+        button_layout.addWidget(self.save_pulse_sequence_button)
+       
+        # Add button for load pulse sequence
+        self.load_pulse_sequence_button = QPushButton("Load pulse sequence")
+        self.load_pulse_sequence_button.clicked.connect(self.on_load_button_clicked)
+        button_layout.addWidget(self.load_pulse_sequence_button)
 
         # Connect signals
-        self.new_event_button.clicked.connect(self.on_new_event_button_clicked)
         self.module.model.events_changed.connect(self.on_events_changed)
 
-        button_layout.addWidget(self.new_event_button)
+        
         button_layout.addStretch(1)
         layout.addWidget(title)
         layout.addLayout(button_layout)
@@ -155,6 +167,22 @@ class PulseProgrammerView(ModuleView):
             
             self.set_parameter_icons()
 
+    @pyqtSlot()
+    def on_save_button_clicked(self):
+        logger.debug("Save button clicked")
+        file_manager = QFileManager(self)
+        file_name = file_manager.saveFileDialog()
+        if file_name:
+            self.module.controller.save_pulse_sequence(file_name)
+
+    @pyqtSlot()
+    def on_load_button_clicked(self):
+        logger.debug("Load button clicked")
+        file_manager = QFileManager(self)
+        file_name = file_manager.loadFileDialog()
+        if file_name:
+            self.module.controller.load_pulse_sequence(file_name)
+
 class EventOptionsWidget(QWidget):
     """ This class is a widget that can be used to set the options for a pulse parameter.
     This widget is then added to the the first row of the according event column in the pulse table.
@@ -268,7 +296,7 @@ class OptionsDialog(QDialog):
                 def checkbox_result():
                     return check_box.isChecked()
 
-                check_box.setChecked(option.state)
+                check_box.setChecked(option.value)
                 self.layout.addWidget(check_box)
                 self.return_functions[option] = checkbox_result
 
@@ -334,3 +362,33 @@ class AddEventDialog(QDialog):
         else:
             self.accept()
 
+
+class QFileManager:
+    def __init__(self, parent=None):
+        self.parent = parent
+
+    def loadFileDialog(self):
+        fileName, _ = QFileDialog.getOpenFileName(self.parent,
+                                                  "QFileManager - Open File",
+                                                  "",
+                                                  "Quack Files (*.quack);;All Files (*)",
+                                                  options = QFileDialog.Option.ReadOnly)
+        if fileName:
+            return fileName
+        else:
+            return None
+
+    def saveFileDialog(self):
+        fileName, _ = QFileDialog.getSaveFileName(self.parent,
+                                                  "QFileManager - Save File",
+                                                  "",
+                                                  "Quack Files (*.quack);;All Files (*)",
+                                                  options=QFileDialog.Option.DontUseNativeDialog)
+        if fileName:
+            # Append the .quack extension if not present
+            if not fileName.endswith('.quack'):
+                fileName += '.quack'
+            return fileName
+        else:
+            return None
+