]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWGCF/FEMTOSCOPY/macros/Train/pPbpionFemto/AOD/3D/pippim/ConfigFemtoAnalysis.C
Merge branch 'feature-movesplit'
[u/mrichter/AliRoot.git] / PWGCF / FEMTOSCOPY / macros / Train / pPbpionFemto / AOD / 3D / pippim / ConfigFemtoAnalysis.C
CommitLineData
acb5803e 1
2/*********************************************************************
3 * *
4 * ConfigFemtoAnalysis.C - configuration macro for the femtoscopic *
5 * analysis, meant as a QA process for two-particle effects *
6 * *
7 * Author: Adam Kisiel (Adam.Kisiel@cern.ch) *
8 * *
9 *********************************************************************/
10
11#if !defined(__CINT__) || defined(__MAKECINT_)
12#include "AliFemtoManager.h"
13#include "AliFemtoEventReaderESDChain.h"
14#include "AliFemtoEventReaderESDChainKine.h"
15#include "AliFemtoEventReaderAODChain.h"
16#include "AliFemtoSimpleAnalysis.h"
17#include "AliFemtoBasicEventCut.h"
18#include "AliFemtoESDTrackCut.h"
19#include "AliFemtoCorrFctn.h"
20#include "AliFemtoCutMonitorParticleYPt.h"
21#include "AliFemtoCutMonitorParticleVertPos.h"
22#include "AliFemtoCutMonitorParticleMomRes.h"
23#include "AliFemtoCutMonitorParticlePID.h"
24#include "AliFemtoCutMonitorEventMult.h"
25#include "AliFemtoCutMonitorEventVertex.h"
26#include "AliFemtoShareQualityTPCEntranceSepPairCut.h"
27#include "AliFemtoPairCutAntiGamma.h"
28#include "AliFemtoPairCutRadialDistance.h"
29#include "AliFemtoQinvCorrFctn.h"
30#include "AliFemtoShareQualityCorrFctn.h"
31#include "AliFemtoTPCInnerCorrFctn.h"
32#include "AliFemtoVertexMultAnalysis.h"
33#include "AliFemtoCorrFctn3DSpherical.h"
34#include "AliFemtoChi2CorrFctn.h"
35#include "AliFemtoCorrFctnTPCNcls.h"
36#include "AliFemtoBPLCMS3DCorrFctn.h"
37#include "AliFemtoCorrFctn3DLCMSSym.h"
38#include "AliFemtoModelBPLCMSCorrFctn.h"
39#include "AliFemtoModelCorrFctn3DSpherical.h"
40#include "AliFemtoModelGausLCMSFreezeOutGenerator.h"
41#include "AliFemtoModelGausRinvFreezeOutGenerator.h"
42#include "AliFemtoModelManager.h"
43#include "AliFemtoModelWeightGeneratorBasic.h"
44#include "AliFemtoModelWeightGeneratorLednicky.h"
45#include "AliFemtoCorrFctnDirectYlm.h"
46#include "AliFemtoModelCorrFctnDirectYlm.h"
47#include "AliFemtoModelCorrFctnSource.h"
48#include "AliFemtoCutMonitorParticlePtPDG.h"
49#include "AliFemtoKTPairCut.h"
50#include "AliFemtoCutMonitorCollections.h"
51#endif
52
53//________________________________________________________________________
54AliFemtoManager* ConfigFemtoAnalysis() {
55
56 double PionMass = 0.13956995;
57 double KaonMass = 0.493677;
58
59 //multiplicity bins
60 int runmults[10] = {1, 1, 1, 1, 0, 0, 0, 0, 0, 0};
61 int multbins[11] = {0.01, 200, 400, 600, 900, 950, 500, 600, 700, 800, 900};
62
63 int runch[3] = {0, 0, 1};
64 const char *chrgs[3] = { "pip", "pim", "pippim" };
65
66 int runktdep = 1;
67 double ktrng[8] = {0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 1.0};
68
69 int run3d = 1; // Do 3D cartesian analysis?
70 int runshlcms = 1;
71
72 //PhysicsSelection set only in runBatch with trigger + Physics Selection Task
73
74
75 double shqmax;
76 int nbinssh = 200;
77
78 if (runshlcms) shqmax = 2.0;
79 else shqmax = 0.9;
80
81 AliFemtoEventReaderAODChain *Reader = new AliFemtoEventReaderAODChain();
82 Reader->SetFilterBit(0);
83 //Reader->SetCentralityPreSelection(500, 950);
84
85 //AliFemtoEventReaderESDChainKine* Reader=new AliFemtoEventReaderESDChainKine();
86 //Reader->SetConstrained(true);
87 // Reader->SetUseTPCOnly(false);
88
89 //AliFemtoEventReaderESDChain *Reader = new AliFemtoEventReaderESDChain();
90 //Reader->SetUseMultiplicity(AliFemtoEventReaderESDChain::kV0Centrality);
91 //Reader->SetReadTrackType(AliFemtoEventReaderESDChain::kGlobal);
92
93
94 AliFemtoManager* Manager=new AliFemtoManager();
95 Manager->SetEventReader(Reader);
96
97 AliFemtoVertexMultAnalysis *anetaphitpc[320];
98 AliFemtoBasicEventCut *mecetaphitpc[320];
99 AliFemtoCutMonitorEventMult *cutPassEvMetaphitpc[320];
100 AliFemtoCutMonitorEventMult *cutFailEvMetaphitpc[320];
101 AliFemtoCutMonitorEventVertex *cutPassEvVetaphitpc[320];
102 AliFemtoCutMonitorEventVertex *cutFailEvVetaphitpc[320];
103 AliFemtoCutMonitorCollections *cutPassColletaphitpc[320];
104 AliFemtoCutMonitorCollections *cutFailColletaphitpc[320];
105 AliFemtoESDTrackCut *dtc1etaphitpc[320];
106 AliFemtoESDTrackCut *dtc2etaphitpc[320];
107 AliFemtoCutMonitorParticleYPt *cutPass1YPtetaphitpc[320];
108 AliFemtoCutMonitorParticleYPt *cutFail1YPtetaphitpc[320];
109 AliFemtoCutMonitorParticlePID *cutPass1PIDetaphitpc[320];
110 AliFemtoCutMonitorParticlePID *cutFail1PIDetaphitpc[320];
111 AliFemtoCutMonitorParticleYPt *cutPass2YPtetaphitpc[320];
112 AliFemtoCutMonitorParticleYPt *cutFail2YPtetaphitpc[320];
113 AliFemtoCutMonitorParticlePID *cutPass2PIDetaphitpc[320];
114 AliFemtoCutMonitorParticlePID *cutFail2PIDetaphitpc[320];
115 // AliFemtoPairCutAntiGamma *sqpcetaphitpc[320];
116 AliFemtoShareQualityTPCEntranceSepPairCut *sqpcetaphitpc[320];
117 //AliFemtoPairCutRadialDistance *sqpcetaphitpc[320];
118 AliFemtoCorrFctnDirectYlm *cylmetaphitpc[320];
119 AliFemtoCorrFctnDEtaDPhi *cdedpetaphi[320];
120 AliFemtoChi2CorrFctn *cchiqinvetaphitpc[320];
121 AliFemtoKTPairCut *ktpcuts[320*7];
122 AliFemtoCorrFctnDirectYlm *cylmkttpc[320*7];
123 AliFemtoQinvCorrFctn *cqinvkttpc[320*7];
124 AliFemtoCorrFctn3DLCMSSym *cq3dlcmskttpc[320*7];
125 AliFemtoCorrFctnTPCNcls *cqinvnclstpc[320];
126 AliFemtoShareQualityCorrFctn *cqinvsqtpc[320*10];
127 AliFemtoChi2CorrFctn *cqinvchi2tpc[320];
128 AliFemtoTPCInnerCorrFctn *cqinvinnertpc[320*10];
129
130 // *** Begin pion-pion analysis ***
131 int aniter = 0;
132
133 for (int imult=0; imult<10; imult++) {
134 if (runmults[imult]) {
135 for (int ichg=0; ichg<3; ichg++) {
136 if (runch[ichg]) {
137 aniter = ichg*10+imult;
138
139 anetaphitpc[aniter] = new AliFemtoVertexMultAnalysis(10, -10.0, 10.0, 4, multbins[imult], multbins[imult+1]);
140 anetaphitpc[aniter]->SetNumEventsToMix(5);
141 anetaphitpc[aniter]->SetMinSizePartCollection(1);
142
143 mecetaphitpc[aniter] = new AliFemtoBasicEventCut();
144 mecetaphitpc[aniter]->SetEventMult(0,10000);
145 mecetaphitpc[aniter]->SetVertZPos(-10,10);
146
147 cutPassEvMetaphitpc[aniter] = new AliFemtoCutMonitorEventMult(Form("cutPass%stpcM%i", chrgs[ichg], imult),500);
148 cutFailEvMetaphitpc[aniter] = new AliFemtoCutMonitorEventMult(Form("cutFail%stpcM%i", chrgs[ichg], imult),500);
149 mecetaphitpc[aniter]->AddCutMonitor(cutPassEvMetaphitpc[aniter], cutFailEvMetaphitpc[aniter]);
150
151 cutPassEvVetaphitpc[aniter] = new AliFemtoCutMonitorEventVertex(Form("cutPass%stpcM%i", chrgs[ichg], imult));
152 cutFailEvVetaphitpc[aniter] = new AliFemtoCutMonitorEventVertex(Form("cutFail%stpcM%i", chrgs[ichg], imult));
153 mecetaphitpc[aniter]->AddCutMonitor(cutPassEvVetaphitpc[aniter], cutFailEvVetaphitpc[aniter]);
154
155 cutPassColletaphitpc[aniter] = new AliFemtoCutMonitorCollections(Form("cutPass%stpcM%i", chrgs[ichg], imult));
156 cutFailColletaphitpc[aniter] = new AliFemtoCutMonitorCollections(Form("cutFail%stpcM%i", chrgs[ichg], imult));
157 mecetaphitpc[aniter]->AddCutMonitor(cutPassColletaphitpc[aniter], cutFailColletaphitpc[aniter]);
158
159 dtc1etaphitpc[aniter] = new AliFemtoESDTrackCut();
160
161 dtc1etaphitpc[aniter]->SetCharge(1.0);
162
163 dtc1etaphitpc[aniter]->SetPt(0.12,4.0);
164 dtc1etaphitpc[aniter]->SetEta(-1.2,1.2);
165
166 //PID method
167 dtc1etaphitpc[aniter]->SetMass(PionMass);
168 dtc1etaphitpc[aniter]->SetMostProbablePion();
169 //dtc1etaphitpc[aniter]->SetPIDMethod(AliFemtoESDTrackCut::kContour);
170
171 //Track quality cuts
172 dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCrefit|AliESDtrack::kITSrefit);
173 dtc1etaphitpc[aniter]->SetminTPCncls(50);
174 dtc1etaphitpc[aniter]->SetRemoveKinks(kTRUE);
175
176
177 dtc1etaphitpc[aniter]->SetMaxTPCChiNdof(4.0);
178 dtc1etaphitpc[aniter]->SetMaxITSChiNdof(36);
179 dtc1etaphitpc[aniter]->SetLabel(kFALSE);
180
181 //primary particles: hits in ITS + DCA cut
182 dtc1etaphitpc[aniter]->SetClusterRequirementITS(AliESDtrackCuts::kSPD,
183 AliESDtrackCuts::kAny);
184 dtc1etaphitpc[aniter]->SetMaxImpactZ(2.0);
185 //dtc1etaphitpc[aniter]->SetMaxImpactXY(2.4);
186 dtc1etaphitpc[aniter]->SetMaxImpactXYPtDep(0.0105, 0.0350, -1.1);
187 //dtc1etaphitpc[aniter]->SetMaxImpactXYPtDep(0.0182, 0.0350, -1.01);
188 //dtc1etaphitpc[aniter]->SetMaxSigmaToVertex(6.0);
189
190 dtc2etaphitpc[aniter] = new AliFemtoESDTrackCut();
191
192 dtc2etaphitpc[aniter]->SetCharge(-1.0);
193
194 dtc2etaphitpc[aniter]->SetPt(0.12,4.0);
195 dtc2etaphitpc[aniter]->SetEta(-1.2,1.2);
196
197 //PID method
198 dtc2etaphitpc[aniter]->SetMass(PionMass);
199 dtc2etaphitpc[aniter]->SetMostProbablePion();
200 //dtc2etaphitpc[aniter]->SetPIDMethod(AliFemtoESDTrackCut::kContour);
201
202 //Track quality cuts
203 dtc2etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCrefit|AliESDtrack::kITSrefit);
204 dtc2etaphitpc[aniter]->SetminTPCncls(50);
205 dtc2etaphitpc[aniter]->SetRemoveKinks(kTRUE);
206
207
208 dtc2etaphitpc[aniter]->SetMaxTPCChiNdof(4.0);
209 dtc2etaphitpc[aniter]->SetMaxITSChiNdof(36);
210 dtc2etaphitpc[aniter]->SetLabel(kFALSE);
211
212 //primary particles: hits in ITS + DCA cut
213 dtc2etaphitpc[aniter]->SetClusterRequirementITS(AliESDtrackCuts::kSPD,
214 AliESDtrackCuts::kAny);
215 dtc2etaphitpc[aniter]->SetMaxImpactZ(2.0);
216 //dtc2etaphitpc[aniter]->SetMaxImpactXY(2.4);
217 dtc2etaphitpc[aniter]->SetMaxImpactXYPtDep(0.0105, 0.0350, -1.1);
218 //dtc2etaphitpc[aniter]->SetMaxImpactXYPtDep(0.0182, 0.0350, -1.01);
219 //dtc2etaphitpc[aniter]->SetMaxSigmaToVertex(6.0);
220
221 cutPass1YPtetaphitpc[aniter] = new AliFemtoCutMonitorParticleYPt(Form("cutPass1%stpcM%i", chrgs[ichg], imult), 0.13957);
222 cutFail1YPtetaphitpc[aniter] = new AliFemtoCutMonitorParticleYPt(Form("cutFail1%stpcM%i", chrgs[ichg], imult), 0.13957);
223 dtc1etaphitpc[aniter]->AddCutMonitor(cutPass1YPtetaphitpc[aniter], cutFail1YPtetaphitpc[aniter]);
224
225 cutPass1PIDetaphitpc[aniter] = new AliFemtoCutMonitorParticlePID(Form("cutPass1%stpcM%i", chrgs[ichg], imult),0);
226 cutFail1PIDetaphitpc[aniter] = new AliFemtoCutMonitorParticlePID(Form("cutFail1%stpcM%i", chrgs[ichg], imult),0);
227 dtc1etaphitpc[aniter]->AddCutMonitor(cutPass1PIDetaphitpc[aniter], cutFail1PIDetaphitpc[aniter]);
228
229 //sqpcetaphitpc[aniter] = new AliFemtoPairCutAntiGamma();
230 sqpcetaphitpc[aniter] = new AliFemtoShareQualityTPCEntranceSepPairCut();
231 //sqpcetaphitpc[aniter] = new AliFemtoPairCutRadialDistance();
232 sqpcetaphitpc[aniter]->SetShareQualityMax(1.0);
233 sqpcetaphitpc[aniter]->SetShareFractionMax(0.05);
234 sqpcetaphitpc[aniter]->SetRemoveSameLabel(kFALSE);
235 //sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.016);
236 //sqpcetaphitpc[aniter]->SetPhiStarDifferenceMinimum(0.02);
237 //runtype==0
238 // sqpcetaphitpc[aniter]->SetMaxEEMinv(0.0);
239 // sqpcetaphitpc[aniter]->SetMaxThetaDiff(0.0);
240 // sqpcetaphitpc[aniter]->SetTPCEntranceSepMinimum(1.5);
241 // sqpcetaphitpc[aniter]->SetPhiStarDistanceMinimum(0.03);
242 // sqpcetaphitpc[aniter]->SetRadialDistanceMinimum(0.12, 0.03);
243 //sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.02);
244 //runtype==1
245 // sqpcetaphitpc[aniter]->SetTPCEntranceSepMinimum(5.0);
246 // sqpcetaphitpc[aniter]->SetPhiStarDistanceMinimum(0.03);
247 //sqpcetaphitpc[aniter]->SetRadialDistanceMinimum(1.2, 0.03);
248 //sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.02);
249 //runtype==2
250 // sqpcetaphitpc[aniter]->SetTPCEntranceSepMinimum(1.0);
251 // sqpcetaphitpc[aniter]->SetPhiStarDistanceMinimum(0.03);
252 //sqpcetaphitpc[aniter]->SetRadialDistanceMinimum(1.2, 0.045);
253 //sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.016);
254 //sqpcetaphitpc[aniter]->SetPhiStarDifferenceMinimum(0.02);
255
256 anetaphitpc[aniter]->SetEventCut(mecetaphitpc[aniter]);
257 if(ichg==0)
258 {
259 anetaphitpc[aniter]->SetFirstParticleCut(dtc1etaphitpc[aniter]);
260 anetaphitpc[aniter]->SetSecondParticleCut(dtc1etaphitpc[aniter]);
261 }
262 else if(ichg==1)
263 {
264 anetaphitpc[aniter]->SetFirstParticleCut(dtc2etaphitpc[aniter]);
265 anetaphitpc[aniter]->SetSecondParticleCut(dtc2etaphitpc[aniter]);
266 }
267 else if(ichg==2)
268 {
269 anetaphitpc[aniter]->SetFirstParticleCut(dtc1etaphitpc[aniter]);
270 anetaphitpc[aniter]->SetSecondParticleCut(dtc2etaphitpc[aniter]);
271 }
272 anetaphitpc[aniter]->SetPairCut(sqpcetaphitpc[aniter]);
273
274 //Correlation functions
275
276 //Spherical harmonics (without kT bins)
277 //cylmetaphitpc[aniter] = new AliFemtoCorrFctnDirectYlm(Form("cylm%stpcM%i", chrgs[ichg], imult),3,nbinssh,0.0,shqmax,runshlcms);
278 //anetaphitpc[aniter]->AddCorrFctn(cylmetaphitpc[aniter]);
279
280 //Qinv (without kT bins)
281 //cqinvkttpc[aniter] = new AliFemtoQinvCorrFctn(Form("cqinv%stpcM%i", chrgs[ichg], imult),nbinssh,0.0,shqmax);
282 //anetaphitpc[aniter]->AddCorrFctn(cqinvkttpc[aniter]);
283
284 //3D cartesian (without kT bins)
285 if(run3d){
286 cq3dlcmskttpc[aniter] = new AliFemtoCorrFctn3DLCMSSym(Form("cq3d%stpcM%i", chrgs[ichg], imult),100,0.5);
287 anetaphitpc[aniter]->AddCorrFctn(cq3dlcmskttpc[aniter]);
288 }
289
290 // cqinvnclstpc[aniter] = new AliFemtoCorrFctnTPCNcls(Form("cqinvncls%stpcM%i", chrgs[ichg], imult),nbinssh,0.0,shqmax);
291 // anetaphitpc[aniter]->AddCorrFctn(cqinvnclstpc[aniter]);
292
293 // cqinvchi2tpc[aniter] = new AliFemtoChi2CorrFctn(Form("cqinvchi2%stpcM%i", chrgs[ichg], imult),nbinssh,0.0,shqmax);
294 // anetaphitpc[aniter]->AddCorrFctn(cqinvchi2tpc[aniter]);
295
296 if (runktdep) {
297 int ktm;
298 for (int ikt=0; ikt<7; ikt++) {
299 ktm = aniter*7 + ikt;
300 ktpcuts[ktm] = new AliFemtoKTPairCut(ktrng[ikt], ktrng[ikt+1]);
301
302 //cylmkttpc[ktm] = new AliFemtoCorrFctnDirectYlm(Form("cylm%stpcM%ikT%i", chrgs[ichg], imult, ikt),3,
303 // nbinssh, 0.0, shqmax, runshlcms);
304 //cylmkttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
305 //anetaphitpc[aniter]->AddCorrFctn(cylmkttpc[ktm]);
306
307 //cqinvkttpc[ktm] = new AliFemtoQinvCorrFctn(Form("cqinv%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0, shqmax);
308 //cqinvkttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
309 //anetaphitpc[aniter]->AddCorrFctn(cqinvkttpc[ktm]);
310
311 //cqinvsqtpc[ktm] = new AliFemtoShareQualityCorrFctn(Form("cqinvsq%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0,shqmax);
312 //cqinvsqtpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
313 //anetaphitpc[aniter]->AddCorrFctn(cqinvsqtpc[ktm]);
314
315 //cqinvinnertpc[ktm] = new AliFemtoTPCInnerCorrFctn(Form("cqinvinner%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0,shqmax);
316 //cqinvinnertpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
317 //cqinvinnertpc[ktm]->SetRadius(1.2);
318 //anetaphitpc[aniter]->AddCorrFctn(cqinvinnertpc[ktm]);
319
320 if (run3d) {
321 // cq3dlcmskttpc[ktm] = new AliFemtoCorrFctn3DLCMSSym(Form("cq3d%stpcM%ikT%i", chrgs[ichg], imult, ikt),60,(imult>3)?((imult>6)?((imult>7)?0.6:0.4):0.25):0.15);
322 cq3dlcmskttpc[ktm] = new AliFemtoCorrFctn3DLCMSSym(Form("cq3d%stpcM%ikT%i", chrgs[ichg], imult, ikt),100,0.5);
323 cq3dlcmskttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
324 anetaphitpc[aniter]->AddCorrFctn(cq3dlcmskttpc[ktm]);
325 }
326 }
327 }
328
329 //cdedpetaphi[aniter] = new AliFemtoCorrFctnDEtaDPhi(Form("cdedp%stpcM%i", chrgs[ichg], imult),39, 39);
330 //anetaphitpc[aniter]->AddCorrFctn(cdedpetaphi[aniter]);
331
332 Manager->AddAnalysis(anetaphitpc[aniter]);
333 }
334 }
335 }
336 }
337 // *** End pion-pion analysis
338
339 return Manager;
340}