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