adding possibility of switching ON/OFF Eventplane bins for Event Mixing

diff --git a/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskBFPsi.cxx b/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskBFPsi.cxx
index f8489691f3af572e76e02153885c398ed9a164f1..0f0576189c007e73cd007dab681ff28b0bd63817 100755 (executable)
--- a/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskBFPsi.cxx
+++ b/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskBFPsi.cxx
@@ -55,6 +55,7 @@ AliAnalysisTaskBFPsi::AliAnalysisTaskBFPsi(const char *name)
   fRunShuffling(kFALSE),\r
   fShuffledBalance(0),\r
   fRunMixing(kFALSE),\r
+  fRunMixingEventPlane(kFALSE),\r
   fMixingTracks(50000),\r
   fMixedBalance(0),\r
   fPoolMgr(0),\r
@@ -349,25 +350,33 @@ void AliAnalysisTaskBFPsi::UserCreateOutputObjects() {
 \r
 \r
   // Event Mixing\r
-  Int_t trackDepth = fMixingTracks; \r
-  Int_t poolsize   = 1000;  // Maximum number of events, ignored in the present implemented of AliEventPoolManager\r
-   \r
-  // centrality bins\r
-  Double_t centralityBins[] = {0.,1.,2.,3.,4.,5.,6.,7.,8.,9.,10.,15.,20.,25.,30.,35.,40.,45.,50.,55.,60.,65.,70.,75.,80.,90.,100.}; // SHOULD BE DEDUCED FROM CREATED ALITHN!!!\r
-  Double_t* centbins        = centralityBins;\r
-  Int_t nCentralityBins     = sizeof(centralityBins) / sizeof(Double_t) - 1;\r
-  \r
-  // Zvtx bins\r
-  Double_t vertexBins[] = {-10., -7., -5., -3., -1., 1., 3., 5., 7., 10.}; // SHOULD BE DEDUCED FROM CREATED ALITHN!!!\r
-  Double_t* vtxbins     = vertexBins;\r
-  Int_t nVertexBins     = sizeof(vertexBins) / sizeof(Double_t) - 1;\r
-\r
-  // Event plane angle (Psi) bins\r
-  Double_t psiBins[] = {0.,45.,135.,215.,305.,360.}; // SHOULD BE DEDUCED FROM CREATED ALITHN!!!\r
-  Double_t* psibins     = psiBins;\r
-  Int_t nPsiBins     = sizeof(psiBins) / sizeof(Double_t) - 1;\r
-\r
-  fPoolMgr = new AliEventPoolManager(poolsize, trackDepth, nCentralityBins, centbins, nVertexBins, vtxbins, nPsiBins, psibins);\r
+  if(fRunMixing){\r
+    Int_t trackDepth = fMixingTracks; \r
+    Int_t poolsize   = 1000;  // Maximum number of events, ignored in the present implemented of AliEventPoolManager\r
+    \r
+    // centrality bins\r
+    Double_t centralityBins[] = {0.,1.,2.,3.,4.,5.,6.,7.,8.,9.,10.,15.,20.,25.,30.,35.,40.,45.,50.,55.,60.,65.,70.,75.,80.,90.,100.}; // SHOULD BE DEDUCED FROM CREATED ALITHN!!!\r
+    Double_t* centbins        = centralityBins;\r
+    Int_t nCentralityBins     = sizeof(centralityBins) / sizeof(Double_t) - 1;\r
+    \r
+    // Zvtx bins\r
+    Double_t vertexBins[] = {-10., -7., -5., -3., -1., 1., 3., 5., 7., 10.}; // SHOULD BE DEDUCED FROM CREATED ALITHN!!!\r
+    Double_t* vtxbins     = vertexBins;\r
+    Int_t nVertexBins     = sizeof(vertexBins) / sizeof(Double_t) - 1;\r
+    \r
+    // Event plane angle (Psi) bins\r
+    Double_t psiBins[] = {0.,45.,135.,215.,305.,360.}; // SHOULD BE DEDUCED FROM CREATED ALITHN!!!\r
+    Double_t* psibins     = psiBins;\r
+    Int_t nPsiBins     = sizeof(psiBins) / sizeof(Double_t) - 1;\r
+    \r
+    // run the event mixing also in bins of event plane (statistics!)\r
+    if(fRunMixingEventPlane){\r
+      fPoolMgr = new AliEventPoolManager(poolsize, trackDepth, nCentralityBins, centbins, nVertexBins, vtxbins, nPsiBins, psibins);\r
+    }\r
+    else{\r
+      fPoolMgr = new AliEventPoolManager(poolsize, trackDepth, nCentralityBins, centbins, nVertexBins, vtxbins);\r
+    }\r
+  }\r
 \r
   if(fESDtrackCuts) fList->Add(fESDtrackCuts);\r
 \r

diff --git a/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskBFPsi.h b/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskBFPsi.h
index 4b7b134781c18c39e2c0b81cd6be1db8faa0f851..e695e8583c65991934cebb4ac33a90928d2ca9ff 100755 (executable)
--- a/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskBFPsi.h
+++ b/PWGCF/EBYE/BalanceFunctions/AliAnalysisTaskBFPsi.h
@@ -44,6 +44,7 @@ class AliAnalysisTaskBFPsi : public AliAnalysisTaskSE {
     fRunMixing = kTRUE;\r
     fMixedBalance = analysisMixed;\r
   }\r
+  void SetMixingWithEventPlane(Bool_t bMixingWithEventPlane = kTRUE) { fRunMixingEventPlane = bMixingWithEventPlane; }\r
   void SetMixingTracks(Int_t tracks) { fMixingTracks = tracks; }\r
   void SetAnalysisCutObject(AliESDtrackCuts *const trackCuts) {\r
     fESDtrackCuts = trackCuts;}\r
@@ -143,6 +144,7 @@ class AliAnalysisTaskBFPsi : public AliAnalysisTaskSE {
   Bool_t fRunShuffling;//run shuffling or not\r
   AliBalancePsi *fShuffledBalance; //BF object (shuffled)\r
   Bool_t fRunMixing;//run mixing or not\r
+  Bool_t fRunMixingEventPlane;//run mixing with Event Plane\r
   Int_t  fMixingTracks;\r
   AliBalancePsi *fMixedBalance; //TriggeredBF object (mixed)\r
   AliEventPoolManager*     fPoolMgr;         //! event pool manager\r

diff --git a/PWGCF/EBYE/macros/AddTaskBalancePsiCentralityTrain.C b/PWGCF/EBYE/macros/AddTaskBalancePsiCentralityTrain.C
index 2ecb0d09f85193d6dc585b295c1cb641db52f74e..2d33dec0640226801c34406da43300f9c3d90eb1 100644 (file)
--- a/PWGCF/EBYE/macros/AddTaskBalancePsiCentralityTrain.C
+++ b/PWGCF/EBYE/macros/AddTaskBalancePsiCentralityTrain.C
@@ -19,6 +19,7 @@ AliAnalysisTaskBFPsi *AddTaskBalancePsiCentralityTrain(Double_t centrMin=0.,
                                                       Double_t centrMax=100.,\r
                                                       Bool_t gRunShuffling=kFALSE,\r
                                                       Bool_t gRunMixing=kFALSE,\r
+                                                      Bool_t gRunMixingWithEventPlane=kFALSE,\r
                                                       TString centralityEstimator="V0M",\r
                                                       Double_t vertexZ=10.,\r
                                                       Double_t DCAxy=-1,\r
@@ -91,6 +92,9 @@ AliAnalysisTaskBFPsi *AddTaskBalancePsiCentralityTrain(Double_t centrMin=0.,
   if(gRunMixing){\r
     taskBF->SetMixingObject(bfm);\r
     taskBF->SetMixingTracks(50000);\r
+    if(gRunMixingWithEventPlane){\r
+      taskBF->SetMixingWithEventPlane(gRunMixingWithEventPlane);\r
+    }\r
   }\r
 \r
   taskBF->SetCentralityPercentileRange(centrMin,centrMax);\r