From 568a95f4c9aaea815fba7dccc730c4500e73ce6c Mon Sep 17 00:00:00 2001
From: appleb_m <martin.appleby@psi.ch>
Date: Fri, 27 Mar 2026 17:00:10 +0100
Subject: [PATCH] sample_queue_model.py: added to_state and from_state
 functions WIP

---
 src/aare/gui/models/sample_queue_model.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/src/aare/gui/models/sample_queue_model.py b/src/aare/gui/models/sample_queue_model.py
index 2721e29f..481a7a61 100644
--- a/src/aare/gui/models/sample_queue_model.py
+++ b/src/aare/gui/models/sample_queue_model.py
@@ -116,4 +116,19 @@ class SampleQueueSpreadsheet(QAbstractTableModel):
     def clearSamples(self):
         self.beginResetModel()
         self.samples = []
+        self.endResetModel()
+
+    def to_state(self) -> dict:
+        return {
+            "samples": [s.to_dict() for s in self.samples],
+        }
+
+    def from_state(self, state: dict):
+        self.beginResetModel()
+
+        self.samples = [
+            SampleShortInfo.from_dict(d)
+            for d in state.get("samples", [])
+        ]
+
         self.endResetModel()
\ No newline at end of file