version used in July Alvra experiment (conversion and compression of files using Ivan's export_file.py code based on jungfrau_utils); make list of frames in crystfel notation to separate laser light/dark

2026-08-10 06:30:28 +02:00 · 2020-07-27 14:47:41 +02:00
parent 44a60a8d27
commit df51bb6ca5
3 changed files with 282 additions and 3 deletions
@@ -0,0 +1,114 @@
+import argparse
+
+import numpy as np
+import os
+
+import h5py
+import json
+
+def is_it_dark(laser_mode, detector_rate, pulseid):
+
+    dark_rate = 1
+    if 100/detector_rate == int(100/detector_rate):
+        dark_rate = 100/detector_rate
+        if laser_mode == 11: # 50/50 mode
+            dark_rate *= 2
+        elif laser_mode == 41: # 4 lights, 1 dark sequence
+            dark_rate *= 5
+        elif laser_mode == 111: # 11 lights, 1 dark sequence
+            dark_rate *= 12
+        elif laser_mode == 191: # 19 lights, 1 dark sequence
+            dark_rate *=20
+
+    dark = True
+
+    if laser_mode == 0:
+        dark = True
+    elif laser_mode == 1:
+        dark = False
+    else:
+        if (pulseid + int(100/detector_rate) ) % dark_rate == 0:
+            dark = True
+        else:
+            dark = False
+
+    return dark
+
+
+parser = argparse.ArgumentParser()
+parser.add_argument("data_file", type=str)
+parser.add_argument("run_info", type=str)
+args = parser.parse_args()
+
+data_file = args.data_file
+run_info_file = args.run_info
+
+try:
+    with open(run_info_file) as json_file:
+        parameters = json.load(json_file)
+except:
+    print("Can't read provided run file {run_info_file}, may be not json?")
+    exit()
+
+laser_mode = parameters.get("laser_mode", 0)
+rate_multiplicator = parameters.get("rate_multiplicator", 1)
+detector_rate = 100//rate_multiplicator
+
+print("Laser mode: ", laser_mode, ", detector runs at ", detector_rate, "Hz")
+
+try:
+    f=h5py.File(data_file, "r")
+except:
+    print(f"Can't open {data_file}")
+    exit()
+
+detector = 'JF06T08V01'
+
+pulseids = f[f'/data/{detector}/pulse_id'][:]
+n_pulse_id = len(pulseids)
+if f'/data/{detector}/is_good_frame' in f.keys():
+    is_good_frame = f[f'/data/{detector}/is_good_frame'][:]
+else:
+    is_good_frame = [1] * n_pulse_id
+
+nGoodFrames = 0
+nProcessedFrames = 0
+
+index_dark = []
+index_light = []
+for i in range(len(pulseids)):
+    if not is_good_frame[i]:
+        continue
+    nGoodFrames += 1
+    p = pulseids[i]
+    nProcessedFrames += 1
+    if is_it_dark(laser_mode, detector_rate, p):
+        index_dark.append(i)
+    else:
+        index_light.append(i)
+
+f.close()
+
+print("Total number of frames: %s, number of good frames : %s, processed frames: %s, outputed frames: %s(dark) %s(light) " % (len(pulseids), nGoodFrames, nProcessedFrames, len(index_dark), len(index_light)) )
+
+delim = '//'
+
+if len(index_dark) > 0:
+    file_dark = data_file[:-3] + ".dark.lst"
+    if laser_mode == -1:
+        file_dark = data_file[:-3] + ".undefined.lst"
+    print(f"List of dark frames : {file_dark} , {len(index_dark)} frames")
+    f_list = open(file_dark, "w")
+    for frame_number in index_dark:
+        print(f'{data_file} //{frame_number}', file = f_list)
+    f_list.close()
+
+if len(index_light) > 0:
+    file_light = data_file[:-3] + ".light.lst"
+    print(f"List of light frames : {file_light} , {len(index_light)} frames")
+    f_list = open(file_light, "w")
+    for frame_number in index_light:
+        print(f'{data_file} {delim}{frame_number}', file = f_list)
+    f_list.close()
+
+