Made these classes derive from AliCFGridSparse.
[u/mrichter/AliRoot.git] / CORRFW / AliCFGridSparse.h
1 #ifndef ALICFGRIDSPARSE_H
2 #define ALICFGRIDSPARSE_H
3 //--------------------------------------------------------------------//
4 //                                                                    //
5 // AliCFGridSparse.cxx Class                                          //
6 // Class to handle N-dim maps for the correction Framework            // 
7 // uses a THnSparse to store the grid                                 //
8 // Author:S.Arcelli, silvia.arcelli@cern.ch
9 //--------------------------------------------------------------------//
10
11 #include "AliCFVGrid.h"
12 #include "THnSparse.h"
13 #include "AliLog.h"
14 class TH1D;
15 class TH2D;
16 class TH3D;
17
18 class AliCFGridSparse : public AliCFVGrid
19 {
20  public:
21   AliCFGridSparse();
22   AliCFGridSparse(const Char_t* name,const Char_t* title);
23   AliCFGridSparse(const Char_t* name, const Char_t* title, const Int_t nVarIn, const Int_t* nBinIn, const Double_t  *binLimitsIn=0);
24   AliCFGridSparse(const AliCFGridSparse & c);
25   
26   virtual ~AliCFGridSparse();
27   AliCFGridSparse& operator=(const AliCFGridSparse& corr) ;
28   virtual void  SetBinLimits(Int_t ivar, Double_t * array);
29   
30   virtual void  Fill(Double_t *var, Double_t weight=1.);
31
32   virtual Float_t GetOverFlows(Int_t var) const;
33   virtual Float_t GetUnderFlows(Int_t var)const;
34   virtual Float_t GetEntries()const;
35
36   virtual Float_t GetElement(Int_t iel)const; 
37   virtual Float_t GetElement(Int_t *bin)const; 
38   virtual Float_t GetElement(Double_t *var)const; 
39
40   virtual Float_t GetElementError(Int_t iel)const; 
41   virtual Float_t GetElementError(Int_t *bin)const; 
42   virtual Float_t GetElementError(Double_t *var)const; 
43
44   virtual void SetElement(Int_t iel, Float_t val); 
45   virtual void SetElement(Int_t *bin, Float_t val); 
46   virtual void SetElement(Double_t *var, Float_t val); 
47   virtual void SetElementError(Int_t iel, Float_t val); 
48   virtual void SetElementError(Int_t *bin, Float_t val) ; 
49   virtual void SetElementError(Double_t *var, Float_t val); 
50
51   virtual TH1D* Project( Int_t ivar) const;
52   virtual TH2D* Project( Int_t ivar1, Int_t ivar2) const;
53   virtual TH3D* Project( Int_t ivar1, Int_t ivar2,Int_t ivar3) const;
54   virtual TH1D* Slice( Int_t, Double_t *, Double_t *) const {AliInfo("Not yet implemented!"); return 0x0;};
55
56
57   //basic operations
58
59   virtual void SumW2();
60   virtual void Add(AliCFVGrid* aGrid, Double_t c=1.);
61   virtual void Add(AliCFVGrid* aGrid1 ,AliCFVGrid* aGrid2, Double_t c1=1.,Double_t c2=1.);
62   virtual void Multiply(AliCFVGrid* aGrid, Double_t c=1.);
63   virtual void Multiply(AliCFVGrid* aGrid1,AliCFVGrid* aGrid2, Double_t c1=1.,Double_t c2=1.);
64   virtual void Divide(AliCFVGrid* aGrid, Double_t c=1.);
65   virtual void Divide(AliCFVGrid* aGrid1, AliCFVGrid* aGrid2, Double_t c1=1., Double_t c2=1.,Option_t *option=0);
66
67   virtual void Rebin(const Int_t* group);
68
69   THnSparse  *GetGrid() const {return fData;};//  Getter for the data Container: a THnSparse
70
71   virtual void Copy(TObject& c) const;
72
73   
74  protected:
75
76   THnSparse  *fData;//  The data Container: a THnSparse  
77   ClassDef(AliCFGridSparse,2);
78 };
79     
80 #endif
81