]> git.uio.no Git - u/mrichter/AliRoot.git/blob - PWGCF/FEMTOSCOPY/macros/Train/NZ/pPb_TTC_kt13/ConfigFemtoAnalysis.C
Merge branch 'feature-movesplit'
[u/mrichter/AliRoot.git] / PWGCF / FEMTOSCOPY / macros / Train / NZ / pPb_TTC_kt13 / 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 "AliFemtoKKTrackCut.h"
20 #include "AliFemtoCorrFctn.h"
21 #include "AliFemtoCutMonitorParticleYPt.h"
22 #include "AliFemtoCutMonitorParticleVertPos.h"
23 #include "AliFemtoCutMonitorParticleMomRes.h"
24 #include "AliFemtoCutMonitorParticlePID.h"
25 #include "AliFemtoCutMonitorEventMult.h"
26 #include "AliFemtoCutMonitorEventVertex.h"
27 #include "AliFemtoShareQualityTPCEntranceSepPairCut.h"
28 #include "AliFemtoPairCutAntiGamma.h"
29 #include "AliFemtoPairCutRadialDistanceKK.h"
30 #include "AliFemtoQinvCorrFctn.h"
31 #include "AliFemtoShareQualityCorrFctn.h"
32 #include "AliFemtoTPCInnerCorrFctn.h"
33 #include "AliFemtoVertexMultAnalysis.h"
34 #include "AliFemtoCorrFctn3DSpherical.h"
35 #include "AliFemtoChi2CorrFctn.h"
36 #include "AliFemtoCorrFctnTPCNcls.h"
37 #include "AliFemtoBPLCMS3DCorrFctn.h"
38 #include "AliFemtoCorrFctn3DLCMSSym.h"
39 #include "AliFemtoModelBPLCMSCorrFctn.h"
40 #include "AliFemtoModelCorrFctn3DSpherical.h"
41 #include "AliFemtoModelGausLCMSFreezeOutGenerator.h"
42 #include "AliFemtoModelGausRinvFreezeOutGenerator.h"
43 #include "AliFemtoModelManager.h"
44 #include "AliFemtoModelWeightGeneratorBasic.h"
45 #include "AliFemtoModelWeightGeneratorLednicky.h"
46 #include "AliFemtoCorrFctnDirectYlm.h"
47 #include "AliFemtoModelCorrFctnDirectYlm.h"
48 #include "AliFemtoModelCorrFctnSource.h"
49 #include "AliFemtoCutMonitorParticlePtPDG.h"
50 #include "AliFemtoKTPairCut.h"
51 #include "AliFemtoCutMonitorCollections.h"
52 #endif
53
54 //________________________________________________________________________
55 AliFemtoManager* ConfigFemtoAnalysis() {
56
57   double PionMass = 0.13956995;
58   double KaonMass = 0.493677;
59   
60   //multiplicity bins
61   // int runmults[1] = {1};//3
62    // int multbins[2] = {0.01, 990};//4
63     
64      int runmults[3] = {1, 1, 1};
65     int multbins[4] = {0.01, 200, 400, 900};
66
67
68   int runch[2] = {1, 1};
69   // const char *chrgs[2] = { "pip", "pim" };
70   const char *chrgs[2] = { "Kp", "Km"};
71
72   int runktdep = 1;
73   //double ktrng[9] = {0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 1.0, 1.3};
74   double ktrng[3] = {0.2, 0.5, 1.3};
75
76   int run3d = 0; // Do 3D cartesian analysis?
77   //int runshlcms = 1;
78   int runshlcms = 0;
79
80   //PhysicsSelection set only in runBatch with trigger + Physics Selection Task
81
82   double shqmax;
83   //int nbinssh = 200;
84   int nbinssh = 100;
85
86   //if (runshlcms) shqmax = 2.0;
87   if (runshlcms) shqmax = 0.25;
88   else shqmax = 2.0;
89
90   AliFemtoEventReaderAODChain *Reader = new AliFemtoEventReaderAODChain();
91   Reader->SetUseMultiplicity(AliFemtoEventReaderAODChain::kCentrality);
92   Reader->SetFilterBit(7);
93  // Reader->SetDCAglobalTrack(kTRUE);
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   AliFemtoKKTrackCut           *dtc1etaphitpc[20];
108   AliFemtoKKTrackCut           *dtc2etaphitpc[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   AliFemtoPairCutRadialDistanceKK      *sqpcetaphitpc[20];
121   //AliFemtoShareQualityPairCut      *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   AliFemtoKTPairCut             *ktpcuts[20*8];
130   AliFemtoCorrFctnDirectYlm     *cylmkttpc[20*8];
131   AliFemtoQinvCorrFctn          *cqinvkttpc[20*8];
132   AliFemtoCorrFctn3DLCMSSym     *cq3dlcmskttpc[20*8];
133   AliFemtoCorrFctnTPCNcls       *cqinvnclstpc[20];
134   AliFemtoShareQualityCorrFctn  *cqinvsqtpc[20*10];
135   AliFemtoChi2CorrFctn          *cqinvchi2tpc[20];
136   AliFemtoTPCInnerCorrFctn      *cqinvinnertpc[20*10];
137
138   // *** Begin pion-pion analysis ***
139   int aniter = 0;
140
141   for (int imult=0; imult<3; imult++) {
142     if (runmults[imult]) {
143       for (int ichg=0; ichg<2; ichg++) {
144       if (runch[ichg]) {
145         aniter = ichg*3+imult;
146
147           anetaphitpc[aniter] = new AliFemtoVertexMultAnalysis(10, -10.0, 10.0, 4, multbins[imult], multbins[imult+1]);
148             anetaphitpc[aniter]->SetNumEventsToMix(5);
149               anetaphitpc[aniter]->SetMinSizePartCollection(1);
150                 anetaphitpc[aniter]->SetVerboseMode(kTRUE);
151       
152     mecetaphitpc[aniter] = new AliFemtoBasicEventCut();
153     mecetaphitpc[aniter]->SetEventMult(0,10000);
154     mecetaphitpc[aniter]->SetVertZPos(-10,10);
155
156         cutPassEvMetaphitpc[aniter] = new AliFemtoCutMonitorEventMult(Form("cutPass%stpcM%i", chrgs[ichg], imult));
157         cutFailEvMetaphitpc[aniter] = new AliFemtoCutMonitorEventMult(Form("cutFail%stpcM%i", chrgs[ichg], imult));
158         mecetaphitpc[aniter]->AddCutMonitor(cutPassEvMetaphitpc[aniter], cutFailEvMetaphitpc[aniter]);
159                     
160         cutPassEvVetaphitpc[aniter] = new AliFemtoCutMonitorEventVertex(Form("cutPass%stpcM%i", chrgs[ichg], imult));
161         cutFailEvVetaphitpc[aniter] = new AliFemtoCutMonitorEventVertex(Form("cutFail%stpcM%i", chrgs[ichg], imult));
162         mecetaphitpc[aniter]->AddCutMonitor(cutPassEvVetaphitpc[aniter], cutFailEvVetaphitpc[aniter]);
163                             
164         cutPassColletaphitpc[aniter] = new AliFemtoCutMonitorCollections(Form("cutPass%stpcM%i", chrgs[ichg], imult));
165         cutFailColletaphitpc[aniter] = new AliFemtoCutMonitorCollections(Form("cutFail%stpcM%i", chrgs[ichg], imult));
166         mecetaphitpc[aniter]->AddCutMonitor(cutPassColletaphitpc[aniter], cutFailColletaphitpc[aniter]);
167
168         //dtc1etaphitpc[aniter] = new AliFemtoESDTrackCut();
169          dtc1etaphitpc[aniter] = new AliFemtoKKTrackCut();
170
171           if (ichg == 0)
172            dtc1etaphitpc[aniter]->SetCharge(1.0);
173             else if (ichg == 1)
174            dtc1etaphitpc[aniter]->SetCharge(-1.0);
175
176                 // dtc1etaphitpc[aniter]->SetPt(0.12,4.0);
177                 // dtc1etaphitpc[aniter]->SetEta(-1.2,1.2);
178                  dtc1etaphitpc[aniter]->SetPt(0.14,1.5);
179                  dtc1etaphitpc[aniter]->SetEta(-0.8,0.8);
180
181                  //PID method
182                 // dtc1etaphitpc[aniter]->SetMass(PionMass);
183                  //  dtc1etaphitpc[aniter]->SetMostProbablePion();
184                  dtc1etaphitpc[aniter]->SetMass(KaonMass);
185                  dtc1etaphitpc[aniter]->SetMostProbableKaon();
186                                                    //dtc1etaphitpc[aniter]->SetPIDMethod(AliFemtoESDTrackCut::kContour);
187 //------------------- November 2013 -----------------------------------< 
188                 // new cuts to remove electron (do not take into analysis if 400<p<500) 
189                 dtc1etaphitpc[aniter]->SetNsigmaTPCle250(2.0);
190                 dtc1etaphitpc[aniter]->SetNsigmaTPC250_400(2.0);
191                  dtc1etaphitpc[aniter]->SetNsigmaTPC400_450(2.0);
192                 dtc1etaphitpc[aniter]->SetNsigmaTPC450_500(2.0);
193                 dtc1etaphitpc[aniter]->SetNsigmaTPCge500(3.0);    
194                 // new cuts are stronger, better separation of pion in TOF 
195                 // when momentum is greater then 800 MeV/c
196                 dtc1etaphitpc[aniter]->SetNsigmaTOF500_800(2.0);
197                 dtc1etaphitpc[aniter]->SetNsigmaTOF800_1000(1.5);
198                 dtc1etaphitpc[aniter]->SetNsigmaTOFge1000(1.0);
199        //------------------- November 2013 ----------------------------------->
200          //Track quality cuts
201          // dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCrefit|AliESDtrack::kITSrefit);
202            dtc1etaphitpc[aniter]->SetminTPCncls(50);
203             dtc1etaphitpc[aniter]->SetRemoveKinks(kTRUE);
204
205
206            dtc1etaphitpc[aniter]->SetMaxTPCChiNdof(4.0);
207            // dtc1etaphitpc[aniter]->SetMaxITSChiNdof(36);        
208             dtc1etaphitpc[aniter]->SetLabel(kFALSE);
209                   
210          //primary particles: hits in ITS + DCA cut
211         //dtc1etaphitpc[aniter]->SetClusterRequirementITS(AliESDtrackCuts::kSPD,
212          //                              AliESDtrackCuts::kAny);
213          dtc1etaphitpc[aniter]->SetMaxImpactZ(3.2);
214         dtc1etaphitpc[aniter]->SetMaxImpactXY(2.4);
215         //dtc1etaphitpc[aniter]->SetMaxImpactXYPtDep(0.0105, 0.0350, -1.1);
216          //dtc1etaphitpc[aniter]->SetMaxImpactXYPtDep(0.0182, 0.0350, -1.01);
217          //dtc1etaphitpc[aniter]->SetMaxSigmaToVertex(6.0);
218
219          //cutPass1YPtetaphitpc[aniter] = new AliFemtoCutMonitorParticleYPt(Form("cutPass1%stpcM%i", chrgs[ichg], imult), 0.13957);
220         //cutFail1YPtetaphitpc[aniter] = new AliFemtoCutMonitorParticleYPt(Form("cutFail1%stpcM%i", chrgs[ichg], imult), 0.13957);
221          cutPass1YPtetaphitpc[aniter] = new AliFemtoCutMonitorParticleYPt(Form("cutPass1%stpcM%i", chrgs[ichg], imult), 0.493677);
222         cutFail1YPtetaphitpc[aniter] = new AliFemtoCutMonitorParticleYPt(Form("cutFail1%stpcM%i", chrgs[ichg], imult), 0.493677);
223          dtc1etaphitpc[aniter]->AddCutMonitor(cutPass1YPtetaphitpc[aniter], cutFail1YPtetaphitpc[aniter]);
224                                             
225         //cutPass1PIDetaphitpc[aniter] = new AliFemtoCutMonitorParticlePID(Form("cutPass1%stpcM%i", chrgs[ichg], imult),0);
226         //cutFail1PIDetaphitpc[aniter] = new AliFemtoCutMonitorParticlePID(Form("cutFail1%stpcM%i", chrgs[ichg], imult),0);
227          cutPass1PIDetaphitpc[aniter] = new AliFemtoCutMonitorParticlePID(Form("cutPass1%stpcM%i", chrgs[ichg], imult),1);
228          cutFail1PIDetaphitpc[aniter] = new AliFemtoCutMonitorParticlePID(Form("cutFail1%stpcM%i", chrgs[ichg], imult),1);
229         dtc1etaphitpc[aniter]->AddCutMonitor(cutPass1PIDetaphitpc[aniter], cutFail1PIDetaphitpc[aniter]);
230                                                         
231          dtc1etaphitpc[aniter]->SetStatus(AliESDtrack::kTPCin); ///my
232
233          //sqpcetaphitpc[aniter] = new AliFemtoPairCutAntiGamma();
234         // sqpcetaphitpc[aniter] = new AliFemtoPairCutRadialDistance();
235         //sqpcetaphitpc[aniter] = new AliFemtoShareQualityTPCEntranceSepPairCut();
236         // sqpcetaphitpc[aniter] = new AliFemtoShareQualityPairCut();
237           sqpcetaphitpc[aniter] = new AliFemtoPairCutRadialDistanceKK();
238          sqpcetaphitpc[aniter]->SetShareQualityMax(1.0);
239           sqpcetaphitpc[aniter]->SetShareFractionMax(0.05);
240          sqpcetaphitpc[aniter]->SetRemoveSameLabel(kFALSE);
241         
242           sqpcetaphitpc[aniter]->SetPhiStarDifferenceMinimum(0.04);
243           sqpcetaphitpc[aniter]->SetEtaDifferenceMinimum(0.02);
244           
245                                                     
246                   anetaphitpc[aniter]->SetEventCut(mecetaphitpc[aniter]);
247                  anetaphitpc[aniter]->SetFirstParticleCut(dtc1etaphitpc[aniter]);
248                  anetaphitpc[aniter]->SetSecondParticleCut(dtc1etaphitpc[aniter]);
249                  anetaphitpc[aniter]->SetPairCut(sqpcetaphitpc[aniter]);
250
251
252                   cqinvkttpc[aniter] = new AliFemtoQinvCorrFctn(Form("cqinv%stpcM%i", chrgs[ichg], imult),nbinssh,0.0,shqmax);
253             anetaphitpc[aniter]->AddCorrFctn(cqinvkttpc[aniter]);
254
255                               //3D cartesian (without kT bins)
256                               if(run3d){
257                     //cq3dlcmskttpc[aniter] = new AliFemtoCorrFctn3DLCMSSym(Form("cq3d%stpcM%i", chrgs[ichg], imult),60,0.5);
258                             cq3dlcmskttpc[aniter] = new AliFemtoCorrFctn3DLCMSSym(Form("cq3d%stpcM%i", chrgs[ichg], imult),100,0.5);
259                             anetaphitpc[aniter]->AddCorrFctn(cq3dlcmskttpc[aniter]);
260                                               }
261
262                           if (runktdep) {
263                       int ktm;
264                          for (int ikt=0; ikt<3; ikt++) {
265                                 ktm = aniter*2 + ikt;
266                               ktpcuts[ktm] = new AliFemtoKTPairCut(ktrng[ikt], ktrng[ikt+1]);
267               
268                               cqinvkttpc[ktm] = new AliFemtoQinvCorrFctn(Form("cqinv%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0, shqmax);
269                               cqinvkttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
270                               anetaphitpc[aniter]->AddCorrFctn(cqinvkttpc[ktm]);
271
272                               cqinvsqtpc[ktm] = new AliFemtoShareQualityCorrFctn(Form("cqinvsq%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0,shqmax);
273                                             cqinvsqtpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
274                                                   anetaphitpc[aniter]->AddCorrFctn(cqinvsqtpc[ktm]);
275
276                 cqinvinnertpc[ktm] = new AliFemtoTPCInnerCorrFctn(Form("cqinvinner%stpcM%ikT%i", chrgs[ichg], imult, ikt),nbinssh,0.0,shqmax);
277               cqinvinnertpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
278                     cqinvinnertpc[ktm]->SetRadius(-1.6);
279                                                           anetaphitpc[aniter]->AddCorrFctn(cqinvinnertpc[ktm]);
280
281                                 if (run3d) {
282                            //       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);
283                     cq3dlcmskttpc[ktm] = new AliFemtoCorrFctn3DLCMSSym(Form("cq3d%stpcM%ikT%i", chrgs[ichg], imult, ikt),60,0.5);
284                                cq3dlcmskttpc[ktm]->SetPairSelectionCut(ktpcuts[ktm]);
285                                 anetaphitpc[aniter]->AddCorrFctn(cq3dlcmskttpc[ktm]);
286                       }
287                                                           }
288                             }
289                         
290                                 cdedpetaphi[aniter] = new AliFemtoCorrFctnDEtaDPhi(Form("cdedp%stpcM%i", chrgs[ichg], imult),39, 39);
291                                   anetaphitpc[aniter]->AddCorrFctn(cdedpetaphi[aniter]);
292                 
293                                       Manager->AddAnalysis(anetaphitpc[aniter]); 
294                                               }
295       }
296     }
297   }
298   // *** End pion-pion analysis
299
300   return Manager;
301 }                         
302