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