DAQ: updated FD operation scripts
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user