Fix for leak

[u/mrichter/AliRoot.git] / PWGHF / vertexingHF / AliAnalysisTaskCombinHF.cxx

diff --git a/PWGHF/vertexingHF/AliAnalysisTaskCombinHF.cxx b/PWGHF/vertexingHF/AliAnalysisTaskCombinHF.cxx
index 6173bde3e147038346e75382bfcbae538cbe1094..1c3a85fc584faa976a006fcdd2439892816d1074 100644 (file)
--- a/PWGHF/vertexingHF/AliAnalysisTaskCombinHF.cxx
+++ b/PWGHF/vertexingHF/AliAnalysisTaskCombinHF.cxx
@@ -410,9 +410,9 @@ void AliAnalysisTaskCombinHF::UserExec(Option_t */*option*/){
   // build the combinatorics
   Int_t nSelected=0;
   Int_t nFiltered=0;
-
-  AliAODVertex* v2=new AliAODVertex(); 
-  AliAODVertex* v3=new AliAODVertex();
+  Double_t dummypos[3]={0.,0.,0.};
+  AliAODVertex* v2=new AliAODVertex(dummypos,999.,-1,2);
+  AliAODVertex* v3=new AliAODVertex(dummypos,999.,-1,3);
   // dummy values of track impact parameter, needed to build an AliAODRecoDecay object
   Double_t d02[2]={0.,0.};
   Double_t d03[3]={0.,0.,0.};
@@ -594,7 +594,12 @@ Bool_t AliAnalysisTaskCombinHF::FillHistos(Int_t pdgD,Int_t nProngs, AliAODRecoD
            Int_t pdgCode = TMath::Abs( part->GetPdgCode() );
            if(pdgCode==321){
              AliAODMCParticle* dmes =  dynamic_cast<AliAODMCParticle*>(arrayMC->At(labD));
-             if(dmes) fPtVsYReco->Fill(dmes->Pt(),dmes->Y());
+             if(dmes){
+               Int_t orig=AliVertexingHFUtils::CheckOrigin(arrayMC,dmes,fGoUpToQuark);
+               if((fPromptFeeddown==kFeeddown && orig==5)|| (fPromptFeeddown==kPrompt && orig==4) || (fPromptFeeddown==kBoth && orig>=4)) {
+                 fPtVsYReco->Fill(dmes->Pt(),dmes->Y());
+               }
+             }
              fMassVsPtVsYSig->Fill(mass,pt,rapid);
            }else{
              fMassVsPtVsYRefl->Fill(mass,pt,rapid);