Compatibility with the Root trunk
[u/mrichter/AliRoot.git] / PWGCF / EBYE / BalanceFunctions / AliBalancePsi.h
1 #ifndef ALIBALANCEPSI_H
2 #define ALIBALANCEPSI_H
3 /*  See cxx source for full Copyright notice */
4
5
6 /* $Id: AliBalancePsi.h 54125 2012-01-24 21:07:41Z miweber $ */
7
8 //-------------------------------------------------------------------------
9 //                          Class AliBalancePsi
10 //   This is the class for the Balance Function writ Psi analysis
11 //
12 //    Origin: Panos Christakoglou, Nikhef, Panos.Christakoglou@cern.ch
13 //-------------------------------------------------------------------------
14
15 #include <vector>
16 #include <TObject.h>
17 #include "TString.h"
18
19 #include "AliTHn.h"
20
21 using std::vector;
22
23 #define ANALYSIS_TYPES  7
24 #define MAXIMUM_NUMBER_OF_STEPS 1024
25 #define MAXIMUM_STEPS_IN_PSI 360
26
27 class TH1D;
28 class TH2D;
29 class TH3D;
30
31 const Int_t nTrackVariablesSingle = 2;       // track variables in histogram (centrality, phi-Psi2, eta, phi, pTtrig)
32 const Int_t nTrackVariablesPair   = 5;       // track variables in histogram (centrality, phi-Psi2, dEta, dPhi, pTtrig, ptAssociated)
33 const TString gBFPsiAnalysisType[ANALYSIS_TYPES] = {"y","eta","qlong","qout","qside","qinv","phi"};
34
35 class AliBalancePsi : public TObject {
36  public:
37   enum EAnalysisType {
38     kRapidity = 0, 
39     kEta      = 1, 
40     kQlong    = 2, 
41     kQout     = 3, 
42     kQside    = 4, 
43     kQinv     = 5, 
44     kPhi      = 6 
45   };
46
47   AliBalancePsi();
48   AliBalancePsi(const AliBalancePsi& balance);
49   ~AliBalancePsi();
50
51   void SetCentralityIdentifier(const char* centralityId) {
52     fCentralityId = centralityId;}
53   
54   void SetAnalysisLevel(const char* analysisLevel) {
55     fAnalysisLevel = analysisLevel;}
56   void SetShuffle(Bool_t shuffle) {bShuffle = shuffle;}
57   void SetCentralityInterval(Double_t cStart, Double_t cStop)  { fCentStart = cStart; fCentStop = cStop;};
58
59   void InitHistograms(void);
60
61   const char* GetAnalysisLevel() {return fAnalysisLevel.Data();}
62   Int_t GetNumberOfAnalyzedEvent() {return fAnalyzedEvents;}
63
64   void CalculateBalance(Double_t gReactionPlane, 
65                         vector<Double_t> **chargeVector);
66   
67   TH2D   *GetCorrelationFunctionPN(Double_t psiMin, Double_t psiMax);
68   TH2D   *GetCorrelationFunctionNP(Double_t psiMin, Double_t psiMax);
69   TH2D   *GetCorrelationFunctionPP(Double_t psiMin, Double_t psiMax);
70   TH2D   *GetCorrelationFunctionNN(Double_t psiMin, Double_t psiMax);
71
72   AliTHn *GetHistNp() {return fHistP;}
73   AliTHn *GetHistNn() {return fHistN;}
74   AliTHn *GetHistNpn() {return fHistPN;}
75   AliTHn *GetHistNnp() {return fHistNP;}
76   AliTHn *GetHistNpp() {return fHistPP;}
77   AliTHn *GetHistNnn() {return fHistNN;}
78
79   void SetHistNp(AliTHn *gHist) {
80     fHistP = gHist; }//fHistP->FillParent(); fHistP->DeleteContainers();}
81   void SetHistNn(AliTHn *gHist) {
82     fHistN = gHist; }//fHistN->FillParent(); fHistN->DeleteContainers();}
83   void SetHistNpn(AliTHn *gHist) {
84     fHistPN = gHist; }//fHistPN->FillParent(); fHistPN->DeleteContainers();}
85   void SetHistNnp(AliTHn *gHist) {
86     fHistNP = gHist; }//fHistNP->FillParent(); fHistNP->DeleteContainers();}
87   void SetHistNpp(AliTHn *gHist) {
88     fHistPP = gHist; }//fHistPP->FillParent(); fHistPP->DeleteContainers();}
89   void SetHistNnn(AliTHn *gHist) {
90     fHistNN = gHist; }//fHistNN->FillParent(); fHistNN->DeleteContainers();}
91
92   TH1D *GetBalanceFunctionHistogram(Int_t iVariableSingle,
93                                     Int_t iVariablePair,
94                                     Double_t psiMin, Double_t psiMax);
95
96   TH2D *GetBalanceFunctionDeltaEtaDeltaPhi(Double_t psiMin, Double_t psiMax);
97   
98  private:
99   Bool_t bShuffle; //shuffled balance function object
100   TString fAnalysisLevel; //ESD, AOD or MC
101   Int_t fAnalyzedEvents; //number of events that have been analyzed
102
103   TString fCentralityId;//Centrality identifier to be used for the histo naming
104
105   Double_t fCentStart;
106   Double_t fCentStop;
107
108   AliTHn *fHistP; //N+
109   AliTHn *fHistN; //N-
110   AliTHn *fHistPN; //N+-
111   AliTHn *fHistNP; //N-+
112   AliTHn *fHistPP; //N++
113   AliTHn *fHistNN; //N--
114
115   Double_t fPsiInterval;// interval in Psi-phi1
116
117   AliBalancePsi & operator=(const AliBalancePsi & ) {return *this;}
118
119   ClassDef(AliBalancePsi, 1)
120 };
121
122 #endif