AliFMDCorrAcceptance.h

Go to the documentation of this file.

//
// This class contains the acceptance correction due to dead channels
//
//
#ifndef ALIFMDCORRACCEPTANCE_H
#define ALIFMDCORRACCEPTANCE_H

#include <TObject.h>
#include <TObjArray.h>
#include <TAxis.h>
class TH2D;
class TH1D;

class AliFMDCorrAcceptance : public TObject 
{
public:
  AliFMDCorrAcceptance();
  AliFMDCorrAcceptance(const AliFMDCorrAcceptance& o);
  virtual ~AliFMDCorrAcceptance();
  AliFMDCorrAcceptance& operator=(const AliFMDCorrAcceptance& o);
  TH2D* GetCorrection(UShort_t d, Char_t r, Double_t v) const;
  TH2D* GetCorrection(UShort_t d, Char_t r, UShort_t b) const;
  TH1D* GetPhiAcceptance(UShort_t d, Char_t r, Double_t v) const;
  TH1D* GetPhiAcceptance(UShort_t d, Char_t r, UShort_t b) const;
  const TAxis& GetVertexAxis() const { return fVertexAxis; }
  Bool_t HasOverflow() const { return fHasOverflow; }
  /* @} */

  Bool_t SetCorrection(UShort_t d, Char_t r, Double_t v, TH2D* h);
  Bool_t SetCorrection(UShort_t d, Char_t r, UShort_t b, TH2D* h);
  void SetVertexAxis(const TAxis& axis);
  void SetVertexAxis(Int_t nBins, Double_t min, Double_t max);
  void SetHasOverflow(Bool_t present=true) { fHasOverflow = present; }
  /* @} */

  Bool_t IsFolder() const { return true; }
  void Browse(TBrowser* b);
  void Print(Option_t* option="R") const; //*MENU*
  void ls(Option_t* option="") const; //*MENU*
  // void Draw(Option_t* option="");
  // void SaveAs(const Char_t* filename, Option_t* option="colz") const;
  /* @} */
protected:
  Int_t FindVertexBin(Double_t vertex) const;
  Int_t GetRingIndex(UShort_t d, Char_t r) const;
  TObjArray* GetRingArray(const TObjArray& m, UShort_t d, Char_t r) const;
  TObjArray* GetOrMakeRingArray(TObjArray& m, UShort_t d, Char_t r) const;
  TObject* GetObject(const TObjArray& m, UShort_t d, 
         Char_t r, UShort_t b) const;

  void FillCache() const;

  TObjArray fRingArray;      // Array of per-ring, per-vertex 2d map
  mutable TObjArray* fCache;  
  TAxis     fVertexAxis;     // The vertex axis 
  Bool_t    fHasOverflow;    // Whether we have the overflow bin set
  ClassDef(AliFMDCorrAcceptance,2); // Acceptance correction due to dead areas
};

//____________________________________________________________________
inline void 
AliFMDCorrAcceptance::SetVertexAxis(Int_t nBins, Double_t min, 
             Double_t max)
{
  fVertexAxis.Set(nBins, min, max);
}
//____________________________________________________________________
inline void 
AliFMDCorrAcceptance::SetVertexAxis(const TAxis& e)
{
  fVertexAxis.Set(e.GetNbins(), e.GetXmin(), e.GetXmax());
}
#endif
// Local Variables:
//  mode: C++
// End: