Belle II Software  release-05-01-25
trgcdctsf1DQM.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.superlayer1.root" % basename)
45 
46 # Unpacker
47 trgcdctsfUnpacker = register_module("TRGCDCTSFUnpacker")
48 trgcdctsfUnpacker.param('TSFMOD', 1)
49 main.add_module(trgcdctsfUnpacker)
50 main.add_module(histo)
51 
52 # DQM
53 trgcdctsfdqm = register_module('TRGCDCTSFDQM')
54 trgcdctsfdqm.param('generatePostscript', True)
55 trgcdctsfdqm.param('TSFMOD', 1)
56 # postscript file name
57 psname = "dqm.%s.ps" % basename
58 trgcdctsfdqm.param('postScriptName', psname)
59 
60 main.add_module(trgcdctsfdqm)
61 
62 progress = register_module('Progress')
63 main.add_module(progress)
64 
65 process(main)
66 
67 print(statistics)