anaKalSimpTuple_cfg.py

Go to the documentation of this file.

import HpstrConf
import sys
import os
import baseConfig as base
 
base.parser.add_argument("-w", "--tracking", type=str, dest="tracking",
                         help="Which tracking to use to make plots", metavar="tracking", default="KF")
base.parser.add_argument("-f", "--makeFlatTuple", type=int, dest="makeFlatTuple",
                         help="Make True to make vertex ana flat tuple", metavar="makeFlatTuple", default=1)
base.parser.add_argument("-r", "--isRadPDG", type=int, dest="isRadPDG",
                         help="Set radiative trident PDG ID", metavar="isRadPDG", default=622)
base.parser.add_argument("-TS", "--trackstate", type=str, dest="trackstate",
                         help="Specify Track State | 'AtECal' or 'AtTarget'. Default " " is origin (AtIP)", metavar="trackstate", default="")
base.parser.add_argument("-bpc", "--beamPosCorr", type=str, dest="beamPosCorr",
                         help="Load beam position corrections from json", metavar="beamPosCorr", default="")
options = base.parser.parse_args()
 
# Use the input file to set the output file name
infile = options.inFilename
outfile = options.outFilename
 
outfile = outfile.split(".root")[0]+"_"+options.tracking+".root"
 
print('Input file: %s' % infile)
print('Output file: %s' % outfile)
 
p = HpstrConf.Process()
 
p.run_mode = 1
#p.max_events = 1000
 
# Library containing processors
p.add_library("libprocessors")
 
 
 
recoana_kf = HpstrConf.Processor('vtxana_kf', 'VertexAnaProcessor')
recoana_gbl = HpstrConf.Processor('vtxana_gbl', 'VertexAnaProcessor')
mcana = HpstrConf.Processor('mcpartana', 'MCAnaProcessor')
 
recoana_kf.parameters["anaName"] = "vtxana_kf"
recoana_kf.parameters["trkColl"] = "KalmanFullTracks%s"%(options.trackstate)
recoana_kf.parameters["tsColl"] = "TSData"
recoana_kf.parameters["vtxColl"] = "UnconstrainedV0Vertices_KF"
recoana_kf.parameters["mcColl"] = "MCParticle"
recoana_kf.parameters["hitColl"] = "SiClusters"
recoana_kf.parameters["ecalColl"] = "RecoEcalClusters"
recoana_kf.parameters["vtxSelectionjson"] = os.environ['HPSTR_BASE']+"/analysis/selections/vertexSelection.json"
recoana_kf.parameters["histoCfg"] = os.environ['HPSTR_BASE']+"/analysis/plotconfigs/tracking/vtxAnalysis.json"
recoana_kf.parameters["mcHistoCfg"] = os.environ['HPSTR_BASE']+'/analysis/plotconfigs/mc/basicMC.json'
 
recoana_kf.parameters["beamE"] = base.beamE[str(options.year)]
recoana_kf.parameters["isData"] = options.isData
recoana_kf.parameters["analysis"] = options.analysis
recoana_kf.parameters["debug"] = 0
recoana_kf.parameters["isRadPDG"] = options.isRadPDG
recoana_kf.parameters["makeFlatTuple"] = options.makeFlatTuple
recoana_kf.parameters["beamPosCfg"] = options.beamPosCorr
 
CalTimeOffset = -999
 
if (options.isData == 1):
    CalTimeOffset = 56.
    print("Running on data file: Setting CalTimeOffset %d" % CalTimeOffset)
 
elif (options.isData == 0):
    CalTimeOffset = 43.
    print("Running on MC file: Setting CalTimeOffset %d" % CalTimeOffset)
else:
    print("Specify which type of ntuple you are running on: -t 1 [for Data] / -t 0 [for MC]")
 
recoana_kf.parameters["CalTimeOffset"] = CalTimeOffset
#Region definitions
RegionPath = os.environ['HPSTR_BASE']+"/analysis/selections/"
 
recoana_kf.parameters["regionDefinitions"] = [RegionPath+'Tight.json',
                                              RegionPath+'radMatchTight.json']
 
#RecoHitAna
recoana_gbl.parameters = recoana_kf.parameters.copy()
recoana_gbl.parameters["anaName"] = "vtxana_gbl"
recoana_gbl.parameters["vtxColl"] = "UnconstrainedV0Vertices"
recoana_gbl.parameters["tsColl"] = "TSData"
recoana_gbl.parameters["hitColl"] = "RotatedHelicalOnTrackHits"
recoana_gbl.parameters["trkColl"] = "GBLTracks"
recoana_gbl.parameters["mcColl"] = "MCParticle"
recoana_gbl.parameters["ecalColl"] = "RecoEcalClusters"
recoana_gbl.parameters["vtxSelectionjson"] = os.environ['HPSTR_BASE']+"/analysis/selections/vertexSelection.json"
recoana_gbl.parameters["histoCfg"] = os.environ['HPSTR_BASE']+"/analysis/plotconfigs/tracking/vtxAnalysis.json"
recoana_gbl.parameters["mcHistoCfg"] = os.environ['HPSTR_BASE']+'/analysis/plotconfigs/mc/basicMC.json'
 
recoana_gbl.parameters["beamE"] = base.beamE[str(options.year)]
recoana_gbl.parameters["isData"] = options.isData
recoana_gbl.parameters["analysis"] = options.analysis
recoana_gbl.parameters["debug"] = 0
recoana_gbl.parameters["isRadPDG"] = options.isRadPDG
recoana_gbl.parameters["makeFlatTuple"] = options.makeFlatTuple
recoana_gbl.parameters["beamPosCfg"] = options.beamPosCorr
 
recoana_gbl.parameters["CalTimeOffset"] = CalTimeOffset
#Region definitions
RegionPath = os.environ['HPSTR_BASE']+"/analysis/selections/"
recoana_gbl.parameters["regionDefinitions"] = [RegionPath+'Tight.json',
                                               RegionPath+'radMatchTight.json']
#MCParticleAna
mcana.parameters["debug"] = 0
mcana.parameters["anaName"] = "mcAna"
mcana.parameters["partColl"] = "MCParticle"
mcana.parameters["trkrHitColl"] = "TrackerHits"
mcana.parameters["ecalHitColl"] = "EcalHits"
mcana.parameters["histCfg"] = os.environ['HPSTR_BASE']+'/analysis/plotconfigs/mc/basicMC.json'
 
#
#    RegionPath+'ESumCR.json',
#    RegionPath+'TightNoSharedL0.json',
#    RegionPath+'TightNoShared.json',
 
# Sequence which the processors will run.
#p.sequence = [recoana_kf,recoana_gbl]
if (options.tracking == "KF"):
    print("Run KalmanFullTracks analysis")
    p.sequence = [recoana_kf]  # ,mcana]
elif (options.tracking == "GBL"):
    print("Run GBL analysis")
    p.sequence = [recoana_gbl]  # ,mcana]
else:
    print("ERROR::Need to specify which tracks KF or GBL")
    exit(1)
 
p.skip_events = options.skip_events
p.max_events = options.nevents
 
p.input_files = infile
p.output_files = [outfile]
 
p.printProcess()