1 import FWCore.ParameterSet.Config
as cms
2 from Configuration.Eras.Modifier_tracker_apv_vfp30_2016_cff
import tracker_apv_vfp30_2016
as _tracker_apv_vfp30_2016
3 from Configuration.Eras.Modifier_fastSim_cff
import fastSim
6 from Configuration.ProcessModifiers.trackdnn_cff
import trackdnn
7 from RecoTracker.IterativeTracking.dnnQualityCuts
import qualityCutDictionary
17 import RecoTracker.TkSeedingLayers.PixelLayerQuadruplets_cfi
18 initialStepSeedLayers = RecoTracker.TkSeedingLayers.PixelLayerTriplets_cfi.PixelLayerTriplets.clone()
19 from Configuration.Eras.Modifier_trackingPhase1_cff
import trackingPhase1
20 trackingPhase1.toModify(initialStepSeedLayers,
21 layerList = RecoTracker.TkSeedingLayers.PixelLayerQuadruplets_cfi.PixelLayerQuadruplets.layerList.value()
23 trackingPhase2PU140.toModify(initialStepSeedLayers,
24 layerList = RecoTracker.TkSeedingLayers.PixelLayerQuadruplets_cfi.PixelLayerQuadruplets.layerList.value()
28 from RecoTracker.TkTrackingRegions.globalTrackingRegionFromBeamSpot_cfi
import globalTrackingRegionFromBeamSpot
as _globalTrackingRegionFromBeamSpot
29 initialStepTrackingRegions = _globalTrackingRegionFromBeamSpot.clone(RegionPSet = dict(
34 from Configuration.Eras.Modifier_trackingPhase2PU140_cff
import trackingPhase2PU140
35 trackingPhase1.toModify(initialStepTrackingRegions, RegionPSet = dict(ptMin = 0.5))
36 from Configuration.Eras.Modifier_highBetaStar_2018_cff
import highBetaStar_2018
37 highBetaStar_2018.toModify(initialStepTrackingRegions,RegionPSet = dict(
41 trackingPhase2PU140.toModify(initialStepTrackingRegions, RegionPSet = dict(ptMin = 0.6,originRadius = 0.03))
44 from RecoTracker.TkHitPairs.hitPairEDProducer_cfi
import hitPairEDProducer
as _hitPairEDProducer
45 initialStepHitDoublets = _hitPairEDProducer.clone(
46 seedingLayers =
'initialStepSeedLayers',
47 trackingRegions =
'initialStepTrackingRegions',
48 maxElement = 50000000,
49 produceIntermediateHitDoublets =
True,
51 from RecoPixelVertexing.PixelTriplets.pixelTripletHLTEDProducer_cfi
import pixelTripletHLTEDProducer
as _pixelTripletHLTEDProducer
53 import RecoPixelVertexing.PixelLowPtUtilities.LowPtClusterShapeSeedComparitor_cfi
54 initialStepHitTriplets = _pixelTripletHLTEDProducer.clone(
55 doublets =
'initialStepHitDoublets',
56 produceSeedingHitSets =
True,
57 SeedComparitorPSet = RecoPixelVertexing.PixelLowPtUtilities.LowPtClusterShapeSeedComparitor_cfi.LowPtClusterShapeSeedComparitor.clone()
59 from RecoTracker.TkSeedGenerator.seedCreatorFromRegionConsecutiveHitsEDProducer_cff
import seedCreatorFromRegionConsecutiveHitsEDProducer
as _seedCreatorFromRegionConsecutiveHitsEDProducer
60 initialStepSeeds = _seedCreatorFromRegionConsecutiveHitsEDProducer.clone(
61 seedingHitSets =
'initialStepHitTriplets',
63 from RecoPixelVertexing.PixelTriplets.caHitQuadrupletEDProducer_cfi
import caHitQuadrupletEDProducer
as _caHitQuadrupletEDProducer
64 _initialStepCAHitQuadruplets = _caHitQuadrupletEDProducer.clone(
65 doublets =
'initialStepHitDoublets',
66 extraHitRPhitolerance = initialStepHitTriplets.extraHitRPhitolerance,
67 SeedComparitorPSet = initialStepHitTriplets.SeedComparitorPSet,
70 value1 = 200, value2 = 50,
72 useBendingCorrection =
True,
74 fitFastCircleChi2Cut =
True,
78 highBetaStar_2018.toModify(_initialStepCAHitQuadruplets,
82 initialStepHitQuadruplets = _initialStepCAHitQuadruplets.clone()
84 trackingPhase1.toModify(initialStepHitDoublets, layerPairs = [0,1,2])
86 trackingPhase2PU140.toModify(initialStepHitDoublets, layerPairs = [0,1,2])
87 trackingPhase2PU140.toModify(initialStepHitQuadruplets,
92 from RecoTracker.TkSeedGenerator.seedCreatorFromRegionConsecutiveHitsTripletOnlyEDProducer_cff
import seedCreatorFromRegionConsecutiveHitsTripletOnlyEDProducer
as _seedCreatorFromRegionConsecutiveHitsTripletOnlyEDProducer
93 _initialStepSeedsConsecutiveHitsTripletOnly = _seedCreatorFromRegionConsecutiveHitsTripletOnlyEDProducer.clone(
94 seedingHitSets =
'initialStepHitTriplets',
95 SeedComparitorPSet = dict(
96 ComponentName =
'PixelClusterShapeSeedComparitor',
97 FilterAtHelixStage = cms.bool(
False),
98 FilterPixelHits = cms.bool(
True),
99 FilterStripHits = cms.bool(
False),
100 ClusterShapeHitFilterName = cms.string(
'ClusterShapeHitFilter'),
101 ClusterShapeCacheSrc = cms.InputTag(
'siPixelClusterShapeCache')
104 trackingPhase1.toReplaceWith(initialStepSeeds, _initialStepSeedsConsecutiveHitsTripletOnly.clone(
105 seedingHitSets =
'initialStepHitQuadruplets'
107 trackingPhase2PU140.toReplaceWith(initialStepSeeds, _initialStepSeedsConsecutiveHitsTripletOnly.clone(
108 seedingHitSets =
'initialStepHitQuadruplets'
110 import FastSimulation.Tracking.TrajectorySeedProducer_cfi
111 from FastSimulation.Tracking.SeedingMigration
import _hitSetProducerToFactoryPSet
112 _fastSim_initialStepSeeds = FastSimulation.Tracking.TrajectorySeedProducer_cfi.trajectorySeedProducer.clone(
113 trackingRegions =
'initialStepTrackingRegions',
115 layerList = initialStepSeedLayers.layerList.value()
119 trackingPhase1.toModify(_fastSim_initialStepSeeds, seedFinderSelector = dict(
120 pixelTripletGeneratorFactory =
None,
124 TTRHBuilder =
'WithoutRefit',
125 HitProducer =
'TrackingRecHitProducer',
128 TTRHBuilder =
'WithoutRefit',
129 HitProducer =
'TrackingRecHitProducer',
131 layerPairs = initialStepHitDoublets.layerPairs.value()
135 fastSim.toReplaceWith(initialStepSeeds,_fastSim_initialStepSeeds)
140 _initialStepTrajectoryFilterBase = TrackingTools.TrajectoryFiltering.TrajectoryFilter_cff.CkfBaseTrajectoryFilter_block.clone(
141 minimumNumberOfHits = 3,
144 initialStepTrajectoryFilterBase = _initialStepTrajectoryFilterBase.clone(
146 minGoodStripCharge = cms.PSet(refToPSet_ = cms.string(
'SiStripClusterChargeCutLoose'))
148 from Configuration.Eras.Modifier_tracker_apv_vfp30_2016_cff
import tracker_apv_vfp30_2016
149 _tracker_apv_vfp30_2016.toModify(initialStepTrajectoryFilterBase, maxCCCLostHits = 2)
151 from Configuration.Eras.Modifier_pp_on_XeXe_2017_cff
import pp_on_XeXe_2017
152 from Configuration.ProcessModifiers.pp_on_AA_cff
import pp_on_AA
153 (pp_on_XeXe_2017 | pp_on_AA).toModify(initialStepTrajectoryFilterBase, minPt=0.6)
154 highBetaStar_2018.toModify(initialStepTrajectoryFilterBase, minPt = 0.05)
156 initialStepTrajectoryFilterInOut = initialStepTrajectoryFilterBase.clone(
157 minimumNumberOfHits = 4,
159 strictSeedExtension =
True,
160 pixelSeedExtension =
True,
162 from Configuration.Eras.Modifier_trackingLowPU_cff
import trackingLowPU
163 trackingLowPU.toReplaceWith(initialStepTrajectoryFilterBase, _initialStepTrajectoryFilterBase)
164 trackingPhase2PU140.toReplaceWith(initialStepTrajectoryFilterBase, _initialStepTrajectoryFilterBase)
166 import RecoPixelVertexing.PixelLowPtUtilities.StripSubClusterShapeTrajectoryFilter_cfi
167 initialStepTrajectoryFilterShape = RecoPixelVertexing.PixelLowPtUtilities.StripSubClusterShapeTrajectoryFilter_cfi.StripSubClusterShapeTrajectoryFilterTIX12.clone()
168 initialStepTrajectoryFilter = cms.PSet(
169 ComponentType = cms.string(
'CompositeTrajectoryFilter'),
171 cms.PSet( refToPSet_ = cms.string(
'initialStepTrajectoryFilterBase')),
176 trackingPhase2PU140.toReplaceWith(initialStepTrajectoryFilter, TrackingTools.TrajectoryFiltering.TrajectoryFilter_cff.CkfBaseTrajectoryFilter_block.clone(
177 minimumNumberOfHits = 3,
180 import RecoTracker.MeasurementDet.Chi2ChargeMeasurementEstimator_cfi
181 initialStepChi2Est = RecoTracker.MeasurementDet.Chi2ChargeMeasurementEstimator_cfi.Chi2ChargeMeasurementEstimator.clone(
182 ComponentName =
'initialStepChi2Est',
185 clusterChargeCut = cms.PSet(refToPSet_ = cms.string(
'SiStripClusterChargeCutLoose')),
186 pTChargeCutThreshold = 15.
188 _tracker_apv_vfp30_2016.toModify(initialStepChi2Est,
189 clusterChargeCut = dict(refToPSet_ =
'SiStripClusterChargeCutTiny')
191 trackingPhase2PU140.toModify(initialStepChi2Est,
192 clusterChargeCut = dict(refToPSet_ =
'SiStripClusterChargeCutNone'),
197 initialStepTrajectoryBuilder = RecoTracker.CkfPattern.GroupedCkfTrajectoryBuilder_cfi.GroupedCkfTrajectoryBuilder.clone(
198 trajectoryFilter = cms.PSet(refToPSet_ = cms.string(
'initialStepTrajectoryFilter')),
199 alwaysUseInvalidHits =
True,
201 estimator =
'initialStepChi2Est',
202 maxDPhiForLooperReconstruction = cms.double(2.0),
203 maxPtForLooperReconstruction = cms.double(0.7)
205 trackingLowPU.toModify(initialStepTrajectoryBuilder, maxCand = 5)
206 trackingPhase1.toModify(initialStepTrajectoryBuilder,
207 minNrOfHitsForRebuild = 1,
208 keepOriginalIfRebuildFails =
True,
210 trackingPhase2PU140.toModify(initialStepTrajectoryBuilder,
211 minNrOfHitsForRebuild = 1,
212 keepOriginalIfRebuildFails =
True,
216 initialStepTrackCandidates = RecoTracker.CkfPattern.CkfTrackCandidates_cfi.ckfTrackCandidates.clone(
217 src =
'initialStepSeeds',
219 numHitsForSeedCleaner = cms.int32(50),
220 onlyPixelHitsForSeedCleaner = cms.bool(
True),
221 TrajectoryBuilderPSet = cms.PSet(refToPSet_ = cms.string(
'initialStepTrajectoryBuilder')),
222 doSeedingRegionRebuilding =
True,
223 useHitsSplitting =
True
226 from Configuration.ProcessModifiers.trackingMkFit_cff
import trackingMkFit
227 import RecoTracker.MkFit.mkFitInputConverter_cfi
as mkFitInputConverter_cfi
228 import RecoTracker.MkFit.mkFitProducer_cfi
as mkFitProducer_cfi
229 import RecoTracker.MkFit.mkFitOutputConverter_cfi
as mkFitOutputConverter_cfi
230 initialStepTrackCandidatesMkFitInput = mkFitInputConverter_cfi.mkFitInputConverter.clone(
231 seeds =
'initialStepSeeds',
233 initialStepTrackCandidatesMkFit = mkFitProducer_cfi.mkFitProducer.clone(
234 hitsSeeds =
'initialStepTrackCandidatesMkFitInput',
236 trackingMkFit.toReplaceWith(initialStepTrackCandidates, mkFitOutputConverter_cfi.mkFitOutputConverter.clone(
237 seeds =
'initialStepSeeds',
238 hitsSeeds =
'initialStepTrackCandidatesMkFitInput',
239 tracks =
'initialStepTrackCandidatesMkFit',
242 import FastSimulation.Tracking.TrackCandidateProducer_cfi
243 fastSim.toReplaceWith(initialStepTrackCandidates,
244 FastSimulation.Tracking.TrackCandidateProducer_cfi.trackCandidateProducer.clone(
245 src =
'initialStepSeeds',
246 MinNumberOfCrossedLayers = 3
252 initialStepTracks = RecoTracker.TrackProducer.TrackProducer_cfi.TrackProducer.clone(
253 src =
'initialStepTrackCandidates',
254 AlgorithmName =
'initialStep',
255 Fitter =
'FlexibleKFFittingSmoother'
257 fastSim.toModify(initialStepTracks, TTRHBuilder =
'WithoutRefit')
261 firstStepPrimaryVerticesUnsorted = _offlinePrimaryVertices.clone(
262 TrackLabel =
'initialStepTracks',
263 vertexCollections = [_offlinePrimaryVertices.vertexCollections[0].
clone()]
265 (pp_on_XeXe_2017 | pp_on_AA).toModify(firstStepPrimaryVerticesUnsorted, TkFilterParameters = dict(trackQuality =
'any'))
271 firstStepPrimaryVerticesBeforeMixing = firstStepPrimaryVerticesUnsorted.clone()
272 fastSim.toModify(firstStepPrimaryVerticesUnsorted, TrackLabel =
'generalTracks')
276 initialStepTrackRefsForJets = trackRefsForJets.clone(
277 src =
'initialStepTracks'
279 fastSim.toModify(initialStepTrackRefsForJets, src =
'generalTracks')
282 firstStepPrimaryVertices = _sortedPrimaryVertices.clone(
283 vertices =
'firstStepPrimaryVerticesUnsorted',
284 particles =
'initialStepTrackRefsForJets',
292 initialStepClassifier1 = TrackMVAClassifierPrompt.clone(
293 src =
'initialStepTracks',
294 mva = dict(GBRForestLabel =
'MVASelectorIter0_13TeV'),
295 qualityCuts = [-0.9,-0.8,-0.7]
297 fastSim.toModify(initialStepClassifier1,vertices =
'firstStepPrimaryVerticesBeforeMixing')
301 initialStepClassifier2 = detachedTripletStepClassifier1.clone(
302 src =
'initialStepTracks'
304 fastSim.toModify(initialStepClassifier2,vertices =
'firstStepPrimaryVerticesBeforeMixing')
305 initialStepClassifier3 = lowPtTripletStep.clone(
306 src =
'initialStepTracks'
308 fastSim.toModify(initialStepClassifier3,vertices =
'firstStepPrimaryVerticesBeforeMixing')
311 initialStep = ClassifierMerger.clone(
312 inputClassifiers=[
'initialStepClassifier1',
'initialStepClassifier2',
'initialStepClassifier3']
314 trackingPhase1.toReplaceWith(initialStep, initialStepClassifier1.clone(
315 mva = dict(GBRForestLabel =
'MVASelectorInitialStep_Phase1'),
316 qualityCuts = [-0.95,-0.85,-0.75]
321 trackdnn.toReplaceWith(initialStep, TrackTfClassifier.clone(
322 src =
'initialStepTracks',
323 qualityCuts = qualityCutDictionary[
"InitialStep"]
325 (trackdnn & fastSim).toModify(initialStep,vertices =
'firstStepPrimaryVerticesBeforeMixing')
327 pp_on_AA.toModify(initialStep,
328 mva = dict(GBRForestLabel =
'HIMVASelectorInitialStep_Phase1'),
329 qualityCuts = [-0.9, -0.5, 0.2],
333 import RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi
334 initialStepSelector = RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.multiTrackSelector.clone(
335 src =
'initialStepTracks',
336 useAnyMVA = cms.bool(
False),
337 GBRForestLabel = cms.string(
'MVASelectorIter0'),
339 RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.looseMTS.clone(
340 name =
'initialStepLoose',
342 RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.tightMTS.clone(
343 name =
'initialStepTight',
344 preFilterName =
'initialStepLoose',
346 RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.highpurityMTS.clone(
347 name =
'QualityMasks',
348 preFilterName =
'initialStepTight',
352 trackingPhase2PU140.toModify(initialStepSelector,
354 GBRForestLabel =
None,
355 trackSelectors= cms.VPSet(
356 RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.looseMTS.clone(
357 name =
'initialStepLoose',
359 res_par = ( 0.003, 0.002 ),
361 maxNumberLostLayers = 3,
362 minNumber3DLayers = 3,
363 d0_par1 = ( 0.8, 4.0 ),
364 dz_par1 = ( 0.9, 4.0 ),
365 d0_par2 = ( 0.6, 4.0 ),
366 dz_par2 = ( 0.8, 4.0 )
368 RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.tightMTS.clone(
369 name =
'initialStepTight',
370 preFilterName =
'initialStepLoose',
372 res_par = ( 0.003, 0.002 ),
374 maxNumberLostLayers = 2,
375 minNumber3DLayers = 3,
376 d0_par1 = ( 0.7, 4.0 ),
377 dz_par1 = ( 0.8, 4.0 ),
378 d0_par2 = ( 0.5, 4.0 ),
379 dz_par2 = ( 0.7, 4.0 )
381 RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.highpurityMTS.clone(
382 name =
'initialStep',
383 preFilterName =
'initialStepTight',
387 res_par = ( 0.003, 0.001 ),
389 maxNumberLostLayers = 2,
390 minNumber3DLayers = 3,
391 d0_par1 = ( 0.6, 4.0 ),
392 dz_par1 = ( 0.7, 4.0 ),
393 d0_par2 = ( 0.45, 4.0 ),
394 dz_par2 = ( 0.55, 4.0 )
402 InitialStepTask = cms.Task(initialStepSeedLayers,
403 initialStepTrackingRegions,
404 initialStepHitDoublets,
405 initialStepHitTriplets,
407 initialStepTrackCandidates,
409 firstStepPrimaryVerticesUnsorted,
410 initialStepTrackRefsForJets,
411 firstStepPrimaryVertices,
412 initialStepClassifier1,initialStepClassifier2,initialStepClassifier3,
413 initialStep,caloJetsForTrkTask)
414 InitialStep = cms.Sequence(InitialStepTask)
416 _InitialStepTask_trackingMkFit = InitialStepTask.copy()
417 _InitialStepTask_trackingMkFit.add(initialStepTrackCandidatesMkFitInput, initialStepTrackCandidatesMkFit)
418 trackingMkFit.toReplaceWith(InitialStepTask, _InitialStepTask_trackingMkFit)
420 _InitialStepTask_LowPU = InitialStepTask.copyAndExclude([firstStepPrimaryVerticesUnsorted, initialStepTrackRefsForJets, caloJetsForTrkTask, firstStepPrimaryVertices, initialStepClassifier1, initialStepClassifier2, initialStepClassifier3])
421 _InitialStepTask_LowPU.replace(initialStep, initialStepSelector)
422 trackingLowPU.toReplaceWith(InitialStepTask, _InitialStepTask_LowPU)
424 _InitialStepTask_Phase1 = InitialStepTask.copyAndExclude([initialStepClassifier2, initialStepClassifier3])
425 _InitialStepTask_Phase1.replace(initialStepHitTriplets, initialStepHitQuadruplets)
426 trackingPhase1.toReplaceWith(InitialStepTask, _InitialStepTask_Phase1)
428 _InitialStepTask_trackingPhase2 = InitialStepTask.copyAndExclude([initialStepClassifier1, initialStepClassifier2, initialStepClassifier3])
429 _InitialStepTask_trackingPhase2.replace(initialStepHitTriplets, initialStepHitQuadruplets)
430 _InitialStepTask_trackingPhase2.replace(initialStep, initialStepSelector)
431 trackingPhase2PU140.toReplaceWith(InitialStepTask, _InitialStepTask_trackingPhase2)
433 from Configuration.Eras.Modifier_fastSim_cff
import fastSim
434 _InitialStepTask_fastSim = cms.Task(initialStepTrackingRegions
436 ,initialStepTrackCandidates
438 ,firstStepPrimaryVerticesBeforeMixing
439 ,initialStepClassifier1,initialStepClassifier2,initialStepClassifier3
442 fastSim.toReplaceWith(InitialStepTask, _InitialStepTask_fastSim)