CMS 3D CMS Logo
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Groups Pages
PFTrackAlgoTools.cc
Go to the documentation of this file.
1 #include "FWCore/MessageLogger/interface/MessageLogger.h"
2 #include "RecoParticleFlow/PFTracking/interface/PFTrackAlgoTools.h"
3 
4 namespace PFTrackAlgoTools {
5 
6  double dPtCut(const reco::TrackBase::TrackAlgorithm& algo,
7  const std::vector<double>& cuts,
8  bool hltIterativeTracking = true) {
9  switch (algo) {
10  case reco::TrackBase::ctf:
11  case reco::TrackBase::duplicateMerge:
12  case reco::TrackBase::initialStep:
13  case reco::TrackBase::highPtTripletStep:
14  case reco::TrackBase::lowPtQuadStep:
15  case reco::TrackBase::lowPtTripletStep:
16  case reco::TrackBase::pixelPairStep:
17  case reco::TrackBase::jetCoreRegionalStep:
18  return cuts[0];
19  case reco::TrackBase::detachedQuadStep:
20  case reco::TrackBase::detachedTripletStep:
21  return cuts[1];
22  case reco::TrackBase::mixedTripletStep:
23  return cuts[2];
24  case reco::TrackBase::pixelLessStep:
25  return cuts[3];
26  case reco::TrackBase::tobTecStep:
27  return cuts[4];
28  case reco::TrackBase::muonSeededStepInOut:
29  case reco::TrackBase::muonSeededStepOutIn:
30  return cuts.at(5);
31  case reco::TrackBase::hltIter0:
32  case reco::TrackBase::hltIter1:
33  case reco::TrackBase::hltIter2:
34  return cuts[0];
35  case reco::TrackBase::hltIter3:
36  return hltIterativeTracking ? cuts[1] : cuts[0];
37  case reco::TrackBase::hltIter4:
38  return hltIterativeTracking ? cuts[2] : cuts[0];
39  case reco::TrackBase::hltIterX:
40  return cuts[0];
41  default:
42  return hltIterativeTracking ? cuts.at(5) : cuts[0];
43  }
44  }
45 
46  unsigned int nHitCut(const reco::TrackBase::TrackAlgorithm& algo,
47  const std::vector<unsigned int>& cuts,
48  bool hltIterativeTracking = true) {
49  switch (algo) {
50  case reco::TrackBase::ctf:
51  case reco::TrackBase::duplicateMerge:
52  case reco::TrackBase::initialStep:
53  case reco::TrackBase::highPtTripletStep:
54  case reco::TrackBase::lowPtQuadStep:
55  case reco::TrackBase::lowPtTripletStep:
56  case reco::TrackBase::pixelPairStep:
57  case reco::TrackBase::jetCoreRegionalStep:
58  return cuts[0];
59  case reco::TrackBase::detachedQuadStep:
60  case reco::TrackBase::detachedTripletStep:
61  return cuts[1];
62  case reco::TrackBase::mixedTripletStep:
63  return cuts[2];
64  case reco::TrackBase::pixelLessStep:
65  return cuts[3];
66  case reco::TrackBase::tobTecStep:
67  return cuts[4];
68  case reco::TrackBase::muonSeededStepInOut:
69  case reco::TrackBase::muonSeededStepOutIn:
70  return cuts.at(5);
71  case reco::TrackBase::hltIter0:
72  case reco::TrackBase::hltIter1:
73  case reco::TrackBase::hltIter2:
74  return cuts[0];
75  case reco::TrackBase::hltIter3:
76  return hltIterativeTracking ? cuts[1] : cuts[0];
77  case reco::TrackBase::hltIter4:
78  return hltIterativeTracking ? cuts[2] : cuts[0];
79  case reco::TrackBase::hltIterX:
80  return cuts[0];
81  default:
82  return hltIterativeTracking ? cuts.at(5) : cuts[0];
83  }
84  }
85 
86  double errorScale(const reco::TrackBase::TrackAlgorithm& algo, const std::vector<double>& errorScale) {
87  switch (algo) {
88  case reco::TrackBase::ctf:
89  case reco::TrackBase::duplicateMerge:
90  case reco::TrackBase::initialStep:
91  case reco::TrackBase::highPtTripletStep:
92  case reco::TrackBase::lowPtQuadStep:
93  case reco::TrackBase::lowPtTripletStep:
94  case reco::TrackBase::pixelPairStep:
95  case reco::TrackBase::jetCoreRegionalStep:
96  case reco::TrackBase::muonSeededStepInOut:
97  case reco::TrackBase::muonSeededStepOutIn:
98  case reco::TrackBase::detachedQuadStep:
99  case reco::TrackBase::detachedTripletStep:
100  case reco::TrackBase::mixedTripletStep:
101  case reco::TrackBase::hltIter0:
102  case reco::TrackBase::hltIter1:
103  case reco::TrackBase::hltIter2:
104  case reco::TrackBase::hltIter3:
105  case reco::TrackBase::hltIter4:
106  case reco::TrackBase::hltIterX:
107  return 1.0;
108  case reco::TrackBase::pixelLessStep:
109  return errorScale[0];
110  case reco::TrackBase::tobTecStep:
111  return errorScale[1];
112  default:
113  return 1E9;
114  }
115  }
116 
117  bool isGoodForEGM(const reco::TrackBase::TrackAlgorithm& algo) {
118  switch (algo) {
119  case reco::TrackBase::ctf:
120  case reco::TrackBase::duplicateMerge:
121  case reco::TrackBase::initialStep:
122  case reco::TrackBase::highPtTripletStep:
123  case reco::TrackBase::lowPtQuadStep:
124  case reco::TrackBase::lowPtTripletStep:
125  case reco::TrackBase::pixelPairStep:
126  case reco::TrackBase::jetCoreRegionalStep:
127  case reco::TrackBase::detachedQuadStep:
128  case reco::TrackBase::detachedTripletStep:
129  case reco::TrackBase::mixedTripletStep:
130  case reco::TrackBase::muonSeededStepInOut:
131  case reco::TrackBase::muonSeededStepOutIn:
132  case reco::TrackBase::hltIter0:
133  case reco::TrackBase::hltIter1:
134  case reco::TrackBase::hltIter2:
135  case reco::TrackBase::hltIter3:
136  case reco::TrackBase::hltIter4:
137  case reco::TrackBase::hltIterX:
138  return true;
139  default:
140  return false;
141  }
142  }
143 
144  bool isGoodForEGMPrimary(const reco::TrackBase::TrackAlgorithm& algo) {
145  switch (algo) {
146  case reco::TrackBase::ctf:
147  case reco::TrackBase::duplicateMerge:
148  case reco::TrackBase::cosmics:
149  case reco::TrackBase::initialStep:
150  case reco::TrackBase::highPtTripletStep:
151  case reco::TrackBase::lowPtQuadStep:
152  case reco::TrackBase::lowPtTripletStep:
153  case reco::TrackBase::pixelPairStep:
154  case reco::TrackBase::detachedQuadStep:
155  case reco::TrackBase::detachedTripletStep:
156  case reco::TrackBase::mixedTripletStep:
157  return true;
158  default:
159  return false;
160  }
161  }
162 
163  bool isFifthStep(const reco::TrackBase::TrackAlgorithm& algo) {
164  switch (algo) {
165  case reco::TrackBase::ctf:
166  case reco::TrackBase::duplicateMerge:
167  case reco::TrackBase::initialStep:
168  case reco::TrackBase::highPtTripletStep:
169  case reco::TrackBase::lowPtQuadStep:
170  case reco::TrackBase::lowPtTripletStep:
171  case reco::TrackBase::pixelPairStep:
172  case reco::TrackBase::jetCoreRegionalStep:
173  case reco::TrackBase::detachedQuadStep:
174  case reco::TrackBase::detachedTripletStep:
175  case reco::TrackBase::mixedTripletStep:
176  case reco::TrackBase::pixelLessStep:
177  case reco::TrackBase::muonSeededStepInOut:
178  case reco::TrackBase::muonSeededStepOutIn:
179  case reco::TrackBase::hltIter0:
180  case reco::TrackBase::hltIter1:
181  case reco::TrackBase::hltIter2:
182  case reco::TrackBase::hltIter3:
183  case reco::TrackBase::hltIter4:
184  case reco::TrackBase::hltIterX:
185  return false;
186  case reco::TrackBase::tobTecStep:
187  return true;
188  default:
189  return true;
190  }
191  }
192 
193  bool highQuality(const reco::TrackBase::TrackAlgorithm& algo) {
194  switch (algo) {
195  case reco::TrackBase::initialStep:
196  case reco::TrackBase::highPtTripletStep:
197  case reco::TrackBase::lowPtQuadStep:
198  case reco::TrackBase::lowPtTripletStep:
199  case reco::TrackBase::pixelPairStep:
200  case reco::TrackBase::detachedQuadStep:
201  case reco::TrackBase::detachedTripletStep:
202  case reco::TrackBase::duplicateMerge:
203  case reco::TrackBase::jetCoreRegionalStep:
204  return true;
205  default:
206  return false;
207  }
208  }
209 
210  bool nonIterative(const reco::TrackBase::TrackAlgorithm& algo) {
211  switch (algo) {
212  case reco::TrackBase::undefAlgorithm:
213  case reco::TrackBase::ctf:
214  case reco::TrackBase::cosmics:
215  return true;
216  default:
217  return false;
218  }
219  }
220 
221  bool step45(const reco::TrackBase::TrackAlgorithm& algo) {
222  switch (algo) {
223  case reco::TrackBase::mixedTripletStep:
224  case reco::TrackBase::pixelLessStep:
225  case reco::TrackBase::tobTecStep:
226  return true;
227  default:
228  return false;
229  }
230  }
231 
232  bool step5(const reco::TrackBase::TrackAlgorithm& algo) {
233  return (algo == reco::TrackBase::tobTecStep || algo == reco::TrackBase::pixelLessStep);
234  }
235 
236  bool goodPtResolution(const reco::TrackRef& trackref,
237  const std::vector<double>& DPtovPtCut,
238  const std::vector<unsigned>& NHitCut,
239  bool useIterTracking,
240  reco::TrackBase::TrackQuality trackQuality) {
241  //check quality of tracks
242  if (!trackref->quality(trackQuality))
243  return false;
244 
245  const double p = trackref->p();
246  const double pT = trackref->pt();
247  const double dpT = trackref->ptError();
248  const unsigned int nHit = trackref->hitPattern().trackerLayersWithMeasurement();
249  const unsigned int nLostHit = trackref->hitPattern().trackerLayersWithoutMeasurement(reco::HitPattern::TRACK_HITS);
250  const unsigned int lostHits = trackref->numberOfLostHits();
251  const double sigmaHad = std::sqrt(1.20 * 1.20 / p + 0.06 * 0.06) / (1. + lostHits);
252 
253  // Protection against 0 momentum tracks
254  if (p < 0.05)
255  return false;
256 
257  LogDebug("goodPtResolution") << " PFBlockAlgo: PFrecTrack->Track Pt= " << pT << " DPt = " << dpT << std::endl;
258 
259  double dptCut = dPtCut(trackref->algo(), DPtovPtCut, useIterTracking);
260  unsigned int nhitCut = nHitCut(trackref->algo(), NHitCut, useIterTracking);
261 
262  if ((dptCut > 0. && dpT / pT > dptCut * sigmaHad) || nHit < nhitCut) {
263  LogDebug("goodPtResolution") << " PFBlockAlgo: skip badly measured track"
264  << ", P = " << p << ", Pt = " << pT << " DPt = " << dpT << ", N(hits) = " << nHit
265  << " (Lost : " << lostHits << "/" << nLostHit << ")"
266  << ", Algo = " << trackref->algo() << std::endl;
267  LogDebug("goodPtResolution") << " cut is DPt/Pt < " << dptCut * sigmaHad << std::endl;
268  LogDebug("goodPtResolution") << " cut is NHit >= " << nhitCut << std::endl;
269  return false;
270  }
271 
272  return true;
273  }
274 
275 } // namespace PFTrackAlgoTools
PFTrackAlgoTools::goodPtResolution
bool goodPtResolution(const reco::TrackRef &, const std::vector< double > &DPtovPtCut, const std::vector< unsigned > &NHitCut, bool useIterTracking, const reco::TrackBase::TrackQuality trackQuality)
Definition: PFTrackAlgoTools.cc:236
PFTrackAlgoTools::errorScale
double errorScale(const reco::TrackBase::TrackAlgorithm &, const std::vector< double > &)
Definition: PFTrackAlgoTools.cc:86
reco::TrackBase::TrackQuality
TrackQuality
track quality
Definition: TrackBase.h:150
reco::TrackBase::TrackAlgorithm
TrackAlgorithm
track algorithm
Definition: TrackBase.h:89
PFTrackAlgoTools::nHitCut
unsigned int nHitCut(const reco::TrackBase::TrackAlgorithm &, const std::vector< unsigned int > &, bool)
Definition: PFTrackAlgoTools.cc:46
cuts
TkSoA const *__restrict__ CAHitNtupletGeneratorKernelsGPU::QualityCuts cuts
Definition: CAHitNtupletGeneratorKernelsImpl.h:416
PFTrackAlgoTools::isGoodForEGM
bool isGoodForEGM(const reco::TrackBase::TrackAlgorithm &)
Definition: PFTrackAlgoTools.cc:117
mathSSE::sqrt
T sqrt(T t)
Definition: SSEVec.h:19
PFTrackAlgoTools::nonIterative
bool nonIterative(const reco::TrackBase::TrackAlgorithm &)
Definition: PFTrackAlgoTools.cc:210
PFTrackAlgoTools::step45
bool step45(const reco::TrackBase::TrackAlgorithm &)
Definition: PFTrackAlgoTools.cc:221
PFTrackAlgoTools::highQuality
bool highQuality(const reco::TrackBase::TrackAlgorithm &)
Definition: PFTrackAlgoTools.cc:193
Reconstruction_hiPF_cff.trackQuality
tuple trackQuality
Definition: Reconstruction_hiPF_cff.py:60
PFTrackAlgoTools::isFifthStep
bool isFifthStep(const reco::TrackBase::TrackAlgorithm &)
Definition: PFTrackAlgoTools.cc:163
PFTrackAlgoTools::dPtCut
double dPtCut(const reco::TrackBase::TrackAlgorithm &, const std::vector< double > &, bool)
Definition: PFTrackAlgoTools.cc:6
AlCaHLTBitMon_ParallelJobs.p
tuple p
Definition: AlCaHLTBitMon_ParallelJobs.py:153
PFTrackAlgoTools::isGoodForEGMPrimary
bool isGoodForEGMPrimary(const reco::TrackBase::TrackAlgorithm &)
Definition: PFTrackAlgoTools.cc:144
PFTrackAlgoTools::step5
bool step5(const reco::TrackBase::TrackAlgorithm &)
Definition: PFTrackAlgoTools.cc:232
LogDebug
#define LogDebug(id)
Definition: MessageLogger.h:233
Generated for CMSSW Reference Manual by doxygen 1.8.5