]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWGCF/FEMTOSCOPY/macros/Train/pPbpionFemto/ESD/SH/ZNA/ConfigFemtoAnalysis.C
Merge branch 'feature-movesplit'
[u/mrichter/AliRoot.git] / PWGCF / FEMTOSCOPY / macros / Train / pPbpionFemto / ESD / SH / ZNA / ConfigFemtoAnalysis.C
CommitLineData
c863b24a 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[2] = {1, 1};
64 const char *chrgs[2] = { "pip", "pim" };
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 = 0; // 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(4);
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::kCentralityZNA);
91 Reader->SetReadTrackType(AliFemtoEventReaderESDChain::kGlobal);
92
93
94 AliFemtoManager* Manager=new AliFemtoManager();
95 Manager->SetEventReader(Reader);
96
97 AliFemtoVertexMultAnalysis *anetaphitpc[20];
98 AliFemtoBasicEventCut *mecetaphitpc[20];
99 AliFemtoCutMonitorEventMult *cutPassEvMetaphitpc[20];
100 AliFemtoCutMonitorEventMult *cutFailEvMetaphitpc[20];
101 AliFemtoCutMonitorEventVertex *cutPassEvVetaphitpc[20];
102 AliFemtoCutMonitorEventVertex *cutFailEvVetaphitpc[20];
103 AliFemtoCutMonitorCollections *cutPassColletaphitpc[20];
104 AliFemtoCutMonitorCollections *cutFailColletaphitpc[20];
105 AliFemtoESDTrackCut *dtc1etaphitpc[20];
106 AliFemtoESDTrackCut *dtc2etaphitpc[20];
107 AliFemtoCutMonitorParticleYPt *cutPass1YPtetaphitpc[20];
108 AliFemtoCutMonitorParticleYPt *cutFail1YPtetaphitpc[20];
109 AliFemtoCutMonitorParticlePID *cutPass1PIDetaphitpc[20];
110 AliFemtoCutMonitorParticlePID *cutFail1PIDetaphitpc[20];
111 AliFemtoCutMonitorParticleYPt *cutPass2YPtetaphitpc[20];
112 AliFemtoCutMonitorParticleYPt *cutFail2YPtetaphitpc[20];
113 AliFemtoCutMonitorParticlePID *cutPass2PIDetaphitpc[20];
114 AliFemtoCutMonitorParticlePID *cutFail2PIDetaphitpc[20];
115 // AliFemtoPairCutAntiGamma *sqpcetaphitpc[20];
116 AliFemtoShareQualityTPCEntranceSepPairCut *sqpcetaphitpc[20];
117 //AliFemtoPairCutRadialDistance *sqpcetaphitpc[20];
118 AliFemtoCorrFctnDirectYlm *cylmetaphitpc[20];
119 AliFemtoCorrFctnDEtaDPhi *cdedpetaphi[20];
120 AliFemtoChi2CorrFctn *cchiqinvetaphitpc[20];
121 AliFemtoKTPairCut *ktpcuts[20*7];
122 AliFemtoCorrFctnDirectYlm *cylmkttpc[20*7];
123 AliFemtoQinvCorrFctn *cqinvkttpc[20*7];
124 AliFemtoCorrFctn3DLCMSSym *cq3dlcmskttpc[20*7];
125 AliFemtoCorrFctnTPCNcls *cqinvnclstpc[20];
126 AliFemtoShareQualityCorrFctn *cqinvsqtpc[20*10];
127 AliFemtoChi2CorrFctn *cqinvchi2tpc[20];
128 AliFemtoTPCInnerCorrFctn *cqinvinnertpc[20*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<2; 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 if (ichg == 0)
162 dtc1etaphitpc[aniter]->SetCharge(1.0);
163 else if (ichg == 1)
164 dtc1etaphitpc[aniter]->SetCharge(-1.0);
165
166 dtc1etaphitpc[aniter]->SetPt(0.12,4.0);
167 dtc1etaphitpc[aniter]->SetEta(-1.2,1.2);
168
169 //PID method
170 dtc1etaphitpc[aniter]->SetMass(PionMass);
171 dtc1etaphitpc[aniter]->SetMostProbablePion();
172 //dtc1etaphitpc[aniter]->SetPIDMethod(AliFemtoESDTrackCut::kContour);
173
174 //Track quality cuts
175 dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCrefit|AliESDtrack::kITSrefit);
176 dtc1etaphitpc[aniter]->SetminTPCncls(50);
177 dtc1etaphitpc[aniter]->SetRemoveKinks(kTRUE);
178
179
180 dtc1etaphitpc[aniter]->SetMaxTPCChiNdof(4.0);
181 dtc1etaphitpc[aniter]->SetMaxITSChiNdof(36);
182 dtc1etaphitpc[aniter]->SetLabel(kFALSE);
183
184 //primary particles: hits in ITS + DCA cut
185 dtc1etaphitpc[aniter]->SetClusterRequirementITS(AliESDtrackCuts::kSPD,
186 AliESDtrackCuts::kAny);
187 dtc1etaphitpc[aniter]->SetMaxImpactZ(2.0);
188 //dtc1etaphitpc[aniter]->SetMaxImpactXY(2.4);
189 dtc1etaphitpc[aniter]->SetMaxImpactXYPtDep(0.0105, 0.0350, -1.1);
190 //dtc1etaphitpc[aniter]->SetMaxImpactXYPtDep(0.0182, 0.0350, -1.01);
191 //dtc1etaphitpc[aniter]->SetMaxSigmaToVertex(6.0);
192
193 cutPass1YPtetaphitpc[aniter] = new AliFemtoCutMonitorParticleYPt(Form("cutPass1%stpcM%i", chrgs[ichg], imult), 0.13957);
194 cutFail1YPtetaphitpc[aniter] = new AliFemtoCutMonitorParticleYPt(Form("cutFail1%stpcM%i", chrgs[ichg], imult), 0.13957);
195 dtc1etaphitpc[aniter]->AddCutMonitor(cutPass1YPtetaphitpc[aniter], cutFail1YPtetaphitpc[aniter]);
196
197 cutPass1PIDetaphitpc[aniter] = new AliFemtoCutMonitorParticlePID(Form("cutPass1%stpcM%i", chrgs[ichg], imult),0);
198 cutFail1PIDetaphitpc[aniter] = new AliFemtoCutMonitorParticlePID(Form("cutFail1%stpcM%i", chrgs[ichg], imult),0);
199 dtc1etaphitpc[aniter]->AddCutMonitor(cutPass1PIDetaphitpc[aniter], cutFail1PIDetaphitpc[aniter]);
200
201 //sqpcetaphitpc[aniter] = new AliFemtoPairCutAntiGamma();
202 sqpcetaphitpc[aniter] = new AliFemtoShareQualityTPCEntranceSepPairCut();
203 //sqpcetaphitpc[aniter] = new AliFemtoPairCutRadialDistance();
204 sqpcetaphitpc[aniter]->SetShareQualityMax(1.0);
205 sqpcetaphitpc[aniter]->SetShareFractionMax(0.05);
206 sqpcetaphitpc[aniter]->SetRemoveSameLabel(kFALSE);
207 //sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.016);
208 //sqpcetaphitpc[aniter]->SetPhiStarDifferenceMinimum(0.02);
209 //runtype==0
210 // sqpcetaphitpc[aniter]->SetMaxEEMinv(0.0);
211 // sqpcetaphitpc[aniter]->SetMaxThetaDiff(0.0);
212 // sqpcetaphitpc[aniter]->SetTPCEntranceSepMinimum(1.5);
213 // sqpcetaphitpc[aniter]->SetPhiStarDistanceMinimum(0.03);
214 // sqpcetaphitpc[aniter]->SetRadialDistanceMinimum(0.12, 0.03);
215 //sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.02);
216 //runtype==1
217 // sqpcetaphitpc[aniter]->SetTPCEntranceSepMinimum(5.0);
218 // sqpcetaphitpc[aniter]->SetPhiStarDistanceMinimum(0.03);
219 //sqpcetaphitpc[aniter]->SetRadialDistanceMinimum(1.2, 0.03);
220 //sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.02);
221 //runtype==2
222 // sqpcetaphitpc[aniter]->SetTPCEntranceSepMinimum(1.0);
223 // sqpcetaphitpc[aniter]->SetPhiStarDistanceMinimum(0.03);
224 //sqpcetaphitpc[aniter]->SetRadialDistanceMinimum(1.2, 0.045);
225 //sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.016);
226 //sqpcetaphitpc[aniter]->SetPhiStarDifferenceMinimum(0.02);
227
228 anetaphitpc[aniter]->SetEventCut(mecetaphitpc[aniter]);
229 anetaphitpc[aniter]->SetFirstParticleCut(dtc1etaphitpc[aniter]);
230 anetaphitpc[aniter]->SetSecondParticleCut(dtc1etaphitpc[aniter]);
231 anetaphitpc[aniter]->SetPairCut(sqpcetaphitpc[aniter]);
232
233 //Correlation functions
234
235 //Spherical harmonics (without kT bins)
236 cylmetaphitpc[aniter] = new AliFemtoCorrFctnDirectYlm(Form("cylm%stpcM%i", chrgs[ichg], imult),3,nbinssh,0.0,shqmax,runshlcms);
237 anetaphitpc[aniter]->AddCorrFctn(cylmetaphitpc[aniter]);
238
239 //Qinv (without kT bins)
240 cqinvkttpc[aniter] = new AliFemtoQinvCorrFctn(Form("cqinv%stpcM%i", chrgs[ichg], imult),nbinssh,0.0,shqmax);
241 anetaphitpc[aniter]->AddCorrFctn(cqinvkttpc[aniter]);
242
243 //3D cartesian (without kT bins)
244 if(run3d){
245 cq3dlcmskttpc[aniter] = new AliFemtoCorrFctn3DLCMSSym(Form("cq3d%stpcM%i", chrgs[ichg], imult),60,0.5);
246 anetaphitpc[aniter]->AddCorrFctn(cq3dlcmskttpc[aniter]);
247 }
248
249 // cqinvnclstpc[aniter] = new AliFemtoCorrFctnTPCNcls(Form("cqinvncls%stpcM%i", chrgs[ichg], imult),nbinssh,0.0,shqmax);
250 // anetaphitpc[aniter]->AddCorrFctn(cqinvnclstpc[aniter]);
251
252 // cqinvchi2tpc[aniter] = new AliFemtoChi2CorrFctn(Form("cqinvchi2%stpcM%i", chrgs[ichg], imult),nbinssh,0.0,shqmax);
253 // anetaphitpc[aniter]->AddCorrFctn(cqinvchi2tpc[aniter]);
254
255 if (runktdep) {
256 int ktm;
257 for (int ikt=0; ikt<7; ikt++) {
258 ktm = aniter*7 + ikt;
259 ktpcuts[ktm] = new AliFemtoKTPairCut(ktrng[ikt], ktrng[ikt+1]);
260
261 cylmkttpc[ktm] = new AliFemtoCorrFctnDirectYlm(Form("cylm%stpcM%ikT%i", chrgs[ichg], imult, ikt),3,
262 nbinssh, 0.0, shqmax, runshlcms);
263 cylmkttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
264 anetaphitpc[aniter]->AddCorrFctn(cylmkttpc[ktm]);
265
266 cqinvkttpc[ktm] = new AliFemtoQinvCorrFctn(Form("cqinv%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0, shqmax);
267 cqinvkttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
268 anetaphitpc[aniter]->AddCorrFctn(cqinvkttpc[ktm]);
269
270 cqinvsqtpc[ktm] = new AliFemtoShareQualityCorrFctn(Form("cqinvsq%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0,shqmax);
271 cqinvsqtpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
272 anetaphitpc[aniter]->AddCorrFctn(cqinvsqtpc[ktm]);
273
274 cqinvinnertpc[ktm] = new AliFemtoTPCInnerCorrFctn(Form("cqinvinner%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0,shqmax);
275 cqinvinnertpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
276 cqinvinnertpc[ktm]->SetRadius(1.2);
277 anetaphitpc[aniter]->AddCorrFctn(cqinvinnertpc[ktm]);
278
279 if (run3d) {
280 // 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);
281 cq3dlcmskttpc[ktm] = new AliFemtoCorrFctn3DLCMSSym(Form("cq3d%stpcM%ikT%i", chrgs[ichg], imult, ikt),60,0.5);
282 cq3dlcmskttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
283 anetaphitpc[aniter]->AddCorrFctn(cq3dlcmskttpc[ktm]);
284 }
285 }
286 }
287
288 cdedpetaphi[aniter] = new AliFemtoCorrFctnDEtaDPhi(Form("cdedp%stpcM%i", chrgs[ichg], imult),39, 39);
289 anetaphitpc[aniter]->AddCorrFctn(cdedpetaphi[aniter]);
290
291 Manager->AddAnalysis(anetaphitpc[aniter]);
292 }
293 }
294 }
295 }
296 // *** End pion-pion analysis
297
298 return Manager;
299}