JetPlusTracks/python/PATJetPlusTrackCorrections_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002 # ---------- Add assigned jet-track association
0003
0004 from RecoJets.JetAssociationProducers.trackExtrapolator_cfi import *
0005 trackExtrapolatorJPTPAT = trackExtrapolator.clone(
0006                       trackSrc = "trackFromPackedCandidate",
0007                       trackQuality = 'highPurity'
0008 )
0009
0010 from RecoJets.JetAssociationProducers.ak4JTA_cff import *
0011 ak4JetTracksAssociatorAtVertexJPTPAT = ak4JetTracksAssociatorAtVertex.clone(
0012                                        useAssigned = True,
0013                                        pvSrc = "offlineSlimmedPrimaryVertices",
0014                                        jets = "slimmedCaloJets",
0015                                        tracks = "trackFromPackedCandidate"
0016 )
0017 ak4JetTracksAssociatorAtCaloFaceJPTPAT = ak4JetTracksAssociatorAtCaloFace.clone(
0018                                          jets = "slimmedCaloJets",
0019                                          tracks = "trackFromPackedCandidate",
0020                                          extrapolations = "trackExtrapolatorJPTPAT"
0021 )
0022 ak4JetExtenderJPTPAT = ak4JetExtender.clone(
0023                        jets = "slimmedCaloJets",
0024                        jet2TracksAtCALO = "ak4JetTracksAssociatorAtCaloFaceJPTPAT",
0025                        jet2TracksAtVX = "ak4JetTracksAssociatorAtVertexJPTPAT"
0026 )
0027
0028 # ---------- Supported Modules
0029
0030 from CommonTools.RecoAlgos.trackFromPackedCandidateProducer_cfi import *
0031 trackFromPackedCandidate = trackFromPackedCandidateProducer.clone(PFCandidates = 'packedPFCandidates')
0032
0033
0034 from RecoJets.JetPlusTracks.jetPlusTrackAddonSeedProducer_cfi import *
0035 JetPlusTrackAddonSeedPAT = jetPlusTrackAddonSeedProducer.clone(
0036     srcCaloJets = "slimmedCaloJets",
0037     srcTrackJets = "ak4TrackJetsJPTPAT",
0038     srcPVs = 'offlineSlimmedPrimaryVertices',
0039     PFCandidates = 'packedPFCandidates',
0040     UsePAT = True
0041 )
0042
0043
0044 from CommonTools.RecoAlgos.TrackWithVertexRefSelector_cfi import *
0045 trackWithVertexRefSelectorJPTPAT = trackWithVertexRefSelector.clone(
0046     vertexTag = 'offlineSlimmedPrimaryVertices',
0047     src = 'trackFromPackedCandidate'
0048 )
0049 from RecoJets.JetProducers.TracksForJets_cff import *
0050 trackRefsForJetsJPTPAT = trackRefsForJets.clone(
0051     src = 'trackWithVertexRefSelectorJPTPAT'
0052 )
0053 from RecoJets.Configuration.RecoTrackJets_cff import *
0054 ak4TrackJetsJPTPAT = ak4TrackJets.clone(
0055     srcPVs = 'offlineSlimmedPrimaryVertices',
0056     UseOnlyOnePV = True,
0057     src = 'trackRefsForJetsJPTPAT'
0058 )
0059
0060 # ---------- Module definition
0061
0062 from RecoJets.JetPlusTracks.JetPlusTrackCorrections_cfi import *
0063 JetPlusTrackZSPCorJetAntiKt4PAT = JetPlusTrackZSPCorJetAntiKt4.clone(
0064     JetTracksAssociationAtVertex = 'ak4JetTracksAssociatorAtVertexJPTPAT',
0065     JetTracksAssociationAtCaloFace = 'ak4JetTracksAssociatorAtCaloFaceJPTPAT',
0066     Muons = 'slimmedMuons',
0067     Electrons = 'slimmedElectrons',
0068     JetSplitMerge = 2,
0069     UsePAT = True
0070 )
0071
0072 ### ---------- Sequences
0073
0074 # Task
0075 PATJetPlusTrackCorrectionsAntiKt4Task = cms.Task(
0076     trackFromPackedCandidate,
0077     trackWithVertexRefSelectorJPTPAT,
0078     trackRefsForJetsJPTPAT,
0079     ak4TrackJetsJPTPAT,
0080     JetPlusTrackAddonSeedPAT,
0081     trackExtrapolatorJPTPAT,
0082     ak4JetTracksAssociatorAtVertexJPTPAT,
0083     ak4JetTracksAssociatorAtCaloFaceJPTPAT,
0084     ak4JetExtenderJPTPAT,
0085     JetPlusTrackZSPCorJetAntiKt4PAT
0086     )
0087
0088 PATJetPlusTrackCorrectionsAntiKt4 = cms.Sequence(PATJetPlusTrackCorrectionsAntiKt4Task)
0089