Belle II Software  release-05-01-25
trggdlDST.py
1 #!/usr/bin/env python3
2 # -*- coding: utf-8 -*-
3 # -----------------------------------------------------------------------------------
4 #
5 # GDL trigger DST Module
6 #
7 # usage : %> basf2 trggdlDST.py [input sroot file name]
8 #
9 # -----------------------------------------------------------------------------------
10 
11 from basf2 import *
12 
13 import sys # get argv
14 argvs = sys.argv # get arg
15 argc = len(argvs) # of arg
16 if argc != 2:
17  sys.exit("trggdlDST.py> # of arg is strange. Exit.")
18 if argc == 2:
19  f_in_root = argvs[1]
20 
21 
22 set_log_level(LogLevel.INFO)
23 
24 main = create_path()
25 
26 # input
27 if f_in_root[-6:] == ".sroot":
28  rootfiletype = "sroot"
29  input = register_module('SeqRootInput')
30 if f_in_root[-5:] == ".root":
31  rootfiletype = "root"
32  input = register_module('RootInput')
33 
34 input.param('inputFileName', f_in_root)
35 main.add_module(input)
36 
37 # Unpacker
38 trggdlUnpacker = register_module("TRGGDLUnpacker")
39 main.add_module(trggdlUnpacker)
40 
41 # Fill DST
42 trggdldst = register_module('TRGGDLDST')
43 main.add_module(trggdldst)
44 
45 output = register_module('RootOutput')
46 output.param("outputFileName", "gdldst.root")
47 main.add_module(output, branchNames=["TRGGDLDST"])
48 
49 progress = register_module('Progress')
50 main.add_module(progress)
51 
52 process(main)
53 
54 print(statistics)