AliFMDCorrector.h

Go to the documentation of this file.

// This class calculates the exclusive charged particle density
// in each for the 5 FMD rings. 
//
#ifndef ALIFMDCORRECTOR_H
#define ALIFMDCORRECTOR_H

#include <TNamed.h>
#include <TList.h>
#include "AliForwardUtil.h"
class TH2D;
class TH2;
class AliFMDCorrector : public TNamed
{
public:
  AliFMDCorrector();
  AliFMDCorrector(const char* name);
  AliFMDCorrector(const AliFMDCorrector& o);
  virtual ~AliFMDCorrector();
  AliFMDCorrector& operator=(const AliFMDCorrector&);
  virtual void SetupForData(const TAxis& etaAxis);
  virtual Bool_t Correct(AliForwardUtil::Histos& hists, UShort_t vtxBin);
  virtual void Terminate(const TList* dir, TList* output, Int_t nEvents);
  virtual void CreateOutputObjects(TList* dir);
  void SetDebug(Int_t dbg=1) { fDebug = dbg; }
  void SetUseSecondaryMap(Bool_t use=true) { fUseSecondaryMap = use; }
  Bool_t IsUseSecondaryMap() const { return fUseSecondaryMap; }
  void SetUseVertexBias(Bool_t use=true) { fUseVertexBias = use; }
  Bool_t IsUseVertexBias() const { return fUseVertexBias; }
  void SetUseAcceptance(Bool_t use=true) { fUseAcceptance = use; }
  Bool_t IsUseAcceptance() const { return fUseAcceptance; }
  void SetUseMergingEfficiency(Bool_t use=true) { fUseMergingEfficiency = use; }
  Bool_t IsUseMergingEfficiency() const { return fUseMergingEfficiency; }
  /* @} */

  virtual void Print(Option_t* option="") const;
protected:
  struct RingHistos : public AliForwardUtil::RingHistos 
  { 
    RingHistos();
    RingHistos(UShort_t d, Char_t r);
    RingHistos(const RingHistos& o);
    RingHistos& operator=(const RingHistos& o);
    ~RingHistos();
    void CreateOutputObjects(TList* dir);
    void Terminate(TList* dir, Int_t nEvents);
    TH2D*     fDensity;      // Distribution primary Nch
    // ClassDef(RingHistos,2);
  };
  RingHistos* GetRingHistos(UShort_t d, Char_t r) const;
  void DivideMap(TH2* num, const TH2* denom, Bool_t alsoUnderOver=false) const;
  TList    fRingHistos;           // List of histogram containers
  Bool_t   fUseSecondaryMap;      // Whether to do correction for secondaries
  Bool_t   fUseVertexBias;        // Whether to do correction for vertex bias
  Bool_t   fUseAcceptance;        // Whether to do correction for dead ch's.
  Bool_t   fUseMergingEfficiency; // Whether to use the merging efficiency
  Int_t    fDebug;                //  Debug level 

  ClassDef(AliFMDCorrector,4); // Correct the inclusive d2N/detadphi
};

#endif
// Local Variables:
//   mode: C++
// End: