IterativeTracking/python/JetCoreRegionalStep_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 # for dnn classifier
0004 from Configuration.ProcessModifiers.trackdnn_cff import trackdnn
0005 from RecoTracker.IterativeTracking.dnnQualityCuts import qualityCutDictionary
0006
0007 # for no-loopers
0008 from Configuration.ProcessModifiers.trackingNoLoopers_cff import trackingNoLoopers
0009
0010 # This step runs over all clusters
0011
0012 # run only if there are high pT jets
0013 jetsForCoreTracking = cms.EDFilter('CandPtrSelector', src = cms.InputTag('ak4CaloJetsForTrk'), cut = cms.string('pt > 100 && abs(eta) < 2.5'), filter = cms.bool(False))
0014
0015 jetsForCoreTrackingBarrel = jetsForCoreTracking.clone( cut = 'pt > 100 && abs(eta) < 2.5' )
0016 jetsForCoreTrackingEndcap = jetsForCoreTracking.clone( cut = 'pt > 100 && abs(eta) > 1.4 && abs(eta) < 2.5' )
0017
0018 # care only at tracks from main PV
0019 firstStepGoodPrimaryVertices = cms.EDFilter('PrimaryVertexObjectFilter',
0020      filterParams = cms.PSet(
0021              minNdof = cms.double(25.0),
0022              maxZ = cms.double(15.0),
0023              maxRho = cms.double(2.0)
0024      ),
0025      src=cms.InputTag('firstStepPrimaryVertices')
0026 )
0027
0028 import RecoTracker.TkSeedingLayers.seedingLayersEDProducer_cfi as _mod
0029
0030 # SEEDING LAYERS
0031 jetCoreRegionalStepSeedLayers = _mod.seedingLayersEDProducer.clone(
0032     layerList = ['BPix1+BPix2', 'BPix1+BPix3', 'BPix2+BPix3',
0033                  'BPix1+FPix1_pos', 'BPix1+FPix1_neg',
0034                  'BPix2+FPix1_pos', 'BPix2+FPix1_neg',
0035                  'FPix1_pos+FPix2_pos', 'FPix1_neg+FPix2_neg',
0036                  #'BPix2+TIB1','BPix2+TIB2',
0037                  'BPix3+TIB1','BPix3+TIB2'],
0038     TIB = dict(
0039         matchedRecHits = cms.InputTag('siStripMatchedRecHits','matchedRecHit'),
0040         TTRHBuilder = cms.string('WithTrackAngle'),
0041         clusterChargeCut = cms.PSet(refToPSet_ = cms.string('SiStripClusterChargeCutNone'))
0042     ),
0043     BPix = dict(
0044         useErrorsFromParam = cms.bool(True),
0045         hitErrorRPhi = cms.double(0.0027),
0046         hitErrorRZ = cms.double(0.006),
0047         TTRHBuilder = cms.string('WithTrackAngle'),
0048         HitProducer = cms.string('siPixelRecHits'),
0049         #skipClusters = cms.InputTag('jetCoreRegionalStepClusters')
0050     ),
0051     FPix = dict(
0052         useErrorsFromParam = cms.bool(True),
0053         hitErrorRPhi = cms.double(0.0051),
0054         hitErrorRZ = cms.double(0.0036),
0055         TTRHBuilder = cms.string('WithTrackAngle'),
0056         HitProducer = cms.string('siPixelRecHits'),
0057         #skipClusters = cms.InputTag('jetCoreRegionalStepClusters')
0058     )
0059 )
0060 from Configuration.Eras.Modifier_trackingPhase1_cff import trackingPhase1
0061 _layerListForPhase1 = [
0062         'BPix1+BPix2', 'BPix1+BPix3', 'BPix1+BPix4',
0063         'BPix2+BPix3', 'BPix2+BPix4',
0064         'BPix3+BPix4',
0065         'BPix1+FPix1_pos', 'BPix1+FPix1_neg',
0066         'BPix2+FPix1_pos', 'BPix2+FPix1_neg',
0067         'FPix1_pos+FPix2_pos', 'FPix1_neg+FPix2_neg',
0068         'FPix1_pos+FPix3_pos', 'FPix1_neg+FPix3_neg',
0069         'FPix2_pos+FPix3_pos', 'FPix2_neg+FPix3_neg',
0070         #'BPix3+TIB1','BPix3+TIB2'
0071         'BPix4+TIB1','BPix4+TIB2'
0072     ]
0073 trackingPhase1.toModify(jetCoreRegionalStepSeedLayers, layerList = _layerListForPhase1)
0074
0075 # TrackingRegion
0076 from RecoTauTag.HLTProducers.tauRegionalPixelSeedTrackingRegions_cfi import tauRegionalPixelSeedTrackingRegions as _tauRegionalPixelSeedTrackingRegions
0077 jetCoreRegionalStepTrackingRegions = _tauRegionalPixelSeedTrackingRegions.clone(
0078     RegionPSet=dict(
0079         ptMin          = 10,
0080         deltaPhiRegion = 0.20,
0081         deltaEtaRegion = 0.20,
0082         JetSrc         = 'jetsForCoreTracking',
0083         vertexSrc      = 'firstStepGoodPrimaryVertices',
0084         howToUseMeasurementTracker = 'Never')
0085 )
0086 jetCoreRegionalStepEndcapTrackingRegions = jetCoreRegionalStepTrackingRegions.clone(
0087     RegionPSet=dict(
0088         JetSrc = 'jetsForCoreTrackingEndcap')
0089 )
0090
0091 # Seeding
0092 from RecoTracker.TkHitPairs.hitPairEDProducer_cfi import hitPairEDProducer as _hitPairEDProducer
0093 jetCoreRegionalStepHitDoublets = _hitPairEDProducer.clone(
0094     seedingLayers         = 'jetCoreRegionalStepSeedLayers',
0095     trackingRegions       = 'jetCoreRegionalStepTrackingRegions',
0096     produceSeedingHitSets = True,
0097     maxElementTotal       = 12000000,
0098 )
0099 jetCoreRegionalStepEndcapHitDoublets = jetCoreRegionalStepHitDoublets.clone(
0100     trackingRegions = 'jetCoreRegionalStepEndcapTrackingRegions',
0101 )
0102
0103 from RecoTracker.TkSeedGenerator.seedCreatorFromRegionConsecutiveHitsEDProducer_cff import seedCreatorFromRegionConsecutiveHitsEDProducer as _seedCreatorFromRegionConsecutiveHitsEDProducer
0104 jetCoreRegionalStepSeeds = _seedCreatorFromRegionConsecutiveHitsEDProducer.clone(
0105     seedingHitSets = 'jetCoreRegionalStepHitDoublets',
0106     forceKinematicWithRegionDirection = True
0107 )
0108 import RecoTracker.TkSeedGenerator.deepCoreSeedGenerator_cfi
0109 jetCoreRegionalStepSeedsBarrel = RecoTracker.TkSeedGenerator.deepCoreSeedGenerator_cfi.deepCoreSeedGenerator.clone(#to run MCtruthSeedGenerator clone here from Validation.RecoTrack
0110     vertices = "firstStepPrimaryVertices",
0111     cores    = "jetsForCoreTrackingBarrel"
0112 )
0113
0114 jetCoreRegionalStepSeedsEndcap = jetCoreRegionalStepSeeds.clone(
0115     seedingHitSets = 'jetCoreRegionalStepEndcapHitDoublets',
0116 )
0117
0118 # QUALITY CUTS DURING TRACK BUILDING
0119 import TrackingTools.TrajectoryFiltering.TrajectoryFilter_cff
0120 jetCoreRegionalStepTrajectoryFilter = TrackingTools.TrajectoryFiltering.TrajectoryFilter_cff.CkfBaseTrajectoryFilter_block.clone(
0121     minimumNumberOfHits = 4,
0122     seedPairPenalty     = 0,
0123     minPt               = 0.1
0124 )
0125 jetCoreRegionalStepBarrelTrajectoryFilter = jetCoreRegionalStepTrajectoryFilter.clone(
0126     minimumNumberOfHits = 2,
0127     maxConsecLostHits   = 2,
0128     maxLostHitsFraction = 1.1,
0129     seedPairPenalty     = 0,
0130     minPt               = 0.9 ## should it be slightly decrease ?
0131 )
0132 jetCoreRegionalStepEndcapTrajectoryFilter = jetCoreRegionalStepTrajectoryFilter.clone()
0133
0134
0135 from Configuration.Eras.Modifier_pp_on_XeXe_2017_cff import pp_on_XeXe_2017
0136 from Configuration.ProcessModifiers.pp_on_AA_cff import pp_on_AA
0137 (pp_on_XeXe_2017 | pp_on_AA).toModify(jetCoreRegionalStepTrajectoryFilter, minPt=5.0)
0138
0139 import TrackingTools.KalmanUpdators.Chi2MeasurementEstimator_cfi
0140 jetCoreRegionalStepChi2Est = TrackingTools.KalmanUpdators.Chi2MeasurementEstimator_cfi.Chi2MeasurementEstimator.clone(
0141     ComponentName = 'jetCoreRegionalStepChi2Est',
0142     nSigma        = 3.0,
0143     MaxChi2       = 30.0
0144 )
0145
0146 # TRACK BUILDING
0147 import RecoTracker.CkfPattern.GroupedCkfTrajectoryBuilder_cfi
0148 #need to also load the refToPSet_ used by GroupedCkfTrajectoryBuilder
0149 CkfBaseTrajectoryFilter_block = RecoTracker.CkfPattern.GroupedCkfTrajectoryBuilder_cfi.CkfBaseTrajectoryFilter_block
0150 jetCoreRegionalStepTrajectoryBuilder = RecoTracker.CkfPattern.GroupedCkfTrajectoryBuilder_cfi.GroupedCkfTrajectoryBuilderIterativeDefault.clone(
0151     trajectoryFilter = dict(refToPSet_ = 'jetCoreRegionalStepTrajectoryFilter'),
0152     maxCand = 50,
0153     estimator = 'jetCoreRegionalStepChi2Est',
0154     maxDPhiForLooperReconstruction = 2.0,
0155     maxPtForLooperReconstruction = 0.7,
0156 )
0157 trackingNoLoopers.toModify(jetCoreRegionalStepTrajectoryBuilder,
0158                            maxPtForLooperReconstruction = 0.0)
0159 jetCoreRegionalStepBarrelTrajectoryBuilder = RecoTracker.CkfPattern.GroupedCkfTrajectoryBuilder_cfi.GroupedCkfTrajectoryBuilderIterativeDefault.clone(
0160     trajectoryFilter = dict(refToPSet_ = 'jetCoreRegionalStepBarrelTrajectoryFilter'),
0161     maxCand = 50,
0162     estimator = 'jetCoreRegionalStepChi2Est',
0163     keepOriginalIfRebuildFails = True,
0164     lockHits = False,
0165     requireSeedHitsInRebuild = False,
0166 )
0167 trackingNoLoopers.toModify(jetCoreRegionalStepBarrelTrajectoryBuilder,
0168                            maxPtForLooperReconstruction = cms.double(0.0))
0169 jetCoreRegionalStepEndcapTrajectoryBuilder = jetCoreRegionalStepTrajectoryBuilder.clone(
0170     trajectoryFilter = cms.PSet(refToPSet_ = cms.string('jetCoreRegionalStepEndcapTrajectoryFilter')),
0171     #clustersToSkip = cms.InputTag('jetCoreRegionalStepClusters'),
0172 )
0173 trackingNoLoopers.toModify(jetCoreRegionalStepEndcapTrajectoryBuilder,
0174                            maxPtForLooperReconstruction = cms.double(0.0))
0175 #customized cleaner for DeepCore
0176 from TrackingTools.TrajectoryCleaning.TrajectoryCleanerBySharedHits_cfi import trajectoryCleanerBySharedHits
0177 jetCoreRegionalStepDeepCoreTrajectoryCleaner = trajectoryCleanerBySharedHits.clone(
0178     ComponentName = 'jetCoreRegionalStepDeepCoreTrajectoryCleaner',
0179     fractionShared = 0.45
0180 )
0181
0182 ############## to run MCtruthSeedGenerator ####################
0183 #import RecoTracker.TkSeedGenerator.deepCoreSeedGenerator_cfi
0184 #import Validation.RecoTrack.JetCoreMCtruthSeedGenerator_cfi
0185 #seedingDeepCore.toReplaceWith(jetCoreRegionalStepSeedsBarrel,
0186 #    RecoTracker.TkSeedGenerator.deepCoreSeedGenerator_cfi.deepCoreSeedGenerator.clone(#to run MCtruthSeedGenerator clone here from Validation.RecoTrack
0187 #       vertices="firstStepPrimaryVertices"
0188 #    )
0189 #)
0190
0191
0192 # MAKING OF TRACK CANDIDATES
0193 import RecoTracker.CkfPattern.CkfTrackCandidates_cfi
0194 jetCoreRegionalStepTrackCandidates = RecoTracker.CkfPattern.CkfTrackCandidates_cfi.ckfTrackCandidatesIterativeDefault.clone(
0195     src                    = 'jetCoreRegionalStepSeeds',
0196     maxSeedsBeforeCleaning = 10000,
0197     TrajectoryBuilderPSet  = dict(refToPSet_ = 'jetCoreRegionalStepTrajectoryBuilder'),
0198     NavigationSchool       = 'SimpleNavigationSchool',
0199     ### these two parameters are relevant only for the CachingSeedCleanerBySharedInput
0200     #numHitsForSeedCleaner = 50,
0201     #onlyPixelHitsForSeedCleaner = True,
0202 )
0203 jetCoreRegionalStepBarrelTrackCandidates = jetCoreRegionalStepTrackCandidates.clone(
0204     src                    = 'jetCoreRegionalStepSeedsBarrel',
0205     TrajectoryBuilderPSet  = cms.PSet( refToPSet_ = cms.string('jetCoreRegionalStepBarrelTrajectoryBuilder')),
0206     ### these two parameters are relevant only for the CachingSeedCleanerBySharedInput
0207     #numHitsForSeedCleaner = cms.int32(50),
0208     #onlyPixelHitsForSeedCleaner = cms.bool(True),
0209     TrajectoryCleaner         = 'jetCoreRegionalStepDeepCoreTrajectoryCleaner',
0210     doSeedingRegionRebuilding = True,
0211 )
0212 jetCoreRegionalStepEndcapTrackCandidates = jetCoreRegionalStepTrackCandidates.clone(
0213     src                    = 'jetCoreRegionalStepSeedsEndcap',
0214     TrajectoryBuilderPSet  = cms.PSet( refToPSet_ = cms.string('jetCoreRegionalStepEndcapTrajectoryBuilder')),
0215     ### these two parameters are relevant only for the CachingSeedCleanerBySharedInput
0216     #numHitsForSeedCleaner = cms.int32(50),
0217     #onlyPixelHitsForSeedCleaner = cms.bool(True),
0218 )
0219
0220 # TRACK FITTING
0221 import RecoTracker.TrackProducer.TrackProducerIterativeDefault_cfi
0222 jetCoreRegionalStepTracks = RecoTracker.TrackProducer.TrackProducerIterativeDefault_cfi.TrackProducerIterativeDefault.clone(
0223     AlgorithmName = 'jetCoreRegionalStep',
0224     src           = 'jetCoreRegionalStepTrackCandidates',
0225     Fitter        = 'FlexibleKFFittingSmoother'
0226 )
0227 jetCoreRegionalStepBarrelTracks = jetCoreRegionalStepTracks.clone(
0228     src           = 'jetCoreRegionalStepBarrelTrackCandidates',
0229 )
0230 jetCoreRegionalStepEndcapTracks = jetCoreRegionalStepTracks.clone(
0231     src           = 'jetCoreRegionalStepEndcapTrackCandidates',
0232 )
0233
0234 from Configuration.Eras.Modifier_fastSim_cff import fastSim
0235 import RecoTracker.FinalTrackSelectors.trackListMerger_cfi
0236 _fastSim_jetCoreRegionalStepTracks = RecoTracker.FinalTrackSelectors.trackListMerger_cfi.trackListMerger.clone(
0237     TrackProducers     = [],
0238     hasSelector        = [],
0239     selectedTrackQuals = [],
0240     copyExtras         = True
0241 )
0242 fastSim.toReplaceWith(jetCoreRegionalStepTracks,_fastSim_jetCoreRegionalStepTracks)
0243
0244
0245 # Final selection
0246 from RecoTracker.FinalTrackSelectors.TrackCutClassifier_cff import *
0247 jetCoreRegionalStep = TrackCutClassifier.clone(
0248     src = 'jetCoreRegionalStepTracks',
0249     mva = dict(
0250     minPixelHits  = [1,1,1],
0251         maxChi2       = [9999.,9999.,9999.],
0252         maxChi2n      = [1.6,1.0,0.7],
0253         minLayers     = [3,5,5],
0254         min3DLayers   = [1,2,3],
0255         maxLostLayers = [4,3,2],
0256         maxDz         = [0.5,0.35,0.2],
0257         maxDr         = [0.3,0.2,0.1]
0258     ),
0259     vertices = 'firstStepGoodPrimaryVertices'
0260 )
0261 jetCoreRegionalStepBarrel = jetCoreRegionalStep.clone(
0262     src = 'jetCoreRegionalStepBarrelTracks',
0263     mva = dict(
0264 #   minPixelHits  = [1,1,1], # they could be easily increased to at least 2 or 3 !
0265         min3DLayers   = [1,2,2],
0266     ),
0267 )
0268
0269 from RecoTracker.FinalTrackSelectors.TrackMVAClassifierPrompt_cfi import *
0270 trackingPhase1.toReplaceWith(jetCoreRegionalStep, TrackMVAClassifierPrompt.clone(
0271      mva = dict(GBRForestLabel = 'MVASelectorJetCoreRegionalStep_Phase1'),
0272      src = 'jetCoreRegionalStepTracks',
0273      qualityCuts = [-0.2,0.0,0.4]
0274 ))
0275
0276 trackingPhase1.toReplaceWith(jetCoreRegionalStepBarrel, jetCoreRegionalStep.clone(
0277      src = 'jetCoreRegionalStepBarrelTracks',
0278 ))
0279
0280 pp_on_AA.toModify(jetCoreRegionalStep, qualityCuts = [-0.2, 0.0, 0.8])
0281 pp_on_AA.toModify(jetCoreRegionalStepBarrel, qualityCuts = [-0.2, 0.0, 0.8])
0282
0283 from RecoTracker.FinalTrackSelectors.trackTfClassifier_cfi import *
0284 from RecoTracker.FinalTrackSelectors.trackSelectionTf_cfi import *
0285 from RecoTracker.FinalTrackSelectors.trackSelectionTf_CKF_cfi import *
0286 trackdnn.toReplaceWith(jetCoreRegionalStep, trackTfClassifier.clone(
0287      src = 'jetCoreRegionalStepTracks',
0288      qualityCuts = qualityCutDictionary.JetCoreRegionalStep.value()
0289 ))
0290 trackdnn.toReplaceWith(jetCoreRegionalStepBarrel, trackTfClassifier.clone(
0291      src = 'jetCoreRegionalStepBarrelTracks',
0292      qualityCuts = qualityCutDictionary.JetCoreRegionalStep.value()
0293 ))
0294
0295 fastSim.toModify(jetCoreRegionalStep,vertices = 'firstStepPrimaryVerticesBeforeMixing')
0296
0297
0298 jetCoreRegionalStepEndcap = jetCoreRegionalStep.clone(
0299     src = 'jetCoreRegionalStepEndcapTracks',
0300 )
0301
0302 # Final sequence
0303 JetCoreRegionalStepTask = cms.Task(jetsForCoreTracking,
0304                                    firstStepGoodPrimaryVertices,
0305                                    #jetCoreRegionalStepClusters,
0306                                    jetCoreRegionalStepSeedLayers,
0307                                    jetCoreRegionalStepTrackingRegions,
0308                                    jetCoreRegionalStepHitDoublets,
0309                                    jetCoreRegionalStepSeeds,
0310                                    jetCoreRegionalStepTrackCandidates,
0311                                    jetCoreRegionalStepTracks,
0312                                    #jetCoreRegionalStepClassifier1,jetCoreRegionalStepClassifier2,
0313                                    jetCoreRegionalStep)
0314 JetCoreRegionalStep = cms.Sequence(JetCoreRegionalStepTask)
0315
0316 JetCoreRegionalStepBarrelTask = cms.Task(jetsForCoreTrackingBarrel,
0317                                          firstStepGoodPrimaryVertices,
0318                                          #jetCoreRegionalStepClusters,
0319                                          jetCoreRegionalStepSeedLayers,
0320                                          jetCoreRegionalStepSeedsBarrel,
0321                                          jetCoreRegionalStepBarrelTrackCandidates,
0322                                          jetCoreRegionalStepBarrelTracks,
0323                                          jetCoreRegionalStepBarrel)
0324
0325 JetCoreRegionalStepEndcapTask = cms.Task(jetsForCoreTrackingEndcap,
0326                                          firstStepGoodPrimaryVertices,
0327                                          #jetCoreRegionalStepClusters,
0328                                          jetCoreRegionalStepSeedLayers,
0329                                          jetCoreRegionalStepEndcapTrackingRegions,
0330                                          jetCoreRegionalStepEndcapHitDoublets,
0331                                          jetCoreRegionalStepSeedsEndcap,
0332                                          jetCoreRegionalStepEndcapTrackCandidates,
0333                                          jetCoreRegionalStepEndcapTracks,
0334                                          jetCoreRegionalStepEndcap)
0335
0336
0337 from Configuration.ProcessModifiers.seedingDeepCore_cff import seedingDeepCore
0338
0339 from RecoTracker.FinalTrackSelectors.TrackCollectionMerger_cfi import *
0340 seedingDeepCore.toReplaceWith(jetCoreRegionalStepTracks, TrackCollectionMerger.clone(
0341     trackProducers   = ["jetCoreRegionalStepBarrelTracks",
0342                         "jetCoreRegionalStepEndcapTracks",],
0343     inputClassifiers = ["jetCoreRegionalStepBarrel",
0344                         "jetCoreRegionalStepEndcap",],
0345     foundHitBonus    = 100.0,
0346     lostHitPenalty   = 1.0
0347 ))
0348
0349 seedingDeepCore.toReplaceWith(jetCoreRegionalStep, jetCoreRegionalStepTracks.clone()) #(*)
0350
0351 seedingDeepCore.toReplaceWith(JetCoreRegionalStepTask, cms.Task(
0352     JetCoreRegionalStepBarrelTask,
0353     JetCoreRegionalStepEndcapTask,
0354     cms.Task(jetCoreRegionalStepTracks,jetCoreRegionalStep)
0355 ))
0356
0357 fastSim.toReplaceWith(JetCoreRegionalStepTask,
0358                       cms.Task(jetCoreRegionalStepTracks,
0359                                    jetCoreRegionalStep))