IterativeTracking/python/LowPtQuadStep_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002 import RecoTracker.IterativeTracking.iterativeTkConfig as _cfg
0003 from Configuration.Eras.Modifier_fastSim_cff import fastSim
0004
0005 # for dnn classifier
0006 from Configuration.ProcessModifiers.trackdnn_cff import trackdnn
0007 from RecoTracker.IterativeTracking.dnnQualityCuts import qualityCutDictionary
0008
0009 # for no-loopers
0010 from Configuration.ProcessModifiers.trackingNoLoopers_cff import trackingNoLoopers
0011
0012 # NEW CLUSTERS (remove previously used clusters)
0013 lowPtQuadStepClusters = _cfg.clusterRemoverForIter('LowPtQuadStep')
0014 for _eraName, _postfix, _era in _cfg.nonDefaultEras():
0015     _era.toReplaceWith(lowPtQuadStepClusters, _cfg.clusterRemoverForIter('LowPtQuadStep', _eraName, _postfix))
0016
0017 from Configuration.ProcessModifiers.trackingLST_cff import trackingLST
0018 # with LST, this is the first iteration with proper cluster masking
0019 trackingLST.toModify(lowPtQuadStepClusters, oldClusterRemovalInfo = "")
0020
0021 # SEEDING LAYERS
0022 import RecoTracker.TkSeedingLayers.PixelLayerQuadruplets_cfi
0023 lowPtQuadStepSeedLayers = RecoTracker.TkSeedingLayers.PixelLayerQuadruplets_cfi.PixelLayerQuadruplets.clone(
0024     BPix = dict(skipClusters = cms.InputTag('lowPtQuadStepClusters')),
0025     FPix = dict(skipClusters = cms.InputTag('lowPtQuadStepClusters'))
0026 )
0027
0028 # TrackingRegion
0029 from RecoTracker.TkTrackingRegions.globalTrackingRegionFromBeamSpot_cfi import globalTrackingRegionFromBeamSpot as _globalTrackingRegionFromBeamSpot
0030 lowPtQuadStepTrackingRegions = _globalTrackingRegionFromBeamSpot.clone(RegionPSet = dict(
0031     ptMin        = 0.15,
0032     originRadius = 0.02,
0033     nSigmaZ      = 4.0 )
0034 )
0035 from Configuration.Eras.Modifier_trackingPhase2PU140_cff import trackingPhase2PU140
0036 trackingPhase2PU140.toModify(lowPtQuadStepTrackingRegions, RegionPSet = dict(ptMin = 0.35,originRadius = 0.025))
0037
0038 from Configuration.Eras.Modifier_pp_on_XeXe_2017_cff import pp_on_XeXe_2017
0039 from Configuration.ProcessModifiers.pp_on_AA_cff import pp_on_AA
0040 from RecoTracker.TkTrackingRegions.globalTrackingRegionWithVertices_cff import globalTrackingRegionWithVertices as _globalTrackingRegionWithVertices
0041 (pp_on_XeXe_2017 | pp_on_AA).toReplaceWith(lowPtQuadStepTrackingRegions,
0042                 _globalTrackingRegionWithVertices.clone(RegionPSet=dict(
0043                     fixedError   = 0.5,
0044                     ptMin        = 0.49,
0045                     originRadius = 0.02 )
0046                 )
0047 )
0048 from Configuration.Eras.Modifier_highBetaStar_cff import highBetaStar
0049 highBetaStar.toModify(lowPtQuadStepTrackingRegions,RegionPSet = dict(
0050      ptMin        = 0.05,
0051      originRadius = 0.2, )
0052 )
0053
0054 # seeding
0055 from RecoTracker.TkHitPairs.hitPairEDProducer_cfi import hitPairEDProducer as _hitPairEDProducer
0056 lowPtQuadStepHitDoublets = _hitPairEDProducer.clone(
0057     seedingLayers   = 'lowPtQuadStepSeedLayers',
0058     trackingRegions = 'lowPtQuadStepTrackingRegions',
0059     layerPairs      = [0,1,2], # layer pairs (0,1), (1,2), (2,3)
0060     maxElement      = 50000000,
0061     produceIntermediateHitDoublets = True,
0062 )
0063 from RecoTracker.PixelSeeding.caHitQuadrupletEDProducer_cfi import caHitQuadrupletEDProducer as _caHitQuadrupletEDProducer
0064 from RecoTracker.PixelSeeding.pixelTripletHLTEDProducer_cfi import pixelTripletHLTEDProducer as _pixelTripletHLTEDProducer
0065 from RecoTracker.PixelLowPtUtilities.ClusterShapeHitFilterESProducer_cfi import *
0066 import RecoTracker.PixelLowPtUtilities.LowPtClusterShapeSeedComparitor_cfi
0067 lowPtQuadStepHitQuadruplets = _caHitQuadrupletEDProducer.clone(
0068     doublets = 'lowPtQuadStepHitDoublets',
0069     extraHitRPhitolerance = _pixelTripletHLTEDProducer.extraHitRPhitolerance,
0070     SeedComparitorPSet = RecoTracker.PixelLowPtUtilities.LowPtClusterShapeSeedComparitor_cfi.LowPtClusterShapeSeedComparitor.clone(),
0071     maxChi2 = dict(
0072         pt1    = 0.7, pt2    = 2,
0073         value1 = 1000, value2 = 150,
0074     ),
0075     useBendingCorrection = True,
0076     fitFastCircle        = True,
0077     fitFastCircleChi2Cut = True,
0078     CAThetaCut           = 0.0017,
0079     CAPhiCut             = 0.3,
0080 )
0081 trackingPhase2PU140.toModify(lowPtQuadStepHitQuadruplets,CAThetaCut = 0.0015,CAPhiCut = 0.25)
0082 highBetaStar.toModify(lowPtQuadStepHitQuadruplets,CAThetaCut = 0.0034,CAPhiCut = 0.6)
0083
0084 from RecoTracker.TkSeedGenerator.seedCreatorFromRegionConsecutiveHitsEDProducer_cff import seedCreatorFromRegionConsecutiveHitsEDProducer as _seedCreatorFromRegionConsecutiveHitsEDProducer
0085 lowPtQuadStepSeeds = _seedCreatorFromRegionConsecutiveHitsEDProducer.clone(
0086     seedingHitSets = 'lowPtQuadStepHitQuadruplets',
0087 )
0088
0089 #For FastSim phase1 tracking
0090 import FastSimulation.Tracking.TrajectorySeedProducer_cfi
0091 from FastSimulation.Tracking.SeedingMigration import _hitSetProducerToFactoryPSet
0092 _fastSim_lowPtQuadStepSeeds = FastSimulation.Tracking.TrajectorySeedProducer_cfi.trajectorySeedProducer.clone(
0093     trackingRegions = 'lowPtQuadStepTrackingRegions',
0094     hitMasks        = cms.InputTag('lowPtQuadStepMasks'),
0095     seedFinderSelector = dict( CAHitQuadrupletGeneratorFactory = _hitSetProducerToFactoryPSet(lowPtQuadStepHitQuadruplets).clone(
0096             SeedComparitorPSet = dict(ComponentName = 'none')),
0097                                layerList = lowPtQuadStepSeedLayers.layerList.value(),
0098                                #new parameters required for phase1 seeding
0099                                BPix      = dict(TTRHBuilder = 'WithoutRefit', HitProducer = 'TrackingRecHitProducer',),
0100                                FPix      = dict(TTRHBuilder = 'WithoutRefit', HitProducer = 'TrackingRecHitProducer',),
0101                                layerPairs = lowPtQuadStepHitDoublets.layerPairs.value()
0102                                )
0103 )
0104 fastSim.toReplaceWith(lowPtQuadStepSeeds,_fastSim_lowPtQuadStepSeeds)
0105
0106 # QUALITY CUTS DURING TRACK BUILDING
0107 import TrackingTools.TrajectoryFiltering.TrajectoryFilter_cff as _TrajectoryFilter_cff
0108 _lowPtQuadStepTrajectoryFilterBase = _TrajectoryFilter_cff.CkfBaseTrajectoryFilter_block.clone(
0109     minimumNumberOfHits = 3,
0110     minPt               = 0.075,
0111 )
0112 lowPtQuadStepTrajectoryFilterBase = _lowPtQuadStepTrajectoryFilterBase.clone(
0113     maxCCCLostHits     = 0,
0114     minGoodStripCharge = dict(refToPSet_ = 'SiStripClusterChargeCutLoose')
0115 )
0116 trackingPhase2PU140.toReplaceWith(lowPtQuadStepTrajectoryFilterBase, _lowPtQuadStepTrajectoryFilterBase)
0117
0118 (pp_on_XeXe_2017 | pp_on_AA).toModify(lowPtQuadStepTrajectoryFilterBase, minPt=0.49)
0119
0120 from RecoTracker.PixelLowPtUtilities.ClusterShapeTrajectoryFilter_cfi import *
0121 # Composite filter
0122 lowPtQuadStepTrajectoryFilter = _TrajectoryFilter_cff.CompositeTrajectoryFilter_block.clone(
0123     filters = [cms.PSet(refToPSet_ = cms.string('lowPtQuadStepTrajectoryFilterBase'))]
0124 )
0125 trackingPhase2PU140.toModify(lowPtQuadStepTrajectoryFilter,
0126     filters = lowPtQuadStepTrajectoryFilter.filters.value() + [cms.PSet(refToPSet_ = cms.string('ClusterShapeTrajectoryFilter'))]
0127 )
0128
0129 import RecoTracker.MeasurementDet.Chi2ChargeMeasurementEstimator_cfi
0130 lowPtQuadStepChi2Est = RecoTracker.MeasurementDet.Chi2ChargeMeasurementEstimator_cfi.Chi2ChargeMeasurementEstimator.clone(
0131     ComponentName    = 'lowPtQuadStepChi2Est',
0132     nSigma           = 3.0,
0133     MaxChi2          = 9.0,
0134     clusterChargeCut = dict(refToPSet_ = ('SiStripClusterChargeCutTight')),
0135 )
0136 trackingPhase2PU140.toModify(lowPtQuadStepChi2Est,
0137     MaxChi2          = 16.0,
0138     clusterChargeCut = dict(refToPSet_ = 'SiStripClusterChargeCutNone')
0139 )
0140
0141 # TRACK BUILDING
0142 import RecoTracker.CkfPattern.GroupedCkfTrajectoryBuilder_cfi
0143 lowPtQuadStepTrajectoryBuilder = RecoTracker.CkfPattern.GroupedCkfTrajectoryBuilder_cfi.GroupedCkfTrajectoryBuilderIterativeDefault.clone(
0144     trajectoryFilter       = dict(refToPSet_ = 'lowPtQuadStepTrajectoryFilter'),
0145     maxCand                = 4,
0146     estimator              = 'lowPtQuadStepChi2Est',
0147     maxDPhiForLooperReconstruction = 2.0,
0148     # 0.63 GeV is the maximum pT for a charged particle to loop within the 1.1m radius
0149     # of the outermost Tracker barrel layer (with B=3.8T)
0150     maxPtForLooperReconstruction = 0.7,
0151 )
0152 trackingNoLoopers.toModify(lowPtQuadStepTrajectoryBuilder,
0153                            maxPtForLooperReconstruction = 0.0)
0154 trackingPhase2PU140.toModify(lowPtQuadStepTrajectoryBuilder,
0155     minNrOfHitsForRebuild      = 1,
0156     keepOriginalIfRebuildFails = True,
0157 )
0158
0159
0160 # MAKING OF TRACK CANDIDATES
0161 from TrackingTools.TrajectoryCleaning.TrajectoryCleanerBySharedHits_cfi import trajectoryCleanerBySharedHits as _trajectoryCleanerBySharedHits
0162 lowPtQuadStepTrajectoryCleanerBySharedHits = _trajectoryCleanerBySharedHits.clone(
0163     ComponentName       = 'lowPtQuadStepTrajectoryCleanerBySharedHits',
0164     fractionShared      = 0.16,
0165     allowSharedFirstHit = True
0166 )
0167 trackingPhase2PU140.toModify(lowPtQuadStepTrajectoryCleanerBySharedHits, fractionShared = 0.09)
0168
0169 import RecoTracker.CkfPattern.CkfTrackCandidates_cfi
0170 lowPtQuadStepTrackCandidates = RecoTracker.CkfPattern.CkfTrackCandidates_cfi.ckfTrackCandidatesIterativeDefault.clone(
0171     src = 'lowPtQuadStepSeeds',
0172     ### these two parameters are relevant only for the CachingSeedCleanerBySharedInput
0173     numHitsForSeedCleaner       = 50,
0174     onlyPixelHitsForSeedCleaner = True,
0175     TrajectoryBuilderPSet       = dict(refToPSet_ = 'lowPtQuadStepTrajectoryBuilder'),
0176     TrajectoryCleaner           = 'lowPtQuadStepTrajectoryCleanerBySharedHits',
0177     clustersToSkip              = 'lowPtQuadStepClusters',
0178     doSeedingRegionRebuilding   = True,
0179     useHitsSplitting            = True,
0180 )
0181 trackingPhase2PU140.toModify(lowPtQuadStepTrackCandidates,
0182     clustersToSkip       = '',
0183     phase2clustersToSkip = 'lowPtQuadStepClusters'
0184 )
0185
0186 from Configuration.ProcessModifiers.trackingMkFitLowPtQuadStep_cff import trackingMkFitLowPtQuadStep
0187 import RecoTracker.MkFit.mkFitSeedConverter_cfi as mkFitSeedConverter_cfi
0188 import RecoTracker.MkFit.mkFitIterationConfigESProducer_cfi as mkFitIterationConfigESProducer_cfi
0189 import RecoTracker.MkFit.mkFitProducer_cfi as mkFitProducer_cfi
0190 import RecoTracker.MkFit.mkFitOutputConverter_cfi as mkFitOutputConverter_cfi
0191 lowPtQuadStepTrackCandidatesMkFitSeeds = mkFitSeedConverter_cfi.mkFitSeedConverter.clone(
0192     seeds = 'lowPtQuadStepSeeds',
0193 )
0194 lowPtQuadStepTrackCandidatesMkFitConfig = mkFitIterationConfigESProducer_cfi.mkFitIterationConfigESProducer.clone(
0195     ComponentName = 'lowPtQuadStepTrackCandidatesMkFitConfig',
0196     config = 'RecoTracker/MkFit/data/mkfit-phase1-lowPtQuadStep.json',
0197 )
0198 lowPtQuadStepTrackCandidatesMkFit = mkFitProducer_cfi.mkFitProducer.clone(
0199     seeds = 'lowPtQuadStepTrackCandidatesMkFitSeeds',
0200     config = ('', 'lowPtQuadStepTrackCandidatesMkFitConfig'),
0201     clustersToSkip = 'lowPtQuadStepClusters',
0202 )
0203 trackingMkFitLowPtQuadStep.toReplaceWith(lowPtQuadStepTrackCandidates, mkFitOutputConverter_cfi.mkFitOutputConverter.clone(
0204     seeds = 'lowPtQuadStepSeeds',
0205     mkFitSeeds = 'lowPtQuadStepTrackCandidatesMkFitSeeds',
0206     tracks = 'lowPtQuadStepTrackCandidatesMkFit',
0207 ))
0208 (pp_on_XeXe_2017 | pp_on_AA).toModify(lowPtQuadStepTrackCandidatesMkFitConfig, minPt=0.49)
0209
0210 #For FastSim phase1 tracking
0211 import FastSimulation.Tracking.TrackCandidateProducer_cfi
0212 _fastSim_lowPtQuadStepTrackCandidates = FastSimulation.Tracking.TrackCandidateProducer_cfi.trackCandidateProducer.clone(
0213     src                      = 'lowPtQuadStepSeeds',
0214     MinNumberOfCrossedLayers = 3,
0215     hitMasks                 = cms.InputTag('lowPtQuadStepMasks')
0216 )
0217 fastSim.toReplaceWith(lowPtQuadStepTrackCandidates,_fastSim_lowPtQuadStepTrackCandidates)
0218
0219 # TRACK FITTING
0220 import RecoTracker.TrackProducer.TrackProducerIterativeDefault_cfi
0221 lowPtQuadStepTracks = RecoTracker.TrackProducer.TrackProducerIterativeDefault_cfi.TrackProducerIterativeDefault.clone(
0222     src           = 'lowPtQuadStepTrackCandidates',
0223     AlgorithmName = 'lowPtQuadStep',
0224     Fitter        = 'FlexibleKFFittingSmoother',
0225 )
0226 fastSim.toModify(lowPtQuadStepTracks,TTRHBuilder = 'WithoutRefit')
0227
0228 from Configuration.Eras.Modifier_phase2_timing_layer_cff import phase2_timing_layer
0229 phase2_timing_layer.toModify(lowPtQuadStepTracks, TrajectoryInEvent = True)
0230
0231 # Final selection
0232 from RecoTracker.FinalTrackSelectors.TrackMVAClassifierPrompt_cfi import *
0233 lowPtQuadStep = TrackMVAClassifierPrompt.clone(
0234      mva         = dict(GBRForestLabel = 'MVASelectorLowPtQuadStep_Phase1'),
0235      src         = 'lowPtQuadStepTracks',
0236      qualityCuts = [-0.7,-0.35,-0.15]
0237 )
0238 pp_on_AA.toModify(lowPtQuadStep,
0239         mva         = dict(GBRForestLabel = 'HIMVASelectorLowPtQuadStep_Phase1'),
0240         qualityCuts = [-0.9, -0.4, 0.3],
0241 )
0242
0243 from RecoTracker.FinalTrackSelectors.trackTfClassifier_cfi import *
0244 from RecoTracker.FinalTrackSelectors.trackSelectionTf_cfi import *
0245 from RecoTracker.FinalTrackSelectors.trackSelectionTf_CKF_cfi import *
0246 trackdnn.toReplaceWith(lowPtQuadStep, trackTfClassifier.clone(
0247     src = 'lowPtQuadStepTracks',
0248     qualityCuts = qualityCutDictionary.LowPtQuadStep.value()
0249 ))
0250 highBetaStar.toModify(lowPtQuadStep,qualityCuts = [-0.9,-0.35,-0.15])
0251 fastSim.toModify(lowPtQuadStep,vertices = 'firstStepPrimaryVerticesBeforeMixing')
0252
0253 ((~trackingMkFitLowPtQuadStep) & trackdnn).toModify(lowPtQuadStep, mva = dict(tfDnnLabel  = 'trackSelectionTf_CKF'),
0254                                                     qualityCuts = [-0.33,  0.13,  0.35])
0255
0256 (pp_on_AA & trackdnn).toModify(lowPtQuadStep, qualityCuts =  [-0.35, 0.8, 0.98] )
0257
0258 # For Phase2PU140
0259 import RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi
0260 lowPtQuadStepSelector = RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.multiTrackSelector.clone(
0261     src = 'lowPtQuadStepTracks',
0262     trackSelectors = [
0263         RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.looseMTS.clone(
0264             name = 'lowPtQuadStepLoose',
0265             chi2n_par = 2.0,
0266             res_par = ( 0.003, 0.002 ),
0267             minNumberLayers = 3,
0268             maxNumberLostLayers = 2,
0269             minNumber3DLayers = 3,
0270             d0_par1 = ( 0.8, 4.0 ),
0271             dz_par1 = ( 0.7, 4.0 ),
0272             d0_par2 = ( 0.6, 4.0 ),
0273             dz_par2 = ( 0.6, 4.0 )
0274             ), #end of pset
0275         RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.tightMTS.clone(
0276             name = 'lowPtQuadStepTight',
0277             preFilterName = 'lowPtQuadStepLoose',
0278             chi2n_par = 1.4,
0279             res_par = ( 0.003, 0.002 ),
0280             minNumberLayers = 3,
0281             maxNumberLostLayers = 2,
0282             minNumber3DLayers = 3,
0283             d0_par1 = ( 0.7, 4.0 ),
0284             dz_par1 = ( 0.6, 4.0 ),
0285             d0_par2 = ( 0.5, 4.0 ),
0286             dz_par2 = ( 0.5, 4.0 )
0287             ),
0288         RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.highpurityMTS.clone(
0289             name = 'lowPtQuadStep',
0290             preFilterName = 'lowPtQuadStepTight',
0291             min_eta = -4.0,
0292             max_eta = 4.0,
0293             chi2n_par = 1.2,
0294             res_par = ( 0.003, 0.001 ),
0295             minNumberLayers = 3,
0296             maxNumberLostLayers = 2,
0297             minNumber3DLayers = 3,
0298             d0_par1 = ( 0.5, 4.0 ),
0299             dz_par1 = ( 0.5, 4.0 ),
0300             d0_par2 = ( 0.45, 4.0 ),
0301             dz_par2 = ( 0.45, 4.0 )
0302             ),
0303     ] #end of vpset
0304 ) #end of clone
0305
0306
0307 # Final sequence
0308 LowPtQuadStepTask = cms.Task(lowPtQuadStepClusters,
0309                              lowPtQuadStepSeedLayers,
0310                              lowPtQuadStepTrackingRegions,
0311                              lowPtQuadStepHitDoublets,
0312                              lowPtQuadStepHitQuadruplets,
0313                              lowPtQuadStepSeeds,
0314                              lowPtQuadStepTrackCandidates,
0315                              lowPtQuadStepTracks,
0316                              lowPtQuadStep)
0317 LowPtQuadStep = cms.Sequence(LowPtQuadStepTask)
0318
0319 _LowPtQuadStepTask_trackingMkFit = LowPtQuadStepTask.copy()
0320 _LowPtQuadStepTask_trackingMkFit.add(lowPtQuadStepTrackCandidatesMkFitSeeds, lowPtQuadStepTrackCandidatesMkFit, lowPtQuadStepTrackCandidatesMkFitConfig)
0321 trackingMkFitLowPtQuadStep.toReplaceWith(LowPtQuadStepTask, _LowPtQuadStepTask_trackingMkFit)
0322
0323 _LowPtQuadStepTask_Phase2PU140 = LowPtQuadStepTask.copy()
0324 _LowPtQuadStepTask_Phase2PU140.replace(lowPtQuadStep, lowPtQuadStepSelector)
0325 trackingPhase2PU140.toReplaceWith(LowPtQuadStepTask, _LowPtQuadStepTask_Phase2PU140)
0326
0327 # fast tracking mask producer
0328 from FastSimulation.Tracking.FastTrackerRecHitMaskProducer_cfi import maskProducerFromClusterRemover
0329 lowPtQuadStepMasks = maskProducerFromClusterRemover(lowPtQuadStepClusters)
0330 fastSim.toReplaceWith(LowPtQuadStepTask,
0331                       cms.Task(lowPtQuadStepMasks
0332                                ,lowPtQuadStepTrackingRegions
0333                                ,lowPtQuadStepSeeds
0334                                ,lowPtQuadStepTrackCandidates
0335                                ,lowPtQuadStepTracks
0336                                ,lowPtQuadStep
0337                                ) )