Merge pull request #332 from dmgav/srx-incomplete-scan

FIX: fix incorrect number of rows in incomplete scans (SRX)
NSLS2 · Dec 13, 2024 · d55ddf0 · d55ddf0
2 parents b0fd12a + 4409e8c
commit d55ddf0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pyxrf/model/load_data_from_db.py b/pyxrf/model/load_data_from_db.py
@@ -2360,6 +2360,12 @@ def map_data2D_srx_new_tiled(
             fast_pos = data_stream0[fast_key]
             slow_pos = data_stream0[slow_key]
         else:
+            # Compute the number of actually existing rows in case the scan was interrupted
+            if d_xs_sum is not None:
+                slow_pts = d_xs_sum.shape[0]
+            elif d_xs2_sum is not None:
+                slow_pts = d_xs2_sum.shape[0]
+
             row_pos_fast = da.arange(fast_pts) * fast_step + fast_start
             fast_pos = da.broadcast_to(row_pos_fast, (slow_pts, fast_pts))
             if snaking_enabled: