]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - EMCAL/AliEMCALDigit.cxx
Bug fix for HMPID bits in readout list.
[u/mrichter/AliRoot.git] / EMCAL / AliEMCALDigit.cxx
index f070fdf641f0aa2b4c19f1cb1ad88733494bcf44..2037198d444de84d35336814c94b8b7e791f9cb9 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(), 
+  fAmpFloat(0.),
+  fNSamples(0),
+  fSamples(0x0),
+  fNSamplesHG(0),
+  fSamplesHG(0x0),
+  fNprimary(0),
+  fNMaxPrimary(5),
+  fPrimary(0x0),
+  fDEPrimary(0x0),
+  fNiparent(0),
+  fNMaxiparent(5), 
+  fIparent(0x0),
+  fDEParent(0x0),
+  fMaxIter(0),
+  fTime(0.), 
+  fTimeR(0.),
+  fChi2(0.),
+  fNDF(0),
+  fDigitType(kUnknown)
+
 {
   // default ctor 
 
-  fIndexInList = 0 ; 
-  fNprimary    = 0 ;  
-  fNMaxPrimary = 0 ; 
-  fNiparent     = 0 ;
+  // 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 ;
+  } 
 
-  fNMaxiparent = 0; 
-  fPrimary = 0 ;
-  fIparent = 0 ;
-  fMaxIter = 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, Float_t digEnergy, Float_t time, Int_t type, Int_t index, Float_t chi2, Int_t ndf, Float_t dE) 
+  : AliDigitNew(),
+    fAmpFloat(digEnergy),
+    fNSamples(0),
+    fSamples(0x0),
+    fNSamplesHG(0),
+    fSamplesHG(0x0),
+    fNprimary(0),
+    fNMaxPrimary(25),
+    fPrimary(0x0),
+    fDEPrimary(0x0),
+    fNiparent(0),
+    fNMaxiparent(150),
+    fIparent(0x0),
+    fDEParent(0x0),
+    fMaxIter(5),
+    fTime(time),
+    fTimeR(time),
+    fChi2(chi2),
+    fNDF(ndf),
+    fDigitType(type)
 {  
   // ctor with all data 
 
-  fNMaxPrimary = 25 ; 
-  fNMaxiparent = 50 ; 
-  fPrimary = new Int_t[fNMaxPrimary] ;
-  fIparent = new Int_t[fNMaxiparent] ; 
-  fAmp         = DigEnergy ;
-  fTime        = time ;
+  // data memebrs of the base class (AliNewDigit)
+  fAmp         = 0;
   fId          = id ;
   fIndexInList = index ; 
-  fMaxIter     = 5;
+
+  // data member  
+  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 ;  
+    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 ( i =1; i< fNMaxiparent ; i++)
+  for ( i = 1; i< fNMaxiparent ; i++) {
     fIparent[i] = -1 ;  
+    fDEParent[i] = 0 ;
+  }  
 }
 
 //____________________________________________________________________________
 AliEMCALDigit::AliEMCALDigit(const AliEMCALDigit & digit) 
