Changed operator+ to make sure that fTimeR is time of earliest hit
[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 #include <Riostream.h>
31 #include <TMath.h>
32
33 // --- Standard library ---
34
35 // --- AliRoot header files ---
36
37 #include "AliEMCALDigit.h"
38 #include "AliEMCALGeometry.h"
39
40 ClassImp(AliEMCALDigit)
41
42 //____________________________________________________________________________
43 AliEMCALDigit::AliEMCALDigit() :  
44   AliDigitNew(),
45   fNprimary(0),
46   fNMaxPrimary(5),
47   fPrimary(0x0),
48   fDEPrimary(0x0),
49   fNiparent(0),
50   fNMaxiparent(5), 
51   fIparent(0x0),
52   fDEParent(0x0),
53   fMaxIter(0),
54   fTime(0.), 
55   fTimeR(0.) 
56
57 {
58   // default ctor 
59
60   // Need to initialise for reading old files
61   fDEPrimary = new Float_t[fNMaxPrimary] ;
62   fDEParent = new Float_t[fNMaxiparent] ; 
63   for ( Int_t i = 0; i < fNMaxPrimary ; i++) {
64     fDEPrimary[i]  = 0 ;
65   } 
66
67   for ( Int_t i = 0; i< fNMaxiparent ; i++) {
68     fDEParent[i] = 0 ;
69   }
70 }
71
72 //____________________________________________________________________________
73 AliEMCALDigit::AliEMCALDigit(Int_t primary, Int_t iparent, Int_t id, Int_t DigEnergy, Float_t time, Int_t index, Float_t dE) 
74   : AliDigitNew(),
75     fNprimary(0),
76     fNMaxPrimary(25),
77     fPrimary(0x0),
78     fDEPrimary(0x0),
79     fNiparent(0),
80     fNMaxiparent(150),
81     fIparent(0x0),
82     fDEParent(0x0),
83     fMaxIter(5),
84     fTime(time),
85     fTimeR(time)
86 {  
87   // ctor with all data 
88
89   // data memebrs of the base class (AliNewDigit)
90   fAmp         = DigEnergy ;
91   fId          = id ;
92   fIndexInList = index ; 
93
94   // data members
95   fPrimary = new Int_t[fNMaxPrimary] ;
96   fDEPrimary = new Float_t[fNMaxPrimary] ;
97   fIparent = new Int_t[fNMaxiparent] ; 
98   fDEParent = new Float_t[fNMaxiparent] ; 
99   if( primary != -1){
100     fNprimary    = 1 ; 
101     fPrimary[0]  = primary ;  
102     fDEPrimary[0] = dE ;  
103     fNiparent   = 1 ;
104     fIparent[0] = iparent ;    
105     fDEParent[0] = dE ;  
106   }
107   else{  //If the contribution of this primary smaller than fDigitThreshold (AliEMCALv1)
108     fNprimary    = 0 ; 
109     fPrimary[0]  = -1 ;
110     fDEPrimary[0]  = 0 ;
111     fNiparent   = 0 ;
112     fIparent[0] = -1 ;  
113     fDEParent[0]  = 0 ;  
114   }
115   Int_t i ;
116   for ( i = 1; i < fNMaxPrimary ; i++) {
117     fPrimary[i]  = -1 ; 
118     fDEPrimary[i]  = 0 ;
119   } 
120
121   for ( i = 1; i< fNMaxiparent ; i++) {
122     fIparent[i] = -1 ;  
123     fDEParent[i] = 0 ;
124   }  
125 }
126
127 //____________________________________________________________________________
128 AliEMCALDigit::AliEMCALDigit(const AliEMCALDigit & digit) 
129   : AliDigitNew(digit),
130     fNprimary(digit.fNprimary),
131     fNMaxPrimary(digit.fNMaxPrimary),
132     fPrimary(0x0),
133     fDEPrimary(0x0),
134     fNiparent(digit.fNiparent),
135     fNMaxiparent(digit.fNMaxiparent),
136     fIparent(0x0),
137     fDEParent(0x0),
138     fMaxIter(digit.fMaxIter),
139     fTime(digit.fTime),
140     fTimeR(digit.fTimeR)
141 {
142   // copy ctor
143   
144   // data memebrs of the base class (AliNewDigit)
145   fAmp         = digit.fAmp ;
146   fId          = digit.fId;
147   fIndexInList = digit.fIndexInList ; 
148
149   // data members
150   fPrimary = new Int_t[fNMaxPrimary] ;  
151   fDEPrimary = new Float_t[fNMaxPrimary] ;
152   fIparent = new Int_t[fNMaxiparent] ;
153   fDEParent = new Float_t[fNMaxiparent] ;
154   Int_t i ;
155   for ( i = 0; i < fNMaxPrimary ; i++) {
156     fPrimary[i]  = digit.fPrimary[i] ;
157     fDEPrimary[i]  = digit.fDEPrimary[i] ;
158   }
159   Int_t j ;
160   for (j = 0; j< fNMaxiparent ; j++) {
161     fIparent[j]  = digit.fIparent[j] ;
162     fDEParent[j]  = digit.fDEParent[j] ;
163   }
164 }
165
166 //____________________________________________________________________________
167 AliEMCALDigit::~AliEMCALDigit() 
168 {
169   // Delete array of primiries if any
170     delete [] fPrimary ;
171     delete [] fDEPrimary ;
172     delete [] fIparent ; 
173     delete [] fDEParent ;
174 }
175
176 //____________________________________________________________________________
177 Int_t AliEMCALDigit::Compare(const TObject * obj) const
178 {
179   // Compares two digits with respect to its Id
180   // to sort according increasing Id
181
182   Int_t rv ;
183
184   AliEMCALDigit * digit = (AliEMCALDigit *)obj ; 
185
186   Int_t iddiff = fId - digit->GetId() ; 
187
188   if ( iddiff > 0 ) 
189     rv = 1 ;
190   else if ( iddiff < 0 )
191     rv = -1 ; 
192   else
193     rv = 0 ;
194   
195   return rv ; 
196
197 }
198
199 //____________________________________________________________________________
200 Float_t AliEMCALDigit::GetEta() const
201
202   //return pseudorapidity for this digit
203   // should be change in EMCALGeometry - 19-nov-04
204   Float_t eta=-10., phi=-10.;
205   Int_t id = GetId();
206   const AliEMCALGeometry *g = AliEMCALGeometry::GetInstance();
207   g->EtaPhiFromIndex(id,eta,phi);
208   return eta ;
209 }
210
211 //____________________________________________________________________________
212 Float_t AliEMCALDigit::GetPhi() const
213
214   //return phi coordinate of digit
215   // should be change in EMCALGeometry - 19-nov-04
216   Float_t eta=-10., phi=-10.;
217   Int_t id = GetId();
218   const AliEMCALGeometry *g = AliEMCALGeometry::GetInstance();
219   g->EtaPhiFromIndex(id,eta,phi);
220   return phi ;
221 }
222
223 //____________________________________________________________________________
224 Int_t AliEMCALDigit::GetPrimary(Int_t index) const
225 {
226   // retrieves the primary particle number given its index in the list 
227   if ( (index <= fNprimary) && (index > 0)){
228     return fPrimary[index-1] ;
229   } 
230
231   return -1 ; 
232 }
233
234 //____________________________________________________________________________
235 Float_t AliEMCALDigit::GetDEPrimary(Int_t index) const
236 {
237   // retrieves the primary particle energy contribution 
238   // given its index in the list 
239   if ( (index <= fNprimary) && (index > 0)){
240     return fDEPrimary[index-1] ;
241   } 
242
243   return 0 ; 
244   
245 }
246
247 //____________________________________________________________________________
248 Int_t AliEMCALDigit::GetIparent(Int_t index) const
249 {
250   // retrieves the primary particle number given its index in the list 
251   if ( index <= fNiparent && index > 0){
252     return fIparent[index-1] ;
253   } 
254
255   return -1 ; 
256   
257 }
258
259 //____________________________________________________________________________
260 Float_t AliEMCALDigit::GetDEParent(Int_t index) const
261 {
262   // retrieves the parent particle energy contribution 
263   // given its index in the list 
264   if ( (index <= fNiparent) && (index > 0)){
265     return fDEParent[index-1] ;
266   } 
267
268   return 0; 
269 }
270
271 //____________________________________________________________________________
272 void AliEMCALDigit::ShiftPrimary(Int_t shift){
273   //shifts primary number to BIG offset, to separate primary in different TreeK
274   Int_t index  ;
275   for(index = 0; index <fNprimary; index++ ){
276     fPrimary[index] = fPrimary[index]+ shift * 10000000   ;}
277   for(index =0; index <fNiparent; index++){
278     fIparent[index] = fIparent[index] + shift * 10000000 ;}
279 }
280 //____________________________________________________________________________
281 Bool_t AliEMCALDigit::operator==(AliEMCALDigit const & digit) const 
282 {
283   // Two digits are equal if they have the same Id
284   
285   if ( fId == digit.fId ) 
286     return kTRUE ;
287   else 
288     return kFALSE ;
289 }
290  
291 //____________________________________________________________________________
292 AliEMCALDigit AliEMCALDigit::operator+(const AliEMCALDigit &digit) 
293 {
294   // Adds the amplitude of digits and completes the list of primary particles
295   // if amplitude is larger than 
296   
297   fAmp += digit.fAmp ;
298   if(fTime > digit.fTime)
299     fTime = digit.fTime ;
300   if (digit.fTimeR < fTimeR)
301     fTimeR = digit.fTimeR ; 
302
303   Int_t max1 = fNprimary ; 
304   Int_t max2 = fNiparent ;  
305   Int_t index ; 
306   for (index = 0 ; index < digit.fNprimary  ; index++){
307     Bool_t newPrim = kTRUE ;
308     Int_t old ;
309     for ( old = 0 ; (old < max1) && newPrim; old++) { //already have this primary?
310       if(fPrimary[old] == digit.fPrimary[index]) {
311         newPrim = kFALSE;
312         fDEPrimary[old] += digit.fDEPrimary[index];
313       }
314     }
315     if (newPrim) {
316       if(max1<fNMaxPrimary){ 
317         fPrimary[max1] = digit.fPrimary[index] ; 
318         fDEPrimary[max1] = digit.fDEPrimary[index] ; 
319         fNprimary++ ;
320         max1++;
321       }
322       if(fNprimary==fNMaxPrimary) {
323         
324         TString mess = " NMaxPrimary  =  " ; 
325         mess += fNMaxPrimary ; 
326         mess += " is too small" ; 
327         Fatal("AliEMCALDigit::Operator+ -->" , mess.Data()) ; 
328
329       }
330     }
331   }
332   
333   for (index = 0 ; index < digit.fNiparent ; index++){
334     Bool_t newParent = kTRUE ;
335     Int_t old ;
336     for ( old = 0 ; (old < max2) && newParent; old++) { //already have this primary?
337       if(fIparent[old] == digit.fIparent[index]) {
338         newParent = kFALSE;
339         fDEParent[old] += digit.fDEParent[index];
340       }
341     }
342     if(newParent){
343       if(max2<fNMaxiparent) { 
344         fIparent[max2] = digit.fIparent[index] ; 
345         fDEParent[max2] = digit.fDEParent[index] ; 
346         fNiparent++ ;
347         max2++;
348       }
349       if(fNiparent==fNMaxiparent) {
350         
351         TString mess = " NMaxiparent  =  " ; 
352         mess += fNMaxiparent ; 
353         mess += " is too small" ; 
354         Fatal("AliEMCALDigit::Operator+ -->", mess.Data()) ; 
355
356       }
357     }
358   }
359   
360   return *this ;
361 }
362
363 //____________________________________________________________________________
364 AliEMCALDigit AliEMCALDigit::operator*(Float_t factor) 
365 {
366   // Multiplies the amplitude by a factor
367   
368   Float_t tempo = static_cast<Float_t>(fAmp) ; 
369   tempo *= factor ; 
370   fAmp = static_cast<Int_t>(TMath::Ceil(tempo)) ; 
371   for(Int_t i=0; i < fNprimary; i++) 
372     fDEPrimary[i] *= factor;
373   for(Int_t i=0; i < fNiparent; i++) 
374     fDEParent[i] *= factor;
375
376   return *this ;
377 }
378
379 //____________________________________________________________________________
380 ostream& operator << ( ostream& out , const AliEMCALDigit & digit)
381 {
382   // Prints the data of the digit
383   
384   out << "ID " << digit.fId << " Energy = " << digit.fAmp <<  " Time = " << digit.fTime << endl ; 
385   Int_t i,j ;
386   for(i=0;i<digit.fNprimary;i++) 
387     out << "Primary " << i+1 << " = " << digit.fPrimary[i] 
388         << " : DE " << digit.fDEPrimary[i] << endl ;
389    
390   for(j=0;j<digit.fNiparent;j++)
391     out << "Iparent " << j+1 << " = " << digit.fIparent[j] 
392         << " : DE " << digit.fDEParent[j] << endl ;
393   out << "Position in list = " << digit.fIndexInList << endl ; 
394   return out ;
395 }
396
397