Adding cut on pythia process type

[u/mrichter/AliRoot.git] / PWGJE / AliPWG4HighPtTrackQA.cxx

diff --git a/PWGJE/AliPWG4HighPtTrackQA.cxx b/PWGJE/AliPWG4HighPtTrackQA.cxx
index 2a99def4b6e558487d8dfc660ff146ba9802adec..a077014cf9acd7b9fa6ba96c3abff5d94c68c564 100644 (file)
--- a/PWGJE/AliPWG4HighPtTrackQA.cxx
+++ b/PWGJE/AliPWG4HighPtTrackQA.cxx
@@ -842,8 +842,8 @@ Bool_t AliPWG4HighPtTrackQA::SelectEvent()
        }
       }
       else if(fDataType==kAOD) {
-       if(dynamic_cast<AliAODEvent*>(fEvent)->GetHeader()->GetCentrality())
-         cent = dynamic_cast<AliAODEvent*>(fEvent)->GetHeader()->GetCentrality();
+       if(((AliVAODHeader*)dynamic_cast<AliAODEvent*>(fEvent)->GetHeader())->GetCentrality())
+         cent = ((AliVAODHeader*)dynamic_cast<AliAODEvent*>(fEvent)->GetHeader())->GetCentrality();
        }
       if(cent>90.) {
        fNEventReject->Fill("cent>90",1);
@@ -901,7 +901,7 @@ Int_t AliPWG4HighPtTrackQA::CalculateCentrality(const AliAODEvent *aod)
   //
 
   if(!aod) return 5;
-  Float_t cent = aod->GetHeader()->GetCentrality();
+  Float_t cent = ((AliVAODHeader*)aod->GetHeader())->GetCentrality();
   if(fDebug>3) printf("centrality: %f\n",cent);
 
   return GetCentralityClass(cent);
@@ -1253,7 +1253,8 @@ void AliPWG4HighPtTrackQA::DoAnalysisAOD()
   AliExternalTrackParam exParam;
   for (Int_t iTrack = 0; iTrack < fEvent->GetNumberOfTracks(); iTrack++) {
 
-    AliAODTrack *aodtrack = aod->GetTrack(iTrack);
+    AliAODTrack *aodtrack = dynamic_cast<AliAODTrack*>(aod->GetTrack(iTrack));
+    if(!aodtrack) AliFatal("Not a standard AOD");
     if( !aodtrack->TestFilterMask(fFilterMask) ) {
       fh1NTracksReject->Fill("noHybridTrack",1);
       continue;