55b67e52a2294c7b23fbbbe5aee989694ab8228e
[u/mrichter/AliRoot.git] / ANALYSIS / AliCollisionNormalization.h
1
2 #ifndef ALICOLLISIONNORMALIZATION_H
3 #define ALICOLLISIONNORMALIZATION_H
4
5 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
6  * See cxx source for full Copyright notice                               */
7
8 //-------------------------------------------------------------------------
9 //                      Implementation of   Class AliCollisionNormalization
10 //
11 //  This class is used to store the vertex ditributions in the data
12 //  and in Monte Carlo, needed to compute the real number of
13 //  collisions a given sample is corresponding to.
14 //  The strategy matches what described in CERN-THESIS-2009-033 p 119.
15 //
16 //    Author:     Michele Floris, CERN
17 //-------------------------------------------------------------------------
18
19 #include "TH2F.h"
20
21 class TH1F;
22 class TH1I;
23 class AliMCEvent;
24
25 class AliCollisionNormalization : public TObject
26
27 {
28
29
30 public:
31   enum { kNevBin0, kNevCollisions, kNevNbin };
32   typedef enum { kProcSD, kProcDD, kProcND, kProcUnknown, kNProcs } ProcType_t; 
33   AliCollisionNormalization();
34   AliCollisionNormalization(Int_t nbinz, Float_t minz, Float_t maxz);
35   AliCollisionNormalization(const char * dataFile, const char * dataListName, 
36                             const char * mcFile,   const char * mcListName,
37                             const char * eventStatFile);
38
39   ~AliCollisionNormalization();
40   
41   void SetMC(Bool_t flag = kTRUE) { fIsMC = flag;}
42
43   void BookAllHistos();
44   TH1 * BookVzHisto(const char * name , const char * title, Bool_t vzOnly=kFALSE);
45
46   void FillVzMCGen(Float_t vz, Int_t ntrk, AliMCEvent * mcEvt);      
47   void FillVzMCRec(Float_t vz, Int_t ntrk, AliMCEvent * mcEvt);      
48   void FillVzMCTrg(Float_t vz, Int_t ntrk, AliMCEvent * mcEvt);      
49   void FillVzData (Float_t vz, Int_t ntrk)      {fHistVzData       ->Fill(vz,ntrk);}
50
51   TH2F *   GetVzMCGen       (Int_t procType) ;
52   TH2F *   GetVzMCRec       (Int_t procType) ;
53   TH2F *   GetVzMCTrg       (Int_t procType) ;
54   TH2F *   GetVzData        () { return fHistVzData       ; }
55   TH1F *   GetStatBin0      () { return fHistStatBin0     ; }
56   TH1F *   GetStat          () { return fHistStat         ; }
57   TH1F *   GetHistProcTypes () { return fHistProcTypes    ; }
58    
59
60   Int_t GetProcessType(AliMCEvent * mcEvt) ;
61   Double_t GetProcessWeight(Int_t proctype);
62
63   void SetReferencsXS(Int_t ref) { fReferenceXS = ref;}
64   
65   Double_t ComputeNint();
66   void SetZRange(Float_t zrange) { fZRange = zrange ;}
67
68   void SetReferenceXS(Int_t ref) { fReferenceXS = ref ;}
69   void GetRelativeFractions(Int_t origin, Float_t& ref_SD, Float_t& ref_DD, Float_t& ref_ND, Float_t& error_SD, Float_t& error_DD, Float_t& error_ND);
70
71   void SetVerbose(Int_t lev) { fVerbose = lev ;}
72
73   void SetEnergy(Float_t en) { fEnergy = en; }
74
75   Long64_t Merge(TCollection* list);
76
77 protected:
78
79   Int_t   fNbinsVz; // number of z bins in the vz histo
80   Float_t fMinVz  ; // lowest Z
81   Float_t fMaxVz  ; // highest Z
82
83   Float_t fZRange; // max |Z| vertex to be considered
84
85   Bool_t fIsMC; // True if processing MC
86   
87   Int_t fReferenceXS;                // index of reference cross section to be used to rescale process types in the calculation of the efficiency
88
89   Int_t fVerbose;                    // Determines the ammount of printout
90
91   Float_t fEnergy;                     // Beam energy in GeV. Defaults to 900.
92
93   TH2F * fHistVzMCGen[kNProcs]    ;    // Vz distribution of generated events vs rec multiplicity
94   TH2F * fHistVzMCRec[kNProcs]    ;     // Vz distribution of reconstructed events vs rec multiplicity
95   TH2F * fHistVzMCTrg[kNProcs]    ;     // Vz distribution of triggered events vs rec multiplicity
96   TH2F * fHistVzData              ;     // Vz distribution of triggered events vs rec multiplicity    
97   TH1F * fHistProcTypes           ;    // Number of evts for different Process types 
98
99   TH1F * fHistStatBin0     ; // event stat histogram, created by physiscs selection; used in ComputeNint;
100   TH1F * fHistStat         ; // event stat histogram, created by physiscs selection; used in ComputeNint;
101
102   static const char * fProcLabel[] ; // labels of the different process types
103   
104   ClassDef(AliCollisionNormalization, 3);
105     
106 private:
107   AliCollisionNormalization(const AliCollisionNormalization&);
108   AliCollisionNormalization& operator=(const AliCollisionNormalization&);
109 };
110
111 #endif