]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWGCF/FEMTOSCOPY/macros/Train/ProtonFemtoscopy/MagMinus/ConfigFemtoAnalysis.C
Merge branch 'feature-movesplit'
[u/mrichter/AliRoot.git] / PWGCF / FEMTOSCOPY / macros / Train / ProtonFemtoscopy / MagMinus / ConfigFemtoAnalysis.C
CommitLineData
318cc34f 1/*********************************************************************
2 * *
3 * Configfemtoanalysis.C - configuration macro for the femtoscopic *
4 * analysis, meant as a QA process for two-particle effects *
5 * *
6 * Author: Adam Kisiel (Adam.Kisiel@cern.ch) *
7 * *
8 *********************************************************************/
9
10#if !defined(__CINT__) || defined(__MAKECINT_)
11#include "AliFemtoManager.h"
12#include "AliFemtoEventReaderESDChain.h"
13#include "AliFemtoEventReaderESDChainKine.h"
14#include "AliFemtoEventReaderAODChain.h"
15#include "AliFemtoSimpleAnalysis.h"
16#include "AliFemtoBasicEventCut.h"
17#include "AliFemtoESDTrackCut.h"
18#include "AliFemtoCorrFctn.h"
19#include "AliFemtoCutMonitorParticleYPt.h"
20#include "AliFemtoCutMonitorParticleVertPos.h"
21#include "AliFemtoCutMonitorParticleMomRes.h"
22#include "AliFemtoCutMonitorParticlePID.h"
23#include "AliFemtoCutMonitorEventMult.h"
24#include "AliFemtoCutMonitorEventVertex.h"
25#include "AliFemtoShareQualityTPCEntranceSepPairCut.h"
26#include "AliFemtoPairCutAntiGamma.h"
27#include "AliFemtoPairCutRadialDistance.h"
28#include "AliFemtoQinvCorrFctn.h"
29#include "AliFemtoCorrFctnNonIdDR.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#endif
51
52//________________________________________________________________________
53AliFemtoManager* ConfigFemtoAnalysis() {
54
55 double PionMass = 0.13956995;
56 double KaonMass = 0.493677;
57 double ProtonMass = 0.938272013;
58
59 int runmults[10] = {1, 1, 1, 1, 1, 1, 0, 0, 0, 0};
60 int multbins[11] = {0.001, 50, 100, 200, 300, 400, 500, 600, 700, 800, 900};
61
62 int runch[3] = {1, 1, 1};
63 const char *chrgs[3] = { "PP", "APAP", "PAP" };
64
65 int runktdep = 1;
66 double ktrng[5] = {0.1, 0.3, 0.6, 1.0, 10.0};
67
68 int numOfMultBins = 10;
69 int numOfChTypes = 3;
70 int numOfkTbins = 4;
71
72 int runqinv = 1;
73 int runshlcms = 0; // 0:PRF(PAP), 1:LCMS(PP,APAP)
74
75 int runtype = 2; // Types 0 - global, 1 - ITS only, 2 - TPC Inner
76 int isrealdata = 1;
77
78 // int gammacut = 1;
79
80 double shqmax = 2.0;
81 int nbinssh = 100;
82
83 AliFemtoEventReaderAODChain *Reader = new AliFemtoEventReaderAODChain();
84 Reader->SetFilterBit(7);
85 Reader->SetCentralityPreSelection(0.001, 510);
86
87 AliFemtoManager* Manager = new AliFemtoManager();
88 Manager->SetEventReader(Reader);
89
90 AliFemtoVertexMultAnalysis *anetaphitpc[10*3];
91 AliFemtoBasicEventCut *mecetaphitpc[10*3];
92
93 AliFemtoCutMonitorEventMult *cutPassEvMetaphitpc[10*3];
94 AliFemtoCutMonitorEventMult *cutFailEvMetaphitpc[10*3];
95
96 AliFemtoCutMonitorEventVertex *cutPassEvVetaphitpc[10*3];
97 AliFemtoCutMonitorEventVertex *cutFailEvVetaphitpc[10*3];
98
99 AliFemtoESDTrackCut *dtc1etaphitpc[10*3];
100 AliFemtoESDTrackCut *dtc2etaphitpc[10*3];
101
102 AliFemtoCutMonitorParticleYPt *cutPass1YPtetaphitpc[10*3];
103 AliFemtoCutMonitorParticleYPt *cutFail1YPtetaphitpc[10*3];
104
105 AliFemtoCutMonitorParticlePID *cutPass1PIDetaphitpc[10*3];
106 AliFemtoCutMonitorParticlePID *cutFail1PIDetaphitpc[10*3];
107
108 AliFemtoPairCutRadialDistance *sqpcetaphitpc[10*3];
109
110 AliFemtoKTPairCut *ktpcuts[100*3*4];
111 AliFemtoCorrFctnDirectYlm *cylmkttpc[100*3];
112
113 AliFemtoQinvCorrFctn *cqinvkttpc[100*3*4];
114 AliFemtoQinvCorrFctn *cqinvtpc[100*3];
115
116 AliFemtoCorrFctnNonIdDR *ckstartpc[100*3];
117 AliFemtoCorrFctnNonIdDR *ckstarkttpc[100*3*4];
118
119 // *** Third QA task - HBT analysis with all pair cuts off, TPC only ***
120 // *** Begin pion-pion (positive) analysis ***
121 int aniter = 0;
122
123 for (int imult = 0; imult < numOfMultBins; imult++) {
124 if (runmults[imult]) {
125
126 for (int ichg = 0; ichg < numOfChTypes; ichg++) {
127 if (runch[ichg]) {
128
129 aniter = ichg * numOfMultBins + imult;
130
131 anetaphitpc[aniter] = new AliFemtoVertexMultAnalysis(8, -8.0, 8.0, 4, multbins[imult], multbins[imult+1]);
132 anetaphitpc[aniter]->SetNumEventsToMix(10);
133 anetaphitpc[aniter]->SetMinSizePartCollection(1);
134
135 mecetaphitpc[aniter] = new AliFemtoBasicEventCut();
136 mecetaphitpc[aniter]->SetEventMult(0.001,100000);
137 mecetaphitpc[aniter]->SetVertZPos(-8,8);
138
139 if (isrealdata)
140 mecetaphitpc[aniter]->SetAcceptOnlyPhysics(kTRUE);
141
142 cutPassEvMetaphitpc[aniter] = new AliFemtoCutMonitorEventMult(Form("cutPass%stpcM%i", chrgs[ichg], imult));
143 cutFailEvMetaphitpc[aniter] = new AliFemtoCutMonitorEventMult(Form("cutFail%stpcM%i", chrgs[ichg], imult));
144 mecetaphitpc[aniter]->AddCutMonitor(cutPassEvMetaphitpc[aniter], cutFailEvMetaphitpc[aniter]);
145
146 // cutPassEvVetaphitpc[aniter] = new AliFemtoCutMonitorEventVertex(Form("cutPass%stpcM%i", chrgs[ichg], imult));
147 // cutFailEvVetaphitpc[aniter] = new AliFemtoCutMonitorEventVertex(Form("cutFail%stpcM%i", chrgs[ichg], imult));
148 // mecetaphitpc[aniter]->AddCutMonitor(cutPassEvVetaphitpc[aniter], cutFailEvVetaphitpc[aniter]);
149
150 dtc1etaphitpc[aniter] = new AliFemtoESDTrackCut();
151 dtc2etaphitpc[aniter] = new AliFemtoESDTrackCut();
152
153 if (ichg == 0) { // pp
154 dtc1etaphitpc[aniter]->SetCharge(1.0);
155 dtc1etaphitpc[aniter]->SetPt(0.5,5.0);
156 dtc1etaphitpc[aniter]->SetEta(-0.8,0.8);
157 dtc1etaphitpc[aniter]->SetMass(ProtonMass);
158 dtc1etaphitpc[aniter]->SetMostProbableProton();
159
160 }
161
162 else if (ichg == 1) { // apap
163 dtc1etaphitpc[aniter]->SetCharge(-1.0);
164 dtc2etaphitpc[aniter]->SetPt(0.2,5.0);
165 dtc1etaphitpc[aniter]->SetEta(-0.8,0.8);
166 dtc1etaphitpc[aniter]->SetMass(ProtonMass);
167 dtc1etaphitpc[aniter]->SetMostProbableProton();
168
169
170 }
171 else if (ichg == 2) { // pap
172 dtc1etaphitpc[aniter]->SetCharge(-1.0);
173 dtc1etaphitpc[aniter]->SetPt(0.2,5.0);
174 dtc1etaphitpc[aniter]->SetEta(-0.8,0.8);
175 dtc1etaphitpc[aniter]->SetMass(ProtonMass);
176 dtc1etaphitpc[aniter]->SetMostProbableProton();
177
178 dtc2etaphitpc[aniter]->SetCharge(1.0);
179 dtc2etaphitpc[aniter]->SetPt(0.5,5.0);
180 dtc2etaphitpc[aniter]->SetEta(-0.8,0.8);
181 dtc2etaphitpc[aniter]->SetMass(ProtonMass);
182 dtc2etaphitpc[aniter]->SetMostProbableProton();
183 }
184
185 // Track quality cuts
186
187 if (runtype == 0) {
188 dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCrefit|AliESDtrack::kITSrefit);
189 // dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCrefit);
190 // dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kITSrefit);
191 dtc1etaphitpc[aniter]->SetminTPCncls(80);
192 dtc1etaphitpc[aniter]->SetRemoveKinks(kTRUE);
193 dtc1etaphitpc[aniter]->SetLabel(kFALSE);
194 // dtc1etaphitpc[aniter]->SetMaxITSChiNdof(6.0);
195 dtc1etaphitpc[aniter]->SetMaxTPCChiNdof(4.0);
196 dtc1etaphitpc[aniter]->SetMaxImpactXY(0.2);
197 // dtc1etaphitpc[aniter]->SetMaxImpactXYPtDep(0.0182, 0.0350, -1.01);
198 dtc1etaphitpc[aniter]->SetMaxImpactZ(0.15);
199 // dtc1etaphitpc[aniter]->SetMaxSigmaToVertex(6.0);
200 }
201 else if (runtype == 1) {
202 // dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCrefit|AliESDtrack::kITSrefit);
203 // dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCrefit);
204 // dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kITSrefit|AliESDtrack::kITSpureSA);
205 // dtc1etaphitpc[aniter]->SetminTPCncls(70);
206 dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kITSrefit);
207 dtc1etaphitpc[aniter]->SetRemoveKinks(kTRUE);
208 dtc1etaphitpc[aniter]->SetLabel(kFALSE);
209 // dtc1etaphitpc[aniter]->SetMaxITSChiNdof(6.0);
210 // dtc1etaphitpc[aniter]->SetMaxTPCChiNdof(6.0);
211 dtc1etaphitpc[aniter]->SetMaxImpactXY(0.2);
212 dtc1etaphitpc[aniter]->SetMaxImpactZ(0.25);
213 // dtc1etaphitpc[aniter]->SetMaxSigmaToVertex(6.0);
214 }
215
216 else if (runtype == 2) {
217
218 dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCin);
219 dtc1etaphitpc[aniter]->SetminTPCncls(70);
220 dtc1etaphitpc[aniter]->SetRemoveKinks(kTRUE);
221 dtc1etaphitpc[aniter]->SetLabel(kFALSE);
222 dtc1etaphitpc[aniter]->SetMaxTPCChiNdof(4.0);
223 // dtc1etaphitpc[aniter]->SetMaxImpactXY(0.1); // 2.4 0.1
224 dtc1etaphitpc[aniter]->SetMaxImpactXYPtDep(0.018, 0.035, -1.01); // DCA xy
225 dtc1etaphitpc[aniter]->SetMaxImpactZ(2.0); // 2.0 0.1
226
227 if (ichg == 2) {
228
229 dtc2etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCin);
230 dtc2etaphitpc[aniter]->SetminTPCncls(70);
231 dtc2etaphitpc[aniter]->SetRemoveKinks(kTRUE);
232 dtc2etaphitpc[aniter]->SetLabel(kFALSE);
233 dtc2etaphitpc[aniter]->SetMaxTPCChiNdof(4.0);
234 // dtc2etaphitpc[aniter]->SetMaxImpactXY(0.1); // 2.4 0.1
235 dtc2etaphitpc[aniter]->SetMaxImpactXYPtDep(0.018, 0.035, -1.01); // DCA xy
236 dtc2etaphitpc[aniter]->SetMaxImpactZ(2.0); // 2.0 0.1
237 }
238
239 }
240
241 cutPass1PIDetaphitpc[aniter] = new AliFemtoCutMonitorParticlePID(Form("cutPass1%stpcM%i", chrgs[ichg], imult),2);//0-pion,1-kaon,2-proton
242 cutFail1PIDetaphitpc[aniter] = new AliFemtoCutMonitorParticlePID(Form("cutFail1%stpcM%i", chrgs[ichg], imult),2);
243 dtc1etaphitpc[aniter]->AddCutMonitor(cutPass1PIDetaphitpc[aniter], cutFail1PIDetaphitpc[aniter]);
244
245
246
247 cutPass1YPtetaphitpc[aniter] = new AliFemtoCutMonitorParticleYPt(Form("cutPass1%stpcM%i", chrgs[ichg], imult),ProtonMass);
248 cutFail1YPtetaphitpc[aniter] = new AliFemtoCutMonitorParticleYPt(Form("cutFail1%stpcM%i", chrgs[ichg], imult),ProtonMass);
249 dtc1etaphitpc[aniter]->AddCutMonitor(cutPass1YPtetaphitpc[aniter], cutFail1YPtetaphitpc[aniter]);
250
251
252
253 // sqpcetaphitpc[aniter] = new AliFemtoPairCutAntiGamma();
254 sqpcetaphitpc[aniter] = new AliFemtoPairCutRadialDistance();
255
256 if (runtype == 0) {
257 sqpcetaphitpc[aniter]->SetShareQualityMax(1.0);
258 sqpcetaphitpc[aniter]->SetShareFractionMax(0.05);
259 sqpcetaphitpc[aniter]->SetRemoveSameLabel(kFALSE);
260 // sqpcetaphitpc[aniter]->SetMaxEEMinv(0.0);
261 // sqpcetaphitpc[aniter]->SetMaxThetaDiff(0.0);
262 // sqpcetaphitpc[aniter]->SetTPCEntranceSepMinimum(1.5);
263 //sqpcetaphitpc[aniter]->SetRadialDistanceMinimum(0.12, 0.03);
264 // sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.02);
265 }
266 else if (runtype == 1) {
267 sqpcetaphitpc[aniter]->SetShareQualityMax(1.0);
268 sqpcetaphitpc[aniter]->SetShareFractionMax(1.05);
269 sqpcetaphitpc[aniter]->SetRemoveSameLabel(kFALSE);
270 // sqpcetaphitpc[aniter]->SetMaxEEMinv(0.002);
271 // sqpcetaphitpc[aniter]->SetMaxThetaDiff(0.008);
272 // sqpcetaphitpc[aniter]->SetTPCEntranceSepMinimum(5.0);
273 //sqpcetaphitpc[aniter]->SetRadialDistanceMinimum(1.2, 0.03);
274 // sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.02);
275 }
276 else if (runtype == 2) {
277
278 sqpcetaphitpc[aniter]->SetShareQualityMax(1.0);
279 sqpcetaphitpc[aniter]->SetShareFractionMax(0.05);
280 sqpcetaphitpc[aniter]->SetRemoveSameLabel(kFALSE);
281
282 // if (gammacut == 0) {
283 //sqpcetaphitpc[aniter]->SetMaxEEMinv(0.0);
284 //sqpcetaphitpc[aniter]->SetMaxThetaDiff(0.0);
285 //}
286 //else if (gammacut == 1) {
287 //sqpcetaphitpc[aniter]->SetMaxEEMinv(0.002);
288 //sqpcetaphitpc[aniter]->SetMaxThetaDiff(0.008);
289 //}
290
291 sqpcetaphitpc[aniter]->SetMagneticFieldSign(-1); // field1 -1, field3 +1
292 sqpcetaphitpc[aniter]->SetPhiStarDifferenceMinimum(0.012); // 0.012 - pions, 0.017 - kaons, 0.018
293 sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.015); // 0.017 - pions, 0.015 - kaons
294
295 }
296
297 anetaphitpc[aniter]->SetEventCut(mecetaphitpc[aniter]);
298
299 if (ichg == 2) {
300 anetaphitpc[aniter]->SetFirstParticleCut(dtc1etaphitpc[aniter]);
301 anetaphitpc[aniter]->SetSecondParticleCut(dtc2etaphitpc[aniter]);
302 }
303 else {
304 anetaphitpc[aniter]->SetFirstParticleCut(dtc1etaphitpc[aniter]);
305 anetaphitpc[aniter]->SetSecondParticleCut(dtc1etaphitpc[aniter]);
306 }
307
308 anetaphitpc[aniter]->SetPairCut(sqpcetaphitpc[aniter]);
309
310
311 if (ichg == 2) {
312 ckstartpc[aniter] = new AliFemtoCorrFctnNonIdDR(Form("ckstar%stpcM%i", chrgs[ichg], imult),nbinssh,0.0,shqmax);
313 anetaphitpc[aniter]->AddCorrFctn(ckstartpc[aniter]);
314 }
315 else {
316 cqinvtpc[aniter] = new AliFemtoQinvCorrFctn(Form("cqinv%stpcM%i", chrgs[ichg], imult),nbinssh,0.0,shqmax);
317 anetaphitpc[aniter]->AddCorrFctn(cqinvtpc[aniter]);
318 }
319
320 cylmkttpc[aniter] = new AliFemtoCorrFctnDirectYlm(Form("cylm%stpcM%i", chrgs[ichg], imult),2,nbinssh, 0.0,shqmax,runshlcms);
321 anetaphitpc[aniter]->AddCorrFctn(cylmkttpc[aniter]);
322
323
324 if (runktdep) {
325 int ktm;
326 for (int ikt=0; ikt<numOfkTbins; ikt++) {
327
328 ktm = aniter * numOfkTbins + ikt;
329 ktpcuts[ktm] = new AliFemtoKTPairCut(ktrng[ikt], ktrng[ikt+1]);
330
331 cylmkttpc[ktm] = new AliFemtoCorrFctnDirectYlm(Form("cylm%stpcM%ikT%i", chrgs[ichg], imult, ikt),2,nbinssh,0.0,shqmax,0);
332 cylmkttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
333 anetaphitpc[aniter]->AddCorrFctn(cylmkttpc[ktm]);
334
335 if (ichg == 2) {
336 ckstarkttpc[ktm] = new AliFemtoCorrFctnNonIdDR(Form("ckstar%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0,shqmax);
337 ckstarkttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
338 anetaphitpc[aniter]->AddCorrFctn(ckstarkttpc[ktm]);
339
340
341 }
342 else {
343 cqinvkttpc[ktm] = new AliFemtoQinvCorrFctn(Form("cqinv%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0,shqmax);
344 cqinvkttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
345 anetaphitpc[aniter]->AddCorrFctn(cqinvkttpc[ktm]);
346 }
347
348 }
349 }
350
351
352 Manager->AddAnalysis(anetaphitpc[aniter]);
353 }
354 }
355 }
356 }
357 // *** End analysis
358
359 return Manager;
360}