Updates from Redmer

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

// $Id$

AliAnalysisTaskRhoFlow* AddTaskRhoFlow(
   const char    *nJets       = "Jets",
   const char    *nTracks     = "PicoTracks",
   const char    *nClusters   = "CaloClusters",  
   const char    *nRho        = "Rho",
   Double_t       jetradius   = 0.2,
   UInt_t         type        = AliAnalysisTaskEmcal::kTPC,
   Double_t       jetareacut  = 0.01,
   Double_t       emcareacut  = 0,
   const UInt_t   exclJets    = 0,
   const char    *taskname    = "RhoFlow"
)
{  
  // Get the pointer to the existing analysis manager via the static access method.
  //==============================================================================
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  if (!mgr)
  {
    ::Error("AddTaskRho", "No analysis manager to connect to.");
    return NULL;
  }  
  
  // Check the analysis type using the event handlers connected to the analysis manager.
  //==============================================================================
  if (!mgr->GetInputEventHandler())
  {
    ::Error("AddTaskRho", "This task requires an input event handler");
    return NULL;
  }
  
  //-------------------------------------------------------
  // Init the task and do settings
  //-------------------------------------------------------

  TString name(Form("%s_%s_", taskname, nJets));
  if (type == AliAnalysisTaskEmcal::kTPC) 
    name += "TPC";
  else if (type == AliAnalysisTaskEmcal::kEMCAL) 
    name += "EMCAL";
  else if (type == AliAnalysisTaskEmcal::kUser) 
    name += "USER";
  AliAnalysisTaskRhoFlow *rhotask = new AliAnalysisTaskRhoFlow(name);
  rhotask->SetAnaType(type);
  rhotask->SetJetsName(nJets);
  rhotask->SetTracksName(nTracks);
  rhotask->SetClusName(nClusters);
  rhotask->SetRhoName(nRho);
  rhotask->SetJetAreaCut(jetareacut);
  rhotask->SetAreaEmcCut(emcareacut);
  rhotask->SetJetPtCut(0);
  rhotask->SetJetRadius(jetradius);
  rhotask->SetExcludeLeadJets(exclJets);

  //-------------------------------------------------------
  // Final settings, pass to manager and set the containers
  //-------------------------------------------------------

  mgr->AddTask(rhotask);

  // Create containers for input/output
  mgr->ConnectInput(rhotask, 0, mgr->GetCommonInputContainer());

  TString contname(name);
  contname += "_histos";
  AliAnalysisDataContainer *coutput1 = mgr->CreateContainer(contname.Data(), 
							    TList::Class(),AliAnalysisManager::kOutputContainer,
							    Form("%s", AliAnalysisManager::GetCommonFileName()));
  mgr->ConnectOutput(rhotask, 1, coutput1);

  return rhotask;
}