PWG0/dNdEta/dNdEtaAnalysis.h

   1 /* $Id$ */
   2
   3 #ifndef DNDETANALYSIS_H
   4 #define DNDETANALYSIS_H
   5
   6 // ------------------------------------------------------
   7 //
   8 // Class for dn/deta analysis
   9 //
  10 // ------------------------------------------------------
  11 //
  12 // TODO:
  13 // - more documentation
  14 // - add debug statements
  15 // - add more histograms
  16 // - add functionality to set the bin sizes
  17 // - figure out correct way to treat the errors
  18 // - add functionality to make dn/deta for different mult classes?
  19
  20 #include <TNamed.h>
  21
  22 class TH3F;
  23 class TH1D;
  24 class TCollection;
  25 class AlidNdEtaCorrection;
  26
  27 class dNdEtaAnalysis : public TNamed
  28 {
  29 public:
  30   enum { kVertexBinning = 1+3 }; // the first is for the whole vertex range, the others divide the vertex range
  31
  32   dNdEtaAnalysis();
  33   dNdEtaAnalysis(Char_t* name, Char_t* title);
  34   virtual ~dNdEtaAnalysis();
  35
  36   dNdEtaAnalysis(const dNdEtaAnalysis &c);
  37   dNdEtaAnalysis &operator=(const dNdEtaAnalysis &c);
  38   virtual void Copy(TObject &c) const;
  39
  40   void FillTrack(Float_t vtx, Float_t eta, Float_t pt, Float_t weight);
  41   void FillEvent(Float_t vtx, Float_t weight);
  42
  43   void Finish(AlidNdEtaCorrection* correction, Float_t ptCut);
  44
  45   void DrawHistograms();
  46   void LoadHistograms();
  47   void SaveHistograms();
  48
  49   virtual Long64_t Merge(TCollection* list);
  50
  51   TH3F* GetHistogram() { return fData; }
  52   TH3F* GetUncorrectedHistogram() { return fDataUncorrected; }
  53   TH1D* GetVtxHistogram() { return fVtx; }
  54   TH1D* GetdNdEtaHistogram(Int_t i = 0) { return fdNdEta[i]; }
  55
  56 protected:
  57   TH3F* fData;              // histogram Eta vs vtx (track count)
  58   TH3F* fDataUncorrected;   // uncorrected histograms Eta vs vtx (track count)
  59
  60   TH1D* fVtx;                   // vtx histogram (event count)
  61
  62   TH1D* fPtDist; // pt distribution
  63
  64   TH1D* fdNdEta[kVertexBinning]; // dndeta results for different vertex bins (0 = full range)
  65   TH1D* fdNdEtaPtCutOffCorrected[kVertexBinning];  // dndeta results for different vertex bins (0 = full range), pt cut off corrected
  66
  67   ClassDef(dNdEtaAnalysis, 1)
  68 };
  69
  70 #endif