Have a seperate coffset variable within 'frag_to_crystfel'

European-XFEL · Aug 30, 2021 · 415c79e · 415c79e
1 parent 8014128
commit 415c79e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/extra_geom/crystfel_fmt.py b/extra_geom/crystfel_fmt.py
@@ -54,8 +54,8 @@ def _crystfel_format_vec(vec):
 
 def frag_to_crystfel(fragment, p, a, ss_slice, fs_slice, dims, pixel_size):
     tile_name = 'p{}a{}'.format(p, a)
-    c = fragment.corner_pos
-    c[:2] = c[:2] / pixel_size
+    c = fragment.corner_pos[:2] / pixel_size
+    coffset = c[2]
     dim_list = []
     for num, value in dims.items():
         if value == 'modno':
@@ -75,7 +75,7 @@ def frag_to_crystfel(fragment, p, a, ss_slice, fs_slice, dims, pixel_size):
         fs_vec=_crystfel_format_vec(fragment.fs_vec/ pixel_size),
         corner_x=c[0],
         corner_y=c[1],
-        coffset=c[2],
+        coffset=coffset,
     )
 
 def write_crystfel_geom(self, filename, *,