]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - EMCAL/AliEMCALDigit.cxx
added low and high flux parameter options
[u/mrichter/AliRoot.git] / EMCAL / AliEMCALDigit.cxx
index f90e10f152a88d8fd763a5e1c362ffec54a35c17..9617673738bec00ec01370fbcf954d64c4e7c46e 100644 (file)
@@ -13,7 +13,7 @@
  * provided "as is" without express or implied warranty.                  *
  **************************************************************************/
 
-/* $Id */
+/* $Id$ */
 
 //_________________________________________________________________________
 //  EMCAL digit: 
 //__________________________________________________________________________
 
 // --- ROOT system ---
+#include <Riostream.h>
+#include <TMath.h>
 
 // --- Standard library ---
 
-#include <iostream.h>
-
 // --- AliRoot header files ---
 
 #include "AliEMCALDigit.h"
 #include "AliEMCALGeometry.h"
-#include "AliEMCALGetter.h"
-
 
 ClassImp(AliEMCALDigit)
 
 //____________________________________________________________________________
-  AliEMCALDigit::AliEMCALDigit() 
+AliEMCALDigit::AliEMCALDigit() :  
+  AliDigitNew(),
+  fNprimary(0),
+  fNMaxPrimary(5),
+  fPrimary(0x0),
+  fDEPrimary(0x0),
+  fNiparent(0),
+  fNMaxiparent(5), 
+  fIparent(0x0),
+  fDEParent(0x0),
+  fMaxIter(0),
+  fTime(0.), 
+  fTimeR(0.) 
+
 {
   // default ctor 
 
-  fIndexInList = -1 ; 
-  fNprimary    = 0 ;  
-  fNMaxPrimary = 5 ; 
-  fNiparent     = 0 ;
-  fNMaxiparent = fNMaxPrimary*10;
+  // Need to initialise for reading old files
   fPrimary = new Int_t[fNMaxPrimary] ;
+  fDEPrimary = new Float_t[fNMaxPrimary] ;
   fIparent = new Int_t[fNMaxiparent] ; 
+  fDEParent = new Float_t[fNMaxiparent] ; 
+  for ( Int_t i = 0; i < fNMaxPrimary ; i++) {
+    fPrimary[i]  = -1 ;
+    fDEPrimary[i]  = 0 ;
+  } 
+
+  for ( Int_t i = 0; i < fNMaxiparent ; i++) {
+    fIparent[i] = -1 ;
+    fDEParent[i] = 0 ;
+  }
 }
 
 //____________________________________________________________________________
-AliEMCALDigit::AliEMCALDigit(Int_t primary, Int_t iparent, Int_t id, Int_t DigEnergy, Float_t time, Int_t index) 
+AliEMCALDigit::AliEMCALDigit(Int_t primary, Int_t iparent, Int_t id, Int_t DigEnergy, Float_t time, Int_t index, Float_t dE) 
+  : AliDigitNew(),
+    fNprimary(0),
+    fNMaxPrimary(25),
+    fPrimary(0x0),
+    fDEPrimary(0x0),
+    fNiparent(0),
+    fNMaxiparent(150),
+    fIparent(0x0),
+    fDEParent(0x0),
+    fMaxIter(5),
+    fTime(time),
+    fTimeR(time)
 {  
   // ctor with all data 
 
-  fNMaxPrimary = 5 ; 
-  fNMaxiparent = fNMaxPrimary*10;
-  fPrimary = new Int_t[fNMaxPrimary] ;
-  fIparent = new Int_t[fNMaxiparent] ; 
+  // data memebrs of the base class (AliNewDigit)
   fAmp         = DigEnergy ;
-  fTime        = time ;
   fId          = id ;
   fIndexInList = index ; 
+
+  // data members
+  fPrimary = new Int_t[fNMaxPrimary] ;
+  fDEPrimary = new Float_t[fNMaxPrimary] ;
+  fIparent = new Int_t[fNMaxiparent] ; 
+  fDEParent = new Float_t[fNMaxiparent] ; 
   if( primary != -1){
     fNprimary    = 1 ; 
-    fPrimary[0]  = primary ;
+    fPrimary[0]  = primary ;  
+    fDEPrimary[0] = dE ;  
     fNiparent   = 1 ;
     fIparent[0] = iparent ;    
-
+    fDEParent[0] = dE ;  
   }
   else{  //If the contribution of this primary smaller than fDigitThreshold (AliEMCALv1)
     fNprimary    = 0 ; 
     fPrimary[0]  = -1 ;
+    fDEPrimary[0]  = 0 ;
     fNiparent   = 0 ;
-    fIparent[0] = -1 ;    
-
+    fIparent[0] = -1 ;  
+    fDEParent[0]  = 0 ;  
   }
   Int_t i ;
-  for ( i = 1; i < fNMaxPrimary ; i++)
+  for ( i = 1; i < fNMaxPrimary ; i++) {
     fPrimary[i]  = -1 ; 
+    fDEPrimary[i]  = 0 ;
+  } 
 
-  for ( Int_t j =1; j< fNMaxiparent ; j++)
-    fIparent[j] = -1 ;  
+  for ( i = 1; i< fNMaxiparent ; i++) {
+    fIparent[i] = -1 ;  
+    fDEParent[i] = 0 ;
+  }  
 }
 
 //____________________________________________________________________________
 AliEMCALDigit::AliEMCALDigit(const AliEMCALDigit & digit) 
