1 #ifndef ALIESDCALOCELLS_H
2 #define ALIESDCALOCELLS_H
3 /* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
8 //-------------------------------------------------------------------------
9 // ESD class to store calorimeter cell data
10 // Clone of AliAODCaloCells made by Markus Oldenburg, CERN
11 // Author: Gustavo Conesa Balbastre INFN-LNF
13 //-------------------------------------------------------------------------
16 #include <AliVCaloCells.h>
19 class AliESDCaloCells : public AliVCaloCells
24 AliESDCaloCells(const char* name, const char* title, VCells_t ttype=kUndef);
25 AliESDCaloCells(const AliESDCaloCells & cells);
26 AliESDCaloCells & operator=(const AliESDCaloCells& source);
27 virtual ~AliESDCaloCells();
29 virtual AliVCaloCells * CopyCaloCells(Bool_t all) const;
30 virtual void Copy(TObject &obj) const;
31 void Clear(const Option_t*);
32 void CreateContainer(Short_t nCells);
33 void DeleteContainer();
36 Bool_t IsEMCAL() const { return (fType == kEMCALCell); }
37 Bool_t IsPHOS() const { return (fType == kPHOSCell) ; }
38 Char_t GetType() const { return fType ; }
39 void SetType(Char_t t){ fType = t ; }
41 inline Bool_t GetCell(Short_t pos, Short_t &cellNumber, Double_t &litude, Double_t &time, Int_t &mclabel, Double_t &efrac) const;
42 Bool_t SetCell(Short_t pos, Short_t cellNumber, Double_t amplitude, Double_t time, Int_t mclabel = -1, Double_t efrac = 0.);
44 Short_t GetNumberOfCells() const { return fNCells ; }
45 void SetNumberOfCells(Int_t n) { fNCells = n ; }
48 inline Double_t GetCellAmplitude(Short_t cellNumber);
49 inline Short_t GetCellPosition(Short_t cellNumber);
50 inline Double_t GetCellTime(Short_t cellNumber);
52 inline Double_t GetAmplitude(Short_t pos) const;
53 inline Double_t GetTime(Short_t pos) const;
54 inline Short_t GetCellNumber(Short_t pos) const;
57 inline Int_t GetCellMCLabel(Short_t cellNumber) ;
58 inline Int_t GetMCLabel(Short_t pos) const ;
60 inline Double_t GetCellEFraction(Short_t cellNumber) ;
61 inline Double_t GetEFraction(Short_t pos) const ;
63 inline void SetEFraction (Short_t pos, Double32_t efrac) ;
64 inline void SetCellEFraction(Short_t cellNumber, Double32_t efrac) ;
68 Int_t fNCells; // Number of cells
69 Short_t *fCellNumber; //[fNCells] array of cell numbers
70 Double32_t *fAmplitude; //[fNCells][0.,0.,16] array with cell amplitudes (= energy!)
71 Double32_t *fTime; //[fNCells][0.,0.,16] array with cell times
72 Double32_t *fEFraction; //[fNCells][0.,0.,16] array with fraction of MC energy and data - for embedding
73 Int_t *fMCLabel; //[fNCells] array of MC labels
74 Bool_t fIsSorted; //! true if cell arrays are sorted by index
75 Char_t fType; // Cell type
77 ClassDef(AliESDCaloCells, 4);
81 Bool_t AliESDCaloCells::GetCell(Short_t pos, Short_t &cellNumber, Double_t &litude,
82 Double_t & time, Int_t & mclabel, Double_t & efrac) const
84 if (pos>=0 && pos<fNCells)
86 cellNumber = fCellNumber[pos];
87 amplitude = fAmplitude[pos];
90 if(fMCLabel) mclabel = fMCLabel[pos];
92 if(fEFraction) efrac = fEFraction[pos];
103 Double_t AliESDCaloCells::GetCellAmplitude(Short_t cellNumber)
110 Short_t pos = TMath::BinarySearch(fNCells, fCellNumber, cellNumber);
111 if (pos>=0 && pos < fNCells && fCellNumber[pos] == cellNumber ) {
112 return fAmplitude[pos];
118 Double_t AliESDCaloCells::GetCellTime(Short_t cellNumber)
125 Short_t pos = TMath::BinarySearch(fNCells, fCellNumber, cellNumber);
126 if (pos>=0 && pos < fNCells && fCellNumber[pos] == cellNumber) {
133 Double_t AliESDCaloCells::GetAmplitude(Short_t pos) const
135 if (pos>=0 && pos<fNCells) {
136 return fAmplitude[pos];
142 Double_t AliESDCaloCells::GetTime(Short_t pos) const
144 if (pos>=0 && pos<fNCells) {
151 Short_t AliESDCaloCells::GetCellNumber(Short_t pos) const
153 if (pos>=0 && pos<fNCells) {
154 return fCellNumber[pos];
160 Short_t AliESDCaloCells::GetCellPosition(Short_t cellNumber)
167 Int_t nabove, nbelow, middle;
170 nabove = fNCells + 1;
172 while (nabove - nbelow > 1) {
173 middle = (nabove + nbelow) / 2;
174 if (cellNumber == fCellNumber[middle-1]) {
178 if (cellNumber < fCellNumber[middle-1]) nabove = middle;
179 else nbelow = middle;
185 Int_t AliESDCaloCells::GetMCLabel(Short_t pos) const
187 if (pos>=0 && pos<fNCells && fMCLabel) {
188 return fMCLabel[pos];
194 Double_t AliESDCaloCells::GetEFraction(Short_t pos) const
196 if (pos>=0 && pos<fNCells && fEFraction) {
197 return fEFraction[pos];
203 Int_t AliESDCaloCells::GetCellMCLabel(Short_t cellNumber)
210 Short_t pos = TMath::BinarySearch(fNCells, fCellNumber, cellNumber);
211 if (pos>=0 && fCellNumber[pos] == cellNumber && fMCLabel) {
212 return fMCLabel[pos];
218 Double_t AliESDCaloCells::GetCellEFraction(Short_t cellNumber)
225 Short_t pos = TMath::BinarySearch(fNCells, fCellNumber, cellNumber);
226 if (pos>=0 && pos < fNCells && fCellNumber[pos] == cellNumber && fEFraction) {
227 return fEFraction[pos];
233 void AliESDCaloCells::SetEFraction(Short_t pos, Double32_t efrac)
235 // Sets the fraction of energy from MC with respect to data at the given position
237 if (pos>=0 && pos < fNCells)
239 if(!fEFraction) fEFraction = new Double32_t[fNCells];
240 fEFraction[pos] = efrac;
244 void AliESDCaloCells::SetCellEFraction(Short_t cellNumber, Double32_t efrac)
251 Short_t pos = TMath::BinarySearch(fNCells, fCellNumber, cellNumber);
252 if (pos>=0 && pos < fNCells && fCellNumber[pos] == cellNumber)
254 if(!fEFraction) fEFraction = new Double32_t[fNCells];
255 fEFraction[pos] = efrac;