]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWGGA/GammaConv/macros/AddTask_Pi0v2.C
fixed bugs while streaming histos for weighting, added new trainconfig for pPb
[u/mrichter/AliRoot.git] / PWGGA / GammaConv / macros / AddTask_Pi0v2.C
CommitLineData
70f90d01 1#include <exception>
2
3class AliAnalysisTaskPi0v2;
4class AliV0ReaderV1;
5
6// Settings
7Int_t nBinsPhi=6;
8Int_t epselectionmask[4]={1,1,1,1};// TPC,TPCEtaGap,V0A,V0C
9const Int_t nCentralityBins=5;
10Double_t fCentralityBins[nCentralityBins+1]={0,5,10,20,30,40};
11Double_t fInvMassRange[2]={0.0,0.3};
12//const Int_t fNRadialBins=9;
13//Float_t fRadialBins[fNRadialBins+1]={0,13,20,26,35,40,55,70,90,200};
14
15Bool_t fFillQA=kTRUE;
16
17//Bool_t fWeightMult=kFALSE; // cut number for mult =9
18
19AliV0ReaderV1 *fV0Reader=NULL;
20AliAnalysisManager *mgr=NULL;
21
22const Int_t numberOfCuts=24;
23TString cutarray[numberOfCuts];
24TString mesoncutarray[numberOfCuts];
25
26
27// Standard Cuts
28cutarray[0] = "1080000042092970023220000"; mesoncutarray[0] = "01522045000"; //standard cut Pi0 PbPb 00-100
29
30// TPC PID
31cutarray[1] = "1080001042093970023220000"; mesoncutarray[1] = "01522045000";
32cutarray[2] = "1080001042096970023220000"; mesoncutarray[2] = "01522045000";
33cutarray[3] = "1080001042092470023220000"; mesoncutarray[3] = "01522045000";
34cutarray[4] = "1080001042092770023220000"; mesoncutarray[4] = "01522045000";
35cutarray[5] = "1080001042092950023220000"; mesoncutarray[5] = "01522045000";
36
37// TOF PID
38cutarray[6] = "1080001042092970033220000"; mesoncutarray[6] = "01522045000";
39cutarray[7] = "1080001042092970043220000"; mesoncutarray[7] = "01522045000";
40
41// Qt max
42cutarray[8] = "1080001042092970024220000"; mesoncutarray[8] = "01522045000";
43cutarray[9] = "1080001042092970022220000"; mesoncutarray[9] = "01522045000";
44
45// Chi2 Gamma
46cutarray[10] = "1080001042092970023120000"; mesoncutarray[10] = "01522045000";
47cutarray[11] = "1080001042092970023820000"; mesoncutarray[11] = "01522045000";
48 // Psi Pair
49cutarray[12] = "1080001042092970023210000"; mesoncutarray[12] = "01522045000";
50cutarray[13] = "1080001042092970023230000"; mesoncutarray[13] = "01522045000";
51
52// R Cut
53cutarray[14] = "1080001044092970023220000"; mesoncutarray[14] = "01522045000"; //5-70
54cutarray[15] = "1080001045092970023220000"; mesoncutarray[15] = "01522045000"; //10-180
55cutarray[16] = "1080001046092970023220000"; mesoncutarray[16] = "01522045000"; //20
56cutarray[17] = "1080001047092970023220000"; mesoncutarray[17] = "01522045000"; //26
57cutarray[18] = "1080001048092970023220000"; mesoncutarray[18] = "01522045000"; //35
58cutarray[19] = "1080001045092970023220000"; mesoncutarray[19] = "01522045000"; //60
59
60// Single Pt
61cutarray[20] = "1080001042492970023220000"; mesoncutarray[20] = "01522045000";
62cutarray[21] = "1080001042192970023220000"; mesoncutarray[21] = "01522045000";
63
64// Alpha
65cutarray[22] = "1080001042092970023220000"; mesoncutarray[22] = "01022085000";
66cutarray[23] = "1080001042092970023220000"; mesoncutarray[23] = "01022005000";
67
85f6a755 68AliAnalysisTask *AddTask_Pi0v2(Int_t harmonic=2,Bool_t IsHeavyIon=kTRUE,Bool_t doSys=kTRUE){
70f90d01 69
70 // standard with task
71 printf("========================================================================================\n");
72 printf("Pi0v2Analysis: Initialising AliAnalysisTaskPi0v2\n");
73 printf("========================================================================================\n");
74
75 //get the current analysis manager
76
77 mgr = AliAnalysisManager::GetAnalysisManager();
78 if (!mgr) {
79 Error("AddTask_dlohner_Pi0v2", "No analysis manager found.");
80 return 0;
81 }
82
83 Bool_t isMC=kFALSE;
84 AliMCEventHandler *mcH = dynamic_cast<AliMCEventHandler*>(mgr->GetMCtruthEventHandler());
85 if(mcH){
86 isMC=kTRUE;
87 }
88
89 // For 2011 data
90 /*
91 gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskVZEROEPSelection.C");
92 AddTaskVZEROEPSelection();
93 gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskEventplane.C");
94 AddTaskEventplane();
95 */
96
97 TString fV0ReaderCut="";
98
99 if(IsHeavyIon){
100 fV0ReaderCut = "1080000002084001001500000";
101 }
102 else{
103 fV0ReaderCut = "0000000002084001001500000";
104 }
85f6a755 105 fV0Reader=new AliV0ReaderV1(Form("PhotonPi0v%d",harmonic));
70f90d01 106 mgr->AddTask(fV0Reader);
107 ConfigV0Reader(fV0Reader,fV0ReaderCut.Data(),IsHeavyIon);
108 mgr->ConnectInput(fV0Reader, 0, mgr->GetCommonInputContainer());
109 fV0Reader->GetConversionCuts()->SetFillCutHistograms("V0ReaderCuts");
110
111 // Setup Task
112
113 //========= Add task to the ANALYSIS manager =====
114
85f6a755 115 AliAnalysisTaskPi0v2 *task = new AliAnalysisTaskPi0v2(Form("dlohnerTask_Pi0v%d",harmonic),harmonic);
116 if(doSys)SetupPi0v2(task,IsHeavyIon,isMC,Form("dlohner_Pi0v%d",harmonic),numberOfCuts);
117 else SetupPi0v2(task,IsHeavyIon,isMC,Form("dlohner_Pi0v%d",harmonic),1);
70f90d01 118
119 return task;
120}
121
122void ConfigV0Reader(AliV0ReaderV1 *fV0Reader,TString analysiscut="",Bool_t IsHeavyIon=kTRUE){
123
124 fV0Reader->SetUseOwnXYZCalculation(kTRUE);
125
126 // Set AnalysisCut Number
127 AliConversionCuts *fCuts=NULL;
128 if(analysiscut!=""){
129 fCuts= new AliConversionCuts(analysiscut.Data(),analysiscut.Data());
130 if(fCuts->InitializeCutsFromCutString(analysiscut.Data())){
131 fV0Reader->SetConversionCuts(fCuts);
132 }
133 }
134 else{
135 // Init standard cuts
136 if(IsHeavyIon){fCuts=AliConversionCuts::GetStandardCuts2010PbPb();}
137 else{fCuts=AliConversionCuts::GetStandardCuts2010pp();}
138 fV0Reader->SetConversionCuts(fCuts);
139 }
140 // Initialize
141 fV0Reader->Init();
142}
143
144
145void SetupPi0v2(AliAnalysisTaskPi0v2 *task,Bool_t IsHeavyIon,Bool_t IsMC=kFALSE,TString outputname,Int_t ncuts=0){
146
147 cout<<"Settings for Task : "<<outputname.Data()<<endl;
148
149 task->SetV0Reader(fV0Reader);
150 task->SetInvMassRange(fInvMassRange);
151 task->SetNBinsPhi(nBinsPhi);
152 task->SetFillQA(fFillQA);
153 //task->SetEPSelectionMask(epselectionmask);
154
155 if(IsHeavyIon){
156 task->SetCentralityBins(fCentralityBins,nCentralityBins);
157 // task->SetWeightMultiplicity(fWeightMult);
158
159 // Set Cuts
160 if(ncuts==0)ncuts=numberOfCuts;
161 if(IsMC)ncuts=1;
162
163 AliConversionSelection **selection=new AliConversionSelection*[ncuts];
164 for(Int_t ii=0;ii<ncuts;ii++){
165 cout<<"AddingCut: "<<cutarray[ii]<<" "<<mesoncutarray[ii]<<endl;
166 selection[ii]=new AliConversionSelection(cutarray[ii],mesoncutarray[ii]);
167 selection[ii]->SetInvMassRange(fInvMassRange);
168 }
169 task->SetCuts(selection,ncuts);
170 task->SetEtaGap(1);
171 }
172 else{
173 // no cuts defined
174 }
175
176 //SetFlattening(task,"LHC10h");
177
178 mgr->AddTask(task);
179
180 AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
181
182 AliAnalysisDataContainer *coutput0 =
183 mgr->CreateContainer("dlohner_tree",TTree::Class(),AliAnalysisManager::kExchangeContainer,"dlohner_default");
184
185 AliAnalysisDataContainer *coutput1 =
186 mgr->CreateContainer(outputname.Data(), TList::Class(),
187 AliAnalysisManager::kOutputContainer,Form("%s.root",outputname.Data()));
188 //connect containers
189 mgr->ConnectInput (task, 0, cinput );
190 // mgr->ConnectOutput (task, 0, coutput0);
191 mgr->ConnectOutput (task, 1, coutput1);
192}
193
194void SetFlattening(AliAnalysisTaskPi0v2 *task,TString period){
195
196 const Int_t nCent=5;
197
198 Int_t harmonic=task->GetHarmonic();
199
200 Int_t periodindex=task->GetPeriodIndex(period);
201
202 if(periodindex==0){
203 // TPC EP
204 if(harmonic==2){
205 Double_t cc2[nCent]={0.00904396,0.00472483,0.00306154,0.00218462,0.00167447};
206 Double_t cs2[nCent]={0.00885519,0.00516223,0.00411065,0.00380145,0.00324424};
207 Double_t cc4[nCent]={-0.00110933,-0.00110521,-0.00124342,0.00104131,0.000651779};
208 Double_t cs4[nCent]={0.00163869,-0.00053565,0.000878745,-0.000563657,-0.000604021};
209 task->SetFlatteningCoeff(AliAnalysisTaskPi0v2::kEPTPC,periodindex,nCent,&cc2[0],&cs2[0],&cc4[0],&cs4[0]);
210 }
211
212 if(harmonic==3){
213 Double_t cc2[5]={0.0116542,0.0103631,0.00897965,0.00707409,0.00605151};
214 Double_t cs2[5]={-0.0171191,-0.013024,-0.0114752,-0.0086613,-0.00706863};
215 Double_t cc4[5]={-0.000602948,0.00144836,-0.000193641,0.000108773,-0.000518333};
216 Double_t cs4[5]={-0.00164769,0.00134327,-0.00106369,7.96546e-06,-0.000261517};
217 task->SetFlatteningCoeff(AliAnalysisTaskPi0v2::kEPTPC,0,nCent,&cc2[0],&cs2[0],&cc4[0],&cs4[0]);
218 }
219
220 //TPC ETA A
221 if(harmonic==2){
222 Double_t cc2[5]={0.00529447,0.00278029,0.00315325,0.00173634,0.000763168};
223 Double_t cs2[5]={0.00314285,0.00170173,0.00263333,0.0018509,0.00223784};
224 Double_t cc4[5]={-0.000737254,-0.00037845,-0.000492715,0.000775897,0.000768656};
225 Double_t cs4[5]={0.000347583,3.79872e-05,0.000387037,-0.000186129,0.000432698};
226
227 task->SetFlatteningCoeff(AliAnalysisTaskPi0v2::kEPTPCEtaA,periodindex,nCent,&cc2[0],&cs2[0],&cc4[0],&cs4[0]);
228 }
229 if(harmonic==3){
230 Double_t cc2[5]={0.000386277,0.000119225,0.00111969,0.000534801,0.000642703};
231 Double_t cs2[5]={-0.00581604,-0.00607255,-0.00443819,-0.00268834,-0.00299961};
232 Double_t cc4[5]={0.00051635,0.00036326,-0.000221272,4.66775e-05,-3.05784e-06};
233 Double_t cs4[5]={1.43285e-05,0.000514099,0.000619339,0.00106466,0.000344196};
234 task->SetFlatteningCoeff(AliAnalysisTaskPi0v2::kEPTPCEtaA,periodindex,nCent,&cc2[0],&cs2[0],&cc4[0],&cs4[0]);
235 }
236
237 //TPC ETA C
238
239 if(harmonic==2){
240 Double_t cc2[5]={-0.00562282,-0.00456735,-0.00306068,-0.0027173,-0.00172432};
241 Double_t cs2[5]={0.0101804,0.00430782,0.00394715,0.00350156,0.00302749};
242 Double_t cc4[5]={0.00150831,-0.00159271,-0.000964157,0.000525894,9.93172e-05};
243 Double_t cs4[5]={0.00119279,-4.74629e-05,0.000118845,0.000278554,3.20868e-05};
244 task->SetFlatteningCoeff(AliAnalysisTaskPi0v2::kEPTPCEtaC,periodindex,nCent,&cc2[0],&cs2[0],&cc4[0],&cs4[0]);
245 }
246 if(harmonic==3){
247 Double_t cc2[5]={0.0116475,0.0102385,0.00801121,0.00552336,0.00423273};
248 Double_t cs2[5]={-0.0112722,-0.00796059,-0.00683678,-0.00531097,-0.00430716};
249 Double_t cc4[5]={-0.000609051,1.36573e-08,-0.000464961,-0.000387943,-2.28363e-05};
250 Double_t cs4[5]={0.00125449,0.00168484,-0.000390491,-0.000219447,8.11997e-07};
251 task->SetFlatteningCoeff(AliAnalysisTaskPi0v2::kEPTPCEtaC,periodindex,nCent,&cc2[0],&cs2[0],&cc4[0],&cs4[0]);
252 }
253
254 //V0A
255
256 if(harmonic==2){
257 Double_t cc2[5]={0.046427,0.0105401,-0.000152992,-0.00578274,-0.0108038};
258 Double_t cs2[5]={0.00551503,0.0158159,0.00965148,0.00135414,-0.00548846};
259 Double_t cc4[5]={0.00362833,0.00170777,0.000152998,0.00223823,0.00215164};
260 Double_t cs4[5]={0.00349056,0.00142802,0.00123298,0.00207995,0.00145625};
261
262 task->SetFlatteningCoeff(AliAnalysisTaskPi0v2::kEPV0A,periodindex,nCent,&cc2[0],&cs2[0],&cc4[0],&cs4[0]);
263 }
264 if(harmonic==3){
265 Double_t cc2[5]={-0.0057427,-0.00482728,-0.00565919,-0.000717094,-0.00933233};
266 Double_t cs2[5]={0.0306554,-0.0144675,-0.0159243,-0.0120465,-0.00814124};
267 Double_t cc4[5]={-0.002868,0.00159533,0.00754171,0.00683898,0.00689441};
268 Double_t cs4[5]={0.00083196,0.00198133,4.68307e-05,-0.00018187,-0.0014258};
269
270 task->SetFlatteningCoeff(AliAnalysisTaskPi0v2::kEPV0A,periodindex,nCent,&cc2[0],&cs2[0],&cc4[0],&cs4[0]);
271 }
272
273 // V0 C
274 if(harmonic==2){
275 Double_t cc2[5]={-0.00473277,-0.000371313,0.000857122,-1.54263e-05,-0.000686139};
276 Double_t cs2[5]={0.00408304,-0.00208615,-0.00149018,-0.000853616,-2.78855e-05};
277 Double_t cc4[5]={-0.00451741,-0.00399036,-0.00318784,-0.00186472,-0.00106299};
278 Double_t cs4[5]={0.00188045,-0.00713956,-0.00484254,-0.00448149,-0.00482164};
279 task->SetFlatteningCoeff(AliAnalysisTaskPi0v2::kEPV0C,periodindex,nCent,&cc2[0],&cs2[0],&cc4[0],&cs4[0]);
280 }
281 if(harmonic==3){
282 Double_t cc2[5]={-0.00259141,-0.00115826,-0.000738658,-4.96667e-05,-0.000346694};
283 Double_t cs2[5]={-0.0111001,0.00258109,0.00110959,-0.000147296,-0.000199817};
284 Double_t cc4[5]={0.000968742,0.00157903,0.000206157,0.000444206,-0.00046573};
285 Double_t cs4[5]={-0.00307319,-0.0047952,-0.00412117,-0.00320344,-0.00386629};
286
287 task->SetFlatteningCoeff(AliAnalysisTaskPi0v2::kEPV0C,periodindex,nCent,&cc2[0],&cs2[0],&cc4[0],&cs4[0]);
288 }
289 }
290
291}
292
293