Changeset 341014c in git for modules/TrackCountingTauTagging.cc

Timestamp:

Feb 12, 2019, 9:29:17 PM (6 years ago)

Author:

Pavel Demin <pavel-demin@…>

Branches:

ImprovedOutputFile, Timing, llp, master

Children:

6455202

Parents:

45e58be

Message:

apply .clang-format to all .h, .cc and .cpp files

File:

• modules/TrackCountingTauTagging.cc (modified) (16 diffs)

modules/TrackCountingTauTagging.cc

@@ -20 +20 @@
 #include "classes/DelphesFormula.h"
 
+#include "ExRootAnalysis/ExRootClassifier.h"
+#include "ExRootAnalysis/ExRootFilter.h"
 #include "ExRootAnalysis/ExRootResult.h"
-#include "ExRootAnalysis/ExRootFilter.h"
-#include "ExRootAnalysis/ExRootClassifier.h"
-
+
+#include "TDatabasePDG.h"
+#include "TFormula.h"
+#include "TLorentzVector.h"
 #include "TMath.h"
+#include "TObjArray.h"
+#include "TRandom3.h"
 #include "TString.h"
-#include "TFormula.h"
-#include "TRandom3.h"
-#include "TObjArray.h"
-#include "TDatabasePDG.h"
-#include "TLorentzVector.h"
 
 #include <algorithm>
-#include <stdexcept>
 #include <iostream>
 #include <sstream>
+#include <stdexcept>
 
 using namespace std;
