Completed changes needed because of previous commit

[u/mrichter/AliRoot.git] / PWG / FLOW / Tasks / AliFlowEventSimpleMaker.cxx

diff --git a/PWG/FLOW/Tasks/AliFlowEventSimpleMaker.cxx b/PWG/FLOW/Tasks/AliFlowEventSimpleMaker.cxx
index e218a0b27918b6a0b468f979fd4579b2a0890e17..651f7de5cf56d81d68382cb882b3a7cea6388d05 100644 (file)
--- a/PWG/FLOW/Tasks/AliFlowEventSimpleMaker.cxx
+++ b/PWG/FLOW/Tasks/AliFlowEventSimpleMaker.cxx
@@ -35,6 +35,7 @@
 #include "AliAODTrack.h"
 #include "AliCFManager.h"
 #include "AliFlowTrackSimpleCuts.h"
+#include "assert.h"
 
 using std::endl;
 using std::cout;
@@ -270,7 +271,8 @@ AliFlowEventSimple* AliFlowEventSimpleMaker::FillTracks(AliAODEvent* anInput,  c
 
     //loop over tracks
     while (iGoodTracks < iN && itrkN < iNumberOfInputTracks) {
-      AliAODTrack* pParticle = anInput->GetTrack(itrkN);   //get input particle
+      AliAODTrack* pParticle = dynamic_cast<AliAODTrack*>(anInput->GetTrack(itrkN));
+      assert((pParticle)&&"Not a standard AOD");   //get input particle
       //make new AliFlowTrackSimple
       AliFlowTrackSimple* pTrack = new AliFlowTrackSimple();
       pTrack->SetPt(pParticle->Pt() );
@@ -718,7 +720,8 @@ AliFlowEventSimple* AliFlowEventSimpleMaker::FillTracks(AliAODEvent* anInput)
   
   //normal loop
   while (iGoodTracks < iN && itrkN < iNumberOfInputTracks) {
-    AliAODTrack* pParticle = anInput->GetTrack(itrkN);   //get input particle
+    AliAODTrack* pParticle = dynamic_cast<AliAODTrack*>(anInput->GetTrack(itrkN));
+    assert((pParticle)&&"Not a standard AOD");   //get input particle
     //cut on tracks
     if (TMath::Abs(pParticle->Eta()) < 0.9)
       {