23108bcb0fcdfb92c4b43129bb3572869374e241
[u/mrichter/AliRoot.git] / EMCAL / AliEMCALDigit.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
16 /* $Id:  */
17
18 //_________________________________________________________________________
19 //  EMCAL digit: 
20 //      A Digit is the sum of the energy lost in an EMCAL Tower
21 //      It also stores information on Primary, and enterring particle
22 //      tracknumbers Digits are created using AliEMCALSDigitizer, followed
23 //      by AliEMCALDigitizer 
24 //
25 //*-- Author: Sahal Yacoob (LBL)
26 // based on : AliPHOSDigit
27 //__________________________________________________________________________
28
29 // --- ROOT system ---
30
31 // --- Standard library ---
32
33 #include <iostream.h>
34
35 // --- AliRoot header files ---
36
37 #include "AliEMCALDigit.h"
38 #include "AliEMCALGeometry.h"
39 #include "AliEMCALGetter.h"
40
41
42 ClassImp(AliEMCALDigit)
43
44 //____________________________________________________________________________
45   AliEMCALDigit::AliEMCALDigit() 
46 {
47   // default ctor 
48
49   fIndexInList = -1 ; 
50   fNprimary    = 0 ;  
51   fNMaxPrimary = 20 ; 
52   fNiparent     = 0 ;
53   fNMaxiparent = fNMaxPrimary*10;
54   fPrimary = new Int_t[fNMaxPrimary] ;
55   fIparent = new Int_t[fNMaxiparent] ; 
56 }
57
58 //____________________________________________________________________________
59 AliEMCALDigit::AliEMCALDigit(Int_t primary, Int_t iparent, Int_t id, Int_t DigEnergy, Float_t time, Int_t index) 
60 {  
61   // ctor with all data 
62
63   fNMaxPrimary = 20 ; 
64   fNMaxiparent = fNMaxPrimary*10;
65   fPrimary = new Int_t[fNMaxPrimary] ;
66   fIparent = new Int_t[fNMaxiparent] ; 
67   fAmp         = DigEnergy ;
68   fTime        = time ;
69   fId          = id ;
70   fIndexInList = index ; 
71   if( primary != -1){
72     fNprimary    = 1 ; 
73     fPrimary[0]  = primary ;  
74     fNiparent   = 1 ;
75     fIparent[0] = iparent ;    
76 }
77   else{  //If the contribution of this primary smaller than fDigitThreshold (AliEMCALv1)
78     fNprimary    = 0 ; 
79     fPrimary[0]  = -1 ;
80     fNiparent   = 0 ;
81     fIparent[0] = -1 ;    
82
83   }
84   Int_t i ;
85   for ( i = 1; i < fNMaxPrimary ; i++)
86     fPrimary[i]  = -1 ; 
87
88   for ( Int_t j =1; j< fNMaxiparent ; j++)
89     fIparent[j] = -1 ;  
90 }
91
92 //____________________________________________________________________________
93 AliEMCALDigit::AliEMCALDigit(const AliEMCALDigit & digit) 
94 {
95   // copy ctor
96   
97
98   fNMaxPrimary = digit.fNMaxPrimary ;  
99   fNMaxiparent = digit.fNMaxiparent ;
100   fPrimary = new Int_t[fNMaxPrimary] ;
101   fIparent = new Int_t[fNMaxiparent] ; 
102   Int_t i ;
103   for ( i = 0; i < fNMaxPrimary ; i++)
104   fPrimary[i]  = digit.fPrimary[i] ;
105   Int_t j ;
106   for (j = 0; j< fNMaxiparent ; j++)
107   fIparent[j]  = digit.fIparent[j] ;
108   fAmp         = digit.fAmp ;
109   fTime        = digit.fTime ;
110   fId          = digit.fId;
111   fIndexInList = digit.fIndexInList ; 
112   fNprimary    = digit.fNprimary ;
113   fNiparent    = digit.fNiparent ;
114 }
115
116 //____________________________________________________________________________
117 AliEMCALDigit::~AliEMCALDigit() 
118 {
119   // Delete array of primiries if any
120     delete [] fPrimary ;
121     delete [] fIparent ; 
122 }
123
124 //____________________________________________________________________________
125 Int_t AliEMCALDigit::Compare(const TObject * obj) const
126 {
127   // Compares two digits with respect to its Id
128   // to sort according increasing Id
129
130   Int_t rv ;
131
132   AliEMCALDigit * digit = (AliEMCALDigit *)obj ; 
133
134   Int_t iddiff = fId - digit->GetId() ; 
135
136   if ( iddiff > 0 ) 
137     rv = 1 ;
138   else if ( iddiff < 0 )
139     rv = -1 ; 
140   else
141     rv = 0 ;
142   
143   return rv ; 
144
145 }
146
147 //____________________________________________________________________________
148 const Float_t AliEMCALDigit::GetEta() const
149 {
150   Float_t eta=-10., phi=-10.;
151   Int_t id = GetId();
152   AliEMCALGeometry *g = AliEMCALGetter::GetInstance()->EMCALGeometry();
153   g->EtaPhiFromIndex(id,eta,phi);
154   return eta ;
155 }
156
157 //____________________________________________________________________________
158 const Float_t AliEMCALDigit::GetPhi() const
159 {
160   Float_t eta=-10., phi=-10.;
161   Int_t id = GetId();
162   AliEMCALGeometry *g = AliEMCALGetter::GetInstance()->EMCALGeometry();
163   g->EtaPhiFromIndex(id,eta,phi);
164   return phi ;
165 }
166
167 //____________________________________________________________________________
168 Int_t AliEMCALDigit::GetPrimary(Int_t index) const
169 {
170   // retrieves the primary particle number given its index in the list 
171   Int_t rv = -1 ;
172   if ( index <= fNprimary && index > 0){
173     rv = fPrimary[index-1] ;
174   } 
175
176   return rv ; 
177   
178 }
179
180 //____________________________________________________________________________
181 Int_t AliEMCALDigit::GetIparent(Int_t index) const
182 {
183   // retrieves the primary particle number given its index in the list 
184   Int_t rv = -1 ;
185   if ( index <= fNiparent ){
186     rv = fIparent[index-1] ;
187   } 
188
189   return rv ; 
190   
191 }
192
193 //______________________________________________________________________
194 const Bool_t AliEMCALDigit::IsInPreShower() const 
195 {
196   Bool_t rv = kFALSE ;
197   const AliEMCALGeometry * geom = AliEMCALGetter::GetInstance()->EMCALGeometry() ;
198   if( GetId() > (geom->GetNZ() * geom->GetNPhi() )) 
199     rv = kTRUE; 
200   return rv; 
201
202
203 //____________________________________________________________________________
204 void AliEMCALDigit::ShiftPrimary(Int_t shift){
205   //shifts primary nimber to BIG offset, to separate primary in different TreeK
206   Int_t index  ;
207   for(index = 0; index <fNprimary; index++ ){
208     fPrimary[index] = fPrimary[index]+ shift * 10000000   ;}
209   for(index =0; index <fNiparent; index++){
210     fIparent[index] = fIparent[index] + shift * 10000000 ;}
211 }
212 //____________________________________________________________________________
213 Bool_t AliEMCALDigit::operator==(AliEMCALDigit const & digit) const 
214 {
215   // Two digits are equal if they have the same Id
216   
217   if ( fId == digit.fId ) 
218     return kTRUE ;
219   else 
220     return kFALSE ;
221 }
222  
223 //____________________________________________________________________________
224 AliEMCALDigit& AliEMCALDigit::operator+(AliEMCALDigit const & digit) 
225 {
226   // Adds the amplitude of digits and completes the list of primary particles
227   // if amplitude is larger than 
228   
229   fAmp += digit.fAmp ;
230   if(fTime > digit.fTime)
231     fTime = digit.fTime ;
232
233   Int_t max1 = fNprimary ; 
234   Int_t max2 = fNiparent ;  
235   Int_t index ; 
236   for (index = 0 ; index < digit.fNprimary  ; index++){
237     Bool_t deja = kTRUE ;
238     Int_t old ;
239     for ( old = 0 ; (old < max1) && deja; old++) { //already have this primary?
240       if(fPrimary[old] == digit.fPrimary[index])
241         deja = kFALSE;
242     }
243     if(deja){
244       if(max1<fNMaxPrimary){ fPrimary[max1] = digit.fPrimary[index] ; 
245       fNprimary++ ;
246       max1++;}
247       if(fNprimary==fNMaxPrimary) {
248       Int_t printindex ;
249       for (printindex = 0 ; printindex < max1 ; printindex++)
250         cout << "printindex = " << printindex << "  primary = " << fPrimary[printindex];
251       cout <<endl;
252       cout << "AliEMCALDigit >> Increase NMaxPrimary "<< endl ;
253 //      return *this ;
254       }
255     }
256   }
257   
258   for (index = 0 ; index < digit.fNiparent ; index++){
259     Bool_t dejavu = kTRUE ;
260     Int_t old ;
261     for ( old = 0 ; (old < max2) && dejavu; old++) { //already have this primary?
262       if(fIparent[old] == digit.fIparent[index])
263         dejavu = kFALSE;
264     }
265     if(dejavu){
266      if(max2<fNMaxiparent){ fIparent[max2] = digit.fIparent[index] ; 
267       fNiparent++ ;
268       max2++;}
269       if(fNiparent==fNMaxiparent) {
270         Int_t printindex ;
271         for (printindex = 0 ; printindex < max1 ; printindex++)
272           cout << "printindex = " << printindex << "  primary = " << fPrimary[printindex];
273         cout <<endl;
274         cout << "AliEMCALDigit >> Increase NMaxiparent "<< endl ;
275         return *this ;
276       }
277     }
278   }
279   
280   return *this ;
281 }
282
283 //____________________________________________________________________________
284 AliEMCALDigit& AliEMCALDigit::operator*(Float_t factor) 
285 {
286   // Multiplies the amplitude by a factor
287   
288   Float_t tempo = static_cast<Float_t>(fAmp) ; 
289   tempo *= factor ; 
290   fAmp = static_cast<Int_t>(TMath::Ceil(tempo)) ; 
291   return *this ;
292 }
293
294 //____________________________________________________________________________
295 ostream& operator << ( ostream& out , const AliEMCALDigit & digit)
296 {
297   // Prints the data of the digit
298   
299   out << "ID " << digit.fId << " Energy = " << digit.fAmp <<  " Time = " << digit.fTime << endl ; 
300   Int_t i,j ;
301   for(i=0;i<digit.fNprimary;i++)
302     out << "Primary " << i+1 << " = " << digit.fPrimary[i] << endl ;
303    
304   for(j=0;j<digit.fNiparent;j++)
305     out << "Iparent " << j+1 << " = " << digit.fIparent[j] << endl ;
306  out << "Position in list = " << digit.fIndexInList << endl ; 
307   return out ;
308 }
309
310