1 #ifndef ALIBALANCEPSI_H
2 #define ALIBALANCEPSI_H
3 /* See cxx source for full Copyright notice */
6 /* $Id: AliBalancePsi.h 54125 2012-01-24 21:07:41Z miweber $ */
8 //-------------------------------------------------------------------------
10 // This is the class for the Balance Function writ Psi analysis
12 // Origin: Panos Christakoglou, Nikhef, Panos.Christakoglou@cern.ch
13 //-------------------------------------------------------------------------
24 #define ANALYSIS_TYPES 7
25 #define MAXIMUM_NUMBER_OF_STEPS 1024
26 #define MAXIMUM_STEPS_IN_PSI 360
32 const Int_t kTrackVariablesSingle = 3; // track variables in histogram (event class, pTtrig, vertexZ)
33 const Int_t kTrackVariablesPair = 6; // track variables in histogram (event class, dEta, dPhi, pTtrig, ptAssociated, vertexZ)
34 const TString gBFPsiAnalysisType[ANALYSIS_TYPES] = {"y","eta","qlong","qout","qside","qinv","phi"};
36 class AliBalancePsi : public TObject {
49 AliBalancePsi(const AliBalancePsi& balance);
52 void SetCentralityIdentifier(const char* centralityId) {
53 fCentralityId = centralityId;}
55 void SetAnalysisLevel(const char* analysisLevel) {
56 fAnalysisLevel = analysisLevel;}
57 void SetShuffle(Bool_t shuffle) {fShuffle = shuffle;}
58 void SetCentralityInterval(Double_t cStart, Double_t cStop) { fCentStart = cStart; fCentStop = cStop;};
59 void SetEventClass(TString receivedEventClass){ fEventClass = receivedEventClass; }
60 void SetDeltaEtaMax(Double_t receivedDeltaEtaMax){ fDeltaEtaMax = receivedDeltaEtaMax; }
61 void SetVertexZBinning(Bool_t receivedVertexBinning=kTRUE){ fVertexBinning = receivedVertexBinning; }
63 void InitHistograms(void);
65 const char* GetAnalysisLevel() {return fAnalysisLevel.Data();}
66 Int_t GetNumberOfAnalyzedEvent() {return fAnalyzedEvents;}
68 void CalculateBalance(Double_t gReactionPlane,
70 TObjArray* particlesMixed,
72 Double_t kMultorCent = -100,
73 Double_t vertexZ = 0);
75 TH2D *GetCorrelationFunction(TString type,
76 Double_t psiMin, Double_t psiMax,
77 Double_t vertexZMin=-1,
78 Double_t vertexZMax=-1,
79 Double_t ptTriggerMin=-1.,
80 Double_t ptTriggerMax=-1.,
81 Double_t ptAssociatedMin=-1.,
82 Double_t ptAssociatedMax=-1,
83 AliBalancePsi *bMixed=NULL);
84 TH2D *GetCorrelationFunctionPN(Double_t psiMin, Double_t psiMax,
85 Double_t vertexZMin=-1,
86 Double_t vertexZMax=-1,
87 Double_t ptTriggerMin=-1.,
88 Double_t ptTriggerMax=-1.,
89 Double_t ptAssociatedMin=-1.,
90 Double_t ptAssociatedMax=-1);
91 TH2D *GetCorrelationFunctionNP(Double_t psiMin, Double_t psiMax,
92 Double_t vertexZMin=-1,
93 Double_t vertexZMax=-1,
94 Double_t ptTriggerMin=-1.,
95 Double_t ptTriggerMax=-1.,
96 Double_t ptAssociatedMin=-1.,
97 Double_t ptAssociatedMax=-1);
98 TH2D *GetCorrelationFunctionPP(Double_t psiMin, Double_t psiMax,
99 Double_t vertexZMin=-1,
100 Double_t vertexZMax=-1,
101 Double_t ptTriggerMin=-1.,
102 Double_t ptTriggerMax=-1.,
103 Double_t ptAssociatedMin=-1.,
104 Double_t ptAssociatedMax=-1);
105 TH2D *GetCorrelationFunctionNN(Double_t psiMin, Double_t psiMax,
106 Double_t vertexZMin=-1,
107 Double_t vertexZMax=-1,
108 Double_t ptTriggerMin=-1.,
109 Double_t ptTriggerMax=-1.,
110 Double_t ptAssociatedMin=-1.,
111 Double_t ptAssociatedMax=-1);
112 TH2D *GetCorrelationFunctionChargeIndependent(Double_t psiMin, Double_t psiMax,
113 Double_t vertexZMin=-1,
114 Double_t vertexZMax=-1,
115 Double_t ptTriggerMin=-1.,
116 Double_t ptTriggerMax=-1.,
117 Double_t ptAssociatedMin=-1.,
118 Double_t ptAssociatedMax=-1);
120 AliTHn *GetHistNp() {return fHistP;}
121 AliTHn *GetHistNn() {return fHistN;}
122 AliTHn *GetHistNpn() {return fHistPN;}
123 AliTHn *GetHistNnp() {return fHistNP;}
124 AliTHn *GetHistNpp() {return fHistPP;}
125 AliTHn *GetHistNnn() {return fHistNN;}
127 void SetHistNp(AliTHn *gHist) {
128 fHistP = gHist; }//fHistP->FillParent(); fHistP->DeleteContainers();}
129 void SetHistNn(AliTHn *gHist) {
130 fHistN = gHist; }//fHistN->FillParent(); fHistN->DeleteContainers();}
131 void SetHistNpn(AliTHn *gHist) {
132 fHistPN = gHist; }//fHistPN->FillParent(); fHistPN->DeleteContainers();}
133 void SetHistNnp(AliTHn *gHist) {
134 fHistNP = gHist; }//fHistNP->FillParent(); fHistNP->DeleteContainers();}
135 void SetHistNpp(AliTHn *gHist) {
136 fHistPP = gHist; }//fHistPP->FillParent(); fHistPP->DeleteContainers();}
137 void SetHistNnn(AliTHn *gHist) {
138 fHistNN = gHist; }//fHistNN->FillParent(); fHistNN->DeleteContainers();}
140 TH1D *GetBalanceFunctionHistogram(Int_t iVariableSingle,
142 Double_t psiMin, Double_t psiMax,
143 Double_t vertexZMin=-1,
144 Double_t vertexZMax=-1,
145 Double_t ptTriggerMin=-1.,
146 Double_t ptTriggerMax=-1.,
147 Double_t ptAssociatedMin=-1.,
148 Double_t ptAssociatedMax=-1); //
150 TH1D *GetBalanceFunctionHistogram2pMethod(Int_t iVariableSingle,
152 Double_t psiMin, Double_t psiMax,
153 Double_t vertexZMin=-1,
154 Double_t vertexZMax=-1,
155 Double_t ptTriggerMin=-1.,
156 Double_t ptTriggerMax=-1.,
157 Double_t ptAssociatedMin=-1.,
158 Double_t ptAssociatedMax=-1,
159 AliBalancePsi *bfMix=NULL);
161 TH2D *GetBalanceFunctionDeltaEtaDeltaPhi(Double_t psiMin, Double_t psiMax,
162 Double_t vertexZMin=-1,
163 Double_t vertexZMax=-1,
164 Double_t ptTriggerMin=-1.,
165 Double_t ptTriggerMax=-1.,
166 Double_t ptAssociatedMin=-1.,
167 Double_t ptAssociatedMax=-1);
169 TH2D *GetBalanceFunctionDeltaEtaDeltaPhi2pMethod(Double_t psiMin, Double_t psiMax,
170 Double_t vertexZMin=-1,
171 Double_t vertexZMax=-1,
172 Double_t ptTriggerMin=-1.,
173 Double_t ptTriggerMax=-1.,
174 Double_t ptAssociatedMin=-1.,
175 Double_t ptAssociatedMax=-1.,
176 AliBalancePsi *bfMix=NULL);
178 TH1D *GetBalanceFunction1DFrom2D2pMethod(Bool_t bPhi,
179 Double_t psiMin, Double_t psiMax,
180 Double_t vertexZMin=-1,
181 Double_t vertexZMax=-1,
182 Double_t ptTriggerMin=-1.,
183 Double_t ptTriggerMax=-1.,
184 Double_t ptAssociatedMin=-1.,
185 Double_t ptAssociatedMax=-1.,
186 AliBalancePsi *bfMix=NULL);
188 Bool_t GetMomentsAnalytical(Int_t fVariable, TH1D* gHist, Bool_t kUseZYAM,
189 Double_t &mean, Double_t &meanError,
190 Double_t &sigma, Double_t &sigmaError,
191 Double_t &skewness, Double_t &skewnessError,
192 Double_t &kurtosis, Double_t &kurtosisError);
194 TH2D *GetQAHistHBTbefore() {return fHistHBTbefore;}
195 TH2D *GetQAHistHBTafter() {return fHistHBTafter;}
196 TH3D *GetQAHistConversionbefore() {return fHistConversionbefore;}
197 TH3D *GetQAHistConversionafter() {return fHistConversionafter;}
198 TH2D *GetQAHistPsiMinusPhi() {return fHistPsiMinusPhi;}
199 TH3D *GetQAHistResonancesBefore() {return fHistResonancesBefore;}
200 TH3D *GetQAHistResonancesRho() {return fHistResonancesRho;}
201 TH3D *GetQAHistResonancesK0() {return fHistResonancesK0;}
202 TH3D *GetQAHistResonancesLambda() {return fHistResonancesLambda;}
203 TH3D *GetQAHistQbefore() {return fHistQbefore;}
204 TH3D *GetQAHistQafter() {return fHistQafter;}
206 void UseResonancesCut() {fResonancesCut = kTRUE;}
207 void UseHBTCut(Double_t setHBTCutValue = 0.02) {
208 fHBTCut = kTRUE; fHBTCutValue = setHBTCutValue;}
209 void UseConversionCut(Double_t setInvMassCutConversion = 0.04) {
210 fConversionCut = kTRUE; fInvMassCutConversion = setInvMassCutConversion; }
211 void UseMomentumDifferenceCut(Double_t gDeltaPtCutMin) {
212 fQCut = kTRUE; fDeltaPtMin = gDeltaPtCutMin;}
215 Float_t GetDPhiStar(Float_t phi1, Float_t pt1, Float_t charge1, Float_t phi2, Float_t pt2, Float_t charge2, Float_t radius, Float_t bSign);
217 Bool_t fShuffle; //shuffled balance function object
218 TString fAnalysisLevel; //ESD, AOD or MC
219 Int_t fAnalyzedEvents; //number of events that have been analyzed
221 TString fCentralityId;//Centrality identifier to be used for the histo naming
228 AliTHn *fHistPN; //N+-
229 AliTHn *fHistNP; //N-+
230 AliTHn *fHistPP; //N++
231 AliTHn *fHistNN; //N--
234 TH2D *fHistHBTbefore; // Delta Eta vs. Delta Phi before HBT inspired cuts
235 TH2D *fHistHBTafter; // Delta Eta vs. Delta Phi after HBT inspired cuts
236 TH3D *fHistConversionbefore; // 3D histogram (Deta,Dphi,Invmass) before Conversion cuts
237 TH3D *fHistConversionafter; // 3D histogram (Deta,Dphi,Invmass) before Conversion cuts
238 TH2D *fHistPsiMinusPhi;// psi - phi QA histogram
239 TH3D *fHistResonancesBefore; // 3D histogram (Deta,Dphi,Invmass) before resonance cuts
240 TH3D *fHistResonancesRho; // 3D histogram (Deta,Dphi,Invmass) after removing rho
241 TH3D *fHistResonancesK0; // 3D histogram (Deta,Dphi,Invmass) after removing rho, K0
242 TH3D *fHistResonancesLambda; // 3D histogram (Deta,Dphi,Invmass) after removing rho, K0, and Lambda
243 TH3D *fHistQbefore; // Delta Eta vs. Delta Phi before cut on momentum difference
244 TH3D *fHistQafter; // Delta Eta vs. Delta Phi after cut on momentum difference
246 Double_t fPsiInterval;// interval in Psi-phi1
247 Double_t fDeltaEtaMax;// maximum delta eta for output THnSparse
249 Bool_t fResonancesCut;//resonances cut
250 Bool_t fHBTCut;//cut for two-track efficiency (like HBT group)
251 Double_t fHBTCutValue;// value for two-track efficiency cut (default = 0.02 from dphicorrelations)
252 Bool_t fConversionCut;//conversion cut
253 Double_t fInvMassCutConversion;//invariant mass for conversion cut
254 Bool_t fQCut;//cut on momentum difference to suppress femtoscopic effect correlations
255 Double_t fDeltaPtMin;//delta pt cut: minimum value
256 Bool_t fVertexBinning;//use vertex z binning in AliTHn
260 AliBalancePsi & operator=(const AliBalancePsi & ) {return *this;}
262 ClassDef(AliBalancePsi, 2)