updates for Effective C++ compiler flags
[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 <Riostream.h>
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   fNiparent(0),
49   fNMaxiparent(5), 
50   fIparent(0x0),
51   fMaxIter(0),
52   fTime(0.), 
53   fTimeR(0.) 
54
55 {
56   // default ctor 
57
58 }
59
60 //____________________________________________________________________________
61 AliEMCALDigit::AliEMCALDigit(Int_t primary, Int_t iparent, Int_t id, Int_t DigEnergy, Float_t time, Int_t index) 
62   : AliDigitNew(),
63     fNprimary(0),
64     fNMaxPrimary(25),
65     fPrimary(0x0),
66     fNiparent(0),
67     fNMaxiparent(150),
68     fIparent(0x0),
69     fMaxIter(5),
70     fTime(time),
71     fTimeR(time)
72 {  
73   // ctor with all data 
74
75   // data memebrs of the base class (AliNewDigit)
76   fAmp         = DigEnergy ;
77   fId          = id ;
78   fIndexInList = index ; 
79
80   // data members
81   fPrimary = new Int_t[fNMaxPrimary] ;
82   fIparent = new Int_t[fNMaxiparent] ; 
83   if( primary != -1){
84     fNprimary    = 1 ; 
85     fPrimary[0]  = primary ;  
86     fNiparent   = 1 ;
87     fIparent[0] = iparent ;    
88 }
89   else{  //If the contribution of this primary smaller than fDigitThreshold (AliEMCALv1)
90     fNprimary    = 0 ; 
91     fPrimary[0]  = -1 ;
92     fNiparent   = 0 ;
93     fIparent[0] = -1 ;    
94
95   }
96   Int_t i ;
97   for ( i = 1; i < fNMaxPrimary ; i++)
98     fPrimary[i]  = -1 ; 
99
100   for ( i =1; i< fNMaxiparent ; i++)
101     fIparent[i] = -1 ;  
102 }
103
104 //____________________________________________________________________________
105 AliEMCALDigit::AliEMCALDigit(const AliEMCALDigit & digit) 
106   : AliDigitNew(digit),
107     fNprimary(digit.fNprimary),
108     fNMaxPrimary(digit.fNMaxPrimary),
109     fPrimary(0x0),
110     fNiparent(digit.fNiparent),
111     fNMaxiparent(digit.fNMaxiparent),
112     fIparent(0x0),
113     fMaxIter(digit.fMaxIter),
114     fTime(digit.fTime),
115     fTimeR(digit.fTimeR)
116 {
117   // copy ctor
118   
119   // data memebrs of the base class (AliNewDigit)
120   fAmp         = digit.fAmp ;
121   fId          = digit.fId;
122   fIndexInList = digit.fIndexInList ; 
123
124   // data members
125   fPrimary = new Int_t[fNMaxPrimary] ;
126   fIparent = new Int_t[fNMaxiparent] ; 
127   Int_t i ;
128   for ( i = 0; i < fNMaxPrimary ; i++)
129   fPrimary[i]  = digit.fPrimary[i] ;
130   Int_t j ;
131   for (j = 0; j< fNMaxiparent ; j++)
132   fIparent[j]  = digit.fIparent[j] ;
133 }
134
135 //____________________________________________________________________________
136 AliEMCALDigit::~AliEMCALDigit() 
137 {
138   // Delete array of primiries if any
139     delete [] fPrimary ;
140     delete [] fIparent ; 
141 }
142
143 //____________________________________________________________________________
144 Int_t AliEMCALDigit::Compare(const TObject * obj) const
145 {
146   // Compares two digits with respect to its Id
147   // to sort according increasing Id
148
149   Int_t rv ;
150
151   AliEMCALDigit * digit = (AliEMCALDigit *)obj ; 
152
153   Int_t iddiff = fId - digit->GetId() ; 
154
155   if ( iddiff > 0 ) 
156     rv = 1 ;
157   else if ( iddiff < 0 )
158     rv = -1 ; 
159   else
160     rv = 0 ;
161   
162   return rv ; 
163
164 }
165
166 //____________________________________________________________________________
167 Float_t AliEMCALDigit::GetEta() const
168
169   //return pseudorapidity for this digit
170   // should be change in EMCALGeometry - 19-nov-04
171   Float_t eta=-10., phi=-10.;
172   Int_t id = GetId();
173   const AliEMCALGeometry *g = AliEMCALGeometry::GetInstance();
174   g->EtaPhiFromIndex(id,eta,phi);
175   return eta ;
176 }
177
178 //____________________________________________________________________________
179 Float_t AliEMCALDigit::GetPhi() const
180
181   //return phi coordinate of digit
182   // should be change in EMCALGeometry - 19-nov-04
183   Float_t eta=-10., phi=-10.;
184   Int_t id = GetId();
185   const AliEMCALGeometry *g = AliEMCALGeometry::GetInstance();
186   g->EtaPhiFromIndex(id,eta,phi);
187   return phi ;
188 }
189
190 //____________________________________________________________________________
191 Int_t AliEMCALDigit::GetPrimary(Int_t index) const
192 {
193   // retrieves the primary particle number given its index in the list 
194   Int_t rv = -1 ;
195   if ( (index <= fNprimary) && (index > 0)){
196     rv = fPrimary[index-1] ;
197   } 
198
199   return rv ; 
200   
201 }
202
203 //____________________________________________________________________________
204 Int_t AliEMCALDigit::GetIparent(Int_t index) const
205 {
206   // retrieves the primary particle number given its index in the list 
207   Int_t rv = -1 ;
208   if ( index <= fNiparent ){
209     rv = fIparent[index-1] ;
210   } 
211
212   return rv ; 
213   
214 }
215
216 //____________________________________________________________________________
217 void AliEMCALDigit::ShiftPrimary(Int_t shift){
218   //shifts primary number to BIG offset, to separate primary in different TreeK
219   Int_t index  ;
220   for(index = 0; index <fNprimary; index++ ){
221     fPrimary[index] = fPrimary[index]+ shift * 10000000   ;}
222   for(index =0; index <fNiparent; index++){
223     fIparent[index] = fIparent[index] + shift * 10000000 ;}
224 }
225 //____________________________________________________________________________
226 Bool_t AliEMCALDigit::operator==(AliEMCALDigit const & digit) const 
227 {
228   // Two digits are equal if they have the same Id
229   
230   if ( fId == digit.fId ) 
231     return kTRUE ;
232   else 
233     return kFALSE ;
234 }
235  
236 //____________________________________________________________________________
237 AliEMCALDigit AliEMCALDigit::operator+(const AliEMCALDigit &digit) 
238 {
239   // Adds the amplitude of digits and completes the list of primary particles
240   // if amplitude is larger than 
241   
242   fAmp += digit.fAmp ;
243   if(fTime > digit.fTime)
244     fTime = digit.fTime ;
245   fTimeR = fTime ; 
246
247   Int_t max1 = fNprimary ; 
248   Int_t max2 = fNiparent ;  
249   Int_t index ; 
250   for (index = 0 ; index < digit.fNprimary  ; index++){
251     Bool_t deja = kTRUE ;
252     Int_t old ;
253     for ( old = 0 ; (old < max1) && deja; old++) { //already have this primary?
254       if(fPrimary[old] == digit.fPrimary[index])
255         deja = kFALSE;
256     }
257     if(deja){
258       if(max1<fNMaxPrimary){ fPrimary[max1] = digit.fPrimary[index] ; 
259       fNprimary++ ;
260       max1++;}
261       if(fNprimary==fNMaxPrimary) {
262
263         TString mess = " NMaxPrimary  =  " ; 
264         mess += fNMaxPrimary ; 
265         mess += " is too small" ; 
266         Fatal("AliEMCALDigit::Operator+ -->" , mess.Data()) ; 
267
268       }
269     }
270   }
271   
272   for (index = 0 ; index < digit.fNiparent ; index++){
273     Bool_t dejavu = kTRUE ;
274     Int_t old ;
275     for ( old = 0 ; (old < max2) && dejavu; old++) { //already have this primary?
276       if(fIparent[old] == digit.fIparent[index])
277         dejavu = kFALSE;
278     }
279     if(dejavu){
280      if(max2<fNMaxiparent){ fIparent[max2] = digit.fIparent[index] ; 
281       fNiparent++ ;
282       max2++;}
283       if(fNiparent==fNMaxiparent) {
284
285         TString mess = " NMaxiparent  =  " ; 
286         mess += fNMaxiparent ; 
287         mess += " is too small" ; 
288         Fatal("AliEMCALDigit::Operator+ -->", mess.Data()) ; 
289
290       }
291     }
292   }
293   
294   return *this ;
295 }
296
297 //____________________________________________________________________________
298 AliEMCALDigit AliEMCALDigit::operator*(Float_t factor) 
299 {
300   // Multiplies the amplitude by a factor
301   
302   Float_t tempo = static_cast<Float_t>(fAmp) ; 
303   tempo *= factor ; 
304   fAmp = static_cast<Int_t>(TMath::Ceil(tempo)) ; 
305   return *this ;
306 }
307
308 //____________________________________________________________________________
309 ostream& operator << ( ostream& out , const AliEMCALDigit & digit)
310 {
311   // Prints the data of the digit
312   
313   out << "ID " << digit.fId << " Energy = " << digit.fAmp <<  " Time = " << digit.fTime << endl ; 
314   Int_t i,j ;
315   for(i=0;i<digit.fNprimary;i++)
316     out << "Primary " << i+1 << " = " << digit.fPrimary[i] << endl ;
317    
318   for(j=0;j<digit.fNiparent;j++)
319     out << "Iparent " << j+1 << " = " << digit.fIparent[j] << endl ;
320  out << "Position in list = " << digit.fIndexInList << endl ; 
321   return out ;
322 }
323
324