rapidity cut according to the mass and not using AliSpectraNameSpace

[u/mrichter/AliRoot.git] / PWGLF / SPECTRA / PiKaPr / TestAOD / AliAnalysisTaskSpectraAOD.cxx

diff --git a/PWGLF/SPECTRA/PiKaPr/TestAOD/AliAnalysisTaskSpectraAOD.cxx b/PWGLF/SPECTRA/PiKaPr/TestAOD/AliAnalysisTaskSpectraAOD.cxx
index f5af53758e8b291054374b96f5551c6565e6cbce..43b548e145110f9bd39b0bce9cf9b63214c37aa5 100644 (file)
--- a/PWGLF/SPECTRA/PiKaPr/TestAOD/AliAnalysisTaskSpectraAOD.cxx
+++ b/PWGLF/SPECTRA/PiKaPr/TestAOD/AliAnalysisTaskSpectraAOD.cxx
@@ -135,6 +135,7 @@ void AliAnalysisTaskSpectraAOD::UserExec(Option_t *)
        }\r
     }\r
   \r
+  Double_t mass[3]={1.39570000000000000e-01,4.93676999999999977e-01,9.38271999999999995e-01};//FIXME masses to be taken from AliHelperPID\r
   //main loop on tracks\r
   for (Int_t iTracks = 0; iTracks < fAOD->GetNumberOfTracks(); iTracks++) {\r
     AliAODTrack* track = fAOD->GetTrack(iTracks);\r
@@ -160,8 +161,9 @@ void AliAnalysisTaskSpectraAOD::UserExec(Option_t *)
     Int_t charge = track->Charge() > 0 ? kChPos : kChNeg;\r
     \r
     // Fill histograms, only if inside y and nsigma acceptance\r
-    if(idRec != kSpUndefined && fTrackCuts->CheckYCut ((AODParticleSpecies_t)idRec))fHistMan->GetHistogram2D(kHistPtRecSigma,idRec,charge)->Fill(track->Pt(),dca);\r
-    //can't put a continue because we still have to fill allcharged primaries, done later\r
+    if(idRec != kSpUndefined){\r
+      if(fTrackCuts->CheckYCut (mass[idRec]))fHistMan->GetHistogram2D(kHistPtRecSigma,idRec,charge)->Fill(track->Pt(),dca);\r
+    }//can't put a continue because we still have to fill allcharged primaries, done later\r
     \r
     /* MC Part */\r
     if (arrayMC) {\r
@@ -196,7 +198,7 @@ void AliAnalysisTaskSpectraAOD::UserExec(Option_t *)
       if(idRec == kSpUndefined) continue;\r
       \r
       // rapidity cut (reconstructed pt and identity)\r
-      if(!fTrackCuts->CheckYCut ((AODParticleSpecies_t)idRec)) continue;\r
+      if(!fTrackCuts->CheckYCut (mass[idRec])) continue;\r
       \r
       // Get true ID\r
       Int_t idGen     = fPID->GetParticleSpecie(partMC);\r