DAQ: updated FD operation scripts

This commit is contained in:
2026-05-22 15:24:26 +02:00
parent b5124707ed
commit 9b878055e7
2 changed files with 7 additions and 8 deletions
@@ -1,10 +1,9 @@
from dataclasses import dataclass
from typing import Callable
from aare.common.sample_geometry import SampleGeometryModel
from aare.daq.config import BeamlineConfig
from aare.daq.devices import BeamlineDevices
from aare.daq.mlbox import MlBox
from aare.daq.operations.common.runtime import DAQRuntimeState, FaceDetectionProgressEmitter
@dataclass
@@ -12,8 +11,8 @@ class FaceDetectionContext:
cfg: BeamlineConfig
devs: BeamlineDevices
mlbox: MlBox
sample_geometry_provider: Callable[[], SampleGeometryModel]
emit_progress: Callable[[dict], None]
runtime: DAQRuntimeState
progress_emitter: FaceDetectionProgressEmitter
@dataclass
@@ -32,7 +32,7 @@ class FaceDetectionService:
angle: int,
boxes_face: dict[int, tuple[float, float, float, float]],
) -> None:
self.ctx.emit_progress(
self.ctx.progress_emitter.emit_progress(
{
"running": True,
"current_angle_deg": angle,
@@ -49,11 +49,11 @@ class FaceDetectionService:
"height_fit": {},
"area_fit": {},
}
self.ctx.emit_progress(payload)
self.ctx.progress_emitter.emit_progress(payload)
return payload
def _centre_correction(self, model: MLBoxModel, tolerance: float = 0.2) -> None:
geom = self.ctx.sample_geometry_provider()
geom = self.ctx.runtime.sample_geometry
beam_y = geom.beam_location_pxl.y
beam_x = geom.beam_location_pxl.x
@@ -206,7 +206,7 @@ class FaceDetectionService:
"best_angle_deg": best_fit_angle_area,
},
}
self.ctx.emit_progress(payload)
self.ctx.progress_emitter.emit_progress(payload)
return FaceDetectionResult(success=True, payload=payload)
except Exception as e: