hooks for PMD flow analysis

[u/mrichter/AliRoot.git] / PWG4 / macros / AddTaskThreeJets.C

AliAnalysisTaskThreeJets * AddTaskThreeJets(char *bRec = "jets",char * bGen = "jetsAODMC_UA104")\r
{\r
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();\r
  if (!mgr) {\r
    ::Error("AddTaskThreeJets", "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
    ::Error("AddTaskThreeJets", "This task requires an input event handler");\r
    return NULL;\r
   }\r
  \r
  // Create the task and configure it.\r
  //===========================================================================\r
  \r
  AliAnalysisTaskThreeJets * threeJets = new  AliAnalysisTaskThreeJets("Three Jet Analysis");\r
  \r
  threeJets->SetBranchRec(bRec);\r
  threeJets->SetBranchGen(bGen); \r
  //  threeJets->SetDebugLevel(10);\r
  threeJets->SetR(.5); \r
  threeJets->SelectCollisionCandidates();\r
  //  threeJets->SetUseMC(kFALSE); // explicitly switch of use of MC/search for MC Jets\r
\r
  \r
  TString type = mgr->GetInputEventHandler()->GetDataType();\r
  if(type == "AOD"){\r
    threeJets->SetAODInput(kTRUE);\r
  }\r
  \r
\r
\r
  mgr->AddTask(threeJets);\r
   \r
  \r
\r
\r
      \r
  // Create ONLY the output containers for the data produced by the task.\r
  // Get and connect other common input/output containers via the manager as below\r
  //==============================================================================\r
  AliAnalysisDataContainer *coutput1_Corr = mgr->CreateContainer(Form("threeJets_%s_%s",bRec,bGen), TList::Class(),AliAnalysisManager::kOutputContainer,Form("%s:PWG4_threeJets_%s_%s",AliAnalysisManager::GetCommonFileName(),bRec,bGen));\r
\r
  mgr->ConnectInput  (threeJets, 0, mgr->GetCommonInputContainer());\r
  mgr->ConnectOutput (threeJets, 0, mgr->GetCommonOutputContainer());\r
  mgr->ConnectOutput (threeJets,  1, coutput1_Corr );\r
  \r
  return threeJets;\r
}\r
\r
\r
AliAnalysisTaskThreeJets * AddTaskJetCorrections(AliAnalysisManager* mgr,AliAnalysisDataContainer *cinput)\r
{\r
  if (!mgr) {\r
    ::Error("AddTaskJetSpectrum", "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
    ::Error("AddTaskJetSpectrum", "This task requires an input event handler");\r
    return NULL;\r
   }\r
  \r
  AliAnalysisTaskThreeJets * threeJets = new  AliAnalysisTaskThreeJets("ThreeJetAnalysis");\r
  \r
  threeJets->SetBranchGen("jetsMC"); \r
  threeJets->SetBranchRec("jets");\r
  threeJets->SetR(.5); \r
  mgr->AddTask(threeJets);\r
\r
  AliAnalysisDataContainer * coutpu0 = mgr->CreateContainer("coutpu0", TTree::Class(),\r
				  AliAnalysisManager::kExchangeContainer);\r
  AliAnalysisDataContainer *coutput1_threeJets = mgr->CreateContainer("threeJets", TList::Class(),AliAnalysisManager::kOutputContainer,"threeJets.root");\r
\r
   mgr->ConnectInput  (threeJets, 0, cinput);\r
   mgr->ConnectOutput (threeJets, 0, coutpu0);\r
   mgr->ConnectOutput (threeJets,  1, coutput1_Corr );\r
   \r
   return threeJets;\r
}\r
\r