]>
Commit | Line | Data |
---|---|---|
563113d0 | 1 | /* $Id$ */ |
2 | ||
3 | //--------------------------------------------------------------------// | |
4 | // // | |
5 | // AliCFContainer Class // | |
6 | // Class to accumulate data on an N-dimensional grids, at different // | |
7 | // selection stages. To be used as an input to get corrections for // | |
8 | // Reconstruction & Trigger efficiency // | |
9 | // // | |
10 | // -- Author : S.Arcelli // | |
11 | //--------------------------------------------------------------------// | |
12 | // | |
13 | // | |
14 | #include <AliLog.h> | |
15 | #include "AliCFGrid.h" | |
16 | #include "AliCFContainer.h" | |
17 | ||
18 | //____________________________________________________________________ | |
19 | ClassImp(AliCFContainer) | |
20 | ||
21 | //____________________________________________________________________ | |
22 | AliCFContainer::AliCFContainer() : | |
23 | AliCFFrame(), | |
24 | fNStep(0), | |
25 | fGrid(0x0) | |
26 | { | |
27 | // | |
28 | // default constructor | |
29 | // | |
30 | } | |
31 | //____________________________________________________________________ | |
32 | AliCFContainer::AliCFContainer(const Char_t* name, const Char_t* title) : | |
33 | AliCFFrame(name,title), | |
34 | fNStep(0), | |
35 | fGrid(0x0) | |
36 | { | |
37 | // default constructor | |
38 | } | |
39 | ||
40 | //____________________________________________________________________ | |
41 | AliCFContainer::AliCFContainer(const Char_t* name, const Char_t* title,const Int_t nSelSteps, const Int_t nVarIn, const Int_t * nBinIn, const Float_t *binLimitsIn) : | |
42 | AliCFFrame(name,title,nVarIn,nBinIn,binLimitsIn), | |
43 | fNStep(0), | |
44 | fGrid(0x0) | |
45 | { | |
46 | // | |
47 | // main constructor | |
48 | // | |
49 | ||
50 | // The selection steps | |
51 | fNStep=nSelSteps; | |
52 | ||
53 | // The grids | |
54 | fGrid = new AliCFGrid*[fNStep]; //the grids at the various selection steps | |
55 | char gname[30]; | |
56 | for(Int_t istep=0;istep<fNStep;istep++){ | |
57 | sprintf(gname,"%s%s%i",GetName(),"_SelStep", istep); | |
58 | fGrid[istep] = new AliCFGrid(gname,title,nVarIn,nBinIn,binLimitsIn); | |
59 | } | |
60 | } | |
61 | //____________________________________________________________________ | |
62 | AliCFContainer::AliCFContainer(const AliCFContainer& c) : | |
63 | AliCFFrame(), | |
64 | fNStep(0), | |
65 | fGrid(0x0) | |
66 | { | |
67 | // | |
68 | // copy constructor | |
69 | // | |
70 | ((AliCFContainer &)c).Copy(*this); | |
71 | } | |
72 | //____________________________________________________________________ | |
73 | AliCFContainer::~AliCFContainer() | |
74 | { | |
75 | // | |
76 | // destructor | |
77 | // | |
78 | if(fGrid)delete [] fGrid; | |
79 | ||
80 | } | |
81 | //____________________________________________________________________ | |
82 | AliCFContainer &AliCFContainer::operator=(const AliCFContainer &c) | |
83 | { | |
84 | // | |
85 | // assigment operator | |
86 | // | |
87 | if (this != &c) | |
88 | ((AliCFContainer &) c).Copy(*this); | |
89 | return *this; | |
90 | } | |
91 | //____________________________________________________________________ | |
92 | void AliCFContainer::SetBinLimits(Int_t varindex, Float_t *array) | |
93 | { | |
94 | // | |
95 | // setting the arrays containing the bin limits | |
96 | // | |
97 | Int_t nbins=fNVarBins[varindex]+1; | |
98 | for(Int_t i=0;i<nbins;i++){ | |
99 | fVarBinLimits[fOffset[varindex]+i] =array[i]; | |
100 | } | |
101 | for(Int_t istep=0;istep<fNStep;istep++){ | |
102 | fGrid[istep]->SetBinLimits(varindex,array); | |
103 | } | |
104 | } | |
105 | //____________________________________________________________________ | |
106 | void AliCFContainer::Copy(TObject& c) const | |
107 | { | |
108 | // | |
109 | // copy function | |
110 | // | |
111 | AliCFContainer& target = (AliCFContainer &) c; | |
112 | target.fNStep=fNStep; | |
113 | target.fNVar=fNVar; | |
114 | target.fNDim=fNDim; | |
115 | target.fNVarBinLimits=fNVarBinLimits; | |
116 | if (fNVarBins) | |
117 | target.fNVarBins = fNVarBins; | |
118 | if (fVarBinLimits) | |
119 | target.fVarBinLimits = fVarBinLimits; | |
120 | if (fGrid) | |
121 | target.fGrid = fGrid; | |
122 | for(Int_t istep=0;istep<fNStep;istep++){ | |
123 | for(Int_t iel=0;iel<fNDim;iel++){ | |
124 | target.fGrid[istep]->SetElement(iel,fGrid[istep]->GetElement(iel)); | |
125 | } | |
126 | } | |
127 | } | |
128 | //____________________________________________________________________ | |
129 | void AliCFContainer::Fill(Float_t *var, Int_t istep, Float_t weight) | |
130 | { | |
131 | // | |
132 | // Fills the grid at selection step istep for a set of values of the | |
133 | // input variables, with a given weight (by default w=1) | |
134 | // | |
135 | fGrid[istep]->Fill(var,weight); | |
136 | } | |
137 | //___________________________________________________________________ | |
138 | TH1F *AliCFContainer::ShowProjection(Int_t ivar, Int_t istep) const | |
139 | { | |
140 | // | |
141 | // returns 1-D projection along variable ivar at selection step istep | |
142 | // | |
143 | return fGrid[istep]->Project(ivar); | |
144 | } | |
145 | //___________________________________________________________________ | |
146 | TH2F *AliCFContainer::ShowProjection(Int_t ivar1, Int_t ivar2, Int_t istep) const | |
147 | { | |
148 | // | |
149 | // returns 2-D projection along variables ivar1,ivar2 at selection step istep | |
150 | // | |
151 | return fGrid[istep]->Project(ivar1,ivar2); | |
152 | } | |
153 | //___________________________________________________________________ | |
154 | TH3F *AliCFContainer::ShowProjection(Int_t ivar1, Int_t ivar2, Int_t ivar3, Int_t istep) const | |
155 | { | |
156 | // | |
157 | // returns 3-D projection along variables ivar1,ivar2,ivar3 | |
158 | // at selection step istep | |
159 | // | |
160 | return fGrid[istep]->Project(ivar1,ivar2,ivar3); | |
161 | } | |
162 | //___________________________________________________________________ | |
163 | TH1F *AliCFContainer::ShowSlice(Int_t ivar, Float_t *varMin, Float_t* varMax, Int_t istep) const | |
164 | { | |
165 | // | |
166 | // Make a slice along variable ivar at selection level istep in range [varMin,varMax] | |
167 | // | |
168 | return (TH1F*)fGrid[istep]->Slice(ivar,varMin,varMax); | |
169 | } | |
170 | //____________________________________________________________________ | |
171 | Long64_t AliCFContainer::Merge(TCollection* list) | |
172 | { | |
173 | // Merge a list of AliCorrection objects with this (needed for | |
174 | // PROOF). | |
175 | // Returns the number of merged objects (including this). | |
176 | ||
177 | if (!list) | |
178 | return 0; | |
179 | ||
180 | if (list->IsEmpty()) | |
181 | return 1; | |
182 | ||
183 | TIterator* iter = list->MakeIterator(); | |
184 | TObject* obj; | |
185 | ||
186 | Int_t count = 0; | |
187 | while ((obj = iter->Next())) { | |
188 | AliCFContainer* entry = dynamic_cast<AliCFContainer*> (obj); | |
189 | if (entry == 0) | |
190 | continue; | |
191 | this->Add(entry); | |
192 | count++; | |
193 | } | |
194 | ||
195 | return count+1; | |
196 | } | |
197 | ||
198 | //____________________________________________________________________ | |
199 | void AliCFContainer::Add(AliCFContainer* aContainerToAdd, Float_t c) | |
200 | { | |
201 | // | |
202 | //add the content of container aContainerToAdd to the current one | |
203 | // | |
204 | ||
205 | if(aContainerToAdd->GetNStep()!=fNStep)AliError("Different number of steps, cannot add the containers"); | |
206 | if(aContainerToAdd->GetNVar()!=fNVar)AliError("Different number of variables, cannot add the containers"); | |
207 | if(aContainerToAdd->GetNDim()!=fNDim)AliError("Different number of dimensions, cannot add the containers!"); | |
208 | ||
209 | for(Int_t istep=0;istep<fNStep;istep++){ | |
210 | fGrid[istep]->Add(aContainerToAdd->GetGrid(istep),c); | |
211 | } | |
212 | } | |
213 | //____________________________________________________________________ | |
214 | Float_t AliCFContainer::GetOverFlows( Int_t ivar, Int_t istep) const { | |
215 | // | |
216 | // Get overflows in variable var at selection level istep | |
217 | // | |
218 | return fGrid[istep]->GetOverFlows(ivar); | |
219 | } | |
220 | //____________________________________________________________________ | |
221 | Float_t AliCFContainer::GetOverFlows( Int_t istep) const { | |
222 | // | |
223 | // Get overflows in variable var at selection level istep | |
224 | // | |
225 | return fGrid[istep]->GetOverFlows(); | |
226 | } | |
227 | //____________________________________________________________________ | |
228 | Float_t AliCFContainer::GetUnderFlows( Int_t ivar, Int_t istep) const { | |
229 | // | |
230 | // Get overflows in variable var at selection level istep | |
231 | // | |
232 | return fGrid[istep]->GetUnderFlows(ivar); | |
233 | } | |
234 | //____________________________________________________________________ | |
235 | Float_t AliCFContainer::GetUnderFlows( Int_t istep) const { | |
236 | // | |
237 | // Get overflows in variable var at selection level istep | |
238 | // | |
239 | return fGrid[istep]->GetUnderFlows(); | |
240 | } | |
241 | //____________________________________________________________________ | |
242 | Float_t AliCFContainer::GetEntries( Int_t istep) const { | |
243 | // | |
244 | // Get overflows in variable var at selection level istep | |
245 | // | |
246 | return fGrid[istep]->GetEntries(); | |
247 | } | |
248 | //____________________________________________________________________ | |
249 | Int_t AliCFContainer::GetEmptyBins( Int_t istep) const { | |
250 | // | |
251 | // Get overflows in variable var at selection level istep | |
252 | // | |
253 | return fGrid[istep]->GetEmptyBins(); | |
254 | } | |
255 | //____________________________________________________________________ | |
256 | Int_t AliCFContainer::GetEmptyBins( Int_t istep, Float_t *varMin, Float_t* varMax) const { | |
257 | // | |
258 | // Get overflows in variable var at selection level istep | |
259 | // | |
260 | return fGrid[istep]->GetEmptyBins(varMin,varMax); | |
261 | } | |
262 | //_____________________________________________________________________ | |
263 | Float_t AliCFContainer::GetIntegral( Int_t istep) const | |
264 | { | |
265 | // | |
266 | // Get Integral at selection level istep | |
267 | // | |
268 | return fGrid[istep]->GetIntegral(); | |
269 | } | |
270 | //_____________________________________________________________________ | |
271 | Float_t AliCFContainer::GetIntegral( Int_t istep, Float_t *varMin, Float_t* varMax ) const | |
272 | { | |
273 | // | |
274 | // Get Integral at selection level istep | |
275 | // | |
276 | return fGrid[istep]->GetIntegral(varMin,varMax); | |
277 | } |