DUNE-DAQ
DUNE Trigger and Data Acquisition software
Loading...
Searching...
No Matches
hdf5_wib2_to_binary Namespace Reference

Functions

 Debug (x)
 
 main (args)
 

Variables

 parser = argparse.ArgumentParser(description = 'Python script to write WIB2 frames from DUNE-DAQ HDF5 files to binary files.')
 
 dest
 
 help
 
 type
 
 int
 
 required
 
 default
 
 action
 
 args = parser.parse_args()
 
 debug = args.debug
 

Detailed Description

Created on: 21/02/2023 15:28

Author: Shyam Bhuller

Description: Python script to write WIB2 frames from DUNE-DAQ HDF5 files to binary files.

Function Documentation

◆ Debug()

hdf5_wib2_to_binary.Debug ( x)
 print if we are in debug mode

Args:
    x (any): thing to prints

Definition at line 19 of file hdf5_wib2_to_binary.py.

19def Debug(x):
20 """ print if we are in debug mode
21
22 Args:
23 x (any): thing to prints
24 """
25 if debug:
26 print(x)
27
28

◆ main()

hdf5_wib2_to_binary.main ( args)

Definition at line 29 of file hdf5_wib2_to_binary.py.

29def main(args):
30 h5file = HDF5RawDataFile(args.file_name)
31
32 total_records = len(h5file.get_all_record_ids())
33 if args.n_records == -1:
34 args.n_records = total_records
35 if args.n_records > total_records:
36 raise Exception(f"Number of specified records is greater than the total {total}")
37
38 n_links = len([f for f in h5file.get_fragment_dataset_paths(1) if f.split("_")[-1] == "WIB"]) # exclude other fragments in the record that are not WIB frames
39 if args.link >= n_links:
40 raise Exception(f"Link number out of range.")
41
42 out_name = f"wib_link_{args.link}.bin"
43 with open(out_name, "wb") as bf:
44 total_frames = 0
45 # loop over all triggers
46 for i in range(args.n_records):
47 header = h5file.get_record_header_dataset_path(i+1) # trigger number starts at 1
48 Debug(header)
49
50 fragments = h5file.get_fragment_dataset_paths(i+1)
51 fragments = [f for f in fragments if f.split("_")[-1] == "WIB"] # exclude other fragments in the record that are not WIB frames
52
53 Debug(f"loading fragment: {fragments[args.link]}")
54 f = h5file.get_frag(fragments[args.link])
55
56 WIB2Frame_size = fddetdataformats.WIB2Frame.sizeof()
57
58 n_frames = (f.get_size() - f.get_header().sizeof()) // WIB2Frame_size # calculate the number of wib frames per fragment
59 for j in range(n_frames):
60 Debug(f.get_fragment_type())
61 Debug(f.get_element_id())
62
63 data = fddetdataformats.WIB2Frame(f.get_data(j * WIB2Frame_size)) # unpack fragment to WIB2Frame
64
65 Debug(f"{data.sizeof()=}")
66
67 bf.write(bytearray(data.get_bytes())) # write binary data to the file
68 total_frames += j
69 print(f"writing {total_frames} WIB2 frames to binary file.", "\r")
70 print(f"wrote {args.n_records} fragments from wib link {args.link} to file {out_name}.")
71 return
72
int main(int argc, char **argv)

Variable Documentation

◆ action

hdf5_wib2_to_binary.action

Definition at line 78 of file hdf5_wib2_to_binary.py.

◆ args

hdf5_wib2_to_binary.args = parser.parse_args()

Definition at line 79 of file hdf5_wib2_to_binary.py.

◆ debug

hdf5_wib2_to_binary.debug = args.debug

Definition at line 80 of file hdf5_wib2_to_binary.py.

◆ default

hdf5_wib2_to_binary.default

Definition at line 77 of file hdf5_wib2_to_binary.py.

◆ dest

hdf5_wib2_to_binary.dest

Definition at line 75 of file hdf5_wib2_to_binary.py.

◆ help

hdf5_wib2_to_binary.help

Definition at line 75 of file hdf5_wib2_to_binary.py.

◆ int

hdf5_wib2_to_binary.int

Definition at line 76 of file hdf5_wib2_to_binary.py.

◆ parser

hdf5_wib2_to_binary.parser = argparse.ArgumentParser(description = 'Python script to write WIB2 frames from DUNE-DAQ HDF5 files to binary files.')

Definition at line 74 of file hdf5_wib2_to_binary.py.

◆ required

hdf5_wib2_to_binary.required

Definition at line 76 of file hdf5_wib2_to_binary.py.

◆ type

hdf5_wib2_to_binary.type

Definition at line 76 of file hdf5_wib2_to_binary.py.