f1e7ff24 |
1 | #include "alles.h" |
2 | #include "/afs/cern.ch/user/s/skowron/aliroot/my/TPC/alles.h" |
3 | #include "AliHBTAnalysis.h" |
4 | #include "AliHBTReader.h" |
5 | #include "AliHBTReaderKineTree.h" |
6 | #include "AliHBTReaderITSv2.h" |
7 | #include "AliHBTReaderITSv1.h" |
8 | #include "AliHBTReaderTPC.h" |
9 | #include "AliHBTReaderInternal.h" |
10 | #include "AliHBTParticleCut.h" |
11 | #include "AliHBTEvent.h" |
12 | #include "AliHBTPairCut.h" |
13 | #include "AliHBTQResolutionFctns.h" |
14 | #include "AliHBTQDistributionFctns.h" |
15 | #include "AliHBTTwoTrackEffFctn.h" |
16 | #include "AliHBTCorrelFctn.h" |
17 | #include "TSystem.h" |
18 | #include "TObjString.h" |
19 | #include "TString.h" |
20 | #include "AliPDG.h" |
21 | #include "AliHBTMonDistributionFctns.h" |
22 | #include "AliHBTMonResolutionFctns.h" |
23 | |
e477958c |
24 | void hbtanalysis(Option_t* datatype, Option_t* processopt="TracksAndParticles", |
25 | Int_t first = -1,Int_t last = -1, |
26 | char *outfile = "hbtanalysis.root") |
0a6dde4c |
27 | { |
28 | //HBT Anlysis Macro |
29 | //Anlyzes TPC recontructed tracks and simulated particles that corresponds to them |
e477958c |
30 | |
31 | //datatype defines type of data to be read |
32 | // Kine - analyzes Kine Tree: simulated particles |
33 | // TPC - analyzes TPC tracking + particles corresponding to these tracks |
34 | // ITSv1 - analyzes ITSv1 ----------------------//-------------------------- |
35 | // ITSv2 - analyzes ITSv2 ----------------------//-------------------------- |
36 | |
37 | //processopt defines option passed to AliHBTAnlysis::Process method |
38 | // default: TracksAndParticles - process both recontructed tracks and sim. particles corresponding to them |
39 | // Tracks - process only recontructed tracks |
40 | // Particles - process only simulated particles |
41 | |
0a6dde4c |
42 | //Reads data from diroctories from first to last(including) |
43 | // For examples if first=3 and last=5 it reads from |
44 | // ./3/ |
45 | // ./4/ |
46 | // ./5/ |
47 | //if first or last is negative (or both), it reads from current directory |
48 | // |
49 | //these names I use when analysis is done directly from CASTOR files via RFIO |
50 | // const char* basedir="rfio:/castor/cern.ch/user/s/skowron/"; |
51 | // const char* serie="standard"; |
52 | // const char* field = "0.4"; |
53 | const char* basedir="."; |
54 | const char* serie=""; |
55 | const char* field = ""; |
56 | |
f1e7ff24 |
57 | Bool_t threeDcuts = kFALSE; |
58 | |
0a6dde4c |
59 | // Dynamically link some shared libs |
f1e7ff24 |
60 | // gROOT->LoadMacro("loadlibs.C"); |
61 | // loadlibs(); |
0a6dde4c |
62 | |
71094efa |
63 | gSystem->Load("$(ALICE_ROOT)/lib/tgt_$(ALICE_TARGET)/libHBTAN"); |
0a6dde4c |
64 | |
65 | |
66 | /***********************************************************/ |
67 | //Create analysis and reader |
68 | AliHBTAnalysis * analysis = new AliHBTAnalysis(); |
69 | |
70 | AliHBTReader* reader; |
e477958c |
71 | Int_t kine = strcmp(datatype,"Kine"); |
72 | Int_t TPC = strcmp(datatype,"TPC"); |
73 | Int_t ITSv1 = strcmp(datatype,"ITSv1"); |
74 | Int_t ITSv2 = strcmp(datatype,"ITSv2"); |
91923a38 |
75 | Int_t intern = strcmp(datatype,"Intern"); |
76 | |
e477958c |
77 | if(!kine) |
78 | { |
79 | reader = new AliHBTReaderKineTree(); |
80 | processopt="Particles"; //this reader by definition reads only simulated particles |
81 | } |
82 | else if(!TPC) |
83 | { |
84 | reader = new AliHBTReaderTPC(); |
85 | } |
86 | else if(!ITSv1) |
87 | { |
88 | reader = new AliHBTReaderITSv1(); |
89 | } |
90 | else if(!ITSv2) |
91 | { |
92 | reader = new AliHBTReaderITSv2(); |
93 | } |
91923a38 |
94 | else if(!intern) |
95 | { |
f1e7ff24 |
96 | reader = new AliHBTReaderInternal("ITS.root"); |
91923a38 |
97 | } |
e477958c |
98 | else |
99 | { |
100 | cerr<<"Option "<<datatype<<" not recognized. Exiting"<<endl; |
101 | return; |
102 | } |
103 | |
0a6dde4c |
104 | TObjArray* dirs=0; |
e477958c |
105 | if ( (first >= 0) && (last>=0) && ( (last-first)>=0 ) ) |
106 | {//read from many dirs dirs |
107 | char buff[50]; |
108 | dirs = new TObjArray(last-first+1); |
109 | for (Int_t i = first; i<=last; i++) |
110 | { |
111 | sprintf(buff,"%s/%s/%s/%d",basedir,field,serie,i); |
112 | TObjString *odir= new TObjString(buff); |
113 | dirs->Add(odir); |
114 | } |
0a6dde4c |
115 | } |
e477958c |
116 | reader->SetDirs(dirs); |
0a6dde4c |
117 | |
118 | /***********************************************************/ |
f1e7ff24 |
119 | /***** R E A D E R ***************************************/ |
120 | /***********************************************************/ |
0a6dde4c |
121 | |
122 | //we want have only low pt pi+ so set a cut to reader |
123 | AliHBTParticleCut* readerpartcut= new AliHBTParticleCut(); |
e4e0ed45 |
124 | readerpartcut->SetPtRange(0.0,1.5); |
0a6dde4c |
125 | readerpartcut->SetPID(kPiPlus); |
126 | reader->AddParticleCut(readerpartcut);//read this particle type with this cut |
127 | |
128 | analysis->SetReader(reader); |
f1e7ff24 |
129 | |
130 | /************************************************************/ |
131 | /**** Q INV Correlation Function ************************/ |
0a6dde4c |
132 | /************************************************************/ |
133 | |
d4ebbc5f |
134 | AliHBTPairCut *paircut = new AliHBTPairCut(); |
f1e7ff24 |
135 | Float_t qinvmin = 0.0; |
136 | Float_t qinvmax = 0.05;//50MeV |
137 | paircut->SetQInvRange(qinvmin,qinvmax); |
d4ebbc5f |
138 | analysis->SetGlobalPairCut(paircut); |
0a6dde4c |
139 | |
f1e7ff24 |
140 | AliHBTQInvCorrelFctn * qinvcfT = new AliHBTQInvCorrelFctn(100,qinvmax); |
141 | AliHBTQInvCorrelFctn * qinvcfP = new AliHBTQInvCorrelFctn(100,qinvmax); |
0a6dde4c |
142 | |
143 | analysis->AddTrackFunction(qinvcfT); |
144 | analysis->AddParticleFunction(qinvcfP); |
145 | |
0a6dde4c |
146 | |
147 | qinvcfP->Rename("qinvcfP","Particle (simulated) Qinv CF \\pi^{+} \\pi^{+}"); |
148 | qinvcfT->Rename("qinvcfT","Track (recontructed) Qinv CF \\pi^{+} \\pi^{+}"); |
91923a38 |
149 | |
f1e7ff24 |
150 | /************************************************************/ |
151 | /**** Q OUT Correlation Function ************************/ |
152 | /************************************************************/ |
153 | |
91923a38 |
154 | AliHBTQOutCMSLCCorrelFctn* qoutP = new AliHBTQOutCMSLCCorrelFctn(); |
155 | qoutP->Rename("qoutP","Particle (simulated) Q_{out} CF \\pi^{+} \\pi^{+}"); |
156 | AliHBTQOutCMSLCCorrelFctn* qoutT = new AliHBTQOutCMSLCCorrelFctn(); |
157 | qoutT->Rename("qoutT","Track (recontructed) Q_{out} CF \\pi^{+} \\pi^{+}"); |
158 | |
159 | AliHBTPairCut *outPairCut = new AliHBTPairCut(); |
160 | outPairCut->SetQOutCMSLRange(0.0,0.15); |
161 | outPairCut->SetQSideCMSLRange(0.0,0.02); |
162 | outPairCut->SetQLongCMSLRange(0.0,0.02); |
163 | qoutP->SetPairCut(outPairCut); |
164 | qoutT->SetPairCut(outPairCut); |
f1e7ff24 |
165 | |
166 | analysis->AddTrackFunction(qoutP); |
167 | analysis->AddParticleFunction(qoutT); |
168 | |
169 | /************************************************************/ |
170 | /**** Q SIDE Correlation Function ***********************/ |
171 | /************************************************************/ |
91923a38 |
172 | |
f1e7ff24 |
173 | AliHBTQSideCMSLCCorrelFctn* qsideP = new AliHBTQSideCMSLCCorrelFctn(100,qinvmax); |
91923a38 |
174 | qsideP->Rename("qsideP","Particle (simulated) Q_{side} CF \\pi^{+} \\pi^{+}"); |
f1e7ff24 |
175 | AliHBTQSideCMSLCCorrelFctn* qsideT = new AliHBTQSideCMSLCCorrelFctn(100,qinvmax); |
91923a38 |
176 | qsideT->Rename("qsideT","Track (recontructed) Q_{side} CF \\pi^{+} \\pi^{+}"); |
177 | |
178 | AliHBTPairCut *sidePairCut = new AliHBTPairCut(); |
179 | sidePairCut->SetQOutCMSLRange(0.0,0.02); |
180 | sidePairCut->SetQSideCMSLRange(0.0,0.15); |
181 | sidePairCut->SetQLongCMSLRange(0.0,0.02); |
182 | qsideP->SetPairCut(sidePairCut); |
183 | qsideT->SetPairCut(sidePairCut); |
0a6dde4c |
184 | |
f1e7ff24 |
185 | analysis->AddTrackFunction(qsideP); |
186 | analysis->AddParticleFunction(qsideT); |
187 | |
188 | /************************************************************/ |
189 | /**** Q LONG Correlation Function ***********************/ |
190 | /************************************************************/ |
91923a38 |
191 | |
f1e7ff24 |
192 | AliHBTQLongCMSLCCorrelFctn* qlongP = new AliHBTQLongCMSLCCorrelFctn(100,qinvmax); |
91923a38 |
193 | qlongP->Rename("qlongP","Particle (simulated) Q_{long} CF \\pi^{+} \\pi^{+}"); |
f1e7ff24 |
194 | AliHBTQLongCMSLCCorrelFctn* qlongT = new AliHBTQLongCMSLCCorrelFctn(100,qinvmax); |
91923a38 |
195 | qlongT->Rename("qlongT","Track (recontructed) Q_{long} CF \\pi^{+} \\pi^{+}"); |
196 | |
197 | AliHBTPairCut *longPairCut = new AliHBTPairCut(); |
198 | longPairCut->SetQOutCMSLRange(0.0,0.02); |
199 | longPairCut->SetQSideCMSLRange(0.0,0.02); |
200 | longPairCut->SetQLongCMSLRange(0.0,0.15); |
201 | qlongP->SetPairCut(longPairCut); |
202 | qlongT->SetPairCut(longPairCut); |
203 | |
91923a38 |
204 | analysis->AddTrackFunction(qlongT); |
205 | analysis->AddParticleFunction(qlongT); |
f1e7ff24 |
206 | |
207 | |
208 | /************************************************************/ |
209 | /************************************************************/ |
210 | /************************************************************/ |
211 | /**** R E S O L U T I O N S ***********************/ |
212 | /************************************************************/ |
213 | /************************************************************/ |
214 | /************************************************************/ |
215 | |
216 | |
217 | |
218 | AliHBTQInvResolVsKtFctn qinvVsktF(200,1.0,0.0,300,0.015,-0.015); //QInvCMSLC Res Vs Kt |
219 | if (threeDcuts) qinvVsktF.SetPairCut(paircut); |
220 | analysis->AddResolutionFunction(&qinvVsktF); |
221 | |
222 | AliHBTQOutResolVsKtFctn qoutVsktF(200,1.0,0.0,300,0.05,-0.05); //QOutCMSLC Res Vs Kt |
223 | if (threeDcuts) qoutVsktF.SetPairCut(outPairCut); |
224 | analysis->AddResolutionFunction(&qoutVsktF); |
225 | |
226 | AliHBTQSideResolVsKtFctn qsideVsktF(200,1.0,0.0,300,0.015,-0.015); //QSideCMSLC Res Vs Kt |
227 | if (threeDcuts) qsideVsktF.SetPairCut(sidePairCut); |
228 | analysis->AddResolutionFunction(&qsideVsktF); |
229 | |
230 | AliHBTQLongResolVsKtFctn qlongVsktF(200,1.0,0.0,300,0.015,-0.015); //QLongCMSLC Res Vs Kt |
231 | if (threeDcuts) qlongVsktF.SetPairCut(longPairCut); |
232 | analysis->AddResolutionFunction(&qlongVsktF); |
233 | |
234 | AliHBTQInvResolVsQInvFctn qInvVsqinvF(200,qinvmax,qinvmin,300,0.015,-0.015); //QInvCMSLC Res Vs QInvCMSLC |
235 | analysis->AddResolutionFunction(&qInvVsqinvF); |
236 | |
237 | AliHBTQOutResolVsQInvFctn qoutVsqinvF(200,qinvmax,qinvmin,300,0.05,-0.05); //QOutCMSLC Res Vs QInvCMSLC |
238 | if (threeDcuts) qoutVsqinvF.SetPairCut(outPairCut); |
239 | analysis->AddResolutionFunction(&qoutVsqinvF); |
240 | |
241 | AliHBTQSideResolVsQInvFctn qsideVsqinvF(200,qinvmax,qinvmin,300,0.015,-0.015); //QSideCMSLC Res Vs QInvCMSLC |
242 | if (threeDcuts) qsideVsqinvF.SetPairCut(sidePairCut); |
243 | analysis->AddResolutionFunction(&qsideVsqinvF); |
244 | |
245 | AliHBTQLongResolVsQInvFctn qlongVsqinvF(200,qinvmax,qinvmin,300,0.015,-0.015); //QLongCMSLC Res Vs QInvCMSLC |
246 | if (threeDcuts) qlongVsqinvF.SetPairCut(longPairCut); |
247 | analysis->AddResolutionFunction(&qlongVsqinvF); |
248 | |
249 | |
250 | AliHBTPairThetaResolVsQInvFctn pairThetaVsqinv(200,qinvmax,qinvmin,300,0.015,-0.015); |
251 | if (threeDcuts) pairThetaVsqinv.SetPairCut(longPairCut); |
252 | analysis->AddResolutionFunction(&pairThetaVsqinv); |
253 | |
254 | AliHBTPairThetaResolVsKtFctn pairThetaVsKt(200,1.0,0.0,300,0.015,-0.015); |
255 | if (threeDcuts) pairThetaVsKt.SetPairCut(longPairCut); |
256 | analysis->AddResolutionFunction(&pairThetaVsKt); |
257 | |
258 | AliHBTPairCut phipc; |
259 | phipc.SetQLongCMSLRange(0.0,0.02); |
260 | |
261 | AliHBTPairPhiResolVsQInvFctn pairPhiVsqinv(200,qinvmax,qinvmin,300,0.015,-0.015); |
262 | if (threeDcuts) pairPhiVsqinv.SetPairCut(&phipc); |
263 | analysis->AddResolutionFunction(&pairPhiVsqinv); |
264 | |
265 | AliHBTPairPhiResolVsKtFctn pairPhiVsKt(200,1.0,0.0,300,0.015,-0.015); |
266 | if (threeDcuts) pairPhiVsKt.SetPairCut(&phipc); |
267 | analysis->AddResolutionFunction(&pairPhiVsKt); |
268 | |
269 | |
270 | AliHBTQOutResolVsQOutFctn qoutVsqoutF; //QOutCMSLC Res Vs QOut |
271 | if (threeDcuts) qoutVsqoutF.SetPairCut(outPairCut); |
272 | analysis->AddResolutionFunction(&qoutVsqoutF); |
273 | |
274 | AliHBTQSideResolVsQSideFctn qsideVsqsideF;//QSideCMSLC Res Vs QSide |
275 | if (threeDcuts) qsideVsqsideF.SetPairCut(sidePairCut); |
276 | analysis->AddResolutionFunction(&qsideVsqsideF); |
277 | |
91923a38 |
278 | |
f1e7ff24 |
279 | AliHBTQLongResolVsQLongFctn qlongVsqlongF;//QLongCMSLC Res Vs QLong |
280 | if (threeDcuts) qlongVsqlongF.SetPairCut(longPairCut); |
281 | analysis->AddResolutionFunction(&qlongVsqlongF); |
282 | |
283 | |
284 | /************************************************************/ |
285 | /************************************************************/ |
286 | /************************************************************/ |
287 | /**** D I S T R B U T I O N S ***********************/ |
288 | /************************************************************/ |
289 | /************************************************************/ |
290 | /************************************************************/ |
91923a38 |
291 | |
f1e7ff24 |
292 | /************************************************************/ |
293 | /**** OUT VS KT ***********************/ |
294 | /************************************************************/ |
295 | AliHBTQOutDistributionVsKtFctn qoutVsKtDistP; |
296 | qoutVsKtDistP.Rename("qoutVsKtDistP",qoutVsKtDistP.GetTitle()); |
297 | AliHBTQOutDistributionVsKtFctn qoutVsKtDistT; |
298 | qoutVsKtDistT.Rename("qoutVsKtDistT",qoutVsKtDistT.GetTitle()); |
299 | if (threeDcuts) qoutVsKtDistP.SetPairCut(outPairCut); |
300 | if (threeDcuts) qoutVsKtDistT.SetPairCut(outPairCut); |
301 | analysis->AddParticleFunction(&qoutVsKtDistP); |
302 | analysis->AddTrackFunction(&qoutVsKtDistT); |
303 | /************************************************************/ |
304 | /**** Side VS KT ***********************/ |
305 | /************************************************************/ |
306 | AliHBTQSideDistributionVsKtFctn qSideVsKtDistP; |
307 | qSideVsKtDistP.Rename("qSideVsKtDistP",qSideVsKtDistP.GetTitle()); |
308 | AliHBTQSideDistributionVsKtFctn qSideVsKtDistT; |
309 | qSideVsKtDistT.Rename("qSideVsKtDistT",qSideVsKtDistT.GetTitle()); |
310 | if (threeDcuts) qSideVsKtDistP.SetPairCut(sidePairCut); |
311 | if (threeDcuts) qSideVsKtDistT.SetPairCut(sidePairCut); |
312 | analysis->AddParticleFunction(&qSideVsKtDistP); |
313 | analysis->AddTrackFunction(&qSideVsKtDistT); |
314 | /************************************************************/ |
315 | /**** Long VS KT ***********************/ |
316 | /************************************************************/ |
317 | AliHBTQLongDistributionVsKtFctn qLongVsKtDistP; |
318 | qLongVsKtDistP.Rename("qLongVsKtDistP",qLongVsKtDistP.GetTitle()); |
319 | AliHBTQLongDistributionVsKtFctn qLongVsKtDistT; |
320 | qLongVsKtDistT.Rename("qLongVsKtDistT",qLongVsKtDistT.GetTitle()); |
321 | if (threeDcuts) qLongVsKtDistP.SetPairCut(longPairCut); |
322 | if (threeDcuts) qLongVsKtDistT.SetPairCut(longPairCut); |
323 | analysis->AddParticleFunction(&qLongVsKtDistP); |
324 | analysis->AddTrackFunction(&qLongVsKtDistT); |
325 | |
326 | /************************************************************/ |
327 | /************************************************************/ |
328 | /************************************************************/ |
329 | /**** M O N I T O R R E S O L U T I O N S *************/ |
330 | /************************************************************/ |
331 | /************************************************************/ |
332 | /************************************************************/ |
333 | |
334 | AliHBTMonPxResolutionVsPtFctn PxVsPtResF(200,1.0,0.0,300,0.05,-0.05); |
335 | analysis->AddParticleAndTrackMonitorFunction(&PxVsPtResF); |
336 | |
337 | AliHBTMonPyResolutionVsPtFctn PyVsPtResF(200,1.0,0.0,300,0.05,-0.05); |
338 | analysis->AddParticleAndTrackMonitorFunction(&PyVsPtResF); |
339 | |
340 | AliHBTMonPzResolutionVsPtFctn PzVsPtResF(200,1.0,0.0,300,0.05,-0.05); |
341 | analysis->AddParticleAndTrackMonitorFunction(&PzVsPtResF); |
342 | |
343 | AliHBTMonPResolutionVsPtFctn PVsPtResF(200,1.0,0.0,300,0.05,-0.05); |
344 | analysis->AddParticleAndTrackMonitorFunction(&PVsPtResF); |
345 | |
346 | AliHBTMonPtResolutionVsPtFctn PtVsPtResF(200,1.0,0.0,300,0.05,-0.05); |
347 | analysis->AddParticleAndTrackMonitorFunction(&PtVsPtResF); |
348 | |
349 | AliHBTMonPhiResolutionVsPtFctn PhiVsPtResF(200,1.0,0.0,300,0.02,-0.02); |
350 | analysis->AddParticleAndTrackMonitorFunction(&PhiVsPtResF); |
351 | |
352 | AliHBTMonThetaResolutionVsPtFctn ThetaVsPtResF(200,1.0,0.0,300,0.02,-0.02); |
353 | analysis->AddParticleAndTrackMonitorFunction(&ThetaVsPtResF); |
354 | |
355 | |
356 | |
357 | |
358 | /************************************************************/ |
359 | /**** P R O C E S S ***********************/ |
360 | /************************************************************/ |
91923a38 |
361 | analysis->Process(processopt); |
362 | |
0a6dde4c |
363 | TFile histoOutput(outfile,"recreate"); |
364 | analysis->WriteFunctions(); |
365 | histoOutput.Close(); |
366 | |
367 | delete qinvcfP; |
368 | delete qinvcfT; |
d4ebbc5f |
369 | delete paircut; |
0a6dde4c |
370 | delete readerpartcut; |
371 | if (dirs) |
372 | { |
373 | dirs->SetOwner(); |
374 | delete dirs; |
375 | } |
376 | delete reader; |
377 | delete analysis; |
378 | |
379 | } |
380 | |