implemented analysis using 3d information (vtx_z, eta, pt)
[u/mrichter/AliRoot.git] / PWG0 / AliCorrectionMatrix3D.h
CommitLineData
bf21645b 1#ifndef ALICORRECTIONMATRIX3D_H
2#define ALICORRECTIONMATRIX3D_H
3
4/* $Id$ */
5
6// ------------------------------------------------------
7//
8// Class to handle 3d-corrections.
9//
10// ------------------------------------------------------
11
12#include <AliCorrectionMatrix.h>
13
14class TH3F;
15
16class AliCorrectionMatrix3D : public AliCorrectionMatrix
17{
18public:
19 AliCorrectionMatrix3D();
20 AliCorrectionMatrix3D(const AliCorrectionMatrix3D& c);
21 AliCorrectionMatrix3D(const Char_t* name, const Char_t* title,
22 Int_t nBinX=10, Float_t Xmin=0., Float_t Xmax=10.,
23 Int_t nBinY=10, Float_t Ymin=0., Float_t Ymax=10.,
24 Int_t nBinZ=10, Float_t Zmin=0., Float_t Zmax=10.);
25
26 virtual ~AliCorrectionMatrix3D();
27
28 TH3F* GetGeneratedHistogram();
29 TH3F* GetMeasuredHistogram();
30 TH3F* GetCorrectionHistogram();
31
32 void FillMeas(Float_t ax, Float_t ay, Float_t az);
33 void FillGene(Float_t ax, Float_t ay, Float_t az);
34
35 Float_t GetCorrection(Float_t ax, Float_t ay, Float_t az) const;
36
37 void RemoveEdges(Float_t cut=2, Int_t nBinsXedge = 0, Int_t nBinsYedge = 0, Int_t nBinsZedge = 0);
38
39 virtual void SaveHistograms();
bf21645b 40
41protected:
42 ClassDef(AliCorrectionMatrix3D,1)
43};
44
45#endif
46