From 5a502471a1bf7d5487a09eef057fb6a70c29a879 Mon Sep 17 00:00:00 2001 From: ekryshen Date: Wed, 12 Mar 2014 14:45:02 +0100 Subject: [PATCH] Protection against missing MC header --- PWGCF/Correlations/Base/AliAnalysisTaskCFTree.cxx | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/PWGCF/Correlations/Base/AliAnalysisTaskCFTree.cxx b/PWGCF/Correlations/Base/AliAnalysisTaskCFTree.cxx index c38105e6e76..10ce2365acd 100644 --- a/PWGCF/Correlations/Base/AliAnalysisTaskCFTree.cxx +++ b/PWGCF/Correlations/Base/AliAnalysisTaskCFTree.cxx @@ -196,8 +196,12 @@ void AliAnalysisTaskCFTree::Exec(Option_t *){ fZvtx = mcEvent->GetPrimaryVertex()->GetZ(); } else if (mcTracks) { AliAODMCHeader* mcHeader = (AliAODMCHeader*) event->FindListObject(AliAODMCHeader::StdBranchName()); - nProduced = mcTracks->GetEntriesFast(); - nPrimGen = mcHeader->GetCocktailHeaders() ? mcHeader->GetCocktailHeader(0)->NProduced() : nProduced; + if (!mcHeader) { printf("AliAODMCHeader not found\n"); return; } + if (mcHeader->GetCocktailHeaders()) { + AliGenEventHeader* header0 = mcHeader->GetCocktailHeader(0); + if (!header0) { printf("first header expected but not found\n"); return; } + nPrimGen = header0->NProduced(); + } else nPrimGen = nProduced; fZvtx = mcHeader->GetVtxZ(); } else return; fEventStatistics->Fill("after mc header check",1); -- 2.43.0