Changeset e40b9cf in git for modules/TrackCountingBTagging.cc

Timestamp:

Oct 30, 2017, 12:56:01 AM (7 years ago)

Author:

Kevin Pedro <kpedro88@…>

Branches:

ImprovedOutputFile, Timing, dual_readout, llp, master

Children:

9458496b

Parents:

cde9f31

Message:

add 3D IP significance option

File:

• modules/TrackCountingBTagging.cc (modified) (5 diffs)

modules/TrackCountingBTagging.cc

@@ -72 +72 @@
   fNtracks = GetInt("Ntracks", 3);
 
+  fUse3D = GetBool("Use3D", false);
+
   // import input array(s)
 
@@ -95 +97 @@
   Candidate *jet, *track;
 
-  Double_t jpx, jpy;
+  Double_t jpx, jpy, jpz;
   Double_t dr, tpt;
-  Double_t xd, yd, d0, dd0, ip, sip;
+  Double_t xd, yd, zd, d0, dd0, dz, ddz, sip;
 
   Int_t sign;
@@ -110 +112 @@
     jpx = jetMomentum.Px();
     jpy = jetMomentum.Py();
+    jpz = jetMomentum.Pz();
 
     // loop over all input tracks
@@ -122 +125 @@
       xd = track->Xd;
       yd = track->Yd;
+      zd = track->Zd;
       d0 = TMath::Abs(track->D0);
-      dd0 = track->ErrorD0;
+      dd0 = TMath::Abs(track->ErrorD0);
+      dz = TMath::Abs(track->DZ);
+      ddz = TMath::Abs(track->ErrorDZ);
 
       if(tpt < fPtMin) continue;
@@ -129 +135 @@
       if(d0 > fIPmax) continue;
 
-      sign = (jpx*xd + jpy*yd > 0.0) ? 1 : -1;
-
-      ip = sign*d0;
-      sip = ip / TMath::Abs(dd0);
+      if(fUse3D){
+        sign = (jpx*xd + jpy*yd + jpz*zd > 0.0) ? 1 : -1;
+        //add transvers and longitudinal significances in quadrature
+        sip = sign * TMath::Sqrt( TMath::Power(d0 / dd0, 2) + TMath::Power(dz / ddz, 2) );
+      }
+      else {
+        sign = (jpx*xd + jpy*yd > 0.0) ? 1 : -1;
+        sip = sign * d0 / TMath::Abs(dd0);
+      }
 
       if(sip > fSigMin) count++;