]> git.uio.no Git - u/mrichter/AliRoot.git/blob - PWGJE/macros/CreateTrackCutsPWGJE.C
c4d6576a58041ca9b9e3b07286dd7d797b92eb27
[u/mrichter/AliRoot.git] / PWGJE / macros / CreateTrackCutsPWGJE.C
1 AliESDtrackCuts *CreateTrackCutsPWGJE(Int_t cutMode) {
2
3   //
4   // Macro to create track cuts for PWG Jet analysis
5   // User can select a specific set by indicating cutMode
6   // cutMode has 8 digits: first 4 digits additional cuts, last 4 digits standard cuts
7   //                       additional cuts are variations of standard cuts (used for hybrid track selection and QA)
8   // Numbering starts from 1000 For standard and additional cut numbers
9
10   AliESDtrackCuts *trackCuts  = new AliESDtrackCuts("AliESDtrackCuts");
11
12   TString tag;
13
14   Int_t mod = 10000;
15
16   Bool_t bStdCutsDefined = kFALSE;
17
18
19   //_____________________________________________________________________
20   //                     STANDARD CUTS
21
22   //Get standard cuts: last 4 digits of cutMode
23   Int_t stdCutMode = cutMode%mod;
24
25   if(stdCutMode == 1000) {
26
27     bStdCutsDefined = kTRUE;
28
29     trackCuts = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010(kTRUE,1);
30     trackCuts->SetMinNCrossedRowsTPC(120);
31     trackCuts->SetMinRatioCrossedRowsOverFindableClustersTPC(0.8);
32     trackCuts->SetMaxChi2PerClusterITS(36);
33     trackCuts->SetMaxFractionSharedTPCClusters(0.4);
34     trackCuts->SetMaxChi2TPCConstrainedGlobal(36);
35
36     trackCuts->SetEtaRange(-0.9,0.9);
37     trackCuts->SetPtRange(0.15, 1e10);
38
39     tag = "Global track RAA analysis QM2011 + Chi2ITS<36";
40
41   }
42
43   if(stdCutMode == 1001) {
44
45     bStdCutsDefined = kTRUE;
46
47     // TPC  
48     trackCuts->SetMinNClustersTPC(90);
49     trackCuts->SetMaxChi2PerClusterTPC(4);
50     trackCuts->SetRequireTPCStandAlone(kTRUE); //cut on NClustersTPC and chi2TPC Iter1
51     trackCuts->SetAcceptKinkDaughters(kFALSE);
52     trackCuts->SetRequireTPCRefit(kTRUE);
53     trackCuts->SetMaxFractionSharedTPCClusters(0.4);
54     // ITS
55     trackCuts->SetRequireITSRefit(kTRUE);
56     //accept secondaries
57     trackCuts->SetMaxDCAToVertexXY(2.4);
58     trackCuts->SetMaxDCAToVertexZ(3.2);
59     trackCuts->SetDCAToVertex2D(kTRUE);
60     //reject fakes
61     trackCuts->SetMaxChi2PerClusterITS(36);
62
63     trackCuts->SetRequireSigmaToVertex(kFALSE);
64
65     trackCuts->SetEtaRange(-0.9,0.9);
66     trackCuts->SetPtRange(0.15, 100.);
67
68  
69     tag = "Global tracks jet analysis with ITSrefit and NclsIter1=90, noSPD requirement";
70
71   }
72
73   if(stdCutMode == 1002) {
74
75     bStdCutsDefined = kTRUE;
76
77     // TPC  
78     trackCuts->SetMinNClustersTPC(80);
79     trackCuts->SetMaxChi2PerClusterTPC(4);
80     trackCuts->SetAcceptKinkDaughters(kFALSE);
81     trackCuts->SetRequireTPCRefit(kTRUE);
82     trackCuts->SetMaxFractionSharedTPCClusters(0.4);
83     // ITS
84     trackCuts->SetRequireITSRefit(kTRUE);
85     //accept secondaries
86     trackCuts->SetMaxDCAToVertexXY(2.4);
87     trackCuts->SetMaxDCAToVertexZ(3.2);
88     trackCuts->SetDCAToVertex2D(kTRUE);
89     //reject fakes
90     trackCuts->SetMaxChi2PerClusterITS(36);
91
92     trackCuts->SetRequireSigmaToVertex(kFALSE);
93
94     trackCuts->SetEtaRange(-0.9,0.9);
95     trackCuts->SetPtRange(0.15, 100.);
96
97  
98     tag = "Global tracks jet analysis with ITSrefit and Ncls=80, noSPD requirement";
99
100   }
101
102   if(stdCutMode == 1003) {
103
104     bStdCutsDefined = kTRUE;
105
106     // tight global tracks
107     trackCuts = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010(kFALSE,1);
108     trackCuts->SetMinNClustersTPC(0);
109     trackCuts->SetMinNCrossedRowsTPC(120);
110     trackCuts->SetMinRatioCrossedRowsOverFindableClustersTPC(0.1);// essentially switches it off
111     trackCuts->SetMaxDCAToVertexXY(2.4);
112     trackCuts->SetMaxDCAToVertexZ(3.2);
113     trackCuts->SetDCAToVertex2D(kTRUE);
114     trackCuts->SetMaxChi2PerClusterITS(36);
115     trackCuts->SetMaxFractionSharedTPCClusters(0.4);
116
117     tag = "Global tracks ITSTPC2010 + NCrossedRows + loose ITS";
118
119   }
120   
121   if(stdCutMode == 1004) {
122
123     bStdCutsDefined = kTRUE;
124
125     // TPC  
126     trackCuts->SetMinNClustersTPC(70);
127     trackCuts->SetMaxChi2PerClusterTPC(4);
128     trackCuts->SetRequireTPCStandAlone(kTRUE); //cut on NClustersTPC and chi2TPC Iter1
129     trackCuts->SetAcceptKinkDaughters(kFALSE);
130     trackCuts->SetRequireTPCRefit(kTRUE);
131     trackCuts->SetMaxFractionSharedTPCClusters(0.4);
132     // ITS
133     trackCuts->SetRequireITSRefit(kTRUE);
134     //accept secondaries
135     trackCuts->SetMaxDCAToVertexXY(2.4);
136     trackCuts->SetMaxDCAToVertexZ(3.2);
137     trackCuts->SetDCAToVertex2D(kTRUE);
138     //reject fakes
139     trackCuts->SetMaxChi2PerClusterITS(36);
140
141     trackCuts->SetRequireSigmaToVertex(kFALSE);
142
143     trackCuts->SetEtaRange(-0.9,0.9);
144     trackCuts->SetPtRange(0.15, 100.);
145  
146     tag = "Global tracks jet analysis with ITSrefit and NclsIter1=70, noSPD requirement";
147
148   }
149   if(stdCutMode == 1005) {
150
151     bStdCutsDefined = kTRUE;
152
153     // TPC  
154     trackCuts->SetMinNClustersTPC(70);
155     trackCuts->SetMaxChi2PerClusterTPC(4);
156     trackCuts->SetRequireTPCStandAlone(kTRUE); //cut on NClustersTPC and chi2TPC Iter1
157     trackCuts->SetAcceptKinkDaughters(kFALSE);
158     trackCuts->SetRequireTPCRefit(kTRUE);
159     trackCuts->SetMaxFractionSharedTPCClusters(0.4);
160     // ITS
161     trackCuts->SetRequireITSRefit(kTRUE);
162     //accept secondaries
163     trackCuts->SetMaxDCAToVertexXY(2.4);
164     trackCuts->SetMaxDCAToVertexZ(3.2);
165     trackCuts->SetDCAToVertex2D(kTRUE);
166     //reject fakes
167     trackCuts->SetMaxChi2PerClusterITS(36);
168
169     trackCuts->SetRequireSigmaToVertex(kFALSE);
170
171     trackCuts->SetEtaRange(-0.9,0.9);
172     trackCuts->SetPtRange(0.15, 1E+15.);
173  
174     tag = "Global tracks jet analysis with ITSrefit and NclsIter1=70, noSPD requirement, no upper pt cut";
175
176   }
177   
178   if(stdCutMode == 1006) {
179
180     bStdCutsDefined = kTRUE;
181
182     // TPC  
183     TFormula *f1NClustersTPCLinearPtDep = new TFormula("f1NClustersTPCLinearPtDep","70.+30./20.*x");
184     trackCuts->SetMinNClustersTPCPtDep(f1NClustersTPCLinearPtDep,20.);
185     trackCuts->SetMinNClustersTPC(70);
186     trackCuts->SetMaxChi2PerClusterTPC(4);
187     trackCuts->SetRequireTPCStandAlone(kTRUE); //cut on NClustersTPC and chi2TPC Iter1
188     trackCuts->SetAcceptKinkDaughters(kFALSE);
189     trackCuts->SetRequireTPCRefit(kTRUE);
190     trackCuts->SetMaxFractionSharedTPCClusters(0.4);
191     // ITS
192     trackCuts->SetRequireITSRefit(kTRUE);
193     //accept secondaries
194     trackCuts->SetMaxDCAToVertexXY(2.4);
195     trackCuts->SetMaxDCAToVertexZ(3.2);
196     trackCuts->SetDCAToVertex2D(kTRUE);
197     //reject fakes
198     trackCuts->SetMaxChi2PerClusterITS(36);
199     trackCuts->SetMaxChi2TPCConstrainedGlobal(36);
200
201     trackCuts->SetRequireSigmaToVertex(kFALSE);
202
203     trackCuts->SetEtaRange(-0.9,0.9);
204     trackCuts->SetPtRange(0.15, 1E+15.);
205  
206     tag = "Global tracks jet analysis with ITSrefit and NclsIter1=PtDep, noSPD requirement, no upper pt cut, golden chi2";
207
208   }
209
210   if(stdCutMode == 1007) {
211     
212     bStdCutsDefined = kTRUE;
213     
214     trackCuts = AliESDtrackCuts::GetStandardITSTPCTrackCuts2011(kFALSE,1);
215     //accept secondaries
216     trackCuts->SetMaxDCAToVertexXY(2.4);
217     trackCuts->SetMaxDCAToVertexZ(3.2);
218     trackCuts->SetDCAToVertex2D(kTRUE);
219
220     //
221     trackCuts->SetMaxChi2TPCConstrainedGlobal(36);
222
223     trackCuts->SetEtaRange(-0.9,0.9);
224     trackCuts->SetPtRange(0.15, 1E+15.);
225  
226     tag = "Global tracks with AliESDtrackCuts::GetStandardITSTPCTrackCuts2011(kFALSE)";
227
228   }
229
230   if(stdCutMode == 1008) {
231     
232     bStdCutsDefined = kTRUE;
233     
234     trackCuts = AliESDtrackCuts::GetStandardITSTPCTrackCuts2011(kFALSE,1);
235     //accept secondaries
236     trackCuts->SetMaxDCAToVertexXY(2.4);
237     trackCuts->SetMaxDCAToVertexZ(3.2);
238     trackCuts->SetDCAToVertex2D(kTRUE);
239
240     //
241     trackCuts->SetMaxChi2TPCConstrainedGlobal(36);
242     trackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kNone);
243
244     trackCuts->SetMaxFractionSharedTPCClusters(0.4);
245
246     trackCuts->SetEtaRange(-0.9,0.9);
247     trackCuts->SetPtRange(0.15, 1E+15.);
248  
249     tag = "Global tracks 2011 with NCrossedRows cut";
250
251   }
252
253
254   if(stdCutMode == 2000) {
255
256     bStdCutsDefined = kTRUE;
257
258     // TPC  
259     trackCuts->SetMinNClustersTPC(90);
260     trackCuts->SetMaxChi2PerClusterTPC(4);
261     trackCuts->SetRequireTPCStandAlone(kTRUE); //cut on NClustersTPC and chi2TPC Iter1
262     trackCuts->SetAcceptKinkDaughters(kFALSE);
263     trackCuts->SetRequireTPCRefit(kTRUE);
264     trackCuts->SetMaxFractionSharedTPCClusters(0.4);
265     //accept secondaries
266     trackCuts->SetMaxDCAToVertexXY(2.4);
267     trackCuts->SetMaxDCAToVertexZ(3.2);
268     trackCuts->SetDCAToVertex2D(kTRUE);
269
270     trackCuts->SetRequireSigmaToVertex(kFALSE);
271
272     trackCuts->SetEtaRange(-0.9,0.9);
273     trackCuts->SetPtRange(0.15, 100.);
274
275  
276     tag = "Global tracks jet analysis, loose cuts, NClsIter1=90, no ITS requirements";
277
278   }
279
280   if(stdCutMode == 2001) {
281
282     bStdCutsDefined = kTRUE;
283
284     trackCuts = AliESDtrackCuts::GetStandardTPCOnlyTrackCuts(); 
285     // trackCuts->SetRequireTPCRefit(kTRUE);
286     trackCuts->SetMinNClustersTPC(70);
287
288     trackCuts->SetEtaRange(-0.9,0.9);
289     trackCuts->SetPtRange(0.15, 100.);
290
291  
292     tag = "TPConly track cuts, loose cuts, NCls=70, no ITS requirements";
293
294   }
295
296   if(stdCutMode == 2002) {
297
298     bStdCutsDefined = kTRUE;
299
300     trackCuts = AliESDtrackCuts::GetStandardTPCOnlyTrackCuts(); 
301     //  trackCuts->SetRequireTPCRefit(kTRUE);
302     trackCuts->SetMinNClustersTPC(0);
303     trackCuts->SetMinNCrossedRowsTPC(120);
304     trackCuts->SetMinRatioCrossedRowsOverFindableClustersTPC(0.1);// essentially switches it off  
305
306     trackCuts->SetEtaRange(-0.9,0.9);
307     trackCuts->SetPtRange(0.15, 100.);
308
309  
310     tag = "TPConly track cuts, loose cuts, NCls=70, no ITS requirements";
311
312   }
313
314   if(!bStdCutsDefined) {
315     printf("last 4 digits do not represent a predefined set of standard cuts. Returning 0\n");
316     return 0;
317
318   }
319
320
321   //_____________________________________________________________________
322   //                     ADDITIONAL CUTS
323
324   //Get additional cut mode: first 4 digits of cutMode
325   Int_t addCutMode = (int)((float)cutMode/(float)mod);
326
327   if(addCutMode == 1000) {
328
329     trackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kAny);
330  
331     tag += " + additonal: SPD any requirement";
332
333   }
334
335   if(addCutMode == 1001) {
336
337     trackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kNone);
338  
339     tag += " + additional: w/o hits in SPD";
340
341   }
342
343   if(addCutMode == 1002) {
344
345     trackCuts->SetMaxChi2PerClusterITS(1E10);
346
347     tag += " + additional: maxITSChi2=1e10";
348
349   }
350
351   if(addCutMode == 1003) {
352
353     trackCuts->SetMinNClustersTPC(0);
354     trackCuts->SetMinNCrossedRowsTPC(0);
355     trackCuts->SetMinRatioCrossedRowsOverFindableClustersTPC(0.);
356
357     tag += " + additional: minClusters=0 minCrossedRows=0 minCrossedRowsOverFindable=0";
358
359   }
360
361   if(addCutMode == 1004) {
362
363     trackCuts->SetRequireITSRefit(kFALSE);
364
365     tag += " + additional: ITSrefit=kFALSE";
366
367   }
368
369   Printf("Created track cuts for: %s", tag.Data());
370
371   return trackCuts;
372
373 }