From 2493b5d3bd62d43cfa53935f6c692699dac89412 Mon Sep 17 00:00:00 2001
From: prino <prino@f7af4fe6-9843-0410-8265-dc069ae4e863>
Date: Wed, 1 Aug 2012 12:42:44 +0000
Subject: [PATCH] Enable usage of MC vertex also for D+

---
 PWGHF/vertexingHF/AliRDHFCutsDplustoKpipi.cxx | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/PWGHF/vertexingHF/AliRDHFCutsDplustoKpipi.cxx b/PWGHF/vertexingHF/AliRDHFCutsDplustoKpipi.cxx
index 4bf0a5b7a72..e07dd223ca7 100644
--- a/PWGHF/vertexingHF/AliRDHFCutsDplustoKpipi.cxx
+++ b/PWGHF/vertexingHF/AliRDHFCutsDplustoKpipi.cxx
@@ -410,7 +410,7 @@ Int_t AliRDHFCutsDplustoKpipi::IsSelected(TObject* obj,Int_t selectionLevel, Ali
     
     //recalculate vertex w/o daughters
     AliAODVertex *origownvtx=0x0;
-    if(fRemoveDaughtersFromPrimary) {
+    if(fRemoveDaughtersFromPrimary && !fUseMCVertex) {
       if(d->GetOwnPrimaryVtx()) origownvtx=new AliAODVertex(*d->GetOwnPrimaryVtx());
       if(!RecalcOwnPrimaryVtx(d,aod)) {
 	CleanOwnPrimaryVtx(d,aod,origownvtx);
@@ -418,6 +418,14 @@ Int_t AliRDHFCutsDplustoKpipi::IsSelected(TObject* obj,Int_t selectionLevel, Ali
       }
     }
 
+    if(fUseMCVertex) {
+      if(d->GetOwnPrimaryVtx()) origownvtx=new AliAODVertex(*d->GetOwnPrimaryVtx());
+      if(!SetMCPrimaryVtx(d,aod)) {
+	CleanOwnPrimaryVtx(d,aod,origownvtx);
+	return 0;
+      }
+    }
+
     Int_t ptbin=PtBin(pt);
     if (ptbin==-1) {
       CleanOwnPrimaryVtx(d,aod,origownvtx);
-- 
2.39.3