Possibility to apply a cut on kaon PID in the 3 prong filtering
[u/mrichter/AliRoot.git] / PWGHF / vertexingHF / AliAnalysisTaskSEVertexingHF.cxx
index 774a6b6..90b4066 100644 (file)
@@ -41,6 +41,8 @@
 #include "AliESDUtils.h"
 #include "AliAODHFUtil.h"
 
+#include "AliInputEventHandler.h"
+#include "AliPIDResponse.h"
 
 ClassImp(AliAnalysisTaskSEVertexingHF)
 
@@ -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,