AliForwardMCCorrectionsTask.h

Go to the documentation of this file.

// 
// Calculate the corrections in the forward regions
// 
#ifndef ALIFORWARDMCCORRECTIONS_H
#define ALIFORWARDMCCORRECTIONS_H

#include "AliBaseMCCorrectionsTask.h"
#include <AliESDFMD.h>
#include "AliFMDMCTrackDensity.h"
#include "AliForwardUtil.h"
#include <TH1I.h>
class AliESDEvent;
class AliFMDCorrSecondaryMap;
class TList;
class TVector3;

class AliForwardMCCorrectionsTask : public AliBaseMCCorrectionsTask
{
public:
  AliForwardMCCorrectionsTask(const char* name);
  AliForwardMCCorrectionsTask();
  virtual Bool_t PreEvent();
  void Print(Option_t* option="") const;
  AliBaseMCTrackDensity& GetTrackDensity() { return fTrackDensity; }
  const AliBaseMCTrackDensity& GetTrackDensity() const { return fTrackDensity; }
protected: 
  AliForwardMCCorrectionsTask(const AliForwardMCCorrectionsTask& o);
  AliForwardMCCorrectionsTask& operator=(const AliForwardMCCorrectionsTask& o);
  struct VtxBin : public AliBaseMCCorrectionsTask::VtxBin
  {
    VtxBin();
    VtxBin(Double_t low, Double_t high, const TAxis& etaAxis);
    VtxBin(const VtxBin& o){;}
    VtxBin& operator=(const VtxBin& o){return *this;}
    TList* CreateOutputObjects(TList* list);
    TH2D* MakeBg(const TH2D* hits, const TH2D* primary) const;
    void Terminate(const TList*            i, 
    TList*                  o,
    UShort_t                iVz, 
    AliFMDCorrSecondaryMap* map);

    AliForwardUtil::Histos fHists;    // Cache of per-ring histograms
    // ClassDef(VtxBin,2); // Vertex bin 
  };
  AliBaseMCCorrectionsTask::VtxBin* CreateVtxBin(Double_t low, Double_t high);
  Bool_t ProcessESD(const AliESDEvent& esd, const AliMCEvent& mc, 
        AliBaseMCCorrectionsTask::VtxBin& bin,
        const TVector3& ip);
  virtual void CreateCorrections(TList* results);
  virtual Bool_t FinalizeVtxBin(AliBaseMCCorrectionsTask::VtxBin* bin, 
        UShort_t     iVz);


  AliFMDMCTrackDensity    fTrackDensity; // Get the track density 
  AliESDFMD               fESDFMD;       // Cache object
  AliFMDCorrSecondaryMap* fSecCorr;
  ClassDef(AliForwardMCCorrectionsTask,4) // Forward corrections class
};

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