Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2024-05-10 02:21:12

0001 import FWCore.ParameterSet.Config as cms
0002 
0003 # Reco Vertex
0004 # initialize magnetic field #########################
0005 from TrackingTools.TransientTrack.TransientTrackBuilder_cfi import *
0006 from RecoVertex.PrimaryVertexProducer.OfflinePrimaryVertices_cfi import *
0007 from RecoVertex.PrimaryVertexProducer.OfflinePrimaryVerticesWithBS_cfi import *
0008 from RecoVertex.V0Producer.generalV0Candidates_cff import *
0009 from RecoVertex.AdaptiveVertexFinder.inclusiveVertexing_cff import *
0010 
0011 from CommonTools.RecoAlgos.TrackWithVertexRefSelector_cfi import *
0012 from RecoJets.JetProducers.TracksForJets_cff import *
0013 from CommonTools.RecoAlgos.sortedPrimaryVertices_cfi import *
0014 from RecoJets.JetProducers.caloJetsForTrk_cff import *
0015 
0016 unsortedOfflinePrimaryVertices=offlinePrimaryVertices.clone()
0017 offlinePrimaryVertices=sortedPrimaryVertices.clone(
0018     vertices="unsortedOfflinePrimaryVertices", 
0019     particles="trackRefsForJetsBeforeSorting"
0020 )
0021 offlinePrimaryVerticesWithBS=sortedPrimaryVertices.clone(
0022     vertices="unsortedOfflinePrimaryVertices:WithBS", 
0023     particles="trackRefsForJetsBeforeSorting"
0024 )
0025 trackWithVertexRefSelectorBeforeSorting = trackWithVertexRefSelector.clone(
0026     vertexTag="unsortedOfflinePrimaryVertices",
0027     ptMax=9e99,
0028     ptErrorCut=9e99
0029 )
0030 trackRefsForJetsBeforeSorting = trackRefsForJets.clone(src="trackWithVertexRefSelectorBeforeSorting")
0031 
0032 
0033 vertexrecoTask = cms.Task(unsortedOfflinePrimaryVertices,
0034                           trackWithVertexRefSelectorBeforeSorting,
0035                           trackRefsForJetsBeforeSorting,
0036                           offlinePrimaryVertices,
0037                           offlinePrimaryVerticesWithBS,
0038                           generalV0Candidates,
0039                           caloJetsForTrkTask,
0040                           inclusiveVertexingTask
0041                           )
0042 vertexreco = cms.Sequence(vertexrecoTask)
0043 
0044 #modifications for timing
0045 from RecoVertex.Configuration.RecoVertex_phase2_timing_cff import (tpClusterProducer ,
0046                                                                   quickTrackAssociatorByHits ,
0047                                                                   trackTimeValueMapProducer ,
0048                                                                   unsortedOfflinePrimaryVertices4DwithPID ,
0049                                                                   offlinePrimaryVertices4DwithPID ,
0050                                                                   offlinePrimaryVertices4DwithPIDWithBS,
0051                                                                   tofPID,
0052                                                                   tofPID3D,
0053                                                                   tofPID4DnoPID,
0054                                                                   unsortedOfflinePrimaryVertices4D,
0055                                                                   trackWithVertexRefSelectorBeforeSorting4D,
0056                                                                   trackRefsForJetsBeforeSorting4D,
0057                                                                   offlinePrimaryVertices4D,
0058                                                                   offlinePrimaryVertices4DWithBS)
0059 
0060 _phase2_tktiming_vertexrecoTask = cms.Task( vertexrecoTask.copy() ,
0061                                             tpClusterProducer ,
0062                                             quickTrackAssociatorByHits ,
0063                                             trackTimeValueMapProducer ,
0064                                             unsortedOfflinePrimaryVertices4D,
0065                                             trackWithVertexRefSelectorBeforeSorting4D ,
0066                                             trackRefsForJetsBeforeSorting4D,
0067                                             offlinePrimaryVertices4D,
0068                                             offlinePrimaryVertices4DWithBS,
0069                                             )
0070 
0071 _phase2_tktiming_layer_vertexrecoTask = cms.Task( _phase2_tktiming_vertexrecoTask.copy() ,
0072                                             tofPID3D,
0073                                             tofPID,
0074                                             tofPID4DnoPID,
0075                                             )
0076 
0077 from Configuration.Eras.Modifier_phase2_timing_cff import phase2_timing
0078 phase2_timing.toReplaceWith(vertexrecoTask, _phase2_tktiming_vertexrecoTask)
0079 
0080 from Configuration.Eras.Modifier_phase2_timing_layer_cff import phase2_timing_layer
0081 phase2_timing_layer.toReplaceWith(vertexrecoTask, _phase2_tktiming_layer_vertexrecoTask)
0082 phase2_timing_layer.toReplaceWith(unsortedOfflinePrimaryVertices4D, unsortedOfflinePrimaryVertices4DwithPID.clone())
0083 phase2_timing_layer.toReplaceWith(offlinePrimaryVertices4D, offlinePrimaryVertices4DwithPID.clone())
0084 phase2_timing_layer.toReplaceWith(offlinePrimaryVertices4DWithBS, offlinePrimaryVertices4DwithPIDWithBS.clone())
0085 phase2_timing_layer.toModify(offlinePrimaryVertices4D, vertices = "unsortedOfflinePrimaryVertices4D", particles = "trackRefsForJetsBeforeSorting4D")
0086 phase2_timing_layer.toModify(offlinePrimaryVertices4DWithBS, vertices = "unsortedOfflinePrimaryVertices4D:WithBS", particles = "trackRefsForJetsBeforeSorting4D")
0087 phase2_timing_layer.toModify(unsortedOfflinePrimaryVertices,
0088     vertexCollections = {0: dict(vertexTimeParameters = cms.PSet( algorithm = cms.string('fromTracksPID'))),
0089                          1: dict(vertexTimeParameters = cms.PSet( algorithm = cms.string('fromTracksPID')))}
0090 )
0091 
0092 from Configuration.ProcessModifiers.vertex4DTrackSelMVA_cff import vertex4DTrackSelMVA
0093 vertex4DTrackSelMVA.toModify(unsortedOfflinePrimaryVertices4D, useMVACut = True)
0094 vertex4DTrackSelMVA.toModify(unsortedOfflinePrimaryVertices4DwithPID, useMVACut = True)