Coverity warning 22314

[u/mrichter/AliRoot.git] / PWGPP / pid / AddTaskLambdaBayes.C

diff --git a/PWGPP/pid/AddTaskLambdaBayes.C b/PWGPP/pid/AddTaskLambdaBayes.C
index 32f9f6c3f5f47bbc5eadd9144244ea285201b38b..8ed9ffe8627cb10d61d9e212362a3d6a32460882 100644 (file)
--- a/PWGPP/pid/AddTaskLambdaBayes.C
+++ b/PWGPP/pid/AddTaskLambdaBayes.C
@@ -1,4 +1,4 @@
-AliAnalysisTask *AddTaskLambdaBayes(Bool_t ismc=kFALSE,Bool_t qa=kTRUE,Int_t filterbit=4,Int_t typeCol=2){
+AliAnalysisTask *AddTaskLambdaBayes(Bool_t ismc=kFALSE,Bool_t qa=kTRUE,Int_t filterbit=4,Int_t typeCol=2,Bool_t toEP=kFALSE,Int_t species=4){
 
   //get the current analysis manager
   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
@@ -22,8 +22,12 @@ AliAnalysisTask *AddTaskLambdaBayes(Bool_t ismc=kFALSE,Bool_t qa=kTRUE,Int_t fil
   task->SetEtaCut(0.8);
   task->SetFilterBit(filterbit);
   task->SetTypeCollisions(typeCol);
+  task->SetCorrEP(toEP);
+  task->SetRefSpecies(species);
 
   AliPIDmaxProb *userCut = new AliPIDmaxProb("maxProbProton");
+  userCut->RequireTPC();
+  userCut->RequireTOF();
   task->SetPIDuserCut(userCut);
 
   mgr->AddTask(task);