New structure of PWG3: PWG3base, PWG3muon, PWG3vertexingHF and PWG3vertexingOld ...

[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliAnalysisTaskSESelectHFTest.C

void AliAnalysisTaskSESelectHFTest()\r
{\r
  //\r
  // Test macro for the AliAnalysisTaskSE for heavy-flavour selection\r
  // and creation of a stand-alone AOD\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("aodTree");\r
  chain->Add("./AliAOD.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("AOD");\r
  TAlienCollection *coll   = TAlienCollection::Open(collectionfile);\r
  TGridResult      *tagResult = coll->GetGridResult("",0,0);\r
  tagResult->Print();\r
  tagAna->ChainGridTags(tagResult);\r
  //Create a new aod 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
\r
  // Input\r
  AliAODInputHandler *inputHandler = new AliAODInputHandler();\r
  inputHandler->AddFriend("AliAOD.VertexingHF.root");\r
  mgr->SetInputEventHandler(inputHandler);\r
\r
  // Output \r
  AliAODHandler *aodHandler   = new AliAODHandler();\r
  aodHandler->SetOutputFileName("AliAOD.VertexingHF.sa.root");\r
  aodHandler->SetCreateNonStandardAOD();\r
  mgr->SetOutputEventHandler(aodHandler);\r
\r
  \r
  // Aanalysis task    \r
  AliAnalysisTaskSESelectHF *hfTask = new AliAnalysisTaskSESelectHF("SelectHFAnalysis");\r
  hfTask->SetDebugLevel(2);\r
  Double_t cutsD0[9]=\r
    // cutsD0[0] = inv. mass half width [GeV]\r
    // cutsD0[1] = dca [cm]\r
    // cutsD0[2] = cosThetaStar\r
    // cutsD0[3] = pTK [GeV/c]\r
    // cutsD0[4] = pTPi [GeV/c]\r
    // cutsD0[5] = d0K [cm]   upper limit!\r
    // cutsD0[6] = d0Pi [cm]  upper limit!\r
    // cutsD0[7] = d0d0 [cm^2]\r
    // cutsD0[8] = cosThetaPoint\r
                     {1,\r
		      100000.,\r
		      1.1,\r
		      0.,\r
		      0.,\r
		      100000.,\r
		      100000.,\r
		      100000000.,\r
		      -0.9}; \r
  hfTask->SetD0toKpiCuts(cutsD0);\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