1) adding also the change in the AddTask macro for the efficiency correction file

diff --git a/PWGCF/EBYE/macros/AddTaskBalancePsiCentralityTrain.C b/PWGCF/EBYE/macros/AddTaskBalancePsiCentralityTrain.C
index eacc81c0a14c97ded53f996172d6234da1f8ce86..737a0eadca96f83bb81d38439cd8fe1b2242673a 100644 (file)
--- a/PWGCF/EBYE/macros/AddTaskBalancePsiCentralityTrain.C
+++ b/PWGCF/EBYE/macros/AddTaskBalancePsiCentralityTrain.C
@@ -40,7 +40,8 @@ AliAnalysisTaskBFPsi *AddTaskBalancePsiCentralityTrain(Double_t centrMin=0.,
                                                       TString fileNameBase="AnalysisResults",\r
                                                       TString fArgEventClass="Centrality",\r
                                                       TString analysisTypeUser="AOD",\r
-                                                      Bool_t bVertexBinning=kTRUE) {\r
+                                                      Bool_t bVertexBinning=kTRUE,\r
+                                                      TString correctionFileName = "$ALICE_ROOT/PWGCF/EBYE/BalanceFunctions/Corrections/CorrectionMaps.root") {\r
   // Creates a balance function analysis task and adds it to the analysis manager.\r
   // Get the pointer to the existing analysis manager via the static access method.\r
   TString outputFileName(fileNameBase);\r
@@ -111,15 +112,9 @@ AliAnalysisTaskBFPsi *AddTaskBalancePsiCentralityTrain(Double_t centrMin=0.,
   taskBF->SetEventClass(fArgEventClass);\r
  \r
   //++++++++++++++++++++++\r
-  TString correctionFileName = "$ALICE_ROOT/PWGCF/EBYE/BalanceFunctions/Corrections/CorrectionMaps.root"; //to put the path for the correction maps\r
-  TFile *fCorrectionMatrix  = TFile::Open(correctionFileName.Data());\r
-  if(!fCorrectionMatrix){\r
-    Printf("WARNING CORRECTION histogram file not found");\r
-    return NULL;\r
-  }\r
-  \r
+  // Efficiency + Contamination corrections\r
+  // If correctionFileName = "", do not use corrections\r
   taskBF->SetInputCorrection(correctionFileName.Data(),"");\r
-\r
   //+++++++++++++++++++++\r
 \r
   taskBF->SetAnalysisObject(bf);\r