Merge branch 'master' into TPCdev

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

diff --git a/PWGJE/EMCALJetTasks/macros/AddTaskEmcalJetMass.C b/PWGJE/EMCALJetTasks/macros/AddTaskEmcalJetMass.C
index 43507cc25e594439e36cab2a3f51b0e8840df902..8765e860316043b3505e0734cb808319be646dfc 100644 (file)
--- a/PWGJE/EMCALJetTasks/macros/AddTaskEmcalJetMass.C
+++ b/PWGJE/EMCALJetTasks/macros/AddTaskEmcalJetMass.C
@@ -8,7 +8,8 @@ AliAnalysisTaskEmcalJetMass* AddTaskEmcalJetMass(const char * njetsBase,
                                                 Int_t       pSel,
                                                 TString     trigClass      = "",
                                                 TString     kEmcalTriggers = "",
-                                                TString     tag            = "") {
+                                                TString     tag            = "",
+                                                TString     nJetsUnsub    = "") {
 
   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
   if (!mgr)
@@ -48,6 +49,15 @@ AliAnalysisTaskEmcalJetMass* AddTaskEmcalJetMass(const char * njetsBase,
     jetContBase->ConnectClusterContainer(clusterCont);
     jetContBase->SetPercAreaCut(0.6);
   }
+  if(!nJetsUnsub.IsNull()) {
+    AliJetContainer *jetContUS = task->AddJetContainer(nJetsUnsub.Data(),strType,R);
+    if(jetContUS) {
+      jetContUS->SetRhoName(nrhoBase);
+      jetContUS->ConnectParticleContainer(trackCont);
+      jetContUS->ConnectClusterContainer(clusterCont);
+      jetContUS->SetPercAreaCut(0.6);
+    }
+  }
 
   task->SetCaloTriggerPatchInfoName(kEmcalTriggers.Data());
   task->SetCentralityEstimator(CentEst);