Added the possibility to the user to manage own PID

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

diff --git a/PWGPP/pid/AddTaskLambdaBayes.C b/PWGPP/pid/AddTaskLambdaBayes.C
index c1dd5c6fddb6764f46642711a6424894b28f0f99..32f9f6c3f5f47bbc5eadd9144244ea285201b38b 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){
+AliAnalysisTask *AddTaskLambdaBayes(Bool_t ismc=kFALSE,Bool_t qa=kTRUE,Int_t filterbit=4,Int_t typeCol=2){
 
   //get the current analysis manager
   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
@@ -21,6 +21,10 @@ AliAnalysisTask *AddTaskLambdaBayes(Bool_t ismc=kFALSE,Bool_t qa=kTRUE,Int_t fil
   if(qa) task->SetQA();
   task->SetEtaCut(0.8);
   task->SetFilterBit(filterbit);
+  task->SetTypeCollisions(typeCol);
+
+  AliPIDmaxProb *userCut = new AliPIDmaxProb("maxProbProton");
+  task->SetPIDuserCut(userCut);
 
   mgr->AddTask(task);