PWGCF/Correlations/macros/dphicorrelations/AddTaskContMC.C

   1 using namespace AliHelperPIDNameSpace;\r
   2   \r
   3 AliAnalysisTaskContMC* AddTaskContMC(Bool_t mc=kFALSE){\r
   4   \r
   5   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();\r
   6   if (!mgr) \r
   7     {\r
   8       ::Error("AddAliAnalysisTaskContMC", "No analysis manager to connect to.");\r
   9       return NULL;\r
  10     }   \r
  11   \r
  12   // Check the analysis type using the event handlers connected to the analysis manager.\r
  13   //==============================================================================\r
  14   if (!mgr->GetInputEventHandler()) \r
  15     {\r
  16       ::Error("AddTaskITSsaTracks", "This task requires an input event handler");\r
  17       return NULL;\r
  18     }   \r
  19   \r
  20   TString type = mgr->GetInputEventHandler()->GetDataType(); // can be "ESD" or "AOD"\r
  21   if(type.Contains("ESD"))\r
  22     {\r
  23       ::Error("AddTaskITSsaTracks", "This task requires to run on AOD");\r
  24       return NULL;\r
  25     }\r
  26   \r
  27   \r
  28   AliAnalysisTaskContMC *task = new AliAnalysisTaskContMC("ContMC");\r
  29   task->SetIsMC(mc);\r
  30   mgr->AddTask(task);\r
  31   \r
  32   TString outputFileName = AliAnalysisManager::GetCommonFileName();\r
  33   cout<<"-------------- outputFileName:  "<<outputFileName<<endl;\r
  34   \r
  35   AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();      \r
  36   AliAnalysisDataContainer *coutputpt2 = mgr->CreateContainer("fOutput", TList::Class(),  AliAnalysisManager::kOutputContainer,outputFileName);\r
  37   \r
  38   //connect containers\r
  39   mgr->ConnectInput(task, 0, cinput);\r
  40   mgr->ConnectOutput(task, 1, coutputpt2);\r
  41   return task;\r
  42 }\r