Commit modifications done to take care of the problems

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

diff --git a/PWG3/hfe/AliHFEsecVtx.cxx b/PWG3/hfe/AliHFEsecVtx.cxx
index 9396dd2601e45538a08cabe7620502c9ca981f8a..092aa6d434565828f9d5c156a75c0cfd6b6d4676 100644 (file)
--- a/PWG3/hfe/AliHFEsecVtx.cxx
+++ b/PWG3/hfe/AliHFEsecVtx.cxx
@@ -214,6 +214,11 @@ void AliHFEsecVtx::Process(AliVTrack *signalTrack){
 
   AliESDtrack *track = dynamic_cast<AliESDtrack *>(signalTrack);
 
+  if(!track){
+    AliDebug(1, "no esd track pointer, return\n");
+    return;
+  }
+
   FillHistos(0,track); // wo any cuts
 
   InitHFEpairs();
@@ -1683,7 +1688,7 @@ void AliHFEsecVtx::FillHistos(Int_t step, const AliESDtrack *track){
   //
   // make container
   //
-  
+
   step = step*7;
 
   AliMCParticle *mctrack = NULL;
@@ -1697,21 +1702,27 @@ void AliHFEsecVtx::FillHistos(Int_t step, const AliESDtrack *track){
 
     Int_t esource=fMCQA->GetElecSource(mcpart);
     if(esource==1) {
+      if(!(dynamic_cast<TH1F *>(fSecVtxList->At(step+1)))) return;
       (dynamic_cast<TH1F *>(fSecVtxList->At(step+1)))->Fill(mcpart->Pt()); //charm
     }
     else if(esource==2 || esource==3) {
+      if(!(dynamic_cast<TH1F *>(fSecVtxList->At(step+2)))) return;
       (dynamic_cast<TH1F *>(fSecVtxList->At(step+2)))->Fill(mcpart->Pt()); //beauty
     }
     else if(esource==4) {
+      if(!(dynamic_cast<TH1F *>(fSecVtxList->At(step+3)))) return;
       (dynamic_cast<TH1F *>(fSecVtxList->At(step+3)))->Fill(mcpart->Pt()); //conversion
     }
     else if(esource==7) {
+      if(!(dynamic_cast<TH1F *>(fSecVtxList->At(step+5)))) return;
       (dynamic_cast<TH1F *>(fSecVtxList->At(step+5)))->Fill(mcpart->Pt()); //contamination
     }
     else if(!(esource<0)) {
+      if(!(dynamic_cast<TH1F *>(fSecVtxList->At(step+4)))) return;
       (dynamic_cast<TH1F *>(fSecVtxList->At(step+4)))->Fill(mcpart->Pt()); //e backgrounds
     }
     else {
+      if(!(dynamic_cast<TH1F *>(fSecVtxList->At(step+6)))) return;
       (dynamic_cast<TH1F *>(fSecVtxList->At(step+6)))->Fill(mcpart->Pt()); //something else?
     }
   }