AliCentralMCCorrectionsTask.h

Go to the documentation of this file.

// 
// Calculate the corrections in the central regions
// 
#ifndef ALICENTRALMCCORRECTIONS_H
#define ALICENTRALMCCORRECTIONS_H

#include "AliBaseMCCorrectionsTask.h"
#include "AliSPDMCTrackDensity.h"
class AliCentralCorrSecondaryMap;
class AliCentralCorrAcceptance;


class AliCentralMCCorrectionsTask : public AliBaseMCCorrectionsTask
{
public:
  AliCentralMCCorrectionsTask(const char* name);
  AliCentralMCCorrectionsTask();
  void Print(Option_t* option="") const;

  void SetNPhiBins(UShort_t nBins) { fNPhiBins = nBins; }
  void SetEffectiveCorrection(Bool_t e) { fEffectiveCorr = e; }
  void SetEtaCut(Double_t maxEta=1.9) { fEtaCut = maxEta; }
  void SetAcceptanceCut(Double_t least=0.8) { fCorrCut = least; }
  AliSPDMCTrackDensity& GetTrackDensity() { return fTrackDensity; }
  const AliSPDMCTrackDensity& GetTrackDensity() const { return fTrackDensity; }
protected: 
  AliCentralMCCorrectionsTask(const AliCentralMCCorrectionsTask& o);
  AliCentralMCCorrectionsTask& operator=(const AliCentralMCCorrectionsTask& o);
  struct VtxBin : public AliBaseMCCorrectionsTask::VtxBin
  {
    VtxBin();
    VtxBin(Double_t low, Double_t high, const TAxis& etaAxis, UShort_t nPhi);
    VtxBin(const VtxBin& o){;}
    VtxBin& operator=(const VtxBin& o){return *this;}
    TList* CreateOutputObjects(TList* list);
    void Terminate(const TList* i, 
       TList* o,
       UShort_t iVz, 
       Bool_t effective,
       Double_t etaCut, 
       Double_t accCut,
       AliCentralCorrSecondaryMap* map,
       AliCentralCorrAcceptance* acorr);
    
    TH2D* fHits;     // Cache of MC-truth hits
    TH2D* fClusters; // Cache of reconstructed hits

    // ClassDef(VtxBin,3); // Vertex bin 
  };
  void DefineBins(TList* list);
  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);


  AliSPDMCTrackDensity        fTrackDensity; // Get the track density 
  AliCentralCorrSecondaryMap* fSecCorr;
  AliCentralCorrAcceptance*   fAccCorr;

  UShort_t   fNPhiBins;      // Nunber of phi bins
  Bool_t     fEffectiveCorr; // Whether to make effective corrections
  Double_t   fEtaCut;        // Maximum Eta
  Double_t   fCorrCut;       // Correction cut
  ClassDef(AliCentralMCCorrectionsTask,3) // Central corrections class
};

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