AliForwardMultDists.h

Go to the documentation of this file.

#ifndef ALIFORWARDMULTDIST_H
#define ALIFORWARDMULTDIST_H

#include "AliBaseAODTask.h"
#include <TList.h>
#include <TString.h>
class TH1;
class TH2;
class AliAODForwardMult;

class AliForwardMultDists : public AliBaseAODTask
{
public:
  enum { 
    kInvalidEta = 999
  };
  enum { 
    kAnalysis = 1, 
    kMC       = 2, 
    kTrigger  = 3, 
    kVertex   = 4, 
    kTriggerVertex = 5
  };
  struct BinSpec 
  {
    BinSpec(Double_t etaMin, Double_t etaMax, Double_t nchLow);
    void Push(UShort_t n, Double_t d);
    const TAxis& Axis() const;
    Double_t fEtaMin; // Least pseudo-rapidity
    Double_t fEtaMax; // Largest pseudo-rapidity
    Double_t fLow;    // Low cut 
    TArrayI  fN;      // Number of bins
    TArrayD  fD;      // Bin widths 
    mutable TAxis    fAxis; // Our axis 
  };

  AliForwardMultDists();
  AliForwardMultDists(const char* name);
  virtual ~AliForwardMultDists() {}
  Bool_t Book();
  Bool_t PreData();
  Bool_t PreEvent() { fIsSelected = false; return true; }
  Bool_t Event(AliAODEvent& aod);
  Bool_t Finalize();
  /* @} */
  void AddBin(const BinSpec& spec);
  void AddBin(Double_t etaLow, Double_t etaMax, const TAxis& nAxis); 
  void AddBin(Double_t etaLow, Double_t etaMax, UShort_t nMax, UShort_t nDiv); 
  /* @} */
  void SetUsePhiAcc(Bool_t use) { fUsePhiAcc = use; }
  void Print(Option_t* option="") const;
  struct EtaBin : public TObject
  {
    EtaBin();
    EtaBin(Double_t minEta, Double_t maxEta, const TAxis& mAxis); 
    EtaBin(const EtaBin& o);
    EtaBin& operator=(const EtaBin& o);
    virtual ~EtaBin() {}
    const char* GetName() const { return fName.Data(); }
    Bool_t IsSymmetric() const;
    Bool_t IsNegative() const;
    Bool_t IsPositive() const;
    const char* ParentName() const;
    TList* FindParent(TList* l, Bool_t create=true) const;
    static TH1* CreateH1(const char* name, const char* title, 
       const TAxis& xAxis);
    static TH2* CreateH2(const char* name, const char* title, 
       const TAxis& xAxis, const TAxis& yAxis);
    void SetupForData(TList* list, const TH2& hist, Bool_t useMC);
    void Process(const TH1& sumForward, const TH1& sumCentral,
     const TH2& forward,    const TH2& central,
     Bool_t     accepted,   const TH1* mc);
    void Terminate(TList* in, TList* out);
      
    TString  fName;          // Name of this bin
    TAxis    fMAxis;         // Axis used for measured Nch
    TAxis    fTAxis;         // Axis used for true Nch
    Double_t fMinEta;        // Least @f$\eta@f$ to consider
    Double_t fMaxEta;        // Largest @f$\eta@f$ to consider
    Int_t    fMinBin;        // Least @f$\eta@f$ bin to consider
    Int_t    fMaxBin;        // Largest @f$\eta@f$ bin to consider
    TH1*     fSum;           // Distribution 
    TH2*     fCorr;          // Correlation between forward and central
    TH2*     fResponse;      // Response matrix (for MC)
    TH1*     fTruth;         // `true' distribution 
    TH1*     fTruthAccepted; // `true' distribution for accepted events
    TH1*     fCoverage;      // How much was covered

    ClassDef(EtaBin,2);
  };
protected:
  static void ProjectX(const TH2& input, TH1& cache, Bool_t usePhiAcc=true);
  static void ProjectX(const TH2* input, TH1* cache);
  AliForwardMultDists(const AliForwardMultDists& o);
  AliForwardMultDists& operator=(const AliForwardMultDists& o);
  Bool_t CheckEvent(const AliAODForwardMult& fwd);

  TList    fBins;         // List of bins 
  TList*   fSymmetric;    // Bins symmetric around 0
  TList*   fNegative;     // Bins on negative side only 
  TList*   fPositive;     // Bins on the positive side only
  TH1*     fMCVertex;     // Histogram of MC IpZ
  TH2*     fDiag;         // Diagnostics
  TH1*     fForwardCache; // Projection cache 
  TH1*     fCentralCache; // Projection cache 
  TH1*     fMCCache;      // Projection cache 
  Bool_t   fUsePhiAcc;    // If true, scale by phi acceptance 
  Bool_t   fIsSelected;   // IF the even was selected

  ClassDef(AliForwardMultDists,1);
};

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