add HBT like cut in Balance Function analysis (default: OFF)

diff --git a/PWGCF/EBYE/BalanceFunctions/AliBalance.cxx b/PWGCF/EBYE/BalanceFunctions/AliBalance.cxx
index 30b65752f85a799ea8fa9af0d9033f9c5fd08762..611ed65390694f3fc32c72550c3e025c64343c79 100644 (file)
--- a/PWGCF/EBYE/BalanceFunctions/AliBalance.cxx
+++ b/PWGCF/EBYE/BalanceFunctions/AliBalance.cxx
@@ -43,6 +43,7 @@ ClassImp(AliBalance)
 AliBalance::AliBalance() :
   TObject(), 
   bShuffle(kFALSE),
+  bHBTcut(kFALSE),
   fAnalysisLevel("ESD"),
   fAnalyzedEvents(0) ,
   fCentralityId(0) ,
@@ -96,7 +97,9 @@ AliBalance::AliBalance() :
 
 //____________________________________________________________________//
 AliBalance::AliBalance(const AliBalance& balance):
-  TObject(balance), bShuffle(balance.bShuffle), 
+  TObject(balance), 
+  bShuffle(balance.bShuffle),
+  bHBTcut(balance.bHBTcut), 
   fAnalysisLevel(balance.fAnalysisLevel),
   fAnalyzedEvents(balance.fAnalyzedEvents), 
   fCentralityId(balance.fCentralityId),
