new classes for track segments
[u/mrichter/AliRoot.git] / EMCAL / AliEMCALDigit.cxx
CommitLineData
61e0abb5 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
ffa6d63b 16/* $Id: */
61e0abb5 17
18//_________________________________________________________________________
ffa6d63b 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
61e0abb5 24//
ffa6d63b 25//*-- Author: Sahal Yacoob (LBL)
26// based on : AliPHOSDigit
814ad4bf 27//__________________________________________________________________________
61e0abb5 28
29// --- ROOT system ---
30
31// --- Standard library ---
32
70479d0e 33#include <Riostream.h>
61e0abb5 34
35// --- AliRoot header files ---
36
37#include "AliEMCALDigit.h"
814ad4bf 38#include "AliEMCALGeometry.h"
88d764f0 39#include "AliEMCALGetter.h"
61e0abb5 40
41
42ClassImp(AliEMCALDigit)
43
44//____________________________________________________________________________
da98933e 45 AliEMCALDigit::AliEMCALDigit()
61e0abb5 46{
47 // default ctor
48
0c22a365 49 fIndexInList = 0 ;
61e0abb5 50 fNprimary = 0 ;
ef733077 51 fNMaxPrimary = 0 ;
61e0abb5 52 fNiparent = 0 ;
0c22a365 53
ef733077 54 fNMaxiparent = 0;
55 fPrimary = 0 ;
56 fIparent = 0 ;
0c22a365 57 fMaxIter = 0;
61e0abb5 58}
59
60//____________________________________________________________________________
814ad4bf 61AliEMCALDigit::AliEMCALDigit(Int_t primary, Int_t iparent, Int_t id, Int_t DigEnergy, Float_t time, Int_t index)
61e0abb5 62{
63 // ctor with all data
64
451bc3c5 65 fNMaxPrimary = 25 ;
fb8b329a 66 fNMaxiparent = 150 ;
88d764f0 67 fPrimary = new Int_t[fNMaxPrimary] ;
68 fIparent = new Int_t[fNMaxiparent] ;
61e0abb5 69 fAmp = DigEnergy ;
814ad4bf 70 fTime = time ;
61e0abb5 71 fId = id ;
72 fIndexInList = index ;
0c22a365 73 fMaxIter = 5;
61e0abb5 74 if( primary != -1){
75 fNprimary = 1 ;
7b62cd84 76 fPrimary[0] = primary ;
61e0abb5 77 fNiparent = 1 ;
78 fIparent[0] = iparent ;
7b62cd84 79}
61e0abb5 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
f807c3bd 91 for ( i =1; i< fNMaxiparent ; i++)
92 fIparent[i] = -1 ;
61e0abb5 93}
94
95//____________________________________________________________________________
96AliEMCALDigit::AliEMCALDigit(const AliEMCALDigit & digit)
97{
98 // copy ctor
99
100
101 fNMaxPrimary = digit.fNMaxPrimary ;
102 fNMaxiparent = digit.fNMaxiparent ;
88d764f0 103 fPrimary = new Int_t[fNMaxPrimary] ;
104 fIparent = new Int_t[fNMaxiparent] ;
61e0abb5 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 ;
814ad4bf 112 fTime = digit.fTime ;
61e0abb5 113 fId = digit.fId;
0c22a365 114 fMaxIter = digit.fMaxIter;
61e0abb5 115 fIndexInList = digit.fIndexInList ;
116 fNprimary = digit.fNprimary ;
117 fNiparent = digit.fNiparent ;
118}
119
120//____________________________________________________________________________
121AliEMCALDigit::~AliEMCALDigit()
122{
123 // Delete array of primiries if any
88d764f0 124 delete [] fPrimary ;
125 delete [] fIparent ;
61e0abb5 126}
127
128//____________________________________________________________________________
129Int_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//____________________________________________________________________________
814ad4bf 152const Float_t AliEMCALDigit::GetEta() const
153{
154 Float_t eta=-10., phi=-10.;
563aa66c 155 Int_t id = GetId();
da98933e 156 const AliEMCALGeometry *g = AliEMCALGetter::GetInstance()->EMCALGeometry();
563aa66c 157 g->EtaPhiFromIndex(id,eta,phi);
814ad4bf 158 return eta ;
159}
160
161//____________________________________________________________________________
162const Float_t AliEMCALDigit::GetPhi() const
163{
164 Float_t eta=-10., phi=-10.;
563aa66c 165 Int_t id = GetId();
da98933e 166 const AliEMCALGeometry *g = AliEMCALGetter::GetInstance()->EMCALGeometry();
563aa66c 167 g->EtaPhiFromIndex(id,eta,phi);
814ad4bf 168 return phi ;
169}
170
171//____________________________________________________________________________
61e0abb5 172Int_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 ;
814ad4bf 176 if ( index <= fNprimary && index > 0){
61e0abb5 177 rv = fPrimary[index-1] ;
178 }
179
180 return rv ;
181
182}
183
184//____________________________________________________________________________
185Int_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
61e0abb5 197//____________________________________________________________________________
198void AliEMCALDigit::ShiftPrimary(Int_t shift){
199 //shifts primary nimber to BIG offset, to separate primary in different TreeK
200 Int_t index ;
201 for(index = 0; index <fNprimary; index++ ){
202 fPrimary[index] = fPrimary[index]+ shift * 10000000 ;}
203 for(index =0; index <fNiparent; index++){
204 fIparent[index] = fIparent[index] + shift * 10000000 ;}
205}
206//____________________________________________________________________________
207Bool_t AliEMCALDigit::operator==(AliEMCALDigit const & digit) const
208{
209 // Two digits are equal if they have the same Id
210
211 if ( fId == digit.fId )
212 return kTRUE ;
213 else
214 return kFALSE ;
215}
216
217//____________________________________________________________________________
218AliEMCALDigit& AliEMCALDigit::operator+(AliEMCALDigit const & digit)
219{
220 // Adds the amplitude of digits and completes the list of primary particles
221 // if amplitude is larger than
222
223 fAmp += digit.fAmp ;
814ad4bf 224 if(fTime > digit.fTime)
225 fTime = digit.fTime ;
7b62cd84 226
61e0abb5 227 Int_t max1 = fNprimary ;
228 Int_t max2 = fNiparent ;
229 Int_t index ;
7b62cd84 230 for (index = 0 ; index < digit.fNprimary ; index++){
61e0abb5 231 Bool_t deja = kTRUE ;
232 Int_t old ;
233 for ( old = 0 ; (old < max1) && deja; old++) { //already have this primary?
7b62cd84 234 if(fPrimary[old] == digit.fPrimary[index])
61e0abb5 235 deja = kFALSE;
236 }
237 if(deja){
7b62cd84 238 if(max1<fNMaxPrimary){ fPrimary[max1] = digit.fPrimary[index] ;
61e0abb5 239 fNprimary++ ;
7b62cd84 240 max1++;}
241 if(fNprimary==fNMaxPrimary) {
8704424d 242
451bc3c5 243 TString mess = " NMaxPrimary = " ;
244 mess += fNMaxPrimary ;
245 mess += " is too small" ;
246 Fatal("AliEMCALDigit::Operator+ -->" , mess.Data()) ;
8704424d 247
61e0abb5 248 }
249 }
250 }
251
252 for (index = 0 ; index < digit.fNiparent ; index++){
253 Bool_t dejavu = kTRUE ;
254 Int_t old ;
255 for ( old = 0 ; (old < max2) && dejavu; old++) { //already have this primary?
7b62cd84 256 if(fIparent[old] == digit.fIparent[index])
61e0abb5 257 dejavu = kFALSE;
258 }
259 if(dejavu){
7b62cd84 260 if(max2<fNMaxiparent){ fIparent[max2] = digit.fIparent[index] ;
61e0abb5 261 fNiparent++ ;
7b62cd84 262 max2++;}
263 if(fNiparent==fNMaxiparent) {
8704424d 264
451bc3c5 265 TString mess = " NMaxiparent = " ;
266 mess += fNMaxiparent ;
267 mess += " is too small" ;
268 Fatal("AliEMCALDigit::Operator+ -->", mess.Data()) ;
8704424d 269
61e0abb5 270 }
271 }
272 }
273
274 return *this ;
275}
276
277//____________________________________________________________________________
563aa66c 278AliEMCALDigit& AliEMCALDigit::operator*(Float_t factor)
279{
280 // Multiplies the amplitude by a factor
281
282 Float_t tempo = static_cast<Float_t>(fAmp) ;
283 tempo *= factor ;
284 fAmp = static_cast<Int_t>(TMath::Ceil(tempo)) ;
285 return *this ;
286}
287
288//____________________________________________________________________________
61e0abb5 289ostream& operator << ( ostream& out , const AliEMCALDigit & digit)
290{
291 // Prints the data of the digit
292
814ad4bf 293 out << "ID " << digit.fId << " Energy = " << digit.fAmp << " Time = " << digit.fTime << endl ;
61e0abb5 294 Int_t i,j ;
295 for(i=0;i<digit.fNprimary;i++)
296 out << "Primary " << i+1 << " = " << digit.fPrimary[i] << endl ;
297
298 for(j=0;j<digit.fNiparent;j++)
299 out << "Iparent " << j+1 << " = " << digit.fIparent[j] << endl ;
300 out << "Position in list = " << digit.fIndexInList << endl ;
301 return out ;
302}
303
304