trggdlUnpacker.py

#!/usr/bin/env python
#
 
 
 
# -----------------------------------------------------------------------------------
#
#                        GDL trigger Unpacker module
#
#    usage : %> basf2 TrgGdlUnpacker.py [input sroot file name]
#
# -----------------------------------------------------------------------------------
 
import basf2 as b2
 
import sys
 
argvs = sys.argv  # get arg
argc = len(argvs)  # of arg
 
if argc != 2:
    sys.exit("trggdlUnpacker.py> # of arg is strange.\n 1.rootname\n Exit.")
 
if argc == 2:
    f_in_root = argvs[1]
 
# set_log_level(LogLevel.ERROR)
b2.set_log_level(b2.LogLevel.INFO)
 
b2.use_central_database("TRGGDL_201811")
 
# input
if f_in_root[-6:] == ".sroot":
    rootfiletype = "sroot"
    input = b2.register_module('SeqRootInput')
if f_in_root[-5:] == ".root":
    rootfiletype = "root"
    input = b2.register_module('RootInput')
 
input.param('inputFileName', f_in_root)
 
# unpacker
unpacker = b2.register_module('TRGGDLUnpacker')
# No unpacking. Just print info of trigger readout board
# included in the data.
trgReadoutBoardSearch = False
unpacker.param('trgReadoutBoardSearch', trgReadoutBoardSearch)
 
# output
output = b2.register_module('RootOutput')
 
# Create main path
main = b2.create_path()
 
# Add modules to main path
main.add_module(input)
main.add_module(unpacker)
 
input.param("inputFileName", f_in_root)
output.param("outputFileName", "gdltrg_test.root")
 
main.add_module(output, branchNames=["TRGGDLUnpackerStores"])
 
if trgReadoutBoardSearch:
    b2.process(main, max_event=100)
else:
    b2.process(main)