From 8ee1336b156d0d6f9066972ba936fc807b192db7 Mon Sep 17 00:00:00 2001
From: Filip Leonarski <filip.leonarski@psi.ch>
Date: Mon, 16 Jun 2025 14:57:36 +0200
Subject: [PATCH] DAQ: Update dtz at the beginning of the measurement (before
 going to data collection)

---
 daq/src/aaredaq/daq.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/daq/src/aaredaq/daq.py b/daq/src/aaredaq/daq.py
index 4c29fee3..07b46fc8 100644
--- a/daq/src/aaredaq/daq.py
+++ b/daq/src/aaredaq/daq.py
@@ -243,6 +243,7 @@ class AareDAQ:
 
     def __raster(self, r: RasterGridRequest) -> CompletedRasterGrid:
         max_time = r.exp_time_s * r.n_y * r.n_x + 60
+        self.__cfg.dtz = r.dtz_mm
         self.__set_state(BeamlineStateEnum.DataCollection)
         self.__jfjoch.measure_raster(r, self.diffraction_geometry, self.sample)
         self.__devs.aerotech.move(r.omega_deg, wait=True, speed=360.0)
@@ -270,6 +271,7 @@ class AareDAQ:
             raise e
 
     def __rotation(self, request: RotationScanRequest):
+        self.__cfg.dtz = request.dtz_mm
         self.__set_state(BeamlineStateEnum.DataCollection)
 
         if request.start is not None: