end-of-line normalization
[u/mrichter/AliRoot.git] / PWGLF / SPECTRA / AntiprotonToProton / AliProtonSpectraCorrection.h
CommitLineData
a65a7e70 1#ifndef ALIPROTONSPECTRACORRECTION_H
2#define ALIPROTONSPECTRACORRECTION_H
3
4//-------------------------------------------------------------------------
5// Class AliProtonSpectraCorrection
6// This is the class for the absorption corrections used for
7// the baryon (proton) ratio analysis
8//
9// Origin: Panos Christakoglou | Panos.Christakoglou@cern.ch
10//-------------------------------------------------------------------------
11
12#include "TObject.h"
13#include "TH1I.h"
14#include "AliCFContainer.h"
15#include "AliCFManager.h"
16
17class TF1;
18class TH2D;
19class TH1F;
20class TList;
21
22class AliPID;
23class AliCFDataGrid;
24class AliAODEvent;
25class AliAODtrack;
26class AliESDEvent;
27class AliESDtrack;
28class AliExternalTrackParam;
29class AliStack;
30class AliESDVertex;
31class AliProtonAnalysisBase;
32class AliMCEvent;
33
34class AliProtonSpectraCorrection : public TObject {
35 public:
36 enum {
37 kStepGenerated = 0,
38 kStepReconstructible = 1,
39 kStepReconstructed = 2,
40 kStepIdentified = 3,
41 kStepSelected = 4,
42 kNSteps = 5
43 };
44
45 AliProtonSpectraCorrection();
46 virtual ~AliProtonSpectraCorrection();
47
48 void SetBaseAnalysis(AliProtonAnalysisBase * const baseAnalysis) {
49 fProtonAnalysisBase = baseAnalysis;}
50 AliProtonAnalysisBase *GetProtonAnalysisBaseObject() const {
51 return fProtonAnalysisBase;}
52
53 void InitAnalysisHistograms(Int_t nbinsY, Float_t fLowY, Float_t fHighY,
54 Int_t nbinsPt, Float_t fLowPt, Float_t fHighPt);
55 void FillCorrectionMaps(AliESDEvent *fESD,
56 const AliESDVertex *vertex,
57 AliMCEvent *mcEvent);
58 void FillCorrectionMaps(AliAODEvent *fAOD);
59
60 AliCFContainer *GetProtonContainer() const {
61 return fCFManagerProtons->GetParticleContainer();}
62 AliCFContainer *GetAntiProtonContainer() const {
63 return fCFManagerAntiProtons->GetParticleContainer();}
64
65 private:
66 AliProtonSpectraCorrection(const AliProtonSpectraCorrection&); // Not implemented
67 AliProtonSpectraCorrection& operator=(const AliProtonSpectraCorrection&); // Not implemented
68
69 AliProtonAnalysisBase *fProtonAnalysisBase;//base analysis object
70
71 Int_t fNBinsY; //number of bins in y or eta
72 Double_t fMinY, fMaxY; //min & max value of y or eta
73 Int_t fNBinsPt; //number of bins in pT
74 Double_t fMinPt, fMaxPt; //min & max value of pT
75
76 //Analysis containers
77 AliCFManager *fCFManagerProtons; // CF manager protons
78 AliCFManager *fCFManagerAntiProtons; // CF manager antiprotons
79
80 ClassDef(AliProtonSpectraCorrection,1);
81};
82
83#endif
84