Belle II Software  release-05-01-25
trggrl_unpacker.py
1 #!/usr/bin/env python
2 #
3 # -----------------------------------------------------------------------------------
4 #
5 # CDCT3D trigger Unpacker module
6 #
7 # usage : %> basf2 TrgCdcT3dUnpacker.py [input sroot file name]
8 #
9 # -----------------------------------------------------------------------------------
10 from basf2 import *
11 import os
12 from optparse import OptionParser
13 from reconstruction import add_cosmics_reconstruction
14 home = os.environ['BELLE2_LOCAL_DIR']
15 
16 parser = OptionParser()
17 parser.add_option(
18  '-f',
19  '--file',
20  dest='filename',
21  default='hsm/belle2/bdata/Data/Raw/e0007/r01640/sub00/cosmic.0007.01640.HLT1.f00000.root')
22 parser.add_option('-o', '--output', dest='output', default='trggrl_unpacker.root')
23 parser.add_option('-t', '--tracking', dest='tracking', default=0)
24 parser.add_option('-g', '--gdl', dest='gdl', default=0)
25 (options, args) = parser.parse_args()
26 
27 runID = str(options.run)
28 
29 set_log_level(LogLevel.ERROR)
30 
31 main = create_path()
32 
33 # input
34 input = register_module('RootInput')
35 input.param('inputFileName', options.filename)
36 
37 main.add_module(input)
38 
39 
40 unpacker = register_module('TRGGRLUnpacker')
41 main.add_module(unpacker)
42 
43 
44 if int(options.tracking):
45  cdcunpacker = register_module('CDCUnpacker')
46  cdcunpacker.param('xmlMapFileName', "cdc/data/ch_map.dat")
47  cdcunpacker.param('enablePrintOut', False)
48  main.add_module(cdcunpacker)
49  add_cosmics_reconstruction(main, 'CDC', False)
50 
51 
52 if int(options.gdl):
53  trggdlUnpacker = register_module("TRGGDLUnpacker")
54  main.add_module(trggdlUnpacker)
55  trggdlsummary = register_module('TRGGDLSummary')
56  main.add_module(trggdlsummary)
57 
58 output = register_module('RootOutput')
59 output.param("outputFileName", options.output)
60 main.add_module(output, branchNames=["TRGGRLUnpackerStore"])
61 
62 # Process all events
63 process(main)