Keep only the forward mapping array persistent, the inverse one is created on demand...
[u/mrichter/AliRoot.git] / PWG0 / AliCorrectionMatrix.cxx
1 /* $Id$ */
2
3 // ------------------------------------------------------
4 //
5 // Class to handle corrections.
6 //
7 // ------------------------------------------------------
8 //
9
10 #include <TFile.h>
11 #include <TCanvas.h>
12 #include <TH2F.h>
13
14 #include <AliLog.h>
15
16 #include "AliCorrectionMatrix.h"
17
18 //____________________________________________________________________
19 ClassImp(AliCorrectionMatrix)
20
21 //____________________________________________________________________
22 AliCorrectionMatrix::AliCorrectionMatrix() : TNamed(),
23   fhMeas(0),
24   fhGene(0),
25   fhCorr(0)
26 {
27   // default constructor
28 }
29
30 //____________________________________________________________________
31 AliCorrectionMatrix::AliCorrectionMatrix(const Char_t* name, const Char_t* title) : TNamed(name, title),
32   fhMeas(0),
33   fhGene(0),
34   fhCorr(0)
35 {
36   // constructor initializing tnamed
37 }
38
39 //____________________________________________________________________
40 AliCorrectionMatrix::AliCorrectionMatrix(const AliCorrectionMatrix& c) : TNamed(c),
41   fhMeas(0),
42   fhGene(0),
43   fhCorr(0)
44 {
45   // copy constructor
46   ((AliCorrectionMatrix &)c).Copy(*this);
47 }
48
49 //____________________________________________________________________
50 AliCorrectionMatrix::~AliCorrectionMatrix()
51 {
52   //
53   // destructor
54   //
55
56   if (fhMeas)
57   {
58     delete fhMeas;
59     fhMeas = 0;
60   }
61
62   if (fhGene)
63   {
64     delete fhGene;
65     fhGene = 0;
66   }
67
68   if (fhCorr)
69   {
70     delete fhCorr;
71     fhCorr = 0;
72   }
73 }
74
75 //____________________________________________________________________
76 AliCorrectionMatrix &AliCorrectionMatrix::operator=(const AliCorrectionMatrix &c)
77 {
78   // assigment operator
79
80   if (this != &c)
81     ((AliCorrectionMatrix &) c).Copy(*this);
82
83   return *this;
84 }
85
86 //____________________________________________________________________
87 void AliCorrectionMatrix::Copy(TObject& c) const
88 {
89   // copy function
90
91   AliCorrectionMatrix& target = (AliCorrectionMatrix &) c;
92
93   if (fhMeas)
94     target.fhMeas = dynamic_cast<TH1*> (fhMeas->Clone());
95
96   if (fhGene)
97     target.fhGene = dynamic_cast<TH1*> (fhGene->Clone());
98
99   if (fhCorr)
100     target.fhCorr = dynamic_cast<TH1*> (fhCorr->Clone());
101 }
102
103 //________________________________________________________________________
104 void AliCorrectionMatrix::SetAxisTitles(const Char_t* titleX, const Char_t* titleY, const Char_t* titleZ)
105 {
106   //
107   // method for setting the axis titles of the histograms
108   //
109
110   fhMeas ->SetXTitle(titleX);  fhMeas ->SetYTitle(titleY);  fhMeas ->SetZTitle(titleZ);
111   fhGene ->SetXTitle(titleX);  fhGene ->SetYTitle(titleY);  fhGene ->SetZTitle(titleZ);
112   fhCorr ->SetXTitle(titleX);  fhCorr ->SetYTitle(titleY);  fhCorr ->SetZTitle(titleZ);
113 }
114
115 //____________________________________________________________________
116 Long64_t AliCorrectionMatrix::Merge(TCollection* list)
117 {
118   // Merge a list of AliCorrectionMatrix objects with this (needed for
119   // PROOF). 
120   // Returns the number of merged objects (including this).
121
122   if (!list)
123     return 0;
124   
125   if (list->IsEmpty())
126     return 1;
127
128   TIterator* iter = list->MakeIterator();
129   TObject* obj;
130
131   // collections of measured and generated histograms
132   TList* collectionMeas = new TList;
133   TList* collectionGene = new TList;
134   
135   Int_t count = 0;
136   while ((obj = iter->Next())) {
137     
138     AliCorrectionMatrix* entry = dynamic_cast<AliCorrectionMatrix*> (obj);
139     if (entry == 0) 
140       continue;
141
142     collectionMeas->Add(entry->GetMeasuredHistogram());
143     collectionGene->Add(entry->GetGeneratedHistogram());
144
145     count++;
146   }
147   fhMeas->Merge(collectionMeas);
148   fhGene->Merge(collectionGene);
149
150   delete collectionMeas;
151   delete collectionGene;
152
153   return count+1;
154 }
155
156 //____________________________________________________________________
157 void AliCorrectionMatrix::Divide()
158 {
159   //
160   // divides generated by measured to get the correction
161   //
162
163 //   if (!fhCorr) {
164 //     fhCorr = (TH1*)fhGene->Clone("correction");
165 //     fhCorr->SetTitle(Form("%s correction",GetTitle()));
166 //     fhCorr->Reset();
167 //   }
168
169   if (!fhMeas || !fhGene || !fhCorr) {
170     AliDebug(AliLog::kError, "measured or generated histograms not available");
171     return;
172   }
173
174   fhCorr->Divide(fhGene, fhMeas, 1, 1, "B");
175
176   Int_t emptyBins = 0;
177   for (Int_t x=1; x<=fhCorr->GetNbinsX(); ++x)
178     for (Int_t y=1; y<=fhCorr->GetNbinsY(); ++y)
179       for (Int_t z=1; z<=fhCorr->GetNbinsZ(); ++z)
180         if (fhCorr->GetBinContent(x, y, z) == 0)
181           ++emptyBins;
182
183   if (emptyBins > 0)
184     printf("INFO: In %s we have %d empty bins (of %d) in the correction map\n", GetTitle(), emptyBins, fhCorr->GetNbinsX() * fhCorr->GetNbinsY() * fhCorr->GetNbinsZ());
185 }
186
187 //____________________________________________________________________
188 void AliCorrectionMatrix::Multiply()
189 {
190   //
191   // multiplies measured with correction to get the generated
192   //
193
194   if (!fhMeas || !fhGene || !fhCorr)
195     return;
196
197   fhGene->Multiply(fhMeas, fhCorr, 1, 1, "B");
198 }
199
200 //____________________________________________________________________
201 void AliCorrectionMatrix::Add(AliCorrectionMatrix* aMatrixToAdd, Float_t c) {
202   //
203   // adds the measured and generated of aMatrixToAdd to measured and generated of this
204   // 
205   // NB: the correction will naturally stay the same
206   
207   fhMeas->Add(aMatrixToAdd->GetMeasuredHistogram(), c);
208   fhGene->Add(aMatrixToAdd->GetGeneratedHistogram(), c);
209 }
210
211
212 //____________________________________________________________________
213 Bool_t AliCorrectionMatrix::LoadHistograms(const Char_t* dir)
214 {
215   //
216   // loads the histograms from a file
217   // if dir is empty a directory with the name of this object is taken (like in SaveHistogram)
218   //
219
220   if (!dir)
221     dir = GetName();
222
223   if (!gDirectory->cd(dir))
224     return kFALSE;
225
226   if (fhGene)
227   {
228     delete fhGene;
229     fhGene=0;
230   }
231
232   if (fhCorr)
233   {
234     delete fhCorr;
235     fhCorr=0;
236   }
237
238   if (fhMeas)
239   {
240     delete fhMeas;
241     fhMeas=0;
242   }
243
244   fhMeas  = dynamic_cast<TH1*> (gDirectory->Get("measured"));
245   if (!fhMeas)
246     Info("LoadHistograms", "No measured hist available");
247
248   fhGene  = dynamic_cast<TH1*> (gDirectory->Get("generated"));
249   if (!fhGene)
250     Info("LoadHistograms", "No generated hist available");
251
252   fhCorr  = dynamic_cast<TH1*> (gDirectory->Get("correction"));
253
254   Bool_t success = kTRUE;
255   if (!fhCorr)
256   {
257     Info("LoadHistograms", "No correction hist available");
258     success = kFALSE;
259   }
260
261   gDirectory->cd("..");
262
263   return success;
264 }
265
266 //____________________________________________________________________
267 void AliCorrectionMatrix::SaveHistograms()
268 {
269   //
270   // saves the histograms
271   //
272
273   gDirectory->mkdir(GetName());
274   gDirectory->cd(GetName());
275
276   if (fhMeas)
277     fhMeas ->Write();
278
279   if (fhGene)
280     fhGene ->Write();
281
282   if (fhCorr)
283     fhCorr->Write();
284
285   gDirectory->cd("..");
286 }
287
288 //____________________________________________________________________
289 void AliCorrectionMatrix::DrawHistograms(const Char_t* canvasName)
290 {
291   //
292   // draws all histograms on one TCanvas
293   // if canvasName is 0 the name of this object is taken
294   //
295
296   if (!canvasName)
297     canvasName = Form("%s_canvas", GetName());
298
299   TCanvas* canvas = new TCanvas(canvasName, GetTitle(), 1200, 400);
300   canvas->Divide(3, 1);
301
302   canvas->cd(1);
303   if (fhMeas)
304     fhMeas->Draw("COLZ");
305
306   canvas->cd(2);
307   if (fhGene)
308   {
309     // work around ROOT bug #22011
310     if (fhGene->GetEntries() == 0)
311       fhGene->SetEntries(1);
312     fhGene->Draw("COLZ");
313   }
314
315   canvas->cd(3);
316   if (fhCorr)
317     fhCorr->Draw("COLZ");
318 }
319
320 //____________________________________________________________________
321 void AliCorrectionMatrix::ReduceInformation()
322 {
323   // this function deletes the measured and generated histograms to reduce the amount of data
324   // in memory
325
326   if (fhMeas)
327   {
328     delete fhMeas;
329     fhMeas = 0;
330   }
331
332   if (fhGene)
333   {
334     delete fhGene;
335     fhGene = 0;
336   }
337 }
338
339 //____________________________________________________________________
340 void AliCorrectionMatrix::Reset(Option_t* option)
341 {
342   // resets the histograms
343
344   if (fhGene)
345     fhGene->Reset(option);
346
347   if (fhMeas)
348     fhMeas->Reset(option);
349
350   if (fhCorr)
351     fhCorr->Reset(option);
352 }
353
354 //____________________________________________________________________
355 void AliCorrectionMatrix::SetCorrectionToUnity()
356 {
357   // sets the correction matrix to unity
358
359   if (!fhCorr)
360     return;
361
362   for (Int_t x=1; x<=fhCorr->GetNbinsX(); ++x)
363     for (Int_t y=1; y<=fhCorr->GetNbinsY(); ++y)
364       for (Int_t z=1; z<=fhCorr->GetNbinsZ(); ++z)
365         fhCorr->SetBinContent(x, y, z, 1);
366 }