Possibility to apply a cut on kaon PID in the 3 prong filtering
[u/mrichter/AliRoot.git] / PWGHF / vertexingHF / AliAnalysisTaskSEVertexingHF.cxx
index 9c43b0c..90b4066 100644 (file)
@@ -41,6 +41,8 @@
 #include "AliESDUtils.h"
 #include "AliAODHFUtil.h"
 
+#include "AliInputEventHandler.h"
+#include "AliPIDResponse.h"
 
 ClassImp(AliAnalysisTaskSEVertexingHF)
 
@@ -108,8 +110,8 @@ void AliAnalysisTaskSEVertexingHF::Init()
   if(fDebug > 1) printf("AnalysisTaskSEVertexingHF::Init() \n");
 
   if(gROOT->LoadMacro("ConfigVertexingHF.C")) {
-    printf("AnalysisTaskSEVertexingHF::Init() \n Using $ALICE_ROOT/PWG3/vertexingHF/ConfigVertexingHF.C\n");
-    gROOT->LoadMacro("$ALICE_ROOT/PWG3/vertexingHF/ConfigVertexingHF.C");
+    printf("AnalysisTaskSEVertexingHF::Init() \n Using $ALICE_ROOT/PWGHF/vertexingHF/ConfigVertexingHF.C\n");
+    gROOT->LoadMacro("$ALICE_ROOT/PWGHF/vertexingHF/ConfigVertexingHF.C");
   }
 
   fVHF = (AliAnalysisVertexingHF*)gROOT->ProcessLine("ConfigVertexingHF()");  
@@ -227,6 +229,11 @@ void AliAnalysisTaskSEVertexingHF::UserExec(Option_t */*option*/)
      if (aodhandler) aodhandler->SetFillExtension(kTRUE);
   }   
 
+  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
+  AliInputEventHandler *inputHandler=(AliInputEventHandler*)mgr->GetInputEventHandler();
+  AliPIDResponse *pidResp=inputHandler->GetPIDResponse();
+  fVHF->SetPidResponse(pidResp);
+
   // heavy flavor vertexing
   fVHF->FindCandidates(event,
                       fVerticesHFTClArr,