added a check for TPC gaussianity
[u/mrichter/AliRoot.git] / PWGPP / pid / AddTaskLambdaBayes.C
1 AliAnalysisTask *AddTaskLambdaBayes(Bool_t ismc=kFALSE,Bool_t qa=kTRUE,Int_t filterbit=4,Int_t typeCol=2){
2
3   //get the current analysis manager
4   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
5   if (!mgr) {
6     Error("No manager found in AddTaskVZERO. Why?");
7     return 0;
8   }
9   // currently don't accept AOD input
10   if (!mgr->GetInputEventHandler()->InheritsFrom(AliAODInputHandler::Class())) {
11     Error("AddTaskLambdaBayes","This task works only with AOD input!");
12     return 0;
13   }
14
15   //========= Add tender to the ANALYSIS manager and set default storage =====
16   char mytaskName[100];
17   snprintf(mytaskName,100,"AliAnalysisTaskLambdaBayes.cxx"); 
18
19   AliAnalysisTaskLambdaBayes *task = new AliAnalysisTaskLambdaBayes(mytaskName);
20   if(ismc) task->SetMC();
21   if(qa) task->SetQA();
22   task->SetEtaCut(0.8);
23   task->SetFilterBit(filterbit);
24   task->SetTypeCollisions(typeCol);
25
26   AliPIDmaxProb *userCut = new AliPIDmaxProb("maxProbProton");
27   userCut->RequireTPC();
28   userCut->RequireTOF();
29   task->SetPIDuserCut(userCut);
30
31   mgr->AddTask(task);
32
33   //Attach input to my tasks
34   AliAnalysisDataContainer *cinput = mgr->CreateContainer("cchain1",TChain::Class(),AliAnalysisManager::kInputContainer);
35   mgr->ConnectInput(task,0,mgr->GetCommonInputContainer());
36
37   // Attach output to my tasks
38   AliAnalysisDataContainer *cOutputL= mgr->CreateContainer("contLambdaBayes1",TList::Class(), AliAnalysisManager::kOutputContainer, 
39 AliAnalysisManager::GetCommonFileName());
40   mgr->ConnectOutput(task, 1, cOutputL);
41
42   AliAnalysisDataContainer *cOutputL2= mgr->CreateContainer("contLambdaBayes2",TList::Class(), AliAnalysisManager::kOutputContainer, 
43 AliAnalysisManager::GetCommonFileName());
44   mgr->ConnectOutput(task, 2, cOutputL2);
45
46   AliAnalysisDataContainer *cOutputL3= mgr->CreateContainer("contLambdaBayes3",TList::Class(), AliAnalysisManager::kOutputContainer, 
47 AliAnalysisManager::GetCommonFileName());
48   mgr->ConnectOutput(task, 3, cOutputL3);
49
50   return task;
51 }
52