IterativeTracking/python/iterativeTk_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002 from RecoTracker.TkSeedGenerator.trackerClusterCheck_cfi import *
0003
0004 from RecoTracker.IterativeTracking.InitialStepPreSplitting_cff import *
0005 from RecoTracker.IterativeTracking.InitialStep_cff import *
0006 from RecoTracker.IterativeTracking.DetachedTripletStep_cff import *
0007 from RecoTracker.IterativeTracking.LowPtTripletStep_cff import *
0008 from RecoTracker.IterativeTracking.PixelPairStep_cff import *
0009 from RecoTracker.IterativeTracking.MixedTripletStep_cff import *
0010 from RecoTracker.IterativeTracking.PixelLessStep_cff import *
0011 from RecoTracker.IterativeTracking.TobTecStep_cff import *
0012 from RecoTracker.IterativeTracking.DisplacedGeneralStep_cff import *
0013 from RecoTracker.IterativeTracking.DisplacedRegionalStep_cff import *
0014 from RecoTracker.IterativeTracking.JetCoreRegionalStep_cff import *
0015
0016 # Phase1 specific iterations
0017 from RecoTracker.IterativeTracking.HighPtTripletStep_cff import *
0018 from RecoTracker.IterativeTracking.DetachedQuadStep_cff import *
0019 from RecoTracker.IterativeTracking.LowPtQuadStep_cff import *
0020
0021 from RecoTracker.FinalTrackSelectors.earlyGeneralTracks_cfi import *
0022 from RecoTracker.IterativeTracking.MuonSeededStep_cff import *
0023 from RecoTracker.FinalTrackSelectors.preDuplicateMergingGeneralTracks_cfi import *
0024 from RecoTracker.FinalTrackSelectors.MergeTrackCollections_cff import *
0025 from RecoTracker.ConversionSeedGenerators.ConversionStep_cff import *
0026
0027 import RecoTracker.IterativeTracking.iterativeTkConfig as _cfg
0028 from RecoTracker.FinalTrackSelectors.trackTfClassifier_cfi import *
0029
0030 trackdnn_source = cms.ESSource("EmptyESSource", recordName = cms.string("TfGraphRecord"), firstValid = cms.vuint32(1), iovIsRunNotTime = cms.bool(True) )
0031 iterTrackingEarlyTask = _cfg.createEarlyTask("", "", globals())
0032 for _eraName, _postfix, _era in _cfg.nonDefaultEras():
0033     _era.toReplaceWith(iterTrackingEarlyTask, _cfg.createEarlyTask(_eraName, _postfix, globals()))
0034 iterTrackingEarly = cms.Sequence(iterTrackingEarlyTask)
0035
0036 iterTrackingTask = cms.Task(InitialStepPreSplittingTask,
0037                             trackerClusterCheck,
0038                             iterTrackingEarlyTask,
0039                             earlyGeneralTracks,
0040                             muonSeededStepTask,
0041                             preDuplicateMergingGeneralTracks,
0042                             generalTracksTask,
0043                             ConvStepTask,
0044                             conversionStepTracks
0045                             )
0046
0047 from Configuration.Eras.Modifier_trackingPhase1_cff import trackingPhase1
0048 from Configuration.ProcessModifiers.displacedRegionalTracking_cff import displacedRegionalTracking
0049 (trackingPhase1 & displacedRegionalTracking).toModify(iterTrackingTask, lambda x: x.add(DisplacedRegionalStepTask))
0050
0051 _iterTrackingTask_trackdnn = iterTrackingTask.copy()
0052 _iterTrackingTask_trackdnn.add(trackdnn_source)
0053 trackdnn.toReplaceWith(iterTrackingTask, _iterTrackingTask_trackdnn)
0054 iterTracking = cms.Sequence(iterTrackingTask)