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