]>
Commit | Line | Data |
---|---|---|
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 */ | |
5 | /* $Id$ */ | |
6 | /* $Log $ */ | |
7 | ||
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 | |
12 | // | |
13 | //------------------------------------------------------------------------- | |
14 | ||
15 | ||
16 | #include <AliVCaloCells.h> | |
17 | #include <TMath.h> | |
18 | ||
19 | class AliESDCaloCells : public AliVCaloCells | |
20 | { | |
21 | public: | |
22 | ||
23 | AliESDCaloCells(); | |
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(); | |
28 | ||
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(); | |
34 | void Sort(); | |
35 | ||
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 ; } | |
40 | ||
41 | inline Bool_t GetCell(Short_t pos, Short_t &cellNumber, Double_t &litude, Double_t &time, Short_t &mclabel, Double_t &efrac) const; | |
42 | Bool_t SetCell(Short_t pos, Short_t cellNumber, Double_t amplitude, Double_t time, Short_t mclabel = -1, Double_t efrac = 0.); | |
43 | ||
44 | Short_t GetNumberOfCells() const { return fNCells ; } | |
45 | void SetNumberOfCells(Int_t n) { fNCells = n ; } | |
46 | ||
47 | ||
48 | inline Double_t GetCellAmplitude(Short_t cellNumber); | |
49 | inline Short_t GetCellPosition(Short_t cellNumber); | |
50 | inline Double_t GetCellTime(Short_t cellNumber); | |
51 | ||
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; | |
55 | ||
56 | // MC & embedding | |
57 | inline Short_t GetCellMCLabel(Short_t cellNumber) ; | |
58 | inline Short_t GetMCLabel(Short_t pos) const ; | |
59 | ||
60 | inline Double_t GetCellEFraction(Short_t cellNumber) ; | |
61 | inline Double_t GetEFraction(Short_t pos) const ; | |
62 | ||
63 | inline void SetEFraction (Short_t pos, Double32_t efrac) ; | |
64 | inline void SetCellEFraction(Short_t cellNumber, Double32_t efrac) ; | |
65 | ||
66 | protected: | |
67 | ||
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 | Short_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 | |
76 | ||
77 | ClassDef(AliESDCaloCells, 3); | |
78 | }; | |
79 | ||
80 | ||
81 | Bool_t AliESDCaloCells::GetCell(Short_t pos, Short_t &cellNumber, Double_t &litude, | |
82 | Double_t & time, Short_t & mclabel, Double_t & efrac) const | |
83 | { | |
84 | if (pos>=0 && pos<fNCells) | |
85 | { | |
86 | cellNumber = fCellNumber[pos]; | |
87 | amplitude = fAmplitude[pos]; | |
88 | time = fTime[pos]; | |
89 | ||
90 | if(fMCLabel) mclabel = fMCLabel[pos]; | |
91 | else mclabel =-1 ; | |
92 | if(fEFraction) efrac = fEFraction[pos]; | |
93 | else efrac = 0 ; | |
94 | ||
95 | return kTRUE; | |
96 | ||
97 | } else | |
98 | { | |
99 | return kFALSE; | |
100 | } | |
101 | } | |
102 | ||
103 | Double_t AliESDCaloCells::GetCellAmplitude(Short_t cellNumber) | |
104 | { | |
105 | if (!fIsSorted) { | |
106 | Sort(); | |
107 | fIsSorted=kTRUE; | |
108 | } | |
109 | ||
110 | Short_t pos = TMath::BinarySearch(fNCells, fCellNumber, cellNumber); | |
111 | if (pos>=0 && pos < fNCells && fCellNumber[pos] == cellNumber ) { | |
112 | return fAmplitude[pos]; | |
113 | } else { | |
114 | return 0.; | |
115 | } | |
116 | } | |
117 | ||
118 | Double_t AliESDCaloCells::GetCellTime(Short_t cellNumber) | |
119 | { | |
120 | if (!fIsSorted) { | |
121 | Sort(); | |
122 | fIsSorted=kTRUE; | |
123 | } | |
124 | ||
125 | Short_t pos = TMath::BinarySearch(fNCells, fCellNumber, cellNumber); | |
126 | if (pos>=0 && pos < fNCells && fCellNumber[pos] == cellNumber) { | |
127 | return fTime[pos]; | |
128 | } else { | |
129 | return -1.; | |
130 | } | |
131 | } | |
132 | ||
133 | Double_t AliESDCaloCells::GetAmplitude(Short_t pos) const | |
134 | { | |
135 | if (pos>=0 && pos<fNCells) { | |
136 | return fAmplitude[pos]; | |
137 | } else { | |
138 | return 0.; | |
139 | } | |
140 | } | |
141 | ||
142 | Double_t AliESDCaloCells::GetTime(Short_t pos) const | |
143 | { | |
144 | if (pos>=0 && pos<fNCells) { | |
145 | return fTime[pos]; | |
146 | } else { | |
147 | return -1.; | |
148 | } | |
149 | } | |
150 | ||
151 | Short_t AliESDCaloCells::GetCellNumber(Short_t pos) const | |
152 | { | |
153 | if (pos>=0 && pos<fNCells) { | |
154 | return fCellNumber[pos]; | |
155 | } else { | |
156 | return fNCells; | |
157 | } | |
158 | } | |
159 | ||
160 | Short_t AliESDCaloCells::GetCellPosition(Short_t cellNumber) | |
161 | { | |
162 | if (!fIsSorted) { | |
163 | Sort(); | |
164 | fIsSorted=kTRUE; | |
165 | } | |
166 | ||
167 | Int_t nabove, nbelow, middle; | |
168 | Short_t pos = -1; | |
169 | ||
170 | nabove = fNCells + 1; | |
171 | nbelow = 0; | |
172 | while (nabove - nbelow > 1) { | |
173 | middle = (nabove + nbelow) / 2; | |
174 | if (cellNumber == fCellNumber[middle-1]) { | |
175 | pos = middle - 1; | |
176 | break; | |
177 | } | |
178 | if (cellNumber < fCellNumber[middle-1]) nabove = middle; | |
179 | else nbelow = middle; | |
180 | } | |
181 | ||
182 | return pos; | |
183 | } | |
184 | ||
185 | Short_t AliESDCaloCells::GetMCLabel(Short_t pos) const | |
186 | { | |
187 | if (pos>=0 && pos<fNCells && fMCLabel) { | |
188 | return fMCLabel[pos]; | |
189 | } else { | |
190 | return -1 ; | |
191 | } | |
192 | } | |
193 | ||
194 | Double_t AliESDCaloCells::GetEFraction(Short_t pos) const | |
195 | { | |
196 | if (pos>=0 && pos<fNCells && fEFraction) { | |
197 | return fEFraction[pos]; | |
198 | } else { | |
199 | return 0.; | |
200 | } | |
201 | } | |
202 | ||
203 | Short_t AliESDCaloCells::GetCellMCLabel(Short_t cellNumber) | |
204 | { | |
205 | if (!fIsSorted) { | |
206 | Sort(); | |
207 | fIsSorted=kTRUE; | |
208 | } | |
209 | ||
210 | Short_t pos = TMath::BinarySearch(fNCells, fCellNumber, cellNumber); | |
211 | if (pos>=0 && fCellNumber[pos] == cellNumber && fMCLabel) { | |
212 | return fMCLabel[pos]; | |
213 | } else { | |
214 | return -1; | |
215 | } | |
216 | } | |
217 | ||
218 | Double_t AliESDCaloCells::GetCellEFraction(Short_t cellNumber) | |
219 | { | |
220 | if (!fIsSorted) { | |
221 | Sort(); | |
222 | fIsSorted=kTRUE; | |
223 | } | |
224 | ||
225 | Short_t pos = TMath::BinarySearch(fNCells, fCellNumber, cellNumber); | |
226 | if (pos>=0 && pos < fNCells && fCellNumber[pos] == cellNumber && fEFraction) { | |
227 | return fEFraction[pos]; | |
228 | } else { | |
229 | return 0.; | |
230 | } | |
231 | } | |
232 | ||
233 | void AliESDCaloCells::SetEFraction(Short_t pos, Double32_t efrac) | |
234 | { | |
235 | // Sets the fraction of energy from MC with respect to data at the given position | |
236 | ||
237 | if (pos>=0 && pos < fNCells) | |
238 | { | |
239 | if(!fEFraction) fEFraction = new Double32_t[fNCells]; | |
240 | fEFraction[pos] = efrac; | |
241 | } | |
242 | } | |
243 | ||
244 | void AliESDCaloCells::SetCellEFraction(Short_t cellNumber, Double32_t efrac) | |
245 | { | |
246 | if (!fIsSorted) { | |
247 | Sort(); | |
248 | fIsSorted=kTRUE; | |
249 | } | |
250 | ||
251 | Short_t pos = TMath::BinarySearch(fNCells, fCellNumber, cellNumber); | |
252 | if (pos>=0 && pos < fNCells && fCellNumber[pos] == cellNumber) | |
253 | { | |
254 | if(!fEFraction) fEFraction = new Double32_t[fNCells]; | |
255 | fEFraction[pos] = efrac; | |
256 | } | |
257 | } | |
258 | ||
259 | ||
260 | #endif |