sf-daq/dap
0
0
Fork 1
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

cleanup

Browse Source
This commit is contained in:
augustin_s
2024-03-22 18:46:54 +01:00
parent 7131329fbb
commit 5bb00e5e6a
1 changed files with 39 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
dap/accumulator.py
View File

@ -3,12 +3,13 @@ import os
import zmq import zmq
flags = 0
FLAGS = 0
OUTPUT_DIR_NAME = "/gpfs/photonics/swissfel/buffer/dap/data" OUTPUT_DIR_NAME = "/gpfs/photonics/swissfel/buffer/dap/data"
def main():
def main():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--accumulator", default="localhost", help="name of host where accumulator works") parser.add_argument("--accumulator", default="localhost", help="name of host where accumulator works")
@ -16,52 +17,57 @@ def main():
args = parser.parse_args() args = parser.parse_args()
FA_HOST_ACCUMULATE = args.accumulator # FA_HOST_ACCUMULATE = args.accumulator
FA_PORT_ACCUMULATE = args.accumulator_port FA_PORT_ACCUMULATE = args.accumulator_port
zmq_context = zmq.Context(io_threads=4) zmq_context = zmq.Context(io_threads=4)
poller = zmq.Poller() poller = zmq.Poller()
# Accumulator # Accumulator
if True:
accumulator_socket = zmq_context.socket(zmq.PULL) accumulator_socket = zmq_context.socket(zmq.PULL)
accumulator_socket.bind('tcp://*:%s' % FA_PORT_ACCUMULATE ) accumulator_socket.bind(f"tcp://*:{FA_PORT_ACCUMULATE}")
poller.register(accumulator_socket, zmq.POLLIN) poller.register(accumulator_socket, zmq.POLLIN)
n_frames_received = 0 n_frames_received = 0
run_name_before = 'very_first_start' run_name_before = "very_first_start"
fNameOutput = f'{OUTPUT_DIR_NAME}/{run_name_before}.dap' fNameOutput = f"{OUTPUT_DIR_NAME}/{run_name_before}.dap"
if not os.path.isdir(os.path.dirname(fNameOutput)): if not os.path.isdir(os.path.dirname(fNameOutput)):
os.makedirs(os.path.dirname(fNameOutput)) os.makedirs(os.path.dirname(fNameOutput))
outputDap=open(fNameOutput, 'a') outputDap = open(fNameOutput, "a")
while True: while True:
events = dict(poller.poll(10)) # in accumulator check for worker output every 0.01 seconds events = dict(poller.poll(10)) # in accumulator check for worker output every 0.01 seconds
if accumulator_socket in events: if accumulator_socket in events:
results = accumulator_socket.recv_json(flags) results = accumulator_socket.recv_json(FLAGS)
n_frames_received += 1 n_frames_received += 1
pulse_id = results.get('pulse_id', 0) pulse_id = results.get("pulse_id", 0)
run_name = str(pulse_id//10000*10000) run_name = str(pulse_id//10000*10000)
detector = results.get('detector_name', "") detector = results.get("detector_name", "")
if run_name != run_name_before: if run_name != run_name_before:
run_name_before = run_name run_name_before = run_name
outputDap.close() outputDap.close()
fNameOutput = f'{OUTPUT_DIR_NAME}/{detector}/{run_name_before}.dap' fNameOutput = f"{OUTPUT_DIR_NAME}/{detector}/{run_name_before}.dap"
if not os.path.isdir(os.path.dirname(fNameOutput)): if not os.path.isdir(os.path.dirname(fNameOutput)):
os.makedirs(os.path.dirname(fNameOutput)) os.makedirs(os.path.dirname(fNameOutput))
outputDap=open(fNameOutput, 'a') outputDap = open(fNameOutput, "a")
pr_rois = results.get("roi_intensities",[]) pr_rois = results.get("roi_intensities", [])
print(pulse_id, results.get('is_good_frame', -1), results.get('is_hit_frame', False), results.get('number_of_spots', -1), results.get("laser_on", False), *pr_rois, file=outputDap) print(pulse_id, results.get("is_good_frame", -1), results.get("is_hit_frame", False), results.get("number_of_spots", -1), results.get("laser_on", False), *pr_rois, file=outputDap)
else: else:
outputDap.flush() # may be too intensive outputDap.flush() # may be too intensive
if __name__ == "__main__": if __name__ == "__main__":
main() main()