[u/mrichter/AliRoot.git] / CORRFW / AliCFContainer.h

#ifndef ALICFCONTAINER_H
#define ALICFCONTAINER_H

/* $Id$ */

//--------------------------------------------------------------------//
//                                                                    //
// AliCFContainer Class                                               //
// Class to handle input data for correction Framework                // 
//                                                                    //
//--------------------------------------------------------------------//

#include "AliCFFrame.h"
#include "AliCFGridSparse.h"

class TH1D;
class TH2D;
class TH3D;
class TCollection;

class AliCFContainer : public AliCFFrame
{
 public:
  AliCFContainer();
  AliCFContainer(const Char_t* name,const Char_t* title);
  AliCFContainer(const Char_t* name, const Char_t* title,const Int_t nSelStep, const Int_t nVarIn, const Int_t* nBinIn);
  AliCFContainer(const AliCFContainer& c);
  AliCFContainer& operator=(const AliCFContainer& corr);
  virtual void Copy(TObject& c) const;

  virtual ~AliCFContainer();

  // AliCFFrame functions
  virtual Int_t      GetNVar()                                       const {return fGrid[0]->GetNVar();}
  virtual void       PrintBinLimits()                                const {fGrid[0]->PrintBinLimits();}
  virtual void       PrintNBins()                                    const {fGrid[0]->PrintNBins();}
  virtual void       SetBinLimits(Int_t ivar, const Double_t * array) ;
  virtual void       GetBinLimits(Int_t ivar, Double_t * array)      const {return fGrid[0]->GetBinLimits(ivar,array);}
  virtual Double_t * GetBinLimits(Int_t ivar)                        const {return fGrid[0]->GetBinLimits(ivar);}
  virtual Long_t     GetNBinsTotal()                                 const {return fGrid[0]->GetNBinsTotal()*fNStep;}
  virtual Int_t      GetNBins(Int_t ivar)                            const {return fGrid[0]->GetNBins(ivar);}
  virtual Int_t    * GetNBins()                                      const {return fGrid[0]->GetNBins();}
  virtual Float_t    GetBinCenter(Int_t ivar,Int_t ibin)             const {return fGrid[0]->GetBinCenter(ivar,ibin);}
  virtual Float_t    GetBinSize  (Int_t ivar,Int_t ibin)             const {return fGrid[0]->GetBinSize  (ivar,ibin);}

  //virtual void       GetBinCenters(const Int_t *ibin, const Double_t *binCenter) const {return fGrid[0]->GetBinCenters(ibin,binCenter);}
  //virtual void       GetBinSizes(const Int_t *ibin, const Double_t *binSizes)    const {return fGrid[0]->GetBinSizes(ibin,binSizes);}

  //probably not needed anymore
  //virtual Int_t      GetBinIndex(const Int_t *ibin)                  const {return fGrid[0]->GetBinIndex(ibin);}
  //virtual void       GetBinIndex(Int_t iel, const Int_t *ibin)       const {return fGrid[0]->GetBinIndex(iel,ibin);}
  //virtual Int_t      GetBinIndex(Int_t ivar, Int_t ind)              const {return fGrid[0]->GetBinIndex(ivar,ind);}

  virtual TAxis       * GetAxis(Int_t ivar, Int_t istep) const {return fGrid[istep]->GetAxis(ivar);}
  virtual void          SetVarTitle (Int_t ivar,  const Char_t* title) ;
  virtual void          SetStepTitle(Int_t istep, const Char_t* title) ;
  virtual const Char_t* GetVarTitle (Int_t ivar)  const {return GetAxis(ivar,0)->GetTitle();}
  virtual const Char_t* GetStepTitle(Int_t istep) const {return fGrid[istep]->GetTitle();}
  virtual Int_t         GetStep(const Char_t* title) const ; // returns the step     corresponding to the given title
  virtual Int_t         GetVar (const Char_t* title) const ; // returns the variable corresponding to the given title

