dbefa84ed5275e5b64bc41bdfbad00a3902b6ef4
[u/mrichter/AliRoot.git] / 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 // - add functionality to set the bin sizes
15 // - update MERge function
16 //
17
18 #include <TCollection.h>
19 #include <TNamed.h>
20 #include "AliPWG0Helper.h"
21
22 class AliCorrection;
23 class TH1;
24
25 class AlidNdEtaCorrection : public TNamed
26 {
27 public:
28   enum CorrectionType {
29     kNone = 0,
30     kTrack2Particle,  // measured events
31     kVertexReco,      // MB sample
32     kINEL,
33     kNSD,
34     kND
35   };
36
37   AlidNdEtaCorrection();
38   AlidNdEtaCorrection(const Char_t* name, const Char_t* title, AliPWG0Helper::AnalysisMode analysis = (AliPWG0Helper::AnalysisMode) (AliPWG0Helper::kTPC | AliPWG0Helper::kFieldOn));
39
40   virtual Long64_t Merge(TCollection* list);
41
42   ~AlidNdEtaCorrection();
43
44   void FillMCParticle(Float_t vtx, Float_t eta, Float_t pt, Bool_t trigger, Bool_t vertex, Int_t processType);
45   void FillTrackedParticle(Float_t vtx, Float_t eta, Float_t pt);
46   void FillEvent(Float_t vtx, Float_t n, Bool_t trigger, Bool_t vertex, Int_t processType);
47
48   void Finish();
49
50   AliCorrection* GetTrack2ParticleCorrection()  {return fTrack2ParticleCorrection;}
51   AliCorrection* GetVertexRecoCorrection()      {return fVertexRecoCorrection;}
52   AliCorrection* GetTriggerBiasCorrectionINEL() {return fTriggerBiasCorrectionMBToINEL;}
53   AliCorrection* GetTriggerBiasCorrectionNSD()  {return fTriggerBiasCorrectionMBToNSD;}
54   AliCorrection* GetTriggerBiasCorrectionND()   {return fTriggerBiasCorrectionMBToND;}
55   AliCorrection* GetCorrection(CorrectionType correctionType);
56
57   void    Reset(void);
58   void    Add(AlidNdEtaCorrection* aCorrectionsToAdd, Float_t c=1);
59
60   void    SaveHistograms();
61   Bool_t  LoadHistograms(const Char_t* dir = 0);
62   void    DrawHistograms();
63   void    DrawOverview(const char* canvasName = 0);
64
65   Float_t GetMeasuredFraction(CorrectionType correctionType, Float_t ptCutOff, Float_t eta = -100, Int_t vertexBegin = -1, Int_t vertexEnd = -1, Bool_t debug = kFALSE);
66   TH1*    GetMeasuredEventFraction(CorrectionType correctionType, Int_t multCut);
67
68   void    ReduceInformation();
69
70 protected:
71   AliCorrection* fTrack2ParticleCorrection;       //-> handles the track-to-particle correction (only track level (vtx_z, eta, pt))
72   AliCorrection* fVertexRecoCorrection;           //-> handles the vertex reconstruction efficiency, (n, vtx_z)
73   AliCorrection* fTriggerBiasCorrectionMBToINEL;  //-> handles the trigger bias MB->INEL, function of n and vtx_z
74   AliCorrection* fTriggerBiasCorrectionMBToNSD;   //-> handles the trigger bias MB->NSD,  function of n and vtx_z
75   AliCorrection* fTriggerBiasCorrectionMBToND;    //-> handles the trigger bias MB->ND,   function of n and vtx_z
76
77 private:
78   AlidNdEtaCorrection(const AlidNdEtaCorrection&);
79   AlidNdEtaCorrection& operator=(const AlidNdEtaCorrection&);
80
81   ClassDef(AlidNdEtaCorrection, 1)
82 };
83
84 #endif