+  : AliDigitNew(digit),
+    fNprimary(digit.fNprimary),
+    fNMaxPrimary(digit.fNMaxPrimary),
+    fPrimary(0x0),
+    fDEPrimary(0x0),
+    fNiparent(digit.fNiparent),
+    fNMaxiparent(digit.fNMaxiparent),
+    fIparent(0x0),
+    fDEParent(0x0),
+    fMaxIter(digit.fMaxIter),
+    fTime(digit.fTime),
+    fTimeR(digit.fTimeR)
 {
   // copy ctor
   
-
-  fNMaxPrimary = digit.fNMaxPrimary ;  
-  fNMaxiparent = digit.fNMaxiparent ;
-  fPrimary = new Int_t[fNMaxPrimary] ;
-  fIparent = new Int_t[fNMaxiparent] ; 
-  Int_t i ;
-  for ( i = 0; i < fNMaxPrimary ; i++)
-  fPrimary[i]  = digit.fPrimary[i] ;
-  Int_t j ;
-  for (j = 0; j< fNMaxiparent ; j++)
-  fIparent[j]  = digit.fIparent[j] ;
+  // data memebrs of the base class (AliNewDigit)
   fAmp         = digit.fAmp ;
-  fTime        = digit.fTime ;
   fId          = digit.fId;
   fIndexInList = digit.fIndexInList ; 
-  fNprimary    = digit.fNprimary ;
-  fNiparent    = digit.fNiparent ;
+
+  // data members
+  fPrimary = new Int_t[fNMaxPrimary] ;  
+  fDEPrimary = new Float_t[fNMaxPrimary] ;
+  fIparent = new Int_t[fNMaxiparent] ;
+  fDEParent = new Float_t[fNMaxiparent] ;
+  Int_t i ;
+  for ( i = 0; i < fNMaxPrimary ; i++) {
+    fPrimary[i]  = digit.fPrimary[i] ;
+    fDEPrimary[i]  = digit.fDEPrimary[i] ;
+  }
+  Int_t j ;
+  for (j = 0; j< fNMaxiparent ; j++) {
+    fIparent[j]  = digit.fIparent[j] ;
+    fDEParent[j]  = digit.fDEParent[j] ;
+  }
 }
 
 //____________________________________________________________________________
@@ -119,7 +172,9 @@ AliEMCALDigit::~AliEMCALDigit()
 {
   // Delete array of primiries if any
     delete [] fPrimary ;
+    delete [] fDEPrimary ;
     delete [] fIparent ; 
+    delete [] fDEParent ;
 }
 
 //____________________________________________________________________________
@@ -146,18 +201,26 @@ Int_t AliEMCALDigit::Compare(const TObject * obj) const
 }
 
 //____________________________________________________________________________
