New way to set input and output containers for AliAnalysisManager

[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / 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
  Bool_t inputAOD=kTRUE; // otherwise, ESD\r
  TString mode="local"; // otherwise, "grid" \r
  Bool_t useParFiles=kFALSE;\r
\r
  gROOT->LoadMacro("$ALICE_ROOT/PWG3/vertexingHF/LoadLibraries.C");\r
  LoadLibraries(useParFiles);\r
\r
  TChain *chain = 0;\r
\r
  if(mode=="local") {\r
    // Local files \r
    TString treeName,fileName;\r
    if(inputAOD) {\r
      treeName="aodTree"; \r
      fileName="AliAOD.root";\r
    } else {\r
      treeName="esdTree"; \r
      fileName="AliESDs.root";\r
    }\r
    chain = new TChain(treeName.Data());\r
    chain->Add(fileName.Data());\r
\r
  } else if (mode=="grid") {\r
    //Fetch files with AliEn :\r
    const char *collectionfile = "Collection.xml";\r
    TGrid::Connect("alien://") ;\r
    TAlienCollection *coll   = TAlienCollection::Open(collectionfile);\r
    if(inputAOD) { // input AOD\r
      chain = new TChain("aodTree");\r
      while(coll->Next()) chain->Add(coll->GetTURL(""));\r
    } else { // input ESD\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
      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
      chain = tagAna->QueryTags(runCuts,lhcCuts,detCuts,eventCuts);\r
    }\r
  } else {\r
    printf("ERROR: mode has to be \"local\" or \"grid\" \n");\r
    return;\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
  AliInputEventHandler *inputHandler = 0;\r
  if(inputAOD) {\r
    inputHandler = new AliAODInputHandler();\r
  } else {\r
    inputHandler = new AliESDInputHandler();\r
  }\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
  mgr->ConnectInput(hfTask,0,mgr->GetCommonInputContainer());\r
  mgr->ConnectOutput(hfTask,0,mgr->GetCommonOutputContainer());\r
  /* \r
  // before v4-17-Release\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
  //\r
  // Run the analysis\r
  //    \r
  printf("CHAIN HAS %d ENTRIES\n",(Int_t)chain->GetEntries());\r
  if(!mgr->InitAnalysis()) return;\r
\r
  mgr->PrintStatus();\r
\r
  mgr->StartAnalysis(mode.Data(),chain);\r
\r
  return;\r
}\r