if refined positions are saved, they are also saved to dump files

ptycho · Oct 13, 2023 · b7b9b09 · b7b9b09
1 parent c544fef
commit b7b9b09
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/ptypy/core/ptycho.py b/ptypy/core/ptycho.py
@@ -989,6 +989,11 @@ def save_run(self, alt_file=None, kind='minimal', force_overwrite=True):
                 if len(self.runtime.iter_info) > 0:
                     dump.runtime.iter_info = [self.runtime.iter_info[-1]]
 
+                if self.record_positions:
+                    dump.positions = {}
+                    for ID, S in self.obj.storages.items():
+                        dump.positions[ID] = np.array([v.coord for v in S.views if v.pod.pr_view.layer==0])
+
                 content = dump
 
             elif kind == 'minimal' or kind == 'dls':