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