CORRFW/AliCFContainer.h

   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"
  14 #include "AliCFGridSparse.h"
  15
  16 class TH1D;
  17 class TH2D;
  18 class TH3D;
  19 class TCollection;
  20
  21 class AliCFContainer : public AliCFFrame
  22 {
  23  public:
  24   AliCFContainer();
  25   AliCFContainer(const Char_t* name,const Char_t* title);
  26   AliCFContainer(const Char_t* name, const Char_t* title,const Int_t nSelStep, const Int_t nVarIn, const Int_t* nBinIn);
  27   AliCFContainer(const AliCFContainer& c);
  28   AliCFContainer& operator=(const AliCFContainer& corr);
  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();}
  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
  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);
  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);}
  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);}
  51   virtual const Char_t* GetBinLabel (Int_t ivar,Int_t ibin)          const {return GetAxis(ivar,0)->GetBinLabel(ibin);}
  52
  53   virtual void       Print(const Option_t*) const ;
  54
  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
  70
  71   virtual Int_t GetNStep() const {return fNStep;};
  72   virtual void  SetNStep(Int_t nStep) {fNStep=nStep;}
  73   virtual void  Fill(const Double_t *var, Int_t istep, Double_t weight=1.) ;
  74
  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 ;
  82
  83
  84   //basic operations
  85   virtual void     Add(const AliCFContainer* aContainerToAdd, Double_t c=1.);
  86   virtual Long64_t Merge(TCollection* list);
  87
  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;
  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 ;
 100
 101   virtual void  SetRangeUser(Int_t ivar, Double_t varMin, Double_t varMax, Int_t istep) ;
 102   virtual void  SetRangeUser(Double_t* varMin, Double_t* varMax, Int_t istep) ;
 103   virtual void  SetGrid(Int_t step, AliCFGridSparse* grid) {if (fGrid[step]) delete fGrid[step]; fGrid[step]=grid;}
 104   virtual AliCFGridSparse * GetGrid(Int_t istep) const {return fGrid[istep];};
 105
 106   virtual void  Scale(Double_t factor) const;
 107
 108  private:
 109   Int_t    fNStep; //number of selection steps
 110   AliCFGridSparse **fGrid;//[fNStep]
 111
 112   ClassDef(AliCFContainer,5);
 113 };
 114
 115 inline void AliCFContainer::SetBinLimits(Int_t ivar, const Double_t* array) {
 116   for (Int_t iStep=0; iStep<GetNStep(); iStep++) {
 117     fGrid[iStep]->SetBinLimits(ivar,array);
 118   }
 119 }
 120
 121 inline void AliCFContainer::SetBinLimits(Int_t ivar, Double_t min, Double_t max) {
 122   for (Int_t iStep=0; iStep<GetNStep(); iStep++) {
 123     fGrid[iStep]->SetBinLimits(ivar,min,max);
 124   }
 125 }
 126
 127 inline void AliCFContainer::SetVarTitle(Int_t ivar, const Char_t* title) {
 128   for (Int_t iStep=0; iStep<fNStep; iStep++) {
 129     GetAxis(ivar,iStep)->SetTitle(title);
 130   }
 131 }
 132
 133 inline void AliCFContainer::SetStepTitle(Int_t istep, const Char_t* title) {
 134   fGrid[istep]->SetTitle(title);
 135 }
 136
 137 inline Int_t AliCFContainer::GetStep(const Char_t* title) const {
 138   TString str(title);
 139   for (Int_t iStep=0; iStep<fNStep; iStep++) {
 140     if (!str.CompareTo(GetStepTitle(iStep))) return iStep;
 141   }
 142   AliError("Step not found");
 143   return -1;
 144 }
 145
 146 inline Int_t AliCFContainer::GetVar(const Char_t* title) const {
 147   return fGrid[0]->GetVar(title);
 148 }
 149
 150 inline void AliCFContainer::SetBinLabel(Int_t iVar, Int_t iBin, const Char_t* label) {
 151   for (Int_t iStep=0; iStep<GetNStep(); iStep++) GetAxis(iVar,iStep)->SetBinLabel(iBin,label);
 152 }
 153
 154 inline void  AliCFContainer::Scale(Double_t factor) const {
 155   Double_t fact[2] = {factor,0} ;
 156   for (Int_t iStep=0; iStep<fNStep; iStep++) fGrid[iStep]->Scale(fact);
 157 }
 158
 159 inline void AliCFContainer::SetBinContent(Int_t* bin, Int_t step, Double_t value) {
 160   // sets the content 'value' to the current container, at step 'step'
 161   // 'bin' is the array of the bin coordinates
 162   GetGrid(step)->GetGrid()->SetBinContent(bin,value);
 163 }
 164
 165 inline void AliCFContainer::SetBinError(Int_t* bin, Int_t step, Double_t value) {
 166   // sets the error 'value' to the current container, at step 'step'
 167   // 'bin' is the array of the bin coordinates
 168   GetGrid(step)->GetGrid()->SetBinError(bin,value);
 169 }
 170
 171 #endif
 172