Merge branch 'master' of https://git.cern.ch/reps/AliRoot
[u/mrichter/AliRoot.git] / PWGCF / EBYE / BalanceFunctions / AliBalancePsi.h
index 4d4ae0f03be9dc45c21e8fa15f9d424fc5b7ed04..f81b5e42db9978572ec25d0641209203d23380bc 100644 (file)
@@ -59,6 +59,7 @@ class AliBalancePsi : public TObject {
   void SetEventClass(TString receivedEventClass){ fEventClass = receivedEventClass; } 
   void SetDeltaEtaMax(Double_t receivedDeltaEtaMax){ fDeltaEtaMax = receivedDeltaEtaMax; }
   void SetVertexZBinning(Bool_t receivedVertexBinning=kTRUE){ fVertexBinning = receivedVertexBinning; }
+  void SetCustomBinning(TString receivedCustomBinning) { fCustomBinning = receivedCustomBinning; }
 
   void InitHistograms(void);
 
@@ -211,8 +212,13 @@ class AliBalancePsi : public TObject {
   void UseMomentumDifferenceCut(Double_t gDeltaPtCutMin) {
     fQCut = kTRUE; fDeltaPtMin = gDeltaPtCutMin;}
 
+  // related to customized binning of output AliTHn
+  Bool_t    IsUseVertexBinning() { return fVertexBinning; }
+  TString   GetBinningString()   { return fBinningString; }
+  Double_t* GetBinning(const char* configuration, const char* tag, Int_t& nBins);
+
  private:
-  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); 
+  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); 
 
   Bool_t fShuffle; //shuffled balance function object
   TString fAnalysisLevel; //ESD, AOD or MC
@@ -254,6 +260,8 @@ class AliBalancePsi : public TObject {
   Bool_t fQCut;//cut on momentum difference to suppress femtoscopic effect correlations
   Double_t fDeltaPtMin;//delta pt cut: minimum value
   Bool_t fVertexBinning;//use vertex z binning in AliTHn
+  TString fCustomBinning;//for setting customized binning
+  TString fBinningString;//final binning string
 
   TString fEventClass;