CMS 3D CMS Logo

 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
CombinedSVComputer.h
Go to the documentation of this file.
1 #ifndef RecoBTag_SecondaryVertex_CombinedSVComputer_h
2 #define RecoBTag_SecondaryVertex_CombinedSVComputer_h
3 
4 #include <iostream>
5 #include <cstddef>
6 #include <string>
7 #include <cmath>
8 #include <vector>
9 
10 #include <Math/VectorUtil.h>
11 
15 
31 
36 
37 
38 #define range_for(i, x) \
39  for(int i = (x).begin; i != (x).end; i += (x).increment)
40 
41 
43  public:
44  explicit CombinedSVComputer(const edm::ParameterSet &params);
45 
47  operator () (const reco::TrackIPTagInfo &ipInfo,
48  const reco::SecondaryVertexTagInfo &svInfo) const;
50  operator () (const reco::CandIPTagInfo &ipInfo,
51  const reco::CandSecondaryVertexTagInfo &svInfo) const;
52 
53  struct IterationRange {
55  };
56  double flipValue(double value, bool vertex) const;
57  IterationRange flipIterate(int size, bool vertex) const;
58 
60 
62  threshTrack(const reco::CandIPTagInfo &trackIPTagInfo,
64  const reco::Jet &jet,
65  const GlobalPoint &pv) const;
67  threshTrack(const reco::TrackIPTagInfo &trackIPTagInfo,
69  const reco::Jet &jet,
70  const GlobalPoint &pv) const;
71  template <class SVTI,class IPTI>
73  const IPTI & ipInfo,const SVTI & svInfo,
74  double & vtx_track_ptSum, double & vtx_track_ESum) const;
75 
76  private:
77  bool trackFlip;
78  bool vertexFlip;
79  double charmCut;
84  unsigned int pseudoMultiplicityMin;
85  unsigned int trackMultiplicityMin;
91  std::vector<reco::btau::TaggingVariableName> taggingVariables;
92 };
93 
94 template <class SVTI,class IPTI>
96  const IPTI & ipInfo, const SVTI & svInfo,
97  double & vtx_track_ptSum, double & vtx_track_ESum) const
98 {
99  using namespace ROOT::Math;
100  using namespace reco;
101 
102  edm::RefToBase<Jet> jet = ipInfo.jet();
103  math::XYZVector jetDir = jet->momentum().Unit();
104  bool havePv = ipInfo.primaryVertex().isNonnull();
105  GlobalPoint pv;
106  if (havePv)
107  pv = GlobalPoint(ipInfo.primaryVertex()->x(),
108  ipInfo.primaryVertex()->y(),
109  ipInfo.primaryVertex()->z());
110 
112 
113 
114  vars.insert(btau::jetPt, jet->pt(), true);
115  vars.insert(btau::jetEta, jet->eta(), true);
116 
117  if (ipInfo.selectedTracks().size() < trackMultiplicityMin)
118  return;
119 
120  vars.insert(btau::jetNTracks, ipInfo.selectedTracks().size(), true);
121 
122  TrackKinematics allKinematics;
123  TrackKinematics trackJetKinematics;
124 
125  double jet_track_ESum= 0.;
126 
127  int vtx = -1;
128 
129  IterationRange range = flipIterate(svInfo.nVertices(), true);
130  range_for(i , range) {
131  if (vtx < 0) vtx = i;
132  }
133 
134  if (vtx >= 0) {
135  vtxType = btag::Vertices::RecoVertex;
136 
137  vars.insert(btau::flightDistance2dVal,flipValue(svInfo.flightDistance(vtx, true).value(),true),true);
138  vars.insert(btau::flightDistance2dSig,flipValue(svInfo.flightDistance(vtx, true).significance(),true),true);
139  vars.insert(btau::flightDistance3dVal,flipValue(svInfo.flightDistance(vtx, false).value(),true),true);
140  vars.insert(btau::flightDistance3dSig,flipValue(svInfo.flightDistance(vtx, false).significance(),true),true);
141  vars.insert(btau::vertexJetDeltaR,Geom::deltaR(svInfo.flightDirection(vtx), jetDir),true);
142  vars.insert(btau::jetNSecondaryVertices, svInfo.nVertices(), true);
143  }
144 
145  std::vector<std::size_t> indices = ipInfo.sortedIndexes(sortCriterium);
146  const std::vector<reco::btag::TrackIPData> &ipData = ipInfo.impactParameterData();
147 
148  const typename IPTI::input_container &tracks = ipInfo.selectedTracks();
149  std::vector<const Track *> pseudoVertexTracks;
150 
151  const Track * trackPairV0Test[2];
152  range = flipIterate(indices.size(), false);
153  range_for(i, range) {
154  std::size_t idx = indices[i];
155  const reco::btag::TrackIPData &data = ipData[idx];
156  const Track * trackPtr = reco::btag::toTrack(tracks[idx]);
157  const Track &track = *trackPtr;
158 
159  jet_track_ESum += std::sqrt(track.momentum().Mag2() + ROOT::Math::Square(ParticleMasses::piPlus));
160 
161  // add track to kinematics for all tracks in jet
162  //allKinematics.add(track); // would make more sense for some variables, e.g. vertexEnergyRatio nicely between 0 and 1, but not necessarily the best option for the discriminating power...
163 
164  // filter track -> this track selection can be tighter than the vertex track selection (used to fill the track related variables...)
165  if (!trackSelector(track, data, *jet, pv))
166  continue;
167 
168  // add track to kinematics for all tracks in jet
169  allKinematics.add(track);
170 
171  // if no vertex was reconstructed, attempt pseudo vertex
172  if (vtxType == btag::Vertices::NoVertex && trackPseudoSelector(track, data, *jet, pv)) {
173  pseudoVertexTracks.push_back(trackPtr);
174  vertexKinematics.add(track);
175  }
176 
177  // check against all other tracks for V0 track pairs
178  trackPairV0Test[0] = reco::btag::toTrack(tracks[idx]);
179  bool ok = true;
180  range_for(j, range) {
181  if (i == j)
182  continue;
183 
184  std::size_t pairIdx = indices[j];
185  const reco::btag::TrackIPData &pairTrackData = ipData[pairIdx];
186  const Track * pairTrackPtr = reco::btag::toTrack(tracks[pairIdx]);
187  const Track &pairTrack = *pairTrackPtr;
188 
189  if (!trackSelector(pairTrack, pairTrackData, *jet, pv))
190  continue;
191 
192  trackPairV0Test[1] = pairTrackPtr;
193  if (!trackPairV0Filter(trackPairV0Test, 2)) {
194  ok = false;
195  break;
196  }
197  }
198  if (!ok)
199  continue;
200 
201  trackJetKinematics.add(track);
202 
203  // add track variables
204  math::XYZVector trackMom = track.momentum();
205  double trackMag = std::sqrt(trackMom.Mag2());
206 
207  vars.insert(btau::trackSip3dVal, flipValue(data.ip3d.value(), false), true);
208  vars.insert(btau::trackSip3dSig, flipValue(data.ip3d.significance(), false), true);
209  vars.insert(btau::trackSip2dVal, flipValue(data.ip2d.value(), false), true);
210  vars.insert(btau::trackSip2dSig, flipValue(data.ip2d.significance(), false), true);
212 // vars.insert(btau::trackJetDistSig, data.distanceToJetAxis.significance(), true);
213 // vars.insert(btau::trackFirstTrackDist, data.distanceToFirstTrack, true);
214 // vars.insert(btau::trackGhostTrackVal, data.distanceToGhostTrack.value(), true);
215 // vars.insert(btau::trackGhostTrackSig, data.distanceToGhostTrack.significance(), true);
216  vars.insert(btau::trackDecayLenVal, havePv ? (data.closestToJetAxis - pv).mag() : -1.0, true);
217 
218  vars.insert(btau::trackMomentum, trackMag, true);
219  vars.insert(btau::trackEta, trackMom.Eta(), true);
220 
221  vars.insert(btau::trackPtRel, VectorUtil::Perp(trackMom, jetDir), true);
222  vars.insert(btau::trackPPar, jetDir.Dot(trackMom), true);
223  vars.insert(btau::trackDeltaR, VectorUtil::DeltaR(trackMom, jetDir), true);
224  vars.insert(btau::trackPtRatio, VectorUtil::Perp(trackMom, jetDir) / trackMag, true);
225  vars.insert(btau::trackPParRatio, jetDir.Dot(trackMom) / trackMag, true);
226  }
227 
228  if (vtxType == btag::Vertices::NoVertex && vertexKinematics.numberOfTracks() >= pseudoMultiplicityMin && pseudoVertexV0Filter(pseudoVertexTracks))
229  {
231  for(std::vector<const Track *>::const_iterator track = pseudoVertexTracks.begin(); track != pseudoVertexTracks.end(); ++track)
232  {
233  vars.insert(btau::trackEtaRel, reco::btau::etaRel(jetDir,(*track)->momentum()), true);
234  vtx_track_ptSum += std::sqrt((*track)->momentum().Perp2());
235  vtx_track_ESum += std::sqrt((*track)->momentum().Mag2() + ROOT::Math::Square(ParticleMasses::piPlus));
236  }
237  }
238 
239  vars.insert(btau::vertexCategory, vtxType, true);
240 
241  vars.insert(btau::trackJetPt, trackJetKinematics.vectorSum().Pt(), true);
242  vars.insert(btau::trackSumJetDeltaR,VectorUtil::DeltaR(allKinematics.vectorSum(), jetDir), true);
243  vars.insert(btau::trackSumJetEtRatio,allKinematics.vectorSum().Et() / ipInfo.jet()->et(), true);
244 
245  vars.insert(btau::trackSip3dSigAboveCharm, flipValue(threshTrack(ipInfo, reco::btag::IP3DSig, *jet, pv).ip3d.significance(),false),true);
246  vars.insert(btau::trackSip3dValAboveCharm, flipValue(threshTrack(ipInfo, reco::btag::IP3DSig, *jet, pv).ip3d.value(),false),true);
247  vars.insert(btau::trackSip2dSigAboveCharm, flipValue(threshTrack(ipInfo, reco::btag::IP2DSig, *jet, pv).ip2d.significance(),false),true);
248  vars.insert(btau::trackSip2dValAboveCharm, flipValue(threshTrack(ipInfo, reco::btag::IP2DSig, *jet, pv).ip2d.value(),false),true);
249 
250  if (vtxType != btag::Vertices::NoVertex) {
252  ? allKinematics.weightedVectorSum()
253  : allKinematics.vectorSum();
255  ? vertexKinematics.weightedVectorSum()
256  : vertexKinematics.vectorSum();
257 
258  if (vtxType != btag::Vertices::RecoVertex) {
259  vars.insert(btau::vertexNTracks,vertexKinematics.numberOfTracks(), true);
260  vars.insert(btau::vertexJetDeltaR,VectorUtil::DeltaR(vertexSum, jetDir), true);
261  }
262 
263  double vertexMass = vertexSum.M();
264  if (vtxType == btag::Vertices::RecoVertex &&
266  GlobalVector dir = svInfo.flightDirection(vtx);
267  double vertexPt2 = math::XYZVector(dir.x(), dir.y(), dir.z()).Cross(vertexSum).Mag2() / dir.mag2();
268  vertexMass = std::sqrt(vertexMass * vertexMass + vertexPt2) + std::sqrt(vertexPt2);
269  }
270  vars.insert(btau::vertexMass, vertexMass, true);
271 
272  double varPi = (vertexMass/5.2794) * (vtx_track_ESum /jet_track_ESum); // 5.2794 should be the average B meson mass of PDG! CHECK!!!
273  vars.insert(btau::massVertexEnergyFraction, varPi / (varPi + 0.04), true);
274  double varB = (std::sqrt(5.2794) * vtx_track_ptSum) / ( vertexMass * std::sqrt(jet->pt()));
275  vars.insert(btau::vertexBoostOverSqrtJetPt,varB*varB/(varB*varB + 10.), true);
276 
277  if (allKinematics.numberOfTracks()) {
278  vars.insert(btau::vertexEnergyRatio, vertexSum.E() / allSum.E(), true);
279  }
280  else {
281  vars.insert(btau::vertexEnergyRatio, 1, true);
282  }
283  }
284 
285  reco::PFJet const * pfJet = dynamic_cast<reco::PFJet const *>( &* jet ) ;
286  pat::Jet const * patJet = dynamic_cast<pat::Jet const *>( &* jet ) ;
287  if ( pfJet != 0 )
288  {
289  vars.insert(btau::chargedHadronEnergyFraction,pfJet->chargedHadronEnergyFraction(), true);
290  vars.insert(btau::neutralHadronEnergyFraction,pfJet->neutralHadronEnergyFraction(), true);
291  vars.insert(btau::photonEnergyFraction,pfJet->photonEnergyFraction(), true);
292  vars.insert(btau::electronEnergyFraction,pfJet->electronEnergyFraction(), true);
293  vars.insert(btau::muonEnergyFraction,pfJet->muonEnergyFraction(), true);
294  vars.insert(btau::chargedHadronMultiplicity,pfJet->chargedHadronMultiplicity(), true);
295  vars.insert(btau::neutralHadronMultiplicity,pfJet->neutralHadronMultiplicity(), true);
296  vars.insert(btau::photonMultiplicity,pfJet->photonMultiplicity(), true);
297  vars.insert(btau::electronMultiplicity,pfJet->electronMultiplicity(), true);
298  vars.insert(btau::muonMultiplicity,pfJet->muonMultiplicity(), true);
299  vars.insert(btau::hadronMultiplicity,pfJet->chargedHadronMultiplicity()+pfJet->neutralHadronMultiplicity(), true);
300  vars.insert(btau::hadronPhotonMultiplicity,pfJet->chargedHadronMultiplicity()+pfJet->neutralHadronMultiplicity()+pfJet->photonMultiplicity(), true);
301  vars.insert(btau::totalMultiplicity,pfJet->chargedHadronMultiplicity()+pfJet->neutralHadronMultiplicity()+pfJet->photonMultiplicity()+pfJet->electronMultiplicity()+pfJet->muonMultiplicity(), true);
302 
303  }
304  else if( patJet != 0 && patJet->isPFJet() )
305  {
306  vars.insert(btau::chargedHadronEnergyFraction,patJet->chargedHadronEnergyFraction(), true);
307  vars.insert(btau::neutralHadronEnergyFraction,patJet->neutralHadronEnergyFraction(), true);
308  vars.insert(btau::photonEnergyFraction,patJet->photonEnergyFraction(), true);
309  vars.insert(btau::electronEnergyFraction,patJet->electronEnergyFraction(), true);
310  vars.insert(btau::muonEnergyFraction,patJet->muonEnergyFraction(), true);
311  vars.insert(btau::chargedHadronMultiplicity,patJet->chargedHadronMultiplicity(), true);
312  vars.insert(btau::neutralHadronMultiplicity,patJet->neutralHadronMultiplicity(), true);
313  vars.insert(btau::photonMultiplicity,patJet->photonMultiplicity(), true);
314  vars.insert(btau::electronMultiplicity,patJet->electronMultiplicity(), true);
315  vars.insert(btau::muonMultiplicity,patJet->muonMultiplicity(), true);
316  vars.insert(btau::hadronMultiplicity,patJet->chargedHadronMultiplicity()+patJet->neutralHadronMultiplicity(), true);
317  vars.insert(btau::hadronPhotonMultiplicity,patJet->chargedHadronMultiplicity()+patJet->neutralHadronMultiplicity()+patJet->photonMultiplicity(), true);
318  vars.insert(btau::totalMultiplicity,patJet->chargedHadronMultiplicity()+patJet->neutralHadronMultiplicity()+patJet->photonMultiplicity()+patJet->electronMultiplicity()+patJet->muonMultiplicity(), true);
319 
320  }
321  else
322  {
325  vars.insert(btau::photonEnergyFraction,0., true);
326  vars.insert(btau::electronEnergyFraction,0., true);
327  vars.insert(btau::muonEnergyFraction,0., true);
330  vars.insert(btau::photonMultiplicity,0, true);
331  vars.insert(btau::electronMultiplicity,0, true);
332  vars.insert(btau::muonMultiplicity,0, true);
333  vars.insert(btau::hadronMultiplicity,0, true);
335  vars.insert(btau::totalMultiplicity,0, true);
336  }
337 }
338 
339 
340 #endif // RecoBTag_SecondaryVertex_CombinedSVComputer_h
reco::TrackSelector trackSelector
const double piPlus
Definition: ParticleMasses.h:9
int i
Definition: DBlmapReader.cc:9
T mag2() const
Definition: PV3DBase.h:66
const math::XYZTLorentzVector & weightedVectorSum() const
unsigned int pseudoMultiplicityMin
void add(const reco::Track &track, double weight=1.0)
Base class for all types of Jets.
Definition: Jet.h:20
double etaRel(const math::XYZVector &dir, const math::XYZVector &track)
Measurement1D ip2d
Definition: IPTagInfo.h:31
reco::TrackSelector trackNoDeltaRSelector
reco::V0Filter pseudoVertexV0Filter
Global3DPoint GlobalPoint
Definition: GlobalPoint.h:10
T y() const
Definition: PV3DBase.h:63
virtual reco::TaggingVariableList operator()(const reco::TrackIPTagInfo &ipInfo, const reco::SecondaryVertexTagInfo &svInfo) const
CombinedSVComputer(const edm::ParameterSet &params)
const Vector & momentum() const
track momentum vector
Definition: TrackBase.h:662
const reco::Track * toTrack(const reco::TrackBaseRef &t)
Definition: IPTagInfo.h:24
Jets made from PFObjects.
Definition: PFJet.h:21
XYZTLorentzVectorD XYZTLorentzVector
Lorentz vector with cylindrical internal representation using pseudorapidity.
Definition: LorentzVector.h:29
reco::btag::SortCriteria sortCriterium
reco::TrackSelector trackPseudoSelector
T sqrt(T t)
Definition: SSEVec.h:48
T z() const
Definition: PV3DBase.h:64
int j
Definition: DBlmapReader.cc:9
Measurement1D ip3d
Definition: IPTagInfo.h:32
IterationRange flipIterate(int size, bool vertex) const
reco::V0Filter trackPairV0Filter
double deltaR(double eta1, double eta2, double phi1, double phi2)
Definition: TreeUtility.cc:17
std::vector< reco::btau::TaggingVariableName > taggingVariables
double significance() const
Definition: Measurement1D.h:32
unsigned int numberOfTracks() const
#define range_for(i, x)
GlobalPoint closestToJetAxis
Definition: IPTagInfo.h:29
tuple tracks
Definition: testEve_cfg.py:39
XYZVectorD XYZVector
spatial vector with cartesian internal representation
Definition: Vector3D.h:30
tuple idx
DEBUGGING if hasattr(process,&quot;trackMonIterativeTracking2012&quot;): print &quot;trackMonIterativeTracking2012 D...
double flipValue(double value, bool vertex) const
Analysis-level calorimeter jet class.
Definition: Jet.h:77
edm::ParameterSet dropDeltaR(const edm::ParameterSet &pset) const
double value() const
Definition: Measurement1D.h:28
const reco::btag::TrackIPData & threshTrack(const reco::CandIPTagInfo &trackIPTagInfo, const reco::btag::SortCriteria sort, const reco::Jet &jet, const GlobalPoint &pv) const
Measurement1D distanceToJetAxis
Definition: IPTagInfo.h:33
void fillCommonVariables(reco::TaggingVariableList &vars, reco::TrackKinematics &vertexKinematics, const IPTI &ipInfo, const SVTI &svInfo, double &vtx_track_ptSum, double &vtx_track_ESum) const
const math::XYZTLorentzVector & vectorSum() const
dbl *** dir
Definition: mlp_gen.cc:35
T x() const
Definition: PV3DBase.h:62
tuple size
Write out results.
void insert(const TaggingVariable &variable, bool delayed=false)
unsigned int trackMultiplicityMin