]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWGGA/GammaConv/macros/AddTask_GammaConvDalitzV1_pPb.C
ConversionCuts extended for Eventplane angle cut, thus all addTasks modified to have...
[u/mrichter/AliRoot.git] / PWGGA / GammaConv / macros / AddTask_GammaConvDalitzV1_pPb.C
CommitLineData
86eaf9ae 1void AddTask_GammaConvDalitzV1_pPb( Int_t trainConfig = 1,
2 Bool_t isMC = kFALSE, //run MC
39e8c2ce 3 Bool_t enableQAMesonTask = kTRUE, //enable QA in AliAnalysisTaskGammaConvDalitzV1
4 Bool_t enableDoMesonChic = kFALSE, // enable additional Chic analysis
5 TString cutnumberAODBranch = "0000000060084001001500000"
6 ) {
7
8
9
10 cout<<"Entro -1"<<endl;
11
12 // ================= Load Librariers =================================
13 gSystem->Load("libCore.so");
14 gSystem->Load("libTree.so");
15 gSystem->Load("libGeom.so");
16 gSystem->Load("libVMC.so");
17 gSystem->Load("libPhysics.so");
18 gSystem->Load("libMinuit");
19 gSystem->Load("libSTEERBase");
20 gSystem->Load("libESD");
21 gSystem->Load("libAOD");
22 gSystem->Load("libANALYSIS");
23 gSystem->Load("libANALYSISalice");
24 gSystem->Load("libPWGGAGammaConv.so");
25 gSystem->Load("libCDB.so");
26 gSystem->Load("libSTEER.so");
27 gSystem->Load("libSTEERBase.so");
28 gSystem->Load("libTENDER.so");
29 gSystem->Load("libTENDERSupplies.so");
30
31
32 cout<<"Entro 0"<<endl;
33
34 // ================== GetAnalysisManager ===============================
35 AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
36 if (!mgr) {
86eaf9ae 37 Error(Form("AddTask_GammaConvDalitzV1_pPb_%i",trainConfig), "No analysis manager found.");
39e8c2ce 38 return ;
39 }
40
41 // ================== GetInputEventHandler =============================
42 AliVEventHandler *inputHandler=mgr->GetInputEventHandler();
43
44 //========= Add PID Reponse to ANALYSIS manager ====
45 if(!(AliPIDResponse*)mgr->GetTask("PIDResponseTask")){
46 gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPIDResponse.C");
47 AddTaskPIDResponse(isMC);
48 }
49
50 //========= Set Cutnumber for V0Reader ================================
ae4f2cfb 51 TString ConvCutnumber = "8000000060084001001500000000"; //Online V0 finder
86eaf9ae 52 TString ElecCuts = "9000540000000200000"; //Electron Cuts
5a104c5e 53 Bool_t doEtaShift = kFALSE;
39e8c2ce 54
55
56
57 AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
58
59 //========= Add V0 Reader to ANALYSIS manager if not yet existent =====
60 if( !(AliV0ReaderV1*)mgr->GetTask("V0ReaderV1") ){
61 AliV0ReaderV1 *fV0ReaderV1 = new AliV0ReaderV1("V0ReaderV1");
62
63 fV0ReaderV1->SetUseOwnXYZCalculation(kTRUE);
64 fV0ReaderV1->SetCreateAODs(kFALSE);// AOD Output
65 fV0ReaderV1->SetUseAODConversionPhoton(kTRUE);
66
67 if (!mgr) {
68 Error("AddTask_V0ReaderV1", "No analysis manager found.");
69 return;
70 }
71
72 // Set AnalysisCut Number
73 AliConversionCuts *fCuts=NULL;
74 if( ConvCutnumber !=""){
75 fCuts= new AliConversionCuts(ConvCutnumber.Data(),ConvCutnumber.Data());
76 fCuts->SetPreSelectionCutFlag(kTRUE);
77 if(fCuts->InitializeCutsFromCutString(ConvCutnumber.Data())){
5a104c5e 78 fCuts->DoEtaShift(doEtaShift);
79 fV0ReaderV1->SetConversionCuts(fCuts);
39e8c2ce 80 fCuts->SetFillCutHistograms("",kTRUE);
81 }
82 }
83 if(inputHandler->IsA()==AliAODInputHandler::Class()){
84 // AOD mode
85 fV0ReaderV1->SetDeltaAODBranchName(Form("GammaConv_%s_gamma",cutnumberAODBranch.Data()));
86 }
87 fV0ReaderV1->Init();
88
89 AliLog::SetGlobalLogLevel(AliLog::kInfo);
90
91 //connect input V0Reader
92 mgr->AddTask(fV0ReaderV1);
93 mgr->ConnectInput(fV0ReaderV1,0,cinput);
94
95 }
96
97 //================================================
98 //========= Add Electron Selector ================
99
100
101 if( !(AliDalitzElectronSelector*)mgr->GetTask("ElectronSelector") ){
102
103 AliDalitzElectronSelector *fElectronSelector = new AliDalitzElectronSelector("ElectronSelector");
104
105 // Set AnalysisCut Number
106
107 AliDalitzElectronCuts *fElecCuts=0;
108
5a104c5e 109 //ElecCuts = "900054000000020000";
39e8c2ce 110
111 if( ElecCuts!=""){
112
113 fElecCuts= new AliDalitzElectronCuts(ElecCuts.Data(),ElecCuts.Data());
114
115 if(fElecCuts->InitializeCutsFromCutString(ElecCuts.Data())){
116
117 fElectronSelector->SetDalitzElectronCuts(fElecCuts);
118
119 fElecCuts->SetFillCutHistograms("",kTRUE);
120
121 }
122
123 }
124
125 fElectronSelector->Init();
126 mgr->AddTask(fElectronSelector);
5a104c5e 127
128 AliAnalysisDataContainer *cinput1 = mgr->GetCommonInputContainer();
129
130 //connect input V0Reader
131
132 mgr->ConnectInput (fElectronSelector,0,cinput1);
39e8c2ce 133
134 }
135
136
137
138 cout<<"Entro"<<endl;
139 //================================================
140 //========= Add task to the ANALYSIS manager =====
141 //================================================
142 // find input container
143
144
145
146 AliAnalysisTaskGammaConvDalitzV1 *task=NULL;
147
86eaf9ae 148 task= new AliAnalysisTaskGammaConvDalitzV1(Form("GammaConvDalitzV1_%i",trainConfig));
39e8c2ce 149
150 task->SetIsHeavyIon(2);
151 task->SetIsMC(isMC);
152
153
154
155 // Cut Numbers to use in Analysis
86eaf9ae 156 Int_t numberOfCuts = 6;
39e8c2ce 157
158 TString *ConvCutarray = new TString[numberOfCuts];
159
160 TString *ElecCutarray = new TString[numberOfCuts];
161
162 TString *MesonCutarray = new TString[numberOfCuts];
163
164 Bool_t doEtaShiftIndCuts = kFALSE;
39e8c2ce 165 TString stringShift = "";
166
167 // Shifting in pPb direction
168
169 doEtaShiftIndCuts = kTRUE;
170 stringShift = "pPb";
171
86eaf9ae 172
173 if( trainConfig == 1 ) {
174
ae4f2cfb 175 ConvCutarray[0] = "8000000082093603007200000000"; ElecCutarray[0] = "9047540025810262170"; MesonCutarray[0] = "01039035009000"; //standard cut Pi0 PbPb 00-100
176 ConvCutarray[1] = "8000000082093603007200000000"; ElecCutarray[1] = "9047540025810261170"; MesonCutarray[1] = "01039035009000"; //standard cut Pi0 PbPb 00-100 + Single Pt primary > 0.100 GeV
177 ConvCutarray[2] = "8000000082094603007200000000"; ElecCutarray[2] = "9047540025810262170"; MesonCutarray[2] = "01039035009000"; //standard cut Pi0 PbPb 00-100 + dEdx electron gamma -6 ,7 sigmas
178 ConvCutarray[3] = "8000000082093603007203000000"; ElecCutarray[3] = "9047540025810262170"; MesonCutarray[3] = "01039035009000"; //standard cut Pi0 PbPb 00-100 do Aysemtri cut
179 ConvCutarray[4] = "8000000082093603007200000000"; ElecCutarray[4] = "9051540025810262170"; MesonCutarray[4] = "01039035009000"; //standard cut Pi0 PbPb 00-100
180 ConvCutarray[5] = "8000000082093603007200000000"; ElecCutarray[5] = "9051540025810262170"; MesonCutarray[5] = "01039035009000"; //standard cut Pi0 PbPb 00-100 Standard cut + dEdx primary -3, 5 and 3.0 , -10 pion rejection
86eaf9ae 181
182 } else if( trainConfig == 2 ) {
183
ae4f2cfb 184 ConvCutarray[0] = "8000000082093603007200000000"; ElecCutarray[0] = "9047540025810262170"; MesonCutarray[0] = "01039035009000"; //standard cut Pi0 PbPb 00-100
185 ConvCutarray[1] = "8020000082093603007200000000"; ElecCutarray[1] = "9047540025810262170"; MesonCutarray[1] = "01039035009000"; //standard cut Pi0 PbPb 00-20
186 ConvCutarray[2] = "8240000082093603007200000000"; ElecCutarray[2] = "9047540025810262170"; MesonCutarray[2] = "01039035009000"; //standard cut Pi0 PbPb 20-40
187 ConvCutarray[3] = "8460000082093603007200000000"; ElecCutarray[3] = "9047540025810262170"; MesonCutarray[3] = "01039035009000"; //standard cut Pi0 PbPb 40-60
188 ConvCutarray[4] = "8680000082093603007200000000"; ElecCutarray[4] = "9047540025810262170"; MesonCutarray[4] = "01039035009000"; //standard cut Pi0 PbPb 60-80
189 ConvCutarray[5] = "8600000082093603007200000000"; ElecCutarray[5] = "9047540025810262170"; MesonCutarray[5] = "01039035009000"; //standard cut Pi0 PbPb 60-100
86eaf9ae 190 }
191
192
39e8c2ce 193
194 TList *ConvCutList = new TList();
195 TList *MesonCutList = new TList();
196 TList *ElecCutList = new TList();
197
198 TList *HeaderList = new TList();
199 TObjString *Header1 = new TObjString("pi0_1");
200 HeaderList->Add(Header1);
5a104c5e 201 TObjString *Header3 = new TObjString("eta_2");
202 HeaderList->Add(Header3);
39e8c2ce 203
204 ConvCutList->SetOwner(kTRUE);
205 AliConversionCuts **analysisCuts = new AliConversionCuts*[numberOfCuts];
206 MesonCutList->SetOwner(kTRUE);
207 AliConversionMesonCuts **analysisMesonCuts = new AliConversionMesonCuts*[numberOfCuts];
208 ElecCutList->SetOwner(kTRUE);
209 AliDalitzElectronCuts **analysisElecCuts = new AliDalitzElectronCuts*[numberOfCuts];
210
211
212
213 for(Int_t i = 0; i<numberOfCuts; i++){
214
215
216 analysisCuts[i] = new AliConversionCuts();
217 if( ! analysisCuts[i]->InitializeCutsFromCutString(ConvCutarray[i].Data()) ) {
218 cout<<"ERROR: analysisCuts [" <<i<<"]"<<endl;
219 return 0;
220 }
221 else {
222
5a104c5e 223 if (doEtaShiftIndCuts) {
224 analysisCuts[i]->DoEtaShift(doEtaShiftIndCuts);
225 analysisCuts[i]->SetEtaShift(stringShift);
226 }
227 ConvCutList->Add(analysisCuts[i]);
228 analysisCuts[i]->SetFillCutHistograms("",kFALSE);
229 analysisCuts[i]->SetAcceptedHeader(HeaderList);
39e8c2ce 230 }
231
232
233
234 analysisMesonCuts[i] = new AliConversionMesonCuts();
235
236 if( ! analysisMesonCuts[i]->InitializeCutsFromCutString(MesonCutarray[i].Data()) ) {
237 cout<<"ERROR: analysisMesonCuts [ " <<i<<" ] "<<endl;
238 return 0;
239 }
240 else {
5a104c5e 241 MesonCutList->Add(analysisMesonCuts[i]);
242 analysisMesonCuts[i]->SetFillCutHistograms("");
39e8c2ce 243 }
244
245
246 TString cutName( Form("%s_%s_%s",ConvCutarray[i].Data(),ElecCutarray[i].Data(),MesonCutarray[i].Data() ) );
247
248
249 analysisElecCuts[i] = new AliDalitzElectronCuts();
250 if( !analysisElecCuts[i]->InitializeCutsFromCutString(ElecCutarray[i].Data())) {
251
252 cout<< "ERROR: analysisElecCuts [ " <<i<<" ] "<<endl;
253 return 0;
254 }
255 else {
256 ElecCutList->Add(analysisElecCuts[i]);
257 analysisElecCuts[i]->SetFillCutHistograms("",kFALSE,cutName);
258 }
259
260
5a104c5e 261
39e8c2ce 262
263 }
264
265
266 task->SetConversionCutList(numberOfCuts,ConvCutList);
267 task->SetMesonCutList(MesonCutList);
268 task->SetElectronCutList(ElecCutList);
269
270 task->SetMoveParticleAccordingToVertex(kTRUE);
271
272
273 if(enableQAMesonTask) task->SetDoMesonQA(kTRUE);
5a104c5e 274 if(enableDoMesonChic) task->SetDoChicAnalysis(kTRUE);
39e8c2ce 275
276 //connect containers
277 AliAnalysisDataContainer *coutput =
86eaf9ae 278 mgr->CreateContainer(Form("GammaConvDalitzV1_%i",trainConfig), TList::Class(),
279 AliAnalysisManager::kOutputContainer,Form("GammaConvV1Dalitz_%i.root",trainConfig));
39e8c2ce 280
281 mgr->AddTask(task);
282 mgr->ConnectInput(task,0,cinput);
283 mgr->ConnectOutput(task,1,coutput);
284
285 return;
286
287}