Number of sigma pedestal cut increased to 4
[u/mrichter/AliRoot.git] / STEER / AOD / AliAODCaloCells.h
1 /* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
2  * See cxx source for full Copyright notice                               */
3
4 /* $Id$ */
5
6 //-------------------------------------------------------------------------
7 //     AOD class to store calorimeter cell data
8 //     Author: Markus Oldenburg, CERN
9 //-------------------------------------------------------------------------
10
11 #ifndef ALIAODCELLS_H
12 #define ALIAODCELLS_H
13
14 #include <AliVCaloCells.h>
15 #include <TMath.h>
16
17 class AliAODCaloCells : public AliVCaloCells
18 {
19  public:
20   AliAODCaloCells();
21   AliAODCaloCells(const char* name, const char* title, VCells_t ttype=kUndef);
22   AliAODCaloCells(const AliAODCaloCells& cells); 
23   AliAODCaloCells& operator=(const AliAODCaloCells& cells);
24   virtual void Copy(TObject &obj) const;
25   virtual AliVCaloCells* CopyCaloCells(Bool_t all) const;
26
27   virtual ~AliAODCaloCells();
28   void Clear(const Option_t*);
29
30   void CreateContainer(Short_t nCells);
31   void DeleteContainer();
32   void Sort();
33   
34   Bool_t SetCell(Short_t pos, Short_t cellNumber, Double_t amplitude, Double_t time = -1);
35   
36   Short_t GetNumberOfCells() const { return fNCells; }
37   void    SetNumberOfCells(Int_t n) { fNCells = n ; }
38   inline Bool_t   GetCell(Short_t pos, Short_t &cellNumber, Double_t &amplitude, Double_t &time) const;
39   inline Double_t GetCellAmplitude(Short_t cellNumber);
40   inline Short_t  GetCellPosition(Short_t cellNumber);
41   inline Double_t GetAmplitude(Short_t pos) const;
42   Double_t GetCellTime(Short_t /*cellNumber*/) {return -1;}
43   inline Short_t  GetCellNumber(Short_t pos) const;
44   Double_t GetTime(Short_t /*pos*/) const {return -1;}
45   Bool_t IsEMCAL() const {return (fType == kEMCALCell);}
46   Bool_t IsPHOS() const {return (fType == kPHOSCell);}
47   Char_t  GetType() const { return fType;}
48   void    SetType(Char_t ttype) { fType=ttype; }
49
50
51  protected:
52   Int_t       fNCells;       // Number of cells
53   Short_t    *fCellNumber;   //[fNCells] array of cell numbers
54   Double32_t *fAmplitude;    //[fNCells][0.,0.,16] array with cell amplitudes (= energy!)
55   Bool_t      fIsSorted;     //! true if cell arrays are sorted by index
56   Char_t      fType;         // Cell type
57   
58   
59   ClassDef(AliAODCaloCells, 2);
60 };
61
62
63 Bool_t AliAODCaloCells::GetCell(Short_t pos, Short_t &cellNumber, Double_t &amplitude, Double_t& /*time*/) const 
64
65   if (pos>=0 && pos<fNCells) {
66     cellNumber = fCellNumber[pos];
67     amplitude = fAmplitude[pos];
68     return kTRUE;
69   } else {
70     Warning("GetCell","Invalid cell array index %d", pos);
71     return kFALSE;
72   }
73 }
74
75
76 Double_t AliAODCaloCells::GetCellAmplitude(Short_t cellNumber)
77
78   if (!fIsSorted) {
79     Sort();
80     fIsSorted=kTRUE;
81   }
82
83   Short_t pos = TMath::BinarySearch(fNCells, fCellNumber, cellNumber);
84   if (pos>=0 && fCellNumber[pos] == cellNumber) {
85     return fAmplitude[pos];
86   } else {
87     return 0.;
88   }
89 }
90
91
92 Double_t AliAODCaloCells::GetAmplitude(Short_t pos) const 
93
94   if (pos>=0 && pos<fNCells) {
95     return fAmplitude[pos];
96   } else {
97     Warning("GetAmplitude","Invalid cell array index %d", pos);
98     return 0.;
99   }
100 }
101
102
103 Short_t AliAODCaloCells::GetCellNumber(Short_t pos) const 
104
105   if (pos>=0 && pos<fNCells) {
106     return fCellNumber[pos];
107   } else {
108     Warning("GetCellNumber","Invalid cell array index %d", pos);
109     return fNCells;
110   }
111 }
112
113 Short_t AliAODCaloCells::GetCellPosition(Short_t cellNumber)
114
115   if (!fIsSorted) {
116     Sort();
117     fIsSorted=kTRUE;
118   }
119
120    Int_t nabove, nbelow, middle;
121    Short_t pos = -1;
122
123    nabove = fNCells + 1;
124    nbelow = 0;
125    while (nabove - nbelow > 1) {
126       middle = (nabove + nbelow) / 2;
127       if (cellNumber == fCellNumber[middle-1]) {
128           pos =   middle - 1;
129           break;
130       }
131       if (cellNumber  < fCellNumber[middle-1]) nabove = middle;
132       else                                     nbelow = middle;
133    }
134
135   return pos;
136 }
137
138
139 #endif