AliForwardMultiplicityBase.h

Go to the documentation of this file.

// 
// Base class for classes that calculate the multiplicity in the
// forward regions event-by-event
// 
#ifndef ALIFORWARDMULTIPLICITYBASE_H
#define ALIFORWARDMULTIPLICITYBASE_H

#include "AliBaseESDTask.h"
#include "AliForwardUtil.h"
#include "AliAODForwardMult.h"
#include "AliAODForwardEP.h"
#include "AliAODMultEventClass.h"
// class AliFMDEnergyFitter;
class AliFMDESDFixer;
class AliFMDSharingFilter;
class AliFMDDensityCalculator;
class AliFMDCorrector;
class AliFMDHistCollector;
class AliFMDEventPlaneFinder;
// class AliMultEventClassifier;
class AliAODHandler;
class AliESDEvent;
class TH2D;
class TList;
class TTree;
class TAxis;
class TProfile;

class AliForwardMultiplicityBase : public AliBaseESDTask
{
public:
  enum { 
    kTimingEventInspector    = 1,
    kTimingSharingFilter     = 2, 
    kTimingDensityCalculator = 3, 
    kTimingCorrections       = 4, 
    kTimingHistCollector     = 5, 
    kTimingEventPlaneFinder  = 6, 
    kTimingTotal             = 7
  };
  enum {
    kStatusNoEvent = 1,
    kStatusNoTrigger,
    kStatusNoSPD,
    kStatusNoFMD,
    kStatusNoVertex,
    kStatusPileup,
    kStatusSPDOutlier,
    kStatusIPzOutOfRange,
    kStatusFailSharing,
    kStatusFailDensity,
    kStatusFailEventPlane,
    kStatusOutlier,
    kStatusFailCorrector,
    kStatusFailCollector,
    kStatusNotAdded,
    kStatusAllThrough       
  };
  virtual Bool_t Book();
  virtual Bool_t PreData(const TAxis& vertex, const TAxis& eta);
  virtual Bool_t PostEvent();
  virtual Bool_t Finalize();
  virtual void Print(Option_t* option="") const;
  virtual void SetDoTiming(Bool_t enable=true) { fDoTiming = enable; }
  virtual void SetEnableLowFlux(Bool_t use=true) { fEnableLowFlux = use; }
  // virtual AliMultEventClassifier& GetMultEventClassifier() = 0;
  // virtual const AliMultEventClassifier& GetMultEventClassifier() const = 0;
  virtual AliFMDESDFixer& GetESDFixer() = 0;
  virtual AliFMDSharingFilter& GetSharingFilter() = 0;
  virtual AliFMDDensityCalculator& GetDensityCalculator() = 0;
  virtual AliFMDCorrector& GetCorrections() = 0;
  virtual AliFMDHistCollector& GetHistCollector() = 0;
  virtual const AliFMDESDFixer& GetESDFixer() const = 0;
  virtual const AliFMDSharingFilter& GetSharingFilter() const = 0;
  virtual const AliFMDDensityCalculator& GetDensityCalculator() const = 0;
  virtual const AliFMDCorrector& GetCorrections() const = 0;
  virtual const AliFMDHistCollector& GetHistCollector() const = 0;
  virtual AliFMDEventPlaneFinder& GetEventPlaneFinder() = 0;
  virtual const AliFMDEventPlaneFinder& GetEventPlaneFinder() const = 0;
  /* @} */

  virtual void SetDebug(Int_t dbg);
  void SetStorePerRing(Bool_t use) { fStorePerRing = use; }
  void SetAddMask(UInt_t mask)  { fAddMask = mask; }
protected: 
  AliForwardMultiplicityBase(const char* name); 
  AliForwardMultiplicityBase() 
    : AliBaseESDTask(), 
      fEnableLowFlux(true), 
      fStorePerRing(false),
      fHData(0),
      fHistos(),
      fAODFMD(),
      fAODEP(),
      // fAODRef(),
      fRingSums(),
      fDoTiming(false), 
      fHTiming(0),
      fHStatus(0),
      fAddMask(AliAODForwardMult::kInel)
  {}
  AliForwardMultiplicityBase(const AliForwardMultiplicityBase& o);
  AliForwardMultiplicityBase& operator=(const AliForwardMultiplicityBase& o);
  virtual void InitMembers(const TAxis& pe, const TAxis& pv);
  virtual void CreateBranches(AliAODHandler* ah);
  virtual void EstimatedNdeta(const TList* input, TList* output) const;
  virtual Bool_t MakeSimpledNdeta(const TList* input, 
          TList*       output,
          Double_t&    nTr, 
          Double_t&    nTrVtx, 
          Double_t&    nAcc);
  virtual void MakeRingdNdeta(const TList* input, 
            const char*  inName,
            TList*       output,
            const char*  outName,
            Int_t        style=20) const;
  TAxis* DefaultEtaAxis() const { return new TAxis(200,-4,6); }
  TAxis* DefaultVertexAxis() const { return new TAxis(10,-10,10); }
  Bool_t                 fEnableLowFlux;// Whether to use low-flux specific code
  Bool_t                 fStorePerRing; // Store each ring on separate branch
  TH2D*                  fHData;        // Summed 1/Nd^2N_{ch}/dphideta
  AliForwardUtil::Histos fHistos;       // Cache histograms 
  AliAODForwardMult      fAODFMD;       // Output object
  AliAODForwardEP        fAODEP;        // Output object
  // AliAODMultEventClass   fAODRef;       // Reference multiplicity
  AliForwardUtil::Histos fRingSums;     // Cache histograms 
  Bool_t                 fDoTiming;     // Whether to do timing or not
  TProfile*              fHTiming;      // Timing histogram 
  TH1*                   fHStatus;      // Status histogram
  UInt_t                 fAddMask;      // For which triggers to add internally
  ClassDef(AliForwardMultiplicityBase,7) // Forward multiplicity class
};

#endif

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