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