File indexing completed on 2021-03-05 03:57:51
0001 import FWCore.ParameterSet.Config as cms
0002 from RecoTracker.IterativeTracking.MuonSeededStep_cff import *
0003
0004
0005 from RecoTracker.FinalTrackSelectors.trackAlgoPriorityOrder_cfi import trackAlgoPriorityOrder
0006 import RecoTracker.FinalTrackSelectors.trackListMerger_cfi
0007 hiEarlyGeneralTracks = RecoTracker.FinalTrackSelectors.trackListMerger_cfi.trackListMerger.clone(
0008 TrackProducers = [
0009 'hiGlobalPrimTracks',
0010 'hiDetachedTripletStepTracks',
0011 'hiLowPtTripletStepTracks',
0012 'hiPixelPairGlobalPrimTracks',
0013 'hiJetCoreRegionalStepTracks',
0014 'hiRegitMuInitialStepTracks',
0015 'hiRegitMuPixelPairStepTracks',
0016 'hiRegitMuMixedTripletStepTracks',
0017 'hiRegitMuPixelLessStepTracks',
0018 'hiRegitMuDetachedTripletStepTracks'
0019 ],
0020 hasSelector = [1,1,1,1,1,1,1,1,1,1],
0021 selectedTrackQuals = [
0022 "hiInitialStepSelector:hiInitialStep",
0023 "hiDetachedTripletStepSelector:hiDetachedTripletStep",
0024 "hiLowPtTripletStepSelector:hiLowPtTripletStep",
0025 "hiPixelPairStepSelector:hiPixelPairStep",
0026 "hiJetCoreRegionalStepSelector:hiJetCoreRegionalStep",
0027 "hiRegitMuInitialStepSelector:hiRegitMuInitialStepLoose",
0028 "hiRegitMuPixelPairStepSelector:hiRegitMuPixelPairStep",
0029 "hiRegitMuMixedTripletStepSelector:hiRegitMuMixedTripletStep",
0030 "hiRegitMuPixelLessStepSelector:hiRegitMuPixelLessStep",
0031 "hiRegitMuDetachedTripletStepSelector:hiRegitMuDetachedTripletStep"
0032 ],
0033 setsToMerge = cms.VPSet( cms.PSet( tLists=cms.vint32(0,1,2,3,4,5,6,7,8,9), pQual=cms.bool(True)),
0034 ),
0035 copyExtras = True,
0036 makeReKeyedSeeds = cms.untracked.bool(False)
0037 )
0038
0039 hiEarlyMuons = earlyMuons.clone(
0040 inputCollectionLabels = ["hiEarlyGeneralTracks", "standAloneMuons:UpdatedAtVtx"]
0041 )
0042
0043
0044 import RecoTracker.SpecialSeedGenerators.outInSeedsFromStandaloneMuons_cfi
0045 import RecoTracker.SpecialSeedGenerators.inOutSeedsFromTrackerMuons_cfi
0046 hiRegitMuonSeededSeedsOutIn = RecoTracker.SpecialSeedGenerators.outInSeedsFromStandaloneMuons_cfi.outInSeedsFromStandaloneMuons.clone(
0047 src = "hiEarlyMuons",
0048 )
0049 hiRegitMuonSeededSeedsInOut = RecoTracker.SpecialSeedGenerators.inOutSeedsFromTrackerMuons_cfi.inOutSeedsFromTrackerMuons.clone(
0050 src = "hiEarlyMuons",
0051 )
0052 hiRegitMuonSeededTrackCandidatesInOut = muonSeededTrackCandidatesInOut.clone(
0053 src = "hiRegitMuonSeededSeedsInOut"
0054 )
0055 hiRegitMuonSeededTrackCandidatesOutIn = muonSeededTrackCandidatesOutIn.clone(
0056 src = "hiRegitMuonSeededSeedsOutIn"
0057 )
0058 hiRegitMuonSeededTracksOutIn = muonSeededTracksOutIn.clone(
0059 src = "hiRegitMuonSeededTrackCandidatesOutIn",
0060 AlgorithmName = 'hiRegitMuMuonSeededStepOutIn'
0061 )
0062 hiRegitMuonSeededTracksInOut = muonSeededTracksInOut.clone(
0063 src = "hiRegitMuonSeededTrackCandidatesInOut",
0064 AlgorithmName = 'hiRegitMuMuonSeededStepInOut'
0065 )
0066 import RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi
0067 import RecoHI.HiTracking.hiMultiTrackSelector_cfi
0068 hiRegitMuonSeededTracksInOutSelector = RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiMultiTrackSelector.clone(
0069 src = 'hiRegitMuonSeededTracksInOut',
0070 vertices = "hiSelectedPixelVertex",
0071 useAnyMVA = True,
0072 GBRForestLabel = 'HIMVASelectorIter7',
0073 GBRForestVars = ['chi2perdofperlayer', 'nhits', 'nlayers', 'eta'],
0074 trackSelectors = cms.VPSet(
0075 RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.looseMTS.clone(
0076 name = 'hiRegitMuonSeededTracksInOutLoose',
0077 min_nhits = 8
0078 ),
0079 RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiTightMTS.clone(
0080 name = 'hiRegitMuonSeededTracksInOutTight',
0081 preFilterName = 'hiRegitMuonSeededTracksInOutLoose',
0082 min_nhits = 8,
0083 useMVA = True,
0084 minMVA = -0.2
0085 ),
0086 RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiHighpurityMTS.clone(
0087 name = 'hiRegitMuonSeededTracksInOutHighPurity',
0088 preFilterName = 'hiRegitMuonSeededTracksInOutTight',
0089 min_nhits = 8,
0090 useMVA = True,
0091 minMVA = -0.09
0092 ),
0093 )
0094 )
0095 from Configuration.Eras.Modifier_trackingPhase1_cff import trackingPhase1
0096 trackingPhase1.toModify(hiRegitMuonSeededTracksInOutSelector,
0097 useAnyMVA = False,
0098 trackSelectors= cms.VPSet(
0099 RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.looseMTS.clone(
0100 name = 'hiRegitMuonSeededTracksInOutLoose',
0101 min_nhits = 8
0102 ),
0103 RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiTightMTS.clone(
0104 name = 'hiRegitMuonSeededTracksInOutTight',
0105 preFilterName = 'hiRegitMuonSeededTracksInOutLoose',
0106 min_nhits = 8,
0107 useMVA = False,
0108 minMVA = -0.2
0109 ),
0110 RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiHighpurityMTS.clone(
0111 name = 'hiRegitMuonSeededTracksInOutHighPurity',
0112 preFilterName = 'hiRegitMuonSeededTracksInOutTight',
0113 min_nhits = 8,
0114 useMVA = False,
0115 minMVA = -0.09
0116 ),
0117 )
0118 )
0119
0120 hiRegitMuonSeededTracksOutInSelector = RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiMultiTrackSelector.clone(
0121 src = 'hiRegitMuonSeededTracksOutIn',
0122 vertices = "hiSelectedPixelVertex",
0123 useAnyMVA = True,
0124 GBRForestLabel = 'HIMVASelectorIter7',
0125 GBRForestVars = ['chi2perdofperlayer', 'nhits', 'nlayers', 'eta'],
0126 trackSelectors = cms.VPSet(
0127 RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.looseMTS.clone(
0128 name = 'hiRegitMuonSeededTracksOutInLoose',
0129 min_nhits = 8
0130 ),
0131 RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiTightMTS.clone(
0132 name = 'hiRegitMuonSeededTracksOutInTight',
0133 preFilterName = 'hiRegitMuonSeededTracksOutInLoose',
0134 min_nhits = 8,
0135 useMVA = True,
0136 minMVA = -0.2
0137 ),
0138 RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiHighpurityMTS.clone(
0139 name = 'hiRegitMuonSeededTracksOutInHighPurity',
0140 preFilterName = 'hiRegitMuonSeededTracksOutInTight',
0141 min_nhits = 8,
0142 useMVA = True,
0143 minMVA = -0.09
0144 ),
0145 )
0146 )
0147 from Configuration.Eras.Modifier_trackingPhase1_cff import trackingPhase1
0148 trackingPhase1.toModify(hiRegitMuonSeededTracksOutInSelector,
0149 useAnyMVA = False,
0150 trackSelectors= cms.VPSet(
0151 RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.looseMTS.clone(
0152 name = 'hiRegitMuonSeededTracksOutInLoose',
0153 min_nhits = 8
0154 ),
0155 RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiTightMTS.clone(
0156 name = 'hiRegitMuonSeededTracksOutInTight',
0157 preFilterName = 'hiRegitMuonSeededTracksOutInLoose',
0158 min_nhits = 8,
0159 useMVA = False,
0160 minMVA = -0.2
0161 ),
0162 RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiHighpurityMTS.clone(
0163 name = 'hiRegitMuonSeededTracksOutInHighPurity',
0164 preFilterName = 'hiRegitMuonSeededTracksOutInTight',
0165 min_nhits = 8,
0166 useMVA = False,
0167 minMVA = -0.09
0168 ),
0169 )
0170 )
0171
0172 hiRegitMuonSeededStepCoreTask = cms.Task(
0173 hiRegitMuonSeededSeedsInOut , hiRegitMuonSeededTrackCandidatesInOut , hiRegitMuonSeededTracksInOut ,
0174 hiRegitMuonSeededSeedsOutIn , hiRegitMuonSeededTrackCandidatesOutIn , hiRegitMuonSeededTracksOutIn
0175 )
0176 hiRegitMuonSeededStepExtraTask = cms.Task(
0177 hiRegitMuonSeededTracksInOutSelector ,
0178 hiRegitMuonSeededTracksOutInSelector
0179 )
0180 hiRegitMuonSeededStepTask = cms.Task(
0181 hiEarlyGeneralTracks ,
0182 hiEarlyMuons ,
0183 hiRegitMuonSeededStepCoreTask ,
0184 hiRegitMuonSeededStepExtraTask
0185 )
0186 hiRegitMuonSeededStep = cms.Sequence(hiRegitMuonSeededStepTask)