AliBaseMCCorrectionsTask.h

Go to the documentation of this file.

// 
// Calculate the corrections in the base regions
// 
#ifndef ALIBASEMCCORRECTIONS_H
#define ALIBASEMCCORRECTIONS_H

#include <AliBaseESDTask.h>
#include <AliESDFMD.h>
#include "AliFMDMCEventInspector.h"
#include <TH1I.h>
class AliBaseMCTrackDensity;
class AliCorrectionManagerBase;
class AliESDEvent;
class TH2D;
class TH1D;
class TList;
class TVector3;

class AliBaseMCCorrectionsTask : public AliBaseESDTask
{
public:
  AliBaseMCCorrectionsTask(const char* name,
         AliCorrectionManagerBase* m);
  AliBaseMCCorrectionsTask();
  virtual Bool_t Book();
  virtual Bool_t Event(AliESDEvent& esd);
  virtual Bool_t Finalize();
  void Print(Option_t* option="") const;

  void SetVertexAxis(Int_t nBins, Double_t vzMin, Double_t vzMax=-1000000);
  void SetVertexAxis(const TAxis& axis);
  void SetEtaAxis(Int_t nBins, Double_t etaMin, Double_t etaMax=-1000000);
  void SetEtaAxis(const TAxis& axis);
  void SetSatellite(Bool_t sat);
  virtual AliBaseMCTrackDensity& GetTrackDensity() = 0;
  virtual const AliBaseMCTrackDensity& GetTrackDensity() const = 0;
  AliFMDEventInspector& GetEventInspector() { return fInspector; }
  const AliFMDEventInspector& GetEventInspector() const { return fInspector;}
  void SetUseESDVertex(Bool_t use){ fUseESDVertex = use;}
  void SetAfterEventSel(Bool_t use){ fAfterEventSel = use; }
protected: 
  AliBaseMCCorrectionsTask(const AliBaseMCCorrectionsTask& o);
  AliBaseMCCorrectionsTask& operator=(const AliBaseMCCorrectionsTask& o);
  struct VtxBin : public TNamed
  {
    VtxBin();
    VtxBin(Double_t low, Double_t high, const TAxis& etaAxis, UShort_t nPhi);
    virtual ~VtxBin() {}
    VtxBin(const VtxBin& o){;}
    VtxBin& operator=(const VtxBin& o){return *this;}
    static const char* BinName(Double_t low, Double_t high);
    virtual TList* CreateOutputObjects(TList* list);
    TH2D*                  fPrimary;  // Cache or primary 
    TH1D*                  fCounts;   // Event count 

    // ClassDef(VtxBin,1); // Vertex bin 
  };
  virtual TAxis* DefaultVertexAxis() const { return const_cast<TAxis*>(&fVtxAxis); }
  virtual TAxis* DefaultEtaAxis() const { return const_cast<TAxis*>(&fEtaAxis); }
  virtual VtxBin* CreateVtxBin(Double_t low, Double_t high) = 0;
  virtual Bool_t ProcessESD(const AliESDEvent& esd, 
          const AliMCEvent& mc, 
          VtxBin& bin,
          const TVector3& ip) = 0;
  virtual void CreateCorrections(TList* results) = 0;
  virtual Bool_t FinalizeVtxBin(VtxBin*      bin, 
        UShort_t     iVz) = 0;
  void DefineBins(TList* list);

  AliFMDMCEventInspector fInspector; // Event inspector 

  TObjArray* fVtxBins;      // Vertex bins 
  TH1I*      fHEvents;      // All Events
  TH1I*      fHEventsTr;    // Histogram of events w/trigger
  TH1I*      fHEventsTrVtx; // Events w/trigger and vertex 
  TAxis      fVtxAxis;      // Vertex axis 
  TAxis      fEtaAxis;      // Eta axis 
  Bool_t     fUseESDVertex; // if true use Z vertex from ESD in calculations
  Bool_t     fAfterEventSel; //if true corr. be calc. after events selection

  ClassDef(AliBaseMCCorrectionsTask,1) // Base corrections class
};

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