Belle II Software  release-05-01-25
trggdlUnpacker.py
1 #!/usr/bin/env python
2 #
3 # -----------------------------------------------------------------------------------
4 #
5 # GDL trigger Unpacker module
6 #
7 # usage : %> basf2 TrgGdlUnpacker.py [input sroot file name]
8 #
9 # -----------------------------------------------------------------------------------
10 from basf2 import *
11 
12 import sys
13 
14 argvs = sys.argv # get arg
15 argc = len(argvs) # of arg
16 
17 if argc != 2:
18  sys.exit("trggdlUnpacker.py> # of arg is strange.\n 1.rootname\n Exit.")
19 
20 if argc == 2:
21  f_in_root = argvs[1]
22 
23 # set_log_level(LogLevel.ERROR)
24 set_log_level(LogLevel.INFO)
25 
26 use_central_database("TRGGDL_201811")
27 
28 # input
29 if f_in_root[-6:] == ".sroot":
30  rootfiletype = "sroot"
31  input = register_module('SeqRootInput')
32 if f_in_root[-5:] == ".root":
33  rootfiletype = "root"
34  input = register_module('RootInput')
35 
36 input.param('inputFileName', f_in_root)
37 
38 # unpacker
39 unpacker = register_module('TRGGDLUnpacker')
40 # No unpacking. Just print info of trigger readout board
41 # included in the data.
42 trgReadoutBoardSearch = False
43 unpacker.param('trgReadoutBoardSearch', trgReadoutBoardSearch)
44 
45 # output
46 output = register_module('RootOutput')
47 
48 # Create main path
49 main = create_path()
50 
51 # Add modules to main path
52 main.add_module(input)
53 main.add_module(unpacker)
54 
55 input.param("inputFileName", f_in_root)
56 output.param("outputFileName", "gdltrg_test.root")
57 
58 main.add_module(output, branchNames=["TRGGDLUnpackerStores"])
59 
60 if trgReadoutBoardSearch:
61  process(main, max_event=100)
62 else:
63  process(main)