CMS 3D CMS Logo

All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Modules Pages
HiRegitMuonSeededStep_cff.py
Go to the documentation of this file.
1 import FWCore.ParameterSet.Config as cms
3 
4 
5 from RecoTracker.FinalTrackSelectors.trackAlgoPriorityOrder_cfi import trackAlgoPriorityOrder
7 hiEarlyGeneralTracks = RecoTracker.FinalTrackSelectors.trackListMerger_cfi.trackListMerger.clone(
8  TrackProducers = [
9  'hiGlobalPrimTracks',
10  'hiDetachedTripletStepTracks',
11  'hiLowPtTripletStepTracks',
12  'hiPixelPairGlobalPrimTracks',
13  'hiJetCoreRegionalStepTracks',
14  'hiRegitMuInitialStepTracks',
15  'hiRegitMuPixelPairStepTracks',
16  'hiRegitMuMixedTripletStepTracks',
17  'hiRegitMuPixelLessStepTracks',
18  'hiRegitMuDetachedTripletStepTracks'
19  ],
20  hasSelector = [1,1,1,1,1,1,1,1,1,1],
21  selectedTrackQuals = [
22  "hiInitialStepSelector:hiInitialStep",
23  "hiDetachedTripletStepSelector:hiDetachedTripletStep",
24  "hiLowPtTripletStepSelector:hiLowPtTripletStep",
25  "hiPixelPairStepSelector:hiPixelPairStep",
26  "hiJetCoreRegionalStepSelector:hiJetCoreRegionalStep",
27  "hiRegitMuInitialStepSelector:hiRegitMuInitialStepLoose",
28  "hiRegitMuPixelPairStepSelector:hiRegitMuPixelPairStep",
29  "hiRegitMuMixedTripletStepSelector:hiRegitMuMixedTripletStep",
30  "hiRegitMuPixelLessStepSelector:hiRegitMuPixelLessStep",
31  "hiRegitMuDetachedTripletStepSelector:hiRegitMuDetachedTripletStep"
32  ],
33  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?
34  ),
35  copyExtras = True,
36  makeReKeyedSeeds = cms.untracked.bool(False)
37 )
38 
39 hiEarlyMuons = earlyMuons.clone(
40  inputCollectionLabels = ["hiEarlyGeneralTracks", "standAloneMuons:UpdatedAtVtx"]
41 )
42 
43 
44 import RecoTracker.SpecialSeedGenerators.outInSeedsFromStandaloneMuons_cfi
45 import RecoTracker.SpecialSeedGenerators.inOutSeedsFromTrackerMuons_cfi
46 hiRegitMuonSeededSeedsOutIn = RecoTracker.SpecialSeedGenerators.outInSeedsFromStandaloneMuons_cfi.outInSeedsFromStandaloneMuons.clone(
47  src = "hiEarlyMuons",
48 )
49 hiRegitMuonSeededSeedsInOut = RecoTracker.SpecialSeedGenerators.inOutSeedsFromTrackerMuons_cfi.inOutSeedsFromTrackerMuons.clone(
50  src = "hiEarlyMuons",
51 )
52 hiRegitMuonSeededTrackCandidatesInOut = muonSeededTrackCandidatesInOut.clone(
53  src = "hiRegitMuonSeededSeedsInOut"
54 )
55 hiRegitMuonSeededTrackCandidatesOutIn = muonSeededTrackCandidatesOutIn.clone(
56  src = "hiRegitMuonSeededSeedsOutIn"
57 )
58 hiRegitMuonSeededTracksOutIn = muonSeededTracksOutIn.clone(
59  src = "hiRegitMuonSeededTrackCandidatesOutIn",
60  AlgorithmName = 'hiRegitMuMuonSeededStepOutIn'
61 )
62 hiRegitMuonSeededTracksInOut = muonSeededTracksInOut.clone(
63  src = "hiRegitMuonSeededTrackCandidatesInOut",
64  AlgorithmName = 'hiRegitMuMuonSeededStepInOut'
65 )
66 import RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi
68 hiRegitMuonSeededTracksInOutSelector = RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiMultiTrackSelector.clone(
69  src = 'hiRegitMuonSeededTracksInOut',
70  vertices = "hiSelectedPixelVertex",
71  useAnyMVA = True,
72  GBRForestLabel = 'HIMVASelectorIter7',
73  GBRForestVars = ['chi2perdofperlayer', 'nhits', 'nlayers', 'eta'],
74  trackSelectors = cms.VPSet(
75  RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.looseMTS.clone(
76  name = 'hiRegitMuonSeededTracksInOutLoose',
77  min_nhits = 8
78  ),
79  RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiTightMTS.clone(
80  name = 'hiRegitMuonSeededTracksInOutTight',
81  preFilterName = 'hiRegitMuonSeededTracksInOutLoose',
82  min_nhits = 8,
83  useMVA = True,
84  minMVA = -0.2
85  ),
86  RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiHighpurityMTS.clone(
87  name = 'hiRegitMuonSeededTracksInOutHighPurity',
88  preFilterName = 'hiRegitMuonSeededTracksInOutTight',
89  min_nhits = 8,
90  useMVA = True,
91  minMVA = -0.09
92  ),
93  ) #end of vpset
94 ) #end of clone
95 from Configuration.Eras.Modifier_trackingPhase1_cff import trackingPhase1
96 trackingPhase1.toModify(hiRegitMuonSeededTracksInOutSelector,
97  useAnyMVA = False,
98  trackSelectors= cms.VPSet(
99  RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.looseMTS.clone(
100  name = 'hiRegitMuonSeededTracksInOutLoose',
101  min_nhits = 8
102  ),
103  RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiTightMTS.clone(
104  name = 'hiRegitMuonSeededTracksInOutTight',
105  preFilterName = 'hiRegitMuonSeededTracksInOutLoose',
106  min_nhits = 8,
107  useMVA = False,
108  minMVA = -0.2
109  ),
110  RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiHighpurityMTS.clone(
111  name = 'hiRegitMuonSeededTracksInOutHighPurity',
112  preFilterName = 'hiRegitMuonSeededTracksInOutTight',
113  min_nhits = 8,
114  useMVA = False,
115  minMVA = -0.09
116  ),
117  ) #end of vpset
118 )
119 
120 hiRegitMuonSeededTracksOutInSelector = RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiMultiTrackSelector.clone(
121  src = 'hiRegitMuonSeededTracksOutIn',
122  vertices = "hiSelectedPixelVertex",
123  useAnyMVA = True,
124  GBRForestLabel = 'HIMVASelectorIter7',
125  GBRForestVars = ['chi2perdofperlayer', 'nhits', 'nlayers', 'eta'],
126  trackSelectors = cms.VPSet(
127  RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.looseMTS.clone(
128  name = 'hiRegitMuonSeededTracksOutInLoose',
129  min_nhits = 8
130  ),
131  RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiTightMTS.clone(
132  name = 'hiRegitMuonSeededTracksOutInTight',
133  preFilterName = 'hiRegitMuonSeededTracksOutInLoose',
134  min_nhits = 8,
135  useMVA = True,
136  minMVA = -0.2
137  ),
138  RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiHighpurityMTS.clone(
139  name = 'hiRegitMuonSeededTracksOutInHighPurity',
140  preFilterName = 'hiRegitMuonSeededTracksOutInTight',
141  min_nhits = 8,
142  useMVA = True,
143  minMVA = -0.09
144  ),
145  ) #end of vpset
146 ) #end of clone
147 from Configuration.Eras.Modifier_trackingPhase1_cff import trackingPhase1
148 trackingPhase1.toModify(hiRegitMuonSeededTracksOutInSelector,
149  useAnyMVA = False,
150  trackSelectors= cms.VPSet(
151  RecoTracker.FinalTrackSelectors.multiTrackSelector_cfi.looseMTS.clone(
152  name = 'hiRegitMuonSeededTracksOutInLoose',
153  min_nhits = 8
154  ),
155  RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiTightMTS.clone(
156  name = 'hiRegitMuonSeededTracksOutInTight',
157  preFilterName = 'hiRegitMuonSeededTracksOutInLoose',
158  min_nhits = 8,
159  useMVA = False,
160  minMVA = -0.2
161  ),
162  RecoHI.HiTracking.hiMultiTrackSelector_cfi.hiHighpurityMTS.clone(
163  name = 'hiRegitMuonSeededTracksOutInHighPurity',
164  preFilterName = 'hiRegitMuonSeededTracksOutInTight',
165  min_nhits = 8,
166  useMVA = False,
167  minMVA = -0.09
168  ),
169  ) #end of vpset
170 )
171 
172 hiRegitMuonSeededStepCoreTask = cms.Task(
173  hiRegitMuonSeededSeedsInOut , hiRegitMuonSeededTrackCandidatesInOut , hiRegitMuonSeededTracksInOut ,
174  hiRegitMuonSeededSeedsOutIn , hiRegitMuonSeededTrackCandidatesOutIn , hiRegitMuonSeededTracksOutIn
175  )
176 hiRegitMuonSeededStepExtraTask = cms.Task(
177  hiRegitMuonSeededTracksInOutSelector ,
178  hiRegitMuonSeededTracksOutInSelector
179  )
180 hiRegitMuonSeededStepTask = cms.Task(
181  hiEarlyGeneralTracks ,
182  hiEarlyMuons ,
183  hiRegitMuonSeededStepCoreTask ,
184  hiRegitMuonSeededStepExtraTask
185  )
186 hiRegitMuonSeededStep = cms.Sequence(hiRegitMuonSeededStepTask)
hiMultiTrackSelector_cfi
trackListMerger_cfi
MuonSeededStep_cff