New AliAnalysisTaskSE for vertexing HF (Andrea)

[u/mrichter/AliRoot.git] / PWG3 / AliAnalysisTaskSEVertexingHFTest.C

void AliAnalysisTaskSEVertexingHFTest()\r
{\r
  //\r
  // Test macro for the AliAnalysisTaskSE for heavy-flavour vertexing\r
  // A.Dainese, andrea.dainese@lnl.infn.it\r
  //\r
\r
  gSystem->Load("libTree.so");\r
  gSystem->Load("libGeom.so");\r
  gSystem->Load("libPhysics.so");\r
  gSystem->Load("libVMC.so");\r
  gSystem->Load("libSTEERBase.so");\r
  gSystem->Load("libESD.so");\r
  gSystem->Load("libAOD.so"); \r
  gSystem->Load("libANALYSIS.so");\r
  gSystem->Load("libANALYSISalice.so");\r
  gSystem->Load("libPWG3base.so");\r
\r
\r
  // Local files \r
  TChain *chain = new TChain("esdTree");\r
  chain->Add("./AliESDs.root");\r
\r
  // or:\r
  /*\r
  //Fetch files with AliEn :\r
  const char *collectionfile = "CollectionTags.xml";\r
  TGrid::Connect("alien://") ;\r
  //Create an AliRunTagCuts and an AliEventTagCuts Object and impose some selection criteria\r
  AliRunTagCuts      *runCuts   = new AliRunTagCuts();\r
  AliEventTagCuts    *eventCuts = new AliEventTagCuts();\r
  AliLHCTagCuts      *lhcCuts   = new AliLHCTagCuts();\r
  AliDetectorTagCuts *detCuts   = new AliDetectorTagCuts();\r
  eventCuts->SetMultiplicityRange(0,20000);\r
  //Create an AliTagAnalysis Object and chain the tags\r
  AliTagAnalysis   *tagAna = new AliTagAnalysis();\r
  tagAna->SetType("ESD");\r
  TAlienCollection *coll   = TAlienCollection::Open(collectionfile);\r
  TGridResult      *tagResult = coll->GetGridResult("",0,0);\r
  tagResult->Print();\r
  tagAna->ChainGridTags(tagResult);\r
  //Create a new esd chain and assign the chain that is returned by querying the tags\r
  TChain* chain = tagAna->QueryTags(runCuts,lhcCuts,detCuts,eventCuts);\r
  */\r
\r
\r
  // Create the analysis manager\r
  AliAnalysisManager *mgr  = new AliAnalysisManager("My Manager","My Manager");\r
  mgr->SetDebugLevel(10);\r
  \r
  // Input Handler\r
  //AliAODInputHandler *inputHandler = new AliAODInputHandler();\r
  AliESDInputHandler *inputHandler = new AliESDInputHandler();\r
  inputHandler->SetInactiveBranches("FMD CaloCluster");\r
  mgr->SetInputEventHandler(inputHandler);\r
  \r
  // Output \r
  AliAODHandler *aodHandler = new AliAODHandler();\r
  aodHandler->SetOutputFileName("AliAOD.VertexingHF.root");\r
  aodHandler->SetCreateNonStandardAOD();\r
  mgr->SetOutputEventHandler(aodHandler);\r
  \r
  // Vertexing analysis task    \r
  AliAnalysisTaskSEVertexingHF *hfTask = new AliAnalysisTaskSEVertexingHF("VertexingHFAnalysis");\r
  hfTask->SetDebugLevel(2);\r
  \r
  mgr->AddTask(hfTask);\r
  \r
  //\r
  // Create containers for input/output\r
  AliAnalysisDataContainer *cinput1 = mgr->CreateContainer("cchain",TChain::Class(), \r
							   AliAnalysisManager::kInputContainer);\r
  AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("tree", TTree::Class(),\r
							    AliAnalysisManager::kOutputContainer, \r
							    "default");\r
  mgr->ConnectInput(hfTask,0,cinput1);\r
  mgr->ConnectOutput(hfTask,0,coutput1);\r
\r
  //\r
  // Run the analysis\r
  //    \r
  printf("CHAIN HAS %d ENTRIES\n",(Int_t)chain->GetEntries());\r
  if(mgr->InitAnalysis()) {\r
    mgr->PrintStatus();\r
    mgr->StartAnalysis("local",chain);\r
    //mgr->StartAnalysis("grid",chain);\r
  }\r
\r
  return;\r
}\r