fix const subtr tracks. Note: array has no gaps

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

diff --git a/PWGJE/EMCALJetTasks/macros/AddTaskEmcalJetMassResponse.C b/PWGJE/EMCALJetTasks/macros/AddTaskEmcalJetMassResponse.C
index 78694281c2d6c0e71632a15905f40c95d9257ff5..f97665ce097c6725716a972d1af1faf1804da15e 100644 (file)
--- a/PWGJE/EMCALJetTasks/macros/AddTaskEmcalJetMassResponse.C
+++ b/PWGJE/EMCALJetTasks/macros/AddTaskEmcalJetMassResponse.C
@@ -1,6 +1,7 @@
 AliAnalysisTaskEmcalJetMassResponse* AddTaskEmcalJetMassResponse(const char * njetsBase,
                                                                 const Double_t R,
                                                                 const char * nrhoBase, 
+                                                                const char * nrhoMass, 
                                                                 const char * ntracks, 
                                                                 const char * nclusters,
                                                                 TF1        * fBkgMass,
@@ -45,6 +46,7 @@ AliAnalysisTaskEmcalJetMassResponse* AddTaskEmcalJetMassResponse(const char * nj
   AliJetContainer *jetContBase = task->AddJetContainer(njetsBase,strType,R);
   if(jetContBase) {
     jetContBase->SetRhoName(nrhoBase);
+    jetContBase->SetRhoMassName(nrhoMass);
     jetContBase->ConnectParticleContainer(trackCont);
     jetContBase->ConnectClusterContainer(clusterCont);
     jetContBase->SetPercAreaCut(0.6);
@@ -66,6 +68,8 @@ AliAnalysisTaskEmcalJetMassResponse* AddTaskEmcalJetMassResponse(const char * nj
   TString outputfile = Form("%s",AliAnalysisManager::GetCommonFileName());
   AliAnalysisDataContainer *coutput1 = mgr->CreateContainer(contName.Data(), TList::Class(),AliAnalysisManager::kOutputContainer,outputfile);
   mgr->ConnectOutput(task,1,coutput1);
+  AliAnalysisDataContainer *coutput2 = mgr->CreateContainer(Form("%sTree",contName.Data()), TTree::Class(),AliAnalysisManager::kOutputContainer,outputfile);
+  mgr->ConnectOutput(task,2,coutput2);
 
   return task;