Belle II Software  release-05-01-25
trgcdctsfDQM.py
1 #!/usr/bin/env python3
2 # -*- coding: utf-8 -*-
3 # -----------------------------------------------------------------------------------
4 #
5 # Example of GDL trigger DQM Module
6 #
7 # usage : %> basf2 trgcdctsfDQM.py [input sroot file name]
8 #
9 # -----------------------------------------------------------------------------------
10 
11 from basf2 import *
12 
13 import sys # get argv
14 import re
15 import os.path
16 argvs = sys.argv # get arg
17 argc = len(argvs) # of arg
18 
19 set_log_level(LogLevel.DEBUG)
20 
21 main = create_path()
22 
23 if argc == 2 and argvs[1][-6:] == ".sroot":
24  f_in_root = argvs[1]
25  input = register_module('SeqRootInput')
26  matchobj = re.search(r"([^\/]+)\.sroot", f_in_root)
27  basename = re.sub(r'\.sroot$', '', matchobj.group())
28  input.param('inputFileName', f_in_root)
29 elif argc == 2 and argvs[1][-5:] == ".root":
30  f_in_root = argvs[1]
31  input = register_module('RootInput')
32  matchobj = re.search(r"([^\/]+)\.root", f_in_root)
33  basename = re.sub(r'\.root$', '', matchobj.group())
34  input.param('inputFileName', f_in_root)
35 elif argc == 1:
36  input = register_module('RootInput')
37  input.param('inputFileName', '/home/belle/nkzw/e3.4S/r034*/all/raw/sub00/raw.physics.hlt_hadron.0003.*.root')
38  basename = "e3.4S.r034"
39 else:
40  sys.exit("trgcdctsfDQM.py> # of arg is strange. Exit.")
41 
42 main.add_module(input)
43 histo = register_module('HistoManager')
44 histo.param("histoFileName", "dqm.%s.root" % basename)
45 
46 
47 nmod = [0, 1, 2, 3, 4, 5, 6]
48 for mod in nmod:
49  # Unpacker
50  main.add_module('TRGCDCTSFUnpacker', TSFMOD=mod)
51 
52  # DQM
53  main.add_module('TRGCDCTSFDQM', TSFMOD=mod)
54 
55 main.add_module(histo)
56 
57 progress = register_module('Progress')
58 main.add_module(progress)
59 
60 process(main)
61 
62 print(statistics)