@@ -388,6 +391,13 @@ void AliBalance::CalculateBalance(Float_t fCentrality,vector<Double_t> **chargeV
        dphi = TMath::Abs(phi1 - phi2);
        if(dphi>180) dphi = 360 - dphi;  //dphi should be between 0 and 180!
 
+       // HBT like cut
+       if(bHBTcut){
+         if( dphi < 3 || deta < 0.01 ){
+           continue;
+         }
+       }
+
        //0:y - 1:eta - 2:Qlong - 3:Qout - 4:Qside - 5:Qinv - 6:phi
        if((rap1 >= fP1Start[kRapidity]) && (rap1 <= fP1Stop[kRapidity]) && (rap2 >= fP1Start[kRapidity]) && (rap2 <= fP1Stop[kRapidity])) {
 

diff --git a/PWGCF/EBYE/BalanceFunctions/AliBalance.h b/PWGCF/EBYE/BalanceFunctions/AliBalance.h
index 679ea536ced3c53776591f9a1b332c26147807a1..54a7ed8cd5b8926cc7ba77838ff9b252d1a56a73 100644 (file)
--- a/PWGCF/EBYE/BalanceFunctions/AliBalance.h
+++ b/PWGCF/EBYE/BalanceFunctions/AliBalance.h
@@ -46,6 +46,7 @@ class AliBalance : public TObject {
   void SetAnalysisLevel(const char* analysisLevel) {
     fAnalysisLevel = analysisLevel;}
   void SetShuffle(Bool_t shuffle) {bShuffle = shuffle;}
+  void SetHBTcut(Bool_t HBTcut) {bHBTcut = HBTcut;}
   void SetInterval(Int_t iAnalysisType, Double_t p1Start, Double_t p1Stop,
                   Int_t ibins, Double_t p2Start, Double_t p2Stop);
   void SetCentralityInterval(Double_t cStart, Double_t cStop)  { fCentStart = cStart; fCentStop = cStop;};
@@ -110,7 +111,9 @@ class AliBalance : public TObject {
   void PrintResults(Int_t iAnalysisType, TH1D *gHist);
 
  private:
-  Bool_t bShuffle; //shuffled balance function object
+  Bool_t bShuffle; // shuffled balance function object
+  Bool_t bHBTcut;  // apply HBT like cuts
+
   TString fAnalysisLevel; //ESD, AOD or MC
   Int_t fAnalyzedEvents; //number of events that have been analyzed
 

diff --git a/PWGCF/EBYE/macros/AddTaskBalanceCentralityTrain.C b/PWGCF/EBYE/macros/AddTaskBalanceCentralityTrain.C
index 1bc91d0ac3de6353f299d95fcbe38c1a6d8822a5..6a0fb7e9dcda641f97e54865d9fb38a2233d8315 100644 (file)
--- a/PWGCF/EBYE/macros/AddTaskBalanceCentralityTrain.C
+++ b/PWGCF/EBYE/macros/AddTaskBalanceCentralityTrain.C
@@ -31,6 +31,7 @@ AliAnalysisTaskBF *AddTaskBalanceCentralityTrain(Double_t centrMin=0.,
                                                 Bool_t kUsePID = kFALSE,\r
                                                 Int_t AODfilterBit = 128,\r
                                                 Bool_t bCentralTrigger = kFALSE,\r
+                                                Bool_t bHBTcut = kFALSE,\r
                                                 TString fileNameBase="AnalysisResults") {\r
 \r
   // Creates a balance function analysis task and adds it to the analysis manager.\r
@@ -62,6 +63,7 @@ AliAnalysisTaskBF *AddTaskBalanceCentralityTrain(Double_t centrMin=0.,
   centralityName+="_Bit";\r
   centralityName+=Form("%d",AODfilterBit);\r
   if(bCentralTrigger)   centralityName+="_withCentralTrigger";\r
+  if(bHBTcut)   centralityName+="_withHBTcut";\r
 \r
 \r
 \r
@@ -93,16 +95,16 @@ AliAnalysisTaskBF *AddTaskBalanceCentralityTrain(Double_t centrMin=0.,
   AliBalance *bfs = 0;  // shuffled Balance function object\r
 \r
   if (analysisType=="ESD"){\r
-    bf  = GetBalanceFunctionObject("ESD",centralityEstimator,centrMin,centrMax);\r
-    if(gRunShuffling) bfs = GetBalanceFunctionObject("ESD",centralityEstimator,centrMin,centrMax,kTRUE);\r
+    bf  = GetBalanceFunctionObject("ESD",centralityEstimator,centrMin,centrMax,kFALSE,bHBTcut);\r
+    if(gRunShuffling) bfs = GetBalanceFunctionObject("ESD",centralityEstimator,centrMin,centrMax,kTRUE,bHBTcut);\r
   }\r
   else if (analysisType=="AOD"){\r
-    bf  = GetBalanceFunctionObject("AOD",centralityEstimator,centrMin,centrMax);\r
-    if(gRunShuffling) bfs = GetBalanceFunctionObject("AOD",centralityEstimator,centrMin,centrMax,kTRUE);\r
+    bf  = GetBalanceFunctionObject("AOD",centralityEstimator,centrMin,centrMax,kFALSE,bHBTcut);\r
+    if(gRunShuffling) bfs = GetBalanceFunctionObject("AOD",centralityEstimator,centrMin,centrMax,kTRUE,bHBTcut);\r
   }\r
   else if (analysisType=="MC"){\r
-    bf  = GetBalanceFunctionObject("MC",centralityEstimator,centrMin,centrMax);\r
-    if(gRunShuffling) bfs = GetBalanceFunctionObject("MC",centralityEstimator,centrMin,centrMax,kTRUE);\r
+    bf  = GetBalanceFunctionObject("MC",centralityEstimator,centrMin,centrMax,kFALSE,bHBTcut);\r
+    if(gRunShuffling) bfs = GetBalanceFunctionObject("MC",centralityEstimator,centrMin,centrMax,kTRUE,bHBTcut);\r
   }\r
   else{\r
     ::Error("AddTaskBF", "analysis type NOT known.");\r

diff --git a/PWGCF/EBYE/macros/configBalanceFunctionAnalysis.C b/PWGCF/EBYE/macros/configBalanceFunctionAnalysis.C
index ee4850cf13dd834aeee82663889fcf05d6de8bec..f1f5d3759b6683408e6fa86ea544a962fbb3b534 100644 (file)
--- a/PWGCF/EBYE/macros/configBalanceFunctionAnalysis.C
+++ b/PWGCF/EBYE/macros/configBalanceFunctionAnalysis.C
@@ -3,11 +3,13 @@ AliBalance *GetBalanceFunctionObject(const char* analysisLevel = "ESD",
                                     const char* centralityName = 0x0,\r
                                     Double_t centrMin = 0.,\r
                                     Double_t centrMax = 100.,\r
-                                    Bool_t bShuffle = kFALSE) {\r
+                                    Bool_t bShuffle = kFALSE,\r
+                                    Bool_t bHBTcut = kFALSE) {\r
   //Function to setup the AliBalance object and return it\r
   AliBalance *gBalance = new AliBalance();\r
   gBalance->SetAnalysisLevel(analysisLevel);\r
   gBalance->SetShuffle(bShuffle);\r
+  gBalance->SetHBTcut(bHBTcut);\r
   if(centralityName) gBalance->SetCentralityIdentifier(centralityName);\r
   gBalance->SetCentralityInterval(centrMin,centrMax);\r
 \r