small fix in Add Task

[u/mrichter/AliRoot.git] / PWGCF / Correlations / macros / dphicorrelations / AddTaskContMC.C

using namespace AliHelperPIDNameSpace;\r
  \r
AliAnalysisTaskContMC* AddTaskContMC(Bool_t mc=kFALSE){\r
  \r
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();\r
  if (!mgr) \r
    {\r
      ::Error("AddAliAnalysisTaskContMC", "No analysis manager to connect to.");\r
      return NULL;\r
    }   \r
  \r
  // Check the analysis type using the event handlers connected to the analysis manager.\r
  //==============================================================================\r
  if (!mgr->GetInputEventHandler()) \r
    {\r
      ::Error("AddTaskITSsaTracks", "This task requires an input event handler");\r
      return NULL;\r
    }   \r
  \r
  TString type = mgr->GetInputEventHandler()->GetDataType(); // can be "ESD" or "AOD"\r
  if(type.Contains("ESD"))\r
    {\r
      ::Error("AddTaskITSsaTracks", "This task requires to run on AOD");\r
      return NULL;\r
    }\r
  \r
  \r
  AliAnalysisTaskContMC *task = new AliAnalysisTaskContMC("ContMC");\r
  task->SetIsMC(mc);\r
  mgr->AddTask(task);\r
  \r
  TString outputFileName = AliAnalysisManager::GetCommonFileName();\r
  cout<<"-------------- outputFileName:  "<<outputFileName<<endl;\r
  \r
  AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();      \r
  AliAnalysisDataContainer *coutputpt2 = mgr->CreateContainer("fOutput", TList::Class(),  AliAnalysisManager::kOutputContainer,outputFileName);\r
  \r
  //connect containers\r
  mgr->ConnectInput(task, 0, cinput);\r
  mgr->ConnectOutput(task, 1, coutputpt2);\r
  return task;\r
}\r