Changeset 1372 in svn for trunk/modules/TrackCountingBTagging.cc

Timestamp:

Apr 17, 2014, 10:54:17 AM (11 years ago)

Author:

Pavel Demin

Message:

several minor corrections

File:

• trunk/modules/TrackCountingBTagging.cc (modified) (7 diffs)

trunk/modules/TrackCountingBTagging.cc

@@ -2 +2 @@
 /** \class TrackCountingBTagging
  *
- *  btagging algorithm based on counting tracks with large impact parameterDetermines origin of jet,
+ *  b-tagging algorithm based on counting tracks with large impact parameter
  *
  *  $Date: 2014-03-27 12:39:14 +0200 (Fri, 27 March 2014) $
@@ -25 +25 @@
 #include "TLorentzVector.h"
 
-#include <algorithm> 
+#include <algorithm>
 #include <stdexcept>
 #include <iostream>
@@ -31 +31 @@
 
 using namespace std;
-
 
 //------------------------------------------------------------------------------
@@ -50 +49 @@
 void TrackCountingBTagging::Init()
 {
- 
   fBitNumber = GetInt("BitNumber", 0);
-  
+
   fPtMin     = GetDouble("TrackPtMin", 1.0);
   fDeltaR    = GetDouble("DeltaR", 0.3);
   fIPmax     = GetDouble("TrackIPMax", 2.0);
-  
+
   fSigMin    = GetDouble("SigMin", 6.5);
-  fNtracks   = GetInt("Ntracks", 3); 
-   
+  fNtracks   = GetInt("Ntracks", 3);
+
   // import input array(s)
 
   fTrackInputArray = ImportArray(GetString("TrackInputArray", "Calorimeter/eflowTracks"));
   fItTrackInputArray = fTrackInputArray->MakeIterator();
-  
+
   fJetInputArray = ImportArray(GetString("JetInputArray", "FastJetFinder/jets"));
   fItJetInputArray = fJetInputArray->MakeIterator();
@@ -82 +80 @@
 {
   Candidate *jet, *track;
-  
+
   Double_t jpx, jpy;
   Double_t dr, tpx, tpy, tpt;
   Double_t xd, yd, dxy, ddxy, ip, sip;
- 
+
   Int_t sign;
- 
+
   Int_t count;
- 
+
   // loop over all input jets
   fItJetInputArray->Reset();
@@ -98 +96 @@
     jpx = jetMomentum.Px();
     jpy = jetMomentum.Py();
-    
+
     // loop over all input tracks
     fItTrackInputArray->Reset();
@@ -104 +102 @@
     while((track = static_cast<Candidate*>(fItTrackInputArray->Next())))
     {
-     const TLorentzVector &trkMomentum = track->Momentum;
-     
-     dr = jetMomentum.DeltaR(trkMomentum);
-     
-     tpt = trkMomentum.Pt();
-     tpx = trkMomentum.Px();
-     tpy = trkMomentum.Py();
-    
-     xd   = track->Xd;
-     yd   = track->Yd;
-     dxy  = TMath::Abs(track->Dxy);
-     ddxy = track->SDxy;
-     
-     if( tpt < fPtMin ) continue;
-     if( dr  > fDeltaR ) continue;
-     if( dxy > fIPmax ) continue;
-        
-     sign  = ( jpx*xd + jpy*yd > 0.0 ) ? 1 : -1;
-        
-     ip  = sign*dxy;
-     sip = ip / TMath::Abs(ddxy);
-     
-     if( sip > fSigMin ) count++;
-     
+      const TLorentzVector &trkMomentum = track->Momentum;
+
+      dr = jetMomentum.DeltaR(trkMomentum);
+
+      tpt = trkMomentum.Pt();
+      tpx = trkMomentum.Px();
+      tpy = trkMomentum.Py();
+
+      xd   = track->Xd;
+      yd   = track->Yd;
+      dxy  = TMath::Abs(track->Dxy);
+      ddxy = track->SDxy;
+
+      if(tpt < fPtMin) continue;
+      if(dr  > fDeltaR) continue;
+      if(dxy > fIPmax) continue;
+
+      sign  = (jpx*xd + jpy*yd > 0.0) ? 1 : -1;
+
+      ip  = sign*dxy;
+      sip = ip / TMath::Abs(ddxy);
+
+      if(sip > fSigMin) count++;
     }
-     
+
     // set BTag flag to true if count >= Ntracks
-    jet->BTag |= ( count >= fNtracks ) << fBitNumber;
+    jet->BTag |= (count >= fNtracks) << fBitNumber;
   }
 }