  virtual Int_t GetNStep() const {return fNStep;};
  virtual void  SetNStep(Int_t nStep) {fNStep=nStep;}
  virtual void  Fill(const Double_t *var, Int_t istep, Double_t weight=1.) ;

  virtual Float_t  GetOverFlows (Int_t var,Int_t istep,Bool_t excl=kFALSE) const;
  virtual Float_t  GetUnderFlows(Int_t var,Int_t istep,Bool_t excl=kFALSE) const ;
  virtual Float_t  GetEntries  (Int_t istep) const ;
  virtual Long_t   GetEmptyBins(Int_t istep) const {return fGrid[istep]->GetEmptyBins();}
  //virtual Int_t    GetEmptyBins(Int_t istep, Double_t *varMin,Double_t *varMax) const ;
  virtual Double_t GetIntegral (Int_t istep) const ;
  //virtual Double_t GetIntegral (Int_t istep, Double_t *varMin,Double_t *varMax) const ;


  //basic operations
  virtual void     Add(const AliCFContainer* aContainerToAdd, Double_t c=1.);
  virtual Long64_t Merge(TCollection* list);

  virtual TH1D* ShowProjection( Int_t ivar, Int_t istep)                           const {return Project(ivar             ,istep);}
  virtual TH2D* ShowProjection( Int_t ivar1, Int_t ivar2, Int_t istep)             const {return Project(ivar1,ivar2      ,istep);}
  virtual TH3D* ShowProjection( Int_t ivar1, Int_t ivar2,Int_t ivar3, Int_t istep) const {return Project(ivar1,ivar2,ivar3,istep);}
  virtual TH1D* Project( Int_t ivar, Int_t istep) const;
  virtual TH2D* Project( Int_t ivar1, Int_t ivar2, Int_t istep) const;
  virtual TH3D* Project( Int_t ivar1, Int_t ivar2,Int_t ivar3, Int_t istep) const;
  virtual TH1D* ShowSlice( Int_t ivar, Double_t *varMin, Double_t *varMax, Int_t istep) const;
  virtual TH2D* ShowSlice( Int_t ivar1, Int_t ivar2, Double_t *varMin, Double_t *varMax, Int_t istep) const;
  virtual TH3D* ShowSlice( Int_t ivar1, Int_t ivar2, Int_t ivar3, Double_t *varMin, Double_t *varMax, Int_t istep) const;
  virtual AliCFContainer*  MakeSlice (Int_t nVars, const Int_t* vars, const Double_t* varMin, const Double_t* varMax) const ;
  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) const ;
  virtual void  SetRangeUser(Int_t ivar, Double_t varMin, Double_t varMax, Int_t istep) ;
  virtual void  SetRangeUser(Double_t* varMin, Double_t* varMax, Int_t istep) ;
  virtual void  SetGrid(Int_t step, AliCFGridSparse* grid) {fGrid[step]=grid;}
  virtual AliCFGridSparse * GetGrid(Int_t istep) const {return fGrid[istep];};
  
 private:
  Int_t    fNStep; //number of selection steps
  AliCFGridSparse **fGrid;//[fNStep]
  
  ClassDef(AliCFContainer,5);
};

inline void AliCFContainer::SetBinLimits(Int_t ivar, const Double_t* array) {
  for (Int_t iStep=0; iStep<GetNStep(); iStep++) {
    fGrid[iStep]->SetBinLimits(ivar,array);
  }
}

inline void AliCFContainer::SetVarTitle(Int_t ivar, const Char_t* title) {
  for (Int_t iStep=0; iStep<fNStep; iStep++) {
    GetAxis(ivar,iStep)->SetTitle(title);
  }
}

inline void AliCFContainer::SetStepTitle(Int_t istep, const Char_t* title) {
  fGrid[istep]->SetTitle(title);
}

inline Int_t AliCFContainer::GetStep(const Char_t* title) const {
  TString str(title);
  for (Int_t iStep=0; iStep<fNStep; iStep++) {
    if (!str.CompareTo(GetStepTitle(iStep))) return iStep;
  }
  AliError("Step not found");
  return -1;
}

