PWG3/vertexingHF/macros/AddTaskSelectHF4Prong.C

   1 AliAnalysisTaskSESelectHF4Prong *AddTaskSelectHF4Prong()\r
   2 {\r
   3   //\r
   4   // Test macro for the AliAnalysisTaskSE for heavy-flavour selection\r
   5   // and creation of a stand-alone AOD\r
   6   // A.Dainese, andrea.dainese@lnl.infn.it
   7   // F.Colamaria, fabio.colamaria@ba.infn.it\r
   8   //\r
   9 \r
  10   // Get the pointer to the existing analysis manager via the static access method.\r
  11   //==============================================================================\r
  12   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();\r
  13   if (!mgr) {\r
  14     ::Error("AddTaskSelectHF4Prong", "No analysis manager to connect to.");\r
  15     return NULL;\r
  16   }   \r
  17 \r
  18   // Output \r
  19   AliAODHandler *aodHandler   = new AliAODHandler();\r
  20   aodHandler->SetOutputFileName("AliAOD.VertexingHF.sa.root");\r
  21   aodHandler->SetCreateNonStandardAOD();\r
  22   mgr->SetOutputEventHandler(aodHandler);\r
  23
  24   //Cuts loading
  25   TFile* filecuts=new TFile("Charm4ProngCutsDef.root");
  26   if(!filecuts->IsOpen()){
  27     cout<<"Input file not found: exit"<<endl;
  28     return;
  29   }
  30
  31   AliRDHFCutsD0toKpipipi* RDHFCharm4Prong=new AliRDHFCutsD0toKpipipi();
  32   RDHFCharm4Prong = (AliRDHFCutsD0toKpipipi*)filecuts->Get("Charm4ProngCuts");
  33   RDHFCharm4Prong->SetName(Form("Charm4ProngCuts%d",1));
  34
  35   if(!RDHFCharm4Prong){
  36     cout<<"Specific AliRDHFCuts not found"<<endl;
  37     return;
  38   }
  39
  40   // Analysis task    \r
  41   AliAnalysisTaskSESelectHF4Prong *hfTask = new AliAnalysisTaskSESelectHF4Prong("SelectHFAnalysis",RDHFCharm4Prong);
  42   hfTask->SetDebugLevel(2);\r
  43   mgr->AddTask(hfTask);
  44 \r
  45   // Create containers for input/output\r  mgr->ConnectInput(hfTask,0,mgr->GetCommonInputContainer());
  46
  47   AliAnalysisDataContainer *contHist = mgr->CreateContainer("histos_bin1",TList::Class(),AliAnalysisManager::kOutputContainer,"HistMassInvAndCuts.root");
  48   AliAnalysisDataContainer *contHist2 = mgr->CreateContainer("histos_bin2",TList::Class(),AliAnalysisManager::kOutputContainer,"HistMassInvAndCuts.root");
  49   AliAnalysisDataContainer *contHist3 = mgr->CreateContainer("histos_bin3",TList::Class(),AliAnalysisManager::kOutputContainer,"HistMassInvAndCuts.root");
  50   AliAnalysisDataContainer *contHist4 = mgr->CreateContainer("histos_bin4",TList::Class(),AliAnalysisManager::kOutputContainer,"HistMassInvAndCuts.root");
  51   AliAnalysisDataContainer *contHist5 = mgr->CreateContainer("histos_bin5",TList::Class(),AliAnalysisManager::kOutputContainer,"HistMassInvAndCuts.root");
  52   AliAnalysisDataContainer *contHistCuts = mgr->CreateContainer("histoscuts",TList::Class(),AliAnalysisManager::kOutputContainer,"HistMassInvAndCuts.root");
  53
  54   mgr->ConnectOutput(hfTask,0,mgr->GetCommonOutputContainer());\r
  55   mgr->ConnectOutput(hfTask,1,contHist);
  56   mgr->ConnectOutput(hfTask,2,contHist2);
  57   mgr->ConnectOutput(hfTask,3,contHist3);
  58   mgr->ConnectOutput(hfTask,4,contHist4);
  59   mgr->ConnectOutput(hfTask,5,contHist5);
  60   mgr->ConnectOutput(hfTask,6,contHistCuts);
  61
  62   return hfTask;\r
  63 }\r