New base class for multiplicity tasks

[u/mrichter/AliRoot.git] / PWG2 / SPECTRA / AliAnalysisCentralCutESD.cxx

diff --git a/PWG2/SPECTRA/AliAnalysisCentralCutESD.cxx b/PWG2/SPECTRA/AliAnalysisCentralCutESD.cxx
index 14643d09d378dc4a0020cece8d4de6a32b8c5e33..fc212ad4e865644b32f6e8a2855a2ff8771ce088 100644 (file)
--- a/PWG2/SPECTRA/AliAnalysisCentralCutESD.cxx
+++ b/PWG2/SPECTRA/AliAnalysisCentralCutESD.cxx
@@ -129,19 +129,19 @@ Double_t priors=0;
 
 
 
-Bool_t AliAnalysisCentralCutESD::IsA(AliESDtrack *track, PDG_t fPartType){
+Bool_t AliAnalysisCentralCutESD::IsA(AliESDtrack *track, PDG_t reqPartType){
 // Determines the type of the particle
     Int_t charge;
 
-       if(fPartType < 0){
-               charge = -1;
-       }
-       else{
-               charge = 1;
-       }
+    if(reqPartType < 0){
+      charge = -1;
+    }
+    else{
+      charge = 1;
+    }
 
-    Double_t probability[5];
-    Double_t w[5];
+    Double_t probability[5] = {0.0,0.0,0.0,0.0,0.0};
+    Double_t w[5] = {0.0,0.0,0.0,0.0,0.0};
 
     Long64_t partType = 0;
 
@@ -165,6 +165,7 @@ Bool_t AliAnalysisCentralCutESD::IsA(AliESDtrack *track, PDG_t fPartType){
 
     if(fPIDtype.Contains("Bayesian")) {
                partType = TMath::LocMax(AliPID::kSPECIES,w);
+               if(partType<0.) return kFALSE;
     }
 
     else if(fPIDtype.Contains("Custom")){
@@ -180,7 +181,7 @@ Bool_t AliAnalysisCentralCutESD::IsA(AliESDtrack *track, PDG_t fPartType){
                return kFALSE;
     }
 
-    if((AliPID::ParticleCode(partType)) != fPartType){
+    if((AliPID::ParticleCode(partType)) != reqPartType){
                return kFALSE;
     }