@@ -44 +44 @@
 {
 public:
-
   TrackCountingTauTaggingPartonClassifier(const TObjArray *array);
 
@@ -67 +66 @@
   Candidate *daughter1 = 0;
   Candidate *daughter2 = 0;
-  
+
   const TLorentzVector &momentum = tau->Momentum;
   Int_t pdgCode, i, j;
@@ -80 +79 @@
   if(tau->D2 < tau->D1) return -1;
 
-  if(tau->D1 >= fParticleInputArray->GetEntriesFast() ||
-     tau->D2 >= fParticleInputArray->GetEntriesFast())
+  if(tau->D1 >= fParticleInputArray->GetEntriesFast() || tau->D2 >= fParticleInputArray->GetEntriesFast())
   {
     throw runtime_error("tau's daughter index is greater than the ParticleInputArray size");
@@ -90 +88 @@
     daughter1 = static_cast<Candidate *>(fParticleInputArray->At(i));
     pdgCode = TMath::Abs(daughter1->PID);
-    if(pdgCode == 11 || pdgCode == 13 || pdgCode == 15) return -1;
+    if(pdgCode == 11 || pdgCode == 13 || pdgCode == 15)
+      return -1;
     else if(pdgCode == 24)
     {
-     if(daughter1->D1 < 0) return -1;
-     for(j = daughter1->D1; j <= daughter1->D2; ++j)
-     {
-       daughter2 = static_cast<Candidate*>(fParticleInputArray->At(j));
-       pdgCode = TMath::Abs(daughter2->PID);
-       if(pdgCode == 11 || pdgCode == 13) return -1;
-     }
-        
+      if(daughter1->D1 < 0) return -1;
+      for(j = daughter1->D1; j <= daughter1->D2; ++j)
+      {
+        daughter2 = static_cast<Candidate *>(fParticleInputArray->At(j));
+        pdgCode = TMath::Abs(daughter2->PID);
+        if(pdgCode == 11 || pdgCode == 13) return -1;
+      }
     }
   }
@@ -125 +123 @@
 void TrackCountingTauTagging::Init()
 {
-  map< Int_t, DelphesFormula * >::iterator itEfficiencyMap;
+  map<Int_t, DelphesFormula *>::iterator itEfficiencyMap;
   ExRootConfParam param;
   DelphesFormula *formula;
@@ -135 +133 @@
   fDeltaRTrack = GetDouble("DeltaRTrack", 0.2);
   fTrackPTMin = GetDouble("TrackPTMin", 1.0);
-  
+
   // read efficiency formulas
   param = GetParam("EfficiencyFormula");
@@ -141 +139 @@
 
   fEfficiencyMap.clear();
-  for(i = 0; i < size/2; ++i)
+  for(i = 0; i < size / 2; ++i)
   {
     formula = new DelphesFormula;
-    formula->Compile(param[i*2 + 1].GetString());
-
-    fEfficiencyMap[param[i*2].GetInt()] = formula;
+    formula->Compile(param[i * 2 + 1].GetString());
+
+    fEfficiencyMap[param[i * 2].GetInt()] = formula;
   }
 
@@ -172 +170 @@
   fTrackInputArray = ImportArray(GetString("TrackInputArray", "TrackMerger/tracks"));
   fItTrackInputArray = fTrackInputArray->MakeIterator();
-  
+
   fFilter = new ExRootFilter(fPartonInputArray);
 
@@ -183 +181 @@
 void TrackCountingTauTagging::Finish()
 {
-  map< Int_t, DelphesFormula * >::iterator itEfficiencyMap;
+  map<Int_t, DelphesFormula *>::iterator itEfficiencyMap;
   DelphesFormula *formula;
 
@@ -207 +205 @@
   Double_t pt, eta, phi, e;
   TObjArray *tauArray;
-  map< Int_t, DelphesFormula * >::iterator itEfficiencyMap;
+  map<Int_t, DelphesFormula *>::iterator itEfficiencyMap;
   DelphesFormula *formula;
   Int_t pdgCode, charge, i, identifier;
@@ -232 +230 @@
     e = jetMomentum.E();
 
-
-// loop over all input tracks
+    // loop over all input tracks
     fItTrackInputArray->Reset();
     while((track = static_cast<Candidate *>(fItTrackInputArray->Next())))
     {
-        if((track->Momentum).Pt() < fTrackPTMin) continue;
-        if(jetMomentum.DeltaR(track->Momentum) <= fDeltaRTrack) {
-            identifier -= 1;
-            charge += track->Charge;
-        }
-    }
-    
+      if((track->Momentum).Pt() < fTrackPTMin) continue;
+      if(jetMomentum.DeltaR(track->Momentum) <= fDeltaRTrack)
+      {
+        identifier -= 1;
+        charge += track->Charge;
+      }
+    }
+
     // loop over all input taus
     itTauArray.Reset();
@@ -251 +249 @@
       if(tau->D1 < 0) continue;
 
-      if(tau->D1 >= fParticleInputArray->GetEntriesFast() ||
-         tau->D2 >= fParticleInputArray->GetEntriesFast())
+      if(tau->D1 >= fParticleInputArray->GetEntriesFast() || tau->D2 >= fParticleInputArray->GetEntriesFast())
       {
         throw runtime_error("tau's daughter index is greater than the ParticleInputArray size");
@@ -263 +260 @@
         daughter = static_cast<Candidate *>(fParticleInputArray->At(i));
         if(TMath::Abs(daughter->PID) == 16) continue;
-        tauMomentum += daughter->Momentum;  
+        tauMomentum += daughter->Momentum;
       }
 
       if(jetMomentum.DeltaR(tauMomentum) <= fDeltaR)
-      {        
+      {
         matchedTau = true;
         pdgCode = 15;
@@ -273 +270 @@
     }
     if(matchedTau)
-        identifier *= -1;
+      identifier *= -1;
     // find an efficency formula
     // If the identifier is larger than 2, set it to 2 (multiprong requires at least 2 tracks)
-    if (identifier > 2)
-        identifier = 2;
-    else if (identifier < -2)
-        identifier = -2;
-
-    
+    if(identifier > 2)
+      identifier = 2;
+    else if(identifier < -2)
+      identifier = -2;
+
     itEfficiencyMap = fEfficiencyMap.find(identifier);
     if(itEfficiencyMap == fEfficiencyMap.end())
@@ -293 +289 @@
     // apply an efficency formula
     jet->TauTag |= (gRandom->Uniform() <= formula->Eval(pt, eta, phi, e)) << fBitNumber;
-   
-   
+
     // set tau charge
     jet->Charge = charge;