AliFMDCorrVertexBias.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 ALIFMDCORRVERTEXBIAS_H
#define ALIFMDCORRVERTEXBIAS_H

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

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

  Bool_t SetCorrection(Char_t r, Double_t v, TH2D* h);
  Bool_t SetCorrection(Char_t r, UShort_t b, TH2D* h);
  void SetVertexAxis(const TAxis& axis);
  void SetVertexAxis(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;
  TObjArray* GetVertexArray(UShort_t v) const;
  TObjArray* GetOrMakeVertexArray(UShort_t v);

  TObjArray fVertexArray;    // Array of per-ring, per-vertex 2nd map
  TAxis     fVertexAxis;     // The vertex axis 
  ClassDef(AliFMDCorrVertexBias,1); // 
};

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