inline Int_t AliCFContainer::GetVar(const Char_t* title) const {
  return fGrid[0]->GetVar(title);
}
#endif
Commit	Line	Data
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	16	class TH1D;
	17	class TH2D;
	18	class TH3D;
07b0477d	19	class TCollection;
563113d0	20
	21	class 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();}
	37	virtual void SetBinLimits(Int_t ivar, const Double_t * array) ;
	38	virtual void GetBinLimits(Int_t ivar, Double_t * array) const {return fGrid[0]->GetBinLimits(ivar,array);}
	39	virtual Double_t * GetBinLimits(Int_t ivar) const {return fGrid[0]->GetBinLimits(ivar);}
	40	virtual Long_t GetNBinsTotal() const {return fGrid[0]->GetNBinsTotal()*fNStep;}
	41	virtual Int_t GetNBins(Int_t ivar) const {return fGrid[0]->GetNBins(ivar);}
	42	virtual Int_t * GetNBins() const {return fGrid[0]->GetNBins();}
	43	virtual Float_t GetBinCenter(Int_t ivar,Int_t ibin) const {return fGrid[0]->GetBinCenter(ivar,ibin);}
	44	virtual Float_t GetBinSize (Int_t ivar,Int_t ibin) const {return fGrid[0]->GetBinSize (ivar,ibin);}
	45
	46	//virtual void GetBinCenters(const Int_t ibin, const Double_t binCenter) const {return fGrid[0]->GetBinCenters(ibin,binCenter);}
	47	//virtual void GetBinSizes(const Int_t ibin, const Double_t binSizes) const {return fGrid[0]->GetBinSizes(ibin,binSizes);}
	48
	49	//probably not needed anymore
	50	//virtual Int_t GetBinIndex(const Int_t *ibin) const {return fGrid[0]->GetBinIndex(ibin);}
	51	//virtual void GetBinIndex(Int_t iel, const Int_t *ibin) const {return fGrid[0]->GetBinIndex(iel,ibin);}
	52	//virtual Int_t GetBinIndex(Int_t ivar, Int_t ind) const {return fGrid[0]->GetBinIndex(ivar,ind);}
	53
	54	virtual TAxis * GetAxis(Int_t ivar, Int_t istep) const {return fGrid[istep]->GetAxis(ivar);}
	55	virtual void SetVarTitle (Int_t ivar, const Char_t* title) ;
	56	virtual void SetStepTitle(Int_t istep, const Char_t* title) ;
	57	virtual const Char_t* GetVarTitle (Int_t ivar) const {return GetAxis(ivar,0)->GetTitle();}
	58	virtual const Char_t* GetStepTitle(Int_t istep) const {return fGrid[istep]->GetTitle();}
	59	virtual Int_t GetStep(const Char_t* title) const ; // returns the step corresponding to the given title
	60	virtual Int_t GetVar (const Char_t* title) const ; // returns the variable corresponding to the given title
9105291d	61
fb494025	62	virtual Int_t GetNStep() const {return fNStep;};
9105291d	63	virtual void SetNStep(Int_t nStep) {fNStep=nStep;}
fb494025	64	virtual void Fill(const Double_t *var, Int_t istep, Double_t weight=1.) ;
9105291d	65
fb494025	66	virtual Float_t GetOverFlows (Int_t var,Int_t istep,Bool_t excl=kFALSE) const;
	67	virtual Float_t GetUnderFlows(Int_t var,Int_t istep,Bool_t excl=kFALSE) const ;
	68	virtual Float_t GetEntries (Int_t istep) const ;
	69	virtual Long_t GetEmptyBins(Int_t istep) const {return fGrid[istep]->GetEmptyBins();}
	70	//virtual Int_t GetEmptyBins(Int_t istep, Double_t varMin,Double_t varMax) const ;
	71	virtual Double_t GetIntegral (Int_t istep) const ;
	72	//virtual Double_t GetIntegral (Int_t istep, Double_t varMin,Double_t varMax) const ;
