Merge branch 'feature-movesplit'

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

diff --git a/PWGPP/pid/AddTaskPhiBayes.C b/PWGPP/pid/AddTaskPhiBayes.C
index 606b608e27c963f7350584171fbbed7ece7acb7b..499dc3aad9cc3f19e9b7c7be9167f84a5487c1cd 100644 (file)
--- a/PWGPP/pid/AddTaskPhiBayes.C
+++ b/PWGPP/pid/AddTaskPhiBayes.C
@@ -1,4 +1,4 @@
-AliAnalysisTask *AddTaskPhiBayes(Bool_t ismc=kFALSE,Bool_t qa=kTRUE,Int_t filterbit=16){
+AliAnalysisTask *AddTaskPhiBayes(Bool_t ismc=kFALSE,Bool_t qa=kTRUE,Int_t filterbit=16,Int_t typeCol=2,Bool_t toEP=kFALSE,Int_t species = 2){
 
   //get the current analysis manager
   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
@@ -21,6 +21,14 @@ AliAnalysisTask *AddTaskPhiBayes(Bool_t ismc=kFALSE,Bool_t qa=kTRUE,Int_t filter
   if(qa) task->SetQA();
   task->SetEtaCut(0.8);
   task->SetFilterBit(filterbit);
+  task->SetTypeCollisions(typeCol);
+  task->SetCorrEP(toEP);
+  task->SetRefSpecies(species);
+
+  AliPIDmaxProb *userCut = new AliPIDmaxProb("maxProbKaon");
+  userCut->RequireTPC();
+  userCut->RequireTOF();
+  task->SetPIDuserCut(userCut);
 
   mgr->AddTask(task);