update from Salvatore, Ruediger for some bug fixes

[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / macros / AddTaskJetPreparation.C

diff --git a/PWGJE/EMCALJetTasks/macros/AddTaskJetPreparation.C b/PWGJE/EMCALJetTasks/macros/AddTaskJetPreparation.C
index bb64852d7bc89fde95848a769b514f7603143b0b..807e84bb722f04e6b77116d7a8deaa8f0d21885d 100644 (file)
--- a/PWGJE/EMCALJetTasks/macros/AddTaskJetPreparation.C
+++ b/PWGJE/EMCALJetTasks/macros/AddTaskJetPreparation.C
@@ -1,6 +1,6 @@
 // $Id$
 
-void AddTaskJetPreparation(
+AliAnalysisTaskSE* AddTaskJetPreparation(
   const char*    dataType           = "ESD",
   const char*    periodstr          = "LHC11h",
   const char*    usedTracks         = "PicoTracks",
@@ -13,6 +13,7 @@ void AddTaskJetPreparation(
   const Double_t etaMatch           = 0.015,
   const Double_t minPtEt            = 0.15,
   const UInt_t   pSel               = AliVEvent::kAny,
+  const Bool_t   trackclus          = kTRUE,
   const Bool_t   doHistos           = kFALSE,
   const Bool_t   makePicoTracks     = kTRUE,
   const Bool_t   isEmcalTrain       = kFALSE
@@ -78,7 +79,7 @@ void AddTaskJetPreparation(
   if (isEmcalTrain)
     RequestMemory(emcalClus,100*1024);
   gROOT->LoadMacro("$ALICE_ROOT/PWGJE/EMCALJetTasks/macros/AddTaskHadCorr.C"); 
-  AliHadCorrTask *hCorr = AddTaskHadCorr("EmcalTracks","EmcalClusters",outClusName,hadcorr,minPtEt,phiMatch,etaMatch,Eexcl,doHistos);
+  AliHadCorrTask *hCorr = AddTaskHadCorr("EmcalTracks","EmcalClusters",outClusName,hadcorr,minPtEt,phiMatch,etaMatch,Eexcl,trackclus,doHistos);
   hCorr->SelectCollisionCandidates(pSel);
   if (isEmcalTrain) {
     if (doHistos)
@@ -92,4 +93,7 @@ void AddTaskJetPreparation(
     AliEmcalMCTrackSelector *mcPartTask = AddTaskMCTrackSelector(usedMCParticles, kFALSE, kFALSE);
     mcPartTask->SelectCollisionCandidates(pSel);
   }
+
+  // Return one task that represents the jet preparation on LEGO trains
+  return emcalParts;
 }