small fix
[u/mrichter/AliRoot.git] / PWG0 / dNdEta / dNdEtaAnalysis.h
CommitLineData
539b6cb4 1#ifndef DNDETANALYSIS_H
2#define DNDETANALYSIS_H
3
75ec0f41 4// ------------------------------------------------------
5//
6// Class for dn/deta analysis
7//
8// ------------------------------------------------------
9//
10// TODO:
11// - more documentation
12// - add debug statements
13// - add more histograms
14// - add functionality to set the bin sizes
15// - figure out correct way to treat the errors
16// - add functionality to make dn/deta for different mult classes?
17
ceb5d1b5 18#include <TObject.h>
19#include <TString.h>
75ec0f41 20
ceb5d1b5 21class TH2F;
22class TH1D;
75ec0f41 23
ceb5d1b5 24class dNdEtaAnalysis : public TObject
25{
75ec0f41 26public:
27 dNdEtaAnalysis(Char_t* name="dndeta_correction");
28
29 void FillTrack(Float_t vtx, Float_t eta, Float_t weight);
30 void FillEvent(Float_t vtx);
ceb5d1b5 31
75ec0f41 32 void Finish();
ceb5d1b5 33
34 void DrawHistograms();
75ec0f41 35 void SaveHistograms();
ceb5d1b5 36
37 TH2F* GetEtaVsVtxHistogram() { return hEtaVsVtx; }
38 TH2F* GetEtaVsVtxUncorrectedHistogram() { return hEtaVsVtxUncorrected; }
39 TH1D* GetVtxHistogram() { return hVtx; }
40 TH1D* GetdNdEtaHistogram() { return hdNdEta; }
41
42 void SetEtaVsVtxHistogram(TH2F* aHist) { hEtaVsVtx = aHist; }
43 void SetEtaVsVtxUncorrectedHistogram(TH2F* aHist) { hEtaVsVtxUncorrected = aHist; }
44 void SetVtxHistogram(TH1D* aHist) { hVtx = aHist; }
45 void SetdNdEtaHistogram(TH1D* aHist) { hdNdEta = aHist; }
46
47protected:
48 TString fName;
49
50 TH2F* hEtaVsVtx;
51 TH2F* hEtaVsVtxUncorrected;
52 TH1D* hVtx;
53 TH1D* hdNdEta;
54
75ec0f41 55 ClassDef(dNdEtaAnalysis,0)
56};
57
539b6cb4 58#endif