Merge branch 'feature-movesplit'

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

diff --git a/PWGPP/pid/AddTaskK0sBayes.C b/PWGPP/pid/AddTaskK0sBayes.C
index 972c2358bd3c9af2b3035b18e5d1d8cc07513402..42ec1796f76be36ca979f5fe4ff8b4684ead18b8 100644 (file)
--- a/PWGPP/pid/AddTaskK0sBayes.C
+++ b/PWGPP/pid/AddTaskK0sBayes.C
@@ -1,5 +1,4 @@
-AliAnalysisTask *AddTaskK0sBayes(Bool_t ismc=kFALSE,Bool_t qa=kTRUE,Int_t filterbit=4,Int_t typeCol=2,Bool_t toEP=kFALSE){
-
+AliAnalysisTask *AddTaskK0sBayes(Bool_t ismc=kFALSE,Bool_t qa=kTRUE,Int_t filterbit=4,Int_t typeCol=2,Bool_t toEP=kFALSE,Int_t species = 2){
   //get the current analysis manager
   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
   if (!mgr) {
@@ -23,6 +22,7 @@ AliAnalysisTask *AddTaskK0sBayes(Bool_t ismc=kFALSE,Bool_t qa=kTRUE,Int_t filter
   task->SetFilterBit(filterbit);
   task->SetTypeCollisions(typeCol);
   task->SetCorrEP(toEP);
+  task->SetRefSpecies(species);
 
   AliPIDmaxProb *userCut = new AliPIDmaxProb("maxProbPion");
   task->SetPIDuserCut(userCut);