Always delete TObjArrays created by TString::Tokenize (Ruben)

[u/mrichter/AliRoot.git] / PWG / muon / AliAnalysisTaskFlowSingleMu.cxx

diff --git a/PWG/muon/AliAnalysisTaskFlowSingleMu.cxx b/PWG/muon/AliAnalysisTaskFlowSingleMu.cxx
index ab1674907ccb5d9c52a24c550a17936330a8dbd6..5de1f4cc42f0ce287db4fa852c5631ac9c953c0d 100644 (file)
--- a/PWG/muon/AliAnalysisTaskFlowSingleMu.cxx
+++ b/PWG/muon/AliAnalysisTaskFlowSingleMu.cxx
@@ -284,10 +284,11 @@ void AliAnalysisTaskFlowSingleMu::ProcessEvent(TString physSel, const TObjArray&
   //
   Double_t containerInput[kNvars];
   AliVParticle* track = 0x0;
-  for ( Int_t istep = 0; istep<2; ++istep ) {
-    Int_t nTracks = ( istep == kStepReconstructed ) ? AliAnalysisMuonUtility::GetNTracks(InputEvent()) : AliAnalysisMuonUtility::GetNMCTracks(InputEvent(),MCEvent());
+  Int_t nSteps = MCEvent() ? 2 : 1;
+  for ( Int_t istep = 0; istep<nSteps; ++istep ) {
+    Int_t nTracks = ( istep == kStepReconstructed ) ? AliAnalysisMuonUtility::GetNTracks(InputEvent()) : MCEvent()->GetNumberOfTracks();
     for (Int_t itrack = 0; itrack < nTracks; itrack++) {
-      track = ( istep == kStepReconstructed ) ? AliAnalysisMuonUtility::GetTrack(itrack,InputEvent()) : AliAnalysisMuonUtility::GetMCTrack(itrack,InputEvent(),MCEvent());
+      track = ( istep == kStepReconstructed ) ? AliAnalysisMuonUtility::GetTrack(itrack,InputEvent()) : MCEvent()->GetTrack(itrack);
       
       Bool_t isSelected = ( istep == kStepReconstructed ) ? fMuonTrackCuts->IsSelected(track) : ( TMath::Abs(track->PdgCode()) == 13 );
       if ( ! isSelected ) continue;
@@ -960,6 +961,7 @@ void AliAnalysisTaskFlowSingleMu::Terminate(Option_t *) {
     } // loop on track sources
   } // loop on steps
 
+  delete centralityClasses;
 
   //////////////////////
   // Event statistics //