changes submitted by user saiola

[u/mrichter/AliRoot.git] / PWGGA / EMCALTasks / AliEmcalPicoTrackMaker.cxx

diff --git a/PWGGA/EMCALTasks/AliEmcalPicoTrackMaker.cxx b/PWGGA/EMCALTasks/AliEmcalPicoTrackMaker.cxx
index abd35629c1754ae188a4241f361e0d4db3c67216..b712ff36542862ad748a6b78e76f9e331839f91c 100644 (file)
--- a/PWGGA/EMCALTasks/AliEmcalPicoTrackMaker.cxx
+++ b/PWGGA/EMCALTasks/AliEmcalPicoTrackMaker.cxx
@@ -77,10 +77,16 @@ void AliEmcalPicoTrackMaker::UserExec(Option_t *)
   }
 
   // retrieve tracks from input.
-  fTracksIn = dynamic_cast<TClonesArray*>(InputEvent()->FindListObject(fTracksInName));
-  if (!fTracksIn) {
-    AliError(Form("Could not retrieve tracks %s!", fTracksInName.Data())); 
-    return;
+  if (!fTracksIn) { 
+    fTracksIn = dynamic_cast<TClonesArray*>(InputEvent()->FindListObject(fTracksInName));
+    if (!fTracksIn) {
+      AliError(Form("Could not retrieve tracks %s!", fTracksInName.Data())); 
+      return;
+    }
+    if (!fTracksIn->GetClass()->GetBaseClass("AliVParticle")) {
+      AliError(Form("%s: Collection %s does not contain AliVParticle objects!", GetName(), fTracksInName.Data())); 
+      return;
+    }
   }
 
   // add tracks to event if not yet there
@@ -100,7 +106,7 @@ void AliEmcalPicoTrackMaker::UserExec(Option_t *)
   const Int_t Ntracks = fTracksIn->GetEntriesFast();
   for (Int_t iTracks = 0, nacc = 0; iTracks < Ntracks; ++iTracks) {
 
-    AliVTrack *track = dynamic_cast<AliVTrack*>(fTracksIn->At(iTracks));
+    AliVTrack *track = static_cast<AliVTrack*>(fTracksIn->At(iTracks));
 
     if (!track)
       continue;