Merge branch 'master' of https://git.cern.ch/reps/AliRoot

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

diff --git a/PWGPP/pid/AddTaskLambdaBayes.C b/PWGPP/pid/AddTaskLambdaBayes.C
index 289bbaf7e60903a44e33175d73d3fcfea4fa87b3..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,6 +22,13 @@ 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);