Belle II Software  release-05-01-25
trggrl_unpacker_dqm.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 histo = register_module('HistoManager')
40 histo.param('histoFileName', "dqm.root") # File to save histograms
41 main.add_module(histo)
42 
43 unpacker = register_module('TRGGRLUnpacker')
44 main.add_module(unpacker)
45 
46 trggrldqm = register_module('TRGGRLDQM')
47 main.add_module(trggrldqm)
48 
49 
50 if int(options.tracking):
51  cdcunpacker = register_module('CDCUnpacker')
52  cdcunpacker.param('xmlMapFileName', "cdc/data/ch_map.dat")
53  cdcunpacker.param('enablePrintOut', False)
54  main.add_module(cdcunpacker)
55  add_cosmics_reconstruction(main, 'CDC', False)
56 
57 
58 if int(options.gdl):
59  trggdlUnpacker = register_module("TRGGDLUnpacker")
60  main.add_module(trggdlUnpacker)
61  trggdlsummary = register_module('TRGGDLSummary')
62  main.add_module(trggdlsummary)
63 
64 output = register_module('RootOutput')
65 output.param("outputFileName", options.output)
66 main.add_module(output, branchNames=["TRGGRLUnpackerStore"])
67 
68 # Process all events
69 process(main)