HiMuonAlgos/python/HiRegitMuonSeededStep_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002 from RecoTracker.IterativeTracking.MuonSeededStep_cff import *
0003
0004 ###### Muon reconstruction module #####
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)),  # should this be False?
0034                              ),
0035     copyExtras = True,
0036     makeReKeyedSeeds = cms.untracked.bool(False)
0037 )
0038
0039 hiEarlyMuons = earlyMuons.clone(
0040     inputCollectionLabels = ["hiEarlyGeneralTracks", "standAloneMuons:UpdatedAtVtx"]
0041 )
0042
0043 ###### SEEDER MODELS ######
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      ) #end of vpset
0094 ) #end of clone
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     ) #end of vpset
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      ) #end of vpset
0146 ) #end of clone
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      ) #end of vpset
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)