563113d0	73
563113d0	74
fb494025	75	//basic operations
fb494025	76	virtual void Add(const AliCFContainer* aContainerToAdd, Double_t c=1.);
563113d0	77	virtual Long64_t Merge(TCollection* list);
563113d0	78
fb494025	79	virtual TH1D* ShowProjection( Int_t ivar, Int_t istep) const {return Project(ivar ,istep);}
	80	virtual TH2D* ShowProjection( Int_t ivar1, Int_t ivar2, Int_t istep) const {return Project(ivar1,ivar2 ,istep);}
	81	virtual TH3D* ShowProjection( Int_t ivar1, Int_t ivar2,Int_t ivar3, Int_t istep) const {return Project(ivar1,ivar2,ivar3,istep);}
	82	virtual TH1D* Project( Int_t ivar, Int_t istep) const;
	83	virtual TH2D* Project( Int_t ivar1, Int_t ivar2, Int_t istep) const;
	84	virtual TH3D* Project( Int_t ivar1, Int_t ivar2,Int_t ivar3, Int_t istep) const;
1e9dad92	85	virtual TH1D* ShowSlice( Int_t ivar, Double_t varMin, Double_t varMax, Int_t istep) const;
c8df672e	86	virtual TH2D* ShowSlice( Int_t ivar1, Int_t ivar2, Double_t varMin, Double_t varMax, Int_t istep) const;
c8df672e	87	virtual TH3D* ShowSlice( Int_t ivar1, Int_t ivar2, Int_t ivar3, Double_t varMin, Double_t varMax, Int_t istep) const;
fb494025	88	virtual AliCFContainer* MakeSlice (Int_t nVars, const Int_t* vars, const Double_t* varMin, const Double_t* varMax) const ;
fb494025	89	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) const ;
c8df672e	90	virtual void SetRangeUser(Int_t ivar, Double_t varMin, Double_t varMax, Int_t istep) ;
9105291d	91	virtual void SetRangeUser(Double_t* varMin, Double_t* varMax, Int_t istep) ;
fb494025	92	virtual void SetGrid(Int_t step, AliCFGridSparse* grid) {fGrid[step]=grid;}
fb494025	93	virtual AliCFGridSparse * GetGrid(Int_t istep) const {return fGrid[istep];};
563113d0	94
	95	private:
	96	Int_t fNStep; //number of selection steps
fb494025	97	AliCFGridSparse **fGrid;//[fNStep]
563113d0	98
fb494025	99	ClassDef(AliCFContainer,5);
563113d0	100	};
fb494025	101
	102	inline void AliCFContainer::SetBinLimits(Int_t ivar, const Double_t* array) {
	103	for (Int_t iStep=0; iStep<GetNStep(); iStep++) {
	104	fGrid[iStep]->SetBinLimits(ivar,array);
	105	}
	106	}
	107
	108	inline void AliCFContainer::SetVarTitle(Int_t ivar, const Char_t* title) {
	109	for (Int_t iStep=0; iStep<fNStep; iStep++) {
	110	GetAxis(ivar,iStep)->SetTitle(title);
	111	}
	112	}
	113
	114	inline void AliCFContainer::SetStepTitle(Int_t istep, const Char_t* title) {
	115	fGrid[istep]->SetTitle(title);
	116	}
	117
	118	inline Int_t AliCFContainer::GetStep(const Char_t* title) const {
	119	TString str(title);
	120	for (Int_t iStep=0; iStep<fNStep; iStep++) {
	121	if (!str.CompareTo(GetStepTitle(iStep))) return iStep;
	122	}
	123	AliError("Step not found");
	124	return -1;
	125	}
	126
	127	inline Int_t AliCFContainer::GetVar(const Char_t* title) const {
	128	return fGrid[0]->GetVar(title);
	129	}
563113d0	130	#endif
563113d0	131