run on AOD branches

[u/mrichter/AliRoot.git] / PWG4 / JetTasks / AliAnalysisTaskJetsTM.cxx

diff --git a/PWG4/JetTasks/AliAnalysisTaskJetsTM.cxx b/PWG4/JetTasks/AliAnalysisTaskJetsTM.cxx
index e0cca5438d3e80466a00e65e871318490f3000c9..cc564cce76fe22c65081a7ae83765106b9a94202 100644 (file)
--- a/PWG4/JetTasks/AliAnalysisTaskJetsTM.cxx
+++ b/PWG4/JetTasks/AliAnalysisTaskJetsTM.cxx
@@ -81,7 +81,18 @@ void AliAnalysisTaskJetsTM::UserExec(Option_t *)
   }
 
   AliAODEvent* aodE  = dynamic_cast<AliAODEvent*>  (fInputEvent);
+
+  if (!aodE) {
+    Printf("ERROR: AOD not available");
+    return;
+  }
+
   TClonesArray* jets = dynamic_cast<TClonesArray*> (aodE->FindListObject("jetsAOD_FASTKT04"));
+  if (!jets) {
+    Printf("ERROR: Jet branch not available");
+    return;
+  }
+
   Int_t nJ = jets->GetEntries();
 
   Float_t ptmax = 0.;
@@ -92,6 +103,7 @@ void AliAnalysisTaskJetsTM::UserExec(Option_t *)
 //
   for (Int_t i = 0; i < nJ; i++) {
       AliAODJet* jet = dynamic_cast<AliAODJet*> (jets->At(i));
+      if (!jet) continue;
       Float_t ptJ  = jet->Pt();
       Float_t etaJ = TMath::Abs(jet->Eta());
       if ((ptJ > 20.) && (ptJ  > ptmax) && etaJ < 0.5) {