fixed the tainted variables
[u/mrichter/AliRoot.git] / PWG0 / AliCorrectionMatrix3D.h
1 #ifndef ALICORRECTIONMATRIX3D_H
2 #define ALICORRECTIONMATRIX3D_H
3
4 /* $Id$ */
5
6 // ------------------------------------------------------
7 //
8 // Class to handle 3d-corrections.
9 //
10 // ------------------------------------------------------
11
12 #include <AliCorrectionMatrix.h>
13 #include <AliCorrectionMatrix2D.h>
14
15 class TH3;
16 class TH2;
17 class TH1;
18
19
20 class AliCorrectionMatrix3D : public AliCorrectionMatrix
21 {
22 public:
23   AliCorrectionMatrix3D();
24   AliCorrectionMatrix3D(const AliCorrectionMatrix3D& c);
25   AliCorrectionMatrix3D(const Char_t* name, const Char_t* title,
26                      Int_t nBinX, Float_t Xmin, Float_t Xmax,
27                      Int_t nBinY, Float_t Ymin, Float_t Ymax,
28                      Int_t nBinZ, Float_t Zmin, Float_t Zmax);
29
30   AliCorrectionMatrix3D(const Char_t* name, const Char_t* title,
31          Int_t nBinX, Float_t Xmin, Float_t Xmax,
32          Int_t nBinY, Float_t Ymin, Float_t Ymax,
33          Int_t nBinZ, const Float_t* zbins);
34
35   AliCorrectionMatrix3D(const Char_t* name, const Char_t* title, TH3* hBinning);
36  
37   virtual ~AliCorrectionMatrix3D();
38
39   AliCorrectionMatrix3D& operator= (const AliCorrectionMatrix3D& c);
40
41   void CreateHists(Int_t nBinX, const Float_t* binLimitsX,
42                    Int_t nBinY, const Float_t* binLimitsY,
43                    Int_t nBinZ, const Float_t* binLimitsZ);
44
45   TH3* GetGeneratedHistogram();
46   TH3* GetMeasuredHistogram();
47   TH3* GetCorrectionHistogram();
48
49   AliCorrectionMatrix2D* Get2DCorrection(Option_t* opt, Float_t aMin, Float_t aMax);
50   TH2* Get2DCorrectionHistogram(Option_t* opt, Float_t aMin, Float_t aMax)     {return Get2DCorrection(opt,aMin,aMax)->GetCorrectionHistogram();}
51   TH1* Get1DCorrectionHistogram(Option_t* opt, Float_t aMins1=0, Float_t aMax1=0, Float_t aMins2=0, Float_t aMax2=0);
52
53   void FillMeas(Float_t ax, Float_t ay, Float_t az);
54   void FillGene(Float_t ax, Float_t ay, Float_t az);
55
56   Float_t GetCorrection(Float_t ax, Float_t ay, Float_t az) const;
57
58   void RemoveEdges(Float_t cut=2, Int_t nBinsXedge = 0, Int_t nBinsYedge = 0, Int_t nBinsZedge = 0);
59
60   virtual void SaveHistograms();
61
62   Int_t CheckEmptyBins(Float_t xmin, Float_t xmax, Float_t ymin, Float_t ymax, Float_t zmin, Float_t zmax, Bool_t quiet = kFALSE);
63   TH1* PlotBinErrors(Float_t xmin, Float_t xmax, Float_t ymin, Float_t ymax, Float_t zmin, Float_t zmax);
64
65
66 protected:
67   ClassDef(AliCorrectionMatrix3D,1)
68 };
69
70 #endif
71