-const Float_t AliEMCALDigit::GetEta() const
-{
+Float_t AliEMCALDigit::GetEta() const
+{ 
+  //return pseudorapidity for this digit
+  // should be change in EMCALGeometry - 19-nov-04
   Float_t eta=-10., phi=-10.;
-  AliEMCALGeometry::GetInstance()->EtaPhiFromIndex(fId,eta,phi);
+  Int_t id = GetId();
+  const AliEMCALGeometry *g = AliEMCALGeometry::GetInstance();
+  g->EtaPhiFromIndex(id,eta,phi);
   return eta ;
 }
 
 //____________________________________________________________________________
-const Float_t AliEMCALDigit::GetPhi() const
-{
+Float_t AliEMCALDigit::GetPhi() const
+{ 
+  //return phi coordinate of digit
+  // should be change in EMCALGeometry - 19-nov-04
   Float_t eta=-10., phi=-10.;
-  AliEMCALGeometry::GetInstance()->EtaPhiFromIndex(fId,eta,phi);
+  Int_t id = GetId();
+  const AliEMCALGeometry *g = AliEMCALGeometry::GetInstance();
+  g->EtaPhiFromIndex(id,eta,phi);
   return phi ;
 }
 
@@ -165,12 +228,23 @@ const Float_t AliEMCALDigit::GetPhi() const
 Int_t AliEMCALDigit::GetPrimary(Int_t index) const
 {
   // retrieves the primary particle number given its index in the list 
-  Int_t rv = -1 ;
-  if ( index <= fNprimary && index > 0){
-    rv = fPrimary[index-1] ;
+  if ( (index <= fNprimary) && (index > 0)){
+    return fPrimary[index-1] ;
   } 
 
-  return rv ; 
+  return -1 ; 
+}
+
+//____________________________________________________________________________
+Float_t AliEMCALDigit::GetDEPrimary(Int_t index) const
+{
+  // retrieves the primary particle energy contribution 
+  // given its index in the list 
+  if ( (index <= fNprimary) && (index > 0)){
+    return fDEPrimary[index-1] ;
+  } 
+
+  return 0 ; 
   
 }
 
@@ -178,28 +252,29 @@ Int_t AliEMCALDigit::GetPrimary(Int_t index) const
 Int_t AliEMCALDigit::GetIparent(Int_t index) const
 {
   // retrieves the primary particle number given its index in the list 
-  Int_t rv = -1 ;
-  if ( index <= fNiparent ){
-    rv = fIparent[index-1] ;
+  if ( index <= fNiparent && index > 0){
+    return fIparent[index-1] ;
   } 
 
-  return rv ; 
+  return -1 ; 
   
 }
 
-//______________________________________________________________________
-const Bool_t AliEMCALDigit::IsInPreShower() const 
+//____________________________________________________________________________
+Float_t AliEMCALDigit::GetDEParent(Int_t index) const
 {
-  Bool_t rv = kFALSE ;
-  const AliEMCALGeometry * geom = AliEMCALGetter::GetInstance()->EMCALGeometry() ;
-  if( GetId() > (geom->GetNZ() * geom->GetNPhi() )) 
-    rv = kTRUE; 
-  return rv; 
-} 
+  // retrieves the parent particle energy contribution 
+  // given its index in the list 
+  if ( (index <= fNiparent) && (index > 0)){
+    return fDEParent[index-1] ;
+  } 
+
+  return 0; 
+}
 
 //____________________________________________________________________________
 void AliEMCALDigit::ShiftPrimary(Int_t shift){
-  //shifts primary nimber to BIG offset, to separate primary in different TreeK
+  //shifts primary number to BIG offset, to separate primary in different TreeK
   Int_t index  ;
   for(index = 0; index <fNprimary; index++ ){
     fPrimary[index] = fPrimary[index]+ shift * 10000000   ;}
@@ -218,7 +293,7 @@ Bool_t AliEMCALDigit::operator==(AliEMCALDigit const & digit) const
 }
  
 //____________________________________________________________________________
-AliEMCALDigit& AliEMCALDigit::operator+(AliEMCALDigit const & digit) 
+AliEMCALDigit AliEMCALDigit::operator+(const AliEMCALDigit &digit) 
 {
   // Adds the amplitude of digits and completes the list of primary particles
   // if amplitude is larger than 
@@ -226,40 +301,62 @@ AliEMCALDigit& AliEMCALDigit::operator+(AliEMCALDigit const & digit)
   fAmp += digit.fAmp ;
   if(fTime > digit.fTime)
     fTime = digit.fTime ;
-  
+  if (digit.fTimeR < fTimeR)
+    fTimeR = digit.fTimeR ; 
+
   Int_t max1 = fNprimary ; 
   Int_t max2 = fNiparent ;  
   Int_t index ; 
-  for (index = 0 ; index < digit.fNprimary ; index++){
-    Bool_t deja = kTRUE ;
+  for (index = 0 ; index < digit.fNprimary  ; index++){
+    Bool_t newPrim = kTRUE ;
     Int_t old ;
-    for ( old = 0 ; (old < max1) && deja; old++) { //already have this primary?
-      if(fPrimary[old] == (digit.fPrimary)[index])
-       deja = kFALSE;
+    for ( old = 0 ; (old < max1) && newPrim; old++) { //already have this primary?
+      if(fPrimary[old] == digit.fPrimary[index]) {
+       newPrim = kFALSE;
+       fDEPrimary[old] += digit.fDEPrimary[index];
+      }
     }
-    if(deja){
-      fPrimary[fNprimary] = (digit.fPrimary)[index] ; 
-      fNprimary++ ;
-      if(fNprimary>fNMaxPrimary) {
-       cout << "AliEMCALDigit >> Increase NMaxPrimary "<< endl ;
-       return *this ;
+    if (newPrim) {
+      if(max1<fNMaxPrimary){ 
+       fPrimary[max1] = digit.fPrimary[index] ; 
+       fDEPrimary[max1] = digit.fDEPrimary[index] ; 
+       fNprimary++ ;
+       max1++;
+      }
+      if(fNprimary==fNMaxPrimary) {
+       
+       TString mess = " NMaxPrimary  =  " ; 
+       mess += fNMaxPrimary ; 
+       mess += " is too small" ; 
+       Fatal("AliEMCALDigit::Operator+ -->" , mess.Data()) ; 
+
       }
     }
   }
   
   for (index = 0 ; index < digit.fNiparent ; index++){
-    Bool_t dejavu = kTRUE ;
+    Bool_t newParent = kTRUE ;
     Int_t old ;
-    for ( old = 0 ; (old < max2) && dejavu; old++) { //already have this primary?
-      if(fIparent[old] == (digit.fIparent)[index])
-       dejavu = kFALSE;
+    for ( old = 0 ; (old < max2) && newParent; old++) { //already have this primary?
+      if(fIparent[old] == digit.fIparent[index]) {
+       newParent = kFALSE;
+       fDEParent[old] += digit.fDEParent[index];
+      }
     }
-    if(dejavu){
-      fIparent[fNiparent] = (digit.fIparent)[index] ; 
-      fNiparent++ ;
-      if(fNiparent>fNMaxiparent) {
-       cout << "AliEMCALDigit >> Increase NMaxiparent "<< endl ;
-       return *this ;
+    if(newParent){
+      if(max2<fNMaxiparent) { 
+       fIparent[max2] = digit.fIparent[index] ; 
+       fDEParent[max2] = digit.fDEParent[index] ; 
+       fNiparent++ ;
+       max2++;
+      }
+      if(fNiparent==fNMaxiparent) {
+       
+       TString mess = " NMaxiparent  =  " ; 
+       mess += fNMaxiparent ; 
+       mess += " is too small" ; 
+       Fatal("AliEMCALDigit::Operator+ -->", mess.Data()) ; 
+
       }
     }
   }
@@ -267,6 +364,22 @@ AliEMCALDigit& AliEMCALDigit::operator+(AliEMCALDigit const & digit)
   return *this ;
 }
 
+//____________________________________________________________________________
+AliEMCALDigit AliEMCALDigit::operator*(Float_t factor) 
+{
+  // Multiplies the amplitude by a factor
+  
+  Float_t tempo = static_cast<Float_t>(fAmp) ; 
+  tempo *= factor ; 
+  fAmp = static_cast<Int_t>(TMath::Ceil(tempo)) ; 
+  for(Int_t i=0; i < fNprimary; i++) 
+    fDEPrimary[i] *= factor;
+  for(Int_t i=0; i < fNiparent; i++) 
+    fDEParent[i] *= factor;
+
+  return *this ;
+}
+
 //____________________________________________________________________________
 ostream& operator << ( ostream& out , const AliEMCALDigit & digit)
 {
@@ -274,12 +387,14 @@ ostream& operator << ( ostream& out , const AliEMCALDigit & digit)
   
   out << "ID " << digit.fId << " Energy = " << digit.fAmp <<  " Time = " << digit.fTime << endl ; 
   Int_t i,j ;
-  for(i=0;i<digit.fNprimary;i++)
-    out << "Primary " << i+1 << " = " << digit.fPrimary[i] << endl ;
+  for(i=0;i<digit.fNprimary;i++) 
+    out << "Primary " << i+1 << " = " << digit.fPrimary[i] 
+       << " : DE " << digit.fDEPrimary[i] << endl ;
    
   for(j=0;j<digit.fNiparent;j++)
-    out << "Iparent " << j+1 << " = " << digit.fIparent[j] << endl ;
- out << "Position in list = " << digit.fIndexInList << endl ; 
+    out << "Iparent " << j+1 << " = " << digit.fIparent[j] 
+       << " : DE " << digit.fDEParent[j] << endl ;
+  out << "Position in list = " << digit.fIndexInList << endl ; 
   return out ;
 }