New set of default cuts suitable for pp events.
[u/mrichter/AliRoot.git] / PWG0 / AliCorrection.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 #include "AliCorrectionMatrix2D.h"
16 #include "AliCorrectionMatrix3D.h"
17
18 #include "AliCorrection.h"
19
20 //____________________________________________________________________
21 ClassImp(AliCorrection)
22
23 //____________________________________________________________________
24 AliCorrection::AliCorrection() : TNamed(),
25   fEventCorr(0),
26   fTrackCorr(0)
27 {
28   // default constructor
29 }
30
31 //____________________________________________________________________
32 AliCorrection::AliCorrection(const Char_t* name, const Char_t* title) : TNamed(name, title),
33   fEventCorr(0),
34   fTrackCorr(0)
35 {
36   // constructor initializing tnamed
37
38   Float_t binLimitsPt[] = {0.0, 0.05, 0.1, 0.125, 0.15, 0.175, 0.2, 0.225, 0.25, 0.275, 0.3, 0.325, 0.35, 0.375, 0.4, 0.425, 0.45, 0.475, 0.5, 0.6, 0.7, 0.8, 0.9, 1.0, 1.5, 2.0, 5.0, 10.0, 100.0};
39   Float_t binLimitsN[]   = {-0.5, 0.5, 1.5, 2.5, 3.5, 4.5, 5.5, 6.5, 7.5, 8.5, 9.5, 10.5, 12.5, 14.5, 16.5, 18.5, 20.5, 25.5, 30.5, 40.5, 50.5, 100.5, 300.5};
40   //Float_t binLimitsVtx[] = {-20,-15,-10,-6,-3,0,3,6,10,15,20};
41   Float_t binLimitsVtx[] = {-20,-19,-18,-17,-16,-15,-14,-13,-12,-11,-10,-9,-8,-7,-6,-5,-4,-3,-2,-1,0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20};
42
43   fEventCorr = new AliCorrectionMatrix2D("EventCorrection", Form("%s EventCorrection", title), 40, binLimitsVtx, 22, binLimitsN);
44   fTrackCorr = new AliCorrectionMatrix3D("TrackCorrection", Form("%s TrackCorrection", title), 40, -20, 20, 20, -2, 2, 28, binLimitsPt);
45
46   fEventCorr->SetAxisTitles("vtx z [cm]", "Ntracks");
47   fTrackCorr->SetAxisTitles("vtx z [cm]", "#eta", "p_{T} [GeV/c]");
48 }
49
50 //____________________________________________________________________
51 AliCorrection::AliCorrection(const AliCorrection& c) : TNamed(c),
52   fEventCorr(0),
53   fTrackCorr(0)
54 {
55   // copy constructor
56   ((AliCorrection &)c).Copy(*this);
57 }
58
59 //____________________________________________________________________
60 AliCorrection::~AliCorrection()
61 {
62   //
63   // destructor
64   //
65
66   if (fEventCorr)
67   {
68     delete fEventCorr;
69     fEventCorr = 0;
70   }
71
72   if (fTrackCorr)
73   {
74     delete fTrackCorr;
75     fTrackCorr = 0;
76   }
77 }
78
79 //____________________________________________________________________
80 AliCorrection &AliCorrection::operator=(const AliCorrection &c)
81 {
82   // assigment operator
83
84   if (this != &c)
85     ((AliCorrection &) c).Copy(*this);
86
87   return *this;
88 }
89
90 //____________________________________________________________________
91 void AliCorrection::Copy(TObject& c) const
92 {
93   // copy function
94
95   AliCorrection& target = (AliCorrection &) c;
96
97   if (fEventCorr)
98     target.fEventCorr = dynamic_cast<AliCorrectionMatrix2D*> (fEventCorr->Clone());
99
100   if (fTrackCorr)
101     target.fTrackCorr = dynamic_cast<AliCorrectionMatrix3D*> (fTrackCorr->Clone());
102 }
103
104 //____________________________________________________________________
105 Long64_t AliCorrection::Merge(TCollection* list)
106 {
107   // Merge a list of AliCorrection objects with this (needed for
108   // PROOF). 
109   // Returns the number of merged objects (including this).
110
111   if (!list)
112     return 0;
113   
114   if (list->IsEmpty())
115     return 1;
116
117   TIterator* iter = list->MakeIterator();
118   TObject* obj;
119
120   // collections of measured and generated histograms
121   TList* collectionEvent = new TList;
122   TList* collectionTrack = new TList;
123   
124   Int_t count = 0;
125   while ((obj = iter->Next())) {
126     
127     AliCorrection* entry = dynamic_cast<AliCorrection*> (obj);
128     if (entry == 0) 
129       continue;
130
131     collectionEvent->Add(entry->fEventCorr);
132     collectionTrack->Add(entry->fTrackCorr);
133
134     count++;
135   }
136   fEventCorr->Merge(collectionEvent);
137   fTrackCorr->Merge(collectionTrack);
138
139   delete collectionEvent;
140   delete collectionTrack;
141
142   return count+1;
143 }
144
145 //____________________________________________________________________
146 void AliCorrection::Divide()
147 {
148   //
149   // divide the histograms to get the correction
150   //
151   
152   if (!fEventCorr || !fTrackCorr)
153     return;
154     
155   fEventCorr->Divide();
156   fTrackCorr->Divide();
157
158   Int_t emptyBins = fTrackCorr->CheckEmptyBins(-9.99, 9.99, -0.79, 0.79, 0.3, 9.9);
159   printf("INFO: In the central region the track correction of %s has %d empty bins\n", GetName(), emptyBins);
160 }
161
162 //____________________________________________________________________
163 Bool_t AliCorrection::LoadHistograms(const Char_t* dir)
164 {
165   //
166   // loads the histograms from a file
167   // if dir is empty a directory with the name of this object is taken (like in SaveHistogram)
168   //
169
170   if (!fEventCorr || !fTrackCorr)
171     return kFALSE;
172
173   if (!dir)
174     dir = GetName();
175
176   if (!gDirectory->cd(dir))
177     return kFALSE;
178
179   Bool_t success = fEventCorr->LoadHistograms();
180   success &= fTrackCorr->LoadHistograms();
181
182   gDirectory->cd("..");
183
184   return success;
185 }
186
187 //____________________________________________________________________
188 void AliCorrection::SaveHistograms()
189 {
190   //
191   // saves the histograms in a directory with the name of this object (GetName)
192   //
193   
194   gDirectory->mkdir(GetName());
195   gDirectory->cd(GetName());
196
197   if (fEventCorr)
198     fEventCorr->SaveHistograms();
199
200   if (fTrackCorr)
201     fTrackCorr->SaveHistograms();
202     
203   gDirectory->cd("..");
204 }
205
206 //____________________________________________________________________
207 void AliCorrection::ReduceInformation()
208 {
209   // this function deletes the measured and generated histograms to reduce the amount of data
210   // in memory
211
212   if (!fEventCorr || !fTrackCorr)
213     return;
214
215   fEventCorr->ReduceInformation();
216   fTrackCorr->ReduceInformation();
217 }
218
219 //____________________________________________________________________
220 void AliCorrection::Reset(Option_t* option)
221 {
222   // resets the histograms
223
224   if (fEventCorr)
225     fEventCorr->Reset(option);
226
227   if (fTrackCorr)
228     fTrackCorr->Reset(option);
229 }
230
231 //____________________________________________________________________
232 void AliCorrection::DrawHistograms(const Char_t* name)
233 {
234   // draws the corrections
235
236   if (!name)
237     name = GetName();
238
239   if (fEventCorr)
240     fEventCorr->DrawHistograms(Form("%s event", name));
241
242   if (fTrackCorr)
243     fTrackCorr->DrawHistograms(Form("%s track", name));
244 }
245
246 //____________________________________________________________________
247 void AliCorrection::SetCorrectionToUnity()
248 {
249   // set the corrections to unity
250
251   if (fEventCorr)
252     fEventCorr->SetCorrectionToUnity();
253
254   if (fTrackCorr)
255     fTrackCorr->SetCorrectionToUnity();
256 }
257
258 //____________________________________________________________________
259 void AliCorrection::Multiply()
260 {
261   // call Multiply
262
263   if (fEventCorr)
264   {
265     fEventCorr->Multiply();
266     // now we manually copy the overflow bin of the y axis (multiplicity) over. This is important to get the event count correct
267     TH2F* hist = fEventCorr->GetMeasuredHistogram();
268     for (Int_t x = 1; x <= hist->GetNbinsX(); ++x)
269       fEventCorr->GetGeneratedHistogram()->SetBinContent(x, hist->GetNbinsY() + 1, hist->GetBinContent(x, hist->GetNbinsY() + 1));
270   }
271
272   if (fTrackCorr)
273     fTrackCorr->Multiply();
274 }