add protection against truncated events + coverity - Rachid
[u/mrichter/AliRoot.git] / EMCAL / AliEMCALDigit.h
1 #ifndef ALIEMCALDIGIT_H
2 #define ALIEMCALDIGIT_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 /* $Id$ */
7
8 //_________________________________________________________________________
9 //  EMCAL digit: 
10 // 
11 //  A  Digit is the sum of energy in a Tower (Hit sum) and stores information, about primaries
12 //  and entering particle contributing to a Digit
13 //
14 //*-- Author: Sahal Yacoob (LBL)
15 // based on : AliPHOSDigit
16 //___________________________________________________________________________
17
18 // --- ROOT system ---
19
20 #include "TObject.h" 
21
22 // --- Standard library ---
23
24 // --- AliRoot header files ---
25 #include "AliDigitNew.h"
26
27 class AliEMCALDigit : public AliDigitNew {
28
29   friend ostream& operator << ( ostream& , const AliEMCALDigit&) ;
30
31  public:
32   
33   AliEMCALDigit() ;
34   AliEMCALDigit(Int_t primary, Int_t iparent, Int_t id, Float_t digEnergy, Float_t time, Int_t type,Int_t index = -1, Float_t chi2=0, Int_t ndf=0, Float_t dE = 0) ;
35   AliEMCALDigit(const AliEMCALDigit & digit) ;
36   virtual ~AliEMCALDigit() ;
37
38   Bool_t operator==(const AliEMCALDigit &rValue) const;
39   AliEMCALDigit operator+(const AliEMCALDigit &rValue) ;
40   AliEMCALDigit operator*(Float_t factor) ; 
41   const AliEMCALDigit& operator = (const AliEMCALDigit &) {return *this;}
42   enum digitType{kUnknown=-1, kHG=0, kLG=1, kLGnoHG=2, kTrigger=3};
43
44   void    Clear(const Option_t*) ;      
45   Int_t   Compare(const TObject * obj) const ;
46   Float_t GetAmplitude()   const { if(!fAmp)return fAmpFloat ; else return fAmp ;}//Keep backward compatibility.
47   Float_t GetEta()         const ; 
48   Int_t   GetNprimary()    const { return fNprimary ;}
49   Int_t   GetPrimary(Int_t index)   const ; 
50   Float_t GetDEPrimary(Int_t index) const ; 
51   Int_t   GetNiparent()    const { return fNiparent ;}
52   Int_t   GetIparent(Int_t index)   const ;
53   Float_t GetDEParent(Int_t index)  const ; 
54   Float_t GetPhi()         const ;
55   Float_t GetTime(void)    const { return fTime  ;}
56   Float_t GetTimeR(void)   const { return fTimeR ;}
57   Float_t GetChi2(void)    const { return fChi2  ;}
58   Int_t   GetNDF(void)     const { return fNDF   ;}
59   Bool_t  IsSortable()     const { return kTRUE  ;}
60   Int_t   GetType()        const { return fDigitType ;}
61         
62   void    SetAmp(Int_t amp)         { fAmp       = amp  ;} //old
63   void    SetAmplitude(Float_t amp) { fAmpFloat  = amp  ;}
64   void    SetId(Int_t idt)          { fId        = idt  ;}
65   void    SetTime(Float_t time)     { fTime      = time ;}
66   void    SetTimeR(Float_t time)    { fTimeR     = time ;}
67   void    SetChi2(Float_t chi)      { fChi2      = chi  ;}
68   void    SetNDF(Int_t ndf)         { fNDF       = ndf  ;}
69   void    SetType(Int_t t)          { fDigitType = t    ;}
70   void    ShiftPrimary(Int_t shift); // shift to separate different TreeK in merging
71
72   //Raw time sample
73   //ALTRO
74   Int_t   GetNALTROSamplesLG() const {if(fDigitType==kLG)return fNSamples; else return 0;}
75   Bool_t  GetALTROSampleLG(const Int_t iSample, Int_t& timeBin, Int_t& amp) const;
76   Int_t   GetNALTROSamplesHG() const {if(fDigitType==kHG) return fNSamplesHG; else return 0;}
77   Bool_t  GetALTROSampleHG(const Int_t iSample, Int_t& timeBin, Int_t& amp) const;
78   //FALTRO, trigger. Same data members as Low Gain      
79   Int_t   GetNFALTROSamples() const 
80         {if(fDigitType==kTrigger) return fNSamples; else return 0;}
81   Bool_t  GetFALTROSample(const Int_t iSample, Int_t& timeBin, Int_t& amp) const ;
82         
83   void SetALTROSamplesHG (const Int_t nSamplesHG, Int_t *samplesHG);
84   void SetALTROSamplesLG (const Int_t nSamplesLG, Int_t *samplesLG);
85   void SetFALTROSamples  (const Int_t nSamples,   Int_t *samples) { if(fDigitType==kTrigger) SetALTROSamplesLG(nSamples, samples);} 
86
87   void    SetCalibAmp(Float_t amp)  { fAmpCalib = amp; }
88   Double_t GetCalibAmp()   const    { return fAmpCalib; }
89
90   void Print(const Option_t* /*opt*/) const;
91         
92  private: 
93         
94   Float_t fAmpFloat;    // Cell amplitude, float
95   Int_t   fNSamples;    // Number of time samples, Low Gain for ALTRO, used also for FALTRO 
96   Int_t*  fSamples;         //[fNSamples], list of time bin constents, Low Gain for ALTRO, used also for FALTRO 
97   Int_t   fNSamplesHG;  // Number of time samples, High Gain for ALTRO
98   Int_t*  fSamplesHG;   //[fNSamples], list of time bin constents, High Gain for ALTRO, used also for FALTRO 
99         
100   Int_t fNprimary ;     // Number of primaries
101   Int_t fNMaxPrimary ;  // Max Number of primaries
102   Int_t *fPrimary ;     //[fNMaxPrimary]  Array of primaries       
103   Float_t *fDEPrimary;  //[fNMaxPrimary]  Array of primary energy contributions
104     
105   Int_t fNiparent ;     // Number of initial parents 
106   Int_t fNMaxiparent ;  // Max Number of parents 
107   Int_t *fIparent ;     //[fNMaxiparent] Array of parents       
108   Float_t *fDEParent;   //[fNMaxiparent]  Array of parent energy contributions
109   Int_t fMaxIter  ;     // Number to Increment Maxiparent, and MaxPrimary if default is not sufficient
110   Float_t fTime ;       // Calculated time  
111   Float_t fTimeR ;      // Earliest time: to be used by Digits2Raw
112   //Fit quality parameters
113   Float_t fChi2;        // Fit Chi aquare       
114   Int_t   fNDF;         // Fit Number of Degrees of Freedom
115         
116   Int_t fDigitType;     // This is a trigger digit(0), HG (1) or LG (3)
117   Float_t fAmpCalib;    //!Calibrated energy
118
119   ClassDef(AliEMCALDigit,6)   // Digit in EMCAL 
120 } ;
121
122 #endif //  ALIEMCALDIGIT_H