Improved particle selection in TOF

diff --git a/PWGLF/STRANGENESS/Hypernuclei/AliAnalysisTaskNucleiv2SP.cxx b/PWGLF/STRANGENESS/Hypernuclei/AliAnalysisTaskNucleiv2SP.cxx
index 1122e57040a11d7923b020a5cf5189dc268d5148..c8b5891057aa566ff48e9ddf06f54c42b585482f 100644 (file)
--- a/PWGLF/STRANGENESS/Hypernuclei/AliAnalysisTaskNucleiv2SP.cxx
+++ b/PWGLF/STRANGENESS/Hypernuclei/AliAnalysisTaskNucleiv2SP.cxx
@@ -667,6 +667,9 @@ void AliAnalysisTaskNucleiv2SP::UserExec(Option_t *)
     Float_t  mass  = -99;
     Double_t pt  = esdtrack->Pt();
 
+    if(fptc==3)
+      pt = 2*pt;
+
     if(TMath::Abs(ptot) < pmax && TMath::Abs(pullTPC) <= 3 && TMath::Abs(pt) < ptmax){
 
       fhBBDeu->Fill(ptot*esdtrack->GetSign(),TPCSignal);
@@ -680,8 +683,18 @@ void AliAnalysisTaskNucleiv2SP::UserExec(Option_t *)
        mass = ptot/TMath::Sqrt(gamma*gamma - 1); // using inner TPC mom. as approx.
        
        fhTOF->Fill(ptot*esdtrack->GetSign(),beta);
-       if(TMath::Abs(mass) > 2.7)continue;
-       if(TMath::Abs(mass) < 1. )continue;
+       if(fptc==1){
+         if(TMath::Abs(mass) > 2.7)continue;
+         if(TMath::Abs(mass) < 1. )continue;
+       }
+       if(fptc==2){
+         if(TMath::Abs(mass) > 5.0)continue;
+         if(TMath::Abs(mass) < 1.8 )continue;
+       }
+       if(fptc==3){
+         if(TMath::Abs(mass) > 5.0)continue;
+         if(TMath::Abs(mass) < 1.8 )continue;
+       }
        fhMassTOF->Fill(mass);
       }