Improved messages and protections

[u/mrichter/AliRoot.git] / PWGLF / STRANGENESS / Hypernuclei / AddTaskNucleiv2SP.C

class AliAnalysisDataContainer;

AliAnalysisTask *AddTaskNucleiv2SP(TString name="name",Int_t ptc =1,Bool_t dcacut = kTRUE, Float_t pull =3,Float_t maxVz=10){
  
  //get the current analysis manager
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  if (!mgr) {
    Error("AddTaskNucleiv2", "No analysis manager found.");
    return 0;
  }
  
  //========= Add task to the ANALYSIS manager =====
  AliAnalysisTaskNucleiv2SP *task = new AliAnalysisTaskNucleiv2SP(name);
  task-> SetIsPrimCut(dcacut); 
  task-> SetParticle (ptc);
  task-> SetMaxPull  (pull);
  task-> SetMaxVz    (maxVz);
  mgr->AddTask(task);
  
  //================================================
  //              data containers
  //================================================
  //            find input container

  AliAnalysisDataContainer *cinput   = mgr->GetCommonInputContainer();
  
  TString outputFileName = AliAnalysisManager::GetCommonFileName();
 
  AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("clisthist", TList::Class(), AliAnalysisManager::kOutputContainer, outputFileName);
  AliAnalysisDataContainer *coutput2 = mgr->CreateContainer("treeNuclei", TTree::Class(),AliAnalysisManager::kOutputContainer, outputFileName);
  //           connect containers
  mgr->ConnectInput  (task,  0, cinput );
  mgr->ConnectOutput (task,  1, coutput1);
  mgr->ConnectOutput (task,  2, coutput2);

  return task;
}