added option for variable binning depending on the dataset for high pt QA and spectra...
[u/mrichter/AliRoot.git] / PWG4 / macros / AddTaskPWG4HighPtQATPConly.C
1 //DEFINITION OF A FEW CONSTANTS
2
3 AliPWG4HighPtQATPConly* AddTaskPWG4HighPtQATPConly(char *prodType = "LHC10e14",int cuts=2)//1: Standard Cuts 2009 2: GetStandardITSTPCTrackCuts2009
4 {
5   // Creates HighPtQATPConly analysis task and adds it to the analysis manager.
6   
7   // A. Get the pointer to the existing analysis manager via the static access method.
8   //==============================================================================
9   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
10   if (!mgr) {
11     Error("AddTaskPWG4HighPtQATPConly", "No analysis manager to connect to.");
12     return NULL;
13   }  
14
15   // B. Check the analysis type using the event handlers connected to the analysis
16   //    manager. The availability of MC handler can also be checked here.
17   //==============================================================================
18   if (!mgr->GetInputEventHandler()) {
19     ::Error("AddTaskPWG4HighPtQATPConly", "This task requires an input event handler");
20     return NULL;
21   }  
22   TString type = mgr->GetInputEventHandler()->GetDataType(); // can be "ESD" or "AOD"
23   const char *analysisType = "ESD";//"TPC"
24
25   // C. Create the task, add it to manager.
26   //===========================================================================
27  
28   //CREATE THE  CUTS -----------------------------------------------
29   //Use AliESDtrackCuts
30   AliESDtrackCuts *trackCuts = new AliESDtrackCuts("AliESDtrackCuts","Standard Cuts");
31   if(cuts==1) {
32     //Standard Cuts
33     trackCuts->SetAcceptKinkDaughters(kFALSE);
34     trackCuts->SetRequireTPCStandAlone(kTRUE); 
35     trackCuts->SetRequireTPCRefit(kTRUE);
36     trackCuts->SetMinNClustersTPC(70);
37     trackCuts->SetEtaRange(-0.9,0.9);
38     trackCuts->SetMaxCovDiagonalElements(2,2,0.5,0.5,2);
39     trackCuts->SetPtRange(0.15, 1e10);
40     trackCuts->SetMaxChi2PerClusterTPC(3.5);
41     trackCuts->SetMaxDCAToVertexXY(2.4);
42     trackCuts->SetMaxDCAToVertexZ(3.2);
43     trackCuts->SetDCAToVertex2D(kTRUE);
44   }
45   else if(cuts==2) {
46     trackCuts=trackCuts->GetStandardITSTPCTrackCuts2009(kTRUE);//Primary Track Selection
47     trackCuts->SetEtaRange(-0.9,0.9);
48     trackCuts->SetPtRange(0.15, 1e10);
49     trackCuts->SetRequireITSRefit(kFALSE);
50   }
51
52   AliESDtrackCuts *trackCutsITS = new AliESDtrackCuts("AliESDtrackCuts","Standard Cuts with ITSrefit");
53   if(cuts==1) {
54     trackCutsITS->SetAcceptKinkDaughters(kFALSE);
55     trackCutsITS->SetRequireTPCRefit(kTRUE);
56     trackCutsITS->SetEtaRange(-0.9,0.9);
57     trackCutsITS->SetMaxCovDiagonalElements(2,2,0.5,0.5,2);
58     trackCutsITS->SetPtRange(0.15, 1e10);
59     trackCutsITS->SetMinNClustersTPC(70);
60     trackCutsITS->SetMaxChi2PerClusterTPC(3.5);
61     trackCutsITS->SetRequireITSRefit(kTRUE);
62     trackCutsITS->SetMaxDCAToVertexXY(2.4);
63     trackCutsITS->SetMaxDCAToVertexZ(3.2);
64     trackCutsITS->SetDCAToVertex2D(kTRUE); 
65   }
66  else if(cuts==2) {
67    trackCutsITS=trackCutsITS->GetStandardITSTPCTrackCuts2009(kTRUE);//Primary Track Selection
68    trackCutsITS->SetEtaRange(-0.9,0.9);
69    trackCutsITS->SetPtRange(0.15, 1e10);
70  }
71
72 //Create the task
73   AliPWG4HighPtQATPConly *taskPWG4QA = new AliPWG4HighPtQATPConly(Form("AliPWG4HighPtQATPConly%d",cuts));
74   taskPWG4QA->SetCuts(trackCuts);
75   taskPWG4QA->SetCutsITS(trackCutsITS);
76   taskPWG4QA->SetCutType(cuts);
77   if(!strcmp(prodType, "LHC10e14") || !strcmp(prodType, "PbPb")) taskPWG4QA->SetPtMax(500.);
78   else taskPWG4QA->SetPtMax(100.);
79
80   // E. Create ONLY the output containers for the data produced by the task.
81   // Get and connect other common input/output containers via the manager as below
82   //==============================================================================
83
84   //------ input data ------
85   TString outputfile = "";
86   outputfile = AliAnalysisManager::GetCommonFileName();
87   outputfile += Form(":PWG4_HighPtQATPConly%d",cuts);
88   
89   AliAnalysisDataContainer *cout_hist0;
90   AliAnalysisDataContainer *cout_hist1;
91   AliAnalysisDataContainer *cout_hist2;
92
93   AliAnalysisDataContainer *cout_cuts0;
94   AliAnalysisDataContainer *cout_cuts1;
95
96   cout_hist0 = mgr->CreateContainer(Form("qa_histsCuts%d",cuts), TList::Class(), AliAnalysisManager::kOutputContainer,outputfile);
97   cout_hist1 = mgr->CreateContainer(Form("qa_histsTPCCuts%d",cuts), TList::Class(), AliAnalysisManager::kOutputContainer,outputfile);
98   cout_hist2 = mgr->CreateContainer(Form("qa_histsITSCuts%d",cuts), TList::Class(), AliAnalysisManager::kOutputContainer,outputfile);  
99
100   cout_cuts0 = mgr->CreateContainer(Form("qa_trackCuts%d",cuts), AliESDtrackCuts::Class(), AliAnalysisManager::kParamContainer,outputfile);
101   cout_cuts1 = mgr->CreateContainer(Form("qa_trackCutsITS%d",cuts), AliESDtrackCuts::Class(), AliAnalysisManager::kParamContainer,outputfile);
102
103   //Add task to manager
104   mgr->AddTask(taskPWG4QA);
105
106   //Connect input containter to manager
107   mgr->ConnectInput(taskPWG4QA,0,mgr->GetCommonInputContainer());
108
109   //Connect output containers to manager
110   mgr->ConnectOutput(taskPWG4QA,0,cout_hist0);
111   mgr->ConnectOutput(taskPWG4QA,1,cout_hist1);
112   mgr->ConnectOutput(taskPWG4QA,2,cout_hist2);
113   mgr->ConnectOutput(taskPWG4QA,3,cout_cuts0);
114   mgr->ConnectOutput(taskPWG4QA,4,cout_cuts1);
115
116   // Return task pointer at the end
117   return taskPWG4QA;
118 }