fix: simple end-to-end test works at beamline

This commit is contained in:
e21206 2023-08-17 19:06:37 +02:00
parent 9476fde13a
commit 28b91eeda2

View File

@ -10,6 +10,9 @@ from ophyd.areadetector.plugins import FileBase
from bec_lib.core import BECMessage, MessageEndpoints from bec_lib.core import BECMessage, MessageEndpoints
from bec_lib.core.file_utils import FileWriterMixin from bec_lib.core.file_utils import FileWriterMixin
from bec_lib.core import bec_logger
logger = bec_logger.logger
class PilatusDetectorCamEx(PilatusDetectorCam, FileBase): class PilatusDetectorCamEx(PilatusDetectorCam, FileBase):
@ -50,8 +53,8 @@ class PilatusCsaxs(DetectorBase):
**kwargs, **kwargs,
) )
# TODO how to get base_path # TODO how to get base_path
self.service_cfg = {"base_path": f"/sls/X12SA/data/{self.username}/Data10/pilatus_2"} self.service_cfg = {"base_path": f"/sls/X12SA/data/{self.username}/Data10/pilatus_2/"}
self.filewriter = FileWriterMixin(self.service_cfg["base_path"]) self.filewriter = FileWriterMixin(self.service_cfg)
self.num_frames = 0 self.num_frames = 0
self.readout = 0.003 # 3 ms self.readout = 0.003 # 3 ms
self.triggermode = 0 # 0 : internal, scan must set this if hardware triggered self.triggermode = 0 # 0 : internal, scan must set this if hardware triggered
@ -68,25 +71,26 @@ class PilatusCsaxs(DetectorBase):
# "RID": scan_msg.content["info"]["RID"], # "RID": scan_msg.content["info"]["RID"],
# "queueID": scan_msg.content["info"]["queueID"], # "queueID": scan_msg.content["info"]["queueID"],
# } # }
scan_number = 10 # scan_msg.content["info"]["scan_number"] self.scan_number = 10 # scan_msg.content["info"]["scan_number"]
self.exp_time = 0.5 # scan_msg.content["info"]["exp_time"] self.exp_time = 0.5 # scan_msg.content["info"]["exp_time"]
self.num_frames = 1 # scan_msg.content["info"]["num_points"] self.num_frames = 3 # scan_msg.content["info"]["num_points"]
# TODO remove
# self.username = self.device_manager.producer.get(MessageEndpoints.account()).decode() # self.username = self.device_manager.producer.get(MessageEndpoints.account()).decode()
# set pilatus threshol # set pilatus threshol
self._set_threshold() self._set_threshold()
# set Epic PVs for filewriting # set Epic PVs for filewriting
self.file_path.set(f"/dev/shm/zmq/") self.cam.file_path.set(f"/dev/shm/zmq/")
self.file_name.set(f"{self.username}_2_") self.cam.file_name.set(f"{self.username}_2_{self.scan_number:05d}")
self.auto_increment.set(1) # auto increment self.cam.auto_increment.set(1) # auto increment
self.file_number.set(0) # first iter self.cam.file_number.set(0) # first iter
self.file_format.set(0) # 0: TIFF self.cam.file_format.set(0) # 0: TIFF
self.file_template.set("%s%s_%5.5d.cbf") self.cam.file_template.set("%s%s_%5.5d.cbf")
# compile zmq stream for data transfer # compile zmq stream for data transfer
scan_dir = self.filewriter._get_scan_directory( scan_dir = self.filewriter._get_scan_directory(
scan_bundle=1000, scan_number=scan_number, leading_zeros=5 scan_bundle=1000, scan_number=self.scan_number, leading_zeros=5
) )
self.destination_path = os.path.join( self.destination_path = os.path.join(
self.service_cfg["base_path"] self.service_cfg["base_path"]
@ -100,6 +104,7 @@ class PilatusCsaxs(DetectorBase):
} }
] ]
} }
logger.info(data_msg)
headers = {"Content-Type": "application/json", "Accept": "application/json"} headers = {"Content-Type": "application/json", "Accept": "application/json"}
res = requests.put( res = requests.put(
@ -116,7 +121,7 @@ class PilatusCsaxs(DetectorBase):
"zmqWriter", "zmqWriter",
self.username, self.username,
{ {
"addr": "tcp://x12sa-pd=1:8888", "addr": "tcp://x12sa-pd-2:8888",
"dst": ["file"], "dst": ["file"],
"numFrm": self.num_frames, "numFrm": self.num_frames,
"timeout": 2000, "timeout": 2000,
@ -126,7 +131,7 @@ class PilatusCsaxs(DetectorBase):
] ]
res = requests.put( res = requests.put(
url="http://xbl-daq-34:8091/pilatus_1/run", url="http://xbl-daq-34:8091/pilatus_2/run",
data=json.dumps(data_msg), data=json.dumps(data_msg),
headers=headers, headers=headers,
) )
@ -155,6 +160,7 @@ class PilatusCsaxs(DetectorBase):
"user": self.username, "user": self.username,
}, },
] ]
logger.info(data_msg)
res = requests.put( res = requests.put(
url="http://xbl-daq-34:8091/pilatus_1/run", url="http://xbl-daq-34:8091/pilatus_1/run",
@ -172,8 +178,9 @@ class PilatusCsaxs(DetectorBase):
# TODO readout mono, monitor threshold and set it if mokev is different # TODO readout mono, monitor threshold and set it if mokev is different
# mokev = self.device_manager.devices.mokev.obj.read()["mokev"]["value"] # mokev = self.device_manager.devices.mokev.obj.read()["mokev"]["value"]
# TODO remove # TODO remove
mokev = 20 mokev = 16
pil_threshold = self.cam.threshold_energy.read() # TODO refactor naming from name, pilatus_2
pil_threshold = self.cam.threshold_energy.read()["pilatus_2_cam_threshold_energy"]["value"]
if not np.isclose(mokev / 2, pil_threshold, rtol=0.05): if not np.isclose(mokev / 2, pil_threshold, rtol=0.05):
self.cam.threshold_energy.set(mokev / 2) self.cam.threshold_energy.set(mokev / 2)