JetPlusTracks/python/JetPlusTrackCorrections_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002 # ---------- Add assigned jet-track association
0003
0004 from RecoJets.JetAssociationProducers.ak4JTA_cff import *
0005 ak4JetTracksAssociatorAtVertexJPT = ak4JetTracksAssociatorAtVertex.clone(
0006     useAssigned = True,
0007     pvSrc       = "offlinePrimaryVertices"
0008 )
0009
0010 # ---------- Tight Electron ID
0011
0012 from RecoEgamma.ElectronIdentification.egmGsfElectronIDs_cff import egmGsfElectronIDs
0013 JPTegmGsfElectronIDs = egmGsfElectronIDs.clone(
0014     physicsObjectsIDs = cms.VPSet(),
0015     physicsObjectSrc = 'gedGsfElectrons'
0016 )
0017 from PhysicsTools.SelectorUtils.tools.vid_id_tools import setupVIDSelection
0018 my_id_modules = ['RecoEgamma.ElectronIdentification.Identification.cutBasedElectronID_Winter22_122X_V1_cff']
0019 for id_module_name in my_id_modules:
0020     idmod= __import__(id_module_name, globals(), locals(), ['idName','cutFlow'])
0021     for name in dir(idmod):
0022         item = getattr(idmod,name)
0023         if hasattr(item,'idName') and hasattr(item,'cutFlow'):
0024             setupVIDSelection(JPTegmGsfElectronIDs,item)
0025
0026 # ---------- Seeds from TrackJets
0027
0028 from RecoJets.JetPlusTracks.jetPlusTrackAddonSeedProducer_cfi import *
0029
0030 JetPlusTrackAddonSeedReco = jetPlusTrackAddonSeedProducer.clone()
0031
0032 # ---------- Module definition
0033 from RecoJets.JetPlusTracks.JetPlusTrackCorrections_cfi import *
0034
0035 JetPlusTrackZSPCorJetAntiKt4 = cms.EDProducer(
0036     "JetPlusTrackProducer",
0037     cms.PSet(JPTZSPCorrectorAntiKt4),
0038     src = cms.InputTag("ak4CaloJets"),
0039     srcTrackJets = cms.InputTag("ak4TrackJets"),
0040     srcAddCaloJets = cms.InputTag('JetPlusTrackAddonSeedReco'),
0041     extrapolations = cms.InputTag("trackExtrapolator"),
0042     tagName = cms.vstring('ZSP_CMSSW390_Akt_05_PU0'),
0043     tagNameOffset = cms.vstring(),
0044     PU = cms.int32(-1),
0045     FixedPU = cms.int32(0),
0046     UseZSP = cms.bool(False),
0047     srcPVs = cms.InputTag('offlinePrimaryVertices'),
0048     alias = cms.untracked.string('JetPlusTrackZSPCorJetAntiKt4'),
0049     ptCUT = cms.double(15.),
0050     dRcone = cms.double(0.4)
0051     )
0052
0053 JetPlusTrackZSPCorJetAntiKt4.JetTracksAssociationAtVertex   = "ak4JetTracksAssociatorAtVertexJPT"
0054 JetPlusTrackZSPCorJetAntiKt4.JetTracksAssociationAtCaloFace = "ak4JetTracksAssociatorAtCaloFace"
0055 JetPlusTrackZSPCorJetAntiKt4.JetSplitMerge = 2
0056
0057 ### ---------- Sequences
0058
0059 # Anti-Kt
0060
0061 JetPlusTrackCorrectionsAntiKt4Task = cms.Task(
0062     JetPlusTrackAddonSeedReco,
0063     ak4JetTracksAssociatorAtVertexJPT,
0064     ak4JetTracksAssociatorAtCaloFace,
0065     JetPlusTrackZSPCorJetAntiKt4
0066     )
0067 JetPlusTrackCorrectionsAntiKt4 = cms.Sequence(JetPlusTrackCorrectionsAntiKt4Task)
0068
0069 # For backward-compatiblity (but to be deprecated!)
0070 JetPlusTrackCorrections = cms.Sequence(JetPlusTrackCorrectionsAntiKt4)