1 - Rename AliAnaMaker and AliAnaBaseClass to AliAnaPartCorrMaker and AliAnaPartCorrB...
[u/mrichter/AliRoot.git] / PWG4 / AliFidutialCut.cxx
1 /**************************************************************************
2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3  *                                                                        *
4  * Author: The ALICE Off-line Project.                                    *
5  * Contributors are mentioned in the code where appropriate.              *
6  *                                                                        *
7  * Permission to use, copy, modify and distribute this software and its   *
8  * documentation strictly for non-commercial purposes is hereby granted   *
9  * without fee, provided that the above copyright notice appears in all   *
10  * copies and that both the copyright notice and this permission notice   *
11  * appear in the supporting documentation. The authors make no claims     *
12  * about the suitability of this software for any purpose. It is          *
13  * provided "as is" without express or implied warranty.                  *
14  **************************************************************************/
15 /* $Id: AliFidutialCut.cxx 21839 2007-10-29 13:49:42Z gustavo $ */
16
17 //_________________________________________________________________________
18 // Class for track/cluster acceptance selection
19 // Selection in Central barrel, EMCAL and PHOS
20 //                
21 //*-- Author: Gustavo Conesa (LNF-INFN) 
22 //////////////////////////////////////////////////////////////////////////////
23
24
25 // --- ROOT system ---
26 #include <TMath.h>
27 #include <TLorentzVector.h>
28 #include <TString.h>
29 //#include <TArrayF.h>
30   
31 //---- ANALYSIS system ----
32 #include "AliLog.h"
33 #include "AliFidutialCut.h"
34
35 ClassImp(AliFidutialCut)
36
37
38 //____________________________________________________________________________
39 AliFidutialCut::AliFidutialCut() : 
40   TObject(),
41   fEMCALFidutialCut(0),  fPHOSFidutialCut(0),  fCTSFidutialCut(0),
42   fCTSFidCutMinEta(0x0),fCTSFidCutMinPhi(0x0),fCTSFidCutMaxEta(0x0), fCTSFidCutMaxPhi(0x0),
43   fEMCALFidCutMinEta(0x0),fEMCALFidCutMinPhi(0x0),fEMCALFidCutMaxEta(0x0), fEMCALFidCutMaxPhi(0x0),
44   fPHOSFidCutMinEta(0x0),fPHOSFidCutMinPhi(0x0),fPHOSFidCutMaxEta(0x0), fPHOSFidCutMaxPhi(0x0)
45
46 {
47   //Ctor
48
49   //Initialize parameters
50   InitParameters();
51 }
52
53 //____________________________________________________________________________
54 AliFidutialCut::AliFidutialCut(const AliFidutialCut & g) :   
55   TObject(g), 
56   fEMCALFidutialCut(g.fEMCALFidutialCut),  fPHOSFidutialCut(g.fPHOSFidutialCut),  fCTSFidutialCut(g. fCTSFidutialCut),
57   fCTSFidCutMinEta(g.fCTSFidCutMinEta?new TArrayF(*g.fCTSFidCutMinEta):0x0),
58   fCTSFidCutMinPhi(g.fCTSFidCutMinPhi?new TArrayF(*g.fCTSFidCutMinPhi):0x0),
59   fCTSFidCutMaxEta(g.fCTSFidCutMaxEta?new TArrayF(*g.fCTSFidCutMaxEta):0x0),
60   fCTSFidCutMaxPhi(g.fCTSFidCutMaxPhi?new TArrayF(*g.fCTSFidCutMaxPhi):0x0),
61   fEMCALFidCutMinEta(g.fEMCALFidCutMinEta?new TArrayF(*g.fEMCALFidCutMinEta):0x0),
62   fEMCALFidCutMinPhi(g.fEMCALFidCutMinPhi?new TArrayF(*g.fEMCALFidCutMinPhi):0x0),
63   fEMCALFidCutMaxEta(g.fEMCALFidCutMaxEta?new TArrayF(*g.fEMCALFidCutMaxEta):0x0),
64   fEMCALFidCutMaxPhi(g.fEMCALFidCutMaxPhi?new TArrayF(*g.fEMCALFidCutMaxPhi):0x0),
65   fPHOSFidCutMinEta(g.fPHOSFidCutMinEta?new TArrayF(*g.fPHOSFidCutMinEta):0x0),
66   fPHOSFidCutMinPhi(g.fPHOSFidCutMinPhi?new TArrayF(*g.fPHOSFidCutMinPhi):0x0),
67   fPHOSFidCutMaxEta(g.fPHOSFidCutMaxEta?new TArrayF(*g.fPHOSFidCutMaxEta):0x0),
68   fPHOSFidCutMaxPhi(g.fPHOSFidCutMaxPhi?new TArrayF(*g.fPHOSFidCutMaxPhi):0x0)
69
70 {
71   // cpy ctor
72
73 }
74
75 //_________________________________________________________________________
76 AliFidutialCut & AliFidutialCut::operator = (const AliFidutialCut & source)
77 {
78   // assignment operator
79   
80   if(&source == this) return *this;
81   
82   fEMCALFidutialCut = source.fEMCALFidutialCut;  
83   fPHOSFidutialCut = source.fPHOSFidutialCut;
84   fCTSFidutialCut = source.fCTSFidutialCut;
85   
86   fCTSFidCutMinEta = source.fCTSFidCutMinEta?new TArrayF(*source.fCTSFidCutMinEta):0x0;
87   fCTSFidCutMinPhi = source.fCTSFidCutMinPhi?new TArrayF(*source.fCTSFidCutMinPhi):0x0;
88   fCTSFidCutMaxEta = source.fCTSFidCutMaxEta?new TArrayF(*source.fCTSFidCutMaxEta):0x0;
89   fCTSFidCutMaxPhi = source.fCTSFidCutMaxPhi?new TArrayF(*source.fCTSFidCutMaxPhi):0x0;
90   fEMCALFidCutMinEta = source.fEMCALFidCutMinEta?new TArrayF(*source.fEMCALFidCutMinEta):0x0;
91   fEMCALFidCutMinPhi = source.fEMCALFidCutMinPhi?new TArrayF(*source.fEMCALFidCutMinPhi):0x0;
92   fEMCALFidCutMaxEta = source.fEMCALFidCutMaxEta?new TArrayF(*source.fEMCALFidCutMaxEta):0x0;
93   fEMCALFidCutMaxPhi = source.fEMCALFidCutMaxPhi?new TArrayF(*source.fEMCALFidCutMaxPhi):0x0;
94   fPHOSFidCutMinEta = source.fPHOSFidCutMinEta?new TArrayF(*source.fPHOSFidCutMinEta):0x0;
95   fPHOSFidCutMinPhi = source.fPHOSFidCutMinPhi?new TArrayF(*source.fPHOSFidCutMinPhi):0x0;
96   fPHOSFidCutMaxEta = source.fPHOSFidCutMaxEta?new TArrayF(*source.fPHOSFidCutMaxEta):0x0;
97   fPHOSFidCutMaxPhi = source.fPHOSFidCutMaxPhi?new TArrayF(*source.fPHOSFidCutMaxPhi):0x0;
98
99   return *this;
100
101 }
102
103 //_________________________________
104 AliFidutialCut::~AliFidutialCut() {
105   //Dtor
106
107   delete fCTSFidCutMinEta ;
108   delete fCTSFidCutMinPhi ;
109   delete fCTSFidCutMaxEta ;
110   delete fCTSFidCutMaxPhi ;
111   
112   delete fEMCALFidCutMinEta ;
113   delete fEMCALFidCutMinPhi ;
114   delete fEMCALFidCutMaxEta ; 
115   delete fEMCALFidCutMaxPhi ;
116   
117   delete fPHOSFidCutMinEta ; 
118   delete fPHOSFidCutMinPhi ; 
119   delete fPHOSFidCutMaxEta ;
120   delete fPHOSFidCutMaxPhi ;
121
122 }
123
124
125 //_______________________________________________________________
126 Bool_t AliFidutialCut::IsInFidutialCut(TLorentzVector momentum, TString det) const
127 {
128   //Selects EMCAL or PHOS cluster or CTS track if it is inside eta-phi defined regions
129
130   Double_t phi = momentum.Phi();
131   if(phi<0) phi+=TMath::TwoPi() ;
132   Double_t eta =  momentum.Eta();
133   //printf("IsInFidutialCut::Det: %s, phi = %f, eta = %f\n", det.Data(),phi*TMath::RadToDeg(), eta);
134
135   Float_t * maxeta = new Float_t;
136   Float_t * maxphi = new Float_t;
137   Float_t * mineta = new Float_t;
138   Float_t * minphi = new Float_t;
139   Int_t nphiregions = 0;
140   Int_t netaregions = 0;
141   Bool_t selection = kFALSE;
142
143   if(det == "CTS"){
144     netaregions =  fCTSFidCutMaxEta->GetSize();
145     nphiregions =  fCTSFidCutMaxPhi->GetSize();
146     if(netaregions !=  fCTSFidCutMinEta->GetSize() || nphiregions !=  fCTSFidCutMinPhi->GetSize())
147       AliFatal(Form("Wrong number of CTS fidutial cut regions: nmaxeta %d != nmineta %d; nmaxphi %d != nminphi %d",
148                     netaregions, fCTSFidCutMinEta->GetSize(),  nphiregions, fCTSFidCutMinPhi->GetSize()));
149     
150     maxeta = fCTSFidCutMaxEta->GetArray();
151     maxphi = fCTSFidCutMaxPhi->GetArray();
152     mineta = fCTSFidCutMinEta->GetArray();
153     minphi = fCTSFidCutMinPhi->GetArray();
154     selection =  fCTSFidutialCut ; 
155   }
156   else   if(det == "EMCAL"){
157     netaregions =  fEMCALFidCutMaxEta->GetSize();
158     nphiregions =  fEMCALFidCutMaxPhi->GetSize();
159     if(netaregions !=  fEMCALFidCutMinEta->GetSize() || nphiregions !=  fEMCALFidCutMinPhi->GetSize())
160       AliFatal(Form("Wrong number of EMCAL fidutial cut regions: nmaxeta %d != nmineta %d; nmaxphi %d != nminphi %d",
161                     netaregions, fEMCALFidCutMinEta->GetSize(),  nphiregions, fEMCALFidCutMinPhi->GetSize()));
162     
163     maxeta = fEMCALFidCutMaxEta->GetArray();
164     maxphi = fEMCALFidCutMaxPhi->GetArray();
165     mineta = fEMCALFidCutMinEta->GetArray();
166     minphi = fEMCALFidCutMinPhi->GetArray();
167     selection =  fEMCALFidutialCut ; 
168   }
169   else   if(det == "PHOS"){
170     netaregions =  fPHOSFidCutMaxEta->GetSize();
171     nphiregions =  fPHOSFidCutMaxPhi->GetSize();
172     if(netaregions !=  fPHOSFidCutMinEta->GetSize() || nphiregions !=  fPHOSFidCutMinPhi->GetSize())
173       AliFatal(Form("Wrong number of PHOS fidutial cut regions: nmaxeta %d != nmineta %d; nmaxphi %d != nminphi %d",
174                     netaregions, fPHOSFidCutMinEta->GetSize(),  nphiregions, fPHOSFidCutMinPhi->GetSize()));
175     
176     maxeta = fPHOSFidCutMaxEta->GetArray();
177     maxphi = fPHOSFidCutMaxPhi->GetArray();
178     mineta = fPHOSFidCutMinEta->GetArray();
179     minphi = fPHOSFidCutMinPhi->GetArray();
180     selection =  fPHOSFidutialCut ; 
181   }
182   else
183     AliFatal(Form("Wrong detector name = %s", det.Data()));
184
185   //printf("IsInFidutialCut::nphiregions = %d, netaregions = %d\n",nphiregions, netaregions);
186
187   if(!selection) return kTRUE; //No cuts applied, all tracks/clusters used
188
189   //Eta fidutial cut
190   Bool_t bInEtaFidCut = kFALSE;
191   for(Int_t ieta = 0; ieta < netaregions; ieta++)
192     if(eta > mineta[ieta] && eta < maxeta[ieta]) bInEtaFidCut = kTRUE;
193
194   if(bInEtaFidCut){
195     //printf("Eta cut passed\n");
196     //Phi fidutial cut
197     Bool_t bInPhiFidCut = kFALSE;
198     for(Int_t iphi = 0; iphi < nphiregions; iphi++)
199       if(phi > minphi[iphi] *TMath::DegToRad()&& phi < maxphi[iphi]*TMath::DegToRad()) bInPhiFidCut = kTRUE ;
200     
201     if(bInPhiFidCut) {
202       //printf("IsInFidutialCut:: %s cluster/track accepted\n",det.Data());
203       return kTRUE;
204     }
205     else return kFALSE;
206     
207   }//In eta fid cut
208   else
209     return kFALSE;
210   
211 }
212
213
214 //_______________________________________________________________
215 void AliFidutialCut::InitParameters()
216 {
217  
218   //Initialize the parameters of the analysis.
219
220   fEMCALFidutialCut = kTRUE ;  
221   fPHOSFidutialCut = kTRUE ;
222   fCTSFidutialCut = kTRUE ;
223
224   fCTSFidCutMinEta = new TArrayF(1);
225   fCTSFidCutMinEta->SetAt(-0.9,0); 
226   fCTSFidCutMaxEta = new TArrayF(1);
227   fCTSFidCutMaxEta->SetAt(0.9,0); 
228
229   fCTSFidCutMinPhi = new TArrayF(1);
230   fCTSFidCutMinPhi->SetAt(0,0); 
231   fCTSFidCutMaxPhi = new TArrayF(1);
232   fCTSFidCutMaxPhi->SetAt(360.,0); 
233
234   fEMCALFidCutMinEta = new TArrayF(1);
235   fEMCALFidCutMinEta->SetAt(-0.7,0); 
236   fEMCALFidCutMaxEta = new TArrayF(1);
237   fEMCALFidCutMaxEta->SetAt(0.7,0); 
238
239   fEMCALFidCutMinPhi = new TArrayF(1);
240   fEMCALFidCutMinPhi->SetAt(80.,0); 
241   fEMCALFidCutMaxPhi = new TArrayF(1);
242   fEMCALFidCutMaxPhi->SetAt(190.,0); 
243
244   fPHOSFidCutMinEta = new TArrayF(1);
245   fPHOSFidCutMinEta->SetAt(-0.13,0); 
246   fPHOSFidCutMaxEta = new TArrayF(1);
247   fPHOSFidCutMaxEta->SetAt(0.13,0); 
248
249   fPHOSFidCutMinPhi = new TArrayF(1);
250   fPHOSFidCutMinPhi->SetAt(220.,0); 
251   fPHOSFidCutMaxPhi = new TArrayF(1);
252   fPHOSFidCutMaxPhi->SetAt(320.,0); 
253
254 }
255
256
257 //________________________________________________________________
258 void AliFidutialCut::Print(const Option_t * opt) const
259 {
260
261   //Print some relevant parameters set for the analysis
262   if(! opt)
263     return;
264
265   printf("***** Print: %s %s ******\n", GetName(), GetTitle() ) ;
266
267   if(fCTSFidutialCut){
268     Int_t netaregions =  fCTSFidCutMaxEta->GetSize();
269     Int_t nphiregions =  fCTSFidCutMaxPhi->GetSize();
270     printf(">> CTS Fidutial regions : phi %d eta %d\n", netaregions, nphiregions) ;
271     for(Int_t ieta = 0; ieta < netaregions; ieta++)
272       printf(" region %d : %3.2f < eta < %3.2f\n", ieta, fCTSFidCutMinEta->GetAt(ieta), fCTSFidCutMaxEta->GetAt(ieta)) ;
273     for(Int_t iphi = 0; iphi < nphiregions; iphi++)
274       printf(" region %d : %3.1f < phi < %3.1f\n", iphi, fCTSFidCutMinPhi->GetAt(iphi), fCTSFidCutMaxPhi->GetAt(iphi)) ; 
275   }
276   else printf(">>No fidutial cuts in CTS\n");
277
278   if(fEMCALFidutialCut){
279     Int_t netaregions =  fEMCALFidCutMaxEta->GetSize();
280     Int_t nphiregions =  fEMCALFidCutMaxPhi->GetSize();
281     printf(">>EMCAL Fidutial regions : phi %d eta %d\n", netaregions, nphiregions) ;
282     for(Int_t ieta = 0; ieta < netaregions; ieta++)
283       printf(" region %d : %3.2f < eta < %3.2f\n", ieta, fEMCALFidCutMinEta->GetAt(ieta), fEMCALFidCutMaxEta->GetAt(ieta)) ;
284     for(Int_t iphi = 0; iphi < nphiregions; iphi++)
285       printf(" region %d : %3.1f < phi < %3.1f\n", iphi, fEMCALFidCutMinPhi->GetAt(iphi), fEMCALFidCutMaxPhi->GetAt(iphi)) ; 
286   }
287   else printf(">>No fidutial cuts in EMCAL\n");
288
289   if(fPHOSFidutialCut){
290     Int_t netaregions =  fPHOSFidCutMaxEta->GetSize();
291     Int_t nphiregions =  fPHOSFidCutMaxPhi->GetSize();
292     printf(">>PHOS Fidutial regions : phi %d eta %d\n", netaregions, nphiregions) ;
293     for(Int_t ieta = 0; ieta < netaregions; ieta++)
294       printf(" region %d : %3.2f < eta < %3.2f\n", ieta, fPHOSFidCutMinEta->GetAt(ieta), fPHOSFidCutMaxEta->GetAt(ieta)) ;
295     for(Int_t iphi = 0; iphi < nphiregions; iphi++)
296       printf(" region %d : %3.1f < phi < %3.1f\n", iphi, fPHOSFidCutMinPhi->GetAt(iphi), fPHOSFidCutMaxPhi->GetAt(iphi)) ; 
297   }
298   else printf(">>No fidutial cuts in PHOS\n");
299   printf("    \n") ;
300
301
302
303 //_______________________________________________________________
304 void AliFidutialCut::SetSimpleCTSFidutialCut(const Float_t eta, const Float_t minphi, const Float_t maxphi){
305
306   //Method to set simple acceptance cut to CTS
307   fCTSFidCutMinEta->Set(1);
308   fCTSFidCutMaxEta->Set(1);
309   fCTSFidCutMinPhi->Set(1);
310   fCTSFidCutMaxPhi->Set(1);
311
312   fCTSFidCutMinEta->SetAt(-eta,0);
313   fCTSFidCutMaxEta->SetAt(eta,0);
314   fCTSFidCutMinPhi->SetAt(minphi,0);
315   fCTSFidCutMaxPhi->SetAt(maxphi,0);
316
317
318 }
319
320
321 //_______________________________________________________________
322 void AliFidutialCut::SetSimpleEMCALFidutialCut(const Float_t eta, const Float_t minphi, const Float_t maxphi){
323   //Method to set simple acceptance cut to EMCAL
324
325   fEMCALFidCutMinEta->Set(1);
326   fEMCALFidCutMaxEta->Set(1);
327   fEMCALFidCutMinPhi->Set(1);
328   fEMCALFidCutMaxPhi->Set(1);
329
330   fEMCALFidCutMinEta->SetAt(-eta,0);
331   fEMCALFidCutMaxEta->SetAt(eta,0);
332   fEMCALFidCutMinPhi->SetAt(minphi,0);
333   fEMCALFidCutMaxPhi->SetAt(maxphi,0);
334
335
336 }
337
338 //_______________________________________________________________
339 void AliFidutialCut::SetSimplePHOSFidutialCut(const Float_t eta, const Float_t minphi, const Float_t maxphi){
340
341   //Method to set simple acceptance cut to PHOS
342   fPHOSFidCutMinEta->Set(1);
343   fPHOSFidCutMaxEta->Set(1);
344   fPHOSFidCutMinPhi->Set(1);
345   fPHOSFidCutMaxPhi->Set(1);
346
347   fPHOSFidCutMinEta->SetAt(-eta,0);
348   fPHOSFidCutMaxEta->SetAt(eta,0);
349   fPHOSFidCutMinPhi->SetAt(minphi,0);
350   fPHOSFidCutMaxPhi->SetAt(maxphi,0);
351
352
353 }