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