PID AOD additions (Alis)

[u/mrichter/AliRoot.git] / PWGCF / EBYE / macros / AddTaskBalancePsiCentralityTrain.C

diff --git a/PWGCF/EBYE/macros/AddTaskBalancePsiCentralityTrain.C b/PWGCF/EBYE/macros/AddTaskBalancePsiCentralityTrain.C
index 0562c2c356c6b8b55b55ab13f1159a5f151d76a7..ef1364ed056a03e9f3746894add184c871468a94 100644 (file)
--- a/PWGCF/EBYE/macros/AddTaskBalancePsiCentralityTrain.C
+++ b/PWGCF/EBYE/macros/AddTaskBalancePsiCentralityTrain.C
@@ -9,9 +9,9 @@
 //=============================================//\r
 \r
 //PID config\r
-Bool_t kUseNSigmaPID = kFALSE;\r
+Bool_t kUseNSigmaPID = kTRUE;\r
 Double_t nSigmaMax = 3.0;\r
-Bool_t kUseBayesianPID = kTRUE;\r
+Bool_t kUseBayesianPID = kFALSE;\r
 Double_t gMinAcceptedProbability = 0.7;\r
 \r
 //_________________________________________________________//\r
@@ -30,7 +30,7 @@ AliAnalysisTaskBFPsi *AddTaskBalancePsiCentralityTrain(Double_t centrMin=0.,
                                                       Double_t etaMax=0.8,\r
                                                       Double_t maxTPCchi2 = -1, \r
                                                       Int_t minNClustersTPC = -1,\r
-                                                      Bool_t kUsePID = kFALSE,\r
+                                                      Bool_t kUsePID = kTRUE,\r
                                                       Bool_t bResonancesCut = kTRUE,\r
                                                       Bool_t bHBTcut = kTRUE,\r
                                                       Double_t HBTCutValue = 0.02,\r
@@ -123,6 +123,9 @@ AliAnalysisTaskBFPsi *AddTaskBalancePsiCentralityTrain(Double_t centrMin=0.,
   \r
   //Event characteristics scheme\r
   taskBF->SetEventClass(fArgEventClass);\r
+  //taskBF->SetCustomBinning("centralityVertex:0,80");\r
+  //taskBF->SetCustomBinning("multiplicity:0,260");\r
+  \r
   if(fArgEventClass == "Multiplicity") {\r
     taskBF->SetMultiplicityRange(centrMin,centrMax);\r
     taskBF->SetMultiplicityEstimator(centralityEstimator);\r
@@ -165,7 +168,7 @@ AliAnalysisTaskBFPsi *AddTaskBalancePsiCentralityTrain(Double_t centrMin=0.,
        taskBF->SetUseBayesianPID(gMinAcceptedProbability);\r
       else if(kUseNSigmaPID)\r
        taskBF->SetUseNSigmaPID(nSigmaMax);\r
-      taskBF->SetParticleOfInterest(AliAnalysistaskBFPsi::kProton);\r
+      taskBF->SetParticleOfInterest(AliAnalysisTaskBFPsi::kPion);\r
       taskBF->SetDetectorUsedForPID(AliAnalysisTaskBFPsi::kTOFpid);\r
     }\r
   }\r
@@ -185,6 +188,18 @@ AliAnalysisTaskBFPsi *AddTaskBalancePsiCentralityTrain(Double_t centrMin=0.,
       if(electronExclusiveRejection) taskBF->SetElectronOnlyRejection(sigmaElectronRejection); // no other particle in nsigma \r
       else                           taskBF->SetElectronRejection(sigmaElectronRejection); // check only if electrons in nsigma\r
     }\r
+\r
+    //++++++++++++++++//\r
+    if(kUsePID) {\r
+      if(kUseBayesianPID)\r
+       taskBF->SetUseBayesianPID(gMinAcceptedProbability);\r
+      else if(kUseNSigmaPID)\r
+       taskBF->SetUseNSigmaPID(nSigmaMax);\r
+      taskBF->SetParticleOfInterest(AliAnalysisTaskBFPsi::kPion);\r
+      taskBF->SetDetectorUsedForPID(AliAnalysisTaskBFPsi::kTPCTOF); //TOFpid,TPCpid\r
+    }\r
+    //++++++++++++++++//\r
+\r
   }\r
   else if(analysisType == "MC") {\r
     taskBF->SetKinematicsCutsAOD(ptMin,ptMax,etaMin,etaMax);\r
@@ -246,7 +261,9 @@ AliAnalysisTaskBFPsi *AddTaskBalancePsiCentralityTrain(Double_t centrMin=0.,
   mgr->ConnectOutput(taskBF, 2, coutBF);\r
   if(gRunShuffling) mgr->ConnectOutput(taskBF, 3, coutBFS);\r
   if(gRunMixing) mgr->ConnectOutput(taskBF, 4, coutBFM);\r
-  if((kUsePID && analysisType == "ESD")||sigmaElectronRejection > 0) mgr->ConnectOutput(taskBF, 5, coutQAPID);\r
+  if(kUsePID||sigmaElectronRejection > 0) mgr->ConnectOutput(taskBF, 5, coutQAPID);\r
+  //if((kUsePID && analysisType == "AOD")||sigmaElectronRejection > 0) mgr->ConnectOutput(taskBF, 5, coutQAPID);\r
+  //if((kUsePID && analysisType == "ESD")||sigmaElectronRejection > 0) mgr->ConnectOutput(taskBF, 5, coutQAPID);\r
 \r
   return taskBF;\r
 }\r