mirror of
https://github.com/slsdetectorgroup/aare.git
synced 2025-06-12 15:27:13 +02:00
write rawfiles (single file) and read rawfiles in order (#66)
* read subfiles with unordered and missing frames * save work debugging * Revert "save work debugging" This reverts commite791992a05
. * Revert "read subfiles with unordered and missing frames" This reverts commit1177fd129d
. * throw when two frames have different frame numbers * write single part RawFile (working beta) * correct total number of frames in master file * add new mythen file with syncd frames * read frames with same frame number * clang-tidy fixes, formatting, add tests * improve readability in loop * fix failing tests --------- Co-authored-by: Bechir <bechir.brahem420@gmail.com>
This commit is contained in:
58
data/mythen/correct_frame_numbers.py
Normal file
58
data/mythen/correct_frame_numbers.py
Normal file
@ -0,0 +1,58 @@
|
||||
import numpy as np
|
||||
import shutil
|
||||
|
||||
header_dt = np.dtype(
|
||||
[
|
||||
("Frame Number", "u8"),
|
||||
("SubFrame Number/ExpLength", "u4"),
|
||||
("Packet Number", "u4"),
|
||||
("Bunch ID", "u8"),
|
||||
("Timestamp", "u8"),
|
||||
("Module Id", "u2"),
|
||||
("Row", "u2"),
|
||||
("Column", "u2"),
|
||||
("Reserved", "u2"),
|
||||
("Debug", "u4"),
|
||||
("Round Robin Number", "u2"),
|
||||
("Detector Type", "u1"),
|
||||
("Header Version", "u1"),
|
||||
("Packets caught mask", "8u8")
|
||||
]
|
||||
)
|
||||
|
||||
# Read three frames from a jungfrau file with a single interface
|
||||
|
||||
frames = 1
|
||||
parts = 4
|
||||
frame_per_file = 3
|
||||
bytes_per_pixel = 4
|
||||
frame_cols = 1
|
||||
frame_rows = 5120
|
||||
|
||||
part_cols = 1280
|
||||
part_rows = 1
|
||||
|
||||
header = np.zeros((frames,parts), dtype = header_dt)
|
||||
|
||||
|
||||
|
||||
# verify that all parts have the same frame number
|
||||
frame = 0
|
||||
i = 55
|
||||
for part in range(parts):
|
||||
file_name_r = f'scan242_d{part}_f{frame//frame_per_file}_{3}.raw'
|
||||
file_name_w = f'CORRECTED_scan242_d{part}_f{frame//frame_per_file}_{3}.raw'
|
||||
shutil.copyfile(file_name_r, file_name_w)
|
||||
|
||||
with open(file_name_r) as fr, open(file_name_w, 'r+b') as fw:
|
||||
# get frame
|
||||
offset = (frame%frame_per_file)*(header_dt.itemsize+part_rows*part_cols*bytes_per_pixel)
|
||||
header[frame,part] = np.fromfile(fr, dtype=header_dt, count = 1,offset=offset)
|
||||
# update frame number
|
||||
header[frame,part]['Frame Number'] = i
|
||||
fw.seek(offset)
|
||||
header[frame,part].tofile(fw)
|
||||
|
||||
|
||||
print("[X] Done\n")
|
||||
|
Reference in New Issue
Block a user