Completed changes needed because of previous commit

[u/mrichter/AliRoot.git] / PWG / muon / AliAnalysisTaskMuonDistributions.cxx

diff --git a/PWG/muon/AliAnalysisTaskMuonDistributions.cxx b/PWG/muon/AliAnalysisTaskMuonDistributions.cxx
index 63812473a48b3145448f20dd6900bd56dc8fda61..d021a19fb1e49c6a55704a2af2e3414e165545a3 100755 (executable)
--- a/PWG/muon/AliAnalysisTaskMuonDistributions.cxx
+++ b/PWG/muon/AliAnalysisTaskMuonDistributions.cxx
@@ -228,7 +228,8 @@ void AliAnalysisTaskMuonDistributions::UserExec(Option_t *)
       ptmu1 = mu1->Pt();
       rapiditymu1 = Rapidity(emu1,pzmu1);
     } else if(strcmp(fkAnalysisType,"AOD")==0){
-      AliAODTrack *mu1 = aod->GetTrack(j);
+      AliAODTrack *mu1 = dynamic_cast<AliAODTrack*>(aod->GetTrack(j));
+      if(!mu1) AliFatal("Not a standard AOD");
       if(!mu1->IsMuonTrack()) continue;
       chargemu1 = mu1->Charge();
       pxmu1 = mu1->Px();
@@ -256,7 +257,8 @@ void AliAnalysisTaskMuonDistributions::UserExec(Option_t *)
           pzmu2 = mu2->Pz();
          emu2 = mu2->E();
         } else if(strcmp(fkAnalysisType,"AOD")==0){
-          AliAODTrack *mu2 = aod->GetTrack(jj);
+          AliAODTrack *mu2 = dynamic_cast<AliAODTrack*>(aod->GetTrack(jj));
+          if(!mu2) AliFatal("Not a standard AOD");
           if(!mu2->IsMuonTrack()) continue; 
          chargemu2 = mu2->Charge();
           pxmu2 = mu2->Px();