]> git.uio.no Git - u/mrichter/AliRoot.git/blob - PWGCF/FEMTOSCOPY/macros/Train/AshutoshPionFemtoscopy/ConfigFemtoAnalysisSpherical.C
Merge branch 'feature-movesplit'
[u/mrichter/AliRoot.git] / PWGCF / FEMTOSCOPY / macros / Train / AshutoshPionFemtoscopy / ConfigFemtoAnalysisSpherical.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* ConfigFemtoAnalysisSpherical() {
55
56   double PionMass = 0.13956995;
57   double KaonMass = 0.493677;
58         
59   //multiplicity bins
60   int runmults[10] = {1, 0, 0, 0, 0, 0, 0, 0, 0, 0};
61   int multbins[11] = {0, 50, 200, 300, 600, 800, 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[4] = {0.2, 0.3,0.4,0.5};
68   double ktrng[8] = { 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 1.0};
69   int run3d = 1; // 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(7);
83   //Reader->SetCentralityPreSelection(0.001, 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::kV0Centrality);
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   AliFemtoPairCutRadialDistanceLM      *sqpcetaphitpcRD[20];
119   AliFemtoCorrFctnDirectYlm     *cylmetaphitpc[20];
120   AliFemtoCorrFctnDEtaDPhi      *cdedpetaphi[20];
121   AliFemtoChi2CorrFctn          *cchiqinvetaphitpc[20];
122   AliFemtoKTPairCut             *ktpcuts[20*7];
123   AliFemtoCorrFctnDirectYlm     *cylmkttpc[20*7];
124   AliFemtoQinvCorrFctn          *cqinvkttpc[20*7];
125   AliFemtoCorrFctn3DLCMSSym     *cq3dlcmskttpc[20*7];
126   AliFemtoCorrFctn3DSpherical   *cq3dspherical[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(4);
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
163           if (ichg == 0)
164             dtc1etaphitpc[aniter]->SetCharge(1.0);
165           else if (ichg == 1)
166             dtc1etaphitpc[aniter]->SetCharge(-1.0);
167
168           dtc1etaphitpc[aniter]->SetPt(0.14,1.5);
169           dtc1etaphitpc[aniter]->SetEta(-0.8,0.8);
170           
171           //PID method
172           dtc1etaphitpc[aniter]->SetMass(PionMass);
173           dtc1etaphitpc[aniter]->SetMostProbablePion();
174           //dtc1etaphitpc[aniter]->SetPIDMethod(1);//contour
175           //dtc1etaphitpc[aniter]->SetNsigmaTPCTOF(kTRUE);
176           //dtc1etaphitpc[aniter]->SetPIDMethod(AliFemtoESDTrackCut::kContour);
177
178           //Track quality cuts
179           //dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCrefit|AliESDtrack::kITSrefit);
180           dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCin);
181           dtc1etaphitpc[aniter]->SetminTPCncls(80);
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(3.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           sqpcetaphitpcRD[aniter] = new AliFemtoPairCutRadialDistanceLM();
209           sqpcetaphitpc[aniter]->SetShareQualityMax(1.0);
210           sqpcetaphitpc[aniter]->SetShareFractionMax(0.05);
211           sqpcetaphitpc[aniter]->SetRemoveSameLabel(kFALSE);
212           sqpcetaphitpc[aniter]->SetMinimumRadius(1.6);
213           sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.02);
214           sqpcetaphitpc[aniter]->SetPhiStarDifferenceMinimum(0.045);  
215           sqpcetaphitpcRD[aniter]->SetMinimumRadius(1.6);
216           sqpcetaphitpcRD[aniter]->SetEtaDifferenceMinimum(0.02);
217           sqpcetaphitpcRD[aniter]->SetPhiStarDifferenceMinimum(0.045);
218           //runtype==0
219           // sqpcetaphitpc[aniter]->SetMaxEEMinv(0.0);
220           // sqpcetaphitpc[aniter]->SetMaxThetaDiff(0.0);
221           // sqpcetaphitpc[aniter]->SetTPCEntranceSepMinimum(1.5);
222           // sqpcetaphitpc[aniter]->SetPhiStarDistanceMinimum(0.03);
223           // sqpcetaphitpc[aniter]->SetRadialDistanceMinimum(0.12, 0.03);
224           //sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.02);
225           //runtype==1
226           //        sqpcetaphitpc[aniter]->SetTPCEntranceSepMinimum(5.0);
227           //        sqpcetaphitpc[aniter]->SetPhiStarDistanceMinimum(0.03);
228           //sqpcetaphitpc[aniter]->SetRadialDistanceMinimum(1.2, 0.03);
229           //sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.02);
230           //runtype==2
231           //        sqpcetaphitpc[aniter]->SetTPCEntranceSepMinimum(1.0);
232           //        sqpcetaphitpc[aniter]->SetPhiStarDistanceMinimum(0.03);
233           //sqpcetaphitpc[aniter]->SetRadialDistanceMinimum(1.2, 0.045);
234           //sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.016);
235           //sqpcetaphitpc[aniter]->SetPhiStarDifferenceMinimum(0.02);
236           
237           anetaphitpc[aniter]->SetEventCut(mecetaphitpc[aniter]);
238           anetaphitpc[aniter]->SetFirstParticleCut(dtc1etaphitpc[aniter]);
239           anetaphitpc[aniter]->SetSecondParticleCut(dtc1etaphitpc[aniter]);
240           anetaphitpc[aniter]->SetPairCut(sqpcetaphitpc[aniter]);
241           
242           
243
244           if (runktdep) {
245             int ktm;
246             for (int ikt=0; ikt<7; ikt++) {
247               ktm = aniter*7 + ikt;
248               ktpcuts[ktm] = new AliFemtoKTPairCut(ktrng[ikt], ktrng[ikt+1]);
249               
250               /*cqinvkttpc[ktm] = new AliFemtoQinvCorrFctn(Form("cqinv%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0, shqmax);
251               cqinvkttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
252               anetaphitpc[aniter]->AddCorrFctn(cqinvkttpc[ktm]);*/
253
254               if (run3d) {
255
256                 /*cq3dlcmskttpc[ktm] = new AliFemtoCorrFctn3DLCMSSym(Form("cq3d%stpcM%ikT%i", chrgs[ichg], imult, ikt),100,0.5);
257                 cq3dlcmskttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
258                 anetaphitpc[aniter]->AddCorrFctn(cq3dlcmskttpc[ktm]);*/
259
260                cq3dspherical[ktm] = new AliFemtoCorrFctn3DSpherical(Form("cq3d%stpcM%ikT%i",chrgs[ichg], imult, ikt), 500,0.0,5.0,200,200);
261                cq3dspherical[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
262                anetaphitpc[aniter]->AddCorrFctn(cq3dspherical[ktm]);
263
264               }
265             }
266           }
267           
268         
269           
270           Manager->AddAnalysis(anetaphitpc[aniter]);    
271         }
272       }
273     }
274   }
275   // *** End pion-pion analysis
276
277   return Manager;
278 }