Completed changes needed because of previous commit

[u/mrichter/AliRoot.git] / PWGDQ / dielectron / AliAnalysisTaskDielectronFilter.cxx

diff --git a/PWGDQ/dielectron/AliAnalysisTaskDielectronFilter.cxx b/PWGDQ/dielectron/AliAnalysisTaskDielectronFilter.cxx
index 8d9b7ab2b0efa5250392dc2a74f7d4f14259f73c..59cd123d7775032b74edb82fb7fd12c4e33616e9 100644 (file)
--- a/PWGDQ/dielectron/AliAnalysisTaskDielectronFilter.cxx
+++ b/PWGDQ/dielectron/AliAnalysisTaskDielectronFilter.cxx
@@ -355,7 +355,9 @@ void AliAnalysisTaskDielectronFilter::UserExec(Option_t *)
          nanoEv->GetCaloCluster(posCaloCls)->AddTrackMatched((AliAODTrack*)nanoEv->GetTrack(posit)); 
          }
          // set references for vtx
-         nanoEv->GetTrack(posit)->SetProdVertex(nanoEv->GetVertex(posVtx));
+         AliAODTrack * trk = dynamic_cast<AliAODTrack*>(nanoEv->GetTrack(posit));
+         if(!trk) AliFatal("Not a standard AOD");
+         trk->SetProdVertex(nanoEv->GetVertex(posVtx));
          }
 
          for(int kj=0; kj<(fDielectron->GetTrackArray(1))->GetEntries(); kj++){
@@ -375,7 +377,9 @@ void AliAnalysisTaskDielectronFilter::UserExec(Option_t *)
          for(int u=0; u<clCls->GetNTracksMatched(); u++) clCls->RemoveTrackMatched(clCls->GetTrackMatched(u));
          nanoEv->GetCaloCluster(negCaloCls)->AddTrackMatched((AliAODTrack*)nanoEv->GetTrack(negat));
          }
-         nanoEv->GetTrack(negat)->SetProdVertex(nanoEv->GetVertex(negVtx)); 
+         AliAODTrack * trk = dynamic_cast<AliAODTrack*>(nanoEv->GetTrack(negat));
+         if(!trk) AliFatal("Not a standard AOD");
+         trk->SetProdVertex(nanoEv->GetVertex(negVtx)); 
          }  
         delete tmp; delete tmpSpd; 
         nanoEv->GetTracks()->Expand(nTracks);