]> git.uio.no Git - u/mrichter/AliRoot.git/blame - CORRFW/AliCFContainer.h
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Update the mult corr histograms
[u/mrichter/AliRoot.git] / CORRFW / AliCFContainer.h
CommitLineData
563113d0 1#ifndef ALICFCONTAINER_H
2#define ALICFCONTAINER_H
3
4/* $Id$ */
5
6//--------------------------------------------------------------------//
7// //
8// AliCFContainer Class //
9// Class to handle input data for correction Framework //
10// //
11//--------------------------------------------------------------------//
12
13#include "AliCFFrame.h"
fb494025 14#include "AliCFGridSparse.h"
563113d0 15
1e9dad92 16class TH1D;
17class TH2D;
18class TH3D;
07b0477d 19class TCollection;
563113d0 20
21class AliCFContainer : public AliCFFrame
22{
23 public:
24 AliCFContainer();
25 AliCFContainer(const Char_t* name,const Char_t* title);
fb494025 26 AliCFContainer(const Char_t* name, const Char_t* title,const Int_t nSelStep, const Int_t nVarIn, const Int_t* nBinIn);
563113d0 27 AliCFContainer(const AliCFContainer& c);
563113d0 28 AliCFContainer& operator=(const AliCFContainer& corr);
fb494025 29 virtual void Copy(TObject& c) const;
30
31 virtual ~AliCFContainer();
32
33 // AliCFFrame functions
34 virtual Int_t GetNVar() const {return fGrid[0]->GetNVar();}
35 virtual void PrintBinLimits() const {fGrid[0]->PrintBinLimits();}
36 virtual void PrintNBins() const {fGrid[0]->PrintNBins();}
aa29aca9 37 virtual void SetBinLimits(Int_t ivar, Double_t min, Double_t max) ; // for uniform bin width only
38 virtual void SetBinLimits(Int_t ivar, const Double_t * array) ; // for variable or uniform bin width
1db34d6e 39 virtual void SetBinLabel (Int_t ivar,Int_t ibin, const Char_t* label);// set a label to bin ibin on axis ivar
40 virtual void SetBinContent(Int_t* bin, Int_t step, Double_t value);
41 virtual void SetBinError (Int_t* bin, Int_t step, Double_t value);
fb494025 42 virtual void GetBinLimits(Int_t ivar, Double_t * array) const {return fGrid[0]->GetBinLimits(ivar,array);}
43 virtual Double_t * GetBinLimits(Int_t ivar) const {return fGrid[0]->GetBinLimits(ivar);}
44 virtual Long_t GetNBinsTotal() const {return fGrid[0]->GetNBinsTotal()*fNStep;}
45 virtual Int_t GetNBins(Int_t ivar) const {return fGrid[0]->GetNBins(ivar);}
46 virtual Int_t * GetNBins() const {return fGrid[0]->GetNBins();}
47 virtual Float_t GetBinCenter(Int_t ivar,Int_t ibin) const {return fGrid[0]->GetBinCenter(ivar,ibin);}
48 virtual Float_t GetBinSize (Int_t ivar,Int_t ibin) const {return fGrid[0]->GetBinSize (ivar,ibin);}
4b7819c4 49 virtual Float_t GetBinContent(const Int_t* coordinates, Int_t step) const {return fGrid[step]->GetGrid()->GetBinContent(coordinates);}
50 virtual Float_t GetBinError (const Int_t* coordinates, Int_t step) const {return fGrid[step]->GetGrid()->GetBinError (coordinates);}
1db34d6e 51 virtual const Char_t* GetBinLabel (Int_t ivar,Int_t ibin) const {return GetAxis(ivar,0)->GetBinLabel(ibin);}
fb494025 52
71b2614f 53 virtual void Print(const Option_t*) const ;
54
fb494025 55 //virtual void GetBinCenters(const Int_t *ibin, const Double_t *binCenter) const {return fGrid[0]->GetBinCenters(ibin,binCenter);}
56 //virtual void GetBinSizes(const Int_t *ibin, const Double_t *binSizes) const {return fGrid[0]->GetBinSizes(ibin,binSizes);}
57
58 //probably not needed anymore
59 //virtual Int_t GetBinIndex(const Int_t *ibin) const {return fGrid[0]->GetBinIndex(ibin);}
60 //virtual void GetBinIndex(Int_t iel, const Int_t *ibin) const {return fGrid[0]->GetBinIndex(iel,ibin);}
61 //virtual Int_t GetBinIndex(Int_t ivar, Int_t ind) const {return fGrid[0]->GetBinIndex(ivar,ind);}
62
63 virtual TAxis * GetAxis(Int_t ivar, Int_t istep) const {return fGrid[istep]->GetAxis(ivar);}
64 virtual void SetVarTitle (Int_t ivar, const Char_t* title) ;
65 virtual void SetStepTitle(Int_t istep, const Char_t* title) ;
66 virtual const Char_t* GetVarTitle (Int_t ivar) const {return GetAxis(ivar,0)->GetTitle();}
67 virtual const Char_t* GetStepTitle(Int_t istep) const {return fGrid[istep]->GetTitle();}
68 virtual Int_t GetStep(const Char_t* title) const ; // returns the step corresponding to the given title
69 virtual Int_t GetVar (const Char_t* title) const ; // returns the variable corresponding to the given title
9105291d 70
fb494025 71 virtual Int_t GetNStep() const {return fNStep;};
9105291d 72 virtual void SetNStep(Int_t nStep) {fNStep=nStep;}
fb494025 73 virtual void Fill(const Double_t *var, Int_t istep, Double_t weight=1.) ;
9105291d 74
fb494025 75 virtual Float_t GetOverFlows (Int_t var,Int_t istep,Bool_t excl=kFALSE) const;
76 virtual Float_t GetUnderFlows(Int_t var,Int_t istep,Bool_t excl=kFALSE) const ;
77 virtual Float_t GetEntries (Int_t istep) const ;
78 virtual Long_t GetEmptyBins(Int_t istep) const {return fGrid[istep]->GetEmptyBins();}
79 //virtual Int_t GetEmptyBins(Int_t istep, Double_t *varMin,Double_t *varMax) const ;
80 virtual Double_t GetIntegral (Int_t istep) const ;
81 //virtual Double_t GetIntegral (Int_t istep, Double_t *varMin,Double_t *varMax) const ;
563113d0 82
563113d0 83
fb494025 84 //basic operations
85 virtual void Add(const AliCFContainer* aContainerToAdd, Double_t c=1.);
563113d0 86 virtual Long64_t Merge(TCollection* list);
87
fb494025 88 virtual TH1D* ShowProjection( Int_t ivar, Int_t istep) const {return Project(ivar ,istep);}
89 virtual TH2D* ShowProjection( Int_t ivar1, Int_t ivar2, Int_t istep) const {return Project(ivar1,ivar2 ,istep);}
90 virtual TH3D* ShowProjection( Int_t ivar1, Int_t ivar2,Int_t ivar3, Int_t istep) const {return Project(ivar1,ivar2,ivar3,istep);}
91 virtual TH1D* Project( Int_t ivar, Int_t istep) const;
92 virtual TH2D* Project( Int_t ivar1, Int_t ivar2, Int_t istep) const;
93 virtual TH3D* Project( Int_t ivar1, Int_t ivar2,Int_t ivar3, Int_t istep) const;
98a5f772 94
95 virtual TH1D* ShowSlice(Int_t ivar, const Double_t *varMin, const Double_t *varMax, Int_t istep, Bool_t useBins=0) const ;
96 virtual TH2D* ShowSlice(Int_t ivar1, Int_t ivar2, const Double_t *varMin, const Double_t *varMax, Int_t istep, Bool_t useBins=0) const ;
97 virtual TH3D* ShowSlice(Int_t ivar1, Int_t ivar2, Int_t ivar3, const Double_t *varMin, const Double_t *varMax, Int_t istep, Bool_t useBins=0) const ;
98 virtual AliCFContainer* MakeSlice(Int_t nVars, const Int_t* vars, const Double_t* varMin, const Double_t* varMax, Bool_t useBins=0) const ;
99 virtual AliCFContainer* MakeSlice(Int_t nVars, const Int_t* vars, const Double_t* varMin, const Double_t* varMax, Int_t nStep, const Int_t* steps, Bool_t useBins=0) const ;
7036630f 100 virtual void Smooth(Int_t istep) {GetGrid(istep)->Smooth();}
98a5f772 101
c8df672e 102 virtual void SetRangeUser(Int_t ivar, Double_t varMin, Double_t varMax, Int_t istep) ;
9105291d 103 virtual void SetRangeUser(Double_t* varMin, Double_t* varMax, Int_t istep) ;
4b7819c4 104 virtual void SetGrid(Int_t step, AliCFGridSparse* grid) {if (fGrid[step]) delete fGrid[step]; fGrid[step]=grid;}
fb494025 105 virtual AliCFGridSparse * GetGrid(Int_t istep) const {return fGrid[istep];};
1db34d6e 106
107 virtual void Scale(Double_t factor) const;
563113d0 108
109 private:
110 Int_t fNStep; //number of selection steps
fb494025 111 AliCFGridSparse **fGrid;//[fNStep]
563113d0 112
fb494025 113 ClassDef(AliCFContainer,5);
563113d0 114};
fb494025 115
116inline void AliCFContainer::SetBinLimits(Int_t ivar, const Double_t* array) {
117 for (Int_t iStep=0; iStep<GetNStep(); iStep++) {
118 fGrid[iStep]->SetBinLimits(ivar,array);
119 }
120}
121
aa29aca9 122inline void AliCFContainer::SetBinLimits(Int_t ivar, Double_t min, Double_t max) {
123 for (Int_t iStep=0; iStep<GetNStep(); iStep++) {
124 fGrid[iStep]->SetBinLimits(ivar,min,max);
125 }
126}
127
fb494025 128inline void AliCFContainer::SetVarTitle(Int_t ivar, const Char_t* title) {
129 for (Int_t iStep=0; iStep<fNStep; iStep++) {
130 GetAxis(ivar,iStep)->SetTitle(title);
131 }
132}
133
134inline void AliCFContainer::SetStepTitle(Int_t istep, const Char_t* title) {
135 fGrid[istep]->SetTitle(title);
136}
137
138inline Int_t AliCFContainer::GetStep(const Char_t* title) const {
139 TString str(title);
140 for (Int_t iStep=0; iStep<fNStep; iStep++) {
141 if (!str.CompareTo(GetStepTitle(iStep))) return iStep;
142 }
143 AliError("Step not found");
144 return -1;
145}
146
147inline Int_t AliCFContainer::GetVar(const Char_t* title) const {
148 return fGrid[0]->GetVar(title);
149}
1db34d6e 150
151inline void AliCFContainer::SetBinLabel(Int_t iVar, Int_t iBin, const Char_t* label) {
152 for (Int_t iStep=0; iStep<GetNStep(); iStep++) GetAxis(iVar,iStep)->SetBinLabel(iBin,label);
153}
154
155inline void AliCFContainer::Scale(Double_t factor) const {
156 Double_t fact[2] = {factor,0} ;
157 for (Int_t iStep=0; iStep<fNStep; iStep++) fGrid[iStep]->Scale(fact);
158}
159
160inline void AliCFContainer::SetBinContent(Int_t* bin, Int_t step, Double_t value) {
4b7819c4 161 // sets the content 'value' to the current container, at step 'step'
162 // 'bin' is the array of the bin coordinates
1db34d6e 163 GetGrid(step)->GetGrid()->SetBinContent(bin,value);
164}
165
166inline void AliCFContainer::SetBinError(Int_t* bin, Int_t step, Double_t value) {
4b7819c4 167 // sets the error 'value' to the current container, at step 'step'
168 // 'bin' is the array of the bin coordinates
1db34d6e 169 GetGrid(step)->GetGrid()->SetBinError(bin,value);
170}
171
563113d0 172#endif
173