vtxhProcess.py

Go to the documentation of this file.

import HpstrConf
import sys
import baseConfig
 
baseConfig.parser.add_argument("-w", "--tracking", dest="tracking", default="KF")
#(options, args) = parser.parse_args()
args = baseConfig.parser.parse_args()
 
# Use the input file to set the output file name
histo_file = args.inFilename[0]
out_file = args.outFilename[0]
 
print('Histo file: %s' % histo_file)
print('Out file: %s' % out_file)
 
p = HpstrConf.Process()
 
p.run_mode = 2
p.skip_events = options.skip_events
p.max_events = options.nevents
 
#p.max_events = 1000
 
# Library containing processors
p.add_library("libprocessors")
 
 
 
vtxPostProc = HpstrConf.Processor('vtxPostProc', 'VtxHistoProcessor')
 
 
vtxPostProc.parameters["debug"] = 1
vtxPostProc.parameters["rebin"] = 4
#KF
#vtxPostProc.parameters["selection"] = "vtxana_kf_vtxSelection"
#GBL
if (args.tracking == "KF"):
    vtxPostProc.parameters["selections"] = ["vtxana_Tight"]
elif (args.tracking == "GBL"):
    vtxPostProc.parameters["selections"] = ["vtxana_Tight"]
else:
    print("Error specify -w KF or -w GBL")
    exit(1)
 
 
vtxPostProc.parameters["projections"] = ["ele_d0_vs_p_hh", "pos_d0_vs_p_hh",
                                         "ele_z0_vs_p_hh", "pos_z0_vs_p_hh",
                                         "vtx_InvM_vtx_svt_z_hh", "vtx_p_svt_z_hh"]
 
 
# Sequence which the processors will run.
p.sequence = [vtxPostProc]
 
p.input_files = [histo_file]
p.output_files = [out_file]
 
p.printProcess()