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