+  : AliDigitNew(digit),
+    fAmpFloat(digit.fAmpFloat),        
+    fNSamples(digit.fNSamples),
+    fSamples(0x0),
+    fNSamplesHG(digit.fNSamplesHG),
+    fSamplesHG(0x0),
+    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), 
+    fChi2(digit.fChi2), 
+    fNDF(digit.fNDF),
+    fDigitType(digit.fDigitType)
 {
   // 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;
-  fMaxIter     = digit.fMaxIter;
   fIndexInList = digit.fIndexInList ; 
-  fNprimary    = digit.fNprimary ;
-  fNiparent    = digit.fNiparent ;
+
+  // data members  
+  if(fNSamples){
+    fSamples     = new Int_t[fNSamples]; 
+    for (Int_t i=0; i < digit.fNSamples; i++) fSamples[i] = digit.fSamples[i];
+  }
+  else fNSamples = 0;
+
+  if(fNSamplesHG){
+    fSamplesHG   = new Int_t[fNSamplesHG]; 
+    for (Int_t i=0; i < digit.fNSamplesHG; i++) fSamplesHG[i] = digit.fSamplesHG[i];
+  }
+  else fNSamplesHG = 0;
+
+  if(fNMaxPrimary){
+    fPrimary   = new Int_t  [fNMaxPrimary] ;  
+    fDEPrimary = new Float_t[fNMaxPrimary] ;
+    for ( Int_t i = 0; i < fNMaxPrimary ; i++) {
+      fPrimary[i]   = digit.fPrimary[i] ;
+      fDEPrimary[i] = digit.fDEPrimary[i] ;
+    }
+  }
+  else{
+    fPrimary   = 0 ;
+    fDEPrimary = 0 ;
+
+  }
+  if(fNMaxiparent){
+    fIparent  = new Int_t  [fNMaxiparent] ;
+    fDEParent = new Float_t[fNMaxiparent] ;
+    for (Int_t j = 0; j< fNMaxiparent ; j++) {
+    fIparent[j]  = digit.fIparent[j] ;
+    fDEParent[j] = digit.fDEParent[j] ;
+    }
+  }
+  else {
+    fIparent  = 0 ;
+    fDEParent = 0 ;
+  }
 }
 
 //____________________________________________________________________________
@@ -122,7 +219,12 @@ AliEMCALDigit::~AliEMCALDigit()
 {
   // Delete array of primiries if any
     delete [] fPrimary ;
+    delete [] fDEPrimary ;
     delete [] fIparent ; 
+    delete [] fDEParent ;
+    delete [] fSamples;
+    delete [] fSamplesHG;
+
 }
 
 //____________________________________________________________________________
@@ -149,35 +251,105 @@ 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.;
   Int_t id = GetId();
-  const AliEMCALGeometry *g = AliEMCALGetter::GetInstance()->EMCALGeometry();
+  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.;
   Int_t id = GetId();
-  const AliEMCALGeometry *g = AliEMCALGetter::GetInstance()->EMCALGeometry();
+  const AliEMCALGeometry *g = AliEMCALGeometry::GetInstance();
   g->EtaPhiFromIndex(id,eta,phi);
   return phi ;
 }
 
+//____________________________________________________________________________
+Bool_t AliEMCALDigit::GetFALTROSample(const Int_t iSample, Int_t& timeBin, Int_t& amp) const
+{
+       //Get FALTRO sample in time bin iSample
+       if (iSample >= fNSamples || iSample < 0 || fDigitType==kTrigger) return kFALSE;
+       
+       amp     =  fSamples[iSample] & 0xFFF;
+       timeBin = (fSamples[iSample] >> 12) & 0xFF;
+       
+       return kTRUE;
+}
+//____________________________________________________________________________
+Bool_t AliEMCALDigit::GetALTROSampleLG(const Int_t iSample, Int_t& timeBin, Int_t& amp) const
+{
+       //Get Low Gain ALTRO sample in time bin iSample
+       if (iSample >= fNSamples || iSample < 0 || fDigitType==kLG) return kFALSE;
+       
+       amp     =  fSamples[iSample] & 0xFFF;
+       timeBin = (fSamples[iSample] >> 12) & 0xFF;
+       
+       return kTRUE;
+}
+
+//____________________________________________________________________________
+void AliEMCALDigit::SetALTROSamplesLG(const Int_t nSamples, Int_t *samples)
+{
+    //Set array of ALTRO samples, Low Gain or FALTRO 
+       fNSamples = nSamples;
+       fSamples  = new Int_t[fNSamples]; 
+       for (Int_t i=0; i < fNSamples; i++) fSamples[i] = samples[i];
+}
+
+//____________________________________________________________________________
+void AliEMCALDigit::SetALTROSamplesHG(const Int_t nSamples, Int_t *samples)
+{
+       //Set array of ALTRO samples, High Gain.
+       fNSamplesHG = nSamples;
+       fSamplesHG  = new Int_t[fNSamplesHG]; 
+       for (Int_t i=0; i < fNSamplesHG; i++) fSamplesHG[i] = samples[i];
+}
+
+
+//____________________________________________________________________________
+Bool_t AliEMCALDigit::GetALTROSampleHG(const Int_t iSample, Int_t& timeBin, Int_t& amp) const
+{
+       //Get High Gain ALTRO sample in time bin iSample
+       if (iSample >= fNSamplesHG || iSample < 0 || fDigitType==kHG) return kFALSE;
+       
+       amp     =  fSamplesHG[iSample] & 0xFFF;
+       timeBin = (fSamplesHG[iSample] >> 12) & 0xFF;
+       
+       return kTRUE;
+}
+
+
 //____________________________________________________________________________
 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 ; 
   
 }
 
