AliFMDCorrSecondaryMap.h

Go to the documentation of this file.

//
// This class contains the secondary correction and the double hit
// correction used in low-flux events.
//
#ifndef ALIFMDCORRSECONDARYMAP_H
#define ALIFMDCORRSECONDARYMAP_H

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

class AliFMDCorrSecondaryMap : public TObject 
{
public:
  AliFMDCorrSecondaryMap();
  AliFMDCorrSecondaryMap(const AliFMDCorrSecondaryMap& o);
  virtual ~AliFMDCorrSecondaryMap();
  AliFMDCorrSecondaryMap& operator=(const AliFMDCorrSecondaryMap& 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;
  const TAxis& GetVertexAxis() const { return fVertexAxis; }
  const TAxis& GetEtaAxis() const { return fEtaAxis; }
  /* @} */

  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 SetEtaAxis(const TAxis& axis);
  void SetEtaAxis(Int_t nBins, Double_t min, Double_t max);
  /* @} */

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

  TObjArray fRingArray;      // Array of per-ring, per-vertex 2nd map
  TAxis     fVertexAxis;     // The vertex axis 
  TAxis     fEtaAxis;        // The eta axis
  ClassDef(AliFMDCorrSecondaryMap,1); // 
};

//____________________________________________________________________
inline void 
AliFMDCorrSecondaryMap::SetVertexAxis(Int_t nBins, Double_t min, Double_t max)
{
  fVertexAxis.Set(nBins, min, max);
}
//____________________________________________________________________
inline void 
AliFMDCorrSecondaryMap::SetVertexAxis(const TAxis& e)
{
  if (e.GetXbins() && e.GetXbins()->GetArray())
    fVertexAxis.Set(e.GetNbins(), e.GetXbins()->GetArray());
  else 
    fVertexAxis.Set(e.GetNbins(), e.GetXmin(), e.GetXmax());
}
//____________________________________________________________________
inline void 
AliFMDCorrSecondaryMap::SetEtaAxis(Int_t nBins, Double_t min, Double_t max)
{
  fEtaAxis.Set(nBins, min, max);
}
//____________________________________________________________________
inline void 
AliFMDCorrSecondaryMap::SetEtaAxis(const TAxis& e)
{
  fEtaAxis.Set(e.GetNbins(), e.GetXmin(), e.GetXmax());
}
#endif
// Local Variables:
//  mode: C++
// End: