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