Event Mixing for Balance Function Psi Task (PhiCorrelation style)

[u/mrichter/AliRoot.git] / PWGCF / EBYE / BalanceFunctions / AliAnalysisTaskBFPsi.h

diff --git a/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskBFPsi.h b/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskBFPsi.h
index 1be5ee9ddb036791d728a3342170bf693dcab033..b1f1e46329f1a0aa5f426438f466daaca842b9dd 100755 (executable)
--- a/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskBFPsi.h
+++ b/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskBFPsi.h
@@ -11,6 +11,8 @@ class TF1;
 \r
 class AliBalancePsi;\r
 class AliESDtrackCuts;\r
+class AliEventPoolManager;\r
+\r
 \r
 #include "AliAnalysisTaskSE.h"\r
 #include "AliBalancePsi.h"\r
@@ -38,6 +40,11 @@ class AliAnalysisTaskBFPsi : public AliAnalysisTaskSE {
     fRunShuffling = kTRUE;\r
     fShuffledBalance = analysisShuffled;\r
   }\r
+  void SetMixingObject(AliBalancePsi *const analysisMixed) {\r
+    fRunMixing = kTRUE;\r
+    fMixedBalance = analysisMixed;\r
+  }\r
+  void SetMixingTracks(Int_t tracks) { fMixingTracks = tracks; }\r
   void SetAnalysisCutObject(AliESDtrackCuts *const trackCuts) {\r
     fESDtrackCuts = trackCuts;}\r
   void SetVertexDiamond(Double_t vx, Double_t vy, Double_t vz) {\r
@@ -130,12 +137,23 @@ class AliAnalysisTaskBFPsi : public AliAnalysisTaskSE {
     fPidDetectorConfig = detConfig;}\r
 \r
  private:\r
+  Double_t    IsEventAccepted(AliVEvent* event);\r
+  Double_t    GetEventPlane(AliVEvent* event);\r
+  TObjArray* GetAcceptedTracks(AliVEvent* event, Double_t fCentrality, Double_t gReactionPlane);\r
+  TObjArray* GetShuffledTracks(TObjArray* tracks);\r
+\r
   AliBalancePsi *fBalance; //BF object\r
   Bool_t fRunShuffling;//run shuffling or not\r
   AliBalancePsi *fShuffledBalance; //BF object (shuffled)\r
+  Bool_t fRunMixing;//run mixing or not\r
+  Int_t  fMixingTracks;\r
+  AliBalancePsi *fMixedBalance; //TriggeredBF object (mixed)\r
+  AliEventPoolManager*     fPoolMgr;         //! event pool manager\r
+\r
   TList *fList; //fList object\r
   TList *fListBF; //fList object\r
   TList *fListBFS; //fList object\r
+  TList *fListBFM; //fList object\r
   TList *fHistListPIDQA;  //! list of histograms\r
 \r
   TH2F *fHistEventStats; //event stats\r
@@ -241,4 +259,6 @@ class AliAnalysisTaskBFPsi : public AliAnalysisTaskSE {
   ClassDef(AliAnalysisTaskBFPsi, 5); // example of analysis\r
 };\r
 \r
+\r
+\r
 #endif\r