Corrections for Coverity warnings

[u/mrichter/AliRoot.git] / PWG3 / hfe / AliHFEtools.cxx

diff --git a/PWG3/hfe/AliHFEtools.cxx b/PWG3/hfe/AliHFEtools.cxx
index b6c573da9f30ffd23a0439ed33c3599830a02430..ef85a5c8dd990270fa822528e67250451e9f1a29 100644 (file)
--- a/PWG3/hfe/AliHFEtools.cxx
+++ b/PWG3/hfe/AliHFEtools.cxx
@@ -257,10 +257,10 @@ Int_t AliHFEtools::GetPdg(const AliVParticle *track){
   Int_t pdg = 0;
   if(!TString(track->IsA()->GetName()).CompareTo("AliMCParticle")){
     const AliMCParticle *mctrack = dynamic_cast<const AliMCParticle *>(track);
-    pdg = mctrack->Particle()->GetPdgCode();
+    pdg = mctrack ? mctrack->Particle()->GetPdgCode() : 0;
   } else if(!TString(track->IsA()->GetName()).CompareTo("AliAODMCParticle")){
     const AliAODMCParticle *aodmctrack = dynamic_cast<const AliAODMCParticle *>(track);
-    pdg = aodmctrack->GetPdgCode();
+    pdg = aodmctrack ? aodmctrack->GetPdgCode() : 0;
   }
   return pdg;
 }