@@ -185,28 +357,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   ;}
@@ -225,31 +398,42 @@ 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 
   
-  fAmp += digit.fAmp ;
+  fAmpFloat += digit.fAmpFloat ;
+  for (Int_t i=0; i < fNSamples  ; i++) fSamples[i]   += digit.fSamples[i];
+  for (Int_t i=0; i < fNSamplesHG; i++) fSamplesHG[i] += digit.fSamplesHG[i];
+
+  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 ;
+    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){
-      if(max1<fNMaxPrimary){ fPrimary[max1] = digit.fPrimary[index] ; 
-      fNprimary++ ;
-      max1++;}
+    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" ; 
@@ -260,18 +444,23 @@ AliEMCALDigit& AliEMCALDigit::operator+(AliEMCALDigit const & digit)
   }
   
   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){
-     if(max2<fNMaxiparent){ fIparent[max2] = digit.fIparent[index] ; 
-      fNiparent++ ;
-      max2++;}
+    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" ; 
@@ -285,13 +474,23 @@ AliEMCALDigit& AliEMCALDigit::operator+(AliEMCALDigit const & digit)
 }
 
 //____________________________________________________________________________
-AliEMCALDigit& AliEMCALDigit::operator*(Float_t factor) 
+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)) ; 
+  //Float_t tempo = static_cast<Float_t>(fAmp) ; 
+  //tempo *= factor ; 
+  //fAmp = static_cast<Int_t>(TMath::Floor(tempo)) ; 
+       
+  fAmpFloat *= factor;
+  for (Int_t i=0; i < fNSamples  ; i++) fSamples[i]   = Int_t(factor*fSamples[i]);
+  for (Int_t i=0; i < fNSamplesHG; i++) fSamplesHG[i] = Int_t(factor*fSamplesHG[i]);
+
+  for(Int_t i=0; i < fNprimary; i++) 
+    fDEPrimary[i] *= factor;
+  for(Int_t i=0; i < fNiparent; i++) 
+    fDEParent[i] *= factor;
+
   return *this ;
 }
 
@@ -302,13 +501,53 @@ 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 ;
 }
 
+//____________________________________________________________________________
+void AliEMCALDigit::Print(const Option_t* /*opt*/) const
+{
+    //Print
+       printf("===\nDigit id: %4d / Energy %2.3f ; Time %e ; Time samples %d ; Chi2 %2.3f, NDF %d, Type? %d \n",
+                  fId, fAmpFloat,fTime, fNSamples, fChi2, fNDF, fDigitType);
+       if(fDigitType==kTrigger){
+               printf("FALTRO: ");
+               for (Int_t i=0; i < GetNFALTROSamples(); i++) 
+               {
+                       Int_t timeBin, amp;
+                       GetFALTROSample(i, timeBin, amp);
+                       printf(" (%d,%d) ",timeBin,amp);
+               }
+               printf("\n");
+       }//trigger
+       else{
+               printf("ALTRO, Low Gain: ");
+               for (Int_t i=0; i < GetNALTROSamplesLG(); i++) 
+               {
+                       Int_t timeBin, amp;
+                       GetALTROSampleLG(i, timeBin, amp);
+                       printf(" (%d,%d) ",timeBin,amp);
+               }
+               printf("\n");
+               printf("ALTRO, High Gain: ");
+               for (Int_t i=0; i < GetNALTROSamplesHG(); i++) 
+               {
+                       Int_t timeBin, amp;
+                       GetALTROSampleHG(i, timeBin, amp);
+                       printf(" (%d,%d) ",timeBin,amp);
+               }
+               printf("\n");
+       }//trigger
+       
+}
+
+