Changes in modules/Isolation.cc [48b6e45:e72ee4b] in git

File:

• modules/Isolation.cc (modified) (3 diffs)

modules/Isolation.cc

@@ -108 +108 @@
 
   fUsePTSum = GetBool("UsePTSum", false);
+
+  fUseRhoCorr = GetBool("UseRhoCorrection", false);
 
   fClassifier->fPTMin = GetDouble("PTMin", 0.5);
@@ -153 +155 @@
   Candidate *candidate, *isolation, *object;
   TObjArray *isolationArray;
-  Double_t sumCharged, sumNeutral, sumAllParticles, sumChargedPU, sumDBeta, ratioDBeta, sumRhoCorr, ratioRhoCorr;
+  Double_t sumCharged, sumNeutral, sumAllParticles, sumChargedPU, sumDBeta, ratioDBeta, sumRhoCorr, ratioRhoCorr, sum, ratio;
   Int_t counter;
   Double_t eta = 0.0;
@@ -245 +247 @@
     candidate->SumPt = sumAllParticles;
 
-    if((fUsePTSum && sumDBeta > fPTSumMax) || (!fUsePTSum && ratioDBeta > fPTRatioMax)) continue;
+    sum = (fUseRhoCorr) ? sumRhoCorr : sumDBeta;
+    if(fUsePTSum && sum > fPTSumMax) continue;
+
+    ratio = (fUseRhoCorr) ? ratioRhoCorr : ratioDBeta;
+    if(!fUsePTSum && ratio > fPTRatioMax) continue;
+
     fOutputArray->Add(candidate);
   }