PWG0/dNdEta/AlidNdEtaCorrection.h

   1 /* $Id$ */
   2
   3 #ifndef ALIDNDETACORRECTION_H
   4 #define ALIDNDETACORRECTION_H
   5
   6
   7 // ------------------------------------------------------
   8 //
   9 // Class to handle corrections for dN/dEta measurements
  10 //
  11 // ------------------------------------------------------
  12 //
  13 // TODO:
  14 // - make the ntrack to npart correction 3D
  15 // - add documentation
  16 // - add status: generate or use maps
  17 // - add functionality to set the bin sizes
  18 // - update MERge function
  19 //
  20
  21 #include <TNamed.h>
  22
  23 #include <AliCorrectionMatrix2D.h>
  24 #include <AliCorrectionMatrix3D.h>
  25
  26 class AlidNdEtaCorrection : public TNamed
  27 {
  28 public:
  29   AlidNdEtaCorrection();
  30   AlidNdEtaCorrection(const Char_t* name, const Char_t* title);
  31
  32   // fVertexRecoCorrection, fTriggerCorrection
  33   void FillEvent(Float_t vtx, Float_t n)                        {fTriggerCorrection->FillGene(vtx, n);}
  34   void FillEventWithTrigger(Float_t vtx, Float_t n)             {fVertexRecoCorrection->FillGene(vtx, n); fTriggerCorrection->FillMeas(vtx, n);}
  35   void FillEventWithTriggerWithReconstructedVertex(Float_t vtx, Float_t n) {fVertexRecoCorrection->FillMeas(vtx, n);}
  36
  37   // fTrack2ParticleCorrection
  38   void FillParticle(Float_t vtx, Float_t eta, Float_t pt)                  {fTrack2ParticleCorrection->FillGene(vtx, eta, pt);}
  39   void FillParticleWhenMeasuredTrack(Float_t vtx, Float_t eta, Float_t pt) {fTrack2ParticleCorrection->FillMeas(vtx, eta, pt);}
  40
  41   // fTriggerBiasCorrection
  42   void FillParticleAllEvents(Float_t eta, Float_t pt)          {fTriggerBiasCorrection->FillGene(eta, pt);}
  43   void FillParticleWhenEventTriggered(Float_t eta, Float_t pt) {fTriggerBiasCorrection->FillMeas(eta, pt);}
  44
  45   void IncreaseEventCount() { fNEvents++; }
  46   void IncreaseTriggeredEventCount() { fNTriggeredEvents++; }
  47
  48   void Finish();
  49
  50   AliCorrectionMatrix3D* GetTrack2ParticleCorrection()    {return fTrack2ParticleCorrection;}
  51   AliCorrectionMatrix2D* GetVertexRecoCorrection()        {return fVertexRecoCorrection;}
  52   AliCorrectionMatrix2D* GetTriggerBiasCorrection()       {return fTriggerBiasCorrection;}
  53
  54   virtual Long64_t Merge(TCollection* list);
  55
  56   void    SaveHistograms();
  57   Bool_t  LoadHistograms(const Char_t* fileName, const Char_t* dir = "dndeta_correction");
  58   Bool_t  LoadCorrection(const Char_t* fileName, const Char_t* dir = "dndeta_correction")
  59     {return LoadHistograms(fileName, dir);}
  60
  61   void DrawHistograms();
  62
  63   //  void RemoveEdges(Float_t cut=2, Int_t nBinsVtx=0, Int_t nBinsEta=0);
  64
  65   Float_t GetTrack2ParticleCorrection(Float_t vtx, Float_t eta, Float_t pt)
  66     {return fTrack2ParticleCorrection->GetCorrection(vtx, eta, pt);}
  67
  68   Float_t GetVertexRecoCorrection(Float_t vtx, Float_t n) {return fVertexRecoCorrection->GetCorrection(vtx, n);}
  69
  70   Float_t GetTriggerCorrection(Float_t vtx, Float_t n) {return fTriggerCorrection->GetCorrection(vtx, n);}
  71
  72   Float_t GetTriggerBiasCorrection(Float_t eta, Float_t pt=0) {return fTriggerBiasCorrection->GetCorrection(eta, pt);}
  73
  74   Float_t GetMeasuredFraction(Float_t ptCutOff, Float_t eta = -1, Bool_t debug = kFALSE);
  75
  76   void SetNEvents(Long64_t events) { fNEvents = events; }
  77
  78   void ReduceInformation();
  79
  80 protected:
  81   AliCorrectionMatrix3D* fTrack2ParticleCorrection; //-> handles the track-to-particle correction, function of vtx_z, eta, pt
  82   AliCorrectionMatrix2D* fVertexRecoCorrection;     //-> handles the vertex reconstruction efficiency, function of n_clustersITS and vtx_z
  83   AliCorrectionMatrix2D* fTriggerCorrection;       //-> handles the trigger efficiency, function of n_clustersITS and vtx_z
  84
  85   AliCorrectionMatrix2D* fTriggerBiasCorrection;          //-> MB to desired sample
  86
  87   Long64_t fNEvents;
  88   Long64_t fNTriggeredEvents;
  89
  90   ClassDef(AlidNdEtaCorrection, 1)
  91 };
  92
  93 #endif